#include "MCRecoPart.h"

Inheritance diagram for sim::MCRecoPart:

Public Member Functions
	MCRecoPart (fhicl::ParameterSet const &pset)
	Default constructor with fhicl parameters. More...

virtual	~MCRecoPart ()
	Default destructor. More...

void	AddParticles (const std::vector< simb::MCParticle > &mcp_v, const std::vector< simb::Origin_t > &orig_v, const std::vector< simb::MCParticle > &mcmp_v={})

unsigned int	AncestorTrackID (const unsigned int part_index)

unsigned int	MotherTrackID (const unsigned int part_index) const

unsigned int	TrackToParticleIndex (const unsigned int track_id) const

bool	InDetector (const double &x, const double &y, const double &z) const

Public Attributes
std::map< unsigned int, unsigned int >	_track_index
	Track ID => Index Map. More...

std::set< int >	_pdg_list
	PDG code list for which particle's trajectory within the detector is saved. More...

T	elements
	STL member. More...

Protected Attributes
double	_x_max
	x-max of volume box used to determine whether to save track information More...

double	_x_min
	x-min of volume box used to determine whether to save track information More...

double	_y_max
	y-max of volume box used to determine whether to save track information More...

double	_y_min
	y-min of volume box used to determine whether to save track information More...

double	_z_max
	z-max of volume box used to determine whether to save track information More...

double	_z_min
	z-min of volume box used to determine whether to save track information More...

std::vector< unsigned int >	_trackIDOffsets
	Track ID offsets for different MCTruths. More...

Detailed Description

Definition at line 91 of file MCRecoPart.h.

Constructor & Destructor Documentation

sim::MCRecoPart::MCRecoPart ( fhicl::ParameterSet const & pset )

Default constructor with fhicl parameters.

Definition at line 19 of file MCRecoPart.cxx.

References _pdg_list, _track_index, _trackIDOffsets, _x_max, _x_min, _y_max, _y_min, _z_max, _z_min, clear(), and geo::GeometryCore::Iterate().

     : _x_max{std::numeric_limits<double>::min()}
     , _x_min{std::numeric_limits<double>::max()}
     , _y_max{std::numeric_limits<double>::min()}
     , _y_min{std::numeric_limits<double>::max()}
     , _z_max{std::numeric_limits<double>::min()}
     , _z_min{std::numeric_limits<double>::max()}
     , _trackIDOffsets{pset.get<std::vector<unsigned int>>("TrackIDOffsets", {0})}
   {
     this->clear();
     _track_index.clear();
     _pdg_list.clear();
     for (auto const& id : pset.get<std::vector<int>>("SavePathPDGList"))
 
       _pdg_list.insert(id);
 
     art::ServiceHandle<geo::Geometry const> geo;
     // Build "Fiducial" Volume Definition:
     //
     // Iterate over all TPC's to get bounding box that covers volumes of each individual TPC in the detector
     for (auto const& tpc : geo->Iterate<geo::TPCGeo>()) {
       _x_max = std::max(_x_max, tpc.BoundingBox().MaxX());
       _x_min = std::min(_x_min, tpc.BoundingBox().MinX());
       _y_max = std::max(_y_max, tpc.BoundingBox().MaxY());
       _y_min = std::min(_y_min, tpc.BoundingBox().MinY());
       _z_max = std::max(_z_max, tpc.BoundingBox().MaxZ());
       _z_min = std::min(_z_min, tpc.BoundingBox().MinZ());
     }
   }

virtual sim::MCRecoPart::~MCRecoPart ( )

inlinevirtual

Default destructor.

Definition at line 98 of file MCRecoPart.h.

98 {};

Member Function Documentation

void sim::MCRecoPart::AddParticles	(	const std::vector< simb::MCParticle > &	mcp_v,
		const std::vector< simb::Origin_t > &	orig_v,
		const std::vector< simb::MCParticle > &	mcmp_v = `{}`
	)

Definition at line 133 of file MCRecoPart.cxx.

References _pdg_list, _track_index, clear(), InDetector(), and util::size().

