PFMultiDepthClusterizer Class Reference

#include <PFMultiDepthClusterizer.h>

Inheritance diagram for PFMultiDepthClusterizer:

Classes
class	ClusterLink

Public Member Functions
void	buildClusters (const reco::PFClusterCollection &, const std::vector< bool > &, reco::PFClusterCollection &outclus)
B2DGPF &	operator= (const B2DGPF &)=delete
	PFMultiDepthClusterizer (const edm::ParameterSet &conf)
	PFMultiDepthClusterizer (const B2DGPF &)=delete
void	update (const edm::EventSetup &es)
virtual	~PFMultiDepthClusterizer ()
Public Member Functions inherited from PFClusterBuilderBase
std::ostream &	operator<< (std::ostream &o) const
PFCBB &	operator= (const PFCBB &)=delete
	PFClusterBuilderBase (const edm::ParameterSet &conf)
	PFClusterBuilderBase (const PFCBB &)=delete
void	reset ()
	~PFClusterBuilderBase ()

Private Types
typedef PFMultiDepthClusterizer	B2DGPF

Private Member Functions
void	absorbCluster (reco::PFCluster &, const reco::PFCluster &)
void	calculateShowerShapes (const reco::PFClusterCollection &, std::vector< double > &, std::vector< double > &)
void	expandCluster (reco::PFCluster &, unsigned int point, std::vector< bool > &mask, const reco::PFClusterCollection &, const std::vector< ClusterLink > &links)
std::vector< ClusterLink >	link (const reco::PFClusterCollection &, const std::vector< double > &, const std::vector< double > &)
std::vector< ClusterLink >	prune (std::vector< ClusterLink > &, std::vector< bool > &linkedClusters)

Private Attributes
std::unique_ptr < PFCPositionCalculatorBase >	_allCellsPosCalc
double	nSigmaEta_
double	nSigmaPhi_

Additional Inherited Members
Public Types inherited from PFClusterBuilderBase
typedef PFCPositionCalculatorBase	PosCalc
Protected Attributes inherited from PFClusterBuilderBase
const float	_minFractionToKeep
unsigned	_nClustersFound
unsigned	_nSeeds
std::unique_ptr< PosCalc >	_positionCalc

Detailed Description

Definition at line 9 of file PFMultiDepthClusterizer.h.

Member Typedef Documentation

typedef PFMultiDepthClusterizer PFMultiDepthClusterizer::B2DGPF

private

Definition at line 10 of file PFMultiDepthClusterizer.h.

Constructor & Destructor Documentation

PFMultiDepthClusterizer::PFMultiDepthClusterizer

(

const edm::ParameterSet &

conf

)

Definition at line 15 of file PFMultiDepthClusterizer.cc.

References _allCellsPosCalc, edm::ParameterSet::exists(), reco::get(), edm::ParameterSet::getParameter(), edm::ParameterSet::getParameterSet(), nSigmaEta_, nSigmaPhi_, NULL, funct::pow(), and AlCaHLTBitMon_QueryRunRegistry::string.

                                                     :
  PFClusterBuilderBase(conf) 
{
  
  _allCellsPosCalc.reset(NULL);
  if( conf.exists("allCellsPositionCalc") ) {
    const edm::ParameterSet& acConf = 
      conf.getParameterSet("allCellsPositionCalc");
    const std::string& algoac = 
      acConf.getParameter<std::string>("algoName");
    PosCalc* accalc = 
      PFCPositionCalculatorFactory::get()->create(algoac, acConf);
    _allCellsPosCalc.reset(accalc);
  }

  nSigmaEta_ = pow(conf.getParameter<double>("nSigmaEta"),2);
  nSigmaPhi_ = pow(conf.getParameter<double>("nSigmaPhi"),2);

}

virtual PFMultiDepthClusterizer::~PFMultiDepthClusterizer ( )

inlinevirtual

Definition at line 14 of file PFMultiDepthClusterizer.h.

{}

PFMultiDepthClusterizer::PFMultiDepthClusterizer ( const B2DGPF &  )

delete

Member Function Documentation

void PFMultiDepthClusterizer::absorbCluster ( reco::PFCluster &  main, const reco::PFCluster &  added  )

private

Definition at line 212 of file PFMultiDepthClusterizer.cc.

