SiStripQualityHotStripIdentifier.cc

Go to the documentation of this file.

#include "CalibTracker/SiStripQuality/plugins/SiStripQualityHotStripIdentifier.h"

#include "CalibFormats/SiStripObjects/interface/SiStripQuality.h"

#include "DataFormats/TrackReco/interface/Track.h"
#include "DataFormats/TrackReco/interface/TrackFwd.h"
#include "DataFormats/TrackerRecHit2D/interface/ProjectedSiStripRecHit2D.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripMatchedRecHit2D.h"
#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"
#include "DataFormats/TrackingRecHit/interface/TrackingRecHit.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
#include <fstream>
#include <iostream>
#include <memory>
#include <sstream>

//Insert here the include to the algos
#include "CalibTracker/SiStripQuality/interface/SiStripHotStripAlgorithmFromClusterOccupancy.h"

SiStripQualityHotStripIdentifier::SiStripQualityHotStripIdentifier(const edm::ParameterSet& iConfig)
    : ConditionDBWriter<SiStripBadStrip>(iConfig),
      m_cacheID_(0),
      dataLabel_(iConfig.getUntrackedParameter<std::string>("dataLabel", "")),
      conf_(iConfig),
      fp_(iConfig.getUntrackedParameter<edm::FileInPath>(
          "file", edm::FileInPath("CalibTracker/SiStripCommon/data/SiStripDetInfo.dat"))),
      Cluster_src_(iConfig.getParameter<edm::InputTag>("Cluster_src")),
      Track_src_(iConfig.getUntrackedParameter<edm::InputTag>("Track_src")),
      tracksCollection_in_EventTree(iConfig.getUntrackedParameter<bool>("RemoveTrackClusters", false)),
      tTopo(nullptr) {
  reader = new SiStripDetInfoFileReader(fp_.fullPath());

  edm::ParameterSet pset = iConfig.getUntrackedParameter<edm::ParameterSet>("ClusterSelection", edm::ParameterSet());
  MinClusterWidth_ = pset.getUntrackedParameter<uint32_t>("minWidth", 1);
  MaxClusterWidth_ = pset.getUntrackedParameter<uint32_t>("maxWidth", 1000);

  bookHistos();
}

SiStripQualityHotStripIdentifier::~SiStripQualityHotStripIdentifier() {}

std::unique_ptr<SiStripBadStrip> SiStripQualityHotStripIdentifier::getNewObject() {
  auto obj = std::make_unique<SiStripBadStrip>();

  edm::ParameterSet parameters = conf_.getParameter<edm::ParameterSet>("AlgoParameters");
  std::string AlgoName = parameters.getParameter<std::string>("AlgoName");
  if (AlgoName == "SiStripHotStripAlgorithmFromClusterOccupancy") {
    edm::LogInfo("SiStripQualityHotStripIdentifier")
        << " [SiStripQualityHotStripIdentifier::getNewObject] call to SiStripHotStripAlgorithmFromClusterOccupancy"
        << std::endl;

    SiStripHotStripAlgorithmFromClusterOccupancy theIdentifier(conf_, tTopo);
    theIdentifier.setProbabilityThreshold(parameters.getUntrackedParameter<double>("ProbabilityThreshold", 1.E-7));
    theIdentifier.setMinNumEntries(parameters.getUntrackedParameter<uint32_t>("MinNumEntries", 100));
    theIdentifier.setMinNumEntriesPerStrip(parameters.getUntrackedParameter<uint32_t>("MinNumEntriesPerStrip", 5));

    SiStripQuality* qobj = new SiStripQuality();
    theIdentifier.extractBadStrips(qobj, ClusterPositionHistoMap, SiStripQuality_);

    edm::LogInfo("SiStripQualityHotStripIdentifier")
        << " [SiStripQualityHotStripIdentifier::getNewObject] copy SiStripObject in SiStripBadStrip" << std::endl;

    std::stringstream ss;

    SiStripBadStrip::RegistryIterator rIter = qobj->getRegistryVectorBegin();
    SiStripBadStrip::RegistryIterator rIterEnd = qobj->getRegistryVectorEnd();
    for (; rIter != rIterEnd; ++rIter) {
      SiStripBadStrip::Range range(qobj->getDataVectorBegin() + rIter->ibegin,
                                   qobj->getDataVectorBegin() + rIter->iend);
      if (!obj->put(rIter->detid, range))
        edm::LogError("SiStripQualityHotStripIdentifier")
            << "[SiStripQualityHotStripIdentifier::getNewObject] detid already exists" << std::endl;
    }
    edm::LogInfo("SiStripQualityHotStripIdentifier")
        << " [SiStripQualityHotStripIdentifier::getNewObject] " << ss.str() << std::endl;

  } else {
    edm::LogError("SiStripQualityHotStripIdentifier")
        << " [SiStripQualityHotStripIdentifier::getNewObject] call for a unknow HotStrip identification algoritm"
        << std::endl;

    std::vector<uint32_t> a;
    SiStripBadStrip::Range range(a.begin(), a.end());
    if (!obj->put(0xFFFFFFFF, range))
      edm::LogError("SiStripQualityHotStripIdentifier")
          << "[SiStripQualityHotStripIdentifier::getNewObject] detid already exists" << std::endl;
  }

  return obj;
}

