CMSSW_4_4_3_patch1/src/RecoLocalTracker/SiPixelRecHits/plugins/SiPixelRecHitConverter.cc

Go to the documentation of this file.

// Our own stuff
#include "SiPixelRecHitConverter.h"
#include "PixelCPETemplateReco.h"
#include "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