//-----------------------------------------------------------
// File and Version Information:
// $Id$
//
// Description:
//      Add more trackreps to existing tracks
//      Used for adding additional particle hypotheses before
//      final fit
//
// Environment:
//      Software developed for the PANDA Detector at FAIR.
//
// Author List:
//      Felix Boehmer E18,   TUM        (original author)
//
//-----------------------------------------------------------


#ifndef ADDTRACKREPTASK_HH
#define ADDTRACKREPTASK_HH

// Base Class Headers ----------------
#include "FairTask.h"

// Collaborating Class Headers -------
#include "TString.h"

#include <map>

// Collaborating Class Declarations --
class TClonesArray;


class AddTrackRepTask : public FairTask {
public:

  // Constructors/Destructors ---------
  AddTrackRepTask();
  ~AddTrackRepTask();

  // Operators
  

  // Accessors -----------------------


  // Modifiers -----------------------
  void SetTrackBranchName(const TString& name) {fTrackBranchName=name;}
  //add another particle hypothesis. The charge from the existing fit is taken 
  //into account. Valid hypotheses must be 
  void AddHypothesis(const TString& particle);
  

  // Operations ----------------------
  
  virtual InitStatus Init();
  virtual void Exec(Option_t* opt);
  
  

private:

  // Private Data Members ------------
  TClonesArray* fTrackArray;
  TString fTrackBranchName;

  std::map<TString,int> fPdgMap;
  std::vector<TString> fToAdd;
  
  

public:
  ClassDef(AddTrackRepTask,1)

};

#endif

//--------------------------------------------------------------
// $Log$
//--------------------------------------------------------------