BSEmcMergePreclusters.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 BSEMCMERGEPRECLUSTERS_HH
#define BSEMCMERGEPRECLUSTERS_HH

#include <memory>
#include <string>
#include <vector>

#include "Rtypes.h"
#include "RtypesCore.h"

#include "PndContainerI.h"
#include "PndProcess.h"

#include "BSEmcCluster.h"

#include "BSEmcDigi.h"
#include "BSEmcGeoNeighbouringRelationPar.h"
#include "BSEmcPositionAlgo.h"
#include "BSEmcPrecluster.h"

class BSEmcCluster;
class BSEmcCrystalPositionPar;
class BSEmcDigi;
class BSEmcGeoNeighbouringRelationPar;
class BSEmcPrecluster;
class PndParameterRegister;
class TBuffer;
class TClass;
class TMemberInspector;
struct BSEmcClusteringData;
template <class T>
class PndContainerI;
template <class T>
class PndMutableContainerI;

class BSEmcMergePreclusters : public PndProcess {
 public:
  BSEmcMergePreclusters();

  virtual ~BSEmcMergePreclusters();

  virtual void SetDetectorName(const std::string &t_detectorName) /*override*/;
  virtual void RequestDataContainer(PndContainerRegister *t_register) /*override*/;
  virtual void GetDataContainer(PndContainerRegister *t_register) /*override*/;
  virtual void SetupParameters(const PndParameterRegister *t_paramRegister) /*override*/;
  virtual void Process() /*override*/;
  std::vector<BSEmcCluster *> MergePreclusters(const std::vector<const BSEmcDigi *> &t_digiCont, const std::vector<BSEmcPrecluster *> &t_preclusterCont);

  void SetPositionMethod(Int_t method)
  {
    fPosMethod = method;
  } 
  void SetNeighbourMethod(Int_t nbmethod)
  {
    fNbMethod = nbmethod;
  } 
  void SetClusteringParName(const std::string &t_parName) { fClusteringParName = t_parName; }
  void SetClusterPropertiesParName(const std::string &t_parName) { fClusterPropertiesParName = t_parName; }
  void SetPositionParName(const std::string &t_parName) { fPositionParName = t_parName; }
  void SetNeighbouringRelationParName(const std::string &t_parName) { fNeighbouringRelationParName = t_parName; }

  void SetDigiBranchName(const TString &t_branchname) { fDigiBranchname = t_branchname; }
  void SetPreclusterBranchName(const TString &t_branchname) { fPreclusterBranchname = t_branchname; }
  void SetClusterBranchName(const TString &t_branchname) { fClusterBranchname = t_branchname; }

 protected:
  std::vector<Int_t> FindNeighbouringPrecluster(const std::vector<BSEmcPrecluster *> &t_preclusterCont);
  std::vector<Int_t> ClusterNeigbours(const std::vector<Int_t> &t_neighbours, const std::vector<BSEmcPrecluster *> &t_preclusterCont);

  std::vector<BSEmcCluster *> MergePreclustersIntoCluster(const std::vector<Int_t> &t_clusterNrForPrecluster, const std::vector<BSEmcPrecluster *> &t_preclusterCont);

  virtual void FinishClusters(const std::vector<const BSEmcDigi *> &t_digiCont, const std::vector<BSEmcCluster *> &t_clusterCont);
  void FinishCluster(const std::vector<const BSEmcDigi *> &t_digiCont, BSEmcCluster *tmpcluster);

 private:
  TString fDigiBranchname{""};
  TString fPreclusterBranchname{""};
  TString fClusterBranchname{""};

  PndContainerI<BSEmcDigi> *fDigiArray{nullptr};
  PndMutableContainerI<BSEmcPrecluster> *fPreclusterArray{nullptr};
  PndMutableContainerI<BSEmcCluster> *fClusterArray{nullptr};
  std::string fClusteringParName{""};
  std::string fClusterPropertiesParName{""};
  std::string fPositionParName{""};
  std::string fNeighbouringRelationParName{""};
  BSEmcGeoNeighbouringRelationPar *fNeighbouringRelationPar{nullptr};

  BSEmcCrystalPositionPar *fPositionPar{nullptr};
  std::unique_ptr<BSEmcPositionAlgo> fPositionProcess{new BSEmcPositionAlgo};
  Int_t fPosMethod;
  Int_t fNbMethod;
  Int_t evtCounter;
  Int_t precCounter;
  Int_t fNrOfPres;
  Int_t nMrgProg;
  Int_t nTotClusters;
  Int_t fRemovedClusters;
  Int_t nTotDigis;
  Int_t wrongConnection;
  Double_t CNtotRtime;
  Double_t CNtotCtime;

  Int_t fClusterNr{0};
  ClassDef(BSEmcMergePreclusters, 2)
};
#endif /*BSEMCMERGEPRECLUSTERS_HH*/