InputSort.cc

Go to the documentation of this file.

#include "RecoVertex/KinematicFit/interface/InputSort.h"
#include "RecoVertex/KinematicFitPrimitives/interface/KinematicTree.h"
#include "RecoVertex/VertexPrimitives/interface/VertexException.h"

std::pair<std::vector<RefCountedKinematicParticle>, std::vector<FreeTrajectoryState> > 
                    InputSort::sort(std::vector<RefCountedKinematicParticle> particles) const
{
 if(particles.size()==0) throw VertexException("Sorting particles for vertex fitter::number of particles = 0");
 std::vector<RefCountedKinematicParticle> sortedParticles;
 std::vector<FreeTrajectoryState> sortedStates;

//checking that only top particles of the tree are passed by user
//correcting them for the top ones otherwise 
 for(std::vector<RefCountedKinematicParticle>::const_iterator i = particles.begin(); i != particles.end(); i++)
 {
  if((*i)->correspondingTree() != 0)
  {
   sortedParticles.push_back((*i)->correspondingTree()->topParticle());
   sortedStates.push_back((*i)->correspondingTree()->topParticle()->currentState().freeTrajectoryState());
  }else{
   sortedParticles.push_back(*i);
   sortedStates.push_back((*i)->currentState().freeTrajectoryState());
  }
 }
 return std::pair<std::vector<RefCountedKinematicParticle>, std::vector<FreeTrajectoryState> >(sortedParticles, sortedStates);
}

std::vector<RefCountedKinematicParticle> InputSort::sort(std::vector<RefCountedKinematicTree> trees) const
{
 if(trees.size() ==0) throw VertexException("Input Sort::Zero vector of trees passed"); 
 std::vector<RefCountedKinematicParticle> res;
 for(std::vector<RefCountedKinematicTree>::const_iterator i = trees.begin(); i!=trees.end(); i++)
 {
  (*i)->movePointerToTheTop();
  res.push_back((*i)->currentParticle());
 }
 return res;
}