Referenced by MCReco::produce().

   {
     if (orig_v.size() != mcp_v.size())
       throw cet::exception(__FUNCTION__) << "MCParticle and Origin_t vector size not same!";
 
     this->clear();
     _track_index.clear();
 
     for (size_t i = 0; i < mcp_v.size(); ++i) {
 
       auto const& mcp = mcp_v[i];
 
       //std::cout<<" Track ID : "<<mcp.TrackId()<<" ... Index : " <<this->size()<<std::endl;
 
       _track_index.insert(std::make_pair((size_t)(mcp.TrackId()), (size_t)(this->size())));
 
       // Change units to LArSoft (MeV, cm, us)
       // (done inside constructor of MCMiniPart)
       this->push_back(MCMiniPart(mcp));
 
       auto& mini_mcp = (*this->rbegin());
 
       for (size_t i = 0; i < (size_t)(mcp.NumberDaughters()); ++i) {
         mini_mcp.AddDaughter(mcp.Daughter(i));
       }
       mini_mcp._origin = orig_v[i];
 
       if (_pdg_list.find(mcp.PdgCode()) != _pdg_list.end()) {
 
         std::set<size_t> det_path_index;
 
         for (size_t i = 0; i < mcp.NumberTrajectoryPoints(); ++i) {
 
           if (InDetector(mcp.Vx(i), mcp.Vy(i), mcp.Vz(i))) det_path_index.insert(i);
         }
 
         if (det_path_index.size()) {
           if ((*det_path_index.begin())) det_path_index.insert((*det_path_index.begin()) - 1);
           if (det_path_index.size() > 1) {
             if (((*det_path_index.rbegin()) + 1) < mcp.NumberTrajectoryPoints())
               det_path_index.insert((*det_path_index.rbegin()) + 1);
           }
           std::vector<std::pair<TLorentzVector, TLorentzVector>> det_path;
           det_path.reserve(det_path_index.size());
           for (auto const& index : det_path_index) {
 
             TLorentzVector vec(mcp.Momentum(index));
             for (size_t i = 0; i < 4; ++i)
               vec[i] *= 1.e3;
 
             det_path.emplace_back(mcp.Position(index), vec);
           }
           mini_mcp._det_path = std::move(det_path);
         }
       } // end if in _pdg_list
     }   // end for loop over mcp_v
 
     // Now loop over dropped particles
     for (auto const& mcmp : mcmp_v) {
 
       _track_index.try_emplace(mcmp.TrackId(), this->size());
 
       this->push_back(sim::MCMiniPart(mcmp));
 
     } // end for loop over mcmp_v
   }   // end AddParticles

unsigned int sim::MCRecoPart::AncestorTrackID ( const unsigned int part_index )

Definition at line 76 of file MCRecoPart.cxx.

References _trackIDOffsets, sim::kINVALID_UINT, MotherTrackID(), util::size(), and TrackToParticleIndex().

Referenced by sim::MCTrackRecoAlg::Reconstruct(), and sim::MCShowerRecoAlg::Reconstruct().

   {
     if (part_index >= this->size()) return kINVALID_UINT;
 
     if ((*this)[part_index]._ancestor != kINVALID_UINT) return (*this)[part_index]._ancestor;
 
     auto result = MotherTrackID(part_index);
 
     if (result == this->at(part_index)._track_id) return result;
 
     //Is a primary particle. Account for possible track ID offsets
     for (auto const& offset : _trackIDOffsets) {
       if (result == offset) return this->at(part_index)._track_id;
     }
 
     auto mother_index = TrackToParticleIndex(result);
 
     while (1) {
 
       if (mother_index != kINVALID_UINT) {
 
         auto const new_result = MotherTrackID(mother_index);
 
         if (new_result == this->at(mother_index)._track_id) break;
 
         result = new_result;
       }
       else {
 
         // Look for a particle that has a daughter = this mother
         auto const old_result = result;
         for (auto const& p : *this) {
 
           if (p.HasDaughter(result)) {
             result = p._track_id;
             break;
           }
         }
         if (result == old_result) break;
       }
 
       mother_index = TrackToParticleIndex(result);
     }
 
     (*this)[part_index]._ancestor = result;
     return result;
   }

bool sim::MCRecoPart::InDetector	(	const double &	x,
		const double &	y,
		const double &	z
	)		const

Definition at line 126 of file MCRecoPart.cxx.

References _x_max, _y_max, _y_min, _z_max, x, and y.

