/afs/cern.ch/work/a/aaltunda/public/www/CMSSW_5_3_13_patch3/src/IORawData/CaloPatterns/src/HtrXmlPattern.cc

Go to the documentation of this file.

#include "DataFormats/Common/interface/Handle.h"
#include "FWCore/Framework/interface/ESHandle.h"
#include "CalibFormats/HcalObjects/interface/HcalDbService.h"
#include "CalibFormats/HcalObjects/interface/HcalDbRecord.h"

#include "DataFormats/HcalDigi/interface/HcalDigiCollections.h"
#include "CondFormats/HcalObjects/interface/HcalElectronicsMap.h"
#include "CalibCalorimetry/HcalAlgos/interface/HcalDbHardcode.h"

#include "HtrXmlPattern.h"
#include "HtrXmlPatternTool.h"
#include "HtrXmlPatternToolParameters.h"

HtrXmlPattern::HtrXmlPattern(const edm::ParameterSet& iConfig)
{
  m_filled=0;
  m_fill_by_hand        = iConfig.getUntrackedParameter<bool>("fill_by_hand");
  m_hand_pattern_number = iConfig.getUntrackedParameter<int> ("hand_pattern_number");
  m_sets_to_show        = iConfig.getUntrackedParameter<int> ("sets_to_show");
  m_write_root_file     = iConfig.getUntrackedParameter<bool>("write_root_file");

  m_toolparameters = new HtrXmlPatternToolParameters;
  m_toolparameters->m_show_errors            = iConfig.getUntrackedParameter<bool>       ("show_errors");
  m_toolparameters->m_presamples_per_event   = iConfig.getUntrackedParameter<int>        ("presamples_per_event");
  m_toolparameters->m_samples_per_event      = iConfig.getUntrackedParameter<int>        ("samples_per_event");

  m_toolparameters->m_XML_file_mode          = iConfig.getUntrackedParameter<int>        ("XML_file_mode");
  m_toolparameters->m_file_tag               = iConfig.getUntrackedParameter<std::string>("file_tag");
  m_toolparameters->m_user_output_directory  = iConfig.getUntrackedParameter<std::string>("user_output_directory");

  std::string out_dir=m_toolparameters->m_user_output_directory;
  while (out_dir.find_last_of('/')==out_dir.length()-1) out_dir.erase(out_dir.find_last_of('/'));
  m_toolparameters->m_output_directory=out_dir+"/"+(m_toolparameters->m_file_tag)+"/";

  m_tool = new HtrXmlPatternTool(m_toolparameters);
}


HtrXmlPattern::~HtrXmlPattern()
{
  delete m_tool;
  delete m_toolparameters;
}

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

   if (m_filled) return;

   edm::ESHandle<HcalDbService> pSetup;
   iSetup.get<HcalDbRecord>().get( pSetup );
   const HcalElectronicsMap* readoutMap=pSetup->getHcalMapping();

   if (m_fill_by_hand) {
     do_hand_fill(readoutMap);
     m_filled=1;
     return;
   }
  
   std::vector<edm::Handle<HBHEDigiCollection> > hbhe;
   std::vector<edm::Handle<HODigiCollection> > ho;
   std::vector<edm::Handle<HFDigiCollection> > hf;
   std::vector<edm::Handle<ZDCDigiCollection> > zdc;
   std::vector<edm::Handle<HcalCalibDigiCollection> > hc;
   std::vector<edm::Handle<HcalTrigPrimDigiCollection> > htp;
   std::vector<edm::Handle<HcalHistogramDigiCollection> > hh;  

   iEvent.getManyByType(hbhe);
   if (hbhe.empty()) {
     cout << "No HB/HE Digis." << endl;
   } else {
     std::vector<edm::Handle<HBHEDigiCollection> >::iterator i;
     for (i=hbhe.begin(); i!=hbhe.end(); i++) {
       const HBHEDigiCollection& c=*(*i);
   
       int count=0;
       for (HBHEDigiCollection::const_iterator j=c.begin(); j!=c.end(); j++) {
         const HcalElectronicsId HEID = readoutMap->lookup(j->id());
         m_tool->Fill(HEID,j);

         if ( count++<m_sets_to_show || m_sets_to_show<0 ) {
           cout << *j << std::endl;
           cout << HEID << endl;
           cout << "count: " << count << endl;
         }
       }
       if (m_sets_to_show!=0) cout << "HB/HE count: " << count << endl;
     }
  }
  
   iEvent.getManyByType(hf);
   if (hf.empty()) {
     cout << "No HF Digis." << endl;
  } else {
     std::vector<edm::Handle<HFDigiCollection> >::iterator i;
     for (i=hf.begin(); i!=hf.end(); i++) {
       const HFDigiCollection& c=*(*i);
   
       int count=0;
       for (HFDigiCollection::const_iterator j=c.begin(); j!=c.end(); j++) {
         const HcalElectronicsId HEID = readoutMap->lookup(j->id());
         m_tool->Fill(HEID,j);
         
         if ( count++<m_sets_to_show || m_sets_to_show<0 ) {
           cout << *j << std::endl;
           cout << HEID << endl;
           cout << "count: " << count << endl;
         }
       }
       if (m_sets_to_show!=0) cout << "HF    count: " << count << endl;
     }
  }
   
   iEvent.getManyByType(ho);
   if (ho.empty()) {
     cout << "No HO Digis." << endl;
  } else {
     std::vector<edm::Handle<HODigiCollection> >::iterator i;
     for (i=ho.begin(); i!=ho.end(); i++) {
       const HODigiCollection& c=*(*i);
       
       int count=0;
       for (HODigiCollection::const_iterator j=c.begin(); j!=c.end(); j++) {
         const HcalElectronicsId HEID = readoutMap->lookup(j->id());
         m_tool->Fill(HEID,j);
         
         if ( count++<m_sets_to_show || m_sets_to_show<0 ) {
           cout << *j << std::endl;
           cout << HEID << endl;
           cout << "count: " << count << endl;
         }
       }
       if (m_sets_to_show!=0) cout << "HO    count: " << count << endl;
     }
  }

   cout << endl;    

}


void HtrXmlPattern::do_hand_fill(const HcalElectronicsMap *emap)
{
  HtrXmlPatternSet *hxps=m_tool->GetPatternSet();

  for (int iCrate=0;iCrate<ChannelPattern::NUM_CRATES;iCrate++) {
    CrateData *cd=hxps->getCrate(iCrate);
    if (!cd) continue;
    for (int iSlot=0;iSlot<ChannelPattern::NUM_SLOTS;iSlot++) {
      for (int iTb=0;iTb<2;iTb++) {
        HalfHtrData* hhd=cd->getHalfHtrData(iSlot,iTb);
        if (!hhd) continue;
        for (int iChannel=1;iChannel<25;iChannel++) {
          ChannelPattern *cp=hhd->getPattern(iChannel);
          if (!cp) continue;
          cp->Fill_by_hand(emap,m_hand_pattern_number);
        }
      }      
    }
  }

}

// ------------ method called once each job just after ending the event loop  ------------
void 
HtrXmlPattern::endJob() {
  bool modeg0=m_toolparameters->m_XML_file_mode>0;
  if (modeg0 || m_write_root_file) m_tool->prepareDirs();
  if (modeg0)                      m_tool->writeXML();
  if (m_write_root_file)           m_tool->createHists();
}