PrepareDMRTrends.h

Go to the documentation of this file.

#ifndef ALIGNMENT_OFFLINEVALIDATION_PREPAREDMRTRENDS_H_
#define ALIGNMENT_OFFLINEVALIDATION_PREPAREDMRTRENDS_H_

#include <iostream>
#include <string>
#include <vector>
#include <algorithm>
#include <map>
#include <iomanip>
#include <fstream>
#include <experimental/filesystem>
#include "TPad.h"
#include "TCanvas.h"
#include "TGraph.h"
#include "TGraphErrors.h"
#include "TMultiGraph.h"
#include "TH1.h"
#include "THStack.h"
#include "TROOT.h"
#include "TFile.h"
#include "TLegend.h"
#include "TLegendEntry.h"
#include "TMath.h"
#include "TRegexp.h"
#include "TPaveLabel.h"
#include "TPaveText.h"
#include "TStyle.h"
#include "TLine.h"
#include "boost/property_tree/ptree.hpp"
#include "boost/property_tree/json_parser.hpp"

#define DUMMY -999.

#define DMRFactor 10000.

struct Point {
  float run, scale, mu, sigma, muplus, muminus, sigmaplus, sigmaminus;

  Point(float Run = DUMMY,
        float ScaleFactor = DMRFactor,
        float y1 = DUMMY,
        float y2 = DUMMY,
        float y3 = DUMMY,
        float y4 = DUMMY,
        float y5 = DUMMY,
        float y6 = DUMMY)
      : run(Run), scale(ScaleFactor), mu(y1), sigma(y2), muplus(y3), muminus(y5), sigmaplus(y4), sigmaminus(y6) {}

  Point(float Run, float ScaleFactor, TH1 *histo, TH1 *histoplus, TH1 *histominus)
      : Point(Run,
              ScaleFactor,
              histo->GetMean(),
              histo->GetMeanError(),
              histoplus->GetMean(),
              histoplus->GetMeanError(),
              histominus->GetMean(),
              histominus->GetMeanError()) {}

  Point(float Run, float ScaleFactor, TH1 *histo) : Point(Run, ScaleFactor, histo->GetMean(), histo->GetMeanError()) {}

  Point &operator=(const Point &p) {
    run = p.run;
    mu = p.mu;
    muplus = p.muplus;
    muminus = p.muminus;
    sigma = p.sigma;
    sigmaplus = p.sigmaplus;
    sigmaminus = p.sigmaminus;
    return *this;
  }

  inline float GetRun() const { return run; }
  inline float GetMu() const { return scale * mu; }
  inline float GetMuPlus() const { return scale * muplus; }
  inline float GetMuMinus() const { return scale * muminus; }
  inline float GetSigma() const { return scale * sigma; }
  inline float GetSigmaPlus() const { return scale * sigmaplus; }
  inline float GetSigmaMinus() const { return scale * sigmaminus; }
  inline float GetDeltaMu() const {
    if (muplus == DUMMY && muminus == DUMMY)
      return DUMMY;
    else
      return scale * (muplus - muminus);
  }
  inline float GetSigmaDeltaMu() const {
    if (sigmaplus == DUMMY && sigmaminus == DUMMY)
      return DUMMY;
    else
      return scale * hypot(sigmaplus, sigmaminus);
  }
};

class Geometry {
public:
  std::vector<Point> points;

private:
  //template<typename T> std::vector<T> GetQuantity (T (Point::*getter)() const) const {
  std::vector<float> GetQuantity(float (Point::*getter)() const) const {
    std::vector<float> v;
    for (Point point : points) {
      float value = (point.*getter)();
      v.push_back(value);
    }
    return v;
  }

public:
  TString title;
  Geometry() : title("") {}
  Geometry(TString Title) : title(Title) {}
  Geometry &operator=(const Geometry &geom) {
    title = geom.title;
    points = geom.points;
    return *this;
  }
  inline void SetTitle(TString Title) { title = Title; }
  inline TString GetTitle() { return title; }
  inline std::vector<float> Run() const { return GetQuantity(&Point::GetRun); }
  inline std::vector<float> Mu() const { return GetQuantity(&Point::GetMu); }
  inline std::vector<float> MuPlus() const { return GetQuantity(&Point::GetMuPlus); }
  inline std::vector<float> MuMinus() const { return GetQuantity(&Point::GetMuMinus); }
  inline std::vector<float> Sigma() const { return GetQuantity(&Point::GetSigma); }
  inline std::vector<float> SigmaPlus() const { return GetQuantity(&Point::GetSigmaPlus); }
  inline std::vector<float> SigmaMinus() const { return GetQuantity(&Point::GetSigmaMinus); }
  inline std::vector<float> DeltaMu() const { return GetQuantity(&Point::GetDeltaMu); }
  inline std::vector<float> SigmaDeltaMu() const { return GetQuantity(&Point::GetSigmaDeltaMu); }
};

class PrepareDMRTrends {
public:
  PrepareDMRTrends(const char *outputFileName, boost::property_tree::ptree &json);
  ~PrepareDMRTrends() {}

  TString getName(TString structure, int layer, TString geometry);
  void compileDMRTrends(std::vector<int> IOVlist,
                        TString Variable,
                        std::vector<std::string> inputFiles,
                        std::vector<TString> structures,
                        const std::map<TString, int> nlayers,
                        bool FORCE = false);

private:
  const char *outputFileName_;
  std::vector<std::string> geometries;
};

#endif  // ALIGNMENT_OFFLINEVALIDATION_PREPAREDMRTRENDS_H_