|
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
|
#include "CommonTools/MVAUtils/interface/TMVAZipReader.h"
#include "FWCore/Utilities/interface/Exception.h"
#include <cstdio>
#include <cstdlib>
#include <zlib.h>
using namespace std;
// From http://stackoverflow.com/questions/874134/find-if-string-endswith-another-string-in-c
bool reco::details::hasEnding(std::string const& fullString, std::string const& ending) {
if (fullString.length() >= ending.length()) {
return (0 == fullString.compare(fullString.length() - ending.length(), ending.length(), ending));
} else {
return false;
}
}
char* reco::details::readGzipFile(const std::string& weightFile) {
FILE* f = fopen(weightFile.c_str(), "r");
if (f == nullptr) {
throw cms::Exception("InvalidFileState") << "Failed to open MVA file = " << weightFile << " !!\n";
}
int magic;
int size;
fread(&magic, 4, 1, f);
fseek(f, -4, SEEK_END);
fread(&size, 4, 1, f);
fclose(f);
// printf("%x, %i\n", magic, size);
gzFile file = gzopen(weightFile.c_str(), "r");
int bytes_read;
char* buffer = (char*)malloc(size);
bytes_read = gzread(file, buffer, size - 1);
buffer[bytes_read] = '\0';
if (!gzeof(file)) {
int err;
const char* error_string;
error_string = gzerror(file, &err);
if (err) {
free(buffer);
throw cms::Exception("InvalidFileState") << "Error while reading gzipped file = " << weightFile << " !!\n"
<< error_string;
}
}
gzclose(file);
return buffer;
}
TMVA::IMethod* reco::details::loadTMVAWeights(TMVA::Reader* reader,
const std::string& method,
const std::string& weightFile,
bool verbose) {
TMVA::IMethod* ptr = nullptr;
verbose = false;
if (verbose)
std::cout << "Booking TMVA Reader with " << method << " and weight file: " << weightFile << std::endl;
if (reco::details::hasEnding(weightFile, ".xml")) {
if (verbose)
std::cout << "Weight file is pure xml." << std::endl;
// Let TMVA read the file
ptr = reader->BookMVA(method, weightFile);
} else if (reco::details::hasEnding(weightFile, ".gz") || reco::details::hasEnding(weightFile, ".gzip")) {
if (verbose)
std::cout << "Unzipping file." << std::endl;
char* c = readGzipFile(weightFile);
// We can't use tmpnam, gcc emits a warning about security.
// This is also technically insecure in the same way, since we append
// a suffix and then open another file.
char tmpFilename[] = "/tmp/tmva.XXXXXX";
int fdToUselessFile = mkstemp(tmpFilename);
std::string weight_file_name(tmpFilename);
weight_file_name += ".xml";
FILE* theActualFile = fopen(weight_file_name.c_str(), "w");
if (theActualFile != nullptr) {
// write xml
fputs(c, theActualFile);
fputs("\n", theActualFile);
fclose(theActualFile);
close(fdToUselessFile);
} else {
throw cms::Exception("CannotWriteFile") << "Error while writing file = " << weight_file_name << " !!\n";
}
if (verbose)
std::cout << "Booking MvA" << std::endl;
ptr = reader->BookMVA(method, weight_file_name);
if (verbose)
std::cout << "Cleaning up" << std::endl;
remove(weight_file_name.c_str());
remove(tmpFilename);
// Someday this will work.
// reader->BookMVA(TMVA::Types::Instance().GetMethodType(TString(method)), c);
if (verbose) {
std::cout << "Reader booked" << std::endl;
}
free(c);
} else {
throw cms::Exception("BadTMVAWeightFilename") << "I don't understand the extension on the filename: " << weightFile
<< ", it should be .xml, .gz, or .gzip" << std::endl;
}
return ptr;
}
|