sim::MCRecoPart Class Reference

#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)
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...

Detailed Description

Definition at line 76 of file MCRecoPart.h.

Constructor & Destructor Documentation

sim::MCRecoPart::MCRecoPart

(

fhicl::ParameterSet const &

pset

)

Default constructor with fhicl parameters.

Definition at line 15 of file MCRecoPart.cxx.

References _pdg_list, _track_index, _x_max, _x_min, _y_max, _y_min, _z_max, _z_min, geo::GeometryCore::begin_TPC(), clear(), geo::GeometryCore::end_TPC(), and fhicl::ParameterSet::get().

  { 
    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> geo;
    // Build "Fiducial" Volume Definition:
    //
    // Iterate over all TPC's to get bounding box that covers volumes of each individual TPC in the detector
    _x_min = std::min_element(geo->begin_TPC(), geo->end_TPC(), [](auto const &lhs, auto const &rhs){ return lhs.BoundingBox().MinX() < rhs.BoundingBox().MinX();})->MinX();
    _y_min = std::min_element(geo->begin_TPC(), geo->end_TPC(), [](auto const &lhs, auto const &rhs){ return lhs.BoundingBox().MinY() < rhs.BoundingBox().MinY();})->MinY();
    _z_min = std::min_element(geo->begin_TPC(), geo->end_TPC(), [](auto const &lhs, auto const &rhs){ return lhs.BoundingBox().MinZ() < rhs.BoundingBox().MinZ();})->MinZ();
    _x_max = std::max_element(geo->begin_TPC(), geo->end_TPC(), [](auto const &lhs, auto const &rhs){ return lhs.BoundingBox().MaxX() < rhs.BoundingBox().MaxX();})->MaxX();
    _y_max = std::max_element(geo->begin_TPC(), geo->end_TPC(), [](auto const &lhs, auto const &rhs){ return lhs.BoundingBox().MaxY() < rhs.BoundingBox().MaxY();})->MaxY();
    _z_max = std::max_element(geo->begin_TPC(), geo->end_TPC(), [](auto const &lhs, auto const &rhs){ return lhs.BoundingBox().MaxZ() < rhs.BoundingBox().MaxZ();})->MaxZ();
  }

virtual sim::MCRecoPart::~MCRecoPart ( )

inlinevirtual

Default destructor.

Definition at line 84 of file MCRecoPart.h.

{};

Member Function Documentation

void sim::MCRecoPart::AddParticles	(	const std::vector< simb::MCParticle > &	mcp_v,
		const std::vector< simb::Origin_t > &	orig_v
	)

Definition at line 124 of file MCRecoPart.cxx.

References _pdg_list, _track_index, clear(), and InDetector().

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())));

      this->push_back(MCMiniPart());

      auto& mini_mcp = (*this->rbegin());

      for(size_t i=0; i<(size_t)(mcp.NumberDaughters()); ++i)
        mini_mcp._daughters.insert(mcp.Daughter(i));

      mini_mcp._track_id  = mcp.TrackId();
      mini_mcp._pdgcode   = mcp.PdgCode();
      mini_mcp._mother    = mcp.Mother();
      mini_mcp._process   = mcp.Process();
      mini_mcp._start_vtx = mcp.Position();
      mini_mcp._start_mom = mcp.Momentum();
      mini_mcp._end_vtx   = mcp.EndPosition();
      mini_mcp._end_mom   = mcp.EndMomentum();
      mini_mcp._origin    = orig_v[i];

      // Change units to LArSoft (MeV, cm, us)
      for(size_t i=0; i<4; ++i) {
        mini_mcp._start_mom[i] *= 1.e3;
        mini_mcp._end_mom[i]   *= 1.e3;
      }
      /*
      for(size_t i=0; i<3; ++i) {
        mini_mcp._start_vtx[i] /= 10.;
        mini_mcp._end_vtx[i] /= 10.;
      }
      mini_mcp.start_vtx[3] /= 1.e-3;
      mini_mcp.end_vtx[3]   /= 1.e-3;
      */

      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 );
          }
          mini_mcp._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;

            mini_mcp._det_path.push_back(std::make_pair(mcp.Position(index),vec));

          }
        }
      }
    }
  }

unsigned int sim::MCRecoPart::AncestorTrackID

(

const unsigned int

part_index

)

Definition at line 64 of file MCRecoPart.cxx.

References sim::kINVALID_UINT, MotherTrackID(), 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;

    unsigned int result = MotherTrackID(part_index);

    if(result == this->at(part_index)._track_id) return result;

    if(!result) 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._daughters.find(result) != p._daughters.end()) {
            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 113 of file MCRecoPart.cxx.

References _x_max, _x_min, _y_max, _y_min, _z_max, _z_min, x, y, and z.

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 38 of file MCRecoPart.cxx.

References sim::kINVALID_UINT, part, 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;
    
    if(!result) 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._daughters.find(daughter_id) != part._daughters.end())

        return part._track_id;

    }
    return result;
  }

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

inline

Definition at line 97 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 115 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 111 of file MCRecoPart.h.

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

double sim::MCRecoPart::_x_max

protected

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

Definition at line 119 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 120 of file MCRecoPart.h.

Referenced by InDetector(), and MCRecoPart().

double sim::MCRecoPart::_y_max

protected

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

Definition at line 121 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 122 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 123 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 124 of file MCRecoPart.h.

Referenced by InDetector(), and MCRecoPart().

T std::vector< T >::elements

inherited

STL member.

---

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

• /cvmfs/larsoft.opensciencegrid.org/products/larsim/v07_09_00/source/larsim/MCSTReco/MCRecoPart.h
• /cvmfs/larsoft.opensciencegrid.org/products/larsim/v07_09_00/source/larsim/MCSTReco/MCRecoPart.cxx