ListIds.cc

Go to the documentation of this file.

#include <string>
#include <vector>
#include <iostream>

#include "FWCore/Framework/interface/EDAnalyzer.h"
#include "FWCore/Framework/interface/Event.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.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/CommonDetUnit/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::EDAnalyzer
{
public:
  ListIds(const edm::ParameterSet &);
  virtual ~ListIds();
  
private:
  void analyze(const edm::Event &, const edm::EventSetup &) override;
  void beginJob() override {}
  void endJob() override;
};

ListIds::ListIds(const edm::ParameterSet &) {
}

ListIds::~ListIds() {
}

void
ListIds::analyze(const edm::Event& evt, const edm::EventSetup& setup){
  std::cout << "______________________________ DDD ______________________________" << std::endl;
  edm::ESTransientHandle<DDCompactView> hDdd;
  setup.get<IdealGeometryRecord>().get( hDdd );

  std::string attribute = "TkDDDStructure";
  CmsTrackerStringToEnum theCmsTrackerStringToEnum;
  DDSpecificsFilter filter;
  filter.setCriteria(DDValue(attribute, "any", 0), DDSpecificsFilter::not_equals);
  DDFilteredView fv(*hDdd); 
  fv.addFilter(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);
  do {
    // print the full hierarchy of all Silicon items
    if (fv.logicalPart().material().name() == "materials:Silicon") {
  
      // 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().name() << '[' << history[h].copyno() << ']';
      }

      // 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;
  edm::ESHandle<TrackerGeometry> hGeo;
  setup.get<TrackerDigiGeometryRecord>().get( hGeo );

  std::cout << std::fixed << std::setprecision(3);
  const std::vector<GeomDet*> & dets = hGeo->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.size()) {
      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);