#include <TrackingMaterialProducer.h>

Inheritance diagram for TrackingMaterialProducer:

Public Member Functions
	TrackingMaterialProducer (const edm::ParameterSet &)

	~TrackingMaterialProducer () override

Public Member Functions inherited from SimProducer
void	registerProducts (edm::ProducerBase &iProd)

	SimProducer ()

Public Member Functions inherited from SimWatcher
	SimWatcher ()

virtual	~SimWatcher ()

Public Member Functions inherited from Observer< const BeginOfJob * >
	Observer ()

void	slotForUpdate (const BeginOfJob * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const EndOfJob * >
	Observer ()

void	slotForUpdate (const EndOfJob * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const BeginOfEvent * >
	Observer ()

void	slotForUpdate (const BeginOfEvent * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const BeginOfTrack * >
	Observer ()

void	slotForUpdate (const BeginOfTrack * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const G4Step * >
	Observer ()

void	slotForUpdate (const G4Step * iT)

virtual	~Observer ()

Public Member Functions inherited from Observer< const EndOfTrack * >
	Observer ()

void	slotForUpdate (const EndOfTrack * iT)

virtual	~Observer ()

Private Member Functions
bool	isSelected (const G4VTouchable *touch)

bool	isSelectedFast (const G4TouchableHistory *touch)

void	produce (edm::Event &, const edm::EventSetup &) override

void	update (const BeginOfJob *) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const BeginOfEvent *) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const BeginOfTrack *) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const G4Step *) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const EndOfTrack *) override
	This routine will be called when the appropriate signal arrives. More...

void	update (const EndOfJob *) override
	This routine will be called when the appropriate signal arrives. More...

Private Attributes
bool	m_primaryTracks

std::vector< std::string >	m_selectedNames

std::vector< const G4LogicalVolume * >	m_selectedVolumes

MaterialAccountingTrack	m_track

std::vector< MaterialAccountingTrack > *	m_tracks

TFile *	output_file_

TProfile *	radLen_vs_eta_

Additional Inherited Members
Protected Member Functions inherited from SimProducer
template<class T >
void	produces ()

template<class T >
void	produces (const std::string &instanceName)

Detailed Description

Definition at line 30 of file TrackingMaterialProducer.h.

Constructor & Destructor Documentation

TrackingMaterialProducer::TrackingMaterialProducer ( const edm::ParameterSet & iPSet )

Definition at line 84 of file TrackingMaterialProducer.cc.

References edm::ParameterSet::getParameter().

 {
   edm::ParameterSet config = iPSet.getParameter<edm::ParameterSet>("TrackingMaterialProducer");
   m_selectedNames       = config.getParameter< std::vector<std::string> >("SelectedVolumes");
   m_primaryTracks       = config.getParameter<bool>("PrimaryTracksOnly");
   m_tracks              = nullptr;
 
   produces< std::vector<MaterialAccountingTrack> >();
   output_file_ = new TFile("radLen_vs_eta_fromProducer.root", "RECREATE");
   output_file_->cd();
   radLen_vs_eta_ = new TProfile("radLen", "radLen", 250., -5., 5., 0, 10.);
 }

TrackingMaterialProducer::~TrackingMaterialProducer ( void )

override

Definition at line 98 of file TrackingMaterialProducer.cc.

99 {

100 }

Member Function Documentation

bool TrackingMaterialProducer::isSelected ( const G4VTouchable * touch )

private

Definition at line 292 of file TrackingMaterialProducer.cc.

References DEFINE_SIMWATCHER, mps_fire::i, and or.

 {
   for (size_t i = 0; i < m_selectedVolumes.size(); ++i)
     if (m_selectedVolumes[i]->IsAncestor( touchable->GetVolume() )
         or m_selectedVolumes[i] == touchable->GetVolume()->GetLogicalVolume())
       return true;
 
   return false;
 }

bool TrackingMaterialProducer::isSelectedFast ( const G4TouchableHistory * touch )

private

Definition at line 146 of file TrackingMaterialProducer.cc.

References edmIntegrityCheck::d, and spr::find().

                                                                                  {
   for (int d = touchable->GetHistoryDepth() -1; d >=0;  --d) {
       if (
            std::find(
                      m_selectedNames.begin(),
                      m_selectedNames.end(),
                      touchable->GetVolume(d)->GetName())
         != m_selectedNames.end())
         return true;
     }
   return false;
 }

