MisalignedTrackerESProducer.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/ModuleFactory.h"
#include "FWCore/Framework/interface/ESProducer.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"

// Conditions database
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"

// Geometry
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeomBuilderFromGeometricDet.h"
#include "Geometry/CommonTopologies/interface/GeometryAligner.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/Records/interface/TrackerTopologyRcd.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"

// Alignment
#include "CondFormats/Alignment/interface/AlignmentErrorsExtended.h"
#include "Alignment/TrackerAlignment/interface/AlignableTracker.h"
#include "Alignment/TrackerAlignment/interface/TrackerScenarioBuilder.h"
#include "Alignment/CommonAlignment/interface/Alignable.h"

// C++
#include <memory>
#include <algorithm>


class MisalignedTrackerESProducer : public edm::ESProducer {
public:
  MisalignedTrackerESProducer(const edm::ParameterSet& p);

  ~MisalignedTrackerESProducer() override;

  std::unique_ptr<TrackerGeometry> produce(const TrackerDigiGeometryRecord& iRecord);

private:
  const bool theSaveToDB;  
  const bool
      theSaveFakeScenario;  
  const edm::ParameterSet theScenario;  
  const std::string theAlignRecordName, theErrorRecordName;
};

//__________________________________________________________________________________________________
//__________________________________________________________________________________________________
//__________________________________________________________________________________________________

//__________________________________________________________________________________________________
MisalignedTrackerESProducer::MisalignedTrackerESProducer(const edm::ParameterSet& p)
    : theSaveToDB(p.getUntrackedParameter<bool>("saveToDbase")),
      theSaveFakeScenario(p.getUntrackedParameter<bool>("saveFakeScenario")),
      theScenario(p.getParameter<edm::ParameterSet>("scenario")),
      theAlignRecordName("TrackerAlignmentRcd"),
      theErrorRecordName("TrackerAlignmentErrorExtendedRcd") {
  setWhatProduced(this);
}

//__________________________________________________________________________________________________
MisalignedTrackerESProducer::~MisalignedTrackerESProducer() {}

//__________________________________________________________________________________________________
std::unique_ptr<TrackerGeometry> MisalignedTrackerESProducer::produce(const TrackerDigiGeometryRecord& iRecord) {
  //Retrieve tracker topology from geometry
  edm::ESHandle<TrackerTopology> tTopoHandle;
  iRecord.getRecord<TrackerTopologyRcd>().get(tTopoHandle);
  const TrackerTopology* const tTopo = tTopoHandle.product();

  edm::LogInfo("MisalignedTracker") << "Producer called";

  // Create the tracker geometry from ideal geometry
  edm::ESHandle<GeometricDet> gD;
  iRecord.getRecord<IdealGeometryRecord>().get(gD);
  edm::ESHandle<PTrackerParameters> ptp;
  iRecord.getRecord<PTrackerParametersRcd>().get(ptp);
  TrackerGeomBuilderFromGeometricDet trackerBuilder;
  std::unique_ptr<TrackerGeometry> theTracker(trackerBuilder.build(&(*gD), *ptp, tTopo));

  // Create the alignable hierarchy
  auto theAlignableTracker = std::make_unique<AlignableTracker>(&(*theTracker), tTopo);

  // Create misalignment scenario, apply to geometry
  TrackerScenarioBuilder scenarioBuilder(&(*theAlignableTracker));
  scenarioBuilder.applyScenario(theScenario);
  Alignments* alignments = theAlignableTracker->alignments();
  AlignmentErrorsExtended* alignmentErrors = theAlignableTracker->alignmentErrors();

  // Store result to EventSetup
  GeometryAligner aligner;
  aligner.applyAlignments<TrackerGeometry>(&(*theTracker),
                                           alignments,
                                           alignmentErrors,
                                           AlignTransform());  // dummy global position

  // Write alignments to DB: have to sort beforhand!
  if (theSaveToDB) {
    // Call service
    edm::Service<cond::service::PoolDBOutputService> poolDbService;
    if (!poolDbService.isAvailable())  // Die if not available
      throw cms::Exception("NotAvailable") << "PoolDBOutputService not available";
    if (theSaveFakeScenario) {  // make empty!
      alignments->clear();
      alignmentErrors->clear();
    }
    poolDbService->writeOne<Alignments>(alignments, poolDbService->currentTime(), theAlignRecordName);
    poolDbService->writeOne<AlignmentErrorsExtended>(alignmentErrors, poolDbService->currentTime(), theErrorRecordName);
  } else {
    // poolDbService::writeOne takes over ownership
    // we have to delete in the case that containers are not written
    delete alignments;
    delete alignmentErrors;
  }

  edm::LogInfo("MisalignedTracker") << "Producer done";
  return theTracker;
}

DEFINE_FWK_EVENTSETUP_MODULE(MisalignedTrackerESProducer);