gtestFluxAstro.cxx File Reference

#include <TFile.h>
#include <TNtuple.h>
#include "Framework/Conventions/Constants.h"
#include "Framework/Conventions/Units.h"
#include "Tools/Flux/GAstroFlux.h"
#include "Framework/Messenger/Messenger.h"
#include "Framework/ParticleData/PDGCodes.h"

Include dependency graph for gtestFluxAstro.cxx:

Go to the source code of this file.

Functions
int	main (int, char **)

Variables
const unsigned int	kNEvents = 1000000

Function Documentation

int main	(	int	,
		char **
	)

Definition at line 34 of file gtestFluxAstro.cxx.

References genie::flux::GAstroFlux::ForceMaxEnergy(), genie::flux::GAstroFlux::ForceMinEnergy(), genie::flux::GAstroFlux::GenerateNext(), genie::flux::GAstroFlux::GenerateWeighted(), genie::units::km, kNEvents, genie::constants::kPi, LOG, genie::flux::GAstroFlux::Momentum(), genie::flux::GAstroFlux::PdgCode(), pINFO, genie::flux::GAstroFlux::Position(), genie::flux::GAstroFlux::SetDetectorPosition(), genie::flux::GAstroFlux::SetEnergyPowLawIdx(), genie::flux::GAstroFlux::SetRelNuPopulations(), and genie::flux::GAstroFlux::Weight().

{
  const double pi = constants::kPi;

  const double latitude  = pi/5;
  const double longitude = pi/4;
  const double depth     = 3.0*units::km;
  const double size      = 2.0*units::km;

  GDiffuseAstroFlux * difflx = new GDiffuseAstroFlux;

  difflx->ForceMinEnergy(1E+2);
  difflx->ForceMaxEnergy(1E+8);
  difflx->GenerateWeighted(true);
  difflx->SetDetectorPosition(latitude,longitude,depth,size);
  difflx->SetRelNuPopulations(1,2,0,1,2,0);
  difflx->SetEnergyPowLawIdx(3.5);

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

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

  TFile f("./genie-astro-flux.root","recreate");
  fluxntp->Write();
  f.Close();

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

  return 0;
}

Variable Documentation

const unsigned int kNEvents = 1000000

Definition at line 31 of file gtestFluxAstro.cxx.

Referenced by createFluxNtuple(), and main().