OfflineCalibSpec.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_OFFLINECALIB_SPEC
#define O2_EMCAL_OFFLINECALIB_SPEC
 
#include <string>
#include "Framework/DataProcessorSpec.h"
#include "EMCALWorkflow/CalibLoader.h"
#include "EMCALCalib/GainCalibrationFactors.h"
#include "EMCALCalibration/EMCALCalibParams.h"
#include "DataFormatsCTP/Digits.h"
#include "DataFormatsCTP/Configuration.h"
#include "Framework/Task.h"
 
#include "TFile.h"
#include "TH1.h"
#include "TH2.h"
#include "THnSparse.h"
 
namespace o2
{
 
namespace emcal
{
 
class OfflineCalibSpec : public framework::Task
{
  using EMCALCalibParams = o2::emcal::EMCALCalibParams;
 
 public:
  OfflineCalibSpec(bool makeCellIDTimeEnergy, bool rejectCalibTriggers, bool rejectL0Trigger, std::shared_ptr<o2::emcal::CalibLoader> calibHandler) : mMakeCellIDTimeEnergy(makeCellIDTimeEnergy), mRejectCalibTriggers(rejectCalibTriggers), mRejectL0Triggers(rejectL0Trigger), mCalibrationHandler(calibHandler){};
 
  ~OfflineCalibSpec() override = default;
 
  void init(framework::InitContext& ctx) final;
 
  void finaliseCCDB(framework::ConcreteDataMatcher& matcher, void* obj) final;
 
  void run(framework::ProcessingContext& ctx) final;
 
  void endOfStream(o2::framework::EndOfStreamContext& ec) final;
 
  static const char* getCTPDigitsBinding() { return "CTPDigits"; }
  static const char* getCTPConfigBinding() { return "CTPConfig"; }
 
 private:
  void updateCalibObjects();
  std::unique_ptr<TH2> mCellAmplitude;              
  std::unique_ptr<TH2> mCellTime;                   
  std::unique_ptr<TH2> mCellTimeLG;                 
  std::unique_ptr<TH2> mCellTimeHG;                 
  std::unique_ptr<TH1> mNevents;                    
  std::unique_ptr<THnSparseF> mCellTimeEnergy;      
  std::shared_ptr<CalibLoader> mCalibrationHandler; 
  GainCalibrationFactors* mGainCalibFactors;        
  bool mMakeCellIDTimeEnergy = true;                
  bool mRejectCalibTriggers = true;                 
  bool mEnableGainCalib = false;                    
  bool mRejectL0Triggers = false;                   
  std::vector<uint64_t> mSelectedClassMasks = {};   
};
 
o2::framework::DataProcessorSpec getEmcalOfflineCalibSpec(bool makeCellIDTimeEnergy, bool rejectCalibTriggers, bool rejectL0Trigger, uint32_t inputsubspec, bool enableGainCalib, bool ctpcfgperrun);
 
} // namespace emcal
 
} // namespace o2
 
#endif // O2_EMCAL_OFFLINECALIB_SPEC