PFHBHERecHitCreatorMaxSample.h

Go to the documentation of this file.

#ifndef RecoParticleFlow_PFClusterProducer_PFHBHeRecHitCreatorMaxSample_h
#define RecoParticleFlow_PFClusterProducer_PFHBHeRecHitCreatorMaxSample_h

#include "RecoParticleFlow/PFClusterProducer/interface/PFRecHitCreatorBase.h"

#include "DataFormats/HcalRecHit/interface/HORecHit.h"
#include "DataFormats/HcalRecHit/interface/HFRecHit.h"
#include "DataFormats/HcalRecHit/interface/HBHERecHit.h"
#include "DataFormats/HcalRecHit/interface/HcalRecHitCollections.h"
#include "DataFormats/HcalDetId/interface/HcalSubdetector.h"

#include "Geometry/CaloGeometry/interface/CaloSubdetectorGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloGeometry.h"
#include "Geometry/CaloGeometry/interface/CaloCellGeometry.h"
#include "Geometry/Records/interface/CaloGeometryRecord.h"
#include "Geometry/CaloTopology/interface/HcalTopology.h"

#include "CalibFormats/HcalObjects/interface/HcalDbRecord.h"
#include "CalibFormats/HcalObjects/interface/HcalCoderDb.h"
#include "CalibFormats/HcalObjects/interface/HcalCalibrations.h"
#include "CondFormats/HcalObjects/interface/HcalQIEShape.h"

#include "DataFormats/HcalDigi/interface/HBHEDataFrame.h"


#include "CalibFormats/HcalObjects/interface/HcalDbService.h"


#include "RecoCaloTools/Navigation/interface/CaloNavigator.h"


class PFHBHERecHitCreatorMaxSample :  public  PFRecHitCreatorBase {

 public:  
  PFHBHERecHitCreatorMaxSample(const edm::ParameterSet& iConfig,edm::ConsumesCollector& iC):
    PFRecHitCreatorBase(iConfig,iC)
    {
      recHitToken_ = iC.consumes<edm::SortedCollection<HBHERecHit> >(iConfig.getParameter<edm::InputTag>("src"));
    }


  ~PFHBHERecHitCreatorMaxSample() {
  }


