DTOccupancyTestML.h

Go to the documentation of this file.

#ifndef DTOccupancyTestML_H
#define DTOccupancyTestML_H
 
#include "FWCore/Framework/interface/Frameworkfwd.h"
#include <FWCore/Framework/interface/EDAnalyzer.h>
#include <FWCore/Framework/interface/ESHandle.h>
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include <DataFormats/MuonDetId/interface/DTLayerId.h>
 
#include <DQMServices/Core/interface/DQMEDHarvester.h>
 
#include "PhysicsTools/TensorFlow/interface/TensorFlow.h"
 
#include "TH2F.h"
 
#include <iostream>
#include <string>
#include <map>
 
class DTGeometry;
class DTChamberId;
 
#include "TFile.h"
#include "TNtuple.h"
 
class DTOccupancyTestML : public DQMEDHarvester {
public:
  DTOccupancyTestML(const edm::ParameterSet &ps);
 
  ~DTOccupancyTestML() override;
 
protected:
  void beginRun(edm::Run const &run, edm::EventSetup const &context) override;
 
  void dqmEndJob(DQMStore::IBooker &, DQMStore::IGetter &) override;
 
 
  void dqmEndLuminosityBlock(DQMStore::IBooker &,
                             DQMStore::IGetter &,
                             edm::LuminosityBlock const &,
                             edm::EventSetup const &) override;
 
private:
  void bookHistos(DQMStore::IBooker &, const int wheelId, std::string folder, std::string histoTag);
 
  std::string getMEName(std::string histoTag, const DTChamberId &chId);
 
  int getIntegral(TH2F *histo, int, int, int, int, bool);
 
  // Run the test on the occupancy histos
  int runOccupancyTest(TH2F *histo,
                       const DTChamberId &chId,
                       float &chamberPercentage,
                       tensorflow::GraphDef *graphDef,
                       tensorflow::Session *session);
 
  std::vector<float> interpolateLayers(std::vector<float> const &inputs, int size, int targetSize);
 
  std::string topFolder(bool isBooking) const;
 
  int nevents;
 
  edm::ESHandle<DTGeometry> muonGeom;
 
  // wheel summary histograms
  std::map<int, MonitorElement *> wheelHistos;
  MonitorElement *summaryHisto;
  MonitorElement *glbSummaryHisto;
 
  std::set<DTLayerId> monitoredLayers;
 
  int lsCounter;
  int nMinEvts;
  int nMinEvtsPC;
  int nZeroEvtsPC;
 
  bool writeRootFile;
  TFile *rootFile;
  TNtuple *ntuple;
  bool tpMode;
 
  bool runOnAllHitsOccupancies;
  bool runOnNoiseOccupancies;
  bool runOnInTimeOccupancies;
  std::string nameMonitoredHisto;
 
  bool bookingdone;
 
  int nChannelTotal;
  int nChannelDead;
};
 
#endif