ClusterizerSpec.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 O2_EMCAL_CLUSTERIZER_SPEC
#define O2_EMCAL_CLUSTERIZER_SPEC
 
#include <vector>
 
#include "DataFormatsEMCAL/Cluster.h"
#include "DataFormatsEMCAL/TriggerRecord.h"
#include "EMCALBase/Geometry.h"
#include "EMCALReconstruction/Clusterizer.h"
#include "Framework/DataProcessorSpec.h"
#include "Framework/Task.h"
#include "TStopwatch.h"
 
namespace o2
{
 
namespace emcal
{
 
namespace reco_workflow
{
 
template <class InputType>
class ClusterizerSpec : public framework::Task
{
 public:
  ClusterizerSpec() : framework::Task(){};
 
  ~ClusterizerSpec() override = default;
 
  void init(framework::InitContext& ctx) final;
 
  void run(framework::ProcessingContext& ctx) final;
  void endOfStream(framework::EndOfStreamContext& ec) final;
 
 private:
  o2::emcal::Clusterizer<InputType> mClusterizer;                               
  o2::emcal::Geometry* mGeometry = nullptr;                                     
  std::vector<o2::emcal::Cluster>* mOutputClusters = nullptr;                   
  std::vector<o2::emcal::ClusterIndex>* mOutputCellDigitIndices = nullptr;      
  std::vector<o2::emcal::TriggerRecord>* mOutputTriggerRecord = nullptr;        
  std::vector<o2::emcal::TriggerRecord>* mOutputTriggerRecordIndices = nullptr; 
  TStopwatch mTimer;
};
 
framework::DataProcessorSpec getClusterizerSpec(bool useDigits);
 
} // namespace reco_workflow
 
} // namespace emcal
 
} // namespace o2
 
#endif // O2_EMCAL_CLUSTERIZER_SPEC