#include <G4TheMTRayTracer.hh>

Inheritance diagram for G4TheMTRayTracer:

Public Member Functions
	G4TheMTRayTracer (G4VFigureFileMaker figMaker=0, G4VRTScanner scanner=0)

virtual	~G4TheMTRayTracer ()

virtual void	Trace (const G4String &fileName)

Public Member Functions inherited from G4TheRayTracer
	G4TheRayTracer (G4VFigureFileMaker figMaker=0, G4VRTScanner scanner=0)

virtual	~G4TheRayTracer ()

void	SetFigureFileMaker (G4VFigureFileMaker *figMaker)

G4VFigureFileMaker *	GetFigureFileMaker ()

void	SetScanner (G4VRTScanner *scanner)

G4VRTScanner *	GetScanner ()

void	SetNColumn (G4int val)

G4int	GetNColumn () const

void	SetNRow (G4int val)

G4int	GetNRow () const

void	SetEyePosition (const G4ThreeVector &val)

G4ThreeVector	GetEyePosition () const

void	SetTargetPosition (const G4ThreeVector &val)

G4ThreeVector	GetTargetPosition () const

void	SetLightDirection (const G4ThreeVector &val)

G4ThreeVector	GetLightDirection () const

void	SetUpVector (const G4ThreeVector &val)

G4ThreeVector	GetUpVector () const

void	SetHeadAngle (G4double val)

G4double	GetHeadAngle () const

void	SetViewSpan (G4double val)

G4double	GetViewSpan () const

void	SetAttenuationLength (G4double val)

G4double	GetAttenuationLength () const

void	SetDistortion (G4bool val)

G4bool	GetDistortion () const

void	SetBackgroundColour (const G4Colour &val)

G4Colour	GetBackgroundColour () const

Protected Member Functions
virtual G4bool	CreateBitMap ()

virtual void	StoreUserActions ()

virtual void	RestoreUserActions ()

Protected Member Functions inherited from G4TheRayTracer
void	CreateFigureFile (const G4String &fileName)

G4bool	GenerateColour (G4Event *anEvent)

G4Colour	GetSurfaceColour (G4RayTrajectoryPoint *point)

G4Colour	GetMixedColour (const G4Colour &surfCol, const G4Colour &transCol, G4double weight=0.5)

G4Colour	Attenuate (G4RayTrajectoryPoint *point, const G4Colour &sourceCol)

G4bool	ValidColour (const G4VisAttributes *visAtt)

Protected Attributes
const G4UserWorkerInitialization *	theUserWorkerInitialization

G4RTWorkerInitialization *	theRTWorkerInitialization

const G4UserRunAction *	theUserRunAction

G4RTRunAction *	theRTRunAction

Protected Attributes inherited from G4TheRayTracer
G4RayShooter *	theRayShooter

G4VFigureFileMaker *	theFigMaker

G4RTMessenger *	theMessenger

G4VRTScanner *	theScanner

G4EventManager *	theEventManager

G4UserEventAction *	theUserEventAction

G4UserStackingAction *	theUserStackingAction

G4UserTrackingAction *	theUserTrackingAction

G4UserSteppingAction *	theUserSteppingAction

G4UserEventAction *	theRayTracerEventAction

G4UserStackingAction *	theRayTracerStackingAction

G4RTTrackingAction *	theRayTracerTrackingAction

G4RTSteppingAction *	theRayTracerSteppingAction

unsigned char *	colorR

unsigned char *	colorG

unsigned char *	colorB

G4int	nColumn

G4int	nRow

G4ThreeVector	eyePosition

G4ThreeVector	targetPosition

G4ThreeVector	eyeDirection

G4ThreeVector	lightDirection

G4ThreeVector	up

G4double	headAngle

G4double	viewSpan

G4double	attenuationLength

G4bool	distortionOn

G4bool	antialiasingOn

G4Colour	rayColour

G4Colour	backgroundColour

Friends
class	G4RTPrimaryGeneratorAction

class	G4RTRun

class	G4RayTracerViewer

Detailed Description

Definition at line 71 of file G4TheMTRayTracer.hh.

Constructor & Destructor Documentation

G4TheMTRayTracer::G4TheMTRayTracer	(	G4VFigureFileMaker *	figMaker = `0`,
		G4VRTScanner *	scanner = `0`
	)

Definition at line 51 of file G4TheMTRayTracer.cc.

