//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      PyROOT Wrapper for GFRaveVertexFactory
//      
//      
//
//
// Environment:
//      Software developed for GEM-TPC detector in FOPI.
//
// Author List:
//      Felix Boehmer    TUM            (original author)
//
//
//-----------------------------------------------------------

#include "FopiPidGFVFWrapper.h" 
#include "GFRaveVertexFactory.h"

#include <cassert>



ClassImp(FopiPidGFVFWrapper)


FopiPidGFVFWrapper::FopiPidGFVFWrapper(int verbosity, bool vacProp) {
  fFactory = new  GFRaveVertexFactory(verbosity, vacProp);
  fResult.clear();
  fTracks.clear();
}


FopiPidGFVFWrapper::~FopiPidGFVFWrapper() {
  delete fFactory;
}

void
FopiPidGFVFWrapper::setMethod(const TString& meth) {
  fFactory->setMethod(meth.Data());
}

void
FopiPidGFVFWrapper::setBeamspot(const TVector3& pos, const TMatrixDSym& cov) {
  fFactory->setBeamspot(pos,cov);
}

unsigned int
FopiPidGFVFWrapper::getNVertices() const {
  return fResult.size();
}


GFRaveVertex* 
FopiPidGFVFWrapper::getVertex(unsigned int i) const {
  assert(i<fResult.size());
  return fResult[i];
}

void
FopiPidGFVFWrapper::clearData() { 
  fResult.clear();
  fTracks.clear();
}


void
FopiPidGFVFWrapper::findVertices(bool beamspot) { 
  fFactory->findVertices(&fResult, fTracks,beamspot);
}

void
FopiPidGFVFWrapper::addTrack(GFTrack* track) { 
  fTracks.push_back(track);
}