RaveVertexFitter.h

/**************************************************************************
 * BASF2 (Belle Analysis Framework 2)                                     *
 * Copyright(C) 2012 - Belle II Collaboration                             *
 *                                                                        *
 * Author: The Belle II Collaboration                                     *
 * Contributors: Moritz Nadler,  Luigi Li Gioi                            *
 *                                                                        *
 * This software is provided "as is" without any warranty.                *
 **************************************************************************/
 
#pragma once
 
#include <analysis/dataobjects/Particle.h>
#include <analysis/VertexFitting/RaveInterface/RaveSetup.h>
 
//std
#include <string>
#include <iostream>
 
//rave
#include <rave/Track.h>
#include <rave/Vertex.h>
 
 
namespace Belle2 {
  class TrackFitResult;
 
  namespace analysis {
    class RaveVertexFitter {
      typedef std::vector<int>::size_type VecSize;
    public:
 
      RaveVertexFitter(/*std::string howToInterfaceRave*/);
 
      ~RaveVertexFitter();
 
      void addTrack(const Particle* const aParticlePtr);
 
      void addTrack(const TrackFitResult* const aTrackPtr);
 
      void addMother(const Particle* const aMotherParticlePtr);
 
      int fit(std::string options = "default");
 
      TVector3 getPos(VecSize vertexId = 0) const;
 
      double getPValue(VecSize vertexId = 0) const;
 
      double getNdf(VecSize vertexId = 0) const;
 
      double getChi2(VecSize vertexId = 0) const;
 
      TMatrixDSym getCov(VecSize vertexId = 0) const;
 
      double getWeight(int trackId, VecSize vertexId = 0) const;
 
      std::vector<int> getTrackIdsForOneVertex(VecSize vertexId = 0) const;
 
      void useBeamSpot(bool beamSpot = true)
      {
        if (beamSpot == true) {
          if (RaveSetup::getRawInstance()->m_useBeamSpot == false) {
            B2ERROR("Beam spot information cannot be used because the beam spot position and covariance was not set in RaveSetup");
            throw;
          }
          m_useBeamSpot = true;
        } else {
          m_useBeamSpot = false;
        }
      }
 
      void clearTracks()
      {
        m_raveTracks.clear();
      }
 
 
      void Print() const
      {
        B2INFO("useBeamSpot: " << m_useBeamSpot);
        B2INFO("number of rave::Tracks: " << m_raveTracks.size());
        B2INFO("number of fitted  rave::Vertices: " << m_raveVertices.size());
 
      }
      int getNTracks() const
      {
        return m_raveTracks.size();
      }
 
      void updateDaughters();
 
 
    protected:
 
//  /** The default constructor checks if RaveSetup was initialized and will set the attributes of RaveVertexFitter */
//  RaveVertexFitter();
      void initBeamSpotMember();
      void isVertexIdValid(const VecSize vertexId) const;
 
      rave::Track TrackFitResultToRaveTrack(const TrackFitResult* const aTrackPtr) const;
      bool m_useBeamSpot; 
      //std::string m_raveAlgorithm;
 
      std::vector<rave::Track> m_raveTracks;
 
      std::vector < rave::Vertex > m_raveVertices;
 
      std::vector<Particle*> m_belleDaughters;
 
 
    };
  }
 
}