ClusterizerTask.h

Go to the documentation of this file.

// Copyright 2019-2020 CERN and copyright holders of ALICE O2.
// See https://alice-o2.web.cern.ch/copyright for details of the copyright holders.
// All rights not expressly granted are reserved.
//
// This software is distributed under the terms of the GNU General Public
// License v3 (GPL Version 3), copied verbatim in the file "COPYING".
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
 
 
#ifndef ALICEO2_EMCAL_CLUSTERIZERTASK
#define ALICEO2_EMCAL_CLUSTERIZERTASK
 
#include "DataFormatsEMCAL/Cluster.h"
#include "DataFormatsEMCAL/Digit.h"
#include "DataFormatsEMCAL/Cell.h"
#include "EMCALBase/Geometry.h"
#include "EMCALReconstruction/DigitReader.h"
#include "EMCALReconstruction/ClusterizerParameters.h"
#include "EMCALReconstruction/Clusterizer.h"
 
namespace o2
{
 
namespace emcal
{
 
template <class InputType>
class ClusterizerTask
{
 
 public:
  ClusterizerTask(ClusterizerParameters* parameters);
  ~ClusterizerTask() = default;
 
  void init();
  void process(const std::string inputFileName, const std::string outputFileName);
  void setGeometry(Geometry* geometry) { mGeometry = geometry; }
  Geometry* getGeometry() { return mGeometry; }
 
 private:
  Clusterizer<InputType> mClusterizer;                                             
  Geometry* mGeometry = nullptr;                                                   
  std::unique_ptr<DigitReader<InputType>> mInputReader;                            
  std::vector<Cluster>* mClustersArray = nullptr;                                  
  std::vector<ClusterIndex>* mClustersInputIndices = nullptr;                      
  std::vector<o2::emcal::TriggerRecord>* mClusterTriggerRecordsClusters = nullptr; 
  std::vector<o2::emcal::TriggerRecord>* mClusterTriggerRecordsIndices = nullptr;  
  ClassDefNV(ClusterizerTask, 1)
};
} // namespace emcal
} // namespace o2
 
#endif /* ALICEO2_EMCAL_CLUSTERIZERTASK */