trk::TrackContainmentAlg Class Reference

#include <TrackContainmentAlg.hh>

Public Member Functions
	TrackContainmentAlg ()
	Default constructor. More...
void	SetupOutputTree (TTree *)
void	Configure (fhicl::ParameterSet const &)
void	SetRunEvent (unsigned int const &, unsigned int const &)
void	ProcessTracks (std::vector< std::vector< recob::Track > > const &, geo::GeometryCore const &)
std::vector< std::vector< int > > const &	GetTrackContainmentValues ()
std::vector< std::vector < double > > const &	GetTrackMinDistanceValues ()
std::vector< std::vector < anab::CosmicTag > > const &	GetTrackCosmicTags ()
void	setMakeCosmicTags (bool flag=true)
void	setFillOutputTree (bool flag=true)

Private Member Functions
bool	IsContained (recob::Track const &, geo::GeometryCore const &)
anab::CosmicTagID_t	GetCosmicTagID (recob::Track const &, geo::GeometryCore const &)
double	MinDistanceStartPt (recob::Track const &, recob::Track const &)
double	MinDistanceEndPt (recob::Track const &, recob::Track const &)

Private Attributes
double	fZBuffer
double	fYBuffer
double	fXBuffer
double	fIsolation
bool	fMakeCosmicTags
bool	fDebug
bool	fFillOutputTree
TTree *	fTrackTree
TrackTree_t	fTrackTreeObj
unsigned int	fRun
unsigned int	fEvent
unsigned int	fCollection
unsigned int	fTrkID
double	fDistance
int	fContainment
std::vector< std::vector< int > >	fTrackContainmentLevel
std::vector< std::vector < std::pair< int, int > > >	fTrackContainmentIndices
std::vector< std::vector < double > >	fMinDistances
std::vector< std::vector < anab::CosmicTag > >	fCosmicTags

Detailed Description

Definition at line 58 of file TrackContainmentAlg.hh.

Constructor & Destructor Documentation

trk::TrackContainmentAlg::TrackContainmentAlg

(

)

Default constructor.

Definition at line 10 of file TrackContainmentAlg.cxx.

{}

Member Function Documentation

void trk::TrackContainmentAlg::Configure

(

fhicl::ParameterSet const &

p

)

Definition at line 24 of file TrackContainmentAlg.cxx.

{
  fZBuffer = p.get<double>("ZBuffer");
  fYBuffer = p.get<double>("YBuffer");
  fXBuffer = p.get<double>("XBuffer");
  fIsolation = p.get<double>("Isolation");

  fDebug = p.get<bool>("Debug",false);
  fMakeCosmicTags = p.get<bool>("MakeCosmicTags",true);
  fFillOutputTree = p.get<bool>("FillOutputTree",true);
}

anab::CosmicTagID_t trk::TrackContainmentAlg::GetCosmicTagID ( recob::Track const &  track, geo::GeometryCore const &  geo  )

private

Definition at line 54 of file TrackContainmentAlg.cxx.

