TG4MediumMap.h

Go to the documentation of this file.

#ifndef TG4_MEDIUM_MAP_H
#define TG4_MEDIUM_MAP_H
 
//------------------------------------------------
// The Geant4 Virtual Monte Carlo package
// Copyright (C) 2007 - 2014 Ivana Hrivnacova
// All rights reserved.
//
// For the licensing terms see geant4_vmc/LICENSE.
// Contact: root-vmc@cern.ch
//-------------------------------------------------

 
#include "globals.hh"
 
#include <map>
#include <vector>
 
class TG4Medium;
 
class G4LogicalVolume;
class G4Material;

 
class TG4MediumMap
{
 public:
  TG4MediumMap();
  ~TG4MediumMap();
 
  // methods
  TG4Medium* AddMedium(G4int mediumID, G4bool warn = true);
  void MapMedium(G4LogicalVolume* lv, G4int mediumID);
  void MapMedium(const G4String& lvName, G4int mediumID);
  void Print() const;
 
  // get methods
  G4int GetNofMedia() const;
  TG4Medium* GetMedium(G4int mediumID, G4bool warn = true) const;
  TG4Medium* GetMedium(const G4String& name, G4bool warn = true) const;
  TG4Medium* GetMedium(G4LogicalVolume* lv, G4bool warn = true) const;
  TG4Medium* GetMedium(const G4Material* material, G4bool warn = true) const;
  void GetMedia(const G4String& namePattern, std::vector<TG4Medium*>& media,
    G4bool warn = true) const;
 
 private:
  TG4MediumMap(const TG4MediumMap& right);
  TG4MediumMap& operator=(const TG4MediumMap& right);
 
  //
  // data members

  std::map<G4int, TG4Medium*> fIdMap;

  std::map<G4LogicalVolume*, TG4Medium*> fLVMap;

  std::map<const G4Material*, TG4Medium*> fMaterialMap;
};
 
#endif // TG4_MEDIUM_MAP_H