BmnLambdaEmbeddingQa.h

Go to the documentation of this file.

#ifndef BMNEMBQA_H
#define BMNEMBQA_H 1
 
#include "BmnLambdaEmbeddingDrawHistos.h"
 
#include <BmnEventHeader.h>
#include <BmnGemStripHit.h>
#include <BmnGemStripStationSet.h>
#include <BmnGlobalTrack.h>
#include <BmnInnerTrackerGeometryDraw.h>
#include <BmnKalmanFilter.h>
#include <BmnLambdaEmbeddingMonitor.h>
#include <BmnLambdaMisc.h>
#include <BmnNewFieldMap.h>
#include <BmnSiliconHit.h>
#include <BmnSiliconStationSet.h>
#include <BmnStripDigit.h>
#include <CbmMCTrack.h>
#include <CbmStsPoint.h>
#include <DstEventHeader.h>
#include <FairMCPoint.h>
#include <FairRunAna.h>
#include <FairTrackParam.h>
#include <TBox.h>
#include <TCanvas.h>
#include <TChain.h>
#include <TClonesArray.h>
#include <TColor.h>
#include <TGeoManager.h>
#include <TH2.h>
#include <TLegend.h>
#include <TMarker.h>
#include <TNamed.h>
#include <TPolyLine.h>
#include <TProfile.h>
#include <TProfile2D.h>
#include <TString.h>
#include <TStyle.h>
#include <TSystem.h>
#include <UniRun.h>
#include <fstream>
#include <iostream>
 
using namespace std;
 
enum detector
{
    GEM,
    SILICON
};
 
class BmnLambdaEmbeddingQa : public TNamed
{
  public:
    BmnLambdaEmbeddingQa();
    BmnLambdaEmbeddingQa(TString);
 
    virtual ~BmnLambdaEmbeddingQa();
 
    // Types of analysis available ...
    void DoDigiAnalysis();
    void DoEffAnalysis();
 
    void DoDrawEventsWithEmbeddedSignals();   // Draw events with embedded hits from Lambda decay products
    // void DoInnerTrackerEmbeddingEfficiency(); // Calculate efficiency of embedding
    void DoInnerTrackerRecoEfficiency();   // Calculate some params. showing hit reconstruction and tracking efficiency
 
    void DoDrawFoundTracks(Bool_t flag) { drawFoundTracks = flag; }
 
    // Setting prefix name if needed ...
    void SetPrefixName(TString name) { fPrefix = name; }
 
    void ShowLays(Int_t lay1, Int_t lay2)
    {
        aLaysToShow[0] = lay1;
        aLaysToShow[1] = lay2;
    }
 
    void SetSignalCuts(TString det, Double_t min, Double_t max)
    {
        Int_t idx = det.Contains("GEM") ? GEM : det.Contains("SILICON") ? SILICON : -1;
        fSignalWindow[idx][0] = min;
        fSignalWindow[idx][1] = max;
    }
 
    void DrawHistos6(
        TClonesArray*,
        TClonesArray*);   // Function to draw histos from BmnLambdaMisc::CheckStripOverlaps(), temporarily solution
 
  private:
    Int_t nInputs;
 
    FairRunAna* fRunAna;
 
    // Geometries
    BmnInnerTrackerGeometryDraw* geoms;
 
    // Files
    TString* fDigi;
    TString* fDst;
    TString* fEmb;
    TString* fPath;
 
    // Pointers to be addressed to corresponding objects if an analysis we do uses them
    BmnFieldMap* fMagField;
    BmnLambdaEmbeddingMonitor* fMon;
    DstEventHeader* hDst;
    BmnEventHeader* hDigi;
    TClonesArray* fGemDigits;
    TClonesArray* fSiliconDigits;
    TClonesArray* fGemHits;
    TClonesArray* fSiliconHits;
    TClonesArray* fSiliconTracks;
    TClonesArray* fGlobTracks;
    TClonesArray* fMcTracks;
    TClonesArray* fSiliconPoints;
    TClonesArray* fGemPoints;
 
    // Containers to be used for analysis
    map<Int_t, Double_t> fSilGemZ;   // <stat, Z-position map>
    // Maps to store <stat, (xy)-coordinates> for GEM and SILICON (MC--> 0, RECO --> 1)
    map<Int_t, vector<pair<Double_t, Double_t>>> gems[2];
    map<Int_t, vector<pair<Double_t, Double_t>>> silicons[2];
 
    // Histos to be shown
    TH2F* hXzMC;
    TH2F* hXzReco;
    TH2F* hYzMC;
    TH2F* hYzReco;
    TH2F** hGemXYProfiles;
    TH2F** hSiliconXYProfiles;
    TH2F* hXzRecoFromTracks;
 
    TH1F* hProtonMomentaEmb;
    TH1F* hPionMomentaEmb;
 
    TH1F* hProtonMomentaReco;
    TH1F* hPionMomentaReco;
 
    TH1F* hProtonNhitsEmb;
    TH1F* hPionNhitsEmb;
 
    TH1F* hProtonNhitsReco;
    TH1F* hPionNhitsReco;
 
    TH1F* hProtonNhitsRecoAll;
    TH1F* hPionNhitsRecoAll;
 
    TProfile* pEffProton;
    TProfile* pEffPion;
 
    Int_t aLaysToShow[2];   // array of strips lays to be shown in histos
 
    TH2F****
        hStripChannel;   // Strip vs. Chan for main part of big zone and common ADC [main, common][stat][Right or Left]
 
    Double_t fSignalWindow[2][2];   // two detectors, two limits
 
    // Options to be reset by user if necessary
    Bool_t drawFoundTracks;
 
  private:
    Bool_t wasReconstructed(BmnGlobalTrack*, Double_t fracLostOrWrongConnectedHits, Double_t momResCut);
    Bool_t wasReconstructed(BmnGlobalTrack*, Double_t momResCut);
 
    // Int_t DefineSiliconStatByZpoint(Double_t);
    // Int_t DefineSiliconModuleByStatAndXY(TBox***, Int_t, Double_t, Double_t);
    void DrawFoundTracks();
 
    // Int_t GetStripBoundValues(TString, Int_t&, Int_t&);
 
    // Additional prefix, if necessary, in output file names...
    TString fPrefix;
 
    void DrawHistos1();   // Histos taken from DoDrawEventsWithEmbeddedSignals-analysis
    // void DrawHistos3(); // Histos taken from DoInnerTrackerRecoEfficiency-analysis
 
    BmnLambdaEmbeddingDrawHistos* histoMan;
 
    ClassDef(BmnLambdaEmbeddingQa, 1)
};
 
#endif