CalibTreeDump.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_CALIBTREEDUMP_H_
#define ALICEO2_TPC_CALIBTREEDUMP_H_
 
#include <vector>
#include <string>
#include <unordered_map>
 
#include <boost/variant.hpp>
 
 
// forward declarations
class TTree;
 
namespace o2
{
namespace tpc
{
 
// forward declarations
template <class T>
class CalDet;
 
template <class T>
class CalArray;
 
 
class CalibTreeDump
{
 public:
  using DataTypes = CalDet<float>; // boost::variant<CalDet<int>, CalDet<float>, CalDet<double>, CalDet<bool>, CalDet<unsigned int>>;
  using CalPadMapType = std::unordered_map<std::string, DataTypes>;
 
  CalibTreeDump() = default;
  ~CalibTreeDump() = default;
 
  template <typename T>
  void add(CalDet<T>* calDet)
  {
    mCalDetObjects.emplace_back(calDet);
  }
  // void add(const CalDet<DataTypes>& calDet) { mCalDetObjects.push_back(calDet); }
 
  template <typename T>
  void add(CalArray<T>* calArray)
  {
    mCalArrayObjects.emplace_back(calArray);
  }
  void add(CalPadMapType& calibs);
 
  void setAddFEEInfo(bool add = true) { mAddFEEInfo = add; }
 
  void addCalPads(const std::string_view file, const std::string_view calPadNames);
 
  void dumpToFile(const std::string filename = "CalibTree.root");
 
  void addInfo(const std::string_view name, float value) { mAddInfo[name.data()] = value; }
 
 private:
  std::unordered_map<std::string, float> mAddInfo{}; 
  std::vector<DataTypes*> mCalDetObjects{};          
  std::vector<DataTypes*> mCalArrayObjects{};        
  bool mAddFEEInfo{false};                           
  std::vector<float> mTraceLengthIROC;               
  std::vector<float> mTraceLengthOROC;               
 
  void addDefaultMapping(TTree* tree);
 
  void addFEEMapping(TTree* tree);
 
  void addCalDetObjects(TTree* tree);
 
  void setDefaultAliases(TTree* tree);
 
  template <class Result, class Func>
  struct forwarding_visitor : boost::static_visitor<Result> {
    Func func;
    forwarding_visitor(const Func& f) : func(f) {}
    forwarding_visitor(Func&& f) : func(std::move(f)) {}
    template <class Arg>
    Result operator()(Arg&& arg) const
    {
      return func(std::forward<Arg>(arg));
    }
  };
 
  template <class Result, class Func>
  forwarding_visitor<Result, std::decay_t<Func>> make_forwarding_visitor(Func&& func)
  {
    return {std::forward<Func>(func)};
  }
};
} // namespace tpc
 
} // namespace o2
#endif