PhotosBranch.h

Go to the documentation of this file.

#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