EcalPulseShapeGrapher.cc

Go to the documentation of this file.

// -*- C++ -*-
//
// Package:    EcalPulseShapeGrapher
// Class:      EcalPulseShapeGrapher
// 
//
// Original Author:  Seth Cooper
//         Created:  Tue Feb  5 11:35:45 CST 2008
//
//

#include "CaloOnlineTools/EcalTools/plugins/EcalPulseShapeGrapher.h"

//
// constants, enums and typedefs
//

//
// static data member definitions
//

//
// constructors and destructor
//
EcalPulseShapeGrapher::EcalPulseShapeGrapher(const edm::ParameterSet& iConfig) :
EBUncalibratedRecHitCollection_ (iConfig.getParameter<edm::InputTag>("EBUncalibratedRecHitCollection")),
EBDigis_ (iConfig.getParameter<edm::InputTag>("EBDigiCollection")),
EEUncalibratedRecHitCollection_ (iConfig.getParameter<edm::InputTag>("EEUncalibratedRecHitCollection")),
EEDigis_ (iConfig.getParameter<edm::InputTag>("EEDigiCollection")),
ampCut_ (iConfig.getUntrackedParameter<int>("AmplitudeCutADC", 13)),
rootFilename_ (iConfig.getUntrackedParameter<std::string>("rootFilename","pulseShapeGrapher"))
{
   //now do what ever initialization is needed

   std::vector<int> listDefaults;
   listDefaults.push_back(-1);
   listChannels_ = iConfig.getUntrackedParameter<std::vector<int> >("listChannels", listDefaults);

   for(std::vector<int>::const_iterator itr = listChannels_.begin(); itr != listChannels_.end(); ++itr)
   {
     std::string title = "Amplitude of cry "+intToString(*itr);
     std::string name = "ampOfCry"+intToString(*itr);
     ampHistMap_[*itr] = new TH1F(name.c_str(),title.c_str(),100,0,100);
     ampHistMap_[*itr]->GetXaxis()->SetTitle("ADC");

     title = "Amplitude (over 13 ADC) of cry "+intToString(*itr);
     name = "cutAmpOfCry"+intToString(*itr);
     cutAmpHistMap_[*itr] = new TH1F(name.c_str(),title.c_str(),100,0,100);
     cutAmpHistMap_[*itr]->GetXaxis()->SetTitle("ADC");

     title = "Pulse shape of cry "+intToString(*itr);
     name = "PulseShapeCry"+intToString(*itr);
     pulseShapeHistMap_[*itr] = new TH2F(name.c_str(),title.c_str(),10,0,10,220,-20,2);
     pulseShapeHistMap_[*itr]->GetXaxis()->SetTitle("sample");

     title = "Raw Pulse shape of cry "+intToString(*itr);
     name = "RawPulseShapeCry"+intToString(*itr);
     rawPulseShapeHistMap_[*itr] = new TH2F(name.c_str(),title.c_str(),10,0,10,500,0,500);
     rawPulseShapeHistMap_[*itr]->GetXaxis()->SetTitle("sample");
     
     title = "Amplitude of first sample, cry "+intToString(*itr);
     name = "AmpOfFirstSampleCry"+intToString(*itr);
     firstSampleHistMap_[*itr] = new TH1F(name.c_str(),title.c_str(),300,100,400);
     firstSampleHistMap_[*itr]->GetXaxis()->SetTitle("ADC");
   }
   
   fedMap_ = new EcalFedMap();
   
   for (int i=0; i<10; i++)
     abscissa[i] = i;
}


EcalPulseShapeGrapher::~EcalPulseShapeGrapher()
{
 
   // do anything here that needs to be done at desctruction time
   // (e.g. close files, deallocate resources etc.)

}


//
// member functions
//

