DTVDriftWriter.cc

Go to the documentation of this file.

/*
 *  See header file for a description of this class.
 *
 *  Author of original version: M. Giunta
 *  \author A. Vilela Pereira
 */

#include "DTVDriftWriter.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "Geometry/DTGeometry/interface/DTGeometry.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
#include "Geometry/DTGeometry/interface/DTSuperLayer.h"

#include "CondFormats/DTObjects/interface/DTMtime.h"
#include "CondFormats/DataRecord/interface/DTMtimeRcd.h"

#include "CalibMuon/DTCalibration/interface/DTCalibDBUtils.h"
#include "CalibMuon/DTCalibration/interface/DTVDriftPluginFactory.h"
#include "CalibMuon/DTCalibration/interface/DTVDriftBaseAlgo.h"

#include <string>
#include <vector>

using namespace std;
using namespace edm;

DTVDriftWriter::DTVDriftWriter(const ParameterSet& pset):
  granularity_( pset.getUntrackedParameter<string>("calibGranularity","bySL") ),
  vDriftAlgo_{DTVDriftPluginFactory::get()->create(pset.getParameter<string>("vDriftAlgo"),
                                                   pset.getParameter<ParameterSet>("vDriftAlgoConfig"))}
 {

  LogVerbatim("Calibration") << "[DTVDriftWriter]Constructor called!";

  if(granularity_ != "bySL")
     throw cms::Exception("Configuration") << "[DTVDriftWriter] Check parameter calibGranularity: " << granularity_ << " option not available.";
}

DTVDriftWriter::~DTVDriftWriter(){
  LogVerbatim("Calibration") << "[DTVDriftWriter]Destructor called!";
}

void DTVDriftWriter::beginRun(const edm::Run& run, const edm::EventSetup& setup) {
  // Get the map of ttrig from the Setup
  ESHandle<DTMtime> mTime;
  setup.get<DTMtimeRcd>().get(mTime);
  mTimeMap_ = &*mTime;

  // Get geometry from Event Setup
  setup.get<MuonGeometryRecord>().get(dtGeom_);
  // Pass EventSetup to concrete implementation
  vDriftAlgo_->setES(setup); 
}

void DTVDriftWriter::endJob() {
  // Create the object to be written to DB
  DTMtime* mTimeNewMap = new DTMtime();

  if(granularity_ == "bySL") {    
     // Get all the sls from the geometry
     const vector<const DTSuperLayer*>& superLayers = dtGeom_->superLayers(); 
     auto sl = superLayers.begin();
     auto sl_end = superLayers.end();
     for(; sl != sl_end; ++sl){
        DTSuperLayerId slId = (*sl)->id();
        // Get original value from DB
        float vDrift = 0., resolution = 0.;
        // vdrift is cm/ns , resolution is cm
        int status = mTimeMap_->get(slId,vDrift,resolution,DTVelocityUnits::cm_per_ns);

        // Compute vDrift
        try{
           dtCalibration::DTVDriftData vDriftData = vDriftAlgo_->compute(slId);
           float vDriftNew = vDriftData.vdrift;
           float resolutionNew = vDriftData.resolution; 
           // vdrift is cm/ns , resolution is cm
           mTimeNewMap->set(slId,
                            vDriftNew,
                    resolutionNew,
                    DTVelocityUnits::cm_per_ns);
           LogVerbatim("Calibration") << "vDrift for: " << slId
                                      << " Mean " << vDriftNew
                                      << " Resolution " << resolutionNew;
        } catch(cms::Exception& e){
           LogError("Calibration") << e.explainSelf();
           // Go back to original value in case of error
           if(!status){  
              mTimeNewMap->set(slId,
                               vDrift,
                               resolution,
                               DTVelocityUnits::cm_per_ns);
              LogVerbatim("Calibration") << "Keep original vDrift for: " << slId
                                         << " Mean " << vDrift
                                         << " Resolution " << resolution;
           }
        }
     } // End of loop on superlayers
  }

  // Write the vDrift object to DB
  LogVerbatim("Calibration") << "[DTVDriftWriter]Writing vdrift object to DB!";
  string record = "DTMtimeRcd";
  DTCalibDBUtils::writeToDB<DTMtime>(record, mTimeNewMap);
}