StoreESCondition.cc

Go to the documentation of this file.

#include "CondTools/Ecal/plugins/StoreESCondition.h"
 
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
#include <fstream>
#include <iostream>
#include <string>
#include <cstring>
#include <ctime>
#include <unistd.h>
 
using std::string;
 
StoreESCondition::StoreESCondition(const edm::ParameterSet& iConfig) {
  prog_name_ = "StoreESCondition";
 
  logfile_ = iConfig.getParameter<std::string>("logfile");
 
  esgain_ = iConfig.getParameter<unsigned int>("gain");
 
  typedef std::vector<edm::ParameterSet> Parameters;
  Parameters toPut = iConfig.getParameter<Parameters>("toPut");
  for (Parameters::iterator itToPut = toPut.begin(); itToPut != toPut.end(); ++itToPut) {
    inpFileName_.push_back(itToPut->getUntrackedParameter<std::string>("inputFile"));
    objectName_.push_back(itToPut->getUntrackedParameter<std::string>("conditionType"));
    since_.push_back(itToPut->getUntrackedParameter<unsigned int>("since"));
  }
}
 
void StoreESCondition::endJob() {
  edm::Service<cond::service::PoolDBOutputService> mydbservice;
  if (!mydbservice.isAvailable()) {
    edm::LogError("StoreESCondition") << "PoolDBOutputService is unavailable"
                                      << "\n";
    return;
  }
 
  bool toAppend = false;
  // copy a string to the char *
  std::string message = "finished OK\n";
  size_t messageSize = message.size() + 1;
  char* messChar = new char[messageSize];
  strncpy(messChar, message.c_str(), messageSize);
 
  for (unsigned int i = 0; i < objectName_.size(); i++) {
    cond::Time_t newTime;
 
    if (mydbservice->isNewTagRequest(objectName_[i] + std::string("Rcd"))) {
      // This is the first object for this tag.
      // Append mode should be off.
      // newTime is the end of this new objects IOV.
      newTime = mydbservice->beginOfTime();
    } else {
      // There should already be an object in the DB for this tag.
      // Append IOV mode should be on.
      // newTime is the beginning of this new objects IOV.
      toAppend = true;
      newTime = (cond::Time_t)since_[i];
    }
    edm::LogInfo("StoreESCondition") << "Reading " << objectName_[i] << " from file and writing to DB with newTime "
                                     << newTime << "\n";
    if (objectName_[i] == "ESChannelStatus") {
      edm::LogInfo("StoreESCondition") << " ESChannelStatus file " << inpFileName_[i] << "\n";
      ESChannelStatus* mycali = readESChannelStatusFromFile(inpFileName_[i].c_str());
      edm::LogInfo("StoreESCondition") << " ESChannelStatus file read "
                                       << "\n";
      if (!toAppend) {
        edm::LogInfo("StoreESCondition") << " before create "
                                         << "\n";
        mydbservice->createNewIOV<ESChannelStatus>(mycali, newTime, mydbservice->endOfTime(), "ESChannelStatusRcd");
        edm::LogInfo("StoreESCondition") << " after create "
                                         << "\n";
      } else {
        edm::LogInfo("StoreESCondition") << " before append "
                                         << "\n";
        mydbservice->appendSinceTime<ESChannelStatus>(mycali, newTime, "ESChannelStatusRcd");
        edm::LogInfo("StoreESCondition") << " after append "
                                         << "\n";
      }
    } else if (objectName_[i] == "ESIntercalibConstants") {
      ESIntercalibConstants* myintercalib = readESIntercalibConstantsFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESIntercalibConstants>(
            myintercalib, newTime, mydbservice->endOfTime(), "ESIntercalibConstantsRcd");
      } else {
        mydbservice->appendSinceTime<ESIntercalibConstants>(myintercalib, newTime, "ESIntercalibConstantsRcd");
      }
    } else if (objectName_[i] == "ESTimeSampleWeights") {
      ESTimeSampleWeights* myintercalib = readESTimeSampleWeightsFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESTimeSampleWeights>(
            myintercalib, newTime, mydbservice->endOfTime(), "ESTimeSampleWeightsRcd");
      } else {
        mydbservice->appendSinceTime<ESTimeSampleWeights>(myintercalib, newTime, "ESTimeSampleWeightsRcd");
      }
    } else if (objectName_[i] == "ESGain") {
      ESGain* myintercalib = readESGainFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESGain>(myintercalib, newTime, mydbservice->endOfTime(), "ESGainRcd");
      } else {
        mydbservice->appendSinceTime<ESGain>(myintercalib, newTime, "ESGainRcd");
      }
    } else if (objectName_[i] == "ESMissingEnergyCalibration") {
      ESMissingEnergyCalibration* myintercalib = readESMissingEnergyFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESMissingEnergyCalibration>(
            myintercalib, newTime, mydbservice->endOfTime(), "ESMissingEnergyCalibrationRcd");
      } else {
        mydbservice->appendSinceTime<ESMissingEnergyCalibration>(
            myintercalib, newTime, "ESMissingEnergyCalibrationRcd");
      }
    } else if (objectName_[i] == "ESRecHitRatioCuts") {
      ESRecHitRatioCuts* myintercalib = readESRecHitRatioCutsFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESRecHitRatioCuts>(
            myintercalib, newTime, mydbservice->endOfTime(), "ESRecHitRatioCutsRcd");
      } else {
        mydbservice->appendSinceTime<ESRecHitRatioCuts>(myintercalib, newTime, "ESRecHitRatioCutsRcd");
      }
    } else if (objectName_[i] == "ESThresholds") {
      ESThresholds* myintercalib = readESThresholdsFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESThresholds>(myintercalib, newTime, mydbservice->endOfTime(), "ESThresholdsRcd");
      } else {
        mydbservice->appendSinceTime<ESThresholds>(myintercalib, newTime, "ESThresholdsRcd");
      }
    } else if (objectName_[i] == "ESPedestals") {
      ESPedestals* myintercalib = readESPedestalsFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESPedestals>(myintercalib, newTime, mydbservice->endOfTime(), "ESPedestalsRcd");
      } else {
        mydbservice->appendSinceTime<ESPedestals>(myintercalib, newTime, "ESPedestalsRcd");
      }
    } else if (objectName_[i] == "ESEEIntercalibConstants") {
      ESEEIntercalibConstants* myintercalib = readESEEIntercalibConstantsFromFile(inpFileName_[i].c_str());
      if (!toAppend) {
        mydbservice->createNewIOV<ESEEIntercalibConstants>(
            myintercalib, newTime, mydbservice->endOfTime(), "ESEEIntercalibConstantsRcd");
      } else {
        mydbservice->appendSinceTime<ESEEIntercalibConstants>(myintercalib, newTime, "ESEEIntercalibConstantsRcd");
      }
    } else {
      edm::LogError("StoreESCondition") << "Object " << objectName_[i] << " is not supported by this program."
                                        << "\n";
    }
    // if more records write here else if ....
 
    writeToLogFileResults(messChar);
 
    edm::LogInfo("StoreESCondition") << "Finished endJob"
                                     << "\n";
  }
 
  delete[] messChar;
}
 
