TPCMShapeCorrection.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 ALICEO2_TPC_TPCMSHAPECORRECTION
#define ALICEO2_TPC_TPCMSHAPECORRECTION
 
#include "DataFormatsTPC/Defs.h"
#include <vector>
 
class TTree;
 
namespace o2::tpc
{
 
struct BoundaryPotentialIFC {
  void addValue(float potential, float z)
  {
    mPotential.emplace_back(potential);
    mZPosition.emplace_back(z);
  }
  std::vector<float> mPotential; 
  std::vector<float> mZPosition; 
  ClassDefNV(BoundaryPotentialIFC, 1);
};
 
struct TPCMShape {
  double getStartTimeMS() const { return mTimeMS.front(); }
 
  double getEndTimeMS() const { return mTimeMS.back(); }
 
  void addBoundaryPotentialIFC(const BoundaryPotentialIFC& pot, double time)
  {
    mTimeMS.emplace_back(time);
    mBoundaryPotentialIFC.emplace_back(pot);
  }
 
  std::vector<double> mTimeMS;                             
  std::vector<BoundaryPotentialIFC> mBoundaryPotentialIFC; 
  int mZ{129};                                             
  int mR{129};                                             
  int mPhi{5};                                             
  int mBField{5};                                          
  ClassDefNV(TPCMShape, 1);
};
 
/*
Class for storing the scalers (obtained from DCAs) which are used to correct for the M-shape distortions
*/
 
class TPCMShapeCorrection
{
 public:
  TPCMShapeCorrection() = default;
 
  void setRun(int run) { mRun = run; }
 
  void dumpToFile(const char* file, const char* name);
 
  void dumpToFileSlices(const char* file, const char* name, int nSlices);
 
  void loadFromFile(const char* inpf, const char* name);
 
  void setFromTree(TTree& tpcMShapeTree);
 
  float setMaxDeltaTimeMS(float deltaTMS) { return mMaxDeltaTimeMS = deltaTMS; }
 
  void setMShapes(const TPCMShape& mshapes) { mMShapes = mshapes; }
 
  int getRun() const { return mRun; }
 
  BoundaryPotentialIFC getBoundaryPotential(const double timestamp) const;
 
  float getMaxDeltaTimeMS() const { return mMaxDeltaTimeMS; }
 
  const auto& getMShapes() const { return mMShapes; }
 
 private:
  int mRun{};                 
  float mMaxDeltaTimeMS = 50; 
  TPCMShape mMShapes;         
 
  ClassDefNV(TPCMShapeCorrection, 1);
};
 
} // namespace o2::tpc
#endif