PhotosBranch.h

#ifndef _PhotosBranch_h_included_
#define _PhotosBranch_h_included_
 
#include <vector>
#include "PhotosParticle.h"
using std::vector;
 
namespace Photospp {
 
  class PhotosBranch {
  public:
    PhotosBranch(PhotosParticle* p);
 
    PhotosParticle*          getDecayingParticle() { return particle;  }
 
    vector<PhotosParticle*> getMothers()          { return mothers;   }
 
    vector<PhotosParticle*> getDaughters()        { return daughters; }
 
    vector<PhotosParticle*> getParticles();
 
    int getSuppressionStatus() { return suppression; }
 
    int getForcingStatus()     { return forcing; }
 
    bool checkMomentumConservation();
 
    void process();
 
    static vector<PhotosBranch*> createBranches(vector<PhotosParticle*> particles);
  private:
    int checkSuppressionLevel() { return checkList(false); }
 
    int checkForcingLevel()     { return checkList(true);  }
 
    int checkList(bool forceOrSuppress);
  private:
    int suppression;
    int forcing;
    PhotosParticle*          particle;
    vector<PhotosParticle*> mothers;
    vector<PhotosParticle*> daughters;
  };
 
} // namespace Photospp
#endif