MCParticleImpl.h

// -*- C++ -*-
#ifndef IMPL_MCPARTICLEIMPL_H
#define IMPL_MCPARTICLEIMPL_H 1
#include <vector>

#include "EVENT/LCIO.h"
#include "AccessChecked.h"

#include "EVENT/MCParticle.h"
#include <bitset>

namespace IMPL {
  
  // fg 01042004 : changed to use vector of pointers !
  //   typedef std::vector<EVENT::MCParticle**> MCParticlePVec ;
  // use  pointers to pointer to MCParticle 
  // as SIO needs the address of the pointer for pointer reallocation....

  class MCParticleImpl : public EVENT::MCParticle, public AccessChecked {
    
  public: 

    MCParticleImpl() ;

//     /** Copy c'tor */
//     MCParticleImpl(const EVENT::MCParticle& p) ;
    
    virtual ~MCParticleImpl() ;

    virtual int id() const { return simpleUID() ; }

    virtual const EVENT::MCParticleVec & getParents() const ;

    virtual const EVENT::MCParticleVec & getDaughters() const ;


    // /** Returns the i-th daughter of this particle.
    //  * Same as getDaughter() except for return type.
    //  *
    //  * @see getNumberOfDaughters
    //  */
    // virtual EVENT::MCParticle * getDaughter(int i) const ;


    // /** Returns the number of parents of this particle - 0 if mother.
    //  */
    // virtual int getNumberOfParents() const ;


    // /** Returns the i-th parent of this particle.
    //  */
    // virtual EVENT::MCParticle * getParent(int i) const ;


    virtual const double* getEndpoint() const ;

    // /** Returns the number of daughters of this particle.
    //  */
    // virtual int getNumberOfDaughters() const ;


    virtual int getPDG() const ;

    virtual int getGeneratorStatus() const ;

    virtual int getSimulatorStatus() const ;

    virtual bool isCreatedInSimulation() const ;

    virtual bool isBackscatter() const ;

    virtual bool vertexIsNotEndpointOfParent() const ;

    virtual bool isDecayedInTracker() const ;

    virtual bool isDecayedInCalorimeter() const ;

    virtual bool hasLeftDetector() const ;

    virtual bool isStopped() const ;

    virtual bool isOverlay() const ;

    virtual const double* getVertex() const ;

    virtual float getTime() const ;

    virtual const double * getMomentum() const ;


    virtual const double* getMomentumAtEndpoint() const ;

    virtual double getMass() const ;

    virtual float getCharge() const ;


    virtual double getEnergy() const ;

    virtual const float* getSpin() const ;


    virtual const int* getColorFlow() const ;


    // set methods
    void addParent(  EVENT::MCParticle *mom ) ;

//     // set methods
//     /** Sets the parent. 
//      */
//     void setParent(  EVENT::MCParticle *mom0 ) ;

//     /** Sets a second parent.
//      */
//     void setSecondParent(  EVENT::MCParticle *mom1 ) ;


    void setPDG(int pdg ) ;

    void setGeneratorStatus( int status ) ;

    void setSimulatorStatus( int status ) ;

    void setVertex( const double vtx[3] ) ;
    
    void setTime( float time) ;

    void setEndpoint( const double pnt[3] ) ;
    
    void setMomentum( const float p[3] );

    void setMomentum( const double p[3] );

    void setMomentumAtEndpoint( const float p[3] );

    void setMomentumAtEndpoint( const double p[3] );

    void setMass( float m ) ;

    void setCharge( float c ) ;

    void setSpin( const float spin[3] );

    void setColorFlow( const int cflow[2] );



    // setters for simulator status
    virtual void setCreatedInSimulation(bool val) ;

    virtual void setBackscatter(bool val) ;

    virtual void setVertexIsNotEndpointOfParent(bool val) ;

    virtual void setDecayedInTracker(bool val) ;

    virtual void setDecayedInCalorimeter(bool val) ;

    virtual void setHasLeftDetector(bool val) ;

    virtual void setStopped(bool val) ;

    virtual void setOverlay(bool val) ;

  protected:

    void addDaughter(  EVENT::MCParticle* daughter) ;

//     EVENT::MCParticle* _mother0 ;
//     EVENT::MCParticle* _mother1 ;
    int _pdg ;
    int _genstatus ;
    //    int _simstatus ;
    std::bitset<32> _simstatus ;
    double _vertex[3] ;
    double _endpoint[3] ;
    double _pEndpoint[3] ;
    double _p[3] ;
    double _mass ;
    float _charge ;
    float _time ;
    EVENT::MCParticleVec _parents ;
    EVENT::MCParticleVec _daughters ;
    bool _endpointSet ;
    float _spin[3] ;
    int _colorFlow[2] ;

}; // class
} // namespace IMPL
#endif /* ifndef IMPL_MCPARTICLEIMPL_H */