KFParticlePVReconstructor.h

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file KFParticlePVReconstructor.h

/*
 * This file is part of KFParticle package
 * Copyright (C) 2007-2019 FIAS Frankfurt Institute for Advanced Studies
 *               2007-2019 Goethe University of Frankfurt
 *               2007-2019 Ivan Kisel <I.Kisel@compeng.uni-frankfurt.de>
 *               2007-2019 Maksym Zyzak
 *
 * KFParticle is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * KFParticle is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program.  If not, see <https://www.gnu.org/licenses/>.
 */


#ifndef KFParticlePVReconstructor_H
#define KFParticlePVReconstructor_H

#include "KFVertex.h"
#include "assert.h"

#include <vector>

class KFParticle;
class KFPTrackVector;

class KFParticlePVReconstructor{
 public:
  KFParticlePVReconstructor():fParticles(0), fNParticles(0), fWeight(0.f), fBeamLine(), fIsBeamLine(0), fClusters(0), fPrimVertices(0), fChi2CutPreparation(100), fChi2Cut(16) {};
  ~KFParticlePVReconstructor(){};
  
  void Init(KFPTrackVector *tracks, int nParticles);
  
  void ReconstructPrimVertex();

  int NPrimaryVertices() const { return fPrimVertices.size(); } 
  KFParticle &GetPrimVertex(int iPV=0)   { return fPrimVertices[iPV]; } 
  KFVertex   &GetPrimKFVertex(int iPV=0)   { return fPrimVertices[iPV]; } 
  std::vector<int>& GetPVTrackIndexArray(int iPV=0) { return fClusters[iPV].fTracks; } 
  KFParticle &GetParticle(int i){ assert( i < fNParticles ); return fParticles[i]; } 
  
  void SetBeamLine(KFParticle& p) { fBeamLine = p; fIsBeamLine = 1; } 
  bool IsBeamLine() const { return fIsBeamLine; } 
  
  void AddPV(const KFVertex &pv, const std::vector<int> &tracks);
  void AddPV(const KFVertex &pv);
  void CleanPV() { fClusters.clear(); fPrimVertices.clear(); } 

  void SetChi2PrimaryCut(float chi2) { fChi2Cut = chi2; fChi2CutPreparation = chi2*5; }
  
 private:
  KFParticlePVReconstructor &operator=(KFParticlePVReconstructor &); 
  KFParticlePVReconstructor(KFParticlePVReconstructor &); 

  void FindPrimaryClusters( int cutNDF = 1);

  std::vector<KFParticle> fParticles; 
  int fNParticles;                    

  std::vector<float> fWeight; 
  
  KFParticle fBeamLine; 
  bool fIsBeamLine;     
  
  struct KFParticleCluster {
    KFParticleCluster():fTracks(0) {};
    std::vector<int> fTracks; 
    float fP[3]; 
    float fC[6]; 
  };

  std::vector< KFParticleCluster > fClusters; 
  std::vector<KFVertex> fPrimVertices;  
  
  float fChi2CutPreparation; 
  float fChi2Cut;            
}; // class KFParticlePVReconstructor


#endif // KFParticlePVReconstructor_H