Referenced by AddParticles().

   {
     return !(x > _x_max || x < _x_min || z > _z_max || z < _z_min || y > _y_max || y < _y_min);
   }

unsigned int sim::MCRecoPart::MotherTrackID ( const unsigned int part_index ) const

Definition at line 50 of file MCRecoPart.cxx.

References _trackIDOffsets, sim::kINVALID_UINT, part, util::size(), and TrackToParticleIndex().

Referenced by AncestorTrackID(), sim::MCTrackRecoAlg::Reconstruct(), and sim::MCShowerRecoAlg::Reconstruct().

   {
     if (this->size() <= part_index) return ::sim::kINVALID_UINT;
 
     unsigned int result = this->at(part_index)._mother;
 
     //Is a primary particle. Account for possible track ID offsets
     for (auto const& offset : _trackIDOffsets) {
       if (result == offset) return this->at(part_index)._track_id;
     }
 
     if (TrackToParticleIndex(result) != ::sim::kINVALID_UINT) return result;
 
     //std::cout<< "\033[95mWarning:\033[00m Mother particle not in the particle list!"<<std::endl;
     // Do brute search
     unsigned int daughter_id = this->at(part_index)._track_id;
 
     for (auto const& part : *this) {
 
       if (part.HasDaughter(daughter_id)) return part._track_id;
     }
     return result;
   }

unsigned int sim::MCRecoPart::TrackToParticleIndex ( const unsigned int track_id ) const

inline

Definition at line 112 of file MCRecoPart.h.

References sim::kINVALID_UINT.

Referenced by AncestorTrackID(), MotherTrackID(), sim::MCTrackRecoAlg::Reconstruct(), and sim::MCShowerRecoAlg::Reconstruct().

     {
       auto const iter(_track_index.find(track_id));
       if (iter == _track_index.end()) return kINVALID_UINT;
       return (*iter).second;
     }

Member Data Documentation

std::set<int> sim::MCRecoPart::_pdg_list

PDG code list for which particle's trajectory within the detector is saved.

Definition at line 127 of file MCRecoPart.h.

Referenced by AddParticles(), MCRecoPart(), and sim::MCTrackRecoAlg::Reconstruct().

std::map<unsigned int, unsigned int> sim::MCRecoPart::_track_index

Track ID => Index Map.

Definition at line 123 of file MCRecoPart.h.

Referenced by AddParticles(), sim::MCShowerRecoPart::ConstructShower(), and MCRecoPart().

std::vector<unsigned int> sim::MCRecoPart::_trackIDOffsets

protected

Track ID offsets for different MCTruths.

Definition at line 136 of file MCRecoPart.h.

Referenced by AncestorTrackID(), MCRecoPart(), and MotherTrackID().

double sim::MCRecoPart::_x_max

protected

x-max of volume box used to determine whether to save track information

Definition at line 130 of file MCRecoPart.h.

Referenced by InDetector(), and MCRecoPart().

double sim::MCRecoPart::_x_min

protected

x-min of volume box used to determine whether to save track information

Definition at line 131 of file MCRecoPart.h.

Referenced by MCRecoPart().

double sim::MCRecoPart::_y_max

protected

y-max of volume box used to determine whether to save track information

Definition at line 132 of file MCRecoPart.h.

Referenced by InDetector(), and MCRecoPart().

double sim::MCRecoPart::_y_min

protected

y-min of volume box used to determine whether to save track information

Definition at line 133 of file MCRecoPart.h.

Referenced by InDetector(), and MCRecoPart().

double sim::MCRecoPart::_z_max

protected

z-max of volume box used to determine whether to save track information

Definition at line 134 of file MCRecoPart.h.

Referenced by InDetector(), and MCRecoPart().

double sim::MCRecoPart::_z_min

protected

z-min of volume box used to determine whether to save track information

Definition at line 135 of file MCRecoPart.h.

Referenced by MCRecoPart().

T std::vector< T >::elements

inherited

STL member.

The documentation for this class was generated from the following files:

larsim/v09_43_00/source/larsim/MCSTReco/MCRecoPart.h
larsim/v09_43_00/source/larsim/MCSTReco/MCRecoPart.cxx

Public Member Functions

Public Attributes

Protected Attributes

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation