PhysicalPartsTree.cc

Go to the documentation of this file.

#include <memory>

#include <FWCore/Framework/interface/EDAnalyzer.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 <DetectorDescription/Core/interface/DDCompactView.h>
#include <DetectorDescription/Core/interface/DDValue.h>
#include <DetectorDescription/Core/interface/DDExpandedView.h>
#include "DetectorDescription/Core/interface/DDName.h"
#include "DetectorDescription/Core/interface/DDSolid.h"
#include "DetectorDescription/Core/interface/DDMaterial.h"

#include <Geometry/Records/interface/IdealGeometryRecord.h>

#include <iostream>
#include <istream>
#include <fstream>
#include <string>
#include <vector>
#include <algorithm>
#include <utility>

class PhysicalPartsTree : public edm::EDAnalyzer {

public:
 
  explicit PhysicalPartsTree( const edm::ParameterSet& );
  ~PhysicalPartsTree();
  
  virtual void analyze( const edm::Event&, const edm::EventSetup& ) override;
  virtual void beginRun( const edm::Run&, const edm::EventSetup& ) override;
  
private: 

  //std::string filename_;  
};

PhysicalPartsTree::PhysicalPartsTree( const edm::ParameterSet& iConfig )
{

}


PhysicalPartsTree::~PhysicalPartsTree()
{
}

void PhysicalPartsTree::analyze( const edm::Event& iEvent, const edm::EventSetup& iSetup ) { 
  std::cout << "analyze does nothing" << std::endl;
}

void PhysicalPartsTree::beginRun( const edm::Run&, const edm::EventSetup& iSetup ) {
  //set a tolerance for "near zero"
  double tolerance = 1.0e-3;

  std::string physicalPartsTreeFileName("PHYSICALPARTSTREE.dat");
  std::string logicalPartTypesFileName("LOGICALPARTTYPES.dat");
  std::string nominalPlacementsFileName("NOMINALPLACEMENTS.dat");
  std::string detectorPartsFileName("DETECTORPARTS.dat");
  std::string pospartsGraphFileName("POSPARTSGRAPH.dat");

  std::ofstream physicalPartsTreeOS(physicalPartsTreeFileName.c_str());
  std::ofstream logicalPartTypesOS(logicalPartTypesFileName.c_str());
  std::ofstream nominalPlacementsOS(nominalPlacementsFileName.c_str());
  std::ofstream detectorPartsOS(detectorPartsFileName.c_str());
  std::ofstream pospartsGraphOS(pospartsGraphFileName.c_str());


  std::string slpname;
  std::string scategory;
  std::vector<std::string> lpname_vec;


  std::cout << "PhysicalPartsTree Analyzer..." << std::endl;
  edm::ESTransientHandle<DDCompactView> pDD;

  iSetup.get<IdealGeometryRecord>().get( "", pDD );
  
  const DDCompactView & cpv = *pDD;
  DDExpandedView epv(cpv);
  int pospartid=0;
  size_t lastfound;
  std::string lgname;
   while(epv.next()){
 
    //for table physicalpartstree
    std::ostringstream parent, child,logicalpartid,parentid;
    child << epv.logicalPart().name();
    const DDGeoHistory & hist = epv.geoHistory();
    parent << hist[hist.size()-2].logicalPart().name();
    logicalpartid << epv.geoHistory();
    parentid<<hist[hist.size()-2];
    lgname=logicalpartid.str();
    lastfound=lgname.find_last_of("/\\");
    lgname=lgname.substr(lastfound+1,lgname.size());
    physicalPartsTreeOS<<lgname<<","<< parentid.str() << "," << child.str()<<std::endl;
//    std::cout << "Logical Part Name= "<<logicalpartid.str() << std::endl;
     
    //for table nominalPlacements
    bool reflection = false;

    DD3Vector x, y, z;
    epv.rotation().GetComponents(x, y, z);
    //Hep3Vector xv = epv.rotation().colX();
    //Hep3Vector yv = epv.rotation().colY();
    //Hep3Vector zv = epv.rotation().colZ();
    //if ( xv.cross(yv) * zv  < 0) {
    //            reflection = true;
    //          }
    if ( (1.0 + (x.Cross(y)).Dot(z)) <= tolerance ) {
      reflection = true;
    }
    std::vector<double> comps(9);
    epv.rotation().GetComponents(comps.begin(), comps.end());
    nominalPlacementsOS<< logicalpartid.str()<<","
               << epv.translation().X() << "," 
               << epv.translation().Y() << ","  
               << epv.translation().Z()<< "," 
//             << epv.rotation().xx()<< "," 
//             << epv.rotation().xy()<< "," 
//             << epv.rotation().xz()<< "," 
//             << epv.rotation().yx()<< "," 
//             << epv.rotation().yy()<< "," 
//             << epv.rotation().yz()<< "," 
//             << epv.rotation().zx()<< "," 
//             << epv.rotation().zy()<< "," 
//             << epv.rotation().zz()<< "," 
               << comps[0]<<","
               << comps[1]<<","
               << comps[2]<<","
               << comps[3]<<","
               << comps[4]<<","
               << comps[5]<<","
               << comps[6]<<","
               << comps[7]<<","
               << comps[8]<<","
               << (int)reflection
               <<std::endl;


    //for table DetectorParts
    detectorPartsOS <<epv.logicalPart().solid().toString() << ","
            <<epv.logicalPart().material().toString() << ","
            <<logicalpartid.str()
            <<std::endl;
   
    //for table PosPartsGraph 

    pospartsGraphOS<<++pospartid<< ","<<epv.copyno()<< ","
    <<parent.str() << "," << child.str()<<std::endl;


    //for table logicalPartTypes
    //this table is special because the DPNAMEs are subset of Physicalpart_id
    //while we're looping all the Physicalpart_ids in epv, so we just 
    //keep the unique DPNAMEs into a vector and save them in looping epv.
    // if there's no lpname in lpname_vec
    //>> the OS file.
 
    slpname=epv.logicalPart().toString();

    std::vector<std::string>::iterator it = find(lpname_vec.begin(), lpname_vec.end(), slpname);    
   
    if (it==lpname_vec.end()){
      if(DDEnums::categoryName(epv.logicalPart().category())){     
    std::ostringstream category;
    category<<DDEnums::categoryName(epv.logicalPart().category());
    scategory=category.str();
    std::string unspec="unspecified";
    if(scategory==unspec){
      lpname_vec.push_back(slpname);
      logicalPartTypesOS <<slpname<< ","
                 <<"DETECTORPARTS"<<std::endl;
    }
    else{
      lpname_vec.push_back(slpname);
      logicalPartTypesOS <<slpname<< ","<<scategory<<std::endl;

    }
      }
      else{
    lpname_vec.push_back(slpname);
    logicalPartTypesOS <<slpname<< ","
               <<"DETECTORPARTS"<<std::endl;

      }
    }



  } //finish looping through expanded view.
 

  physicalPartsTreeOS.close();
  logicalPartTypesOS.close();
  nominalPlacementsOS.close();
  detectorPartsOS.close();
  pospartsGraphOS.close();
}

//define this as a plug-in
DEFINE_FWK_MODULE(PhysicalPartsTree);