New code from Piergiorgio added

[u/mrichter/AliRoot.git] / AliGeant4 / AliGunParticle.cxx

// $Id$
// Category: event
//
// See the class description in the header file.

#include "AliGunParticle.h"

#include <G4ios.hh>

AliGunParticle::AliGunParticle()
  : fParticleDefinition(0),
    fParticleMomentum(G4ThreeVector(0., 0., 1.)),
    fPosition(G4ThreeVector()),
    fTime(0),
    fPolarization(G4ThreeVector())
{
//
}

AliGunParticle::AliGunParticle(G4ParticleDefinition* particleDef,
                   G4ParticleMomentum momentum, G4ThreeVector position, 
                   G4double time, G4ThreeVector polarization )
  : fParticleDefinition(particleDef),
    fParticleMomentum(momentum),
    fPosition(position),
    fTime(time),
    fPolarization(polarization)
{
//
}

AliGunParticle::AliGunParticle( G4ParticleDefinition* particleDef, 
                   G4ParticleMomentum momentumDir, G4double kinEnergy,
		   G4ThreeVector position, G4double time, 
		   G4ThreeVector polarization )
  : fParticleDefinition(particleDef),
    fPosition(position),
    fTime(time),
    fPolarization(polarization)
{
//   
  G4double mass = fParticleDefinition->GetPDGMass();
  G4double energy = kinEnergy + mass;
  G4double pmag = sqrt(energy*energy - mass*mass);
  fParticleMomentum = pmag*momentumDir;
}

AliGunParticle::AliGunParticle(const AliGunParticle& right) {
//
  fParticleDefinition = right.fParticleDefinition;
  fParticleMomentum = right.fParticleMomentum;
  fPosition = right.fPosition; 
  fTime = right.fTime;
  fPolarization = right.fPolarization; 
}

AliGunParticle::~AliGunParticle() {
//
}

// operators

AliGunParticle& AliGunParticle::operator=(const AliGunParticle& right)
{    
  // check assignement to self
  if (this == &right) return *this;

  fParticleDefinition = right.fParticleDefinition;
  fParticleMomentum = right.fParticleMomentum;
  fPosition = right.fPosition; 
  fTime = right.fTime;
  fPolarization = right.fPolarization; 
  
  return *this;
}

G4int AliGunParticle::operator==(const AliGunParticle& right) const 
{
//    
  G4int returnValue = 0;
  if (fParticleDefinition == right.fParticleDefinition 
      && fParticleMomentum == right.fParticleMomentum 
      && fPosition == right.fPosition 
      && fTime == right.fTime 
      && fPolarization == right.fPolarization)  returnValue = 1; 

  return returnValue;  
}

G4int AliGunParticle::operator!=(const AliGunParticle& right) const
{
//    
  G4int returnValue = 1;
  if (*this == right) returnValue = 0; 
  
  return returnValue;
}

// public methods

void AliGunParticle::Print() const
{
// Prints particle properties.
// ---

  G4cout << "    particle name: ";
  G4cout << fParticleDefinition->GetParticleName() << G4endl;
  G4cout << "    particle momentum: ";
  G4cout << fParticleMomentum << G4endl;
  G4cout << "    particle position: ";
  G4cout << fPosition << " mm" << G4endl;
  G4cout << "    particle time: ";
  G4cout << fTime     << " s" << G4endl;
  G4cout << "    particle polarization: ";
  G4cout << fPolarization << G4endl;
}  

void AliGunParticle::SetKineticEnergy(G4double kinEnergy)
{
// Sets kinetic energy.
// ---

  G4double mass = fParticleDefinition->GetPDGMass();
  G4double energy = kinEnergy + mass;
  G4double pmag = sqrt(energy*energy - mass*mass);
  fParticleMomentum.setMag(pmag); 
}

G4double AliGunParticle::GetKineticEnergy() const
{
// Gets kinetic energy.
// ---

  G4double mass = fParticleDefinition->GetPDGMass();
  G4double energy 
   = sqrt(fParticleMomentum.mag()*fParticleMomentum.mag() + mass*mass);
  return energy - mass;
}