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/ESHandle.h"
#include "FWCore/ParameterSet/interface/FileInPath.h"
#include "DQMServices/Core/interface/DQMStore.h"
#include "DataFormats/MuonDetId/interface/DTLayerId.h"
#include "Geometry/Records/interface/MuonGeometryRecord.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::ESGetToken<DTGeometry, MuonGeometryRecord> muonGeomToken_;
  const 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