beamtime/cern-oct11/go4/RICH/TRICHProc.h (r4864/r3566)

Go to the documentation of this file.

#ifndef TRICHPROC_H
#define TRICHPROC_H

#include "TRocProc.h"
#include "TRICHEvent.h"
#include "TTriglogEvent.h"
#include "TBeamMonitorEvent.h"
#include "TMbsCrateEvent.h"
#include "TRICHParam.h"
#include "TGo4Picture.h"
#include "TH1.h"
#include "TH2.h"
#include "TGo4WinCond.h"
#include "TGo4PolyCond.h"
#include "TCBMBeamtimeProc.h"
#include "MAPMT_Cell.h"
#include "TEpicsEvent.h"
#include "TFiberHodEvent.h"

//ring fitter
class CbmRichRingFitterCOP;
class CbmRichRingFitterEllipseTau;
class CbmRichRingFinderHoughImpl;
class CbmRichRingLight;
class CbmRichHitLight;
class TEllipse;
class TLatex;


class TRICHProc : public TCBMBeamtimeProc {
   friend class TCBMBeamtimeProc; // may call protected process methods
   
   public:
   TRICHProc();
   TRICHProc(const char* name);
   virtual ~TRICHProc();
   
   virtual void InitEvent(TGo4EventElement*); 
   
   /* this is the main processing function here*/
   virtual void FinalizeEvent();
   
   
protected:
   // local subroutines      
   bool ReadMapmtGeometry(const char* fname);
   void ProcessRICH();
   void FillCircleFitHistograms(
                   Int_t hIndex,
                   CbmRichRingLight* ring,
                   int nofEventHits);

   void FillEllipseFitHistograms(
                   Int_t hIndex,
                   CbmRichRingLight* ring,
                   int nofEventHits);

   void DoAnalysis(
                   const vector<CbmRichHitLight>&,
                   bool isElectron,
                   bool isPion,
                   bool isMuon);

   void DrawRingHits(
                   CbmRichRingLight* ring,
                   Int_t sedNum);

   void DrawEventHits(
                   const vector<CbmRichHitLight>&,
                   Int_t sedNum);

   void DrawCircles(
                   const vector<CbmRichRingLight*>& rings,
                   Int_t sedNum);

   void DrawEllipses(
                   const vector<CbmRichRingLight*>& rings,
                   Int_t sedNum);
   
   void SaveParameters();

   void SetParticleIdentificationConditions();

   TRICHParam * fPar;
   
   TRocEvent* fRocInputEvent;
   TRICHEvent* fOutputEvent;
   TFiberHodEvent* fHodo1;
   TTriglogEvent* fTriglogEvent;
   TBeamMonitorEvent* fBeamEvent;
   TMbsCrateEvent* fMbsEvent;
   TEpicsEvent* fEpicsEvent;
   
   //ring finders and fitters
   CbmRichRingFitterCOP* fRingFitter;
   CbmRichRingFitterEllipseTau* fEllipseFitter;
   CbmRichRingFinderHoughImpl* fRingFinder;
   
   TH2* hEntries2dIntegral;
   TH2* hEntries2dMean;
   TH2* hEntriesperPMT;
   TH1* hADCall;
   
   TH1* hHittime;
   TH1* hHitMultperPMT[17];
   
   TH2* hHodXYscan;
   Int_t fNofHodWindowsX;
   Int_t fNofHodWindowsY;
   
   //Circle fitter histograms
   //[0] is electron, [1] is pion, [2] is muon, [3] is all
   TH1* hCircleFitRadius[4];
   TH2* hCircleFitCenter[4];
   TH1* hCircleFitChi2[4];
   TH1* hCircleFitDR[4];
   
   //ellipse fitter histograms
   //[0] is electron, [1] is pion, [2] is muon, [3] is all
   TH1* hEllipseFitAaxis[4];
   TH1* hEllipseFitBaxis[4];
   TH1* hEllipseFitPhi[4];
   TH1* hEllipseFitBoverA[4];
   TH2* hEllipseFitCenter[4];
   TH1* hEllipseFitChi2[4];
   
   // efficiency historams vs. nof hits
   //[0] is electron, [1] is pion, [2] is muon, [3] is all
   TH1* hNofHitsEvAll[4];
   TH1* hNofHitsEvRingFinder[4];
   TH1* hNofHitsEvCircleFit[4];
   TH1* hNofHitsEvEllipseFit[4];
   TH1* hNofHitsEvRingFinderEff[4];
   TH1* hNofHitsEvCircleFitEff[4];
   TH1* hNofHitsEvEllipseFitEff[4];

   TH1* hNofHitsRingFinder[4]; // number of hits in found ring after ring finder
   
   TH1* hNofHitsInHodo; //Number of hits in hodoscope
   TH2* hHodoADCvsNofHitInEv; // Hodoscope ADC vs number of hits in ring
   TH2* hNotAssignedHitsXY; // XY position of the hits not assigned to the ring

   //Number of entries 2D
   //[0] is electron, [1] is pion, [2] is muon, [3] is all
   TH2* hEntries2d[4];

   //Number of found rings per event
   TH1* hNofFoundRings;

   TH1* hTimespread;
   
   // additional histos for hodoscope correlation
   TH2* hMAPMT_hcor2d_s1;
   TH2* hMAPMT_hcor2d_s2;
   TH2* hMAPMT_hcor2d_s3;
   TH2* hMAPMT_hcor2d_s4;
   TH2* hMAPMT_hcor2d_s5;
   
   // additional histos for correlation with gas purity
   TH2* hMAPMT_HitMultvsO2;
   
   // conditions on ADC, time
   TGo4WinCond* fMAPMT_cond;
   TGo4WinCond* fMAPMT_tcond;
   
   // conditions for electons, muons and pions
   TGo4PolyCond  *fElectronCondc1c2;  
   TGo4PolyCond  *fMuonCondc1c2;      
   TGo4PolyCond  *fPionCondc1c2;      
   
   // event display 
   vector<TGo4Picture*> fCirclePicture;
   vector<vector<TEllipse*> > fCircle;
   vector<TLatex*> fCircleText;
   vector<vector<TEllipse*> > fEllipse;
   vector<TLatex*> fEllipseText;
   vector<vector<TEllipse*> > fRingHits;
   vector<vector<TEllipse*> > fEventHits;
   vector<TH2*> hSED;
   const static int fMaxNofHitsInEventDraw = 100;
   
   // variables and data structures
   MAPMT_Cell fMAPMTCells[NUM_MAPMP_CELLS];
   
   int fEventNr;
   int fEventTime;
   Double_t fO2level;
   Double_t fH2Olevel;
   
   int fNrSED;
   ClassDef(TRICHProc,1)   
};

#endif //TUNPACKPROCESSOR_H