Tracker_OldtoNewConverter.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    Tracker_OldtoNewConverter.cc
// Class:      Tracker_OldtoNewConverter
//
//
// Original Author:  Nhan Tran
//         Created:  Mon Jul 16m 16:56:34 CDT 2007
// $Id: Tracker_OldtoNewConverter.cc,v 1.2 2010/01/04 18:24:37 mussgill Exp $
//
//
 
// system include files
#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/MakerMacros.h"
 
#include <algorithm>
#include "TTree.h"
#include "TFile.h"
 
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
#include <fstream>
#include <iostream>
 
//
// class decleration
//
 
class Tracker_OldtoNewConverter : public edm::one::EDAnalyzer<> {
public:
  explicit Tracker_OldtoNewConverter(const edm::ParameterSet&);
  ~Tracker_OldtoNewConverter() override;
 
private:
  void beginJob() override;
  void analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) override;
  void endJob() override;
 
  void createMap();
  void addBranches();
 
  // ----------member data ---------------------------
 
  std::string m_conversionType;
  std::string m_textFile;
  std::string m_inputFile;
  std::string m_outputFile;
  std::string m_treeName;
 
  std::map<uint32_t, uint32_t> theMap;
 
  TFile* m_inputTFile;
  TFile* m_outputTFile;
  TTree* m_inputTree;
  TTree* m_outputTree;
 
  uint32_t rawid_i, rawid_f;
  //int rawid_i, rawid_f;
  double x_i, y_i, z_i, a_i, b_i, c_i;
  double x_f, y_f, z_f, a_f, b_f, c_f;
};
 
//
// constants, enums and typedefs
//
 
//
// static data member definitions
//
 
//
// constructors and destructor
//
Tracker_OldtoNewConverter::Tracker_OldtoNewConverter(const edm::ParameterSet& iConfig)
    : m_inputTFile(nullptr),
      m_outputTFile(nullptr),
      m_inputTree(nullptr),
      m_outputTree(nullptr),
      rawid_i(0),
      rawid_f(0),
      x_i(0.),
      y_i(0.),
      z_i(0.),
      a_i(0.),
      b_i(0.),
      c_i(0.),
      x_f(0.),
      y_f(0.),
      z_f(0.),
      a_f(0.),
      b_f(0.),
      c_f(0.) {
  m_conversionType = iConfig.getUntrackedParameter<std::string>("conversionType");
  m_inputFile = iConfig.getUntrackedParameter<std::string>("inputFile");
  m_outputFile = iConfig.getUntrackedParameter<std::string>("outputFile");
  m_textFile = iConfig.getUntrackedParameter<std::string>("textFile");
  m_treeName = iConfig.getUntrackedParameter<std::string>("treeName");
}
 
Tracker_OldtoNewConverter::~Tracker_OldtoNewConverter() {}
 
//
// member functions
//
 
// ------------ method called to for each event  ------------
void Tracker_OldtoNewConverter::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {}
 
// ------------ method called once each job just before starting event loop  ------------
void Tracker_OldtoNewConverter::beginJob() {
  m_inputTFile = new TFile(m_inputFile.c_str());
  m_outputTFile = new TFile(m_outputFile.c_str(), "RECREATE");
 
  m_inputTree = (TTree*)m_inputTFile->Get(m_treeName.c_str());
  m_outputTree = new TTree(m_treeName.c_str(), m_treeName.c_str());
 
  createMap();
  addBranches();
 
  uint32_t nEntries = m_inputTree->GetEntries();
  uint32_t iter = 0;
  for (uint32_t i = 0; i < nEntries; ++i) {
    m_inputTree->GetEntry(i);
    std::map<uint32_t, uint32_t>::const_iterator it = theMap.find(rawid_i);
 
    if (it == theMap.end()) {
      edm::LogInfo("ERROR") << "Error: couldn't find rawId: " << rawid_i;
      iter++;
    } else {
      rawid_f = (it)->second;
      x_f = x_i;
      y_f = y_i;
      z_f = z_i;
      a_f = a_i;
      b_f = b_i;
      c_f = c_i;
      m_outputTree->Fill();
    }
  }
  edm::LogInfo("errors") << "Couldn't find: " << iter;
  m_outputTFile->cd();
  m_outputTree->Write();
  m_outputTFile->Close();
}
 
void Tracker_OldtoNewConverter::createMap() {
  std::ifstream myfile(m_textFile.c_str());
  if (!myfile.is_open())
    throw cms::Exception("FileAccess") << "Unable to open input text file";
 
  uint32_t oldid;
  uint32_t newid;
 
  uint32_t ctr = 0;
  while (!myfile.eof() && myfile.good()) {
    myfile >> oldid >> newid;
 
    //depends on conversion type: OldtoNew or NewtoOld
    std::pair<uint32_t, uint32_t> pairType;
    if (m_conversionType == "OldtoNew") {
      pairType.first = oldid;
      pairType.second = newid;
    }
    if (m_conversionType == "NewtoOld") {
      pairType.first = newid;
      pairType.second = oldid;
    }
 
    theMap.insert(pairType);
 
    if (myfile.fail())
      break;
 
    ctr++;
  }
  edm::LogInfo("Check") << ctr << " alignables read.";
}
 
void Tracker_OldtoNewConverter::addBranches() {
  m_inputTree->SetBranchAddress("rawid", &rawid_i);
  m_inputTree->SetBranchAddress("x", &x_i);
  m_inputTree->SetBranchAddress("y", &y_i);
  m_inputTree->SetBranchAddress("z", &z_i);
  m_inputTree->SetBranchAddress("alpha", &a_i);
  m_inputTree->SetBranchAddress("beta", &b_i);
  m_inputTree->SetBranchAddress("gamma", &c_i);
 
  m_outputTree->Branch("rawid", &rawid_f, "rawid/I");
  m_outputTree->Branch("x", &x_f, "x/D");
  m_outputTree->Branch("y", &y_f, "y/D");
  m_outputTree->Branch("z", &z_f, "z/D");
  m_outputTree->Branch("alpha", &a_f, "alpha/D");
  m_outputTree->Branch("beta", &b_f, "beta/D");
  m_outputTree->Branch("gamma", &c_f, "gamma/D");
}
 
// ------------ method called once each job just after ending the event loop  ------------
void Tracker_OldtoNewConverter::endJob() {}
 
//define this as a plug-in
DEFINE_FWK_MODULE(Tracker_OldtoNewConverter);