BDSMultilayerScreen.hh

/* 
Beam Delivery Simulation (BDSIM) Copyright (C) Royal Holloway, 
University of London 2001 - 2023.
 
This file is part of BDSIM.
 
BDSIM is free software: you can redistribute it and/or modify 
it under the terms of the GNU General Public License as published 
by the Free Software Foundation version 3 of the License.
 
BDSIM is distributed in the hope that it will be useful, but 
WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.
 
You should have received a copy of the GNU General Public License
along with BDSIM.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef BDSMULTILAYERSCREEN_H
#define BDSMULTILAYERSCREEN_H 
 
#include <map>
#include <vector>
 
#include "globals.hh"
#include "G4RotationMatrix.hh"
#include "G4TwoVector.hh"
#include "G4ThreeVector.hh"
 
class G4LogicalVolume;
class G4PVPlacement;
class G4VSolid;
 
class BDSColourWheel;
class BDSScreenLayer;
 
class BDSMultilayerScreen 
{
public:
  BDSMultilayerScreen(const G4TwoVector& xysizeIn,
                      const G4String&    nameIn);
  virtual ~BDSMultilayerScreen();
 
  void AddScreenLayer(G4double thickness,
                      G4String material,
                      G4String name,
                      G4int    isSampler = 0,
                      G4double grooveWidth = 0,
                      G4double grooveSpatialFrequency = 0);
  void AddScreenLayer(BDSScreenLayer* layer, G4int isSampler=0);
 
  inline const G4ThreeVector& GetSize() const {return size;}
  inline BDSScreenLayer* ScreenLayer(G4int layer) {return screenLayers[layer];}
  BDSScreenLayer* ScreenLayer(const G4String& layerName);
  BDSScreenLayer* LastLayer() const {return screenLayers.back();}
  inline G4int NLayers() const {return (G4int)screenLayers.size();}
 
  inline void SetPhys(G4PVPlacement* physIn) {phys = physIn;}
 
  void Build();
 
  virtual void Place(G4RotationMatrix* rot,
                     const G4ThreeVector& pos,
                     G4LogicalVolume* motherVol);
 
  void ReflectiveSurface(G4int layer1, G4int layer2);
  void RoughSurface(G4int layer1, G4int layer2);
 
private:
  BDSMultilayerScreen() = delete;
  BDSMultilayerScreen(const BDSMultilayerScreen&) = delete;
  BDSMultilayerScreen& operator=(const BDSMultilayerScreen&) = delete;
  void ComputeDimensions();
 
  void BuildMotherVolume();
 
  void PlaceLayers();
  
  G4TwoVector   xysize; 
  G4String      name;   
  G4ThreeVector size;   
 
  G4LogicalVolume* log;
  G4PVPlacement*   phys;
  G4VSolid*        solid;
  std::vector<BDSScreenLayer*>        screenLayers;     
  std::map<G4String, BDSScreenLayer*> screenLayerNames; 
  std::vector<G4double> screenLayerZPos;
 
  BDSColourWheel* colourWheel;
};
 
#endif