//---------------------------------------------------------------------------------
// The AliKFParticle class
// .
// @author  M.Zyzak, I.Kulakov, I. Kisel
// @version 1.0
// @since   29.06.11
// 
// Class to reconstruct and store the decayed particle parameters.
// The method is described in CBM-SOFT note 2007-003, 
// ``Reconstruction of decayed particles based on the Kalman filter'', 
// http://www.gsi.de/documents/DOC-2007-May-14-1.pdf
//
// This class is ALICE interface to general mathematics in AliKFParticleBase
// 
//  -= Copyright &copy ALICE HLT Group =-
//_________________________________________________________________________________

#ifndef ALIKFPARTICLETEST_H
#define ALIKFPARTICLETEST_H

#include "TString.h"
#include "TH1F.h"
#include "TH2F.h"
#include <vector>

struct TESDTrackInfo;
class AliESDEvent;

class AliKFParticleTest
{
 public:

  AliKFParticleTest();
  ~AliKFParticleTest();

  //Set functions
  void SetInputFile(TString n) { fInpFileName.clear(); fInpFileName.push_back(n); }
  void AddInputFile(TString n) { fInpFileName.push_back(n); }
  void SetOutputFile(TString n) { fOutFileName = n; }

  void SetTestParticlesPDG(std::vector<Int_t> &p) {fTestParticlesPDG = p;}
  void AddTestParticlesPDG(Int_t pdg) {fTestParticlesPDG.push_back(pdg);}

  //Test itself
  void RunTest();

 private:

  //funcions
  void SelectParticles(AliESDEvent *event, std::vector<TESDTrackInfo> &ESDTrackInfo);
  void CheckInputTracks(AliESDEvent *event, std::vector<TESDTrackInfo> &ESDTrackInfo );
  void TestKFParticle(AliESDEvent *event, std::vector<TESDTrackInfo> &ESDTrackInfo);
  void TestKFParticleSIMD(AliESDEvent *event, std::vector<TESDTrackInfo> &ESDTrackInfo);
  void TestScalarVsSIMD(AliESDEvent *event, std::vector<TESDTrackInfo> &ESDTrackInfo);
  void TestKFParticleSpeed(AliESDEvent *event, float &timer, float &timerSIMD, int &Npart);
  void TestKFParticleBG(AliESDEvent *event);
  void WriteHistos();
  //data

  //file names
  std::vector<TString> fInpFileName;
  TString fOutFileName;

  //Particles for the tests
  std::vector<Int_t> fTestParticlesPDG;

  //histograms
  static const Int_t cNHistos = 34;
  TH1F *fhKFParticle[cNHistos];
  TH1F *fhKFParticleSIMD[cNHistos];
  TH1F *fhKFInputTracks[10];
  
  ClassDef( AliKFParticleTest, 1 );
};

#endif