G4Molecule.cc

Go to the documentation of this file. Or view the newest version in sPHENIX GitHub for file G4Molecule.cc

// ********************************************************************
// * 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.          *
// ********************************************************************
//
//
// ---------------------------------------------------------------------
//  GEANT 4 class header file
//
//  History: first implementation, based on G4DynamicParticle
//           New dependency : G4VUserTrackInformation
//
//      ---------------- G4Molecule  ----------------
//      first design&implementation by Alfonso Mantero, 7 Apr 2009
//      New developments Alfonso Mantero & Mathieu Karamitros
//      Oct/Nov 2009 Class Name changed to G4Molecule
//                   Removed dependency from G4DynamicParticle
//                   New constructors :
//                    copy constructor
//                    direct ionized/excited molecule
//                   New methods :
//                    Get : name,atoms' number,nb electrons,decayChannel
//                    PrintState //To get the electronic level and the
//                                 corresponding name of the excitation
//                    Kinematic :
//                    BuildTrack,GetKineticEnergy,GetDiffusionVelocity
//                    Change the way dynCharge and eNb is calculated
// ---------------------------------------------------------------------

#include "G4Molecule.hh"
#include "G4MolecularConfiguration.hh"
#include "Randomize.hh"
#include "G4PhysicalConstants.hh"
#include "G4SystemOfUnits.hh"
#include "G4Track.hh"
#include "G4VMoleculeCounter.hh"

using namespace std;

ITImp(G4Molecule)

G4Allocator<G4Molecule>*& aMoleculeAllocator()
{
    G4ThreadLocalStatic G4Allocator<G4Molecule>* _instance = nullptr;
    return _instance;
}

//______________________________________________________________________________

template<>
G4KDNode<G4Molecule>::~G4KDNode() {
    fPoint->SetNode(nullptr);
}

//______________________________________________________________________________

G4Molecule* GetMolecule(const G4Track& track)
{
    return (G4Molecule*)(GetIT(track));
}

//______________________________________________________________________________

G4Molecule* GetMolecule(const G4Track* track)
{
    return (G4Molecule*)(GetIT(track));
}

//______________________________________________________________________________

G4Molecule* G4Molecule::GetMolecule(const G4Track* track)
{
    return (G4Molecule*)(GetIT(track));
}

//______________________________________________________________________________

void G4Molecule::Print() const
{
    G4cout << "The user track information is a molecule" << G4endl;
}

//______________________________________________________________________________

G4Molecule::G4Molecule(const G4Molecule& right)
    : G4VUserTrackInformation("G4Molecule")
    , G4IT(right)
{
    fpMolecularConfiguration = right.fpMolecularConfiguration;
}

//______________________________________________________________________________

G4Molecule& G4Molecule::operator=(const G4Molecule& right)
{
    if (&right == this) return *this;
    fpMolecularConfiguration = right.fpMolecularConfiguration;
    return *this;
}

//______________________________________________________________________________

G4bool G4Molecule::operator==(const G4Molecule& right) const
{
    if (fpMolecularConfiguration == right.fpMolecularConfiguration)
    {
        return true;
    }
    return false;
}

//______________________________________________________________________________

G4bool G4Molecule::operator!=(const G4Molecule& right) const
{
    return !(*this == right);
}

//______________________________________________________________________________
G4bool G4Molecule::operator<(const G4Molecule& right) const
{
    return fpMolecularConfiguration < right.fpMolecularConfiguration;
}

//______________________________________________________________________________

G4Molecule::G4Molecule()
    : G4VUserTrackInformation("G4Molecule")
    , G4IT()
{
    fpMolecularConfiguration = nullptr;
}

//______________________________________________________________________________