References reco::PFCluster::addRecHitFraction(), reco::e1, reco::e2, reco::PFCluster::recHitFractions(), reco::CaloCluster::seed(), and reco::CaloCluster::setSeed().

Referenced by expandCluster().

                                                               {
  
  double e1=0.0;
  double e2=0.0;

  //find seeds
  for ( const auto& fraction : main.recHitFractions())
    if (fraction.recHitRef()->detId() == main.seed()) {
      e1=fraction.recHitRef()->energy();
    }

  for ( const auto& fraction : added.recHitFractions()) {
    main.addRecHitFraction(fraction); 
    if (fraction.recHitRef()->detId() == added.seed()) {
      e2=fraction.recHitRef()->energy();
    }
  }
  if (e2>e1)
    main.setSeed(added.seed());

}

void PFMultiDepthClusterizer::buildClusters ( const reco::PFClusterCollection &  input, const std::vector< bool > &  seedable, reco::PFClusterCollection &  outclus  )

virtual

Implements PFClusterBuilderBase.

Definition at line 36 of file PFMultiDepthClusterizer.cc.

References _allCellsPosCalc, calculateShowerShapes(), expandCluster(), i, link(), electronStore::links, and prune().

                                           {

  std::vector<double> etaRMS(input.size(),0.0);
  std::vector<double> phiRMS(input.size(),0.0);

  //We need to sort the clusters for smaller to larger depth
  //  for (unsigned int i=0;i<input.size();++i)
  //   printf(" cluster%f %f \n",input[i].depth(),input[i].energy());



  //calculate cluster shapes
  calculateShowerShapes(input,etaRMS,phiRMS);

  //link
  std::vector<ClusterLink> links = link(input,etaRMS,phiRMS); 
  //  for (const auto& link: links)
  //    printf("link %d %d %f %f\n",link.from(),link.to(),link.dR(),link.dZ());


 
  std::vector<bool> mask(input.size(),false);
  std::vector<bool> linked(input.size(),false);

  //prune
  std::vector<ClusterLink> prunedLinks;
  if (links.size())
    prunedLinks =  prune(links,linked);

  //printf("Pruned links\n")
  //  for (const auto& link: prunedLinks)
  //  printf("link %d %d %f %f\n",link.from(),link.to(),link.dR(),link.dZ());



  //now we need to build clusters 
  for (unsigned int i=0;i<input.size();++i) {
    //if not masked
    if( mask[i])
      continue;
    //if not linked just spit it out
    if (!linked[i]) {
      output.push_back(input[i]);
      //      printf("Added single cluster with energy =%f \n",input[i].energy());
      mask[i] = true;
      continue;
    }


    //now business: if  linked and not  masked gather clusters
    reco::PFCluster cluster = input[i];
    mask[i] = true;
    expandCluster(cluster,i, mask,input,prunedLinks);
    _allCellsPosCalc->calculateAndSetPosition(cluster);
    output.push_back(cluster);
    //    printf("Added linked cluster with energy =%f\n",cluster.energy());
    
  }
}

void PFMultiDepthClusterizer::calculateShowerShapes ( const reco::PFClusterCollection &  clusters, std::vector< double > &  etaRMS, std::vector< double > &  phiRMS  )

private

Definition at line 100 of file PFMultiDepthClusterizer.cc.

References SiPixelRawToDigiRegional_cfi::deltaPhi, reco::PFCluster::energy(), cropTnPTrees::frac, i, max(), reco::CaloCluster::position(), and reco::PFCluster::recHitFractions().

Referenced by buildClusters().

                                                                                                                   {
  float etaSum;
  float phiSum;

  //shower shapes. here do not use the fractions 

  for( unsigned int i=0;i<clusters.size();++i ) {
    const reco::PFCluster& cluster = clusters[i]; 
    etaSum=0.0;phiSum=0.0;
    for (const auto& frac : cluster.recHitFractions()) {
      etaSum +=frac.fraction()*frac.recHitRef()->energy()*fabs(frac.recHitRef()->position().Eta() -cluster.position().Eta() );
      phiSum +=frac.fraction()*frac.recHitRef()->energy()*fabs(deltaPhi(frac.recHitRef()->position().Phi(),cluster.position().Phi()));
    }
    //protection for single line : assign ~ tower
    etaRMS[i] = std::max(etaSum/cluster.energy(),0.1);
    phiRMS[i] = std::max(phiSum/cluster.energy(),0.1);
    
  }

}