{

  auto id = anab::CosmicTagID_t::kNotTagged;

  if(track.Vertex().Z() < (0+fZBuffer) || track.Vertex().Z() > (geo.DetLength()-fZBuffer))
    id = anab::CosmicTagID_t::kGeometry_Z;
  else if(track.Vertex().Y() < (-1*geo.DetHalfHeight()+fYBuffer) || track.Vertex().Y() > (geo.DetHalfHeight()-fYBuffer))
    id = anab::CosmicTagID_t::kGeometry_Y;
  else if( (track.Vertex().X()>0 && track.Vertex().X() < (0+fXBuffer)) ||
           (track.Vertex().X()<2*geo.DetHalfWidth() && track.Vertex().X() > (2*geo.DetHalfWidth()-fXBuffer)) )
    id = anab::CosmicTagID_t::kGeometry_X;

  if( track.End().Z() < (0+fZBuffer) || track.End().Z() > (geo.DetLength()-fZBuffer) ){
    if(id == anab::CosmicTagID_t::kNotTagged)
      id = anab::CosmicTagID_t::kGeometry_Z;
    else if(id == anab::CosmicTagID_t::kGeometry_Z)
      id = anab::CosmicTagID_t::kGeometry_ZZ;
    else if(id == anab::CosmicTagID_t::kGeometry_Y)
      id = anab::CosmicTagID_t::kGeometry_YZ;
    else if(id == anab::CosmicTagID_t::kGeometry_X)
      id = anab::CosmicTagID_t::kGeometry_XZ;
  }
  else if( track.End().Y() < (-1*geo.DetHalfHeight()+fYBuffer) || track.End().Y() > (geo.DetHalfHeight()-fYBuffer) ){
    if(id == anab::CosmicTagID_t::kNotTagged)
      id = anab::CosmicTagID_t::kGeometry_Y;
    else if(id == anab::CosmicTagID_t::kGeometry_Z)
      id = anab::CosmicTagID_t::kGeometry_YZ;
    else if(id == anab::CosmicTagID_t::kGeometry_Y)
      id = anab::CosmicTagID_t::kGeometry_YY;
    else if(id == anab::CosmicTagID_t::kGeometry_X)
      id = anab::CosmicTagID_t::kGeometry_XY;
  }
  else if( (track.End().X()>0 && track.End().X() < (0+fXBuffer)) ||
           (track.End().X()<2*geo.DetHalfWidth() && track.End().X() > (2*geo.DetHalfWidth()-fXBuffer)) ){
    if(id == anab::CosmicTagID_t::kNotTagged)
      id = anab::CosmicTagID_t::kGeometry_X;
    else if(id == anab::CosmicTagID_t::kGeometry_Z)
      id = anab::CosmicTagID_t::kGeometry_XZ;
    else if(id == anab::CosmicTagID_t::kGeometry_Y)
      id = anab::CosmicTagID_t::kGeometry_XY;
    else if(id == anab::CosmicTagID_t::kGeometry_X)
      id = anab::CosmicTagID_t::kGeometry_XX;
  }

  if(track.Vertex().X() < 0 || track.Vertex().X() > 2*geo.DetHalfWidth())
    id = anab::CosmicTagID_t::kOutsideDrift_Partial;
  if(track.End().X() < 0 || track.End().X() > 2*geo.DetHalfWidth()){
    if(id==anab::CosmicTagID_t::kOutsideDrift_Partial)
      id = anab::CosmicTagID_t::kOutsideDrift_Complete;
    else
      id = anab::CosmicTagID_t::kOutsideDrift_Partial;
  }

  return id;
}

std::vector< std::vector<int> > const& trk::TrackContainmentAlg::GetTrackContainmentValues ( )

inline

Definition at line 73 of file TrackContainmentAlg.hh.

  { return fTrackContainmentLevel; }

std::vector< std::vector< anab::CosmicTag > > const & trk::TrackContainmentAlg::GetTrackCosmicTags

(

)

Definition at line 323 of file TrackContainmentAlg.cxx.

{
  if(fMakeCosmicTags)
    return fCosmicTags;
  else
    throw cet::exception("TrackContainmentAlg::GetTrackCosmicTags")
      << "Cosmic tags not created. Set MakeCosmicTags to true in fcl paramters.";
}

std::vector< std::vector<double> > const& trk::TrackContainmentAlg::GetTrackMinDistanceValues ( )

inline

Definition at line 75 of file TrackContainmentAlg.hh.

  { return fMinDistances; }

bool trk::TrackContainmentAlg::IsContained ( recob::Track const &  track, geo::GeometryCore const &  geo  )

private

Definition at line 36 of file TrackContainmentAlg.cxx.

{
  if( track.Vertex().Z() < (0+fZBuffer) || track.Vertex().Z() > (geo.DetLength()-fZBuffer) )
    return false;
  if( track.End().Z() < (0+fZBuffer) || track.End().Z() > (geo.DetLength()-fZBuffer) )
    return false;
  if( track.Vertex().Y() < (-1*geo.DetHalfHeight()+fYBuffer) || track.Vertex().Y() > (geo.DetHalfHeight()-fYBuffer) )
    return false;
  if( track.End().Y() < (-1*geo.DetHalfHeight()+fYBuffer) || track.End().Y() > (geo.DetHalfHeight()-fYBuffer) )
    return false;
  if( track.Vertex().X() < (0+fXBuffer) || track.Vertex().X() > (2*geo.DetHalfWidth()-fXBuffer) )
    return false;
  if( track.End().X() < (0+fXBuffer) || track.End().X() > (2*geo.DetHalfWidth()-fXBuffer) )
    return false;

  return true;
}

double trk::TrackContainmentAlg::MinDistanceEndPt ( recob::Track const &  t_probe, recob::Track const &  t_ref  )

private

Definition at line 127 of file TrackContainmentAlg.cxx.

