reco::details Namespace Reference

Functions
bool	hasEnding (std::string const &fullString, std::string const &ending)
TMVA::IMethod *	loadTMVAWeights (TMVA::Reader *reader, const std::string &method, const std::string &weightFile, bool verbose=false)
char *	readGzipFile (const std::string &weightFile)

Function Documentation

bool reco::details::hasEnding	(	std::string const &	fullString,
		std::string const &	ending
	)

Definition at line 11 of file TMVAZipReader.cc.

Referenced by loadTMVAWeights().

                                                                                {
  if (fullString.length() >= ending.length()) {
    return (0 == fullString.compare(fullString.length() - ending.length(), ending.length(), ending));
  } else {
    return false;
  }
}

TMVA::IMethod * reco::details::loadTMVAWeights	(	TMVA::Reader *	reader,
		const std::string &	method,
		const std::string &	weightFile,
		bool	verbose = false
	)

Definition at line 52 of file TMVAZipReader.cc.

References HltBtagPostValidation_cff::c, gather_cfg::cout, Exception, hasEnding(), readGzipFile(), and AlCaHLTBitMon_QueryRunRegistry::string.

Referenced by BaseMVAValueMapProducer< pat::Jet >::BaseMVAValueMapProducer(), cms::PileupJPTJetIdAlgo::bookMVAReader(), MVAJetPuId::bookReader(), l1t::HGC3DClusterTMVASelector::HGC3DClusterTMVASelector(), TMVAEvaluator::initialize(), and EcalDeadChannelRecoveryBDTG< EBDetId >::loadFile().

                                                            {
  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;
}

char * reco::details::readGzipFile

(

const std::string &

weightFile

)

Definition at line 19 of file TMVAZipReader.cc.

References edmScanValgrind::buffer, runTheMatrix::err, Exception, f, FrontierConditions_GlobalTag_cff::file, and findQualityFiles::size.

Referenced by loadTMVAWeights().

                                                           {
  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;
}