    void importRecHits(std::auto_ptr<reco::PFRecHitCollection>&out,std::auto_ptr<reco::PFRecHitCollection>& cleaned ,const edm::Event& iEvent,const edm::EventSetup& iSetup) {

      beginEvent(iEvent,iSetup);

      edm::Handle<edm::SortedCollection<HBHERecHit> > recHitHandle;

      edm::ESHandle<CaloGeometry> geoHandle;
      iSetup.get<CaloGeometryRecord>().get(geoHandle);


      edm::ESHandle<HcalDbService> conditions;
      iSetup.get<HcalDbRecord > ().get(conditions);
  
      // get the ecal geometry
      const CaloSubdetectorGeometry *hcalBarrelGeo = 
    geoHandle->getSubdetectorGeometry(DetId::Hcal, HcalBarrel);
      const CaloSubdetectorGeometry *hcalEndcapGeo = 
    geoHandle->getSubdetectorGeometry(DetId::Hcal, HcalEndcap);

      iEvent.getByToken(recHitToken_,recHitHandle);
      for( const auto& erh : *recHitHandle ) {      
    const HcalDetId& detid = (HcalDetId)erh.detid();
    HcalSubdetector esd=(HcalSubdetector)detid.subdetId();
    
    
    //CUSTOM ENERGY AND TIME RECO

    CaloSamples tool;
    const HcalCalibrations& calibrations = conditions->getHcalCalibrations(detid);
    const HcalQIECoder* channelCoder = conditions->getHcalCoder(detid);
    const HcalQIEShape* shape = conditions->getHcalShape(channelCoder);
    HcalCoderDb coder(*channelCoder, *shape);
    int auxwd1 = erh.auxHBHE();  // TS = 0,1,2,3 info
    int auxwd2 = erh.aux();      // TS = 4,5,6,7 info   

    int adc[8];
    int capid[8];

    adc[0] = (auxwd1)       & 0x7F;
    adc[1] = (auxwd1 >> 7)  & 0x7F;
    adc[2] = (auxwd1 >> 14) & 0x7F;
    adc[3] = (auxwd1 >> 21) & 0x7F;
    
    capid[0] = (auxwd1 >> 28) & 0x3;  // rotating through 4 values: 0,1,2,3
    capid[1] = (capid[0] + 1 <= 3) ? capid[0] + 1 : 0;
    capid[2] = (capid[1] + 1 <= 3) ? capid[1] + 1 : 0;
    capid[3] = (capid[2] + 1 <= 3) ? capid[2] + 1 : 0;

    adc[4] = (auxwd2)       & 0x7F;
    adc[5] = (auxwd2 >> 7)  & 0x7F;
    adc[6] = (auxwd2 >> 14) & 0x7F;
    adc[7] = (auxwd2 >> 21) & 0x7F;

    capid[4] = (auxwd2 >> 28) & 0x3;
    capid[5] = (capid[4] + 1 <= 3) ? capid[4] + 1 : 0;
    capid[6] = (capid[5] + 1 <= 3) ? capid[5] + 1 : 0;
    capid[7] = (capid[6] + 1 <= 3) ? capid[6] + 1 : 0; 

    // Pack the above into HBHEDataFrame for subsequent linearization to fC     
    HBHEDataFrame digi(detid);
    digi.setSize(10);
    digi.setPresamples(4); 
    for (int ii = 0; ii < 8; ii++) {
      HcalQIESample s (adc[ii], capid[ii], 0, 0);
      digi.setSample(ii,s);
    } 
    coder.adc2fC(digi, tool); 
    HcalGenericDetId hcalGenDetId(erh.id());
    std::array<double,8> samples;


    //store the samples over thresholds
    for (int ii = 0; ii < 8; ii++) {
      double sampleE = calibrations.respcorrgain(capid[ii]) *
        (tool[ii] - calibrations.pedestal(capid[ii]));
      if (sampleE>sampleCut_)
        samples[ii] = sampleE;
      else
        samples[ii] = 0.0;

    }

    
    //run the algorithm
    double energy=0.0;
    double energy2=0.0;
    double time=0.0;
    double s2=0.0;
    std::vector<double> hitEnergies;      
    std::vector<double> hitTimes;     

    for (int ii = 0; ii < 8; ii++) {
      energy=energy+samples[ii];
      s2=samples[ii]*samples[ii];
      time = time+(-100. + ii*25.0)*s2;
      energy2=energy2+s2;

      if (ii>0 && ii<7) {
        if (samples[ii]<=samples[ii-1] && samples[ii]<samples[ii+1] && energy>0) {
          hitEnergies.push_back(energy);
          hitTimes.push_back(time/energy2);
          energy=0.0;
          energy2=0.0;
          time=0.0;
        }
          
      }
      else if (ii==7 && energy>0) {
          hitEnergies.push_back(energy);
          hitTimes.push_back(time/energy2);
          energy=0.0;
          energy2=0.0;
          time=0.0;
      }

    }       

    if (hitEnergies.size()==0)
      continue;

    int depth = detid.depth();
    math::XYZVector position;
    math::XYZVector axis;
    
    const CaloCellGeometry *thisCell=0;
    PFLayer::Layer layer = PFLayer::HCAL_BARREL1;
    switch(esd) {
    case HcalBarrel:
      thisCell =hcalBarrelGeo->getGeometry(detid); 
      layer =PFLayer::HCAL_BARREL1;
      break;

    case HcalEndcap:
      thisCell =hcalEndcapGeo->getGeometry(detid); 
      layer =PFLayer::HCAL_ENDCAP;
      break;
    default:
      break;
    }
  
    // find rechit geometry
    if(!thisCell) {
      edm::LogError("PFHBHERecHitCreatorMaxSample")
        <<"warning detid "<<detid.rawId()
        <<" not found in geometry"<<std::endl;
      continue;
    }


    auto const point = thisCell->getPosition();
    position.SetCoordinates ( point.x(),
                  point.y(),
                  point.z() );


    reco::PFRecHit rh( detid.rawId(),layer,
               energy, 
               position.x(), position.y(), position.z(), 
               0,0,0);

    rh.setDepth(depth);


    const CaloCellGeometry::CornersVec& corners = thisCell->getCorners();
    assert( corners.size() == 8 );

    rh.setNECorner( corners[0].x(), corners[0].y(),  corners[0].z());
    rh.setSECorner( corners[1].x(), corners[1].y(),  corners[1].z());
    rh.setSWCorner( corners[2].x(), corners[2].y(),  corners[2].z());
    rh.setNWCorner( corners[3].x(), corners[3].y(),  corners[3].z());
    
    //  for (unsigned int i=0;i<hitEnergies.size();++i)
    //    printf(" %f / %f ,",hitEnergies[i],hitTimes[i]);
    
    //now find the best hit 
    auto best_hit = std::min_element(hitTimes.begin(),hitTimes.end(),
                      [](const double& a, 
                     const double& b){
                       return fabs(a) < fabs(b);
                      });


    rh.setTime(*best_hit);
    rh.setEnergy(hitEnergies[std::distance(hitTimes.begin(),best_hit)]);
    //  printf("Best = %f %f\n",rh.energy(),rh.time());

    //Apply Q tests
    bool rcleaned = false;
    bool keep=true;

    for( const auto& qtest : qualityTests_ ) {
      if (!qtest->test(rh,erh,rcleaned)) {
        keep = false;
      }
    }
      
    if(keep) {
      out->push_back(rh);
    }
    else if (rcleaned) 
      cleaned->push_back(rh);
      }
    }


 protected:
    edm::EDGetTokenT<edm::SortedCollection<HBHERecHit> > recHitToken_;
    const double sampleCut_ = 0.6; // minimalistic threshold just to reduce the iterations 

};





#endif