{
  double min_distance = 9e12;
  double tmp;
  for(size_t i_p=0; i_p<t_ref.NumberTrajectoryPoints(); ++i_p){
    tmp =
      (t_probe.End().X()-t_ref.LocationAtPoint(i_p).X())*(t_probe.End().X()-t_ref.LocationAtPoint(i_p).X()) +
      (t_probe.End().Y()-t_ref.LocationAtPoint(i_p).Y())*(t_probe.End().Y()-t_ref.LocationAtPoint(i_p).Y()) +
      (t_probe.End().Z()-t_ref.LocationAtPoint(i_p).Z())*(t_probe.End().Z()-t_ref.LocationAtPoint(i_p).Z());
    if(tmp<min_distance)
      min_distance=tmp;
  }
  return std::sqrt(min_distance);
}

double trk::TrackContainmentAlg::MinDistanceStartPt ( recob::Track const &  t_probe, recob::Track const &  t_ref  )

private

Definition at line 111 of file TrackContainmentAlg.cxx.

{
  double min_distance = 9e12;
  double tmp;
  for(size_t i_p=0; i_p<t_ref.NumberTrajectoryPoints(); ++i_p){
    tmp =
      (t_probe.Vertex().X()-t_ref.LocationAtPoint(i_p).X())*(t_probe.Vertex().X()-t_ref.LocationAtPoint(i_p).X()) +
      (t_probe.Vertex().Y()-t_ref.LocationAtPoint(i_p).Y())*(t_probe.Vertex().Y()-t_ref.LocationAtPoint(i_p).Y()) +
      (t_probe.Vertex().Z()-t_ref.LocationAtPoint(i_p).Z())*(t_probe.Vertex().Z()-t_ref.LocationAtPoint(i_p).Z());
    //std::cout << "\t\t\ttmp=" << tmp << std::endl;
    if(tmp<min_distance)
      min_distance=tmp;
  }
  return std::sqrt(min_distance);
}

void trk::TrackContainmentAlg::ProcessTracks	(	std::vector< std::vector< recob::Track > > const &	tracksVec,
		geo::GeometryCore const &	geo
	)

Definition at line 148 of file TrackContainmentAlg.cxx.

