Surface.h

Go to the documentation of this file.

#ifndef SURFACE_H_
#define SURFACE_H_

#include <limits>
#include <math.h>
#include "log.h"
#include "vector.h"
#include "Neutron.h"

#define PI_OVER_TWO 1.57079633

#define THREE_PI_OVER_TWO 4.71238898

#define TWO_PI 6.28318531

#define TINY_MOVE 1E-5


typedef enum surfaceTypes {
    XPLANE,
    YPLANE,
    ZCYLINDER
} surfaceType;


typedef enum boundaryTypes {
    REFLECTIVE,
    VACUUM,
    INTERFACE
} boundaryType;


class Surface {
protected:
    char* _surface_name;
    static int _n;
    int _uid;
    surfaceType _surface_type;
    boundaryType _boundary_type;
public:
    Surface(const char* surface_name=(char*)"");
    virtual ~Surface();

    char* getSurfaceName();
    int getUid() const;    
    boundaryType getBoundaryType() const;
    surfaceType getSurfaceType() const;

    void setBoundaryType(boundaryType type);

    virtual float evaluate(neutron* neutron) =0;

    virtual float evaluate(float x, float y, float z) =0;

    virtual float computeParametrizedDistance(neutron* neutron) =0;

    virtual bool onSurface(neutron* neutron) =0;

    virtual void reflectNeutron(neutron* neutron) =0;
};


class XPlane: public Surface {
private:
    float _x;
public:
    XPlane(const char* surface_name=(char*)"");
    virtual ~XPlane();
    float getX();
    void setX(float x);
    float evaluate(neutron* neutron);
    float evaluate(float x, float y, float z);
    float computeParametrizedDistance(neutron* neutron);
    bool onSurface(neutron* neutron);
    void reflectNeutron(neutron* neutron);
};


class YPlane: public Surface {
private:
    float _y;
public:
    YPlane(const char* surface_name=(char*)"");
    virtual ~YPlane();
    float getY();
    void setY(float y);
    float evaluate(neutron* neutron);
    float evaluate(float x, float y, float z);
    float computeParametrizedDistance(neutron* neutron);
    bool onSurface(neutron* neutron);
    void reflectNeutron(neutron* neutron);
};


class ZCylinder: public Surface {
protected:
    float _r;
    float  _r_squared;
    float _x0;
    float _y0;
public:
    ZCylinder(const char* surface_name=(char*)"");
    virtual ~ZCylinder();
    float getX0();
    float getY0();
    float getRadius();
    void setX0(float x0);
    void setY0(float y0);
    void setRadius(float r);
    float evaluate(neutron* neutron);
    float evaluate(float x, float y, float z);
    float computeParametrizedDistance(neutron* neutron);
    bool onSurface(neutron* neutron);
    void reflectNeutron(neutron* neutron);
};

#endif /* SURFACE_H_ */