TrackContainmentAlg.hh

Go to the documentation of this file.

#ifndef TRK_TRACKCONTAINMENTALG_H
#define TRK_TRACKCONTAINMENTALG_H

#include <string>
#include <vector>

#include "lardataobj/RecoBase/Track.h"
#include "lardataobj/AnalysisBase/CosmicTag.h"

#include "TTree.h"
#include "fhiclcpp/ParameterSet.h"

namespace geo{ class GeometryCore; }

namespace trk{
  class TrackContainmentAlg;

  typedef struct TrackTree{

    TrackTree(){}
    
    TrackTree(recob::Track const& t):
      start_x(t.Vertex().X()),start_y(t.Vertex().Y()),start_z(t.Vertex().Z()),
      start_theta(t.VertexDirection().Theta()),start_phi(t.VertexDirection().Phi()),
      end_x(t.End().X()),end_y(t.End().Y()),end_z(t.End().Z()),
      end_theta(t.EndDirection().Theta()),end_phi(t.EndDirection().Phi()),
      length(t.Length()),length_simple( (t.End()-t.Vertex()).Mag() ){}

    double start_x;
    double start_y;
    double start_z;
    double start_theta;
    double start_phi;
    double end_x;
    double end_y;
    double end_z;
    double end_theta;
    double end_phi;
    double length;
    double length_simple;

    std::string Leaflist()
    { return "start_x/D:start_y/D:start_z/D:start_theta/D:start_phi/D:end_x/D:end_y/D:end_z/D:end_theta/D:end_phi/D:length/D:length_simple/D"; }
    
  } TrackTree_t;
  
}


class trk::TrackContainmentAlg{
  
public:
  
  TrackContainmentAlg();

  virtual ~TrackContainmentAlg(){};

  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()
  { return fTrackContainmentLevel; }
  std::vector< std::vector<double> > const& GetTrackMinDistanceValues()
  { return fMinDistances; }
  std::vector< std::vector<anab::CosmicTag> > const& GetTrackCosmicTags();

  void setMakeCosmicTags(bool flag=true) { fMakeCosmicTags = flag; }
  void setFillOutputTree(bool flag=true) { fFillOutputTree = flag; }
  
  
 private:

  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;
  

  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&);
  
};

#endif