MisalignedMuonESProducer.cc

Go to the documentation of this file.

 
// Framework
#include "FWCore/Utilities/interface/Exception.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Framework/interface/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducts.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
 
// Conditions database
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
 
// Alignment
#include "Alignment/MuonAlignment/interface/AlignableMuon.h"
#include "Geometry/CSCGeometry/interface/CSCGeometry.h"
#include "Alignment/MuonAlignment/interface/MuonScenarioBuilder.h"
#include "Alignment/CommonAlignment/interface/Alignable.h"
#include "Geometry/CommonTopologies/interface/GeometryAligner.h"
#include "Geometry/Records/interface/MuonGeometryRecord.h"
#include "Geometry/DTGeometryBuilder/src/DTGeometryBuilderFromDDD.h"
#include "Geometry/CSCGeometryBuilder/src/CSCGeometryBuilderFromDDD.h"
 
#include <memory>
 
class MisalignedMuonESProducer : public edm::ESProducer {
public:
  MisalignedMuonESProducer(const edm::ParameterSet& p);
 
  ~MisalignedMuonESProducer() override;
 
  edm::ESProducts<std::unique_ptr<DTGeometry>, std::unique_ptr<CSCGeometry> > produce(const MuonGeometryRecord&);
 
  void saveToDB();
 
private:
  const bool theSaveToDB;               
  const edm::ParameterSet theScenario;  
 
  std::string theDTAlignRecordName, theDTErrorRecordName;
  std::string theCSCAlignRecordName, theCSCErrorRecordName;
 
  Alignments* dt_Alignments;
  AlignmentErrorsExtended* dt_AlignmentErrorsExtended;
  Alignments* csc_Alignments;
  AlignmentErrorsExtended* csc_AlignmentErrorsExtended;
};
 
//__________________________________________________________________________________________________
//__________________________________________________________________________________________________
//__________________________________________________________________________________________________
 
//__________________________________________________________________________________________________
MisalignedMuonESProducer::MisalignedMuonESProducer(const edm::ParameterSet& p)
    : theSaveToDB(p.getUntrackedParameter<bool>("saveToDbase")),
      theScenario(p.getParameter<edm::ParameterSet>("scenario")),
      theDTAlignRecordName("DTAlignmentRcd"),
      theDTErrorRecordName("DTAlignmentErrorExtendedRcd"),
      theCSCAlignRecordName("CSCAlignmentRcd"),
      theCSCErrorRecordName("CSCAlignmentErrorExtendedRcd") {
  setWhatProduced(this);
}
 
//__________________________________________________________________________________________________
MisalignedMuonESProducer::~MisalignedMuonESProducer() {}
 
//__________________________________________________________________________________________________
edm::ESProducts<std::unique_ptr<DTGeometry>, std::unique_ptr<CSCGeometry> > MisalignedMuonESProducer::produce(
    const MuonGeometryRecord& iRecord) {
  edm::LogInfo("MisalignedMuon") << "Producer called";
 
  // Create the Muon geometry from ideal geometry
  edm::ESTransientHandle<DDCompactView> cpv;
  iRecord.getRecord<IdealGeometryRecord>().get(cpv);
 
  edm::ESHandle<MuonGeometryConstants> mdc;
  iRecord.getRecord<IdealGeometryRecord>().get(mdc);
 
  DTGeometryBuilderFromDDD DTGeometryBuilder;
  CSCGeometryBuilderFromDDD CSCGeometryBuilder;
 
  auto theDTGeometry = std::make_unique<DTGeometry>();
  DTGeometryBuilder.build(*theDTGeometry, &(*cpv), *mdc);
  auto theCSCGeometry = std::make_unique<CSCGeometry>();
  CSCGeometryBuilder.build(*theCSCGeometry, &(*cpv), *mdc);
 
  // Create the alignable hierarchy
  AlignableMuon* theAlignableMuon = new AlignableMuon(&(*theDTGeometry), &(*theCSCGeometry));
 
  // Create misalignment scenario
  MuonScenarioBuilder scenarioBuilder(theAlignableMuon);
  scenarioBuilder.applyScenario(theScenario);
 
  // Get alignments and errors
  dt_Alignments = theAlignableMuon->dtAlignments();
  dt_AlignmentErrorsExtended = theAlignableMuon->dtAlignmentErrorsExtended();
  csc_Alignments = theAlignableMuon->cscAlignments();
  csc_AlignmentErrorsExtended = theAlignableMuon->cscAlignmentErrorsExtended();
 
  // Misalign the EventSetup geometry
  GeometryAligner aligner;
 
  aligner.applyAlignments<DTGeometry>(&(*theDTGeometry), dt_Alignments, dt_AlignmentErrorsExtended, AlignTransform());
  aligner.applyAlignments<CSCGeometry>(
      &(*theCSCGeometry), csc_Alignments, csc_AlignmentErrorsExtended, AlignTransform());
 
  // Write alignments to DB
  if (theSaveToDB)
    this->saveToDB();
 
  edm::LogInfo("MisalignedMuon") << "Producer done";
 
  return edm::es::products(std::move(theDTGeometry), std::move(theCSCGeometry));
}
 
//__________________________________________________________________________________________________
void MisalignedMuonESProducer::saveToDB(void) {
  // Call service
  edm::Service<cond::service::PoolDBOutputService> poolDbService;
  if (!poolDbService.isAvailable())  // Die if not available
    throw cms::Exception("NotAvailable") << "PoolDBOutputService not available";
 
  // Store DT alignments and errors
  poolDbService->writeOne<Alignments>(&(*dt_Alignments), poolDbService->beginOfTime(), theDTAlignRecordName);
  poolDbService->writeOne<AlignmentErrorsExtended>(
      &(*dt_AlignmentErrorsExtended), poolDbService->beginOfTime(), theDTErrorRecordName);
 
  // Store CSC alignments and errors
  poolDbService->writeOne<Alignments>(&(*csc_Alignments), poolDbService->beginOfTime(), theCSCAlignRecordName);
  poolDbService->writeOne<AlignmentErrorsExtended>(
      &(*csc_AlignmentErrorsExtended), poolDbService->beginOfTime(), theCSCErrorRecordName);
}
//__________________________________________________________________________________________________
 
DEFINE_FWK_EVENTSETUP_MODULE(MisalignedMuonESProducer);