StoreESCondition::~StoreESCondition() {}
 
void StoreESCondition::analyze(const edm::Event& evt, const edm::EventSetup& evtSetup) {}
 
void StoreESCondition::writeToLogFile(string a, string b, unsigned long long since) {
  FILE* outFile;  // output log file for appending
  outFile = fopen(logfile_.c_str(), "a");
  if (!outFile) {
    edm::LogError("StoreESCondition") << "*** Can not open file: " << logfile_;
    return;
  }
  char header[256];
  fillHeader(header);
  char appendMode[10];
  if (since != 0)
    strcpy(appendMode, "append");
  else
    strcpy(appendMode, "create");
 
  //fprintf(outFile, "%s %s condition from file %s written into DB for SM %d (mapped to SM %d) in %s mode (since run %u)\n",
  //header, a.c_str(), b.c_str(),  appendMode, (unsigned int)since);
 
  fclose(outFile);  // close out file
}
 
void StoreESCondition::writeToLogFileResults(char* arg) {
  FILE* outFile;  // output log file for appending
  outFile = fopen(logfile_.c_str(), "a");
  if (!outFile) {
    edm::LogError("StoreESCondition") << "*** Can not open file: " << logfile_;
    return;
  }
  char header[256];
  fillHeader(header);
  fprintf(outFile, "%s %s\n", header, arg);
  fclose(outFile);  // close out file
}
 
void StoreESCondition::fillHeader(char* header) {
  time_t rawtime;
  struct tm* timeinfo;
  time(&rawtime);
  timeinfo = localtime(&rawtime);
  char user[50];
  strcpy(user, getlogin());
  strcpy(header, asctime(timeinfo));
  strcpy(header, user);
}
 
