BSEmcFlagSplitOffs.h

Go to the documentation of this file.

#ifndef BSEMCFLAGSPLITOFFS_HH
#define BSEMCFLAGSPLITOFFS_HH


#include "BSEmcCluster.h"
#include "BSEmcClusteringData.h"
#include "PndContainerI.h"
#include "BSEmcGeoNeighbouringRelationPar.h"
#include "BSEmcSplitOffPar.h"
#include "PndParameterRegister.h"
#include "BSEmcCrystalPositionPar.h"
#include "PndOutputContainerI.h"
#include "PndProcess.h"
#include <vector>
#include "TF1.h"

class FairParGenericSet;

class BSEmcDigi;
class BSEmcCluster;
class BSEmcSubCluster;
class BSEmcRecoHit;


class BSEmcFlagSplitOffs : public PndProcess<BSEmcClusteringData>
{
public:
  BSEmcFlagSplitOffs();
  virtual ~BSEmcFlagSplitOffs();
  virtual void SetDetectorName(const std::string &t_detectorName);
  virtual void SetupParameters(const PndParameterRegister *t_parameterRegister) ;
  virtual void SetData(BSEmcClusteringData *t_pdata) ;
  virtual void Process() ;

  struct Cluster_Info {
    const std::vector<const BSEmcDigi *> &DigiArray;
    const std::vector<const BSEmcCluster *> &ClusterArray;
    const BSEmcCluster* cluster;
    const std::vector<const BSEmcSubCluster*> &SubclusterArray;

    Cluster_Info(const std::vector<const BSEmcDigi *> &digis,const std::vector<const BSEmcCluster *> &clusters, const BSEmcCluster* current_cluster, const std::vector<const BSEmcSubCluster*> &subclusters) 
    : DigiArray(digis),ClusterArray(clusters), cluster(current_cluster),SubclusterArray(subclusters)  {}
  };

  std::map<int,std::vector<const BSEmcSubCluster*>> CreateClusterMap();

  const BSEmcSubCluster* FindSubCluster(const BSEmcDigiInfo_t &max, std::vector<const BSEmcSubCluster*> subcluster_array);

  std::vector<BSEmcDigiInfo_t> FindSecondaryMaxima(const BSEmcDigiInfo_t &centralmax, const std::vector<BSEmcDigiInfo_t> &maxima);

  double CalculateShowerMass(const std::vector<const BSEmcDigi *> &DigiArray, const BSEmcCluster* cluster);

  bool ShowerMassCut(const std::vector<const BSEmcDigi *> &DigiArray, const BSEmcCluster* cluster);

  double CalculateHighPEDShowerMass(const std::vector<const BSEmcDigi *> &DigiArray, const BSEmcSubCluster* subcluster1,const BSEmcSubCluster* subcluster2);

  bool HighPEDCut(const std::vector<const BSEmcDigi *> &DigiArray, const BSEmcSubCluster* subcluster1,const BSEmcSubCluster* subcluster2);

  double CalculateDistance(const std::vector<const BSEmcDigi *> &DigiArray, const BSEmcDigiInfo_t &firstmax, const BSEmcDigiInfo_t &secondmax);

  double FindSmallestDistance(const std::vector<const BSEmcCluster *> &ClusterArray, const std::vector<const BSEmcDigi *> &DigiArray, const BSEmcDigiInfo_t &max);

  double FindEnergyClosestCluster(const std::vector<const BSEmcCluster *> &ClusterArray, const std::vector<const BSEmcDigi *> &DigiArray, const BSEmcDigiInfo_t &max);

  bool RatioCut(const double ratio, const double distance);

  void FlagHighPED(Cluster_Info &ClusterInfo, std::vector<int> &SplitOffSubclusterIds);

  void Flag2PED(Cluster_Info &ClusterInfo, std::vector<int> &SplitOffSubclusterIds);

  void Flag1PED(Cluster_Info &ClusterInfo, std::vector<int> &SplitOffSubclusterIds);

private:

  std::string fPositionParName{""};
  std::string fNeighbouringRelationParName{""};
  std::string fSplitOffParName{""};
  BSEmcGeoNeighbouringRelationPar *fNeighbouringRelationPar{nullptr};
  const BSEmcCrystalPositionPar* fPositionPar;
  BSEmcSplitOffPar *fSplitOffPar{nullptr};
  TString fSubClusterBranchName{""};
  
  PndContainerI<BSEmcDigi> *fDigiArray{nullptr};            
  PndContainerI<BSEmcCluster> *fClusterArray{nullptr};       
  PndContainerI<BSEmcSubCluster> *fSubClusterArray{nullptr}; 
  PndOutputContainerI<BSEmcRecoHit> *fRecoHitArray{nullptr};

  std::unique_ptr<TF1> f1PEDFunction;
  std::unique_ptr<TF1> f2PEDFunction;
  std::unique_ptr<TF1> fHighPEDFunction;

  ClassDef(BSEmcFlagSplitOffs,1);

};

#endif /*BSEMCFLAGSPLITOFFS_HH*/