BSEmcFlagSplitOffs.h

Go to the documentation of this file.

#ifndef BSEMCFLAGSPLITOFFS_HH
#define BSEMCFLAGSPLITOFFS_HH

#include "BSEmcCluster.h"

#include "BSEmcCrystalPositionPar.h"
#include "BSEmcGeoNeighbouringRelationPar.h"
#include "BSEmcSplitOffPar.h"
#include "PndContainerI.h"
#include "PndMutableContainerI.h"
#include "PndParameterRegister.h"
#include "PndProcess.h"
#include "TF1.h"
#include <vector>

class FairParGenericSet;

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

class BSEmcFlagSplitOffs : public PndProcess {
 public:
  BSEmcFlagSplitOffs();
  virtual ~BSEmcFlagSplitOffs();
  virtual void SetDetectorName(const std::string &t_detectorName);
  virtual void SetupParameters(const PndParameterRegister *t_parameterRegister);
  virtual void RequestDataContainer(PndContainerRegister *t_register) /*override*/;
  virtual void GetDataContainer(PndContainerRegister *t_register) /*override*/;
  virtual void Process() /*override*/;

  void SetDigiBranchName(const TString &t_branchname) { fDigiBranchname = t_branchname; }
  void SetClusterBranchName(const TString &t_branchname) { fClusterBranchname = t_branchname; }
  void SetSubClusterBranchName(const TString &t_branchname) { fSubClusterBranchname = t_branchname; }
  void SetRecoHitBranchName(const TString &t_branchname) { fRecoHitBranchname = t_branchname; }
  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 fDigiBranchname{""};
  TString fClusterBranchname{""};
  TString fSubClusterBranchname{""};
  TString fRecoHitBranchname{""};

  PndContainerI<BSEmcDigi> *fDigiArray{nullptr};
  PndContainerI<BSEmcCluster> *fClusterArray{nullptr};
  PndContainerI<BSEmcSubCluster> *fSubClusterArray{nullptr};
  PndMutableContainerI<BSEmcRecoHit> *fRecoHitArray{nullptr};

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

  ClassDef(BSEmcFlagSplitOffs, 1);
};

#endif /*BSEMCFLAGSPLITOFFS_HH*/