void TrackingMaterialProducer::produce	(	edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

overrideprivatevirtual

Implements SimProducer.

Definition at line 283 of file TrackingMaterialProducer.cc.

References eostools::move(), edm::Event::put(), and l1t::tracks.

 {
   // transfer ownership to the Event
   std::unique_ptr<std::vector<MaterialAccountingTrack> > tracks( m_tracks );
   iEvent.put(std::move(tracks));
   m_tracks = nullptr;
 }

void TrackingMaterialProducer::update ( const BeginOfJob * )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfJob * >.

Definition at line 109 of file TrackingMaterialProducer.cc.

References MessageLogger_cfi::cerr, and GetVolume().

 {
   // INFO
   LogInfo("TrackingMaterialProducer") << "TrackingMaterialProducer: List of the selected volumes: " << std::endl;
   for (std::vector<std::string>::const_iterator volume_name = m_selectedNames.begin();
        volume_name != m_selectedNames.end(); ++volume_name) {
     const G4LogicalVolume* volume = GetVolume(*volume_name);
     if (volume) {
       LogInfo("TrackingMaterialProducer") << "TrackingMaterialProducer: " << *volume_name << std::endl;
       m_selectedVolumes.push_back( volume );
     } else {
       // FIXME: throw an exception ?
       std::cerr << "TrackingMaterialProducer::update(const BeginOfJob*): WARNING: selected volume \"" << *volume_name << "\" not found in geometry " << std::endl;
     }
   }
 }

void TrackingMaterialProducer::update ( const BeginOfEvent * )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfEvent * >.

Definition at line 128 of file TrackingMaterialProducer.cc.

 {
   m_tracks = new std::vector<MaterialAccountingTrack>();
 }

void TrackingMaterialProducer::update ( const BeginOfTrack * )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const BeginOfTrack * >.

Definition at line 135 of file TrackingMaterialProducer.cc.

References HiIsolationCommonParameters_cff::track.

 {
   m_track.reset();
 
   // prevent secondary tracks from propagating
   G4Track* track = const_cast<G4Track*>((*event)());
   if (m_primaryTracks and track->GetParentID() != 0) {
     track->SetTrackStatus(fStopAndKill);
   }
 }

void TrackingMaterialProducer::update ( const G4Step * )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const G4Step * >.

Definition at line 160 of file TrackingMaterialProducer.cc.

References GetSensitiveVolume(), GetTransform(), hcalDigis_cfi::level, PV3DBase< T, PVType, FrameType >::perp(), position, create_public_lumi_plots::transform, X0, and PV3DBase< T, PVType, FrameType >::z().

 {
   const G4TouchableHistory* touchable = (G4TouchableHistory*)(step->GetTrack()->GetTouchable());
   if (not isSelectedFast( touchable )) {
     LogInfo("TrackingMaterialProducer") << "TrackingMaterialProducer:\t[...] skipping "
                                          << touchable->GetVolume()->GetName() << std::endl;
     return;
   }
 
   // material and step proterties
   const G4Material* material = touchable->GetVolume()->GetLogicalVolume()->GetMaterial();
   double length = step->GetStepLength() / cm;          // mm -> cm
   double X0 = material->GetRadlen() / cm;              // mm -> cm
   double Ne = material->GetElectronDensity() * cm3;    // 1/mm3 -> 1/cm3
   double Xi = Ne / 6.0221415e23 * 0.307075 / 2;        // MeV / cm
   double radiationLengths = length / X0;               //
   double energyLoss       = length * Xi / 1000.;       // GeV
   //double energyLoss = step->GetDeltaEnergy()/MeV;  should we use this??
 
   G4ThreeVector globalPosPre  = step->GetPreStepPoint()->GetPosition();
   G4ThreeVector globalPosPost = step->GetPostStepPoint()->GetPosition();
   GlobalPoint globalPositionIn(  globalPosPre.x()  / cm, globalPosPre.y()  / cm, globalPosPre.z() / cm );    // mm -> cm
   GlobalPoint globalPositionOut( globalPosPost.x() / cm, globalPosPost.y() / cm, globalPosPost.z() / cm );   // mm -> cm
 
   // check for a sensitive detector
   bool enter_sensitive = false;
   bool leave_sensitive = false;
   double cosThetaPre  = 0.0;
   double cosThetaPost = 0.0;
   int level = 0;
   const G4VPhysicalVolume* sensitive = nullptr;
   GlobalPoint position;
   std::tie(sensitive, level) = GetSensitiveVolume(touchable);
   if (sensitive) {
     const G4VSolid &          solid     = *touchable->GetSolid( level );
     const G4AffineTransform & transform = GetTransform( touchable, level );
     G4ThreeVector pos = transform.Inverse().TransformPoint( G4ThreeVector( 0., 0., 0. ) );
     position = GlobalPoint( pos.x() / cm, pos.y() / cm, pos.z() / cm );  // mm -> cm
 
     G4ThreeVector localPosPre   = transform.TransformPoint( globalPosPre );
     EInside       statusPre     = solid.Inside( localPosPre );
     if (statusPre == kSurface) {
       enter_sensitive = true;
       G4ThreeVector globalDirPre  = step->GetPreStepPoint()->GetMomentumDirection();
       G4ThreeVector localDirPre   = transform.TransformAxis( globalDirPre );
       G4ThreeVector normalPre     = solid.SurfaceNormal( localPosPre );
       cosThetaPre  = normalPre.cosTheta( -localDirPre );
     }
 
     G4ThreeVector localPosPost  = transform.TransformPoint( globalPosPost );
     EInside       statusPost    = solid.Inside( localPosPost );
     if (statusPost == kSurface) {
       leave_sensitive = true;
       G4ThreeVector globalDirPost = step->GetPostStepPoint()->GetMomentumDirection();
       G4ThreeVector localDirPost  = transform.TransformAxis( globalDirPost );
       G4ThreeVector normalPost    = solid.SurfaceNormal( localPosPost );
       cosThetaPost = normalPost.cosTheta( localDirPost );
     }
   }
 
   // update track accounting
   if (enter_sensitive)
     m_track.enterDetector( sensitive, position, cosThetaPre );
   m_track.step(MaterialAccountingStep( length, radiationLengths, energyLoss, globalPositionIn, globalPositionOut ));
   if (leave_sensitive)
     m_track.leaveDetector( sensitive, cosThetaPost );
 
   if (sensitive)
     LogInfo("TrackingMaterialProducer") << "Track was near sensitive     volume "
                                          << sensitive->GetName() << std::endl;
   else
     LogInfo("TrackingMaterialProducer") << "Track was near non-sensitive volume "
                                          << touchable->GetVolume()->GetName() << std::endl;
   LogInfo("TrackingMaterialProducer")  << "Step length:             "
                                        << length << " cm\n"
                                        << "globalPreStep(r,z): (" << globalPositionIn.perp()
                                        << ", " << globalPositionIn.z() << ") cm\n"
                                        << "globalPostStep(r,z): (" << globalPositionOut.perp()
                                        << ", " << globalPositionOut.z() << ") cm\n"
                                        << "position(r,z): ("
                                        << position.perp()
                                        << ", " << position.z() << ") cm\n"
                                        << "Radiation lengths:       "
                                        << radiationLengths << " \t\t(X0: "
                                        << X0 << " cm)\n"
                                        << "Energy loss:             "
                                        << energyLoss << " MeV  \t(Xi: "
                                        << Xi << " MeV/cm)\n"
                                        << "Track was " << (enter_sensitive ? "entering " : "in none ")
                                        << "sensitive volume\n"
                                        << "Track was " << (leave_sensitive ? "leaving  " : "in none ")
                                        << "sensitive volume\n";
 
 }

void TrackingMaterialProducer::update ( const EndOfTrack * )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const EndOfTrack * >.

Definition at line 257 of file TrackingMaterialProducer.cc.

References HiIsolationCommonParameters_cff::track.

 {
   const G4Track * track = (*event)();
   if (m_primaryTracks and track->GetParentID() != 0)
     return;
 
   radLen_vs_eta_->Fill(track->GetMomentum().eta(), m_track.summary().radiationLengths());
   m_tracks->push_back(m_track);
 
   // LogInfo
   LogInfo("TrackingMaterialProducer") << "TrackingMaterialProducer: this track took "
                                        << m_track.steps().size()
                                        << " steps, and passed through "
                                        << m_track.detectors().size()
                                        << " sensitive detectors" << std::endl;
   LogInfo("TrackingMaterialProducer") << "TrackingMaterialProducer: track length:       "
                                        << m_track.summary().length()
                                        << " cm" << std::endl;
   LogInfo("TrackingMaterialProducer") << "TrackingMaterialProducer: radiation lengths: "
                                        << m_track.summary().radiationLengths() << std::endl;
   LogInfo("TrackingMaterialProducer") << "TrackingMaterialProducer: energy loss:        "
                                        << m_track.summary().energyLoss()
                                        << " MeV" << std::endl;
 }

void TrackingMaterialProducer::update ( const EndOfJob * )

overrideprivatevirtual

This routine will be called when the appropriate signal arrives.

Implements Observer< const EndOfJob * >.

Definition at line 103 of file TrackingMaterialProducer.cc.

 {
   radLen_vs_eta_->Write();
   output_file_->Close();
 }

Member Data Documentation

bool TrackingMaterialProducer::m_primaryTracks

private

Definition at line 55 of file TrackingMaterialProducer.h.

std::vector<std::string> TrackingMaterialProducer::m_selectedNames

private

Definition at line 56 of file TrackingMaterialProducer.h.

std::vector<const G4LogicalVolume *> TrackingMaterialProducer::m_selectedVolumes

private

Definition at line 57 of file TrackingMaterialProducer.h.

MaterialAccountingTrack TrackingMaterialProducer::m_track

private

Definition at line 58 of file TrackingMaterialProducer.h.

std::vector<MaterialAccountingTrack>* TrackingMaterialProducer::m_tracks

private

Definition at line 59 of file TrackingMaterialProducer.h.

TFile* TrackingMaterialProducer::output_file_

private

Definition at line 60 of file TrackingMaterialProducer.h.

TProfile* TrackingMaterialProducer::radLen_vs_eta_

private

Definition at line 61 of file TrackingMaterialProducer.h.

Public Member Functions

Private Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation