---

yaml --- r: 74603 b: "refs/heads/CMSSW_7_1_X" c: 8c189c3 h: "refs/heads/CMSSW_7_1_X" i: 74601: 0eb0edf 74599: 574c778 v: v3
cms-sw · Oct 2, 2009 · 8387c4f · 8387c4f
1 parent 49ac97c
commit 8387c4f
Show file tree

Hide file tree

Showing 12 changed files with 961 additions and 6 deletions.
diff --git a/[refs] b/[refs]
@@ -1,3 +1,3 @@
 ---
 refs/heads/gh-pages: 09c786f70121f131b3715aaf3464996502bbeb7e
-"refs/heads/CMSSW_7_1_X": e5c23c20f8f591c417c106e295d101d907840b26
+"refs/heads/CMSSW_7_1_X": 8c189c34b093970176470cc9835afc9b3fce0931
diff --git a/trunk/Alignment/TrackerAlignment/BuildFile b/trunk/Alignment/TrackerAlignment/BuildFile
@@ -13,15 +13,15 @@
 <use name=Geometry/CommonDetUnit>
 <use name=Geometry/Records>
 <use name=Geometry/TrackerGeometryBuilder>
+<use name=DataFormats/Alignment>
+<use name=AnalysisDataFormats/SiStripClusterInfo>
+<use name=RecoTracker/TransientTrackingRecHit>
 <export>
-  <use name=Alignment/CommonAlignment>
+#  <use name=Alignment/CommonAlignment>
   <use name=CondFormats/Alignment>
   <use name=DataFormats/SiPixelDetId>
   <use name=DataFormats/SiStripDetId>
   <use name=Geometry/CommonDetUnit>
   <lib name=AlignmentTrackerAlignment>
 </export>
 
