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