DQM/SiStripMonitorClient/bin/moduleOccupancyTrend.cc - master - githubmirror/cmssw

DQM/SiStripMonitorClient/bin/moduleOccupancyTrend.cc
190 строк4 KB
Shahzad Malik Muzaffar
[DQM] Applying code checks/format
03 окт 2024, 10:24
03 окт 2024, 10:241c51ea5
#include <iostream>
#include <fstream>
#include <sstream>
#include <string>

#include <TROOT.h>
#include <TStyle.h>
#include "TFile.h"
#include "THStack.h"
#include "TH1D.h"
#include "TDirectoryFile.h"
#include "TCanvas.h"
#include "TLine.h"
#include "TLegend.h"
#include "TRint.h"

int main(int argc, char* argv[]) {
  gROOT->SetStyle("Plain");

  char* filelist;
  char* modulelist;

  filelist = argv[1];
  modulelist = argv[2];

  int stripmin = 999;
  int stripmax = 999;

  if (argv[3] && argv[4]) {
    stripmin = atoi(argv[3]);
    stripmax = atoi(argv[4]);
  }

  std::string detid;
  std::string hn;

  TLegend leg(0.1, 0.7, 0.2, 0.9);
  leg.SetFillStyle(0);

  std::ifstream inmodules(modulelist);

  while (true) {
    inmodules >> detid;
    if (!inmodules.good())
      break;

    hn = "ClusterDigiPosition__det__" + detid;  //std::to_string(detid);

    TCanvas c1("c1", "c1", 1600, 900);

    c1.SetBatch(kTRUE);
    c1.SetLogy(1);
    c1.SetGridy(1);

    //cout << detid << endl;
    std::ifstream fileToCountLines(filelist);
    std::size_t lines_count = 0;
    std::string line;

    while (std::getline(fileToCountLines, line))
      ++lines_count;

    const float dim = lines_count;

    std::ifstream filesin(filelist);

    std::string filename;
    int k = 0;

    TH1D* trend = new TH1D("trend", "trend", int(dim), 0.5, dim + 0.5);
    trend->SetMarkerSize(3);
    trend->SetMarkerStyle(8);
    trend->SetMarkerColor(4);

    std::string ttitle = hn + "_trend";
    trend->SetTitle(ttitle.c_str());

    double max = -1;
    double min = 1000000;

    leg.Clear();
    TFile* fin;

    THStack* hs = new THStack("hs", "");

    while (true) {
      filesin >> filename;
      if (!filesin.good())
        break;

      std::string runNum = filename.substr(filename.find("run_") + 4, 6);
      //std::cout << runNum << std::endl;

      fin = TFile::Open(filename.c_str());

      if (!fin) {
        std::cout << "Cannot open file " << filename.c_str() << std::endl;
        return 0;
      }

      TH1D* Events = (TH1D*)fin->Get("TotEvents");

      double EvtNum = Events->GetBinContent(1);

      TH1D* histo = (TH1D*)fin->Get(hn.c_str());

      if (!histo) {
        std::cout << "Cannot open histo " << hn.c_str() << std::endl;
        return 0;
      }

      histo->SetDirectory(nullptr);
      histo->SetStats(kFALSE);

      if (hn.find("Summary") == std::string::npos)
        histo->Scale(1 / EvtNum);

      double numberPerEvent;
      if (stripmin == 999 && stripmax == 999)
        numberPerEvent = histo->Integral();
      else
        numberPerEvent = histo->Integral(stripmin, stripmax);

      if (max <= histo->GetBinContent(histo->GetMaximumBin()))
        max = histo->GetBinContent(histo->GetMaximumBin());
      if (min > histo->GetBinContent(histo->GetMinimumBin()))
        min = histo->GetBinContent(histo->GetMinimumBin());

      histo->SetLineColor(k + 1);
      histo->SetMarkerStyle(9);
      histo->SetMarkerColor(k + 1);

      trend->SetBinContent(k + 1, numberPerEvent);
      trend->GetXaxis()->SetBinLabel(k + 1, runNum.c_str());

      leg.AddEntry(histo, runNum.c_str(), "L");
      hs->Add(histo);
      k++;

      fin->Close();
    }

    if (min == 0)
      min = 1.e-6;

    max = max * 10;  // in a way one can read the legend

    hs->SetMaximum(max);
    hs->SetMinimum(min);
    hs->SetTitle(hn.c_str());

    hs->Draw("nostack");

    TLine l;

    l.SetLineColor(4);
    l.SetLineStyle(2);
    l.SetLineWidth(3);

    l.DrawLine(128, min, 128, max);
    l.DrawLine(256, min, 256, max);
    l.DrawLine(384, min, 384, max);
    l.DrawLine(384, min, 384, max);
    l.DrawLine(512, min, 512, max);
    l.DrawLine(640, min, 640, max);

    leg.Draw();
    std::string outname = hn + "_Super.png";
    c1.SaveAs(outname.c_str());

    c1.SetGridx(1);

    double mintrend = 0;
    if (trend->GetMinimum() == 0)
      mintrend = 1.e-4;
    else
      mintrend = trend->GetMinimum();

    trend->SetStats(kFALSE);

    trend->GetYaxis()->SetRangeUser(mintrend * 0.5, trend->GetMaximum() * 2);
    trend->Draw("P");
    outname = hn + "_Trend.png";
    c1.SaveAs(outname.c_str());
    c1.Clear();

    delete trend;
  }
  return 0;
}