PGeometricDetBuilder.cc

Go to the documentation of this file.

#include "FWCore/Framework/interface/one/EDAnalyzer.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/Event.h"
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESTransientHandle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "FWCore/Framework/interface/MakerMacros.h"
#include "CondFormats/GeometryObjects/interface/PGeometricDet.h"
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "Geometry/TrackerNumberingBuilder/interface/GeometricDet.h"
#include "DetectorDescription/Core/interface/DDCompactView.h"
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "FWCore/MessageLogger/interface/MessageLogger.h"
#include <vector>

class PGeometricDetBuilder : public edm::one::EDAnalyzer<edm::one::WatchRuns>
{
public:
  
  PGeometricDetBuilder( const edm::ParameterSet& ) {}
  
  void beginRun(edm::Run const& iEvent, edm::EventSetup const&) override;
  void analyze(edm::Event const& iEvent, edm::EventSetup const&) override {}
  void endRun(edm::Run const& iEvent, edm::EventSetup const&) override {}
private:
  void putOne ( const GeometricDet* gd, PGeometricDet* pgd, int lev );
};

void
PGeometricDetBuilder::beginRun( const edm::Run&, edm::EventSetup const& es) 
{
  PGeometricDet* pgd = new PGeometricDet;
  edm::Service<cond::service::PoolDBOutputService> mydbservice;
  if( !mydbservice.isAvailable() ){
    edm::LogError("PGeometricDetBuilder")<<"PoolDBOutputService unavailable";
    return;
  }
  edm::ESTransientHandle<DDCompactView> pDD;
  edm::ESHandle<GeometricDet> rDD;
  es.get<IdealGeometryRecord>().get( pDD );
  es.get<IdealGeometryRecord>().get( rDD );
  const GeometricDet* tracker = &(*rDD);

  // so now I have the tracker itself. loop over all its components to store them.
  putOne(tracker, pgd, 0);
  std::vector<const GeometricDet*> tc = tracker->components();
  std::vector<const GeometricDet*>::const_iterator git = tc.begin();
  std::vector<const GeometricDet*>::const_iterator egit = tc.end();
  int count=0;
  int lev = 1;
  for (; git!= egit; ++git) {  // one level below "tracker"
    putOne(*git, pgd, lev);
    std::vector<const GeometricDet*> inone = (*git)->components();
    if ( inone.size() == 0 )  ++count;
    std::vector<const GeometricDet*>::const_iterator git2 = inone.begin();
    std::vector<const GeometricDet*>::const_iterator egit2 = inone.end();
    ++lev;
    for (; git2 != egit2; ++git2) { // level 2
      putOne(*git2, pgd, lev);
      std::vector<const GeometricDet*> intwo= (*git2)->components();
      if ( intwo.size() == 0 )  ++count;
      std::vector<const GeometricDet*>::const_iterator git3 = intwo.begin();
      std::vector<const GeometricDet*>::const_iterator egit3 = intwo.end();
      ++lev;
      for (; git3 != egit3; ++git3) { // level 3
    putOne(*git3, pgd, lev);
    std::vector<const GeometricDet*> inthree= (*git3)->components();
    if ( inthree.size() == 0 )  ++count;
    std::vector<const GeometricDet*>::const_iterator git4 = inthree.begin();
    std::vector<const GeometricDet*>::const_iterator egit4 = inthree.end();
    ++lev;
    for (; git4 != egit4; ++git4) { //level 4
      putOne(*git4, pgd, lev);
      std::vector<const GeometricDet*> infour= (*git4)->components();
      if ( infour.size() == 0 )  ++count;
      std::vector<const GeometricDet*>::const_iterator git5 = infour.begin();
      std::vector<const GeometricDet*>::const_iterator egit5 = infour.end();
      ++lev;
      for (; git5 != egit5; ++git5) { // level 5
        putOne(*git5, pgd, lev);
        std::vector<const GeometricDet*> infive= (*git5)->components();
        if ( infive.size() == 0 )  ++count;
        std::vector<const GeometricDet*>::const_iterator git6 = infive.begin();
        std::vector<const GeometricDet*>::const_iterator egit6 = infive.end();
        ++lev;
        for (; git6 != egit6; ++git6) { //level 6
          putOne(*git6, pgd, lev);
          std::vector<const GeometricDet*> insix= (*git6)->components();
          if ( insix.size() == 0 )  ++count;
        } // level 6
        --lev;
      } // level 5
      --lev;
    } // level 4
    --lev;
      } //level 3
      --lev;
    } // level 2
    --lev;
  }
  std::vector<const GeometricDet*> modules =  tracker->deepComponents();
  if ( mydbservice->isNewTagRequest("IdealGeometryRecord") ) {
    mydbservice->createNewIOV<PGeometricDet>( pgd,mydbservice->beginOfTime(),mydbservice->endOfTime(),"IdealGeometryRecord");
  } else {
    edm::LogError("PGeometricDetBuilder")<<"PGeometricDetBuilder Tag already present";
  }
}
  
