CbmMvdSensorFindHitTask.h

Go to the documentation of this file.

// -------------------------------------------------------------------------
// -----                    CbmMvdSensorFindHitTask header file            -----
// -----                   Created 11/09/13 P.Sitzmann                 -----
// -------------------------------------------------------------------------
 
 
#ifndef CBMMVDSENSORFINDHITTASK_H
#define CBMMVDSENSORFINDHITTASK_H 1
 
#include "CbmMvdDigi.h"
#include "CbmMvdPoint.h"
#include "FairTask.h"
 
#include "TArrayS.h"
#include "TCanvas.h"
#include "TObjArray.h"
#include "TRefArray.h"
 
 
#include "TH1.h"
#include "TH1F.h"
#include "TH2.h"
#include "TMath.h"
#include "TRandom3.h"
#include "TStopwatch.h"
#include "TString.h"
#include <list>
#include <map>
#include <utility>
#include <vector>
 
#include "CbmMvdSensor.h"
#include "CbmMvdSensorTask.h"
 
class TClonesArray;
class TRandom3;
class CbmMvdPileupManager;
 
class CbmMvdSensorFindHitTask : public CbmMvdSensorTask {
 
public:
  CbmMvdSensorFindHitTask();
 
 
  CbmMvdSensorFindHitTask(Int_t mode);
 
 
  virtual ~CbmMvdSensorFindHitTask();
 
 
  void ExecChain();
  void Exec();
 
  void InitTask(CbmMvdSensor* mySensor);
 
  virtual void SetInputDigi(CbmMvdDigi* digi) {
    new ((*fInputBuffer)[fInputBuffer->GetEntriesFast()])
      CbmMvdDigi(*((CbmMvdDigi*) digi));
    inputSet = kTRUE;
  }
 
  //Double_t GetSigmaX()        const { return fSigmaX;     };
 
  void SetSigmaNoise(Double_t sigmaNoise, Bool_t addNoise) {
    fSigmaNoise = sigmaNoise;
    fAddNoise   = addNoise;
  }
  void SetSeedThreshold(Double_t seedCharge) { fSeedThreshold = seedCharge; }
  void SetNeighbourThreshold(Double_t neighCharge) {
    fNeighThreshold = neighCharge;
  }
  void SetAdcDynamic(Int_t adcDynamic) { fAdcDynamic = adcDynamic; };
  void SetAdcOffset(Int_t adcOffset) { fAdcOffset = adcOffset; };
  void SetAdcBits(Int_t adcBits) { fAdcBits = adcBits; };
  Int_t GetAdcCharge(Float_t charge);
 
  void SetHitPosErrX(Double_t errorX) { fHitPosErrX = errorX; }
  void SetHitPosErrY(Double_t errorY) { fHitPosErrY = errorY; }
  void SetHitPosErrZ(Double_t errorZ) { fHitPosErrZ = errorZ; }
  void ShowDebugHistograms() { fShowDebugHistos = kTRUE; }
 
 
  //protected:
protected:
  // ----------   Protected data members  ------------------------------------
 
  Int_t fAdcDynamic;
  Int_t fAdcOffset;
  Int_t fAdcBits;
  Int_t fAdcSteps;
  Float_t fAdcStepSize;
 
  TClonesArray* fDigis;
  TClonesArray* fHits;
  TClonesArray* fClusters;
 
 
  TObjArray* fPixelChargeHistos;
  TObjArray* fTotalChargeInNpixelsArray;
 
  // Debug Histograms
  TH1F* fResolutionHistoX;
  TH1F* fResolutionHistoY;
  TH1F* fResolutionHistoCleanX;
  TH1F* fResolutionHistoCleanY;
  TH1F* fResolutionHistoMergedX;
  TH1F* fResolutionHistoMergedY;
  TH2F* fBadHitHisto;
  Float_t* fGausArray;
  Int_t fGausArrayIt;
  Int_t fGausArrayLimit;
 
  std::map<std::pair<Int_t, Int_t>, Int_t> fDigiMap;
  std::map<std::pair<Int_t, Int_t>, Int_t>::iterator fDigiMapIt;
 
 
  TH2F* h;
  TH2F* h3;
  TH1F* h1;
  TH1F* h2;
  TH1F* Qseed;
  TH1F* fFullClusterHisto;
 
  TCanvas* c1;
 
private:
  Int_t fNEvent;
  Int_t fMode;
  Int_t fCounter;
  Double_t fSigmaNoise;
  Double_t fSeedThreshold;
  Double_t fNeighThreshold;
  Bool_t fShowDebugHistos;
  Bool_t fUseMCInfo;
  Bool_t inputSet;
 
 
  Double_t fLayerRadius;
  Double_t fLayerRadiusInner;
  Double_t fLayerPosZ;
  Double_t fHitPosX;
  Double_t fHitPosY;
  Double_t fHitPosZ;
  Double_t fHitPosErrX;
  Double_t fHitPosErrY;
  Double_t fHitPosErrZ;
 
  TString fBranchName;
 
  Int_t fDigisInCluster;
  static const Short_t fChargeArraySize =
    7;  //must be an odd number >3, recommended numbers= 5 or 7
 
  Bool_t fAddNoise;
 
 
 
  // -----   Private methods   ---------------------------------------------
 
 
  void Reset();
 
  void Finish();
 
 
  InitStatus ReInit();
 
 
  Int_t GetMvdGeometry();
 
 
  void SetMvdGeometry(Int_t detId);
  void AddNoiseToDigis(CbmMvdDigi* digi);
  // void GenerateFakeDigis(Double_t pixelSizeX, Double_t pixelSizeY);
  void CheckForNeighbours(std::vector<Int_t>* clusterArray,
                          Int_t clusterDigi,
                          TArrayS* pixelUsed);
 
  void
  CreateHit(std::vector<Int_t>* clusterArray, TVector3& pos, TVector3& dpos);
  void ComputeCenterOfGravity(std::vector<Int_t>* clusterArray,
                              TVector3& pos,
                              TVector3& dpos);
  void UpdateDebugHistos(std::vector<Int_t>* clusterArray,
                         Int_t seedIndexX,
                         Int_t seedIndexY);
 
private:
  CbmMvdSensorFindHitTask(const CbmMvdSensorFindHitTask&);
  CbmMvdSensorFindHitTask operator=(const CbmMvdSensorFindHitTask&);
 
  ClassDef(CbmMvdSensorFindHitTask, 1);
};
 
 
#endif
 
/* fGausArray=new Float_t[fGausArrayLimit];
    for(Int_t i=0;i<fGausArrayLimit;i++){fGausArray[i]=gRandom->Gaus(0, fSigmaNoise);};
    fGausArrayIt=0;
*/