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"

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

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;
}

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;
}

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;
}

Variable Documentation

const unsigned int kNEvents = 10000

Definition at line 32 of file gtestFluxSimple.cxx.