G4INCLStore.hh

Go to the documentation of this file.

//
// ********************************************************************
// * 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.          *
// ********************************************************************
//
// INCL++ intra-nuclear cascade model
// Pekka Kaitaniemi, CEA and Helsinki Institute of Physics
// Davide Mancusi, CEA
// Alain Boudard, CEA
// Sylvie Leray, CEA
// Joseph Cugnon, University of Liege
//
#define INCLXX_IN_GEANT4_MODE 1

#include "globals.hh"

#ifndef G4INCLParticleStore_hh
#define G4INCLParticleStore_hh 1

#include <map>
#include <list>
#include <vector>
#include <string>
#include <algorithm>

#include "G4INCLParticle.hh"
#include "G4INCLIAvatar.hh"
#include "G4INCLBook.hh"
#include "G4INCLConfig.hh"

#ifdef INCLXX_IN_GEANT4_MODE
#define INCL_AVATAR_SEARCH_MinElement 1
#endif // INCLXX_IN_GEANT4_MODE

namespace G4INCL {

  class Store {
  public:
    Store(Config const * const config);

    ~Store();

    void add(Particle *p);

    void addParticleEntryAvatar(IAvatar *a);

    void addParticleEntryAvatars(IAvatarList const &al);

    void add(IAvatar *a);

    IAvatarList const &getAvatars() const {
      return avatarList;
    }

    void addIncomingParticle(Particle * const p);

    void removeFromIncoming(Particle * const p) { incoming.remove(p); }

    inline void clearIncoming() {
      incoming.clear();
    }

    inline void deleteIncoming() {
      for(ParticleIter iter = incoming.begin(); iter != incoming.end(); ++iter) {
        delete (*iter);
      }
      clearIncoming();
    }

    void particleHasBeenUpdated(long);

    IAvatar* findSmallestTime();

    void timeStep(G4double step);
    
    void particleHasBeenEjected(long);

    void addToOutgoing(Particle *p) { outgoing.push_back(p); }

    void addToOutgoing(ParticleList const &pl) {
      for(ParticleIter p=pl.begin(); p!=pl.end(); ++p)
        addToOutgoing(*p);
    }

    void particleHasBeenDestroyed(long);

    void particleHasEntered(Particle * const particle);

    ParticleList const & getIncomingParticles() const { return incoming; }

    ParticleList const & getOutgoingParticles() const { return outgoing; }

    ParticleList extractDynamicalSpectators() {
      ParticleList spectators;
      std::list<std::list<Particle*>::iterator> toBeErased;
      for(std::list<Particle*>::iterator p=outgoing.begin(); p!=outgoing.end(); ++p) {
        if((*p)->isProjectileSpectator()) {
// assert((*p)->isNucleon());
          spectators.push_back(*p); // add them to the list we will return
          toBeErased.push_back(p); // we will remove them from outgoing later
        }
      }

      // Now erase them from outgoing
      for(std::list<std::list<Particle*>::iterator>::iterator i=toBeErased.begin();
          i!=toBeErased.end(); ++i) {
        outgoing.erase(*i);
      }

      return spectators;
    }

    ParticleList const & getParticles() const { return inside; }

    Book* getBook() {return theBook; };

    G4int countCascading() {
      G4int n=0;
      for(ParticleIter i=inside.begin(); i!=inside.end(); ++i) {
        if(!(*i)->isTargetSpectator())
          ++n;
      }
      return n;
    }

    Config const * getConfig() { return theConfig; };

    ParticleList getParticipants();

    ParticleList getSpectators();

    void clear();

    void clearOutgoing();

    void clearAvatars();

    void initialiseParticleAvatarConnections();

    void loadParticles(std::string filename);

    G4int getLoadedA() { return loadedA; };

    G4int getLoadedZ() { return loadedZ; };

    G4double getLoadedStoppingTime() { return loadedStoppingTime; };

    std::string printParticleConfiguration();

    void writeParticles(std::string filename);

    std::string printAvatars();

    G4bool containsCollisions() const;

#if defined(INCL_AVATAR_SEARCH_FullSort) || defined(INCL_AVATAR_SEARCH_MinElement)

    static G4bool avatarComparisonPredicate(IAvatar *lhs, IAvatar *rhs) {
      return (lhs->getTime() < rhs->getTime());
    }
#elif defined(INCL_AVATAR_SEARCH_INCLSort)

    std::list<IAvatarIter>::iterator binaryIterSearch(IAvatar const * const avatar);
#endif

  private:
    Store(const Store &rhs);

    Store &operator=(Store const &rhs);

    void removeAvatarsFromParticle(long ID);

    G4bool avatarInConnectionMap(long);


    void connectParticleAndAvatar(long particleID, long avatarID);

    void removeAvatarFromParticle(long particleID, long avatarID);
    void removeAvatarByID(long ID);

  private:
    std::map<long, Particle*> particles;

    std::map<long, IAvatar*> avatars;

    std::map<long, std::vector<long>* > particleAvatarConnections;

    IAvatarList avatarList;

    ParticleList incoming;

    ParticleList inside;

    ParticleList outgoing;

    ParticleList geomSpectators;

    G4double currentTime;

    Book *theBook;

    G4int loadedA;

    G4int loadedZ;

    G4double loadedStoppingTime;

    Config const * theConfig;

#ifdef INCL_AVATAR_SEARCH_INCLSort

    std::list<IAvatarIter> avatarIterList;
#endif

  };
}
#endif