//
// ********************************************************************
// * 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.          *
// ********************************************************************
//
// $Id: NTSTPrimaryGeneratorMessenger.cc,v 1.3 2006/06/29 18:26:29 gunter Exp $
// 

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....
//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....

#include "NTSTPrimaryGeneratorMessenger.hh"
#include "G4UIcmdWithAString.hh"
#include "G4UIcmdWithABool.hh"
#include "G4UIdirectory.hh"

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....

NTSTPrimaryGeneratorMessenger::NTSTPrimaryGeneratorMessenger()
  : ChooseCmd(0), PrintCmd(0), generator(0), print(0)
{ // set the available commands
  GenDir = new G4UIdirectory("/gen/");
  GenDir->SetGuidance("NTST generator control");
  ChooseCmd = new G4UIcmdWithAString("/gen/choose",this);
  ChooseCmd->SetGuidance("Choose the generator to use.");
  ChooseCmd->SetDefaultValue("gun");
  G4String Names("gun evt");
  //  ChooseCmd->SetCandidates(Names);
  //  ChooseCmd->AvailableForStates(PreInit,Idle);
  SetNewValue(ChooseCmd, "gun");

  // (en,dis)able printing
  PrintCmd = new G4UIcmdWithABool("/gen/print", this);
  G4cout << "### Instantiated /gen/print cmd" << G4endl;
  PrintCmd->SetGuidance("print events (true) or not (false).");
  G4cout << "### SetGuidance for /gen/print" << G4endl;
  PrintCmd->SetDefaultValue(false);
  G4cout << "### Set default value for /gen/print cmd" << G4endl;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....

NTSTPrimaryGeneratorMessenger::~NTSTPrimaryGeneratorMessenger()
{
  delete ChooseCmd;
  delete PrintCmd;
  delete generator;
}

//....oooOO0OOooo........oooOO0OOooo........oooOO0OOooo........oooOO0OOooo....

#include "NTSTGunGenerator.hh"
#include "NTSTBabarEvtReadGenerator.hh"

void 
NTSTPrimaryGeneratorMessenger::
SetNewValue(G4UIcommand* command, G4String newValue)
{ 
  if( command == ChooseCmd ) {
    if (newValue == "gun") {
      Choice = GUN;
      delete generator;
      generator = new NTSTGunGenerator();
      G4cout << "### instantiated " << newValue << " generator" << G4endl;
      Name = newValue;
    } else if (newValue == "evt") {
      Choice = EVT;
      delete generator;
      generator = new NTSTBabarEvtReadGenerator("babarevt.out");
      G4cout << "### instantiated " << newValue << " generator" << G4endl;
      Name = newValue;
    } else {
      G4cerr << "!!! Illegal choice of generator: \"" << newValue 
	     << "\"" << G4endl;
      G4cerr << "!!! Valid choices are " << GetNames() << G4endl;
      G4cerr << "!!! Old generator: \"" << GetName() << "\" retained" 
	     << G4endl;
    }
  } else if ( command == PrintCmd ) {
    if (newValue == "0" || newValue == "false") DisablePrinting();
    else if (newValue == "1" || newValue == "true") EnablePrinting();
    else G4cerr << "PrintCmd \"" << newValue << "\" not understood" << G4endl;
  }
}