Geometry.h

Go to the documentation of this file.

#ifndef GEOMETRY_H_
#define GEOMETRY_H_

#ifdef __cplusplus
#include <sys/stat.h>
#include <omp.h>
#include <vector>
#include "Region.h"
#include "Fissioner.h"
#include "TallyBank.h"
#include "Timer.h"
#endif


typedef enum spatialTypes {
    INFINITE_HOMOGENEOUS,
    HOMOGENEOUS_EQUIVALENCE,
    HETEROGENEOUS,
} spatialType;


class Geometry {

private:
    const char* _geometry_name;
    int _num_neutrons_per_batch;
    int _num_batches;
    int _num_threads;

    spatialType _spatial_type;
    InfiniteMediumRegion* _infinite_medium;
    EquivalenceRegion* _fuel;
    EquivalenceRegion* _moderator;
    std::vector<BoundedRegion*> _regions;

    float _fuel_radius;
    float _pitch;
    float _buckling_squared;

    float _dancoff;
    float _sigma_e;
    float _beta;
    float _alpha1;
    float _alpha2;

    int _num_prob;
    float* _prob_energies;
    float* _prob_ff;
    float* _prob_mf;

    Fissioner* _fissioner;
    float _source_sampling_radius;

    void initializeProbModFuelRatios();

public:
    Geometry(spatialType spatial_type, const char* name=(const char*)"");
    virtual ~Geometry();
    
    const char* getName();
    int getNumNeutronsPerBatch();
    int getTotalNumNeutrons();
    int getNumBatches();
    int getNumThreads();
    spatialType getSpatialType();
    float getBucklingSquared();
    float getVolume();
    float getSourceSamplingRadius();

    void setName(const char* name);
    void setSourceSamplingRadius(float radius);         
    void setNeutronsPerBatch(int num_neutrons_per_batch);
    void setNumBatches(int num_batches);
    void setNumThreads(int num_threads);
    void setSpatialType(spatialType spatial_type);
    void setFuelPinRadius(float radius);
    void setPinCellPitch(float pitch);
    void setDancoffFactor(float dancoff);
    void addRegion(Region* region);
    void setBucklingSquared(float buckling_squared);

    bool contains(neutron* neutron);
    bool contains(float x, float y, float z);
    void findContainingRegion(neutron* neutron);
    Region* findContainingRegion(float x, float y, float z);    
    void runMonteCarloSimulation();
    void initializeSourceNeutron(neutron* neutron);
};


#endif /* GEOMETRY_H_ */