surface_quantity.h

#ifndef MIDSX_SURFACE_QUANTITY_H
#define MIDSX_SURFACE_QUANTITY_H
 
#include <memory>
#include "quantity.h"
#include <Eigen/Dense>
#include "photon.h"
#include "tally_data.h"
 
enum class VectorSurfaceQuantityType {
    IncidentEnergy,
    EntranceCosine,
};
 
 
namespace VectorSurfaceQuantityHelper {
    std::string toString(VectorSurfaceQuantityType type);
}
 
enum class CountSurfaceQuantityType {
    NumberOfPhotons
};
 
namespace CountSurfaceQuantityHelper {
    std::string toString(CountSurfaceQuantityType type);
}
 
class VectorSurfaceQuantity {
public:
    using ValueExtractor = std::function<double(const TempSurfaceTallyData&)>;
 
    explicit VectorSurfaceQuantity(VectorSurfaceQuantityType type);
 
    VectorSurfaceQuantity operator+(VectorSurfaceQuantity& other) const;
 
    void measure(TempSurfaceTallyData& temp_surface_tally_data);
 
    VectorSurfaceQuantityType getType() const;
 
    // Wrapper functions for VectorValue. Self-explanatory.
    VectorValue& getTotalValues();
    VectorValue& getPrimaryValues();
    VectorValue& getSingleIncoherentScatterValues();
    VectorValue& getSingleCoherentScatterValues();
    VectorValue& getMultipleScatterValues();
private:
    VectorSurfaceQuantityType type_;
    bool totaled_ = false;
    VectorValue total_values_;
    VectorValue primary_values_;
    VectorValue single_incoherent_scatter_values_;
    VectorValue single_coherent_scatter_values_;
    VectorValue multiple_scatter_values_;
    ValueExtractor valueExtractor_;
};
 
class CountSurfaceQuantity {
public:
    using ValueExtractor = std::function<bool(const TempSurfaceTallyData&)>;
 
    explicit CountSurfaceQuantity(CountSurfaceQuantityType type);
 
    CountSurfaceQuantity operator+(const CountSurfaceQuantity& other) const;
 
    void measure(TempSurfaceTallyData& temp_surface_tally_data);
 
    CountSurfaceQuantityType getType() const;
 
    // Wrapper functions for CountValue. Self-explanatory.
    CountValue getTotalValues();
    CountValue getPrimaryValues() const;
    CountValue getSingleIncoherentScatterValues() const;
    CountValue getSingleCoherentScatterValues() const;
    CountValue getMultipleScatterValues() const;
private:
    CountSurfaceQuantityType type_;
    bool totaled_ = false;
    CountValue total_values_;
    CountValue primary_values_;
    CountValue single_incoherent_scatter_values_;
    CountValue single_coherent_scatter_values_;
    CountValue multiple_scatter_values_;
    ValueExtractor valueExtractor_;
};
 
#endif //MIDSX_SURFACE_QUANTITY_H