MeshGenerator.h

Go to the documentation of this file.

//
// Mesh Generator
//
// Copyright (c) 2008-2020
// Paul Scherrer Institut, Villigen PSI, Switzerland
// All rights reserved.
//
// OPAL is licensed under GNU GPL version 3.
//
 
#ifndef MESHGENERATOR_H_
#define MESHGENERATOR_H_
 
#include <string>
#include <utility>
#include <vector>
#include "OPALTypes.h"
 
class ElementBase;
 
class MeshData {
public:
    std::vector<Vector_t<double, 3>> vertices_m;
    std::vector<Vector_t<unsigned int, 3>> triangles_m;
    std::vector<std::pair<Vector_t<double, 3>, Vector_t<double, 3>>> decorations_m;
    int type_m;
};
 
class MeshGenerator {
public:
    MeshGenerator();
 
    static bool getTransverseSupport(const ElementBase& element, double& minor, double& major);
 
    void setDriftReference(double minor, double major);
 
    void add(const ElementBase& element);
 
    void write(const std::string& fname);
 
private:
    enum MeshType {
        OTHER = 0,
        DIPOLE,
        QUADRUPOLE,
        SEXTUPOLE,
        OCTUPOLE,
        SOLENOID,
        RFCAVITY,
        TRAVELINGWAVE,
        DRIFT
    };
 
    static MeshData getCylinder(
            double length, double minor, double major, double formFactor,
            const unsigned int numSegments = 36);
 
    static MeshData getTube(
            double length, double innerMinor, double innerMajor, double outerMinor,
            double outerMajor, const unsigned int numSegments = 36);
 
    static MeshData getQuadrupole(double length, double minor, double major, double formFactor);
 
    static MeshData getSolenoid(double length, double minor, double major);
 
    static MeshData getRFCavity(double length, double minor, double major);
 
    static MeshData getTravelingWave(double length, double minor, double major);
 
    static MeshData getBox(double length, double width, double height, double formFactor);
 
    std::vector<MeshData> elements_m;
    bool hasDriftReference_m;
    double driftMinor_m;
    double driftMajor_m;
};
 
#endif