SiStripPlotGain.cc

Go to the documentation of this file.

#include "DQM/SiStripMonitorSummary/plugins/SiStripPlotGain.h"

#include "DataFormats/TrackerCommon/interface/TrackerTopology.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
#include "DataFormats/SiStripDetId/interface/SiStripDetId.h"

SiStripPlotGain::SiStripPlotGain(const edm::ParameterSet &iConfig) : cacheID(0xFFFFFFFF) {
  // now do what ever initialization is needed
  file = new TFile("correlTest.root", "RECREATE");
  tkmap = new TrackerMap();
}

SiStripPlotGain::~SiStripPlotGain() {}

//

void SiStripPlotGain::beginRun(const edm::Run &run, const edm::EventSetup &es) {
  if (getCache(es) == cacheID)
    return;
  cacheID = getCache(es);

  edm::LogInfo("") << "[SiStripPlotGain::beginRun] cacheID " << cacheID << std::endl;

  es.get<SiStripApvGainRcd>().get(Handle_);
  DoAnalysis(es, *Handle_.product());
}

void SiStripPlotGain::DoAnalysis(const edm::EventSetup &es, const SiStripApvGain &gain) {
  edm::LogInfo("") << "[Doanalysis]";

  // Retrieve tracker topology from geometry
  edm::ESHandle<TrackerTopology> tTopoHandle;
  es.get<TrackerTopologyRcd>().get(tTopoHandle);
  const TrackerTopology *const tTopo = tTopoHandle.product();

  std::vector<TH1F *> histos;

  SiStripApvGain::RegistryPointers p = gain.getRegistryPointers();
  SiStripApvGain::RegistryConstIterator iter, iterE;
  iter = p.detid_begin;
  iterE = p.detid_end;

  float value;

  // Divide result by d
  for (; iter != iterE; ++iter) {
    getHistos(*iter, tTopo, histos);
    SiStripApvGain::Range range = SiStripApvGain::Range(p.getFirstElement(iter), p.getLastElement(iter));

    edm::LogInfo("") << "[Doanalysis] detid " << *iter << " range " << range.second - range.first;
    size_t apv = 0, apvE = (range.second - range.first);
    for (; apv < apvE; apv += 2) {
      value = gain.getApvGain(apv, range);
      tkmap->fill(*iter, value);
      for (size_t i = 0; i < histos.size(); ++i)
        histos[i]->Fill(value);
    }
  }
}

void SiStripPlotGain::getHistos(const uint32_t &detid, const TrackerTopology *tTopo, std::vector<TH1F *> &histos) {
  histos.clear();

  int subdet = -999;
  int component = -999;
  SiStripDetId a(detid);
  if (a.subdetId() == 3) {
    subdet = 0;
    component = tTopo->tibLayer(detid);
  } else if (a.subdetId() == 4) {
    subdet = 1;
    component = tTopo->tidSide(detid) == 2 ? tTopo->tidWheel(detid) : tTopo->tidWheel(detid) + 3;
  } else if (a.subdetId() == 5) {
    subdet = 2;
    component = tTopo->tobLayer(detid);
  } else if (a.subdetId() == 6) {
    subdet = 3;
    component = tTopo->tecSide(detid) == 2 ? tTopo->tecWheel(detid) : tTopo->tecWheel(detid) + 9;
  }

  int index = 100 + subdet * 100 + component;

  histos.push_back(getHisto(subdet + 1));
  histos.push_back(getHisto(index));
}

TH1F *SiStripPlotGain::getHisto(const long unsigned int &index) {
  if (vTH1.size() < index + 1)
    vTH1.resize(index + 1, nullptr);

  if (vTH1[index] == nullptr) {
    char name[128];
    sprintf(name, "%lu", index);
    edm::LogInfo("") << "[getHisto] creating index " << index << std::endl;
    vTH1[index] = new TH1F(name, name, 150, 0., 5.);
  }

  return vTH1[index];
}

void SiStripPlotGain::endJob() {
  for (size_t i = 0; i < vTH1.size(); i++)
    if (vTH1[i] != nullptr)
      vTH1[i]->Write();

  file->Write();
  file->Close();

  tkmap->save(false, 0, 0, "testTkMap.png");
}