void SiStripQualityHotStripIdentifier::algoBeginRun(const edm::Run& run, const edm::EventSetup& iSetup) {
  //Retrieve tracker topology from geometry
  edm::ESHandle<TrackerTopology> tTopoHandle;
  iSetup.get<TrackerTopologyRcd>().get(tTopoHandle);
  tTopo = tTopoHandle.product();

  resetHistos();
  unsigned long long cacheID = iSetup.get<SiStripQualityRcd>().cacheIdentifier();

  if (m_cacheID_ == cacheID)
    return;

  m_cacheID_ = cacheID;

  iSetup.get<SiStripQualityRcd>().get(dataLabel_, SiStripQuality_);
}

void SiStripQualityHotStripIdentifier::algoEndJob() {
  //Clear map
  ClusterPositionHistoMap.clear();
}

void SiStripQualityHotStripIdentifier::resetHistos() {
  edm::LogInfo("SiStripQualityHotStripIdentifier") << " [SiStripQualityHotStripIdentifier::resetHistos] " << std::endl;
  SiStrip::QualityHistosMap::iterator it = ClusterPositionHistoMap.begin();
  SiStrip::QualityHistosMap::iterator iEnd = ClusterPositionHistoMap.end();
  for (; it != iEnd; ++it) {
    it->second->Reset();
  }
}

void SiStripQualityHotStripIdentifier::bookHistos() {
  edm::LogInfo("SiStripQualityHotStripIdentifier") << " [SiStripQualityHotStripIdentifier::bookHistos] " << std::endl;
  char hname[1024];
  std::map<uint32_t, SiStripDetInfoFileReader::DetInfo>::const_iterator it = reader->getAllData().begin();
  std::map<uint32_t, SiStripDetInfoFileReader::DetInfo>::const_iterator iEnd = reader->getAllData().end();
  for (; it != iEnd; ++it) {
    sprintf(hname, "h_%d", it->first);
    SiStrip::QualityHistosMap::iterator ref = ClusterPositionHistoMap.find(it->first);
    if (ref == ClusterPositionHistoMap.end()) {
      ClusterPositionHistoMap[it->first] =
          std::make_shared<TH1F>(hname, hname, it->second.nApvs * 128, -0.5, it->second.nApvs * 128 - 0.5);
    } else
      edm::LogError("SiStripQualityHotStripIdentifier")
          << " [SiStripQualityHotStripIdentifier::bookHistos] DetId " << it->first
          << " already found in map. Ignoring new data" << std::endl;
  }
}

void SiStripQualityHotStripIdentifier::fillHisto(uint32_t detid, float value) {
  SiStrip::QualityHistosMap::iterator ref = ClusterPositionHistoMap.find(detid);
  if (ref != ClusterPositionHistoMap.end())
    ref->second->Fill(value);
  else
    edm::LogError("SiStripQualityHotStripIdentifier")
        << " [SiStripQualityHotStripIdentifier::fillHisto] Histogram not found in the list for DetId " << detid
        << " Ignoring data value " << value << std::endl;
}