void PFMultiDepthClusterizer::expandCluster ( reco::PFCluster &  cluster, unsigned int  point, std::vector< bool > &  mask, const reco::PFClusterCollection &  clusters, const std::vector< ClusterLink > &  links  )

private

Definition at line 237 of file PFMultiDepthClusterizer.cc.

References absorbCluster(), link(), electronStore::links, and point.

Referenced by buildClusters().

                                                                                                                                                             {
  for (const auto& link : links) {
    if (link.from() == point) {
      //found link that starts from this guy if not masked absorb
      if (!mask[link.from()]) {
    absorbCluster(cluster,clusters[link.from()]);
    mask[link.from()] = true;
      }

      if (!mask[link.to()]) {
    absorbCluster(cluster,clusters[link.to()]);
    mask[link.to()]=true;
    expandCluster(cluster,link.to(),mask,clusters,links);
      }
    }
    if (link.to() == point) {
      //found link that starts from this guy if not masked absorb
      if (!mask[link.to()]) {
    absorbCluster(cluster,clusters[link.to()]);
    mask[link.to()] = true;
      }

      if (!mask[link.from()]) {
    absorbCluster(cluster,clusters[link.from()]);
    mask[link.from()]=true;
    expandCluster(cluster,link.from(),mask,clusters,links);
      }
    }



  }

}

std::vector< PFMultiDepthClusterizer::ClusterLink > PFMultiDepthClusterizer::link ( const reco::PFClusterCollection &  clusters, const std::vector< double > &  etaRMS, const std::vector< double > &  phiRMS  )

private

Definition at line 123 of file PFMultiDepthClusterizer.cc.

References SiPixelRawToDigiRegional_cfi::deltaPhi, reco::PFCluster::depth(), reco::PFCluster::energy(), i, j, electronStore::links, nSigmaEta_, nSigmaPhi_, and reco::CaloCluster::position().

Referenced by buildClusters(), and expandCluster().

                                                                                                               {

  std::vector<ClusterLink> links;
  //loop on all pairs
  for (unsigned int i=0;i<clusters.size();++i)
    for (unsigned int j=0;j<clusters.size();++j) {
      if (i==j )
    continue;

      const reco::PFCluster& cluster1 = clusters[i]; 
      const reco::PFCluster& cluster2 = clusters[j]; 

      float dz = (cluster2.depth() - cluster1.depth());

      //Do not link at the same layer and only link inside out!
      if (dz<0.0 || fabs(dz)<0.2)
    continue;

      float deta =(cluster1.position().Eta()-cluster2.position().Eta())*(cluster1.position().Eta()-cluster2.position().Eta())/(etaRMS[i]*etaRMS[i]+etaRMS[j]*etaRMS[j]);
      float dphi = deltaPhi(cluster1.position().Phi(),cluster2.position().Phi())*deltaPhi(cluster1.position().Phi(),cluster2.position().Phi())/(phiRMS[i]*phiRMS[i]+phiRMS[j]*phiRMS[j]);

      //      printf("Testing Link %d -> %d (%f %f %f %f ) \n",i,j,deta,dphi,cluster1.position().Eta()-cluster2.position().Eta(),deltaPhi(cluster1.position().Phi(),cluster2.position().Phi()));

      if (deta<nSigmaEta_ && dphi<nSigmaPhi_ ) 
    links.push_back(ClusterLink(i,j,deta+dphi,fabs(dz),cluster1.energy()+cluster2.energy()));
    }

      return links;
}

B2DGPF& PFMultiDepthClusterizer::operator= ( const B2DGPF &  )

delete

std::vector< PFMultiDepthClusterizer::ClusterLink > PFMultiDepthClusterizer::prune ( std::vector< ClusterLink > &  links, std::vector< bool > &  linkedClusters  )

private

Definition at line 154 of file PFMultiDepthClusterizer.cc.

References PFMultiDepthClusterizer::ClusterLink::dR(), PFMultiDepthClusterizer::ClusterLink::dZ(), PFMultiDepthClusterizer::ClusterLink::energy(), i, j, and PFMultiDepthClusterizer::ClusterLink::to().

