BDSScreenLayer.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 BDSSCREENLAYER_H
#define BDSSCREENLAYER_H 
 
#include "globals.hh"
#include "G4Colour.hh"
#include "G4ThreeVector.hh"
 
class G4LogicalVolume;
class G4PVPlacement;
class G4VSolid;
 
class BDSScreenLayer 
{
public:
  BDSScreenLayer(G4ThreeVector size,  // full widths in each dimension
                 G4String name,
                 G4String material,
                 G4double grooveWidth,
                 G4double grooveSpatialFrequency);
  virtual ~BDSScreenLayer(){;}
  inline G4LogicalVolume* GetLog()  const {return log;}
  inline G4String         GetName() const {return name;}
  inline G4ThreeVector    GetSize() const {return size;}
  inline G4PVPlacement*   GetPhys() const {return phys;}
  void SetPhys(G4PVPlacement* phys);
  void SetColour(G4Colour col);
  /*
  void backInternalMirror();
  void frontInternalMirror();
  */
  void AssignSampler();
 
  //Get the AssignSampler ID.
  G4int GetSamplerID() const {return samplerID;}
 
protected:
  BDSScreenLayer();
  G4ThreeVector size;
  G4String name;
  G4LogicalVolume* log = nullptr;
  G4PVPlacement* phys  = nullptr;
  G4VSolid* solid = nullptr;
 
private:
  /*
  class InternalMirror
  {
  public:
    InternalMirror(G4int            varside,
                   G4ThreeVector    size,
                   G4String         material,
                   G4LogicalVolume* motherLog,
                   G4PVPlacement*   motherPhys);
    ~InternalMirror();
    void geom();
    void compute();
    void place();
    void optical();
    enum sides{BACK, FRONT};
 
  private:
    InternalMirror();
    G4int side;
    G4VSolid* solid;
    G4LogicalVolume* log;
    G4PVPlacement* phys;
    G4ThreeVector motherSize;
    G4String motherMaterial;
    G4LogicalVolume* motherLog;
    G4PVPlacement* motherPhys;
    G4double thickness;
    G4double pos;
  };
  
  InternalMirror* internalMirror = nullptr;
  */
  virtual void Build();
  void BuildGroove();
  virtual void BuildScreen();
  void SetVisAttributes();
  void CutGroove(G4double xPos);
  void CutGrooves();
  G4String material;
  G4String logName;
  G4String solidName;
  // Geometrical objects:
  G4LogicalVolume* grooveLog = nullptr;
  G4VSolid* grooveSolid  = nullptr;
  G4double grooveWidth = 0.0;
  G4double grooveSpatialFrequency = 0.0;
  G4int    nGrooves = 0;
  G4Colour colour;
  G4int    samplerID = 0;
};
 
#endif