G4Molecule::~G4Molecule()
{
    if (fpTrack != nullptr)
    {
        if (G4VMoleculeCounter::Instance()->InUse())
        {
            G4VMoleculeCounter::Instance()->
                RemoveAMoleculeAtTime(fpMolecularConfiguration,
                                      fpTrack->GetGlobalTime(),
                                      &(fpTrack->GetPosition()));
        }
        fpTrack = nullptr;
    }
    fpMolecularConfiguration = nullptr;
}

//______________________________________________________________________________
G4Molecule::G4Molecule(G4MoleculeDefinition* pMoleculeDefinition)
    : G4VUserTrackInformation("G4Molecule")
    , G4IT()
{
    fpMolecularConfiguration = G4MolecularConfiguration::GetOrCreateMolecularConfiguration(pMoleculeDefinition);
}

//______________________________________________________________________________

G4Molecule::G4Molecule(G4MoleculeDefinition* pMoleculeDefinition, int charge)
{
    fpMolecularConfiguration = G4MolecularConfiguration::GetOrCreateMolecularConfiguration(pMoleculeDefinition,
                                                                                           charge);
}

//______________________________________________________________________________
G4Molecule::G4Molecule(G4MoleculeDefinition* pMoleculeDefinition,
                       G4int OrbitalToFree,
                       G4int OrbitalToFill)
   : G4VUserTrackInformation("G4Molecule")
   , G4IT()
{
    if (pMoleculeDefinition->GetGroundStateElectronOccupancy())
    {
        G4ElectronOccupancy dynElectronOccupancy(*pMoleculeDefinition->GetGroundStateElectronOccupancy());

        if (OrbitalToFill != 0)
        {
            dynElectronOccupancy.RemoveElectron(OrbitalToFree - 1, 1);
            dynElectronOccupancy.AddElectron(OrbitalToFill - 1, 1);
            // dynElectronOccupancy.DumpInfo(); // DEBUG
        }

        if (OrbitalToFill == 0)
        {
            dynElectronOccupancy.RemoveElectron(OrbitalToFree - 1, 1);
            // dynElectronOccupancy.DumpInfo(); // DEBUG
        }

        fpMolecularConfiguration =
            G4MolecularConfiguration::GetOrCreateMolecularConfiguration(
                pMoleculeDefinition, dynElectronOccupancy);
    }
    else
    {
        fpMolecularConfiguration = nullptr;
        G4Exception(
            "G4Molecule::G4Molecule(G4MoleculeDefinition* pMoleculeDefinition, "
            "G4int OrbitalToFree, G4int OrbitalToFill)",
            "G4Molecule_wrong_usage_of_constructor",
            FatalErrorInArgument,
            "If you want to use this constructor, the molecule definition has to be "
            "first defined with electron occupancies");
    }
}

//______________________________________________________________________________
G4Molecule::G4Molecule(G4MoleculeDefinition* pMoleculeDefinition,
                       G4int level,
                       G4bool excitation)
    : G4VUserTrackInformation("G4Molecule")
    , G4IT()
{
    if (pMoleculeDefinition->GetGroundStateElectronOccupancy())
    {
        G4ElectronOccupancy dynElectronOccupancy(*pMoleculeDefinition->GetGroundStateElectronOccupancy());

        if (excitation)
        {
            dynElectronOccupancy.RemoveElectron(level, 1);
            dynElectronOccupancy.AddElectron(5, 1);
            // dynElectronOccupancy.DumpInfo(); // DEBUG
        }
        else
        {
            dynElectronOccupancy.RemoveElectron(level, 1);
            // dynElectronOccupancy.DumpInfo(); // DEBUG
        }

        fpMolecularConfiguration = G4MolecularConfiguration::GetOrCreateMolecularConfiguration(pMoleculeDefinition,
                dynElectronOccupancy);
    }
    else
    {
        fpMolecularConfiguration = nullptr;
        G4Exception(
            "G4Molecule::G4Molecule(G4MoleculeDefinition* pMoleculeDefinition, "
            "G4int OrbitalToFree, G4int OrbitalToFill)",
            "G4Molecule_wrong_usage_of_constructor",
            FatalErrorInArgument,
            "If you want to use this constructor, the molecule definition has to be "
            "first defined with electron occupancies");
    }
}

