CbmStsFindTracksQa.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                  CbmStsFindTracksQa header file               -----
// -----                  Created 11/01/06  by V. Friese               -----
// -------------------------------------------------------------------------
 
/** CbmStsFindTracksQa.h
 *@author V.Friese <v.friese@gsi.de>
 **
 ** Quality check task for CbmStsFindTracks
 **/
 
#ifndef CBMSTSTRACKFINDERQA_H
#define CBMSTSTRACKFINDERQA_H 1
 
#include "FairTask.h"
#include "TStopwatch.h"
#include "TVector3.h"
 
#include <map>
 
class TCanvas;
class TClonesArray;
class TH1;
class TH1F;
class TList;
class FairGeoPassivePar;
class CbmGeoStsPar;
 
class CbmStsFindTracksQa : public FairTask
{
 
  public:
    /** Default constructor **/
    CbmStsFindTracksQa(Int_t iVerbose = 1);
 
    /** Standard constructor
     *@param minHits     Minimal number of StsHits for considered MCTracks
     *@param quota       True/all hits for track to be considered reconstructed
     *@param iVerbose    Verbosity level
     **/
    CbmStsFindTracksQa(Int_t minHits, Double_t quota, Int_t iVerbose);
 
    /** Destructor **/
    virtual ~CbmStsFindTracksQa();
 
    /** Set parameter containers **/
    virtual void SetParContainers();
 
    /** Initialisation **/
    virtual InitStatus Init();
 
    /** Reinitialisation **/
    virtual InitStatus ReInit();
 
    /** Execution **/
    virtual void Exec(Option_t* opt);
 
  private:
    /** Finish **/
    virtual void Finish();
 
    /** Read the geometry parameters **/
    InitStatus GetGeometry();
 
    /** Create histograms **/
    void CreateHistos();
 
    /** Reset histograms and counters **/
    void Reset();
 
    /** Fill a map from MCTrack index to number of corresponding StsHits **/
    void FillHitMap();
 
    /** Fill a map from MCTrack index to matched StsTrack index
     *@param nRec  Number of reconstructed tracks (return)
     *@param nGhosts  Number of ghost tracks (return)
     *@param nClones  Number of clone tracks (return)
     **/
    void FillMatchMap(Int_t& nRec, Int_t& nGhosts, Int_t& nClones);
 
    /** Divide histograms (reco/all) with correct error for the efficiency
     *@param histo1  reconstructed tracks
     *@param histo2  all tracks (normalisation)
     *@param histo3  efficiency
     **/
    void DivideHistos(TH1* histo1, TH1* histo2, TH1* histo3);
 
    /** Map from MCTrack index to number of attached StsHits **/
    std::map<Int_t, Int_t> fHitMap;
 
    /** Map from MCTrack index to matched StsTrack index **/
    std::map<Int_t, Int_t> fMatchMap;
 
    /** Map from MCTrack index to percentage of matched hits **/
    std::map<Int_t, Double_t> fQualiMap;
 
    /** Pointers to data arrays **/
    TClonesArray* fMCTracks;    // MCtrack
    TClonesArray* fStsPoints;   // StsPoints
    TClonesArray* fStsHits;     // StsHits
    TClonesArray* fStsTracks;   // StsTrack
    TClonesArray* fMatches;     // StsTrackMatch
 
    /** Geometry parameters **/
    FairGeoPassivePar* fPassGeo;   // Passive geometry parameters
    CbmGeoStsPar* fStsGeo;         // STS geometry parameters
    TVector3 fTargetPos;           // Target centre position
    Int_t fNStations;              // Number of STS stations
 
    /** Task parameters **/
    Int_t fMinHits;    // Minimal number of StsHits for considered MCTrack
    Double_t fQuota;   // True/all hits for track to be considered reconstructed
 
    /** Histograms **/
    TH1F *fhMomAccAll, *fhMomRecAll, *fhMomEffAll;      // eff. vs. p, all
    TH1F *fhMomAccPrim, *fhMomRecPrim, *fhMomEffPrim;   // eff. vs. p, vertex
    TH1F *fhMomAccSec, *fhMomRecSec, *fhMomEffSec;      // eff. vs. p, non-vertex
    TH1F *fhNpAccAll, *fhNpRecAll, *fhNpEffAll;         // eff. vs. np, all
    TH1F *fhNpAccPrim, *fhNpRecPrim, *fhNpEffPrim;      // eff. vs. np, vertex
    TH1F *fhNpAccSec, *fhNpRecSec, *fhNpEffSec;         // eff. vs. np, non-vertex
    TH1F *fhZAccSec, *fhZRecSec, *fhZEffSec;            // eff. vs. z, non-vertex
    TH1F *fhNhClones, *fhNhGhosts;                      // # hits of clones and ghosts
 
    /** List of histograms **/
    TList* fHistoList;
 
    /** Counters **/
    Int_t fNAccAll, fNAccPrim, fNAccRef, fNAccSec;
    Int_t fNRecAll, fNRecPrim, fNRecRef, fNRecSec;
    Int_t fNGhosts, fNClones;
    Int_t fNEvents;       /** Number of events with success **/
    Int_t fNEventsFailed; /** Number of events with failure **/
    Double_t fTime;       /** Total real time used for good events **/
 
    /** Timer **/
    TStopwatch fTimer;
 
    CbmStsFindTracksQa(const CbmStsFindTracksQa&);
    CbmStsFindTracksQa operator=(const CbmStsFindTracksQa&);
 
    ClassDef(CbmStsFindTracksQa, 1);
};
 
#endif