PndEmcDistributedClustering.h

Go to the documentation of this file.

//****************************************************************************
//*                   This file is part of PandaRoot.                        *
//*                                                                          *
//*            PandaRoot is distributed under the terms of the               *
//*              GNU General Public License (GPL) version 3,                 *
//*                 copied verbatim in the file "LICENSE".                   *
//*                                                                          *
//*  Copyright (C) 2006 - 2024 FAIR GmbH and copyright holders of PandaRoot  *
//*     The copyright holders are listed in the file "COPYRIGHTHOLDERS".     *
//*               The authors are listed in the file "AUTHORS".              *
//****************************************************************************

//-----------------------------------------------------------------------
// File and Version Information:
// $Id: $
// 
//---------------------------------------------------------------------
//#pragma once
#ifndef PNDEMCDISTRIBUTEDCLUSTERING_H
#define PNDEMCDISTRIBUTEDCLUSTERING_H

#include "FairTask.h"
#include <vector>       

#include "FairTSBufferFunctional.h"
#include "TStopwatch.h"
#include "TH1.h"
#include "TVector3.h"

class TClonesArray;
class TObjectArray;
class PndEmcDigi;
class PndEmcCluster;
class PndEmcGeoPar;
class PndEmcRecoPar;
class BinaryFunctor;

class PndEmcDistributedClustering : public FairTask
{
public:

  // Constructors

  PndEmcDistributedClustering(Int_t verbose=0, Bool_t storeclusters=kTRUE);

  // Destructor

  virtual ~PndEmcDistributedClustering( );

  virtual InitStatus Init();

  virtual void Exec(Option_t* opt);

  virtual void FinishTask();

  void SetStorageOfData(Bool_t val);    

  void StoreClusterBaseDigis(Bool_t val = kTRUE) { fStoreClusterBase = kTRUE; }
 
  void SetTimebunchCutTime(Double_t time) { fTimebunchCutTime = time; } 
  void SetClusterActiveTime(Double_t dt) { fClusterActiveTime = dt; }

  void SetClusterMinimumEnergy(Double_t minE) { fClusterEnergyCut = minE; }
  void EnableRemovalOfLowEnergyClusters(Bool_t enable) {fRemoveLowEclus = enable;} 
  void SetPositionMethod(Int_t method) { fPosMethod = method; } 
  void SetNeighbourMethod(Int_t nbmethod) { fNbMethod = nbmethod; } 
  void SetAutoDetermineTimecuts(Bool_t autodet) {fAutoDetermine = autodet;}

  void SetFactor(TVector3* factor) {fFactor = factor;}

protected:
  virtual void FinishPreclusters();
  void FinishPrecluster(PndEmcCluster* tmpcluster);
  virtual void FinishClusters();
  void FinishCluster(PndEmcCluster* tmpcluster, int);
  virtual void RemoveLowEnergyClusters();
  virtual void MapDistancesToSingleHitClusters();
  
private:
    TClonesArray* fDigiArray;
    TClonesArray* fPreclusterArray;

    TClonesArray* fClusterArray;    

    PndEmcGeoPar*     fGeoPar;       
    PndEmcRecoPar*    fRecoPar;      
    Double_t fDigiEnergyTresholdBarrel; 
    Double_t fDigiEnergyTresholdFWD; 
    Double_t fDigiEnergyTresholdBWD; 
    Double_t fDigiEnergyTresholdShashlyk;
    Double_t fClusterEnergyCut; 
    Double_t fTimebunchCutTime; // Defines how long clusters are kept open in timebased reconstruction
    Double_t fClusterActiveTime; 
    BinaryFunctor* fDigiFunctor; 

    std::vector<Double_t> fClusterPosParam;

    Int_t fNrOfEvents;
    Int_t fPosMethod;
    Int_t fNbMethod;
    Int_t evtCounter;
    Int_t digiCounter;
    Int_t fNrOfDigis;
    Int_t nDistProg;
    Int_t fRemovedClusters;
    Double_t DCtotRtime;
    Double_t DCtotCtime;
    Double_t CNtotRtime;
    Double_t CNtotCtime;
    bool tagSingleClus;
    bool printClusters;
    Bool_t fAutoDetermine;

    TH1I* hNdigis;
//  TH1I* hClusMultiplicity;
//  TH1D* hTimeDifference;
    TH1D* hSingle;
    TH1D* hDistancesToSingleHitClusters;
//  TH1D* hRadDiff;
//  TH1D* hPosDiff;

    Bool_t fStoreClusters;
    Bool_t fStoreClusterBase;   //restore digis in case of a timebased run
    Bool_t fRemoveLowEclus;

    virtual void SetParContainers();

    TStopwatch fTimer;
    int counter;
    double stoptime;

    TVector3* fFactor;

    ClassDef(PndEmcDistributedClustering, 2)
};
#endif // PNDEMCDISTRIBUTEDCLUSTERING_HH