ESThresholds* StoreESCondition::readESThresholdsFromFile(const char* inputFile) {
  std::ifstream ESThresholdsFile(edm::FileInPath(inputFile).fullPath().c_str());
  float ts2, zs;  //2nd time sample, ZS threshold
  ESThresholdsFile >> ts2;
  ESThresholdsFile >> zs;
  ESThresholds* esThresholds = new ESThresholds(ts2, zs);
 
  return esThresholds;
}
 
ESEEIntercalibConstants* StoreESCondition::readESEEIntercalibConstantsFromFile(const char* inputFile) {
  std::ifstream ESEEIntercalibFile(edm::FileInPath(inputFile).fullPath().c_str());
  float gammaLow0, alphaLow0, gammaHigh0, alphaHigh0, gammaLow1, alphaLow1, gammaHigh1, alphaHigh1, gammaLow2,
      alphaLow2, gammaHigh2, alphaHigh2, gammaLow3, alphaLow3, gammaHigh3, alphaHigh3;
  //  const float ESEEIntercalibValue[16];
  //  for (int i = 0; i < 16; ++i) {
  //    ESEEIntercalibFile >> ESEEIntercalibValue[i];
  //  }
  //  ESEEIntercalibConstants* eseeIntercalibConstants = new ESEEIntercalibConstants(ESEEIntercalibValue);
  ESEEIntercalibFile >> gammaLow0;
  ESEEIntercalibFile >> alphaLow0;
  ESEEIntercalibFile >> gammaHigh0;
  ESEEIntercalibFile >> alphaHigh0;
  ESEEIntercalibFile >> gammaLow1;
  ESEEIntercalibFile >> alphaLow1;
  ESEEIntercalibFile >> gammaHigh1;
  ESEEIntercalibFile >> alphaHigh1;
  ESEEIntercalibFile >> gammaLow2;
  ESEEIntercalibFile >> alphaLow2;
  ESEEIntercalibFile >> gammaHigh2;
  ESEEIntercalibFile >> alphaHigh2;
  ESEEIntercalibFile >> gammaLow3;
  ESEEIntercalibFile >> alphaLow3;
  ESEEIntercalibFile >> gammaHigh3;
  ESEEIntercalibFile >> alphaHigh3;
  ESEEIntercalibConstants* eseeIntercalibConstants = new ESEEIntercalibConstants(gammaLow0,
                                                                                 alphaLow0,
                                                                                 gammaHigh0,
                                                                                 alphaHigh0,
                                                                                 gammaLow1,
                                                                                 alphaLow1,
                                                                                 gammaHigh1,
                                                                                 alphaHigh1,
                                                                                 gammaLow2,
                                                                                 alphaLow2,
                                                                                 gammaHigh2,
                                                                                 alphaHigh2,
                                                                                 gammaLow3,
                                                                                 alphaLow3,
                                                                                 gammaHigh3,
                                                                                 alphaHigh3);
 
  return eseeIntercalibConstants;
}
 
ESMissingEnergyCalibration* StoreESCondition::readESMissingEnergyFromFile(const char* inputFile) {
  std::ifstream ESMissingEnergyFile(edm::FileInPath(inputFile).fullPath().c_str());
  float ConstAEta0, ConstBEta0, ConstAEta1, ConstBEta1, ConstAEta2, ConstBEta2, ConstAEta3, ConstBEta3;
  ESMissingEnergyFile >> ConstAEta0;
  ESMissingEnergyFile >> ConstBEta0;
  ESMissingEnergyFile >> ConstAEta1;
  ESMissingEnergyFile >> ConstBEta1;
  ESMissingEnergyFile >> ConstAEta2;
  ESMissingEnergyFile >> ConstBEta2;
  ESMissingEnergyFile >> ConstAEta3;
  ESMissingEnergyFile >> ConstBEta3;
  ESMissingEnergyCalibration* esMissingEnergy = new ESMissingEnergyCalibration(
      ConstAEta0, ConstBEta0, ConstAEta1, ConstBEta1, ConstAEta2, ConstBEta2, ConstAEta3, ConstBEta3);
 
  return esMissingEnergy;
}
 