{

  if(fDebug){
    std::cout << "Geometry:" << std::endl;
    std::cout << "\t" << geo.DetHalfWidth() << " " << geo.DetHalfHeight() << " " << geo.DetLength() << std::endl;
    std::cout << "\t z:(" << fZBuffer << "," << geo.DetLength()-fZBuffer << ")"
              << "\t y:(" << -1.*geo.DetHalfHeight()+fYBuffer << "," << geo.DetHalfHeight()-fYBuffer << ")"
              << "\t x:(" << 0+fXBuffer << "," << 2*geo.DetHalfWidth()-fXBuffer << ")" << std::endl;
  }


  int containment_level=0;
  bool track_linked = false;
  std::size_t n_tracks=0;

  fTrackContainmentLevel.clear();
  fTrackContainmentLevel.resize(tracksVec.size());
  fMinDistances.clear();
  fMinDistances.resize(tracksVec.size());

  fTrackContainmentIndices.clear();
  fTrackContainmentIndices.push_back( std::vector< std::pair<int,int> >() );

  fCosmicTags.clear();
  fCosmicTags.resize(tracksVec.size());

  //first, loop through tracks and see what's not contained

  for(size_t i_tc=0; i_tc<tracksVec.size(); ++i_tc){
    fTrackContainmentLevel[i_tc].resize(tracksVec[i_tc].size(),-1);
    fMinDistances[i_tc].resize(tracksVec[i_tc].size(),9e12);
    fCosmicTags[i_tc].resize(tracksVec[i_tc].size(),anab::CosmicTag(-1));
    n_tracks += tracksVec[i_tc].size();
    for(size_t i_t=0; i_t<tracksVec[i_tc].size(); ++i_t){

      if(!IsContained(tracksVec[i_tc][i_t],geo)){
        if(!track_linked) track_linked=true;
        fTrackContainmentLevel[i_tc][i_t] = 0;
        fTrackContainmentIndices.back().emplace_back(i_tc,i_t);
        if(fDebug){
          std::cout << "\tTrack (" << i_tc << "," << i_t << ")"
                    << " " << containment_level << std::endl;
        }

      }//end if contained
    }//end loop over tracks

  }//end loop over track collections



  //now, while we are still linking tracks, loop over all tracks and note anything
  //close to an uncontained (or linked) track
  while(track_linked){

    track_linked=false;
    ++containment_level;
    fTrackContainmentIndices.push_back( std::vector< std::pair<int,int> >() );

    for(size_t i_tc=0; i_tc<tracksVec.size(); ++i_tc){
      for(size_t i_t=0; i_t<tracksVec[i_tc].size(); ++i_t){
        if(fTrackContainmentLevel[i_tc][i_t]>=0)
          continue;
        else
          {
            for(auto const& i_tr : fTrackContainmentIndices[containment_level-1]){

              /*
              if(fDebug){
                std::cout << "\t\t" << MinDistanceStartPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second]) << std::endl;
                std::cout << "\t\t" << MinDistanceEndPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second]) << std::endl;
              }
              */

              if(MinDistanceStartPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second])<fMinDistances[i_tc][i_t])
                fMinDistances[i_tc][i_t] = MinDistanceStartPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second]);
              if(MinDistanceEndPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second])<fMinDistances[i_tc][i_t])
                fMinDistances[i_tc][i_t] = MinDistanceEndPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second]);

              if(MinDistanceStartPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second])<fIsolation ||
                 MinDistanceEndPt(tracksVec[i_tc][i_t],tracksVec[i_tr.first][i_tr.second])<fIsolation){
                if(!track_linked) track_linked=true;
                fTrackContainmentLevel[i_tc][i_t] = containment_level;
                fTrackContainmentIndices.back().emplace_back(i_tc,i_t);

                if(fDebug){
                  std::cout << "\tTrackPair (" << i_tc << "," << i_t << ") and (" << i_tr.first << "," << i_tr.second << ")"
                            << " " << containment_level << std::endl;
                }

              }//end if track not isolated

            }//end loop over existing uncontained/linked tracks

          }//end if track not already linked

      }//end loops over tracks
    }//end loop over track collections
  }//end while linking tracks


  if(fDebug)
    std::cout << "All done! Now let's make the tree and tags!" << std::endl;

  //now we're going to will the tree and create tags if requested
  for(size_t i_tc=0; i_tc<tracksVec.size(); ++i_tc){
    for(size_t i_t=0; i_t<tracksVec[i_tc].size(); ++i_t){

      //fill ROOT Tree
      if(fFillOutputTree){
        fTrackTreeObj = TrackTree_t(tracksVec[i_tc][i_t]);
        fDistance = fMinDistances[i_tc][i_t];
        fCollection = i_tc;
        fTrkID = i_t;
        fContainment = fTrackContainmentLevel[i_tc][i_t];
        fTrackTree->Fill();
      }

      if(fMakeCosmicTags){

        //default (if track looks contained and isolated)
        float score=0;
        auto id = anab::CosmicTagID_t::kNotTagged;

        //overwrite if track is not contained or not isolated
        if(fTrackContainmentLevel[i_tc][i_t]>=0){
          score = 1./(1.+(float)fTrackContainmentLevel[i_tc][i_t]);
          if(fTrackContainmentLevel[i_tc][i_t]==0)
            id = GetCosmicTagID(tracksVec[i_tc][i_t],geo);
          else
            id = anab::CosmicTagID_t::kNotIsolated;
        }

        fCosmicTags[i_tc][i_t] =
          anab::CosmicTag(std::vector<float>{(float)tracksVec[i_tc][i_t].Vertex().X(),
                                             (float)tracksVec[i_tc][i_t].Vertex().Y(),
                                             (float)tracksVec[i_tc][i_t].Vertex().Z()},
                          std::vector<float>{(float)tracksVec[i_tc][i_t].End().X(),
                                             (float)tracksVec[i_tc][i_t].End().Y(),
                                             (float)tracksVec[i_tc][i_t].End().Z()},
            score,id);
      }//end cosmic tag making


      //some debug work
      if(fTrackContainmentLevel[i_tc][i_t]<0 && fDebug){
        std::cout << "Track (" << i_tc << "," << i_t << ")"
                  << " " << fTrackContainmentLevel[i_tc][i_t]
                  << " " << fMinDistances[i_tc][i_t] << std::endl;
        std::cout << "\tS_(X,Y,Z) = ("
                  << tracksVec[i_tc][i_t].Vertex().X() << ","
                  << tracksVec[i_tc][i_t].Vertex().Y() << ","
                  << tracksVec[i_tc][i_t].Vertex().Z() << ")" << std::endl;
        std::cout << "\tNearest wire ..." << std::endl;
        for(size_t i_p=0; i_p<geo.Nplanes(); ++i_p)
          std::cout << "\t\tPlane " << i_p << " " << geo.NearestWireID(tracksVec[i_tc][i_t].Vertex(),i_p).Wire << std::endl;
        std::cout << "\tE_(X,Y,Z) = ("
                  << tracksVec[i_tc][i_t].End().X() << ","
                  << tracksVec[i_tc][i_t].End().Y() << ","
                  << tracksVec[i_tc][i_t].End().Z() << ")" << std::endl;
        std::cout << "\tNearest wire ..." << std::endl;
        for(size_t i_p=0; i_p<geo.Nplanes(); ++i_p)
          std::cout << "\t\tPlane " << i_p << " " << geo.NearestWireID(tracksVec[i_tc][i_t].End(),i_p).Wire << std::endl;
        std::cout << "\tLength=" << tracksVec[i_tc][i_t].Length() << std::endl;
        std::cout << "\tSimple_length=" << (tracksVec[i_tc][i_t].End()-tracksVec[i_tc][i_t].Vertex()).R() << std::endl;
      }//end debug statements if track contained

    }//end loops over tracks
  }//end loop over track collections

}//end ProcessTracks