-
-
-
diff --git a/trunk/Alignment/TrackerAlignment/plugins/AlignmentPrescaler.cc b/trunk/Alignment/TrackerAlignment/plugins/AlignmentPrescaler.cc
@@ -0,0 +1,242 @@
+#include "Alignment/TrackerAlignment/plugins/AlignmentPrescaler.h"
+
+
+AlignmentPrescaler::AlignmentPrescaler(const edm::ParameterSet &iConfig):
+  src_(iConfig.getParameter<edm::InputTag>("src")),
+  AM_(iConfig.getParameter<edm::InputTag>("assomap")),
+  prescfilename_(iConfig.getParameter<std::string>("PrescFileName")),
+  presctreename_(iConfig.getParameter<std::string>("PrescTreeName"))
+{
+  // issue the produce<>
+  produces<AliClusterValueMap>();
+  produces<AliTrackTakenClusterValueMap>();
+
+}
+
+AlignmentPrescaler::~AlignmentPrescaler(){
+  //  
+}
+
+void AlignmentPrescaler::beginJob( const edm::EventSetup & ){
+  //
+   fpresc_=new TFile(prescfilename_.c_str(),"READ");
+   tpresc_=(TTree*)fpresc_->Get(presctreename_.c_str());
+   tpresc_->BuildIndex("DetId");
+   tpresc_->SetBranchStatus("*",0);
+   tpresc_->SetBranchStatus("DetId",1);
+   tpresc_->SetBranchStatus("PrescaleFactor",1);
+   tpresc_->SetBranchStatus("PrescaleFactorOverlap",1);
+
+   detid_=0;
+   preschit_=99.0;
+   prescoverlap_=88.0;
+
+   tpresc_->SetBranchAddress("DetId",&detid_);
+   tpresc_->SetBranchAddress("PrescaleFactor",&preschit_);
+   tpresc_->SetBranchAddress("PrescaleFactorOverlap",&prescoverlap_);
+
+   myrand=new TRandom3();
+   //   myrand->SetSeed();
+
+
+   /*
+   fh=new TFile("debugfile1.root","RECREATE");
+   hrr=new TH1F("randdistr","distribution of random numbers",100,0.0,1.0);
+   hnhitssubdet=new TH1F("hitvssubdet","Distribution of hits Vs SubDet",7,0.0,7.0);
+   hnhitsTIBL3=new TH1I("hitvsphiTIBL3","Distribution of hits Vs DetId (TIB L3 only)",5848,369153044,369158892);
+   
+   totnhitspxl_=0;
+   */
+
+
+}
+
+void AlignmentPrescaler::endJob( ){
+  /*
+  cout<<"\n\n%%%%%%%% At the end of AlignmentPrescale the number of PIXEL HITS TAKEN was "<< totnhitspxl_<<endl<<endl;
+
+  fh->cd();
+  hrr->Write();
+  hnhitssubdet->Write();
+  hnhitsTIBL3->Write();
+
+
+  delete hrr;
+  fh->Close();
+  delete fh;
+  */
+
+  //
+  delete tpresc_;
+  fpresc_->Close();
+  delete fpresc_;
+  delete myrand;
+}
+
+void AlignmentPrescaler::produce(edm::Event &iEvent, const edm::EventSetup &iSetup){
+  //  std::cout<<"\n\n#################\n### Starting the AlignmentPrescaler::produce ; Event: "<<iEvent.id().run() <<", "<<iEvent.id().event()<<std::endl;
+  edm::Handle<reco::TrackCollection> Tracks;
+  iEvent.getByLabel(src_, Tracks);
+
+  //take  HitAssomap
+  edm::Handle<AliClusterValueMap> hMap;
+  iEvent.getByLabel(AM_, hMap);
+  AliClusterValueMap InValMap=*hMap;
+
+  //prepare the output of the ValueMap flagging tracks
+  std::vector<int> trackflags(Tracks->size(),0);
+
+
+  //int npxlhits=0;
+
+    //loop on tracks
+  for(std::vector<reco::Track>::const_iterator ittrk = Tracks->begin(), edtrk = Tracks->end(); ittrk != edtrk; ++ittrk){
+    //loop on tracking rechits
+    // std::cout << "Loop on hits of track #" << (ittrk - Tracks->begin()) << std::endl;
+    int nhit=0;
+    int ntakenhits=0;
+    bool firstTakenHit=false;
+
+    for (trackingRecHit_iterator ith = ittrk->recHitsBegin(), edh = ittrk->recHitsEnd(); ith != edh; ++ith) {
+      const TrackingRecHit *hit = ith->get(); // ith is an iterator on edm::Ref to rechit
+      if(! hit->isValid()){
+       	nhit++;
+	continue;
+      }
+      uint32_t tmpdetid = hit->geographicalId().rawId();
+      tpresc_->GetEntryWithIndex(tmpdetid);
+
+
+      //-------------
+      //decide whether to take this hit or not
+      bool takeit=false;  
+      int subdetId=hit->geographicalId().subdetId();   
+
+
+      //check first if the cluster is also in the overlap asso map
+      bool isOverlapHit=false;
+      //  bool first=true;
+      //ugly...
+      const SiStripRecHit2D* striphit = dynamic_cast<const SiStripRecHit2D*>(hit);
+      const SiPixelRecHit*   pixelhit= dynamic_cast<const SiPixelRecHit*>(hit);
+      AlignmentClusterFlag tmpflag(hit->geographicalId());
+      if(subdetId>2){// SST case
+	if(striphit!=0){
+	  SiStripRecHit2D::ClusterRef stripclust(striphit->cluster());
+	  tmpflag=InValMap[stripclust];
+	  tmpflag.SetDetId(hit->geographicalId());
+	  if(tmpflag.isOverlap())isOverlapHit=true;
+	  // cout<<"~*~*~* Prescale for module "<<tmpflag.detId().rawId()<<"("<<InValMap[stripclust].detId().rawId() <<") is "<<preschit_<<flush;
+	  //if(tmpflag.isOverlap())cout<<" (it is Overlap)"<<flush;
+	  //	  else cout<<endl;
+
+	}//end if striphit!=0
+      }//end if is a strip hit
+      else{
+	if(pixelhit!=0){
+	  //npxlhits++;
+	  SiPixelClusterRefNew pixclust(pixelhit->cluster());
+	  tmpflag=InValMap[pixclust];
+	  tmpflag.SetDetId(hit->geographicalId());
+	  if(tmpflag.isOverlap())isOverlapHit=true;
+	}
+      }//end else is a pixel hit
+      //      tmpflag.SetDetId(hit->geographicalId());
+
+      if( isOverlapHit ){
+	//cout<<"  DetId="<<tmpdetid<<" is Overlap! "<<flush;
+	takeit=(float(myrand->Rndm())<=prescoverlap_);
+      }
+      if( !takeit ){
+	float rr=float(myrand->Rndm());
+	takeit=(rr<=preschit_);
+      }
+      if(takeit){//HIT TAKEN !
+	//cout<<"  DetId="<<tmpdetid<<" taken!"<<flush;
+	tmpflag.SetTakenFlag();
+
+	if(subdetId>2){
+	  SiStripRecHit2D::ClusterRef stripclust(striphit->cluster());
+	  InValMap[stripclust]=tmpflag;//.SetTakenFlag();
+
+	}
+	else{
+	  SiPixelClusterRefNew pixclust(pixelhit->cluster());
+	  InValMap[pixclust]=tmpflag;//.SetTakenFlag();
+	}
+
+	if(!firstTakenHit){
+	  firstTakenHit=true;
+	  //std::cout<<"Index of the track iterator is "<< ittrk-Tracks->begin() <<endl;
+
+	}
+	ntakenhits++;
+      }//end if take this hit
+      //cout<<endl;
+
+        nhit++;
+      //cout<<endl;
+    }//end loop on RecHits
+    trackflags[ittrk-Tracks->begin()]=ntakenhits;
+    // cout<<"Entrioes in debug histo: "<<hrr->GetEntries()<<endl;
+  }//end loop on tracks
+
+
+
+  // totnhitspxl_+=ntakenhits;
+  //cout<<"AlignmentPrescaler::produce says that in this event "<<ntakenhits<<" pixel clusters were taken (out of "<<npxlhits<<" total pixel hits."<<endl;
+
+
+
+  //save the asso map, tracks...
+  // prepare output 
+  std::auto_ptr<AliClusterValueMap> OutVM( new AliClusterValueMap);
+  *OutVM=InValMap;
+
+  iEvent.put(OutVM);
+
+
+  std::auto_ptr<AliTrackTakenClusterValueMap> trkVM( new AliTrackTakenClusterValueMap);
+  AliTrackTakenClusterValueMap::Filler trkmapfiller(*trkVM);
+  trkmapfiller.insert(Tracks,trackflags.begin(),trackflags.end() );
+  trkmapfiller.fill();
+  iEvent.put(trkVM);
+
+
+}//end produce
+
+
+int AlignmentPrescaler::layerFromId (const DetId& id) const
+{
+ if ( uint32_t(id.subdetId())==PixelSubdetector::PixelBarrel ) {
+    PXBDetId tobId(id);
+    return tobId.layer();
+  }
+  else if ( uint32_t(id.subdetId())==PixelSubdetector::PixelEndcap ) {
+    PXFDetId tobId(id);
+    return tobId.disk() + (3*(tobId.side()-1));
+  }
+  else if ( id.subdetId()==StripSubdetector::TIB ) {
+    TIBDetId tibId(id);
+    return tibId.layer();
+  }
+  else if ( id.subdetId()==StripSubdetector::TOB ) {
+    TOBDetId tobId(id);
+    return tobId.layer();
+  }
+  else if ( id.subdetId()==StripSubdetector::TEC ) {
+    TECDetId tobId(id);
+    return tobId.wheel() + (9*(tobId.side()-1));
+  }
+  else if ( id.subdetId()==StripSubdetector::TID ) {
+    TIDDetId tobId(id);
+    return tobId.wheel() + (3*(tobId.side()-1));
+  }
+  return -1;
+
+}//end layerfromId
+
+// ========= MODULE DEF ==============
+#include "FWCore/PluginManager/interface/ModuleDef.h"
+#include "FWCore/Framework/interface/MakerMacros.h"
+DEFINE_FWK_MODULE(AlignmentPrescaler);
diff --git a/trunk/Alignment/TrackerAlignment/plugins/AlignmentPrescaler.h b/trunk/Alignment/TrackerAlignment/plugins/AlignmentPrescaler.h
@@ -0,0 +1,80 @@
+#ifndef TrackerAlignment_AlignmentPrescaler_H
+#define TrackerAlignment_AlignmentPrescaler_H
+
+#include <Riostream.h>
+#include <string>
+#include "TFile.h"
+#include "TTree.h"
+#include "TRandom3.h"
+#include "TH1F.h"
+
+
+#include "FWCore/Framework/interface/EDProducer.h"
+#include "FWCore/Framework/interface/EventPrincipal.h" 
+#include "FWCore/Framework/interface/Event.h"
+#include "FWCore/Framework/interface/EventSetup.h"
+#include "FWCore/Framework/interface/ESHandle.h"
+#include "FWCore/MessageLogger/interface/MessageLogger.h"
+#include "FWCore/ParameterSet/interface/ParameterSet.h"
+#include "FWCore/ParameterSet/interface/InputTag.h"
+
+#include "DataFormats/Common/interface/View.h"
+#include "DataFormats/DetId/interface/DetId.h"
+#include "DataFormats/SiPixelDetId/interface/PXBDetId.h"
+#include "DataFormats/SiPixelDetId/interface/PXFDetId.h"
+#include "DataFormats/SiStripDetId/interface/TIBDetId.h"
+#include "DataFormats/SiStripDetId/interface/TIDDetId.h"
+#include "DataFormats/SiStripDetId/interface/TOBDetId.h"
+#include "DataFormats/SiStripDetId/interface/TECDetId.h"
+#include "Geometry/TrackerGeometryBuilder/interface/TrackerGeometry.h"
+#include "Alignment/TrackerAlignment/interface/AlignableTracker.h"
+#include "Alignment/CommonAlignment/interface/Alignable.h"
+#include "Alignment/CommonAlignment/interface/Utilities.h"
+#include "Utilities/General/interface/ClassName.h"
+
+#include "DataFormats/TrackReco/interface/Track.h"
+#include "AnalysisDataFormats/TrackInfo/interface/TrackInfo.h"
+
+#include "DataFormats/TrackerRecHit2D/interface/SiStripRecHit2D.h"
+#include "DataFormats/TrackerRecHit2D/interface/SiPixelRecHit.h"
+#include "RecoTracker/TransientTrackingRecHit/interface/TSiStripRecHit2DLocalPos.h"
+#include "RecoTracker/TransientTrackingRecHit/interface/TSiPixelRecHit.h"
+#include "DataFormats/Alignment/interface/AlignmentClusterFlag.h"
+#include "DataFormats/Alignment/interface/AliClusterValueMap.h"
+
+
+class AlignmentPrescaler : public edm::EDProducer{
+
+ public:
+  AlignmentPrescaler(const edm::ParameterSet &iConfig);
+  ~AlignmentPrescaler();
+  void beginJob( const edm::EventSetup & );
+  void endJob();
+  virtual void produce(edm::Event &iEvent, const edm::EventSetup &iSetup) ;
+
+ private:
+  edm::InputTag src_;//tracks in input
+  edm::InputTag AM_;//Hit-quality association map
+
+  std::string prescfilename_;//name of the file containing the TTree with the prescaling factors
+  std::string presctreename_;//name of the  TTree with the prescaling factors
+
+  TFile *fpresc_;
+  TTree *tpresc_;
+  TRandom3 *myrand;
+  /*
+  //temp, just for debug
+  TFile *fh;
+  TH1F *hrr;
+  TH1F *hnhitssubdet;
+  TH1I *hnhitsTIBL3;
+  */
+
+
+  int layerFromId (const DetId& id) const;
+
+  unsigned int detid_;
+  float preschit_, prescoverlap_;
+  int totnhitspxl_;
+};
+#endif
diff --git a/trunk/Alignment/TrackerAlignment/plugins/BuildFile b/trunk/Alignment/TrackerAlignment/plugins/BuildFile
@@ -5,4 +5,30 @@
 <use name=Geometry/TrackingGeometryAligner>
 <use name=CondCore/DBOutputService>
 
-<library name=AlignmentTrackerAlignmentPlugin file="*.cc"><flags EDM_PLUGIN=1></library>
+<library name=AlignmentTrackerAlignmentPlugin file="*Misalign*.cc">
+<flags EDM_PLUGIN=1>
+</library>
+
+
+
+#<use name=FWCore/MessageLogger>
+#<use name=FWCore/Framework>
+#<use name=FWCore/ParameterSet>
+
+<library name=AlignmentPrescalerPlugin file=AlignmentPrescaler.cc>
+ <use name=DataFormats/TrackerRecHit2D>
+ <use name=DataFormats/Alignment>
+ <flags EDM_PLUGIN=1>
+</library>
+
+<library name="TreeMergerPlugin" file="TreeMerger.cc">
+# <use name=Alignment/TrackerAlignment>
+ <flags EDM_PLUGIN=1>
+</library>
+
+<library name=OverlapTaggerPlugin file=OverlapTagger.cc>
+ <use name=DataFormats/TrackReco>
+ <use name=DataFormats/TrackerRecHit2D>
+ <use name=DataFormats/Alignment>
+ <flags EDM_PLUGIN=1>
+</library>