FillClusMatchTree.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file FillClusMatchTree.h

#ifndef FILLCLUSMATCHTREE__H
#define FILLCLUSMATCHTREE__H

#include "TrkrClusLoc.h"
#include "TrackClusEvaluator.h"

#include <fun4all/SubsysReco.h> 
#include <vector>
#include <string>

class EmbRecoMatchContainer;
class PHCompositeNode;
class PHG4ParticleSvtxMap;
class SvtxPHG4ParticleMap;
class EmbRecoMatchContainer;
class PHCompositeNode;
class PHG4TpcCylinderGeom;
class PHG4TpcCylinderGeomContainer;
class PHG4TruthInfoContainer;
class SvtxTrack;
class SvtxTrackMap;
class TrackSeedContainer;
class TrkrCluster;
/* class TrkrClusterContainer; */
class TrkrTruthTrack;
class TrkrTruthTrackContainer;
class TTree;
class TH2D;
class ClusHitsVerbose;
class TrkrClusterIsMatcher;

class FillClusMatchTree : public SubsysReco
{
 public:
  FillClusMatchTree(
        TrkrClusterIsMatcher* _ismatcher
      , const std::string& tfile_name="trackclusmatch.root"
      , bool  _fill_clusters      = true
      , bool  _fill_clusverbose   = true
      , bool  _fill_svtxnomatch   = false
  );

  virtual ~FillClusMatchTree();

  int Init(PHCompositeNode *) override;
  int InitRun(PHCompositeNode *topNode) override;
  int process_event(PHCompositeNode * /*topNode*/) override;
  int End(PHCompositeNode *topNode) override;

  void clear_clusvecs(const std::string& tag="");

  void print_mvtx_diagnostics();

   TrkrClusterIsMatcher* m_ismatcher;
 private:

   int createNodes(PHCompositeNode *topNode);
   TrackClusEvaluator m_TCEval;

   // contianer used to fill the other track matches
   EmbRecoMatchContainer   *m_EmbRecoMatchContainer   {nullptr};
   PHG4TruthInfoContainer  *m_PHG4TruthInfoContainer  {nullptr};
   SvtxTrackMap            *m_SvtxTrackMap            {nullptr};
   TrkrTruthTrackContainer *m_TrkrTruthTrackContainer {nullptr};
   ClusHitsVerbose         *m_PHG4ClusHitVerb         {nullptr};
   ClusHitsVerbose         *m_SvtxClusHitVerb         {nullptr};
   /* TrkrClusterContainer         *m_TruthClusterContainer        {nullptr}; */
   /* TrkrClusterContainer         *m_RecoClusterContainer         {nullptr}; */
   /* PHG4TpcCylinderGeomContainer *m_PHG4TpcCylinderGeomContainer {nullptr}; */


   TTree* m_ttree;
   std::string m_outfile_name;
 public:
   bool   m_fill_clusters;
   bool   m_fill_clusverbose; // unmatched Svtx tracks
   bool   m_fill_SvUnmatched; // unmatched Svtx tracks
 private:

  // Tree Branch members:
    int   nevent       {-1};
    int   nphg4        {0};
    int   nsvtx        {0};
    int   ntrackmatches {0};
    int   nphg4_part   {0};
    float centrality   {0.};

    // Tracks and clustes
    //
    // lables:
    //  tracks:
    //    g4 : phg4track matched
    //    sv : svtx_track matched
    //    gU : phg4track not-matched
    //    sU : svtx_track not-matched
    //  clusters:
    //    M : matched
    //    U : unmatched
    TH2D* h2_G4_nPixelsPhi;
    TH2D* h2_G4_nPixelsZ;
    TH2D* h2_Sv_nPixelsPhi;
    TH2D* h2_Sv_nPixelsZ;
    
    // Track tree
    int  b_trackid;
    bool b_is_g4track;
    bool b_is_Svtrack;
    bool b_is_matched;

    float b_trkpt;
    float b_trkphi;
    float b_trketa;

    int   b_nclus         {};
    int   b_nclustpc      {};
    int   b_nclusmvtx     {};
    int   b_nclusintt     {};

    float b_matchrat      {};
    float b_matchrat_intt {};
    float b_matchrat_mvtx {};
    float b_matchrat_tpc  {};

    std::vector<bool>  b_clusmatch  {};
    std::vector<float> b_clus_x     {};
    std::vector<float> b_clus_y     {};
    std::vector<float> b_clus_z     {};
    std::vector<float> b_clus_r     {};
    std::vector<int>   b_clus_layer {};
    /* std::vector<int>   b_clus_nphibins   {}; */
    /* std::vector<int>   b_clus_ntbins     {}; */

    std::vector<float> b_clus_lphi     {}; // l for local surface
    std::vector<float> b_clus_lphisize {}; // phi scaled by dphi
    std::vector<float> b_clus_lz       {};
    std::vector<float> b_clus_lzsize   {};

    // Maps of hits within clusters
    /* using BinData = std::pair<int,float>;  // index + energy for a given bin (MVTX, INTT, or TPC) */
    using VecVecInt   = std::vector<std::vector<int>>;
    /* using VecVecFloat = std::vector<std::vector<float>>; */
    VecVecInt b_phibins;
    VecVecInt b_phibins_cut;
    VecVecInt b_zbins;
    VecVecInt b_zbins_cut;

    VecVecInt b_phibinsE;
    VecVecInt b_phibinsE_cut;
    VecVecInt b_zbinsE;
    VecVecInt b_zbinsE_cut;

    void fill_cluster_branches(TrkrClusLoc&, bool isPHG4=true);
    /* void pushback_verb_bins(TrkrDefs::cluskey, bool isPHG4=true); */
    
};

#endif  // FILLTRUTHRECOMATCHTREE_H