GeometryReaders/XMLIdealGeometryESSource/test/PerfectGeometryAnalyzer.cc - master - githubmirror/cmssw

GeometryReaders/XMLIdealGeometryESSource/test/PerfectGeometryAnalyzer.cc
107 строк4 KB
Sunanda
Code check
21 дек 2021, 23:05
21 дек 2021, 23:053c592c6
// -*- C++ -*-
//
// Package:    PerfectGeometryAnalyzer
// Class:      PerfectGeometryAnalyzer
//
/**\class PerfectGeometryAnalyzer PerfectGeometryAnalyzer.cc test/PerfectGeometryAnalyzer/src/PerfectGeometryAnalyzer.cc

 Description: <one line class summary>

 Implementation:
     <Notes on implementation>
*/
//
// Original Author:  Tommaso Boccali
//         Created:  Tue Jul 26 08:47:57 CEST 2005
//
//

#include <memory>
#include <iostream>
#include <fstream>

#include "FWCore/Framework/interface/Frameworkfwd.h"
#include "FWCore/Framework/interface/one/EDAnalyzer.h"

#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"

#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "DetectorDescription/Core/interface/DDCompactView.h"
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "DetectorDescription/OfflineDBLoader/interface/GeometryInfoDump.h"

#include "CondFormats/Common/interface/FileBlob.h"
#include "Geometry/Records/interface/GeometryFileRcd.h"
#include "DetectorDescription/Parser/interface/DDLParser.h"
#include "DetectorDescription/Core/interface/DDRoot.h"

#include "MagneticField/Records/interface/IdealMagneticFieldRecord.h"

class PerfectGeometryAnalyzer : public edm::one::EDAnalyzer<> {
public:
  explicit PerfectGeometryAnalyzer(const edm::ParameterSet&);
  ~PerfectGeometryAnalyzer() override;

  void beginJob() override {}
  void analyze(edm::Event const&, edm::EventSetup const&) override;
  void endJob() override {}

private:
  std::string label_;
  bool isMagField_;
  bool dumpHistory_;
  bool dumpPosInfo_;
  bool dumpSpecs_;
  std::string fname_;
  int nNodes_;
  bool fromDB_;
  std::string ddRootNodeName_;
  edm::ESGetToken<DDCompactView, IdealGeometryRecord> geomToken1_;
  edm::ESGetToken<DDCompactView, IdealMagneticFieldRecord> geomToken2_;
};

PerfectGeometryAnalyzer::PerfectGeometryAnalyzer(const edm::ParameterSet& iConfig)
    : label_(iConfig.getUntrackedParameter<std::string>("label", "")),
      isMagField_(iConfig.getUntrackedParameter<bool>("isMagField", false)),
      dumpHistory_(iConfig.getUntrackedParameter<bool>("dumpGeoHistory", false)),
      dumpPosInfo_(iConfig.getUntrackedParameter<bool>("dumpPosInfo", false)),
      dumpSpecs_(iConfig.getUntrackedParameter<bool>("dumpSpecs", false)),
      fname_(iConfig.getUntrackedParameter<std::string>("outFileName", "GeoHistory")),
      nNodes_(iConfig.getUntrackedParameter<uint32_t>("numNodesToDump", 0)),
      fromDB_(iConfig.getUntrackedParameter<bool>("fromDB", false)),
      ddRootNodeName_(iConfig.getUntrackedParameter<std::string>("ddRootNodeName", "cms:OCMS")) {
  if (isMagField_) {
    label_ = "magfield";
  }
  if (!isMagField_)
    geomToken1_ = esConsumes<DDCompactView, IdealGeometryRecord>(edm::ESInputTag("", label_));
  else
    geomToken2_ = esConsumes<DDCompactView, IdealMagneticFieldRecord>(edm::ESInputTag("", label_));
}

PerfectGeometryAnalyzer::~PerfectGeometryAnalyzer() {}

void PerfectGeometryAnalyzer::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup) {
  using namespace edm;

  std::cout << "Here I am " << std::endl;

  edm::ESTransientHandle<DDCompactView> pDD =
      (!isMagField_) ? iSetup.getTransientHandle(geomToken1_) : iSetup.getTransientHandle(geomToken2_);
  if (pDD.description()) {
    edm::LogInfo("PerfectGeometryAnalyzer") << pDD.description()->type_ << " label: " << pDD.description()->label_;
  } else {
    edm::LogWarning("PerfectGeometryAnalyzer") << "NO label found pDD.description() returned false.";
  }
  if (!pDD.isValid()) {
    edm::LogError("PerfectGeometryAnalyzer") << "ESTransientHandle<DDCompactView> pDD is not valid!";
  }
  GeometryInfoDump gidump;
  gidump.dumpInfo(dumpHistory_, dumpSpecs_, dumpPosInfo_, *pDD, fname_, nNodes_);
  std::cout << "finished" << std::endl;
}

DEFINE_FWK_MODULE(PerfectGeometryAnalyzer);