ESPedestals* StoreESCondition::readESPedestalsFromFile(const char* inputFile) {
  ESPedestals* esPedestals = new ESPedestals();
 
  // int ped[2][2][40][40][32];
  // for (int i=0; i<2; ++i)
  //  for (int j=0; j<2; ++j)
  //    for (int k=0; k<40; ++k)
  //    for (int m=0; m<40; ++m)
  //      for (int n=0; n<32; ++n)
  //        ped[i][j][k][m][n] = 0;
 
  int ped[ESDetId::IZ_NUM][ESDetId::PLANE_MAX][ESDetId::IX_MAX][ESDetId::IY_MAX][ESDetId::ISTRIP_MAX] = {};
 
  int iz, ip, ix, iy, is, ped_, zside;
  std::ifstream pedestalFile(edm::FileInPath(inputFile).fullPath().c_str());
 
  for (int i = 0; i < 137216; ++i) {
    pedestalFile >> iz >> ip >> ix >> iy >> is >> ped_;
 
    zside = (iz == -1) ? 1 : 0;
    ped[zside][ip - 1][ix - 1][iy - 1][is - 1] = ped_;
  }
 
  for (int iz = -1; iz <= 1; ++iz) {
    if (iz == 0)
      continue;
    zside = (iz == -1) ? 1 : 0;
 
    for (int iplane = ESDetId::PLANE_MIN; iplane <= ESDetId::PLANE_MAX; ++iplane)
      for (int ix = ESDetId::IX_MIN; ix <= ESDetId::IX_MAX; ++ix)
        for (int iy = ESDetId::IY_MIN; iy <= ESDetId::IY_MAX; ++iy)
          for (int istrip = ESDetId::ISTRIP_MIN; istrip <= ESDetId::ISTRIP_MAX; ++istrip) {
            ESPedestals::Item ESitem;
            ESitem.mean = ped[zside][iplane - 1][ix - 1][iy - 1][istrip - 1];
            ESitem.rms = 3;  // LG : 3, HG : 6
 
            if (ESDetId::validDetId(istrip, ix, iy, iplane, iz)) {
              ESDetId esId(istrip, ix, iy, iplane, iz);
              esPedestals->insert(std::make_pair(esId.rawId(), ESitem));
            }
          }
  }
 
  return esPedestals;
}
 
ESRecHitRatioCuts* StoreESCondition::readESRecHitRatioCutsFromFile(const char* inputFile) {
  std::ifstream ESRecHitRatioCutsFile(edm::FileInPath(inputFile).fullPath().c_str());
 
  float r12Low, r23Low, r12High, r23High;
  ESRecHitRatioCutsFile >> r12Low;
  ESRecHitRatioCutsFile >> r23Low;
  ESRecHitRatioCutsFile >> r12High;
  ESRecHitRatioCutsFile >> r23High;
  ESRecHitRatioCuts* esRecHitRatioCuts = new ESRecHitRatioCuts(r12Low, r23Low, r12High, r23High);
  //  cout<<"We are in RH ratio cut : gain : " << esgain_<<endl;
 
  // HG (R12Low, R23Low, R12High, R23High)
  //  ESRecHitRatioCuts* esRecHitRatioCuts;
  //if (esgain_ == 2) esRecHitRatioCuts = new ESRecHitRatioCuts(-99999., 0.24, 0.61, 2.23);
  //  if (esgain_ == 2) esRecHitRatioCuts = new ESRecHitRatioCuts(-99999., 0.61, 0.24, 2.23); // HG
  //  else esRecHitRatioCuts = new ESRecHitRatioCuts(-99999., 0.2, 0.39, 2.8);
  // LG (R12Low, R23Low, R12High, R23High)
  //ESRecHitRatioCuts* esRecHitRatioCuts = new ESRecHitRatioCuts(-99999., 99999., -99999., 99999.); // first running
  //ESRecHitRatioCuts* esRecHitRatioCuts = new ESRecHitRatioCuts(-99999., 0.2, 0.39, 2.8);
 
  return esRecHitRatioCuts;
}
 
ESGain* StoreESCondition::readESGainFromFile(const char* inputFile) {
  std::ifstream amplFile(edm::FileInPath(inputFile).fullPath().c_str());
 
  int gain;
  amplFile >> gain;
  edm::LogInfo("StoreESCondition") << "gain : " << gain << "\n";
 
  ESGain* esGain = new ESGain(gain);  // 1: LG, 2: HG
  return esGain;
}
 
