SiPixelRecHitConverter.cc

Go to the documentation of this file.

// Our own stuff
#include "RecoLocalTracker/SiPixelRecHits/interface/SiPixelRecHitConverter.h"
#include "RecoLocalTracker/SiPixelRecHits/interface/PixelCPETemplateReco.h"
#include "RecoLocalTracker/SiPixelRecHits/interface/PixelCPEGeneric.h"

// Geometry
#include "Geometry/Records/interface/TrackerDigiGeometryRecord.h"
#include "Geometry/TrackerGeometryBuilder/interface/PixelGeomDetUnit.h"

// Data Formats
#include "DataFormats/DetId/interface/DetId.h"
#include "DataFormats/Common/interface/Ref.h"
#include "DataFormats/Common/interface/DetSet2RangeMap.h"

// Framework Already defined in the *.h file
//#include "DataFormats/Common/interface/Handle.h"
//#include "FWCore/Framework/interface/ESHandle.h"

// STL
#include <vector>
#include <memory>
#include <string>
#include <iostream>

// MessageLogger
#include "FWCore/MessageLogger/interface/MessageLogger.h"

#include "RecoLocalTracker/Records/interface/TkPixelCPERecord.h"

using namespace std;

namespace cms
{
  //---------------------------------------------------------------------------
  //---------------------------------------------------------------------------
  SiPixelRecHitConverter::SiPixelRecHitConverter(edm::ParameterSet const& conf) 
    : 
    conf_(conf),
    cpeName_("None"),     // bogus
    cpe_(0),              // the default, in case we fail to make one
    ready_(false),        // since we obviously aren't
    src_( conf.getParameter<edm::InputTag>( "src" ) ),
    theVerboseLevel(conf.getUntrackedParameter<int>("VerboseLevel",0))
  {
    //--- Declare to the EDM what kind of collections we will be making.
    produces<SiPixelRecHitCollection>();
   
  }
  
  // Destructor
  SiPixelRecHitConverter::~SiPixelRecHitConverter() 
  { 
  }  
  
  //---------------------------------------------------------------------------
  // Begin job: get magnetic field
  //---------------------------------------------------------------------------
  //void SiPixelRecHitConverter::beginJob() 
  void SiPixelRecHitConverter::beginJob() 
  {
  }
  
  //---------------------------------------------------------------------------
  //---------------------------------------------------------------------------
  void SiPixelRecHitConverter::produce(edm::Event& e, const edm::EventSetup& es)
  {

    // Step A.1: get input data
    edm::Handle< edmNew::DetSetVector<SiPixelCluster> > input;
    e.getByLabel( src_, input);
    
    // Step A.2: get event setup
    edm::ESHandle<TrackerGeometry> geom;
    es.get<TrackerDigiGeometryRecord>().get( geom );

        // Step B: create empty output collection
    std::auto_ptr<SiPixelRecHitCollectionNew> output(new SiPixelRecHitCollectionNew);

    // Step B*: create CPE
    edm::ESHandle<PixelClusterParameterEstimator> hCPE;
    std::string cpeName_ = conf_.getParameter<std::string>("CPE");
    es.get<TkPixelCPERecord>().get(cpeName_,hCPE);
    const PixelClusterParameterEstimator &cpe(*hCPE);
    cpe_ = &cpe;
    
    if(cpe_) ready_ = true;
    
    
    // Step C: Iterate over DetIds and invoke the strip CPE algorithm
    // on each DetUnit

    run( input, *output, geom );


    e.put(output);

  }

  //---------------------------------------------------------------------------
  //---------------------------------------------------------------------------
  void SiPixelRecHitConverter::run(edm::Handle<edmNew::DetSetVector<SiPixelCluster> >  inputhandle,
                   SiPixelRecHitCollectionNew &output,
                   edm::ESHandle<TrackerGeometry> & geom) 
  {
    if ( ! ready_ ) 
      {
    edm::LogError("SiPixelRecHitConverter") << " at least one CPE is not ready -- can't run!";
    // TO DO: throw an exception here?  The user may want to know...
    assert(0);
    return;   // clusterizer is invalid, bail out
      }
    
    int numberOfDetUnits = 0;
    int numberOfClusters = 0;
    
    const edmNew::DetSetVector<SiPixelCluster>& input = *inputhandle;
    
    edmNew::DetSetVector<SiPixelCluster>::const_iterator DSViter=input.begin();
    
    for ( ; DSViter != input.end() ; DSViter++) 
      {
    numberOfDetUnits++;
    unsigned int detid = DSViter->detId();
    DetId detIdObject( detid );  
    const GeomDetUnit * genericDet = geom->idToDetUnit( detIdObject );
    const PixelGeomDetUnit * pixDet = dynamic_cast<const PixelGeomDetUnit*>(genericDet);
    assert(pixDet); 
    SiPixelRecHitCollectionNew::FastFiller recHitsOnDetUnit(output,detid);
    
    edmNew::DetSet<SiPixelCluster>::const_iterator clustIt = DSViter->begin(), clustEnd = DSViter->end();
    
    for ( ; clustIt != clustEnd; clustIt++) 
      {
        numberOfClusters++;
        std::pair<LocalPoint, LocalError> lv = cpe_->localParameters( *clustIt, *genericDet );
        LocalPoint lp( lv.first );
        LocalError le( lv.second );
        // Create a persistent edm::Ref to the cluster
        edm::Ref< edmNew::DetSetVector<SiPixelCluster>, SiPixelCluster > cluster = edmNew::makeRefTo( inputhandle, clustIt);
        // Make a RecHit and add it to the DetSet
        // old : recHitsOnDetUnit.push_back( new SiPixelRecHit( lp, le, detIdObject, &*clustIt) );
        SiPixelRecHit hit( lp, le, detIdObject, cluster);
        // Copy the extra stuff; unfortunately, only the derivatives of PixelCPEBase
        // are capable of doing that.  So until we get rid of CPEFromDetPosition
        // we'll have to dynamic_cast :(
        // &&& This cast can be moved to the setupCPE, so that it's done once per job.
        const PixelCPEBase * cpeBase 
          = dynamic_cast< const PixelCPEBase* >( cpe_ );
        if (cpeBase) 
          {
        hit.setRawQualityWord( cpeBase->rawQualityWord() );
        // hit.setProbabilityX( cpeBase->probabilityX() );
        // hit.setProbabilityY( cpeBase->probabilityY() );
        // hit.setQBin( cpeBase->qBin() );
        // hit.setCotAlphaFromCluster( cpeBase->cotAlphaFromCluster() );
        // hit.setCotBetaFromCluster ( cpeBase->cotBetaFromCluster()  );
          }
        // 
        // Now save it =================
        recHitsOnDetUnit.push_back(hit);
        // =============================
      } //  <-- End loop on Clusters
    
    if ( recHitsOnDetUnit.size()>0 ) 
      {
        if (theVerboseLevel > 2) 
          LogDebug("SiPixelRecHitConverter") << " Found " 
                         << recHitsOnDetUnit.size() << " RecHits on " << detid; 
      }
    
      } //    <-- End loop on DetUnits
    
    if ( theVerboseLevel > 2 ) LogDebug ("SiPixelRecHitConverter") 
      << cpeName_ << " converted " << numberOfClusters 
      << " SiPixelClusters into SiPixelRecHits, in " 
      << numberOfDetUnits << " DetUnits.";  
  }
}  // end of namespace cms