IDCAverageGroup.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_IDCAVERAGEGROUP_H_
#define ALICEO2_IDCAVERAGEGROUP_H_
 
#include "TPCCalibration/RobustAverage.h"
#include "TPCCalibration/IDCAverageGroupBase.h"
#include "TPCBase/Sector.h"
#include "TPCBase/CalDet.h"
#include <boost/property_tree/ptree.hpp>
#include <vector>
#include "Rtypes.h"
 
namespace o2::utils
{
class TreeStreamRedirector;
}
 
namespace o2::tpc
{
 
// forward declaration of helper class
template <class Type>
class IDCAverageGroupHelper;
 
template <class Type>
class IDCAverageGroup : public IDCAverageGroupBase<Type>
{
 public:
  template <bool IsEnabled = true, typename std::enable_if<(IsEnabled && (std::is_same<Type, IDCAverageGroupCRU>::value)), int>::type = 0>
  IDCAverageGroup(const unsigned char groupPads = 4, const unsigned char groupRows = 4, const unsigned char groupLastRowsThreshold = 2, const unsigned char groupLastPadsThreshold = 2, const unsigned int groupPadsSectorEdges = 0, const unsigned short cru = 0, const unsigned char overlapRows = 0, const unsigned char overlapPads = 0)
    : IDCAverageGroupBase<Type>{groupPads, groupRows, groupLastRowsThreshold, groupLastPadsThreshold, groupPadsSectorEdges, cru, sNThreads}, mOverlapRows{overlapRows}, mOverlapPads{overlapPads}
  {
    init();
  }
 
  template <bool IsEnabled = true, typename std::enable_if<(IsEnabled && (std::is_same<Type, IDCAverageGroupTPC>::value)), int>::type = 0>
  IDCAverageGroup(const std::array<unsigned char, Mapper::NREGIONS>& groupPads = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, const std::array<unsigned char, Mapper::NREGIONS>& groupRows = {1, 1, 1, 1, 1, 1, 1, 1, 1, 1}, const std::array<unsigned char, Mapper::NREGIONS>& groupLastRowsThreshold = {}, const std::array<unsigned char, Mapper::NREGIONS>& groupLastPadsThreshold = {}, const unsigned int groupPadsSectorEdges = 0, const unsigned char overlapRows = 0, const unsigned char overlapPads = 0)
    : IDCAverageGroupBase<Type>{groupPads, groupRows, groupLastRowsThreshold, groupLastPadsThreshold, groupPadsSectorEdges, sNThreads}, mOverlapRows{overlapRows}, mOverlapPads{overlapPads}
  {
    init();
  }
 
  void processIDCs(const CalDet<PadFlags>* padStatusFlags = nullptr);
 
  void drawGrouping(const std::string filename = "");
 
  void dumpToFile(const char* outFileName = "IDCAverageGroup.root", const char* outName = "IDCAverageGroup") const;
 
  static int getNThreads() { return sNThreads; }
 
  const auto& getIDCGroup() const { return this->mIDCsGrouped; }
 
  const auto& getIDCsUngrouped() const { return this->mIDCsUngrouped; }
 
  static void setSigma(const float sigma) { o2::conf::ConfigurableParam::setValue<float>("TPCIDCGroupParam", "Sigma", sigma); }
 
  static void setNThreads(const int nThreads) { sNThreads = nThreads; }
 
  bool setFromFile(const char* fileName = "IDCAverageGroup.root", const char* name = "IDCAverageGroup");
 
  void createDebugTree(const char* nameFile);
 
  static void createDebugTreeForAllCRUs(const char* nameFile, const char* filename);
 
 private:
  inline static int sNThreads{1};      
  const unsigned char mOverlapRows{0}; 
  const unsigned char mOverlapPads{0}; 
 
  void init();
 
  static void createDebugTree(const IDCAverageGroupHelper<Type>& idcStruct, o2::utils::TreeStreamRedirector& pcstream);
 
  static float normal_dist(const float x, const float sigma);
 
  template <class LoopType>
  void loopOverGroups(IDCAverageGroupHelper<LoopType>& idcStruct, const CalDet<PadFlags>* padStatusFlags = nullptr);
 
  void drawGroupingInformations(const int region, const int grPads, const int grRows, const int groupLastRowsThreshold, const int groupLastPadsThreshold, const int overlapRows, const int overlapPads, const int nIDCs, const int groupPadsSectorEdges) const;
 
  void drawLatex(IDCAverageGroupHelper<IDCAverageGroupDraw>& idcStruct, const GlobalPadNumber padNum, const unsigned int padInRegion, const bool fillPoly, const int colOffs = 0) const;
 
  ClassDefNV(IDCAverageGroup, 1)
};
 
} // namespace o2::tpc
 
#endif