void PGeometricDetBuilder::putOne ( const GeometricDet* gd, PGeometricDet* pgd, int lev ) {

  PGeometricDet::Item item;
  DDTranslation tran = gd->translation();
  DDRotationMatrix rot = gd->rotation();
  DD3Vector x, y, z;
  rot.GetComponents(x, y, z);
  item._name           = gd->name().name();
  item._ns             = gd->name().ns();
  item._level          = lev;
  item._x              = tran.X();
  item._y              = tran.Y();
  item._z              = tran.Z();
  item._phi            = gd->phi();
  item._rho            = gd->rho();
  item._a11            = x.X();
  item._a12            = y.X();
  item._a13            = z.X();
  item._a21            = x.Y();
  item._a22            = y.Y();
  item._a23            = z.Y();
  item._a31            = x.Z();
  item._a32            = y.Z();
  item._a33            = z.Z();
  item._shape          = gd->shape();
  item._type           = gd->type();
  if(gd->shape()==1){
    item._params0=gd->params()[0];
    item._params1=gd->params()[1];
    item._params2=gd->params()[2];
    item._params3=0;
    item._params4=0;
    item._params5=0;
    item._params6=0;
    item._params7=0;
    item._params8=0;
    item._params9=0;
    item._params10=0;
  }else if(gd->shape()==3){
    item._params0=gd->params()[0];
    item._params1=gd->params()[1];
    item._params2=gd->params()[2];
    item._params3=gd->params()[3];
    item._params4=gd->params()[4];
    item._params5=gd->params()[5];
    item._params6=gd->params()[6];
    item._params7=gd->params()[7];
    item._params8=gd->params()[8];
    item._params9=gd->params()[9];
    item._params10=gd->params()[10];
  }else{
    item._params0=0;
    item._params1=0;
    item._params2=0;
    item._params3=0;
    item._params4=0;
    item._params5=0;
    item._params6=0;
    item._params7=0;
    item._params8=0;
    item._params9=0;
    item._params10=0;
  } 
  item._geographicalID = gd->geographicalID();
  item._radLength      = gd->radLength();
  item._xi             = gd->xi();
  item._pixROCRows     = gd->pixROCRows();
  item._pixROCCols     = gd->pixROCCols();
  item._pixROCx        = gd->pixROCx();
  item._pixROCy        = gd->pixROCy();
  item._stereo         = gd->stereo();
  item._siliconAPVNum = gd->siliconAPVNum();

  GeometricDet::nav_type const & nt = gd->navType();
  size_t nts = nt.size();
  item._numnt = nts;
  std::vector<int> tempnt(nt.begin(),nt.end());
  for ( size_t extrant = nt.size(); extrant < 11; ++extrant ) {
    tempnt.push_back(-1);
  } 
  item._nt0 = tempnt[0];
  item._nt1 = tempnt[1];
  item._nt2 = tempnt[2];
  item._nt3 = tempnt[3];
  item._nt4 = tempnt[4];
  item._nt5 = tempnt[5];
  item._nt6 = tempnt[6];
  item._nt7 = tempnt[7];
  item._nt8 = tempnt[8];
  item._nt9 = tempnt[9];
  item._nt10 = tempnt[10];

  pgd->pgeomdets_.push_back ( item );
}

DEFINE_FWK_MODULE(PGeometricDetBuilder);