KVEventViewer.h

 
#ifndef __KVEVENTVIEWER_H
#define __KVEVENTVIEWER_H
 
#include "KVEvent.h"
#include "TGeoManager.h"
#include "TGeoMaterial.h"
#include "TGeoMedium.h"
#include "TGeoVolume.h"
#include "Riostream.h"
using namespace std;
 
class TBranch;
class TTree;
 
class KVEventViewer : public KVBase {
protected:
    Int_t fproton_color;
    Int_t fneutron_color;
    Int_t fProton_color;
    Int_t fNeutron_color;
    Double_t free_nucleon_radius;
    Double_t nucleon_radius;
    Double_t nuclear_radius_parameter;
 
   Double_t fMaxVelocity;
   Bool_t fFixSeed;
   Int_t fRefresh;
   Int_t fSeed;
   Bool_t fXYMode;
    Bool_t fMomentumSpace;
   Double_t fScaleFactor;
 
    Int_t ivol;
    Bool_t fSavePicture;
 
   TGeoMaterial* matEmptySpace;
   TGeoMaterial* matNuc;
   TGeoMaterial* matBox;
 
   TGeoMedium* EmptySpace;
   TGeoMedium* Nuc;
   TGeoMedium* Box;
   TGeoVolume* top;
   TGeoManager* geom;
 
   KVEvent* theEvent;
 
   ifstream eventFile;
   Int_t eventCounter;
   Bool_t textInput;
 
   TTree* theTree;
   Long64_t NtreeEntries;
 
    Double_t maxV;
    Int_t maxZ;
 
   Int_t fHighlightMode;
   Bool_t fAxesMode;
 
   Bool_t fFixPerspective;
 
public:
   enum EHighlightMode {
      kNoHighlight,
      kHighlightZmax,
      kHighlightParameter
   };
   KVEventViewer(Int_t protoncolor = kRed - 2, Int_t neutroncolor = kBlue - 2,
                 Int_t highlightprotoncolor = kOrange - 2,
                 Int_t highlightneutroncolor = kGreen - 2,
                 Double_t freenucleonradius = 0.2,
                 Double_t nucleonradius = 0.25,
                 Double_t nuclearradiusparameter = 0.2);
   virtual ~KVEventViewer();
 
   void SetProtonColor(Int_t c)
   {
      fproton_color = c;
   }
   void SetNeutronColor(Int_t c)
   {
      fneutron_color = c;
   }
   void SetHighlightProtonColor(Int_t c)
   {
      fProton_color = c;
   }
   void SetHighlightNeutronColor(Int_t c)
   {
      fNeutron_color = c;
   }
   void SetSavePicture(Bool_t yes = kTRUE)
   {
      fSavePicture = yes;
   }
 
   void SetFreeNucleonRadius(Double_t r)
   {
      free_nucleon_radius = r;
   }
   void SetNucleonRadius(Double_t r)
   {
      nucleon_radius = r;
   }
 
   void SetMaxVelocity(Int_t vel)
   {
      fMaxVelocity = vel;
   }
   void SetFixSeed(Bool_t fix, Int_t refresh = 1)
   {
      fFixSeed = fix;
      fRefresh = refresh;
   }
   void SetXYMode(Bool_t mode)
   {
      fXYMode = mode;
   }
   void SetDrawMomentumSpace(Bool_t on = kTRUE)
   {
      fMomentumSpace = on;
   }
 
   void DrawNucleus(KVNucleus&, const Char_t* frame = "");
   void DrawEvent(KVEvent*, const Char_t* frame = "");
   void DrawEvent(KVEvent*, const Char_t* frame = "") const;
 
   void SetInput(TBranch* eventbranch);
   void SetInput(const char* filename);
 
   void ReadEvent();
   void ReadTextEvent();
   void ReadTreeEvent();
 
   void DrawNextEvent();
 
   virtual Bool_t SetHighlight(KVNucleus* n);
   virtual void SetHighlightMode(Int_t m)
   {
      fHighlightMode = m;
   }
 
   void SetAxesMode(Bool_t on = kTRUE)
   {
      fAxesMode = on;
   }
 
   void DumpCurrentEvent()
   {
      if (theEvent) theEvent->Print("ok");
   }
 
   void SetFixPerspective(Bool_t on = kTRUE)
   {
 
      fFixPerspective = on;
   }
 
   ClassDef(KVEventViewer, 1) //Draw events in 3D using OpenGL
};
 
#endif