//*-- Author : Jochen Markert 18.07.2007

#ifndef  HPIDTRACKCLEANER_H
#define  HPIDTRACKCLEANER_H

#include "hreconstructor.h"
#include "hcategory.h"
#include "hpidtracksorter.h"

#include "TString.h"

class HPidTrackCleaner : public HReconstructor {

protected:


    HPidTrackSorter sorter;             //! for doing the final task of flagging
    HCategory* pPidTrackCandCat;        //! HPidTrackCand category
    Bool_t (*pUserSelectLeptons)(HPidTrackCand* ); //!  user provided function pointer to lepton selection
    Bool_t (*pUserSelectHadrons)(HPidTrackCand* ); //!  user provided function pointer to hadron selection

    void   clear(void);
public:
    HPidTrackCleaner(void);
    HPidTrackCleaner(const Text_t *name,const Text_t *title);
    ~HPidTrackCleaner(void);
    Bool_t init      (void);
    Int_t  execute   (void);
    Bool_t finalize  (void);
    static void   setDebug       (void)        { HPidTrackSorter::setDebug();}
    static void   setPrintLevel  (Int_t level) { HPidTrackSorter::setPrintLevel(level);}
    void          setOutputFile(TString filename = "");
    void          setUserSelectionLeptons(Bool_t (*function)(HPidTrackCand* )){ pUserSelectLeptons = function;}
    void          setUserSelectionHadrons(Bool_t (*function)(HPidTrackCand* )){ pUserSelectHadrons = function;}
    ClassDef(HPidTrackCleaner,0);
};
#endif /* !HPIDTRACKCLEANER_H */