df/d27/VtxTest_8h_source.html

#ifndef TRUTHCONVERSIONEVAL_H__

#define TRUTHCONVERSIONEVAL_H__


#include <fun4all/SubsysReco.h>

#include <calobase/RawClusterContainer.h>

#include <queue>


class PHCompositeNode;

class PHG4TruthInfoContainer;

class PHG4Particle;

class PHG4VtxPoint;

class Conversion;

class SvtxTrackEval;

class SvtxHitMap;

class SvtxHit;

class SvtxClusterMap;

class SvtxCluster;

class RawClusterContainer;

class TTree;

class TFile;

class SVReco;

class TrkrClusterContainer;


class VtxTest: public SubsysReco

{


  public:

    VtxTest(const std::string &name,unsigned int runnumber,

        int particleEmbed, int pythiaEmbed,bool makeTTree);

    ~VtxTest();

    int InitRun(PHCompositeNode*);

    int process_event(PHCompositeNode*);

    int End(PHCompositeNode*);

    const RawClusterContainer* getClusters()const;


  private:

    void doNodePointers(PHCompositeNode* topNode);

    std::queue<std::pair<int,int>> numUnique(std::map<int,Conversion>* map,SvtxTrackEval* trackEval,RawClusterContainer* mainClusterContainer);

    void findChildren(std::queue<std::pair<int,int>> missing,PHG4TruthInfoContainer* truthinfo);

    void processBackground(std::map<int,Conversion>* map,SvtxTrackEval* trackEval,TTree* tree);


    int get_embed(PHG4Particle* particle, PHG4TruthInfoContainer* truthinfo) const;

    float vtoR(PHG4VtxPoint* vtx)const;


    const static int s_kMAXParticles=200;

    const static int s_kMAXRecoMatch=20;

    const unsigned int _kRunNumber;

    const int _kParticleEmbed;

    const int _kPythiaEmbed;

    const bool _kMakeTTree;//< if false no TTrees are output

    int _runNumber;

    TFile *_f=NULL;

    TTree *_tree=NULL;

    TTree *_signalCutTree=NULL;

    TTree *_h_backgroundCutTree=NULL;

    TTree *_e_backgroundCutTree=NULL;

    RawClusterContainer *_mainClusterContainer;

    PHG4TruthInfoContainer *_truthinfo;

    TrkrClusterContainer* _clusterMap;

    SvtxHitMap *_hitMap;

    std::string _foutname;

    SVReco *_vertexer=NULL;


    int _b_event;

    int _b_nVtx;

    int _b_Tpair;

    int _b_Rpair;

    double _b_rVtx[s_kMAXParticles];

    bool _b_pythia[s_kMAXParticles];

    float _b_electron_pt[s_kMAXParticles];

    float _b_positron_pt[s_kMAXParticles];

    float _b_electron_reco_pt[s_kMAXParticles];

    float _b_positron_reco_pt[s_kMAXParticles];

    float _b_e_deta[s_kMAXParticles];

    float _b_e_dphi[s_kMAXParticles];

    float _b_parent_pt  [s_kMAXParticles];

    float _b_parent_eta [s_kMAXParticles];

    float _b_parent_phi [s_kMAXParticles];

    int   _b_grandparent_id [s_kMAXParticles];


    int   _b_nCluster [s_kMAXRecoMatch];

    int   _b_fLayer [s_kMAXRecoMatch];

    float _b_cluster_dphi [s_kMAXRecoMatch];

    float _b_cluster_deta [s_kMAXRecoMatch];

    float _b_Mcluster_prob[s_kMAXRecoMatch];

    float _b_Scluster_prob[s_kMAXRecoMatch];


    float _b_track_deta ;

    int _b_track_layer ;

    int _b_track_dlayer ;

    float _b_track_pT;

    float _b_ttrack_pT;

    double _b_approach  ;

    double _b_vtx_radius ;

    double _b_vtx_phi ;

    double _b_vtx_eta ;

    double _b_vtx_x ;

    double _b_vtx_y ;

    double _b_vtx_z ;

    double _b_tvtx_eta ;

    double _b_tvtx_x ;

    double _b_tvtx_y ;

    double _b_tvtx_z ;

    double _b_tvtx_radius ;

    double _b_tvtx_phi ;

    double _b_vtxTrack_dist;

    float _b_vtx_chi2;

    float _b_photon_m;

    float _b_photon_pT;

    float _b_cluster_prob;

    float _b_track_dphi;

    //bb stands for background branch

    float _bb_track_deta ;

    int _bb_track_layer ;

    int _bb_track_dlayer ;

    float _bb_track_pT;

    double _bb_approach ;

    double _bb_vtx_radius;

    double _bb_vtxTrack_dist;

    float _bb_vtx_chi2;

    float _bb_photon_m;

    float _bb_photon_pT;

    float _bb_cluster_prob;

    float _bb_track_dphi;

    int _bb_pid;

    RawClusterContainer _conversionClusters;


    const static int s_kTPCRADIUS=21; //in cm there is a way to get this from the simulation I should implement?

    float _kRAPIDITYACCEPT=1; //<acceptance rapidity currently hard coded to |1|

};


#endif // __TRUTHCONVERSIONEVAL_H__