CMS 3D CMS Logo

/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_6_2_5/src/DPGAnalysis/SiStripTools/plugins/DetIdSelectorTest.cc

Go to the documentation of this file.
00001 // -*- C++ -*-
00002 //
00003 // Package:    SiStripTools
00004 // Class:      DetIdSelectorTest
00005 // 
00013 //
00014 // Original Author:  Andrea Venturi
00015 //         Created:  Tue Jul 19 11:56:00 CEST 2009
00016 //
00017 //
00018 
00019 
00020 // system include files
00021 #include <memory>
00022 
00023 // user include files
00024 #include <vector>
00025 #include <string>
00026 
00027 #include "FWCore/Framework/interface/Frameworkfwd.h"
00028 #include "FWCore/Framework/interface/EDAnalyzer.h"
00029 
00030 #include "FWCore/Framework/interface/Event.h"
00031 #include "FWCore/Framework/interface/MakerMacros.h"
00032 
00033 #include "FWCore/ParameterSet/interface/ParameterSet.h"
00034 
00035 #include "FWCore/MessageLogger/interface/MessageLogger.h"
00036 
00037 #include "FWCore/ServiceRegistry/interface/Service.h"
00038 #include "CommonTools/UtilAlgos/interface/TFileService.h"
00039 
00040 #include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
00041 #include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"
00042 #include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
00043 
00044 #include "CalibTracker/SiStripCommon/interface/SiStripDetInfoFileReader.h"
00045 #include "CalibTracker/SiPixelESProducers/interface/SiPixelDetInfoFileReader.h"
00046 #include "CommonTools/UtilAlgos/interface/DetIdSelector.h"
00047 #include "FWCore/ParameterSet/interface/FileInPath.h"
00048 
00049 //******** Single include for the TkMap *************
00050 #include "DQM/SiStripCommon/interface/TkHistoMap.h" 
00051 #include "CommonTools/TrackerMap/interface/TrackerMap.h"
00052 //***************************************************
00053 
00054 //
00055 // class decleration
00056 //
00057 
00058 class DetIdSelectorTest : public edm::EDAnalyzer {
00059 public:
00060   explicit DetIdSelectorTest(const edm::ParameterSet&);
00061   ~DetIdSelectorTest();
00062 
00063 
00064 private:
00065   virtual void beginJob() ;
00066   virtual void analyze(const edm::Event&, const edm::EventSetup&) override;
00067   virtual void endJob() ;
00068 
00069       // ----------member data ---------------------------
00070 
00071 
00072   std::vector<DetIdSelector> detidsels_;
00073   TkHistoMap *tkhisto_;
00074   TrackerMap tkmap_;
00075 
00076   
00077 };
00078 
00079 //
00080 // constants, enums and typedefs
00081 //
00082 
00083 //
00084 // static data member definitions
00085 //
00086 
00087 //
00088 // constructors and destructor
00089 //
00090 DetIdSelectorTest::DetIdSelectorTest(const edm::ParameterSet& iConfig):
00091   detidsels_(), tkhisto_(new TkHistoMap("SelectorTest","SelectorTest",-1)),tkmap_()
00092 {
00093    //now do what ever initialization is needed
00094 
00095   std::vector<edm::ParameterSet> selconfigs = iConfig.getParameter<std::vector<edm::ParameterSet> >("selections");
00096 
00097   for(std::vector<edm::ParameterSet>::const_iterator selconfig=selconfigs.begin();selconfig!=selconfigs.end();++selconfig) {
00098     DetIdSelector selection(*selconfig);
00099     detidsels_.push_back(selection);
00100   }
00101 
00102   tkmap_.setPalette(1);
00103   tkmap_.addPixel(true);
00104 
00105 }
00106 
00107 
00108 DetIdSelectorTest::~DetIdSelectorTest()
00109 {
00110  
00111    // do anything here that needs to be done at desctruction time
00112    // (e.g. close files, deallocate resources etc.)
00113 
00114 }
00115 
00116 
00117 //
00118 // member functions
00119 //
00120 
00121 // ------------ method called to for each event  ------------
00122 void
00123 DetIdSelectorTest::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
00124 {
00125    using namespace edm;
00126    
00127    {
00128      SiStripDetInfoFileReader * reader=edm::Service<SiStripDetInfoFileReader>().operator->();
00129      
00130      //   SiStripDetInfoFileReader reader(edm::FileInPath("CalibTracker/SiStripCommon/data/SiStripDetInfo.dat"));
00131      //   SiStripDetInfoFileReader reader;
00132      
00133      const std::vector<uint32_t>& detids = reader->getAllDetIds();
00134      
00135      for(std::vector<uint32_t>::const_iterator detid=detids.begin();detid!=detids.end();++detid) {
00136        
00137        LogDebug("DetID") << *detid;
00138        int index=0;
00139        for(std::vector<DetIdSelector>::const_iterator detidsel=detidsels_.begin();detidsel!=detidsels_.end();++detidsel) {
00140          if(detidsel->isSelected(*detid)) {
00141            LogDebug("selected") << "Selected by selection " << index;
00142            unsigned int det = *detid;
00143            tkhisto_->add(det,index);
00144            tkmap_.fill_current_val(det,index);
00145          }
00146          ++index;
00147        }
00148        
00149      }
00150    }
00151 
00152    {
00153      edm::FileInPath fp("CalibTracker/SiPixelESProducers/data/PixelSkimmedGeometry.txt");
00154      
00155      SiPixelDetInfoFileReader pxlreader(fp.fullPath());
00156      const std::vector<uint32_t>& detids = pxlreader.getAllDetIds();
00157      
00158      for(std::vector<uint32_t>::const_iterator detid=detids.begin();detid!=detids.end();++detid) {
00159        
00160        LogDebug("DetID") << *detid;
00161        int index=0;
00162        for(std::vector<DetIdSelector>::const_iterator detidsel=detidsels_.begin();detidsel!=detidsels_.end();++detidsel) {
00163          if(detidsel->isSelected(*detid)) {
00164            LogDebug("selected") << "Selected by selection " << index;
00165            unsigned int det = *detid;
00166            //      tkhisto_->add(det,index);
00167            tkmap_.fill_current_val(det,index);
00168          }
00169          ++index;
00170        }
00171        
00172      }
00173    }
00174     
00175  
00176    /*
00177      edm::ESHandle<TrackerGeometry> pDD;
00178      iSetup.get<TrackerDigiGeometryRecord>().get( pDD );
00179      
00180      for (TrackerGeometry::DetContainer::const_iterator it = pDD->dets().begin(); it != pDD->dets().end(); it++){
00181      
00182      if(dynamic_cast<PixelGeomDetUnit*>((*it))!=0){
00183      DetId detId = (*it)->geographicalId();
00184      LogDebug("DetID") << detId.rawId();
00185      int index=0;
00186      for(std::vector<DetIdSelector>::const_iterator detidsel=detidsels_.begin();detidsel!=detidsels_.end();++detidsel) {
00187      if(detidsel->isSelected(detId)) {
00188      LogDebug("selected") << " Selected by selection " << index;
00189      //  tkhisto_->add(det,index);
00190      tkmap_.fill_current_val(detId.rawId(),index);
00191      }
00192      ++index;
00193      }
00194      
00195      }
00196      
00197      }
00198    */   
00199 }
00200 
00201 // ------------ method called once each job just before starting event loop  ------------
00202 void 
00203 DetIdSelectorTest::beginJob()
00204 {
00205 
00206 }
00207 
00208 // ------------ method called once each job just after ending the event loop  ------------
00209 void 
00210 DetIdSelectorTest::endJob() {
00211 
00212 
00213   //  tkhisto_->dumpInTkMap(&tkmap);
00214   std::string mapname = "SelectorTest.png";
00215   tkmap_.save(true,0,0,mapname,5700,2400);
00216 
00217   std::string rootmapname = "TKMap_Selectortest.root";
00218   tkhisto_->save(rootmapname);
00219 }
00220 
00221 
00222 //define this as a plug-in
00223 DEFINE_FWK_MODULE(DetIdSelectorTest);