#include <RecoTracker/DeDxDiscriminatorLearner/src/DeDxDiscriminatorLearner.cc>

Inheritance diagram for DeDxDiscriminatorLearner:

Classes
class	isEqual

struct	stModInfo

Public Member Functions
	DeDxDiscriminatorLearner (const edm::ParameterSet &)

	~DeDxDiscriminatorLearner ()

Public Member Functions inherited from ConditionDBWriter< PhysicsTools::Calibration::HistogramD3D >
	ConditionDBWriter (const edm::ParameterSet &iConfig)

virtual	~ConditionDBWriter ()

Public Member Functions inherited from edm::EDAnalyzer
void	callWhenNewProductsRegistered (std::function< void(BranchDescription const &)> const &func)

	EDAnalyzer ()

ModuleDescription const &	moduleDescription () const

std::string	workerType () const

virtual	~EDAnalyzer ()

Public Member Functions inherited from edm::EDConsumerBase
	EDConsumerBase ()

ProductHolderIndexAndSkipBit	indexFrom (EDGetToken, BranchType, TypeID const &) const

void	itemsMayGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

void	itemsToGet (BranchType, std::vector< ProductHolderIndexAndSkipBit > &) const

std::vector < ProductHolderIndexAndSkipBit > const &	itemsToGetFromEvent () const

void	labelsForToken (EDGetToken iToken, Labels &oLabels) const

void	modulesDependentUpon (const std::string &iProcessName, std::vector< const char * > &oModuleLabels) const

bool	registeredToConsume (ProductHolderIndex, bool, BranchType) const

bool	registeredToConsumeMany (TypeID const &, BranchType) const

void	updateLookup (BranchType iBranchType, ProductHolderIndexHelper const &)

virtual	~EDConsumerBase ()

Private Member Functions
virtual void	algoAnalyze (const edm::Event &, const edm::EventSetup &)

virtual void	algoBeginJob (const edm::EventSetup &)

virtual void	algoEndJob ()

PhysicsTools::Calibration::HistogramD3D *	getNewObject ()

void	Learn (const SiStripCluster *cluster, TrajectoryStateOnSurface trajState)

Private Attributes
std::string	algoMode

double	Charge_Max

double	Charge_Min

int	Charge_NBins

TH3F *	Charge_Vs_Path

std::string	HistoFile

const TrackerGeometry *	m_tracker

edm::EDGetTokenT < reco::TrackCollection >	m_tracksTag

edm::EDGetTokenT < TrajTrackAssociationCollection >	m_trajTrackAssociationTag

unsigned int	MaxNrStrips

double	MaxTrackChiOverNdf

double	MaxTrackEta

double	MaxTrackMomentum

double	MeVperADCPixel

double	MeVperADCStrip

double	MinTrackEta

unsigned int	MinTrackHits

double	MinTrackMomentum

__gnu_cxx::hash_map< unsigned int, stModInfo *, __gnu_cxx::hash< unsigned int >, isEqual >	MODsColl

double	P_Max

double	P_Min

int	P_NBins

double	Path_Max

double	Path_Min

int	Path_NBins

bool	usePixel

bool	useStrip

Additional Inherited Members
Public Types inherited from edm::EDAnalyzer
typedef EDAnalyzer	ModuleType

Static Public Member Functions inherited from edm::EDAnalyzer
static const std::string &	baseType ()

static void	fillDescriptions (ConfigurationDescriptions &descriptions)

static void	prevalidate (ConfigurationDescriptions &)

Protected Member Functions inherited from ConditionDBWriter< PhysicsTools::Calibration::HistogramD3D >
void	setDoStore (const bool doStore)
	When set to false the payload will not be written to the db. More...

void	storeOnDbNow ()

cond::Time_t	timeOfLastIOV ()

Protected Member Functions inherited from edm::EDConsumerBase
template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	consumes (edm::InputTag const &tag)