//______________________________________________________________________________

G4Molecule::G4Molecule(const G4MolecularConfiguration* pMolecularConfiguration)
{
    fpMolecularConfiguration = pMolecularConfiguration;
}

//______________________________________________________________________________

void G4Molecule::SetElectronOccupancy(const G4ElectronOccupancy* pElectronOcc)
{
    fpMolecularConfiguration = 
        G4MolecularConfiguration::GetOrCreateMolecularConfiguration(fpMolecularConfiguration->GetDefinition(),
                                                                    *pElectronOcc);
}

//______________________________________________________________________________

void G4Molecule::ExciteMolecule(G4int excitationLevel)
{
    fpMolecularConfiguration = fpMolecularConfiguration->ExciteMolecule(excitationLevel);
}

//______________________________________________________________________________

void G4Molecule::IonizeMolecule(G4int ionizationLevel)
{
    fpMolecularConfiguration = fpMolecularConfiguration->IonizeMolecule(ionizationLevel);
}

//______________________________________________________________________________

void G4Molecule::AddElectron(G4int orbit, G4int number)
{
    fpMolecularConfiguration = fpMolecularConfiguration->AddElectron(orbit, number);
}

//______________________________________________________________________________

void G4Molecule::RemoveElectron(G4int orbit, G4int number)
{
    fpMolecularConfiguration =
        fpMolecularConfiguration->RemoveElectron(orbit, number);
}

//______________________________________________________________________________

void G4Molecule::MoveOneElectron(G4int orbitToFree, G4int orbitToFill)
{
    fpMolecularConfiguration =
        fpMolecularConfiguration->MoveOneElectron(orbitToFree, orbitToFill);
}

//______________________________________________________________________________

const G4String& G4Molecule::GetName() const
{
    return fpMolecularConfiguration->GetName();
}

//______________________________________________________________________________

const G4String& G4Molecule::GetFormatedName() const
{
    return fpMolecularConfiguration->GetFormatedName();
}

//______________________________________________________________________________

G4int G4Molecule::GetAtomsNumber() const
{
    return fpMolecularConfiguration->GetAtomsNumber();
}

//______________________________________________________________________________

G4double G4Molecule::GetNbElectrons() const
{
    return fpMolecularConfiguration->GetNbElectrons();
}

//______________________________________________________________________________

void G4Molecule::PrintState() const
{
    fpMolecularConfiguration->PrintState();
}

//______________________________________________________________________________

G4Track* G4Molecule::BuildTrack(G4double globalTime,
                                 const G4ThreeVector& position)
{
    if (fpTrack != nullptr)
    {
        G4Exception("G4Molecule::BuildTrack", "Molecule001", FatalErrorInArgument,
                    "A track was already assigned to this molecule");
    }

    // Kinetic Values
    // Set a random direction to the molecule
    G4double costheta = (2 * G4UniformRand() - 1);
    G4double theta = acos(costheta);
    G4double phi = 2 * pi * G4UniformRand();

    G4double xMomentum = cos(phi) * sin(theta);
    G4double yMomentum = sin(theta) * sin(phi);
    G4double zMomentum = costheta;

    G4ThreeVector MomentumDirection(xMomentum, yMomentum, zMomentum);
    G4double KineticEnergy = GetKineticEnergy();

    G4DynamicParticle* dynamicParticle = new G4DynamicParticle(
        fpMolecularConfiguration->GetDefinition(), MomentumDirection,
        KineticEnergy);

    if (G4VMoleculeCounter::InUse())
    {
        G4VMoleculeCounter::Instance()->
            AddAMoleculeAtTime(fpMolecularConfiguration,
                               globalTime,
                               &(fpTrack->GetPosition()));
    }

    //Set the Track
    fpTrack = new G4Track(dynamicParticle, globalTime, position);
    fpTrack->SetUserInformation(this);

    return fpTrack;
}

