source: trunk/source/geometry/solids/test/fred/src/GridParticleGunMessenger.cc@ 1330

//
// ********************************************************************
// * License and Disclaimer                                           *
// *                                                                  *
// * The  Geant4 software  is  copyright of the Copyright Holders  of *
// * the Geant4 Collaboration.  It is provided  under  the terms  and *
// * conditions of the Geant4 Software License,  included in the file *
// * LICENSE and available at  http://cern.ch/geant4/license .  These *
// * include a list of copyright holders.                             *
// *                                                                  *
// * Neither the authors of this software system, nor their employing *
// * institutes,nor the agencies providing financial support for this *
// * work  make  any representation or  warranty, express or implied, *
// * regarding  this  software system or assume any liability for its *
// * use.  Please see the license in the file  LICENSE  and URL above *
// * for the full disclaimer and the limitation of liability.         *
// *                                                                  *
// * This  code  implementation is the result of  the  scientific and *
// * technical work of the GEANT4 collaboration.                      *
// * By using,  copying,  modifying or  distributing the software (or *
// * any work based  on the software)  you  agree  to acknowledge its *
// * use  in  resulting  scientific  publications,  and indicate your *
// * acceptance of all terms of the Geant4 Software license.          *
// ********************************************************************
//
//
// GridParticleGunMessenger.cc
//
// Implemenation of GridParticleGun's messenger
//

#include "GridParticleGunMessenger.hh"
#include "GridParticleGun.hh"
#include "G4ThreeVector.hh"
#include "G4UIdirectory.hh"
#include "G4UIcmdWith3Vector.hh"
#include "G4UIcmdWith3VectorAndUnit.hh"
#include "G4UIcmdWithAnInteger.hh"
#include "G4ios.hh"

//
// Constructor
// 
GridParticleGunMessenger::GridParticleGunMessenger( GridParticleGun *ourGun )
{
        gun = ourGun;
        
        gunDirectory = new G4UIdirectory( "/gridGun/" );
        gunDirectory->SetGuidance( "Grid Particle Gun control commands." );
        
        directionCmd = new G4UIcmdWith3Vector( "/gridGun/direction", this );
        directionCmd->SetGuidance( "Set direction of particles." );
        directionCmd->SetParameterName( "X", "Y", "Z", true, true );
        
        originCmd = new G4UIcmdWith3VectorAndUnit( "/gridGun/origin", this );
        originCmd->SetGuidance( "Set origin of grid." );
        originCmd->SetParameterName( "X", "Y", "Z", true, true );
        originCmd->SetDefaultUnit( "m" );       

        grid1Cmd = new G4UIcmdWith3VectorAndUnit( "/gridGun/grid1", this );
        grid1Cmd->SetGuidance( "Set first axis of grid." );
        grid1Cmd->SetParameterName( "X", "Y", "Z", true, true );
        grid1Cmd->SetDefaultUnit( "m" );

        grid2Cmd = new G4UIcmdWith3VectorAndUnit( "/gridGun/grid2", this );
        grid2Cmd->SetGuidance( "Set second axis of grid." );
        grid2Cmd->SetParameterName( "X", "Y", "Z", true, true );
        grid2Cmd->SetDefaultUnit( "m" );

        n1Cmd = new G4UIcmdWithAnInteger( "/gridGun/n1", this );
        n1Cmd->SetGuidance( "Set number of grid points along first axis." );
        n1Cmd->SetParameterName( "n1", true, true );
        
        n2Cmd = new G4UIcmdWithAnInteger( "/gridGun/n2", this );
        n2Cmd->SetGuidance( "Set number of grid points along second axis." );
        n2Cmd->SetParameterName( "n2", true, true );
}

//
// Destructor
//
GridParticleGunMessenger::~GridParticleGunMessenger()
{
        delete directionCmd;
        delete originCmd;
        delete grid1Cmd;
        delete grid2Cmd;
        delete n1Cmd;
        delete n2Cmd;
        delete gunDirectory;
}

//
// SetNewValue
//
// Called by the UI when user requests a change
//
void GridParticleGunMessenger::SetNewValue( G4UIcommand *command, G4String newValues )
{
        if (command == directionCmd) {
                gun->SetDirection( directionCmd->GetNew3VectorValue( newValues ) );
        }
        else if (command == originCmd) {
                gun->SetOrigin( originCmd->GetNew3VectorValue( newValues ) );
        }
        else if (command == grid1Cmd) {
                gun->SetGrid1( grid1Cmd->GetNew3VectorValue( newValues ) );
        }
        else if (command == grid2Cmd) {
                gun->SetGrid2( grid2Cmd->GetNew3VectorValue( newValues ) );
        }
        else if (command == n1Cmd) {
                gun->SetN1( n1Cmd->GetNewIntValue( newValues ) );
        }
        else if (command == n2Cmd) {
                gun->SetN2( n2Cmd->GetNewIntValue( newValues ) );
        }
}

//
// GetCurrentValue
//
G4String GridParticleGunMessenger::GetCurrentValue( G4UIcommand *command )
{
        if (command == directionCmd) {
                return directionCmd->ConvertToString( gun->GetDirection() );
        }
        else if (command == originCmd) {
                return originCmd->ConvertToString( gun->GetOrigin(), "m" );
        }
        else if (command == grid1Cmd) {
                return grid1Cmd->ConvertToString( gun->GetGrid1(), "m" );
        }
        else if (command == grid2Cmd) {
                return grid2Cmd->ConvertToString( gun->GetGrid2(), "m" );
        }
        else if (command == n1Cmd) {
                return n1Cmd->ConvertToString( gun->GetN1() );
        }
        else if (command == n2Cmd) {
                return n2Cmd->ConvertToString( gun->GetN2() );
        }
        
        return "baloney";
}