Referenced by buildClusters(), and dirstructure.Directory::prune().

                                                                     {
      std::vector<ClusterLink> goodLinks ;
      std::vector<bool> mask(links.size(),false);

      for (unsigned int i=0;i<links.size()-1;++i) {
    if (mask[i])
      continue;
    for (unsigned int j=i+1;j<links.size();++j) {

      if (mask[j])
        continue;
      
      const ClusterLink& link1  = links[i];
      const ClusterLink& link2  = links[j];

      if (link1.to() == link2.to()) {//found two links going to the same spot,kill one  
        //first prefer nearby layers
        if (link1.dZ() < link2.dZ()) {
          mask[j]=true;
        }
        else if (link1.dZ() > link2.dZ()) {
          mask[i] = true;
        }
        else if (fabs(link1.dZ()-link2.dZ())<0.2) { //if same layer-pick based on transverse compatibility
          if (link1.dR()<link2.dR()) {
        mask[j]=true;
          }
          else if (link1.dR()>link2.dR()) {
        mask[i] = true;
          }
          else {
        //same distance as well -> can happen !!!!! Pick the highest SUME 
        if (link1.energy()<link2.energy())
          mask[i] = true;
        else
          mask[j] = true;
          
          }
        }
      }
    }
      }

      for (unsigned int i=0;i<links.size();++i) {
    if (mask[i])
      continue;
    goodLinks.push_back(links[i]);
    linkedClusters[links[i].from()]=true;
    linkedClusters[links[i].to()]=true;
      }


      return goodLinks;       
}   

void PFMultiDepthClusterizer::update ( const edm::EventSetup &  es )

inlinevirtual

Reimplemented from PFClusterBuilderBase.

Definition at line 18 of file PFMultiDepthClusterizer.h.

References _allCellsPosCalc.

Referenced by progressbar.ProgressBar::__next__(), relval_steps.Matrix::__setitem__(), relval_steps.Steps::__setitem__(), Vispa.Gui.VispaWidget.VispaWidget::autosize(), Vispa.Views.LineDecayView.LineDecayContainer::createObject(), Vispa.Views.LineDecayView.LineDecayContainer::deselectAllObjects(), Vispa.Gui.VispaWidgetOwner.VispaWidgetOwner::deselectAllWidgets(), Vispa.Gui.VispaWidget.VispaWidget::enableAutosizing(), progressbar.ProgressBar::finish(), Vispa.Gui.MenuWidget.MenuWidget::leaveEvent(), Vispa.Gui.VispaWidgetOwner.VispaWidgetOwner::mouseMoveEvent(), Vispa.Gui.MenuWidget.MenuWidget::mouseMoveEvent(), Vispa.Views.LineDecayView.LineDecayContainer::mouseMoveEvent(), Vispa.Gui.VispaWidgetOwner.VispaWidgetOwner::mouseReleaseEvent(), Vispa.Views.LineDecayView.LineDecayContainer::objectMoved(), relval_steps.Steps::overwrite(), Vispa.Views.LineDecayView.LineDecayContainer::removeObject(), Vispa.Gui.ConnectableWidget.ConnectableWidget::removePorts(), Vispa.Gui.FindDialog.FindDialog::reset(), Vispa.Gui.PortConnection.PointToPointConnection::select(), Vispa.Gui.VispaWidget.VispaWidget::select(), Vispa.Views.LineDecayView.LineDecayContainer::select(), Vispa.Gui.VispaWidget.VispaWidget::setText(), Vispa.Gui.VispaWidget.VispaWidget::setTitle(), Vispa.Gui.ZoomableWidget.ZoomableWidget::setZoom(), Vispa.Views.LineDecayView.LineDecayContainer::setZoom(), and Vispa.Gui.PortConnection.PointToPointConnection::updateConnection().

                                       { 
    _allCellsPosCalc->update(es);
  }

Member Data Documentation

std::unique_ptr<PFCPositionCalculatorBase> PFMultiDepthClusterizer::_allCellsPosCalc

private

Definition at line 27 of file PFMultiDepthClusterizer.h.

Referenced by buildClusters(), PFMultiDepthClusterizer(), and update().

double PFMultiDepthClusterizer::nSigmaEta_

private

Definition at line 28 of file PFMultiDepthClusterizer.h.

Referenced by link(), and PFMultiDepthClusterizer().

double PFMultiDepthClusterizer::nSigmaPhi_

private

Definition at line 29 of file PFMultiDepthClusterizer.h.

Referenced by link(), and PFMultiDepthClusterizer().