StarKinematics.h

#ifndef __StarKinematics_h__
#define __StarKinematics_h__

#include "StarGenerator/BASE/StarGenerator.h"
#include "StarGenerator/EVENT/StarGenEvent.h"

#include "TMath.h"
#include "TVector3.h"

#include "TH1.h"
#include "TF1.h"

class StarKinematics : public StarGenerator
{
 public:
  StarKinematics( const Char_t *name="StarKine" );
  ~StarKinematics(){ /* nada */ };

  StarGenParticle *AddParticle();

  StarGenParticle *AddParticle( const Char_t *type );

  void Kine( Int_t ntrack, const Char_t *type="pi+,pi-,K+,K-,proton,antiproton", Double_t ptlow=0.0, Double_t pthigh=500.0,
             Double_t ylow=-10.0, Double_t yhigh=+10.0,
             Double_t philow=0.0, Double_t phihigh=TMath::TwoPi() );

  void Dist( Int_t ntrack, const Char_t *type, TF1 *pt, TF1 *y, TF1 *phi=0 );
  void Dist( Int_t ntrack, const Char_t *type, TH1 *pt, TH1 *y, TH1 *phi=0 );
  
  void Cosmic( int ntrack, const char* type="mu+", double plow=3.0, double phigh=10.0, double radius=300.0, double zmin=-3.0, double zmax=+3.0, double dtheta=15.0 );


  Int_t Generate();
  Int_t PreGenerate();
  Int_t PostGenerate();

  void Clear( const Option_t *opts="" ){ /* nada */ };

  Int_t Init(){ return kStOK; }

 private:
 protected:

  // User's event
  StarGenEvent *mUser;

  TVector3 _momentum;  // Used for internal calculations

  ClassDef(StarKinematics,1);

};

#endif