BDSBeamPipeFactoryPoints.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 BDSBEAMPIPEFACTORYPOINTS_H
#define BDSBEAMPIPEFACTORYPOINTS_H
 
#include "BDSBeamPipeFactoryBase.hh"
#include "BDSBeamPipe.hh"
 
#include "globals.hh"
#include "G4TwoVector.hh"
 
#include <vector>
 
class G4Material;
 
class BDSBeamPipeFactoryPoints: public BDSBeamPipeFactoryBase
{
public:
  BDSBeamPipeFactoryPoints();
  virtual ~BDSBeamPipeFactoryPoints();
 
  virtual BDSBeamPipe* CreateBeamPipe(const G4String& nameIn,
                                      G4double    lengthIn,
                                      G4double    aper1In,
                                      G4double    aper2In,
                                      G4double    aper3In,
                                      G4double    aper4In,
                                      G4Material* vacuumMaterialIn,
                                      G4double    beamPipeThicknessIn,
                                      G4Material* beamPipeMaterialIn,
                                      const G4String& pointsFileIn = "",
                                      const G4String& pointsUnitIn = "");
 
  virtual BDSBeamPipe* CreateBeamPipe(const G4String&      nameIn,
                                      G4double             lengthIn,
                                      const G4ThreeVector& inputFaceNormalIn,
                                      const G4ThreeVector& outputFaceNormalIn,
                                      G4double      aper1In,
                                      G4double      aper2In,
                                      G4double      aper3In,
                                      G4double      aper4In,
                                      G4Material*   vacuumMaterialIn,
                                      G4double      beamPipeThicknessIn,
                                      G4Material*   beamPipeMaterialIn,
                                      const G4String& pointsFileIn = "",
                                      const G4String& pointsUnitIn = "");
  
protected:
  virtual void GeneratePoints(G4double aper1,
                              G4double aper2,
                              G4double aper3,
                              G4double aper4,
                              G4double beamPipeThickness,
                              G4int    pointsPerTwoPi = 40) = 0;
 
  virtual G4double CalculateIntersectionRadius(G4double aper1,
                                               G4double aper2,
                                               G4double aper3,
                                               G4double aper4,
                                               G4double beamPipeThickness) = 0;
 
  virtual void CleanUp();
 
  void AppendPoint(std::vector<G4TwoVector>& vec,
                   G4double x,
                   G4double y);
 
  void AppendAngle(std::vector<G4TwoVector>& vec,
                   G4double startAngle,
                   G4double finishAngle,
                   G4double radius,
                   G4int    nPoints = 10,
                   G4double xOffset = 0,
                   G4double yOffset = 0);
 
  void AppendAngleEllipse(std::vector<G4TwoVector>& vec,
                          G4double startAngle,
                          G4double finishAngle,
                          G4double radiusA,   // radius in horizontal
                          G4double radiusB,   // radius in vertical
                          G4int    nPoints = 10,
                          G4double xOffset = 0,
                          G4double yOffset = 0);
 
  std::vector<G4TwoVector> vacuumEdge;
 
  std::vector<G4TwoVector> beamPipeInnerEdge;
 
  std::vector<G4TwoVector> beamPipeOuterEdge;
 
  std::vector<G4TwoVector> containerEdge;
 
  std::vector<G4TwoVector> containerSubtractionEdge;
  
  G4double extentX;
  G4double extentY;
  
  G4String pointsFile;
  G4String pointsUnit;
 
private:
  G4double intersectionRadius;
  
  G4VSolid* beamPipeInnerSolid;
  
  G4VSolid* beamPipeOuterSolid;
 
  void CleanUpPoints();
 
  void CreateSolids(const G4String& name,
                    G4double length,
                    G4bool   buildLongForIntersection = false);
 
  void CreateSolidsAngled(const G4String&      name,
                          G4double             length,
                          const G4ThreeVector& inputFace,
                          const G4ThreeVector& outputFace);
 
  virtual BDSBeamPipe* CommonFinalConstruction(const G4String& nameIn,
                                               G4Material* vacuumMaterialIn,
                                               G4Material* beamPipeMaterialIn,
                                               G4double    lengthIn);
};
  
#endif