//______________________________________________________________________________

G4double G4Molecule::GetKineticEnergy() const
{
    // Ideal Gaz case
    double v = GetDiffusionVelocity();
    double E = (fpMolecularConfiguration->GetMass() / (c_squared)) * (v * v) / 2.;
    return E;
}

//______________________________________________________________________________

G4double G4Molecule::GetDiffusionVelocity() const
{
    double moleculeMass = fpMolecularConfiguration->GetMass() / (c_squared);

    // Different possibilities
    // Ideal Gaz case : Maxwell Boltzmann Distribution
    //    double sigma = k_Boltzmann * fgTemperature / mass;
    //    return G4RandGauss::shoot( 0, sigma );
    // Ideal Gaz case : mean velocity from equipartition theorem
    return sqrt(3 * k_Boltzmann *
                G4MolecularConfiguration::GetGlobalTemperature() / moleculeMass);
    // Using this approximation for liquid is wrong
    // However the brownian process avoid taking
    // care of energy consideration and plays only
    // with positions
}

//______________________________________________________________________________

// added - to be transformed in a "Decay method"
const vector<const G4MolecularDissociationChannel*>*
G4Molecule::GetDissociationChannels() const
{
    return fpMolecularConfiguration->GetDissociationChannels();
}

//______________________________________________________________________________

G4int G4Molecule::GetFakeParticleID() const
{
    return fpMolecularConfiguration->GetFakeParticleID();
}

//______________________________________________________________________________

G4int G4Molecule::GetMoleculeID() const
{
    return fpMolecularConfiguration->GetMoleculeID();
}

//______________________________________________________________________________

G4double G4Molecule::GetDecayTime() const
{
    return fpMolecularConfiguration->GetDecayTime();
}

//______________________________________________________________________________

G4double G4Molecule::GetVanDerVaalsRadius() const
{
    return fpMolecularConfiguration->GetVanDerVaalsRadius();
}

//______________________________________________________________________________

G4int G4Molecule::GetCharge() const
{
    return fpMolecularConfiguration->GetCharge();
}

//______________________________________________________________________________

G4double G4Molecule::GetMass() const
{
    return fpMolecularConfiguration->GetMass();
}

//______________________________________________________________________________

const G4ElectronOccupancy* G4Molecule::GetElectronOccupancy() const
{
    return fpMolecularConfiguration->GetElectronOccupancy();
}

//______________________________________________________________________________

const G4MoleculeDefinition* G4Molecule::GetDefinition() const
{
    return fpMolecularConfiguration->GetDefinition();
}

//______________________________________________________________________________

G4double G4Molecule::GetDiffusionCoefficient() const
{
    return fpMolecularConfiguration->GetDiffusionCoefficient();
}

//______________________________________________________________________________

G4double G4Molecule::GetDiffusionCoefficient(const G4Material* pMaterial,
                                             double temperature) const
{
    return fpMolecularConfiguration->GetDiffusionCoefficient(pMaterial,
                                                             temperature);
}

//______________________________________________________________________________

const G4MolecularConfiguration* G4Molecule::GetMolecularConfiguration() const
{
    return fpMolecularConfiguration;
}

//______________________________________________________________________________

const G4String& G4Molecule::GetLabel() const
{
    return fpMolecularConfiguration->GetLabel();
}

//______________________________________________________________________________

void G4Molecule::ChangeConfigurationToLabel(const G4String& label)
{
    // TODO check fpMolecularConfiguration already exists
    // and new one as well
    // TODO notify for stack change
    fpMolecularConfiguration = G4MolecularConfiguration::GetMolecularConfiguration(
        fpMolecularConfiguration->GetDefinition(), label);

    assert(fpMolecularConfiguration != nullptr);
}