Painter.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_PAINTER_H_
#define ALICEO2_TPC_PAINTER_H_
 
 
#include <vector>
#include <string>
#include <string_view>
#include "DataFormatsTPC/Defs.h"
#include "DataFormatsTPC/LtrCalibData.h"
 
class TH1;
class TH2;
class TH3F;
class TH2Poly;
class TCanvas;
class TMultiGraph;
class TTree;
 
namespace o2::tpc
{
 
template <class T>
class CalDet;
 
template <class T>
class CalArray;
 
 
struct painter {
 
  enum class Type : int {
    Pad,   
    Stack, 
    FEC    
  };
 
  static std::array<int, 6> colors;
  static std::array<int, 10> markers;
 
  struct PadCoordinates {
    std::vector<double> xVals = std::vector<double>(4);
    std::vector<double> yVals = std::vector<double>(4);
 
    void rotate(float angDeg)
    {
      const auto ang = 0.017453292519943295 * angDeg;
      const auto cs = std::cos(ang);
      const auto sn = std::sin(ang);
      for (size_t i = 0; i < xVals.size(); ++i) {
        const auto x = xVals[i] * cs - yVals[i] * sn;
        const auto y = xVals[i] * sn + yVals[i] * cs;
        xVals[i] = x;
        yVals[i] = y;
      }
    }
  };
 
  static std::vector<PadCoordinates> getPadCoordinatesSector();
 
  static std::vector<PadCoordinates> getStackCoordinatesSector();
 
  static std::vector<PadCoordinates> getFECCoordinatesSector();
 
  static std::vector<o2::tpc::painter::PadCoordinates> getCoordinates(const Type type);
 
  static std::vector<double> getRowBinningCM(uint32_t roc = 72);
 
  static std::string getROCTitle(const int rocNumber);
 
  // using T=float;
  template <class T>
  static TCanvas* draw(const CalDet<T>& calDet, int nbins1D = 300, float xMin1D = 0, float xMax1D = 0, TCanvas* outputCanvas = nullptr);
 
  template <class T>
  static TCanvas* draw(const CalArray<T>& calArray);
 
  template <class T>
  static void fillHistogram2D(TH2& h2D, const CalDet<T>& calDet, Side side);
 
  template <class T>
  static void fillHistogram2D(TH2& h2D, const CalArray<T>& calArray);
 
  template <class T>
  static TH2* getHistogram2D(const CalDet<T>& calDet, Side side);
 
  template <class T>
  static TH2* getHistogram2D(const CalArray<T>& calArray);
 
  static TH2Poly* makeSectorHist(const std::string_view name = "hSector", const std::string_view title = "Sector;local #it{x} (cm);local #it{y} (cm)", const float xMin = 83.65f, const float xMax = 247.7f, const float yMin = -43.7f, const float yMax = 43.7f, const Type type = Type::Pad);
 
  static TH2Poly* makeSideHist(Side side, const Type type = Type::Pad);
 
  template <class T>
  static void fillPoly2D(TH2Poly& h2D, const CalDet<T>& calDet, Side side);
 
  template <class T>
  static std::vector<TCanvas*> makeSummaryCanvases(const CalDet<T>& calDet, int nbins1D = 300, float xMin1D = 0, float xMax1D = 0, bool onlyFilled = true, std::vector<TCanvas*>* outputCanvases = nullptr);
 
  static std::vector<TCanvas*> makeSummaryCanvases(const std::string_view fileName, const std::string_view calPadNames, int nbins1D = 300, float xMin1D = 0, float xMax1D = 0, bool onlyFilled = true);
 
  static std::vector<TCanvas*> makeSummaryCanvases(TTree& tree, const std::string_view draw, std::string_view cut = "1", int nbins1D = 300, float xMin1D = 0, float xMax1D = 0);
 
  static void drawSectorsXY(Side side, int sectorLineColor = 920, int sectorTextColor = 1);
 
  static void drawSectorLocalPadNumberPoly(short padTextColor = kBlack, float lineScalePS = 1);
 
  static void drawSectorInformationPoly(short regionLineColor = kRed, short rowTextColor = kRed);
 
  template <typename DataT>
  static TH3F convertCalDetToTH3(const std::vector<CalDet<DataT>>& calDet, const bool norm = true, const int nRBins = 150, const float rMin = 83.5, const float rMax = 254.5, const int nPhiBins = 720, const float zMax = 1);
 
  static std::vector<TCanvas*> makeSummaryCanvases(const LtrCalibData& ltr, std::vector<TCanvas*>* outputCanvases = nullptr);
 
  static TCanvas* makeJunkDetectionCanvas(const TObjArray* data, TCanvas* outputCanvas = nullptr);
 
  static void adjustPalette(TH1* h, float x2ndc, float tickLength = 0.015);
 
  static TMultiGraph* makeMultiGraph(TTree& tree, std::string_view varX, std::string_view varsY, std::string_view errVarsY = "", std::string_view cut = "", bool makeSparse = true);
 
}; // struct painter
 
} // namespace o2::tpc
 
#endif // ALICEO2_TPC_PAINTER_H_