|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
|
#ifndef DTOccupancyTestML_H
#define DTOccupancyTestML_H
/** \class DTOccupancyTest
* *
* DQM Test Client
*
* \author G. Cerminara - University and INFN Torino
*
* threadsafe version (//-) oct/nov 2014 - WATWanAbdullah -ncpp-um-my
*
*
*/
#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:
/// Constructor
DTOccupancyTestML(const edm::ParameterSet &ps);
/// Destructor
~DTOccupancyTestML() override;
protected:
/// BeginRun
void beginRun(edm::Run const &run, edm::EventSetup const &context) override;
/// Endjob
void dqmEndJob(DQMStore::IBooker &, DQMStore::IGetter &) override;
/// DQM Client Diagnostic
void dqmEndLuminosityBlock(DQMStore::IBooker &,
DQMStore::IGetter &,
edm::LuminosityBlock const &,
edm::EventSetup const &) override;
private:
/// book the summary histograms
void bookHistos(DQMStore::IBooker &, const int wheelId, std::string folder, std::string histoTag);
/// Get the ME name
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
|