void SiStripQualityHotStripIdentifier::algoAnalyze(const edm::Event& e, const edm::EventSetup& eSetup) {
  edm::Handle<edm::DetSetVector<SiStripCluster> > dsv_SiStripCluster;
  e.getByLabel(Cluster_src_, dsv_SiStripCluster);

  edm::Handle<reco::TrackCollection> trackCollection;
  if (tracksCollection_in_EventTree) {
    e.getByLabel(Track_src_, trackCollection);
    if (!trackCollection.isValid()) {
      edm::LogError("SiStripQualityHotStripIdentifier")
          << " [SiStripQualityHotStripIdentifier::algoAnalyze] missing trackCollection with label " << Track_src_
          << std::endl;
    }
  }

  std::set<const void*> vPSiStripCluster;
  //Perform track study
  if (tracksCollection_in_EventTree) {
    const reco::TrackCollection tC = *(trackCollection.product());
    int i = 0;
    for (reco::TrackCollection::const_iterator track = tC.begin(); track != tC.end(); track++) {
      LogTrace("SiStripQualityHotStripIdentifier")
          << "Track number " << i + 1 << "\n\tmomentum: " << track->momentum() << "\n\tPT: " << track->pt()
          << "\n\tvertex: " << track->vertex() << "\n\timpact parameter: " << track->d0()
          << "\n\tcharge: " << track->charge() << "\n\tnormalizedChi2: " << track->normalizedChi2()
          << "\n\tFrom EXTRA : "
          << "\n\t\touter PT " << track->outerPt() << std::endl;

      //Loop on rechits
      for (auto const& recHit : track->recHits()) {
        if (!recHit->isValid()) {
          LogTrace("SiStripQualityHotStripIdentifier") << "\t\t Invalid Hit " << std::endl;
          continue;
        }

        const SiStripRecHit2D* singleHit = dynamic_cast<const SiStripRecHit2D*>(recHit);
        const SiStripMatchedRecHit2D* matchedHit = dynamic_cast<const SiStripMatchedRecHit2D*>(recHit);
        const ProjectedSiStripRecHit2D* projectedHit = dynamic_cast<const ProjectedSiStripRecHit2D*>(recHit);

        if (matchedHit) {
          vPSiStripCluster.insert((void*)&(matchedHit->monoCluster()));
          vPSiStripCluster.insert((void*)&(matchedHit->stereoCluster()));
        } else if (projectedHit) {
          vPSiStripCluster.insert((void*)&*(projectedHit->originalHit().cluster()));
        } else if (singleHit) {
          vPSiStripCluster.insert((void*)&*(singleHit->cluster()));
        } else {
          LogTrace("SiStripQualityHotStripIdentifier") << "NULL hit" << std::endl;
        }
      }
    }
  }

  std::stringstream ss;
  //Loop on Det Clusters
  edm::DetSetVector<SiStripCluster>::const_iterator DSViter = dsv_SiStripCluster->begin();
  for (; DSViter != dsv_SiStripCluster->end(); DSViter++) {
    edm::DetSet<SiStripCluster>::const_iterator ClusIter = DSViter->data.begin();
    edm::DetSet<SiStripCluster>::const_iterator ClusIterEnd = DSViter->data.end();
    for (; ClusIter != ClusIterEnd; ++ClusIter) {
      if (MinClusterWidth_ <= ClusIter->amplitudes().size() && ClusIter->amplitudes().size() <= MaxClusterWidth_) {
        if (std::find(vPSiStripCluster.begin(), vPSiStripCluster.end(), (void*)&*ClusIter) == vPSiStripCluster.end()) {
          if (edm::isDebugEnabled())
            ss << " adding cluster to histo for detid " << DSViter->id << " with barycenter " << ClusIter->barycenter()
               << std::endl;
          fillHisto(DSViter->id, ClusIter->barycenter());
        }
      }
    }
  }
  LogTrace("SiStripQualityHotStripIdentifier") << ss.str();
}