EDGetToken	consumes (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	consumes (TypeToGet const &id, edm::InputTag const &tag)

ConsumesCollector	consumesCollector ()
	Use a ConsumesCollector to gather consumes information from helper functions. More...

template<typename ProductType , BranchType B = InEvent>
void	consumesMany ()

void	consumesMany (const TypeToGet &id)

template<BranchType B>
void	consumesMany (const TypeToGet &id)

template<typename ProductType , BranchType B = InEvent>
EDGetTokenT< ProductType >	mayConsume (edm::InputTag const &tag)

EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

template<BranchType B>
EDGetToken	mayConsume (const TypeToGet &id, edm::InputTag const &tag)

Detailed Description

Description: <one line="" class="" summary>="">

Implementation: <Notes on="" implementation>="">

Definition at line 32 of file DeDxDiscriminatorLearner.h.

Constructor & Destructor Documentation

DeDxDiscriminatorLearner::DeDxDiscriminatorLearner ( const edm::ParameterSet & iConfig )

explicit

Definition at line 36 of file DeDxDiscriminatorLearner.cc.

References algoMode, Charge_Max, Charge_Min, Charge_NBins, edm::ParameterSet::getParameter(), edm::ParameterSet::getUntrackedParameter(), HistoFile, m_tracksTag, m_trajTrackAssociationTag, MaxNrStrips, MaxTrackEta, MaxTrackMomentum, MinTrackEta, MinTrackHits, MinTrackMomentum, P_Max, P_Min, P_NBins, Path_Max, Path_Min, Path_NBins, usePixel, and useStrip.

                                                                                  : ConditionDBWriter<PhysicsTools::Calibration::HistogramD3D>(iConfig)
 {
    m_tracksTag                 = consumes<reco::TrackCollection>(iConfig.getParameter<edm::InputTag>("tracks"));
    m_trajTrackAssociationTag   = consumes<TrajTrackAssociationCollection>(iConfig.getParameter<edm::InputTag>("trajectoryTrackAssociation"));
 
    usePixel = iConfig.getParameter<bool>("UsePixel"); 
    useStrip = iConfig.getParameter<bool>("UseStrip");
    if(!usePixel && !useStrip)
    edm::LogWarning("DeDxDiscriminatorLearner") << "Pixel Hits AND Strip Hits will not be used to estimate dEdx --> BUG, Please Update the config file";
 
    P_Min           = iConfig.getParameter<double>  ("P_Min"  );
    P_Max               = iConfig.getParameter<double>  ("P_Max"  );
    P_NBins             = iConfig.getParameter<int>     ("P_NBins");
    Path_Min            = iConfig.getParameter<double>  ("Path_Min"  );
    Path_Max            = iConfig.getParameter<double>  ("Path_Max"  );
    Path_NBins          = iConfig.getParameter<int>     ("Path_NBins");
    Charge_Min          = iConfig.getParameter<double>  ("Charge_Min"  );
    Charge_Max          = iConfig.getParameter<double>  ("Charge_Max"  );
    Charge_NBins        = iConfig.getParameter<int>     ("Charge_NBins");
 
    MinTrackMomentum    = iConfig.getUntrackedParameter<double>  ("minTrackMomentum"   ,  3.0);
    MaxTrackMomentum    = iConfig.getUntrackedParameter<double>  ("maxTrackMomentum"   ,  99999.0); 
    MinTrackEta         = iConfig.getUntrackedParameter<double>  ("minTrackEta"        , -5.0);
    MaxTrackEta         = iConfig.getUntrackedParameter<double>  ("maxTrackEta"        ,  5.0);
    MaxNrStrips         = iConfig.getUntrackedParameter<unsigned>("maxNrStrips"        ,  255);
    MinTrackHits        = iConfig.getUntrackedParameter<unsigned>("MinTrackHits"       ,  4);
 
    algoMode            = iConfig.getUntrackedParameter<string>  ("AlgoMode"           ,  "SingleJob");
    HistoFile           = iConfig.getUntrackedParameter<string>  ("HistoFile"        ,  "out.root");
 }

DeDxDiscriminatorLearner::~DeDxDiscriminatorLearner ( )

Definition at line 68 of file DeDxDiscriminatorLearner.cc.

68 {}

Member Function Documentation

void DeDxDiscriminatorLearner::algoAnalyze	(	const edm::Event &	iEvent,
		const edm::EventSetup &	iSetup
	)

privatevirtual

Reimplemented from ConditionDBWriter< PhysicsTools::Calibration::HistogramD3D >.

Definition at line 131 of file DeDxDiscriminatorLearner.cc.

References edm::AssociationMap< Tag >::begin(), edm::AssociationMap< edm::OneToOne< std::vector< Trajectory >, reco::TrackCollection, unsigned short > >::const_iterator, edm::AssociationMap< Tag >::end(), reco::TrackBase::eta(), reco::Track::found(), edm::EventSetup::get(), edm::Event::getByToken(), TrajectoryStateOnSurface::isValid(), Learn(), m_tracker, m_tracksTag, m_trajTrackAssociationTag, MaxTrackEta, MaxTrackMomentum, Trajectory::measurements(), MinTrackEta, MinTrackHits, MinTrackMomentum, reco::TrackBase::p(), edm::Handle< T >::product(), and edm::ESHandle< class >::product().

 {
    edm::ESHandle<TrackerGeometry> tkGeom;
    iSetup.get<TrackerDigiGeometryRecord>().get( tkGeom );
    m_tracker = tkGeom.product();
 
 
 
   Handle<TrajTrackAssociationCollection> trajTrackAssociationHandle;
   iEvent.getByToken(m_trajTrackAssociationTag, trajTrackAssociationHandle);
   const TrajTrackAssociationCollection TrajToTrackMap = *trajTrackAssociationHandle.product();
 
   edm::Handle<reco::TrackCollection> trackCollectionHandle;
   iEvent.getByToken(m_tracksTag,trackCollectionHandle);
 
   unsigned track_index = 0;
   for(TrajTrackAssociationCollection::const_iterator it = TrajToTrackMap.begin(); it!=TrajToTrackMap.end(); ++it, track_index++) {
 
       const Track      track = *it->val;
       const Trajectory traj  = *it->key;
 
       if(track.eta()  <MinTrackEta      || track.eta()>MaxTrackEta     ){continue;}
       if(track.p()    <MinTrackMomentum || track.p()  >MaxTrackMomentum){continue;}
       if(track.found()<MinTrackHits                                    ){continue;}
 
       vector<TrajectoryMeasurement> measurements = traj.measurements();
       for(vector<TrajectoryMeasurement>::const_iterator measurement_it = measurements.begin(); measurement_it!=measurements.end(); measurement_it++)
       {
          TrajectoryStateOnSurface trajState = measurement_it->updatedState();
          if( !trajState.isValid() ) continue;
 
          const TrackingRecHit*         hit               = (*measurement_it->recHit()).hit();
          const SiStripRecHit2D*        sistripsimplehit  = dynamic_cast<const SiStripRecHit2D*>(hit);
          const SiStripMatchedRecHit2D* sistripmatchedhit = dynamic_cast<const SiStripMatchedRecHit2D*>(hit);
          const SiStripRecHit1D*        sistripsimple1dhit  = dynamic_cast<const SiStripRecHit1D*>(hit);
 
          if(sistripsimplehit){
              Learn((sistripsimplehit->cluster()).get(), trajState);
          }else if(sistripmatchedhit){
              Learn(&sistripmatchedhit->monoCluster(),trajState);
              Learn(&sistripmatchedhit->stereoCluster(),trajState);
          }else if(sistripsimple1dhit){
              Learn((sistripsimple1dhit->cluster()).get(), trajState);
          }else{
          }
 
       }
    }
 
 }

void DeDxDiscriminatorLearner::algoBeginJob ( const edm::EventSetup & iSetup )

privatevirtual

Reimplemented from ConditionDBWriter< PhysicsTools::Calibration::HistogramD3D >.

Definition at line 72 of file DeDxDiscriminatorLearner.cc.

References Charge_Max, Charge_Min, Charge_NBins, Charge_Vs_Path, compareJSON::const, DeDxDiscriminatorLearner::stModInfo::DetId, reco::tau::disc::Eta(), DeDxDiscriminatorLearner::stModInfo::Eta, edm::EventSetup::get(), i, m_tracker, MODsColl, DeDxDiscriminatorLearner::stModInfo::NAPV, StripTopology::nstrips(), P_Max, P_Min, P_NBins, Path_Max, Path_Min, Path_NBins, edm::ESHandle< class >::product(), dttmaxenums::R, DeDxDiscriminatorLearner::stModInfo::R, DetId::rawId(), DeDxDiscriminatorLearner::stModInfo::SubDet, DetId::subdetId(), StripSubdetector::TEC, DeDxDiscriminatorLearner::stModInfo::Thickness, StripSubdetector::TIB, StripSubdetector::TID, and StripSubdetector::TOB.

 {
 //   Charge_Vs_Path = new TH2F ("Charge_Vs_Path"     , "Charge_Vs_Path" , 24, 0.2, 1.4, 250, 0, 5000);
    Charge_Vs_Path = new TH3F ("Charge_Vs_Path"     , "Charge_Vs_Path" , P_NBins, P_Min, P_Max, Path_NBins, Path_Min, Path_Max, Charge_NBins, Charge_Min, Charge_Max);
 
 
    edm::ESHandle<TrackerGeometry> tkGeom;
    iSetup.get<TrackerDigiGeometryRecord>().get( tkGeom );
    m_tracker = tkGeom.product();
 
    auto const & Det = tkGeom->dets();
    for(unsigned int i=0;i<Det.size();i++){
       DetId  Detid  = Det[i]->geographicalId();
       int    SubDet = Detid.subdetId();
 
       if( SubDet == StripSubdetector::TIB ||  SubDet == StripSubdetector::TID ||
           SubDet == StripSubdetector::TOB ||  SubDet == StripSubdetector::TEC  ){
 
           auto DetUnit     = dynamic_cast<StripGeomDetUnit const*> (Det[i]);
           if(!DetUnit)continue;
 
           const StripTopology& Topo     = DetUnit->specificTopology();
           unsigned int         NAPV     = Topo.nstrips()/128;
 
           double Eta           = DetUnit->position().basicVector().eta();
           double R             = DetUnit->position().basicVector().transverse();
           double Thick         = DetUnit->surface().bounds().thickness();
 
           stModInfo* MOD       = new stModInfo;
           MOD->DetId           = Detid.rawId();
           MOD->SubDet          = SubDet;
           MOD->Eta             = Eta;
           MOD->R               = R;
           MOD->Thickness       = Thick;
           MOD->NAPV            = NAPV;
           MODsColl[MOD->DetId] = MOD;
       }
    }
 
 }

void DeDxDiscriminatorLearner::algoEndJob ( )

privatevirtual

Reimplemented from ConditionDBWriter< PhysicsTools::Calibration::HistogramD3D >.

Definition at line 115 of file DeDxDiscriminatorLearner.cc.

References algoMode, Charge_Vs_Path, HistoFile, Input, and Output.

 {
 
    if( strcmp(algoMode.c_str(),"MultiJob")==0){
     TFile* Output = new TFile(HistoFile.c_str(), "RECREATE");
         Charge_Vs_Path->Write();
     Output->Write();
     Output->Close();
    }else if( strcmp(algoMode.c_str(),"WriteOnDB")==0){
         TFile* Input = new TFile(HistoFile.c_str() );
     Charge_Vs_Path = (TH3F*)(Input->FindObjectAny("Charge_Vs_Path"))->Clone();  
     Input->Close();
    }
 
 }

PhysicsTools::Calibration::HistogramD3D * DeDxDiscriminatorLearner::getNewObject ( )

privatevirtual

Implements ConditionDBWriter< PhysicsTools::Calibration::HistogramD3D >.

Definition at line 204 of file DeDxDiscriminatorLearner.cc.

References Charge_Vs_Path, getGTfromDQMFile::obj, and PhysicsTools::Calibration::Histogram3D< Value_t, AxisX_t, AxisY_t, AxisZ_t >::setBinContent().

 {
 //   if( strcmp(algoMode.c_str(),"MultiJob")==0)return NULL;
 
    PhysicsTools::Calibration::HistogramD3D* obj;
    obj = new PhysicsTools::Calibration::HistogramD3D(
                 Charge_Vs_Path->GetNbinsX(), Charge_Vs_Path->GetXaxis()->GetXmin(),  Charge_Vs_Path->GetXaxis()->GetXmax(),
                 Charge_Vs_Path->GetNbinsY(), Charge_Vs_Path->GetYaxis()->GetXmin(),  Charge_Vs_Path->GetYaxis()->GetXmax(),
             Charge_Vs_Path->GetNbinsZ(), Charge_Vs_Path->GetZaxis()->GetXmin(),  Charge_Vs_Path->GetZaxis()->GetXmax());
 
    for(int ix=0; ix<=Charge_Vs_Path->GetNbinsX()+1; ix++){
       for(int iy=0; iy<=Charge_Vs_Path->GetNbinsY()+1; iy++){
          for(int iz=0; iz<=Charge_Vs_Path->GetNbinsZ()+1; iz++){
             obj->setBinContent(ix, iy, iz, Charge_Vs_Path->GetBinContent(ix,iy, iz) );       
 //          if(Charge_Vs_Path->GetBinContent(ix,iy)!=0)printf("%i %i %i --> %f\n",ix,iy, iz, Charge_Vs_Path->GetBinContent(ix,iy,iz)); 
          }
       }
    }
 
    return obj;
 }

void DeDxDiscriminatorLearner::Learn	(	const SiStripCluster *	cluster,
		TrajectoryStateOnSurface	trajState
	)

private

Definition at line 183 of file DeDxDiscriminatorLearner.cc.

References SiStripCluster::amplitudes(), DeDxDiscriminatorTools::charge(), Charge_Vs_Path, SiStripCluster::firstStrip(), TrackerGeometry::idToDetUnit(), DeDxDiscriminatorTools::IsFarFromBorder(), DeDxDiscriminatorTools::IsSpanningOver2APV(), TrajectoryStateOnSurface::localDirection(), TrajectoryStateOnSurface::localMomentum(), m_tracker, PV3DBase< T, PVType, FrameType >::mag(), MaxNrStrips, MODsColl, DeDxDiscriminatorTools::path(), cmsHarvester::path, DeDxDiscriminatorLearner::stModInfo::Thickness, and PV3DBase< T, PVType, FrameType >::z().

Referenced by algoAnalyze().

 {
    // Get All needed variables
    LocalVector             trackDirection = trajState.localDirection();
    double                  cosine         = trackDirection.z()/trackDirection.mag();
    const vector<uint8_t>&  ampls          = cluster->amplitudes();
    uint32_t                detId          = 0; // zero since long time cluster->geographicalId();
    int                     firstStrip     = cluster->firstStrip();
    stModInfo* MOD                         = MODsColl[detId];
    // Sanity Checks
    if( ampls.size()>MaxNrStrips)                                                                      { return; }
 // if( DeDxDiscriminatorTools::IsSaturatingStrip  (ampls))                                            { return; }
    if( DeDxDiscriminatorTools::IsSpanningOver2APV (firstStrip, ampls.size()))                         { return; }
    if(!DeDxDiscriminatorTools::IsFarFromBorder    (trajState, m_tracker->idToDetUnit(DetId(detId)) )) { return; }
    // Fill Histo Path Vs Charge/Path
    double charge = DeDxDiscriminatorTools::charge(ampls);
    double path   = DeDxDiscriminatorTools::path(cosine,MOD->Thickness);
    Charge_Vs_Path->Fill(trajState.localMomentum().mag(),path,charge/path);
 }