CellRecalibratorSpec.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_CELLRECALIBRATOR_SPEC
#define O2_EMCAL_CELLRECALIBRATOR_SPEC
 
#include <bitset>
#include <cstdint>
#include <optional>
#include "DataFormatsEMCAL/MCLabel.h"
#include "EMCALCalib/CellRecalibrator.h"
#include "EMCALWorkflow/CalibLoader.h"
#include "Framework/ConcreteDataMatcher.h"
#include "Framework/DataProcessorSpec.h"
#include "Framework/Task.h"
#include "SimulationDataFormat/MCTruthContainer.h"
 
namespace o2
{
 
namespace emcal
{
 
class Cell;
class TriggerRecord;
 
class CellRecalibratorSpec : public framework::Task
{
 public:
  enum class LEDEventSettings {
    KEEP,    
    DROP,    
    REDIRECT 
  };
  CellRecalibratorSpec(uint32_t outputspec, LEDEventSettings ledsettings, bool badChannelCalib, bool timeCalib, bool gainCalib, bool isMC, std::shared_ptr<CalibLoader>(calibHandler));
 
  ~CellRecalibratorSpec() final = default;
 
  void init(framework::InitContext& ctx) final;
 
  void run(framework::ProcessingContext& ctx) final;
 
  void finaliseCCDB(framework::ConcreteDataMatcher& matcher, void* obj) final;
 
  void setRunBadChannelCalibration(bool doRun) { mCalibrationSettings.set(BADCHANNEL_CALIB, doRun); };
 
  void setRunTimeCalibration(bool doRun) { mCalibrationSettings.set(TIME_CALIB, doRun); }
 
  void setRunGainCalibration(bool doRun) { mCalibrationSettings.set(GAIN_CALIB, doRun); }
 
  bool isRunBadChannlCalibration() const { return mCalibrationSettings.test(BADCHANNEL_CALIB); }
 
  bool isRunTimeCalibration() const { return mCalibrationSettings.test(TIME_CALIB); }
 
  bool isRunGainCalibration() const { return mCalibrationSettings.test(GAIN_CALIB); }
 
 private:
  void updateCalibObjects();
 
  void writeTrigger(const gsl::span<const o2::emcal::Cell> selectedCells, const o2::emcal::TriggerRecord& eventtrigger, std::vector<o2::emcal::Cell>& outputcontainer, std::vector<o2::emcal::TriggerRecord>& outputtriggers);
 
  void writeMCLabels(const o2::dataformats::MCTruthContainer<o2::emcal::MCLabel>& inputlabels, o2::dataformats::MCTruthContainer<o2::emcal::MCLabel>& outputContainer, const std::vector<int>& keptIndices, int firstindex);
 
  enum CalibrationType_t {
    BADCHANNEL_CALIB = 0, 
    TIME_CALIB = 1,       
    GAIN_CALIB = 2        
  };
 
  uint32_t mOutputSubspec = 0;                            
  bool mIsMC = false;                                     
  LEDEventSettings mLEDsettings = LEDEventSettings::KEEP; 
  std::bitset<8> mCalibrationSettings;                    
  std::shared_ptr<CalibLoader> mCalibrationHandler;       
  CellRecalibrator mCellRecalibrator;                     
};
 
framework::DataProcessorSpec getCellRecalibratorSpec(uint32_t inputSubspec, uint32_t outputSubspec, uint32_t ledsettings, bool badChannelCalib, bool timeCalib, bool gainCalib, bool isMC);
 
} // namespace emcal
 
} // namespace o2
 
#endif // O2_EMCAL_CELLRECALIBRATOR_SPEC