void trk::TrackContainmentAlg::setFillOutputTree ( bool  flag = true )

inline

Definition at line 80 of file TrackContainmentAlg.hh.

{ fFillOutputTree = flag; }

void trk::TrackContainmentAlg::setMakeCosmicTags ( bool  flag = true )

inline

Definition at line 79 of file TrackContainmentAlg.hh.

{ fMakeCosmicTags = flag; }

void trk::TrackContainmentAlg::SetRunEvent	(	unsigned int const &	run,
		unsigned int const &	event
	)

Definition at line 142 of file TrackContainmentAlg.cxx.

{
  fRun = run;
  fEvent = event;
}

void trk::TrackContainmentAlg::SetupOutputTree

(

TTree *

tfs_tree_trk

)

Definition at line 12 of file TrackContainmentAlg.cxx.

                                                               {
  fTrackTree = tfs_tree_trk;
  fTrackTree->SetObject(fTrackTree->GetName(),"Track Tree");
  fTrackTree->Branch("run",&fRun);
  fTrackTree->Branch("event",&fEvent);
  fTrackTree->Branch("trk",&fTrackTreeObj,fTrackTreeObj.Leaflist().c_str());
  fTrackTree->Branch("trk_col",&fCollection);
  fTrackTree->Branch("trk_id",&fTrkID);
  fTrackTree->Branch("trk_mindist",&fDistance);
  fTrackTree->Branch("trk_containment",&fContainment);
}

Member Data Documentation

unsigned int trk::TrackContainmentAlg::fCollection

private

Definition at line 97 of file TrackContainmentAlg.hh.

int trk::TrackContainmentAlg::fContainment

private

Definition at line 100 of file TrackContainmentAlg.hh.

std::vector< std::vector<anab::CosmicTag> > trk::TrackContainmentAlg::fCosmicTags

private

Definition at line 105 of file TrackContainmentAlg.hh.

bool trk::TrackContainmentAlg::fDebug

private

Definition at line 90 of file TrackContainmentAlg.hh.

double trk::TrackContainmentAlg::fDistance

private

Definition at line 99 of file TrackContainmentAlg.hh.

unsigned int trk::TrackContainmentAlg::fEvent

private

Definition at line 96 of file TrackContainmentAlg.hh.

bool trk::TrackContainmentAlg::fFillOutputTree

private

Definition at line 91 of file TrackContainmentAlg.hh.

double trk::TrackContainmentAlg::fIsolation

private

Definition at line 88 of file TrackContainmentAlg.hh.

bool trk::TrackContainmentAlg::fMakeCosmicTags

private

Definition at line 89 of file TrackContainmentAlg.hh.

std::vector< std::vector<double> > trk::TrackContainmentAlg::fMinDistances

private

Definition at line 104 of file TrackContainmentAlg.hh.

unsigned int trk::TrackContainmentAlg::fRun

private

Definition at line 95 of file TrackContainmentAlg.hh.

std::vector< std::vector< std::pair<int,int> > > trk::TrackContainmentAlg::fTrackContainmentIndices

private

Definition at line 103 of file TrackContainmentAlg.hh.

std::vector< std::vector<int> > trk::TrackContainmentAlg::fTrackContainmentLevel

private

Definition at line 102 of file TrackContainmentAlg.hh.

TTree* trk::TrackContainmentAlg::fTrackTree

private

Definition at line 93 of file TrackContainmentAlg.hh.

TrackTree_t trk::TrackContainmentAlg::fTrackTreeObj

private

Definition at line 94 of file TrackContainmentAlg.hh.

unsigned int trk::TrackContainmentAlg::fTrkID

private

Definition at line 98 of file TrackContainmentAlg.hh.

double trk::TrackContainmentAlg::fXBuffer

private

Definition at line 87 of file TrackContainmentAlg.hh.

double trk::TrackContainmentAlg::fYBuffer

private

Definition at line 86 of file TrackContainmentAlg.hh.

double trk::TrackContainmentAlg::fZBuffer

private

Definition at line 85 of file TrackContainmentAlg.hh.

---

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

• TrackContainmentAlg.hh
• TrackContainmentAlg.cxx