ESTimeSampleWeights* StoreESCondition::readESTimeSampleWeightsFromFile(const char* inputFile) {
  std::ifstream amplFile(edm::FileInPath(inputFile).fullPath().c_str());
 
  float w[3];
  for (int k = 0; k < 3; ++k) {
    float ww;
    amplFile >> ww;
    w[k] = ww;
    edm::LogInfo("StoreESCondition") << "weight : " << k << " " << w[k] << "\n";
  }
 
  ESTimeSampleWeights* esWeights = new ESTimeSampleWeights(w[0], w[1], w[2]);
  return esWeights;
}
 
ESIntercalibConstants* StoreESCondition::readESIntercalibConstantsFromFile(const char* inputFile) {
  ESIntercalibConstants* ical = new ESIntercalibConstants();
 
  std::ifstream mipFile(edm::FileInPath(inputFile).fullPath().c_str());
 
  for (int i = 0; i < 137216; ++i) {
    int iz, ip, ix, iy, is;
    double mip;
    mipFile >> iz >> ip >> ix >> iy >> is >> mip;
    //if (mip <20 || mip> 70) cout<<iz<<" "<<ip<<" "<<ix<<" "<<iy<<" "<<is<<" "<<mip<<endl; // HG
    // LG : HG MIP/6/1.14
    //mip = mip/6/1.14;
    // LG : HG MIP/6
    if (esgain_ == 1)
      mip = mip / 6.;  // LG
    if (mip < 20 || mip > 70)
      edm::LogInfo("StoreESCondition") << iz << " " << ip << " " << ix << " " << iy << " " << is << " " << mip
                                       << "\n";  // LG
 
    if (ESDetId::validDetId(is, ix, iy, ip, iz)) {
      ESDetId esId(is, ix, iy, ip, iz);
      ical->setValue(esId.rawId(), mip);
    }
  }
 
  return ical;
}
 
ESChannelStatus* StoreESCondition::readESChannelStatusFromFile(const char* inputFile) {
  int z[1000], p[1000], x[1000], y[1000], nsensors;
  std::ifstream statusFile(edm::FileInPath(inputFile).fullPath().c_str());
  statusFile >> nsensors;
  edm::LogInfo("StoreESCondition") << " nsensors " << nsensors << "\n";
  if (nsensors >= 1000) {
    edm::LogInfo("StoreESCondition") << " *** value too high, modify the method!***"
                                     << "\n";
    exit(-1);
  }
  for (int i = 0; i < nsensors; ++i) {
    statusFile >> z[i] >> p[i] >> x[i] >> y[i];
  }
  ESChannelStatus* ecalStatus = new ESChannelStatus();
  int Nbstatus = 0, Nbstrip = 0;
  for (int istrip = ESDetId::ISTRIP_MIN; istrip <= ESDetId::ISTRIP_MAX; istrip++) {
    for (int ix = ESDetId::IX_MIN; ix <= ESDetId::IX_MAX; ix++) {
      for (int iy = ESDetId::IY_MIN; iy <= ESDetId::IY_MAX; iy++) {
        for (int iplane = 1; iplane <= 2; iplane++) {
          for (int izeta = -1; izeta <= 1; izeta = izeta + 2) {
            //      if (izeta==0) continue;
            //      try {
 
            //ESDetId Plane iplane Zside izeta
            //      if(!ESDetId::validDetId(istrip,ix,iy,iplane,izeta)) cout << " Unvalid DetId" << endl;
            //      else {
            if (ESDetId::validDetId(istrip, ix, iy, iplane, izeta)) {
              ESDetId anESId(istrip, ix, iy, iplane, izeta);
              int status = 0;
              //          std::ifstream statusFile(edm::FileInPath(inputFile).fullPath().c_str());
              Nbstrip++;
              for (int i = 0; i < nsensors; ++i) {
                if (izeta == z[i] && iplane == p[i] && ix == x[i] && iy == y[i])
                  status = 1;
              }
              if (status == 1) {
                Nbstatus++;
                if (istrip == 1)
                  edm::LogInfo("StoreESCondition") << " Bad channel ix " << ix << " iy " << iy << " iplane " << iplane
                                                   << " iz " << izeta << "\n";  // print only once
              }
              ecalStatus->setValue(anESId, status);
              //          statusFile.close();
            }  // valid DetId
            //      catch ( cms::Exception &e ) { }
          }  // loop over z
        }    //  loop over plane
      }      //   loop over y
    }        //    loop over x
  }          //     loop over strips
  edm::LogInfo("StoreESCondition") << " Nb of strips " << Nbstrip << " Number of bad channels " << Nbstatus << "\n";
  statusFile.close();
 
  // overwrite the statuses which are in the file
  return ecalStatus;
}