References FatalException, G4Exception(), theRTRunAction, theRTWorkerInitialization, theUserRunAction, and theUserWorkerInitialization.

 : G4TheRayTracer(figMaker,scanner)
 {
   if(!theInstance) 
   { theInstance = this; }
   else
   { G4Exception("G4TheMTRayTracer::G4TheMTRayTracer","VisRayTracer00100",
          FatalException,"G4TheMTRayTracer has to be a singleton.");}
   theUserWorkerInitialization = 0;
   theRTWorkerInitialization = 0;
   theUserRunAction = 0;
   theRTRunAction = 0;
 }

G4TheMTRayTracer::~G4TheMTRayTracer ( )

virtual

Definition at line 66 of file G4TheMTRayTracer.cc.

References theRTRunAction, and theRTWorkerInitialization.

 {
   if(theRTWorkerInitialization)
   {
     delete theRTWorkerInitialization;
     theRTWorkerInitialization = 0;
   }
   if(theRTRunAction)
   {
     delete theRTRunAction;
     theRTRunAction = 0;
   }
 }

Member Function Documentation

G4bool G4TheMTRayTracer::CreateBitMap ( )

protectedvirtual

Reimplemented from G4TheRayTracer.

Definition at line 152 of file G4TheMTRayTracer.cc.

References G4UImanager::ApplyCommand(), G4TheRayTracer::colorB, G4TheRayTracer::colorG, G4TheRayTracer::colorR, G4UIcommand::ConvertToString(), G4VRTScanner::Coords(), G4VRTScanner::Draw(), G4Colour::GetBlue(), G4VVisManager::GetConcreteInstance(), G4RunManager::GetCurrentRun(), G4Colour::GetGreen(), G4RTRun::GetMap(), G4THitsMap< T >::GetMap(), G4MTRunManager::GetMasterRunManager(), G4Colour::GetRed(), G4UImanager::GetUIpointer(), G4VVisManager::IgnoreStateChanges(), G4VRTScanner::Initialize(), G4TheRayTracer::nColumn, G4TheRayTracer::nRow, RestoreUserActions(), StoreUserActions(), and G4TheRayTracer::theScanner.

Referenced by Trace().

 {
   G4VVisManager* visMan = G4VVisManager::GetConcreteInstance();
   visMan->IgnoreStateChanges(true);
   StoreUserActions();
 
 // Event loop
   G4MTRunManager* mrm = G4MTRunManager::GetMasterRunManager();
   G4int nEvent = nRow*nColumn;
 ////  mrm->BeamOn(nEvent);
 ////  Temporary work-around until direct invokation of G4RunManager::BeamOn() works.
   G4String str = "/run/beamOn " + G4UIcommand::ConvertToString(nEvent);
   G4UImanager::GetUIpointer()->ApplyCommand(str);
 
   RestoreUserActions();
   visMan->IgnoreStateChanges(false);
 
   const G4RTRun* theRun = static_cast<const G4RTRun*>(mrm->GetCurrentRun());
   if(!theRun) return false;
 
   G4THitsMap<G4Colour>* colMap = theRun->GetMap(); 
   std::map<G4int,G4Colour*>::iterator itr = colMap->GetMap()->begin();
   for(;itr!=colMap->GetMap()->end();itr++)
   {
     G4int key = itr->first;
     G4Colour* col = itr->second;
     colorR[key] = (unsigned char)(int(255*col->GetRed()));
     colorG[key] = (unsigned char)(int(255*col->GetGreen()));
     colorB[key] = (unsigned char)(int(255*col->GetBlue()));
   }  
 
   theScanner->Initialize(nRow,nColumn);
   G4int iRow, iColumn;
   while (theScanner->Coords(iRow,iColumn))
   {
     G4int iCoord = iRow * nColumn + iColumn;
     theScanner->Draw(colorR[iCoord],colorG[iCoord],colorB[iCoord]);
   }
 
   return true;
 }

void G4TheMTRayTracer::RestoreUserActions ( )

protectedvirtual

Reimplemented from G4TheRayTracer.

Definition at line 145 of file G4TheMTRayTracer.cc.

References G4MTRunManager::GetMasterRunManager(), G4MTRunManager::SetUserAction(), G4MTRunManager::SetUserInitialization(), theUserRunAction, and theUserWorkerInitialization.

Referenced by CreateBitMap().

 {
   G4MTRunManager* mrm = G4MTRunManager::GetMasterRunManager();
   mrm->SetUserInitialization(const_cast<G4UserWorkerInitialization*>(theUserWorkerInitialization));
   mrm->SetUserAction(const_cast<G4UserRunAction*>(theUserRunAction));
 }

