PhotosEvent.cxx

#include <vector>
#include "PhotosParticle.h"
#include "PhotosBranch.h"
#include "PhotosEvent.h"
#include "Log.h"
using std::vector;

namespace Photospp
{

PhotosEvent::~PhotosEvent()
{
        while(m_branch_points.size()!=0)
        {
                PhotosBranch *temp = m_branch_points.back();
                m_branch_points.pop_back();
                delete temp;
        }
}

void PhotosEvent::process()
{
        //print();
        vector<PhotosParticle*> particles = filterParticles( getParticleList() );
        m_branch_points = PhotosBranch::createBranches(particles);

        for(int i=0;i<(int)m_branch_points.size();i++)
                m_branch_points.at(i)->process();
        //print();
}

vector<PhotosParticle *> PhotosEvent::filterParticles(vector<PhotosParticle *> particles)
{
        vector<PhotosParticle *> filtered;
        for(int i=0;i<(int)particles.size();i++)
        {
                PhotosParticle *p = particles.at(i);
                if(!p) continue;

                //check that the particle decays
                if(p->getStatus()==PhotosParticle::STABLE) continue;

                //check for self decays
                vector<PhotosParticle *> daughters = p->getDaughters();
                int j=0;
                for(j=0;j<(int)daughters.size();j++)
                        if(daughters.at(j)->getPdgID()==p->getPdgID()) break;
                if(j!=(int)daughters.size()) continue;

                Log::Debug(2)<<"Passed particle filter"<<endl;
                filtered.push_back(p);
        }
        return filtered;
}

} // namespace Photospp