//--------------------------------------------------------------------------
// Description:
//      Class PndListProvider
//      
//  List Container for PndSimpleAnalysis
//
//  This software was developed for the PANDA collaboration.  If you
//  use all or part of it, please give an appropriate acknowledgement.
//
// Author List:
//      Klaus Goetzen                    Original Author
//
// Copyright Information:
//      Copyright (C) 2008              GSI
//
//------------------------------------------------------------------------

#ifndef PNDLISTPROVIDER_H
#define PNDLISTPROVIDER_H

//----------------------
// Base Class Headers --
//----------------------

#include <string>

//-------------------------------
// Collaborating class Headers --
//-------------------------------


//--------------------------------------------
// Collaborating class forward declarations --
// -------------------------------------------
#include "ArgList.h"
#include "RhoBase/TCandList.h"

//typedef std::vector<PndListProvider*>;

class TParticlePDG;
class TDatabasePDG;
class TCandList;
class TPidMinusSelector;
class TPidPlusSelector;
class TPidMassSelector;

class TH1F;


class PndListProvider
{
public:

  //--------------------
  // Public interface --
  //--------------------

  //
  // Constructors
  //

  PndListProvider(std::string name, std::string pdgType="");
  PndListProvider(std::string name, int pdgcode);
  
  //
  // Destructor
  //
  virtual ~PndListProvider();

  //
  //    Modifiers + Accessors to contained information
  //
  
  std::string GetName() {return fName;}
  bool IsGeneric() {return fIsGeneric;}
  bool ToDump() {return fToDump;}
  
  void SetType(std::string pdgType);
  void SetType(int pdgcode);
  void SetGeneric(bool isgeneric=true) {fIsGeneric=isgeneric;}
  void SetToDump(bool todump=true) {fToDump=todump;}
  void SetHisto(TH1F *h) {fHisto=h;}
  void SetCandList(TCandList &cl);
  void SetMassSelector(double mean, double width);
  
  void AddDaughterPointer(PndListProvider* p);
  void AddDecayProduct(std::string dname);
  void AddDaughterType(std::string dtype);
  void AddDaughterType(int pdgcode);
  int  GetType();
  int  GetDaughterType(int i);
  std::string GetDecayProdName(int i);
  int  GetNDaughters() {return fNDaughters;}
  
  void GetCandList(TCandList& tl);
  int GetNCandidates();
  
  TH1F* GetHisto() {return fHisto;}
  void FillHisto();
  
  void Reset(); 
  void Print();


private:
  std::string      fName;
  
  TCandList       fOwnList;
  
  TPidMinusSelector   *fMinusSel;
  TPidPlusSelector    *fPlusSel;
  TPidMassSelector    *fMassSel;
  
  TH1F             *fHisto;
  
  TParticlePDG*    fParticlePDG;
  std::vector<TParticlePDG*> fDaughterPDG;
  ArgVector        fDaughterListNames;
  std::vector<PndListProvider*> fDaughterPointers;
  
  int              fNDaughters;
  
  bool             fIsGeneric;  
  bool             fToDump;  
  bool             fCandsUptodate;
  bool             fHistoFilled;
  
  TDatabasePDG*    fdbPdg;
    
};

#endif