gtestROOTGeometry.cxx File Reference

#include <string>
#include <vector>
#include <cassert>
#include <TFile.h>
#include <TNtupleD.h>
#include <TSystem.h>
#include <TLorentzVector.h>
#include <TVector3.h>
#include <TApplication.h>
#include <TPolyMarker3D.h>
#include "Framework/Conventions/Constants.h"
#include "Tools/Geometry/ROOTGeomAnalyzer.h"
#include "Framework/EventGen/PathLengthList.h"
#include "Framework/Messenger/Messenger.h"
#include "Framework/Numerical/RandomGen.h"
#include "Framework/Utils/StringUtils.h"
#include "Framework/Utils/CmdLnArgParser.h"

Include dependency graph for gtestROOTGeometry.cxx:

Go to the source code of this file.

Functions
void	GetCommandLineArgs (int argc, char **argv)
void	GetRandomRay (TLorentzVector &x, TLorentzVector &p)
int	GetTargetMaterial (const PathLengthList &pl)
TVector3	kDefOptRayDirection (1, 0, 0)
TVector3	kDefOptRaySurf (0, 0, 0)
int	main (int argc, char **argv)

Variables
string	gOptGeomFile
string	gOptRootGeomTopVol
TVector3	gOptRayDirection
TVector3	gOptRaySurf
double	gOptRayR
int	gOptNVtx
int	gOptTgtPdg
double	kDefOptRayR = 100

Function Documentation

void GetCommandLineArgs	(	int	argc,
		char **	argv
	)

void GetRandomRay	(	TLorentzVector &	x,
		TLorentzVector &	p
	)

Definition at line 200 of file gtestROOTGeometry.cxx.

References gOptRayDirection, gOptRayR, gOptRaySurf, genie::RandomGen::Instance(), genie::constants::kPi, LOG, pNOTICE, and genie::RandomGen::RndFlux().

Referenced by main().

{
// generate a random ray (~flux neutrino)
//
  RandomGen * rnd = RandomGen::Instance();

  TVector3 vec0(gOptRayDirection);
  TVector3 vec = vec0.Orthogonal();

  double phi = 2.*kPi * rnd->RndFlux().Rndm();
  double Rt  = -1;
  bool accept = false;
  while(!accept) {
    double r = gOptRayR * rnd->RndFlux().Rndm();
    double y = gOptRayR * rnd->RndFlux().Rndm();
    if(y<r) {
      accept = true;
      Rt = r;
    }
  }

  vec.Rotate(phi,vec0);
  vec.SetMag(Rt);

  vec = vec + gOptRaySurf;

  TLorentzVector xx(vec, 0.);
  TLorentzVector pp(gOptRayDirection, gOptRayDirection.Mag());

  x = xx;
  p = pp;

  LOG("test", pNOTICE) 
   << "** Curr ray:";
  LOG("test", pNOTICE) 
   << "    x = " << x.X() << ",  y = " << x.Y() << ",  z = " << x.Z();
  LOG("test", pNOTICE) 
   << "    px = " << p.X() << ", py = " << p.Y() << ", pz = " << p.Z();

}

int GetTargetMaterial

(

const PathLengthList &

pl

)

Definition at line 241 of file gtestROOTGeometry.cxx.

References genie::PathLengthList::AreAllZero(), gOptTgtPdg, genie::RandomGen::Instance(), genie::PathLengthList::PathLength(), and genie::RandomGen::RndFlux().

Referenced by main().

{
  if(pl.AreAllZero()) return -1;

  if(gOptTgtPdg > 0) {
    if(pl.PathLength(gOptTgtPdg) > 0) return gOptTgtPdg;
  } 
  else {
    RandomGen * rnd = RandomGen::Instance();

    PathLengthList::const_iterator pliter;
    double sum = 0;
    for(pliter = pl.begin(); pliter != pl.end(); ++pliter) {
        sum += pliter->second;
    }
    double cpl = sum * rnd->RndFlux().Rndm();
    sum = 0;
    for(pliter = pl.begin(); pliter != pl.end(); ++pliter) {
       sum += pliter->second;
       if(cpl < sum) {
          return pliter->first;
       }
    }
  }
  return -1;
}

TVector3 kDefOptRayDirection	(	1	,
		0	,
		0
	)

TVector3 kDefOptRaySurf	(	0	,
		0	,
		0
	)

int main	(	int	argc,
		char **	argv
	)

Definition at line 106 of file gtestROOTGeometry.cxx.

