gtestFluxSimple.cxx File Reference

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

Include dependency graph for gtestFluxSimple.cxx:

Go to the source code of this file.

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

Variables
const unsigned int	kNEvents = 10000

Function Documentation

createFluxNtuple()

TNtuple * createFluxNtuple

(

GFluxI *

flux

)

Definition at line 98 of file gtestFluxSimple.cxx.

{
  TNtuple * fluxntp =
      new TNtuple("fluxntp",
            "flux data", "x:y:z:t:px:py:pz:E:pdgc");
 
  LOG("test", pINFO) << "Generating flux neutrinos";
 
  unsigned int ievent = 0;
  while(ievent++ < kNEvents) {
 
    flux->GenerateNext();
 
    int pdgc = flux->PdgCode();
    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);
  }
  return fluxntp;
}

References kNEvents, LOG, and pINFO.

Referenced by runGCylindTH1FluxDriver().

main()

int main	(	int	,
		char **	)

Definition at line 38 of file gtestFluxSimple.cxx.

{
  LOG("test", pINFO)  << "Running GCylindTH1Flux driver test";
  TNtuple * ntcylh1f = runGCylindTH1FluxDriver();
  ntcylh1f->SetTitle("GCylindTH1Flux driver data");
 
  LOG("test", pINFO) << "Saving flux ntuples";
 
  TFile f("./genie-flux-drivers.root","recreate");
  ntcylh1f  -> Write("ntcylh1f");
  f.Close();
 
  delete ntcylh1f;
 
  LOG("test", pINFO)  << "Done!";
 
  return 0;
}

References LOG, pINFO, runGCylindTH1FluxDriver(), and Write().

runGCylindTH1FluxDriver()

TNtuple * runGCylindTH1FluxDriver

(

void

)

Definition at line 57 of file gtestFluxSimple.cxx.

{
  LOG("test", pINFO)  << "Creating GCylindTH1Flux flux driver";
 
  GCylindTH1Flux * flux = new GCylindTH1Flux;
 
  LOG("test", pINFO)  << "Setting configuration data";
 
  TF1 * f1 = new TF1("f1","1./x",0.5,5.0);
  TH1D * spectrum1 = new TH1D("spectrum1","numu E",    20,0.5,5);
  spectrum1->FillRandom("f1",100000);
 
  TF1 * f2 = new TF1("f2","x",0.5,5.0);
  TH1D * spectrum2 = new TH1D("spectrum2","numubar E", 20,0.5,5);
  spectrum2->FillRandom("f2",10000);
 
  TVector3 direction(0,0,1);
  TVector3 beam_spot(0,0,-10);
 
  double Rtransverse = 0.5;
 
  LOG("test", pINFO)  << "Configuring GCylindTH1Flux flux driver";
 
  flux -> SetNuDirection      (direction);
  flux -> SetBeamSpot         (beam_spot);
  flux -> SetTransverseRadius (Rtransverse);
  flux -> AddEnergySpectrum   (kPdgNuMu,     spectrum1);
  flux -> AddEnergySpectrum   (kPdgAntiNuMu, spectrum2);
 
  LOG("test", pINFO) << "Creating flux ntuple";
  GFluxI * fluxi = dynamic_cast<GFluxI*>(flux);
 
  TNtuple * fluxntp = createFluxNtuple(fluxi);
 
  delete f1;
  delete f2;
  delete flux;
 
  return fluxntp;
}

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

Referenced by main().

Variable Documentation

kNEvents

const unsigned int kNEvents = 10000

Definition at line 32 of file gtestFluxSimple.cxx.