CalibratorVdExB.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_TRD_CALIBRATORVDEXB_H
#define O2_TRD_CALIBRATORVDEXB_H
 
#include "DetectorsCalibration/TimeSlotCalibration.h"
#include "DetectorsCalibration/TimeSlot.h"
#include "DataFormatsTRD/Constants.h"
#include "DataFormatsTRD/AngularResidHistos.h"
#include "CCDB/CcdbObjectInfo.h"
#include "DataFormatsTRD/CalVdriftExB.h"
#include "TRDCalibration/CalibrationParams.h"
 
#include "Rtypes.h"
#include "TProfile.h"
#include "Fit/Fitter.h"
#include "TFile.h"
#include "TTree.h"
 
#include <array>
#include <cstdlib>
#include <memory>
 
namespace o2
{
namespace trd
{
 
struct FitFunctor {
  double operator()(const double* par) const;
  double calculateDeltaAlphaSim(double vdFit, double laFit, double impactAng) const;
  std::array<std::unique_ptr<TProfile>, constants::MAXCHAMBER> profiles; 
  std::array<double, constants::MAXCHAMBER> vdPreCorr;                   
  std::array<double, constants::MAXCHAMBER> laPreCorr;                   
  int currDet;                                                           
  float lowerBoundAngleFit;                                              
  float upperBoundAngleFit;                                              
  double mAnodePlane;                                                    
};
 
class CalibratorVdExB final : public o2::calibration::TimeSlotCalibration<o2::trd::AngularResidHistos>
{
  using Slot = o2::calibration::TimeSlot<o2::trd::AngularResidHistos>;
 
 public:
  enum ParamIndex : int {
    LA,
    VD
  };
  CalibratorVdExB() = default;
  ~CalibratorVdExB() final = default;
 
  bool hasEnoughData(const Slot& slot) const final { return slot.getContainer()->getNEntries() >= mMinEntriesTotal; }
  void initOutput() final;
  void finalizeSlot(Slot& slot) final;
  Slot& emplaceNewSlot(bool front, TFType tStart, TFType tEnd) final;
 
  void createOutputFile();
 
  void closeOutputFile();
 
  const std::vector<o2::trd::CalVdriftExB>& getCcdbObjectVector() const { return mObjectVector; }
  std::vector<o2::ccdb::CcdbObjectInfo>& getCcdbObjectInfoVector() { return mInfoVector; }
 
  void initProcessing();
 
  void retrievePrev(o2::framework::ProcessingContext& pc);
 
 private:
  bool mInitDone{false};                                     
  const TRDCalibParams& mParams{TRDCalibParams::Instance()}; 
  size_t mMinEntriesTotal{mParams.minEntriesChamber};        
  size_t mMinEntriesChamber{mParams.minEntriesTotal};        
  bool mEnableOutput{false};                                 
  std::unique_ptr<TFile> mOutFile{nullptr};                  
  std::unique_ptr<TTree> mOutTree{nullptr};                  
  FitFunctor mFitFunctor;                                    
  std::vector<o2::ccdb::CcdbObjectInfo> mInfoVector;         
  std::vector<o2::trd::CalVdriftExB> mObjectVector;          
  ROOT::Fit::Fitter mFitter;                                 
  double mParamsStart[2];                                    
 
  ClassDefOverride(CalibratorVdExB, 5);
};
 
} // namespace trd
} // namespace o2
 
#endif // O2_TRD_CALIBRATORVDEXB_H