CellConverterSpec.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_CELLCONVERTER_SPEC
#define O2_EMCAL_CELLCONVERTER_SPEC
 
#include <vector>
 
#include "DataFormatsEMCAL/Cell.h"
#include "DataFormatsEMCAL/MCLabel.h"
#include "DataFormatsEMCAL/TriggerRecord.h"
#include "Framework/DataProcessorSpec.h"
#include "Framework/Task.h"
#include "EMCALBase/Geometry.h"
#include "EMCALReconstruction/CaloRawFitter.h"
#include "EMCALReconstruction/AltroHelper.h"
#include "EMCALWorkflow/CalibLoader.h"
#include "SimulationDataFormat/MCTruthContainer.h"
 
namespace o2
{
 
namespace emcal
{
 
namespace reco_workflow
{
 
class CellConverterSpec : public framework::Task
{
 public:
  CellConverterSpec(bool propagateMC, int inputSubsepc, int outputSubspec, std::shared_ptr<o2::emcal::CalibLoader> calibhandler) : framework::Task(), mPropagateMC(propagateMC), mSubspecificationIn(inputSubsepc), mSubspecificationOut(outputSubspec), mCalibHandler(calibhandler){};
 
  ~CellConverterSpec() override = default;
 
  void init(framework::InitContext& ctx) final;
 
  void run(framework::ProcessingContext& ctx) final;
 
  void finaliseCCDB(o2::framework::ConcreteDataMatcher& matcher, void* obj) final;
 
 protected:
  std::vector<o2::emcal::SRUBunchContainer> digitsToBunches(gsl::span<const o2::emcal::Digit> digits, std::vector<gsl::span<const o2::emcal::MCLabel>>& mcLabels);
 
  std::vector<AltroBunch> findBunches(const std::vector<const o2::emcal::Digit*>& channelDigits, const std::vector<gsl::span<const o2::emcal::MCLabel>>& mcLabels, ChannelType_t channelType);
 
  void mergeLabels(std::vector<o2::emcal::AltroBunch>& channelBunches);
 
  int selectMaximumBunch(const gsl::span<const Bunch>& bunchvector);
 
  void updateCalibrationObjects();
 
 private:
  bool mPropagateMC = false;                                           
  unsigned int mSubspecificationIn = 0;                                
  unsigned int mSubspecificationOut = 0;                               
  o2::emcal::Geometry* mGeometry = nullptr;                            
  std::shared_ptr<o2::emcal::CalibLoader> mCalibHandler;               
  std::unique_ptr<o2::emcal::CaloRawFitter> mRawFitter;                
  std::vector<o2::emcal::Cell> mOutputCells;                           
  std::vector<o2::emcal::TriggerRecord> mOutputTriggers;               
  o2::dataformats::MCTruthContainer<o2::emcal::MCLabel> mOutputLabels; 
};
 
framework::DataProcessorSpec getCellConverterSpec(bool propagateMC, int inputSubsepc = 0, int outputSubspec = 0);
 
} // namespace reco_workflow
 
} // namespace emcal
 
} // namespace o2
 
#endif // O2_EMCAL_CELLCONVERTER_SPEC