gtestFluxAtmo.cxx File Reference

#include <TFile.h>
#include <TNtuple.h>
#include <TSystem.h>
#include <TH1D.h>
#include <TF1.h>
#include "Tools/Flux/GFLUKAAtmoFlux.h"
#include "Tools/Flux/GBGLRSAtmoFlux.h"
#include "Framework/Messenger/Messenger.h"
#include "Framework/ParticleData/PDGCodes.h"

Include dependency graph for gtestFluxAtmo.cxx:

Go to the source code of this file.

Functions
TNtuple *	runGFlukaAtmo3DFluxDriver (void)
TNtuple *	runGBartolAtmoFluxDriver (void)
TNtuple *	createFluxNtuple (GFluxI *flux)
int	main (int, char **)

Variables
const unsigned int	kNEvents = 100000

Function Documentation

TNtuple * createFluxNtuple

(

GFluxI *

flux

)

Definition at line 120 of file gtestFluxAtmo.cxx.

References genie::GFluxI::GenerateNext(), kNEvents, LOG, genie::GFluxI::Momentum(), genie::GFluxI::PdgCode(), pINFO, genie::GFluxI::Position(), and genie::GFluxI::Weight().

Referenced by runGBartolAtmoFluxDriver(), runGCylindTH1FluxDriver(), and runGFlukaAtmo3DFluxDriver().

{
  TNtuple * fluxntp = 
     new TNtuple("fluxntp", "flux", "x:y:z:t:px:py:pz:E:pdgc:wght");

  LOG("test", pINFO) << "Generating flux neutrinos";

  unsigned int ievent = 0;
  while(ievent++ < kNEvents) {
    LOG("test", pINFO)  << "Event number: " << ievent;
    flux->GenerateNext();
    int    pdgc = flux->PdgCode();
    double wght = flux->Weight();
    const TLorentzVector & x4 = flux->Position();
    const TLorentzVector & p4 = flux->Momentum();
    fluxntp->Fill( 
      x4.X(),  x4.Y(),  x4.Z(),  x4.T(),
      p4.Px(), p4.Py(), p4.Pz(), p4.E(), pdgc, wght);
  }
  return fluxntp;
}

int main	(	int	,
		char **
	)

Definition at line 40 of file gtestFluxAtmo.cxx.

References LOG, pINFO, runGBartolAtmoFluxDriver(), runGFlukaAtmo3DFluxDriver(), and Write().

{
  LOG("test", pINFO)  << "Running GFlukaAtmo3DFlux driver test";
  TNtuple * ntfluka = runGFlukaAtmo3DFluxDriver();
  ntfluka->SetTitle("GFlukaAtmo3DFlux driver data");

  LOG("test", pINFO)  << "Running GBartolAtmoFlux driver test";
  TNtuple * ntbartol = runGBartolAtmoFluxDriver();
  ntbartol->SetTitle("GBartolAtmoFlux driver data");

  LOG("test", pINFO) << "Saving flux ntuples";

  TFile f("./genie-flux-drivers.root","recreate");
  ntfluka  -> Write("ntfluka");
  ntbartol -> Write("ntbartol");
  f.Close();

  LOG("test", pINFO)  << "Done!";

  return 0;
}

TNtuple * runGBartolAtmoFluxDriver

(

void

)

Definition at line 93 of file gtestFluxAtmo.cxx.

References createFluxNtuple(), genie::kPdgAntiNuE, genie::kPdgAntiNuMu, genie::kPdgNuE, genie::kPdgNuMu, LOG, and pINFO.

Referenced by main().

{
  string base_dir = 
    (gSystem->Getenv("GFLUX_BGLRS3DATMO") ?
       gSystem->Getenv("GFLUX_BGLRS3DATMO") : ".");

  double Rlongitudinal  = 1000.; //m
  double Rtransverse    = 100.;  //m

  GBGLRSAtmoFlux * flux = new GBGLRSAtmoFlux;

  flux -> AddFluxFile ( kPdgNuMu,     base_dir + "/f210_3_z.kam_num" );
  flux -> AddFluxFile ( kPdgAntiNuMu, base_dir + "/f210_3_z.kam_nbm" );
  flux -> AddFluxFile ( kPdgNuE,      base_dir + "/f210_3_z.kam_nue" );
  flux -> AddFluxFile ( kPdgAntiNuE,  base_dir + "/f210_3_z.kam_nbe" );
  flux -> SetRadii(Rlongitudinal, Rtransverse);
  flux -> LoadFluxData();
  flux -> GenerateWeighted(true);
//flux -> ForceMaxEnergy(300);

  LOG("test", pINFO) << "Generating events";
  TNtuple * fluxntp = createFluxNtuple(dynamic_cast<GFluxI*>(flux));
  delete flux;

  return fluxntp;
}

TNtuple * runGFlukaAtmo3DFluxDriver

(

void

)

Definition at line 62 of file gtestFluxAtmo.cxx.

References createFluxNtuple(), genie::kPdgAntiNuE, genie::kPdgAntiNuMu, genie::kPdgNuE, genie::kPdgNuMu, LOG, and pINFO.

Referenced by main().

{
  string base_dir = 
    (gSystem->Getenv("GFLUX_FLUKA3DATMO") ?
       gSystem->Getenv("GFLUX_FLUKA3DATMO") : ".");

  double Rlongitudinal  = 1000.; //m
  double Rtransverse    = 100.;  //m

  GFLUKAAtmoFlux * flux = new GFLUKAAtmoFlux;

  LOG("test", pINFO) << base_dir + "/sdave_numu07.dat";
  LOG("test", pINFO) << base_dir + "/sdave_anumu07.dat";
  LOG("test", pINFO) << base_dir + "/sdave_nue07.dat";
  LOG("test", pINFO) << base_dir + "/sdave_anue07.dat";


  flux -> AddFluxFile ( kPdgNuMu,     base_dir + "/sdave_numu07.dat"  );
  flux -> AddFluxFile ( kPdgAntiNuMu, base_dir + "/sdave_anumu07.dat" );
  flux -> AddFluxFile ( kPdgNuE,      base_dir + "/sdave_nue07.dat"   );
  flux -> AddFluxFile ( kPdgAntiNuE,  base_dir + "/sdave_anue07.dat"  );
  flux -> SetRadii(Rlongitudinal, Rtransverse);
  flux -> LoadFluxData();
  flux -> GenerateWeighted(true);
//flux -> ForceMaxEnergy(3);

  LOG("test", pINFO) << "Generating events";
  TNtuple * fluxntp = createFluxNtuple(dynamic_cast<GFluxI*>(flux));
  return fluxntp;
}

Variable Documentation

const unsigned int kNEvents = 100000

Definition at line 33 of file gtestFluxAtmo.cxx.