|
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
|
#include "DQM/TrackerRemapper/interface/SiStripTkMaps.h"
#include <cstdint> // For uint32_t
#include <cstdlib> // For std::exit
#include <fstream>
#include <iostream>
#include <numeric> // std::accumulate
#include <sstream>
#include <string>
#include <vector>
#include "TCanvas.h"
#include "TStyle.h"
void showHelp(const std::string& scriptName) {
std::cout << "Usage: " << scriptName << " [options] <detid>\n"
<< " --input-file <filename> Specify the input file\n"
<< " --h or --help Show this help message\n"
<< " <detid> Provide DetId (list of DetIds)\n";
}
int main(int argc, char* argv[]) {
std::string inputFile;
std::vector<std::pair<uint32_t, float>> detidValues;
// If no arguments are passed or --h/--help is passed, show the help message
if (argc == 1) {
showHelp(argv[0]);
return 0;
}
// Parse command line arguments
for (int i = 1; i < argc; ++i) {
std::string arg = argv[i];
if (arg == "--h" || arg == "--help") {
showHelp(argv[0]);
return 0; // Exit after displaying help
} else if (arg == "--input-file" && i + 1 < argc) {
gStyle->SetPalette(kRainbow);
gStyle->SetNumberContours(256);
inputFile = argv[++i];
} else {
gStyle->SetPalette(1);
// Treat as DetId list if no --input-file is provided
try {
uint32_t detid = std::stoul(arg);
detidValues.emplace_back(detid, 1.0); // Default value is 1.0
} catch (const std::invalid_argument&) {
std::cerr << "Invalid DetId: " << arg << "\n";
showHelp(argv[0]);
return 1;
}
}
}
// If --input-file is provided, read from file
if (!inputFile.empty()) {
std::ifstream file(inputFile);
if (!file) {
std::cerr << "Error: Unable to open input file " << inputFile << std::endl;
return 1;
}
std::string line;
while (std::getline(file, line)) {
std::istringstream iss(line);
uint32_t detid;
float value = 1.0; // Default value
iss >> detid;
if (iss >> value) { // If a second column exists, read it as value
detidValues.emplace_back(detid, value);
} else {
detidValues.emplace_back(detid, 1.0);
}
}
}
// Create the map and fill it
SiStripTkMaps theMap("COLZ0 AL");
theMap.bookMap("Strip Tracker Map of Marked modules", "input values");
for (const auto& [detid, value] : detidValues) {
theMap.fill(detid, value);
}
// Check if all values are the same using a lambda function
bool allSame = std::all_of(detidValues.begin(), detidValues.end(), [&](const std::pair<uint32_t, float>& p) {
return p.second == detidValues[0].second;
});
TCanvas c = TCanvas("c", "c");
theMap.drawMap(c, "");
// adjust the z-axis scale
if (allSame)
theMap.setZAxisRange(0., detidValues[0].second);
c.SaveAs("SiStripsTkMaps.png");
std::cout << "Filled tracker map with " << detidValues.size() << " detids." << std::endl;
return 0;
}
|