EdbPrediction.h

Go to the documentation of this file.

#ifndef ROOT_EdbPrediction
#define ROOT_EdbPrediction
 
 
#include "TClonesArray.h"
#include "TNtuple.h"
 
#ifndef ROOT_EdbVirtual
#include "EdbVirtual.h"
#endif
 
 
//______________________________________________________________________________
class EdbPredictionDC : public TObject, public EdbTrack2D {
 
private:
 
  Int_t      eIDp;        
  Int_t      eEvent;      
  Int_t      eFlag;       
 
  Float_t    eTy;         
  Float_t    eTz;         
  Float_t    eYp;         
  Float_t    eZp;         
 
public:
  EdbPredictionDC() { }
  EdbPredictionDC( int id, int event, int flag, 
           float ty,  float tz,  float y,  float z ):
    eIDp(id), eEvent(event), eFlag(flag), 
     eTy(ty), eTz(tz), eYp(y), eZp(z) {}
  virtual ~EdbPredictionDC() { }
 
  virtual Int_t      GetID()    const { return eIDp; }
  virtual Int_t      GetEvent() const { return eEvent; }
  virtual Int_t      GetFlag()  const { return eFlag; }
  virtual Float_t    GetTy()    const { return eTy; }
  virtual Float_t    GetTz()    const { return eTz; }
  virtual Float_t    GetYp()    const { return eYp; }
  virtual Float_t    GetZp()    const { return eZp; }
 
  //  virtual void       Set( float y,  float z,  float ty,  float tz );
 
 
  virtual Float_t    X()          const     { return GetYp(); }
  virtual Float_t    Y()          const     { return GetZp(); }
  virtual void    SetX( float x )           { eYp = x; }
  virtual void    SetY( float y )           { eZp = y; }
  
  virtual Float_t    TX()           const   { return GetTy(); }
  virtual Float_t    TY()           const   { return GetTz(); }
  virtual void    SetTX( float tx )         { eTy = tx; }
  virtual void    SetTY( float ty )         { eTz = ty; }
  
  virtual void    Print( Option_t *opt=0) const;
  void            WriteDC( FILE *file ) const;
  
  ClassDef(EdbPredictionDC,2)  // one prediction (track) a-la $c
};
 
//______________________________________________________________________________
class EdbPredictionsBox : public TObject, public EdbPointsBox2D {
 
 private:
 
  TClonesArray *ePredictions;  
 
 public: 
  EdbPredictionsBox();
  EdbPredictionsBox(const EdbPredictionsBox &pb);
  EdbPredictionsBox( int n );
  virtual ~EdbPredictionsBox();
 
  void                 AddPrediction( int id, int event, int flag, 
                        float ty,  float tz, float y,  float z);
 
  Int_t                 GetN()         const;
  EdbPredictionDC      *GetPrediction(int i) const;
  TClonesArray         *GetPredictions()     const { return ePredictions; }
 
  Int_t         N()       const { return GetN(); }
  EdbPoint     *At(int i) const      { return (EdbPoint*)GetPrediction(i); }
 
  //  void  Draw(int style=23, int   col=4, float size=1. );
 
  void       Print( Option_t *opt="") const;
 
  void      WriteDC( char *file="predictions.dc") const;
  void      WriteDC1( char *file="predictions.dc") const;
 
  TNtuple  *MakeNtuple( char *name="predictions") const;
 
  int       ReadDC( char *file );
  int       ReadDC0( char *file );
 
  void      Generate( int n );
 
  //void      Test();
 
  ClassDef(EdbPredictionsBox,2)  // Predictions a-la $c
};
 
#endif /* EdbPrediction */