SimTracker/TrackerMaterialAnalysis/plugins/ListIds.cc - master - githubmirror/cmssw

SimTracker/TrackerMaterialAnalysis/plugins/ListIds.cc
169 строк7 KB
Shahzad Malik Muzaffar
[SIMULATION] Drop Geometry/CommonDetUnit package
28 май 2026, 12:59
28 май 2026, 12:593cef132
#include <string>
#include <vector>
#include <iostream>
#include <algorithm>

#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/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Utilities/interface/InputTag.h"
#include "FWCore/ParameterSet/interface/types.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/Math/interface/Vector3D.h"
#include "DetectorDescription/Core/interface/DDFilteredView.h"
#include "DetectorDescription/Core/interface/DDCompactView.h"
#include "DetectorDescription/Core/interface/DDMaterial.h"
#include "Geometry/CommonTopologies/interface/GeomDet.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "Geometry/TrackerNumberingBuilder/interface/CmsTrackerStringToEnum.h"
#include "Geometry/TrackerNumberingBuilder/interface/GeometricDet.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"

static bool dddGetStringRaw(const DDFilteredView &view, const std::string &name, std::string &value) {
  DDValue parameter(name);
  std::vector<const DDsvalues_type *> result;
  view.specificsV(result);
  for (std::vector<const DDsvalues_type *>::iterator it = result.begin(); it != result.end(); ++it) {
    if (DDfetch(*it, parameter)) {
      if (parameter.strings().size() == 1) {
        value = parameter.strings().front();
        return true;
      } else {
        return false;
      }
    }
  }
  return false;
}

/*
static inline
double dddGetDouble(const std::string & s, const DDFilteredView & view) {
  std::string value;
  if (dddGetStringRaw(view, s, value))
    return double(::atof(value.c_str()));
  else
    return NAN;
}
*/

static inline std::string dddGetString(const std::string &s, const DDFilteredView &view) {
  std::string value;
  if (dddGetStringRaw(view, s, value))
    return value;
  else
    return std::string();
}

static inline std::ostream &operator<<(std::ostream &out, const math::XYZVector &v) {
  return out << "(" << v.rho() << ", " << v.z() << ", " << v.phi() << ")";
}

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

private:
  void analyze(const edm::Event &, const edm::EventSetup &) override;
  void beginJob() override {}
  void endJob() override;
  // List of material names used to select specific detectors.
  // Names are matched literally, w/o any usage of regexp.
  // Names should also be specified with the correct namespace,
  // otherwise the matching will fail.
  bool printMaterial_;
  std::vector<std::string> materials_;
  edm::ESGetToken<DDCompactView, IdealGeometryRecord> dddToken_;
  edm::ESGetToken<TrackerGeometry, TrackerDigiGeometryRecord> geoToken_;
};

ListIds::ListIds(const edm::ParameterSet &pset)
    : printMaterial_(pset.getUntrackedParameter<bool>("printMaterial")),
      materials_(pset.getUntrackedParameter<std::vector<std::string> >("materials")),
      dddToken_(esConsumes()),
      geoToken_(esConsumes()) {}

ListIds::~ListIds() {}

void ListIds::analyze(const edm::Event &evt, const edm::EventSetup &setup) {
  std::cout << "______________________________ DDD ______________________________" << std::endl;
  auto hDdd = setup.getTransientHandle(dddToken_);

  std::string attribute = "TkDDDStructure";
  CmsTrackerStringToEnum theCmsTrackerStringToEnum;
  DDSpecificsHasNamedValueFilter filter{attribute};
  DDFilteredView fv(*hDdd, filter);
  if (theCmsTrackerStringToEnum.type(dddGetString(attribute, fv)) != GeometricDet::Tracker) {
    fv.firstChild();
    if (theCmsTrackerStringToEnum.type(dddGetString(attribute, fv)) != GeometricDet::Tracker)
      throw cms::Exception("Configuration") << "The first child of the DDFilteredView is not what is expected \n"
                                            << dddGetString(attribute, fv);
  }

  std::cout << std::fixed << std::setprecision(3);
  bool printAnyMaterial = (std::find(materials_.begin(), materials_.end(), "ANY") != materials_.end());
  do {
    // print the full hierarchy of all elements whose material
    // has been specified by the user. An empty list of
    // materials will print no elements. The special
    // keyword ANY (in any location of the vector)
    // will select all elements.
    if (printAnyMaterial ||
        (std::find(materials_.begin(), materials_.end(), fv.logicalPart().material().name().fullname()) !=
         materials_.end())) {
      // start from 2 to skip the leading /OCMS[0]/CMSE[1] part
      const DDGeoHistory &history = fv.geoHistory();
      std::cout << '/';
      for (unsigned int h = 2; h < history.size(); ++h) {
        std::cout << '/' << history[h].logicalPart().name().ns() << ":" << history[h].logicalPart().name().name() << '['
                  << history[h].copyno() << ']';
      }
      if (printMaterial_)
        std::cout << " Material: |" << fv.logicalPart().material().name() << "|";
      // DD3Vector and DDTranslation are the same type as math::XYZVector
      math::XYZVector position = fv.translation() / 10.;  // mm -> cm
      std::cout << "\t" << position << std::endl;
    }
  } while (fv.next());
  std::cout << std::endl;

  std::cout << "______________________________ std::vector<GeomDet*> from TrackerGeometry::dets() "
               "______________________________"
            << std::endl;
  auto const &geo = setup.getData(geoToken_);

  std::cout << std::fixed << std::setprecision(3);
  auto const &dets = geo.dets();
  for (unsigned int i = 0; i < dets.size(); ++i) {
    const GeomDet &det = *dets[i];

    // Surface::PositionType is a typedef for Point3DBase<float,GlobalTag> a.k.a. GlobalPoint
    const Surface::PositionType &p = det.position();
    math::XYZVector position(p.x(), p.y(), p.z());

    std::cout << det.subDetector() << '\t' << det.geographicalId().det() << '\t' << det.geographicalId().subdetId()
              << '\t' << det.geographicalId().rawId() << "\t" << position;
    const std::vector<const GeomDet *> &parts = det.components();
    if (!parts.empty()) {
      std::cout << "\t[" << parts[0]->geographicalId().rawId();
      for (unsigned int j = 1; j < parts.size(); ++j)
        std::cout << '\t' << parts[j]->geographicalId().rawId();
      std::cout << ']';
    }
    std::cout << std::endl;
  }
}

void ListIds::endJob() {}

//-------------------------------------------------------------------------
// define as a plugin
#include "FWCore/PluginManager/interface/ModuleDef.h"
#include "FWCore/Framework/interface/MakerMacros.h"
DEFINE_FWK_MODULE(ListIds);