// ------------ method called to for each event  ------------
void
EcalPulseShapeGrapher::analyze(const edm::Event& iEvent, const edm::EventSetup& iSetup)
{
   using namespace edm;
   using namespace std;

   int numHitsWithActivity = 0;
   
   //int eventNum = iEvent.id().event();
   //vector<EcalUncalibratedRecHit> sampleHitsPastCut;
   
   Handle<EcalUncalibratedRecHitCollection> EBHits;
   iEvent.getByLabel(EBUncalibratedRecHitCollection_, EBHits);
   Handle<EcalUncalibratedRecHitCollection> EEHits;
   iEvent.getByLabel(EEUncalibratedRecHitCollection_, EEHits);
   //cout << "event: " << eventNum << " sample hits collection size: " << sampleHits->size() << endl;
   //Handle<EcalUncalibratedRecHitCollection> fittedHits;
   //iEvent.getByLabel(FittedUncalibratedRecHitCollection_, fittedHits);
   Handle<EBDigiCollection>  EBdigis;
   iEvent.getByLabel(EBDigis_, EBdigis);
   Handle<EEDigiCollection>  EEdigis;
   iEvent.getByLabel(EEDigis_, EEdigis);
   //cout << "event: " << eventNum << " digi collection size: " << digis->size() << endl;

   auto_ptr<EcalElectronicsMapping> ecalElectronicsMap(new EcalElectronicsMapping);

   // Loop over the hits
   for(EcalUncalibratedRecHitCollection::const_iterator hitItr = EBHits->begin(); hitItr != EBHits->end(); ++hitItr)
   {
     EcalUncalibratedRecHit hit = (*hitItr);
     float amplitude = hit.amplitude();
     EBDetId hitDetId = hit.id();

     // Get the Fedid
     EcalElectronicsId elecId = ecalElectronicsMap->getElectronicsId(hitDetId);
     int FEDid = 600+elecId.dccId();
     string SMname = fedMap_->getSliceFromFed(FEDid);
     
     vector<int>::const_iterator itr = listChannels_.begin();
     while(itr != listChannels_.end() && (*itr) != hitDetId.hashedIndex())
     {
       itr++;
     }
     if(itr==listChannels_.end())
       continue;

     ampHistMap_[hitDetId.hashedIndex()]->Fill(amplitude);
     //cout << "Cry hash:" << hitDetId.hashedIndex() << " amplitude: " << amplitude << endl;
     if(amplitude < ampCut_)
       continue;

     cutAmpHistMap_[hitDetId.hashedIndex()]->Fill(amplitude);
     numHitsWithActivity++;
     EBDigiCollection::const_iterator digiItr= EBdigis->begin();
     while(digiItr != EBdigis->end() && digiItr->id() != hitItr->id())
     {
       digiItr++;
     }
     if(digiItr == EBdigis->end())
       continue;

     double sampleADC[10];
     EBDataFrame df(*digiItr);
     double pedestal = 200;

     if(df.sample(0).gainId()!=1 || df.sample(1).gainId()!=1) continue; //goes to the next digi
     else {
       sampleADC[0] = df.sample(0).adc();
       sampleADC[1] = df.sample(1).adc();
       pedestal = (double)(sampleADC[0]+sampleADC[1])/(double)2;
     } 

     for (int i=0; (unsigned int)i< digiItr->size(); ++i ) {
       EBDataFrame df(*digiItr);
       double gain = 12.;
       if(df.sample(i).gainId()==1)
         gain = 1.;
       else if(df.sample(i).gainId()==2)
         gain = 2.;
       sampleADC[i] = pedestal+(df.sample(i).adc()-pedestal)*gain;
     }

     //cout << "1) maxsample amp:" << maxSampleAmp << " maxSampleIndex:" << maxSampleIndex << endl;
     for(int i=0; i<10;++i)
     {
       //cout << "Filling hist for:" << hitDetId.hashedIndex() << " with sample:" << i 
         //<< " amp:" <<(float)(df.sample(i).adc()-baseline)/(maxSampleAmp-baseline) << endl;
       //cout << "ADC of sample:" << df.sample(i).adc() << " baseline:" << baseline << " maxSampleAmp:" 
         //<< maxSampleAmp << endl << endl;
       pulseShapeHistMap_[hitDetId.hashedIndex()]->Fill(i, (float)(sampleADC[i]-pedestal)/amplitude);
       rawPulseShapeHistMap_[hitDetId.hashedIndex()]->Fill(i, (float)(sampleADC[i]));
     }
     firstSampleHistMap_[hitDetId.hashedIndex()]->Fill(sampleADC[0]);
   }

   // Now do the same for the EE hits
   for(EcalUncalibratedRecHitCollection::const_iterator hitItr = EEHits->begin(); hitItr != EEHits->end(); ++hitItr)
   {
     EcalUncalibratedRecHit hit = (*hitItr);
     float amplitude = hit.amplitude();
     EEDetId hitDetId = hit.id();

     // Get the Fedid
     EcalElectronicsId elecId = ecalElectronicsMap->getElectronicsId(hitDetId);
     int FEDid = 600+elecId.dccId();
     string SMname = fedMap_->getSliceFromFed(FEDid);
     
     vector<int>::const_iterator itr = listChannels_.begin();
     while(itr != listChannels_.end() && (*itr) != hitDetId.hashedIndex())
     {
       itr++;
     }
     if(itr==listChannels_.end())
       continue;

     ampHistMap_[hitDetId.hashedIndex()]->Fill(amplitude);
     //cout << "Cry hash:" << hitDetId.hashedIndex() << " amplitude: " << amplitude << endl;
     if(amplitude < ampCut_)
       continue;

     cutAmpHistMap_[hitDetId.hashedIndex()]->Fill(amplitude);
     numHitsWithActivity++;
     EEDigiCollection::const_iterator digiItr= EEdigis->begin();
     while(digiItr != EEdigis->end() && digiItr->id() != hitItr->id())
     {
       digiItr++;
     }
     if(digiItr == EEdigis->end())
       continue;

     double sampleADC[10];
     EEDataFrame df(*digiItr);
     double pedestal = 200;

     if(df.sample(0).gainId()!=1 || df.sample(1).gainId()!=1) continue; //goes to the next digi
     else {
       sampleADC[0] = df.sample(0).adc();
       sampleADC[1] = df.sample(1).adc();
       pedestal = (double)(sampleADC[0]+sampleADC[1])/(double)2;
     } 

     for (int i=0; (unsigned int)i< digiItr->size(); ++i ) {
       EEDataFrame df(*digiItr);
       double gain = 12.;
       if(df.sample(i).gainId()==1)
         gain = 1.;
       else if(df.sample(i).gainId()==2)
         gain = 2.;
       sampleADC[i] = pedestal+(df.sample(i).adc()-pedestal)*gain;
     }

     //cout << "1) maxsample amp:" << maxSampleAmp << " maxSampleIndex:" << maxSampleIndex << endl;
     for(int i=0; i<10;++i)
     {
       //cout << "Filling hist for:" << hitDetId.hashedIndex() << " with sample:" << i 
         //<< " amp:" <<(float)(df.sample(i).adc()-baseline)/(maxSampleAmp-baseline) << endl;
       //cout << "ADC of sample:" << df.sample(i).adc() << " baseline:" << baseline << " maxSampleAmp:" 
         //<< maxSampleAmp << endl << endl;
       pulseShapeHistMap_[hitDetId.hashedIndex()]->Fill(i, (float)(sampleADC[i]-pedestal)/amplitude);
       rawPulseShapeHistMap_[hitDetId.hashedIndex()]->Fill(i, (float)(sampleADC[i]));
     }
     firstSampleHistMap_[hitDetId.hashedIndex()]->Fill(sampleADC[0]);
   }
}


// ------------ method called once each job just after ending the event loop  ------------
void 
EcalPulseShapeGrapher::endJob() {
  
   rootFilename_+=".root";
   file_ = new TFile(rootFilename_.c_str(),"RECREATE");
   TH1::AddDirectory(false);
  
   for(std::vector<int>::const_iterator itr = listChannels_.begin(); itr != listChannels_.end(); ++itr)
   {
     ampHistMap_[*itr]->Write();
     cutAmpHistMap_[*itr]->Write();
     firstSampleHistMap_[*itr]->Write();

     rawPulseShapeHistMap_[*itr]->Write();
     TProfile* t2 = (TProfile*) (rawPulseShapeHistMap_[*itr]->ProfileX());
     t2->Write();
     //TODO: fix the normalization so these are correct
     //pulseShapeHistMap_[*itr]->Write();
     //TProfile* t1 = (TProfile*) (pulseShapeHistMap_[*itr]->ProfileX());
     //t1->Write();
   }

   
  file_->Write();
  file_->Close();
}

std::string EcalPulseShapeGrapher::intToString(int num)
{
  using namespace std;
  ostringstream myStream;
  myStream << num << flush;
  return(myStream.str()); //returns the string form of the stringstream object
}