References genie::geometry::ROOTGeomAnalyzer::ComputeMaxPathLengths(), genie::geometry::ROOTGeomAnalyzer::ComputePathLengths(), genie::geometry::ROOTGeomAnalyzer::GenerateVertex(), GetCommandLineArgs(), genie::geometry::ROOTGeomAnalyzer::GetGeometry(), GetRandomRay(), GetTargetMaterial(), gOptGeomFile, gOptNVtx, gOptRootGeomTopVol, genie::pdg::IonPdgCodeToA(), genie::pdg::IonPdgCodeToZ(), LOG, pERROR, pINFO, pNOTICE, and genie::geometry::ROOTGeomAnalyzer::SetTopVolName().

{
  GetCommandLineArgs(argc, argv);
 
#ifdef __GENIE_GEOM_DRIVERS_ENABLED__

  TApplication theApp("App", &argc, argv);
 
  // Create & configure the geometry driver
  //
  LOG("test", pINFO) 
     << "Creating a geometry driver for ROOT geometry at: " 
     << gOptGeomFile;
  geometry::ROOTGeomAnalyzer * geom_driver = 
               new geometry::ROOTGeomAnalyzer(gOptGeomFile);
  geom_driver ->SetTopVolName(gOptRootGeomTopVol);

  // Draw the geometry
  // & define TPolyMarker3D for drawing vertices later on
  //
  LOG("test", pINFO) 
      << "Drawing the ROOT geometry";
  geom_driver->GetGeometry()->GetTopVolume()->Draw();

  TPolyMarker3D * vtxp = new TPolyMarker3D();
  vtxp->SetMarkerColor(kRed);
  vtxp->SetMarkerStyle(8);
  vtxp->SetMarkerSize(0.4);

  // Compute & Printout the (density weighted) max path lengths
  //
  LOG("test", pINFO) 
       << "Computing max {density-weighted path lengths}";
  const PathLengthList & maxpl = geom_driver->ComputeMaxPathLengths();

  LOG("test", pINFO) << "Maximum math lengths: " << maxpl;

  TFile f("geomtest.root","recreate");
  TNtupleD vtxnt("vtxnt","","x:y:z:A:Z");

  TLorentzVector x(0,0,0,0);
  TLorentzVector p(0,0,0,0);

  int n = 0;
  while (n < gOptNVtx) {

    // generate a random ray 
    GetRandomRay(x,p);

    // compute density-weighted path lengths for each geometry
    // material for the current ray
    const PathLengthList & pl = geom_driver->ComputePathLengths(x,p);
    LOG("test",pINFO)        
       << "Current path lengths: " << pl;

    // select detector material (amongst all materials defined in the 
    // detector geometry -- do so based on density-weighted path lengths) 
    // or force it to the user-selected material
    int tpdg = GetTargetMaterial(pl);
    if (tpdg == -1) continue;
    LOG("test",pINFO) << "Selected target material: " << tpdg;

    // generate an 'interaction vertex' in the selected material
    const TVector3 & vtx = geom_driver->GenerateVertex(x,p,tpdg);
    LOG("test",pINFO) 
      << "Generated vtx: (x = " << vtx.X() 
      << ", y = " << vtx.Y() << ", z = " <<vtx.Z() << ")";

    // add it at the ntuple & at the vtx marker
    vtxnt.Fill(vtx.X(),vtx.Y(),vtx.Z(), 
               pdg::IonPdgCodeToA(tpdg), pdg::IonPdgCodeToZ(tpdg));
    vtxp->SetNextPoint(vtx.X(),vtx.Y(),vtx.Z()); 

    n++;
    LOG("test", pNOTICE) 
      << " *** Vertices generated so far: " << n;
  }
 
  // draw vertices
  vtxp->Draw("same");
  
  vtxnt.Write();
  f.Close();
  
  theApp.Run(kTRUE);

#else
    LOG("test", pERROR) 
       << "*** You should have enabled the geometry drivers first!";
#endif

  return 0;
}

Variable Documentation

string gOptGeomFile

Definition at line 93 of file gtestROOTGeometry.cxx.

Referenced by main().

int gOptNVtx

Definition at line 98 of file gtestROOTGeometry.cxx.

Referenced by main().

TVector3 gOptRayDirection

Definition at line 95 of file gtestROOTGeometry.cxx.

Referenced by GetRandomRay().

double gOptRayR

Definition at line 97 of file gtestROOTGeometry.cxx.

Referenced by GetRandomRay().

TVector3 gOptRaySurf

Definition at line 96 of file gtestROOTGeometry.cxx.

Referenced by GetRandomRay().

string gOptRootGeomTopVol

Definition at line 94 of file gtestROOTGeometry.cxx.

int gOptTgtPdg

Definition at line 99 of file gtestROOTGeometry.cxx.

Referenced by GetTargetMaterial().

double kDefOptRayR = 100

Definition at line 101 of file gtestROOTGeometry.cxx.