void G4TheMTRayTracer::StoreUserActions ( )

protectedvirtual

Reimplemented from G4TheRayTracer.

Definition at line 132 of file G4TheMTRayTracer.cc.

References G4MTRunManager::GetMasterRunManager(), G4RunManager::GetUserRunAction(), G4RunManager::GetUserWorkerInitialization(), G4MTRunManager::SetUserAction(), G4MTRunManager::SetUserInitialization(), theRTRunAction, theRTWorkerInitialization, theUserRunAction, and theUserWorkerInitialization.

Referenced by CreateBitMap().

 { 
   G4MTRunManager* mrm = G4MTRunManager::GetMasterRunManager();
   theUserWorkerInitialization = mrm->GetUserWorkerInitialization();
   theUserRunAction = mrm->GetUserRunAction();
 
   if(!theRTWorkerInitialization) theRTWorkerInitialization = new G4RTWorkerInitialization();
   if(!theRTRunAction) theRTRunAction = new G4RTRunAction();
 
   mrm->SetUserInitialization(theRTWorkerInitialization);
   mrm->SetUserAction(theRTRunAction);
 }

void G4TheMTRayTracer::Trace ( const G4String & fileName )

virtual

Reimplemented from G4TheRayTracer.

Definition at line 80 of file G4TheMTRayTracer.cc.

References G4UImanager::ApplyCommand(), G4TheRayTracer::backgroundColour, G4TheRayTracer::colorB, G4TheRayTracer::colorG, G4TheRayTracer::colorR, G4UIcommand::ConvertToString(), CreateBitMap(), G4TheRayTracer::CreateFigureFile(), G4TheRayTracer::eyeDirection, G4TheRayTracer::eyePosition, G4cerr, G4endl, G4State_Idle, G4Colour::GetBlue(), G4UImanager::GetCurrentIntValue(), G4StateManager::GetCurrentState(), G4Colour::GetGreen(), G4Colour::GetRed(), G4StateManager::GetStateManager(), G4StateManager::GetStateString(), G4UImanager::GetUIpointer(), G4TheRayTracer::nColumn, G4TheRayTracer::nRow, G4TheRayTracer::targetPosition, G4TheRayTracer::theFigMaker, and CLHEP::Hep3Vector::unit().

 {
   G4StateManager* theStateMan = G4StateManager::GetStateManager();
   G4ApplicationState currentState = theStateMan->GetCurrentState();
   if(currentState!=G4State_Idle)
   {
     G4cerr << "Illegal application state <" << theStateMan->GetStateString(currentState)
            << "> - Trace() ignored. " << G4endl;
     return;
   }
 
   if(!theFigMaker)
   {
     G4cerr << "Figure file maker class is not specified - Trace() ignored." << G4endl;
     return;
   }
 
   G4UImanager* UI = G4UImanager::GetUIpointer();
   G4int storeTrajectory = UI->GetCurrentIntValue("/tracking/storeTrajectory");
   UI->ApplyCommand("/tracking/storeTrajectory 1");
 
   G4ThreeVector tmpVec = targetPosition - eyePosition;
   eyeDirection = tmpVec.unit();
   G4int nPixel = nColumn*nRow;
   colorR = new unsigned char[nPixel];
   colorG = new unsigned char[nPixel];
   colorB = new unsigned char[nPixel];
   unsigned char defR = (unsigned char)(int(255*backgroundColour.GetRed()));
   unsigned char defG = (unsigned char)(int(255*backgroundColour.GetGreen()));
   unsigned char defB = (unsigned char)(int(255*backgroundColour.GetBlue()));
   for(G4int ii=0;ii<nPixel;ii++)
   {
     colorR[ii] = defR;
     colorG[ii] = defG;
     colorB[ii] = defB;
   }
 
   G4bool succeeded = CreateBitMap();
   if(succeeded)
   { CreateFigureFile(fileName); }
   else
   { G4cerr << "Could not create figure file" << G4endl;
     G4cerr << "You might set the eye position outside of the world volume" << G4endl; }
 
   G4String str = "/tracking/storeTrajectory " + G4UIcommand::ConvertToString(storeTrajectory);
   UI->ApplyCommand(str);
 
   delete [] colorR;
   delete [] colorG;
   delete [] colorB;
 }