BSEmcClusteringTask.h

Go to the documentation of this file.

//-----------------------------------------------------------------------
//-----------------------------------------------------------------------
// File and Version Information:
// $Id: $
// Software developed for the PANDA Detector at GSI.
// Author List:
//  Jan Zhong
//  Marcel Tiemens
//  Áron Kripkó
//---------------------------------------------------------------------
//#pragma once
#ifndef BSEMCCLUSTERINGTASK_HH
#define BSEMCCLUSTERINGTASK_HH

#include <string>
#include <vector>

#include "FairLogger.h"

#include "PndContainerI.h"
#include "PndInputContainer.h"
#include "PndOutputContainer.h"
#include "PndProcess.h"
#include "PndProcessTask.h"

#include "BSEmcCluster.h"
#include "BSEmcClusteringData.h"
#include "BSEmcClusteringPar.h"
#include "BSEmcDigi.h"
#include "BSEmcSubCluster.h"

class BSEmcClusteringTask : public PndProcessTask<BSEmcClusteringData> {
 public:
  // Constructors
  BSEmcClusteringTask(const std::string &t_detectorName = "Barrel", Bool_t t_storeclusters = kTRUE);
  // Destructor
  virtual ~BSEmcClusteringTask();

  void SetDigiBranchName(const std::string &t_branchname) { fDigiBranchname = t_branchname; }
  void SetDigiBranchAsMutable(Bool_t t_mutable) { fMutableDigiBranch = t_mutable; }
  void SetPreclusterBranchName(const std::string &t_branchname) { fPreclusterBranchname = t_branchname; }
  void SetClusterBranchName(const std::string &t_branchname) { fClusterBranchname = t_branchname; }
  void SetSubClusterBranchName(const std::string &t_branchname) { fSubClusterBranchname = t_branchname; }
  void SetRecoHitBranchName(const std::string &t_branchname) { fRecoHitBranchname = t_branchname; }

 protected:
  virtual void SetupDataArrays() /*override*/;
  virtual void PreProcessing(Option_t * /*unused*/) /*override*/;
  virtual void PostProcessing(Option_t * /*unused*/) /*override*/;

 private:
  PndContainerI<BSEmcDigi> *fDigiInput{nullptr};
  PndOutputContainer<BSEmcPrecluster> *fPreclusterOutput{nullptr};
  PndOutputContainer<BSEmcCluster> *fClusterOutput{nullptr};
  PndOutputContainer<BSEmcSubCluster> *fSubClusterOutput{nullptr};
  PndOutputContainer<BSEmcRecoHit> *fRecoHitOutput{nullptr};

  std::string fDigiBranchname{""};
  std::string fPreclusterBranchname{""};
  std::string fClusterBranchname{""};
  std::string fSubClusterBranchname{""};
  std::string fRecoHitBranchname{""};

  Bool_t fMutableDigiBranch{kFALSE};
  ClassDef(BSEmcClusteringTask, 1)
};

#endif /*BSEMCCLUSTERINGTASK_HH*/