/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/Alignment/SurveyAnalysis/plugins/SurveyInputTrackerFromDB.cc

Go to the documentation of this file.

#include "Alignment/CommonAlignment/interface/SurveyDet.h"
#include "Alignment/TrackerAlignment/interface/AlignableTracker.h"
#include "FWCore/ParameterSet/interface/ParameterSet.h"
#include "FWCore/Framework/interface/EventSetup.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "Geometry/Records/interface/IdealGeometryRecord.h"
#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeomBuilderFromGeometricDet.h"

// Database
#include "CondCore/DBOutputService/interface/PoolDBOutputService.h"
#include "FWCore/ServiceRegistry/interface/Service.h"
#include "CondFormats/AlignmentRecord/interface/TrackerAlignmentRcd.h"
#include "CondFormats/AlignmentRecord/interface/TrackerAlignmentErrorRcd.h"
#include "CondFormats/Alignment/interface/Alignments.h"
#include "CondFormats/Alignment/interface/AlignmentErrors.h"


#include "Alignment/SurveyAnalysis/plugins/SurveyInputTrackerFromDB.h"

SurveyInputTrackerFromDB::SurveyInputTrackerFromDB(const edm::ParameterSet& cfg)
  : textFileName( cfg.getParameter<std::string>("textFileName") ),
    theParameterSet( cfg )
{}

void SurveyInputTrackerFromDB::analyze(const edm::Event&, const edm::EventSetup& setup)
{

  if (theFirstEvent) {

        //  std::cout << "***************ENTERING INITIALIZATION******************" << std::endl;
        
        //Retrieve tracker topology from geometry
        edm::ESHandle<TrackerTopology> tTopoHandle;
        setup.get<IdealGeometryRecord>().get(tTopoHandle);
        const TrackerTopology* const tTopo = tTopoHandle.product();

        //Get map from textreader
        SurveyInputTextReader dataReader;
        dataReader.readFile( textFileName );
        uIdMap = dataReader.UniqueIdMap();
        
        edm::ESHandle<GeometricDet>  geom;
        setup.get<IdealGeometryRecord>().get(geom); 
        TrackerGeometry* tracker = TrackerGeomBuilderFromGeometricDet().build(&*geom, theParameterSet);
        
        addComponent( new AlignableTracker( tracker, tTopo ) );
        addSurveyInfo( detector() );
        
        //write out to a DB ...
        Alignments* myAlignments = detector()->alignments();
        AlignmentErrors* myAlignmentErrors = detector()->alignmentErrors();
        
        // 2. Store alignment[Error]s to DB
        edm::Service<cond::service::PoolDBOutputService> poolDbService;
        // Call service
        
        if( !poolDbService.isAvailable() ) // Die if not available
                throw cms::Exception("NotAvailable") << "PoolDBOutputService not available";
        
        poolDbService->writeOne<Alignments>( myAlignments, poolDbService->beginOfTime(), "TrackerAlignmentRcd" );
        poolDbService->writeOne<AlignmentErrors>( myAlignmentErrors, poolDbService->beginOfTime(), "TrackerAlignmentErrorRcd" );
        
        theFirstEvent = false;
  }
}

void SurveyInputTrackerFromDB::addSurveyInfo(Alignable* ali)
{
        const align::Alignables& comp = ali->components();
        unsigned int nComp = comp.size();
        
        for (unsigned int i = 0; i < nComp; ++i) addSurveyInfo(comp[i]);
        
        align::ErrorMatrix error;

        SurveyInputTextReader::MapType::const_iterator it
          = uIdMap.find(std::make_pair(ali->id(), ali->alignableObjectId()));

        if (it != uIdMap.end()){
                
                const align::Scalars& parameters = (it)->second;
                
                //move the surface
                //displacement
                align::LocalVector lvector (parameters[0], parameters[1], parameters[2]);
                align::GlobalVector gvector = ali->surface().toGlobal(lvector);
                ali->move(gvector);
                //rotation
                Basic3DVector<align::Scalar> rot_aa(parameters[3], parameters[4], parameters[5]);
                align::RotationType rotation(rot_aa, rot_aa.mag());
                ali->rotateInLocalFrame(rotation);
                
                //sets the errors for the hierarchy level
                double* errorData = error.Array();
                for (unsigned int i = 0; i < 21; ++i){errorData[i] = parameters[i+6];}
                
                ali->setSurvey( new SurveyDet(ali->surface(), error*(1e-6)) );
        }
        else {
                error = ROOT::Math::SMatrixIdentity();
                ali->setSurvey( new SurveyDet(ali->surface(), error * 1e-6) );
        }
        
}
// Plug in to framework

#include "FWCore/Framework/interface/MakerMacros.h"


DEFINE_FWK_MODULE(SurveyInputTrackerFromDB);