PndListProvider.h

Go to the documentation of this file.

//--------------------------------------------------------------------------
// 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 "RhoCandList.h"

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

class TParticlePDG;
class TDatabasePDG;
#include "RhoMassParticleSelector.h"
#include "RhoNeutralParticleSelector.h"
#include "RhoPlusParticleSelector.h"
#include "RhoMinusParticleSelector.h"
#include "RhoSimpleElectronSelector.h"
#include "RhoSimpleKaonSelector.h"
#include "RhoSimpleMuonSelector.h"
#include "RhoSimplePionSelector.h"
#include "RhoSimpleProtonSelector.h"

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(RhoCandList &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(RhoCandList &tl);
  int GetNCandidates();

  TH1F *GetHisto() { return fHisto; }
  void FillHisto();

  void Reset();
  void Print();

 private:
  std::string fName;

  RhoCandList fOwnList;

  RhoMinusParticleSelector *fMinusSel;
  RhoPlusParticleSelector *fPlusSel;
  RhoMassParticleSelector *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;
};

#endif