fast-nuclear-sim/src/reactor/rod.hpp

88 lines
1.8 KiB
C++

#pragma once
#include <json/json.h>
#include <ostream>
#include <glm/vec4.hpp>
#include <memory>
namespace Sim::Reactor
{
struct Reactor;
class Rod
{
public:
bool selected = false;
Reactor* reactor = nullptr;
static const int VAL_N = 3;
enum val_t
{
HEAT = 0,
N_SLOW = 1,
N_FAST = 2,
};
constexpr Rod() {};
Rod(const Json::Value& node);
virtual ~Rod() {};
virtual void interact(Rod* o, double secs);
virtual void update(double secs) { }
virtual void add(val_t type, double v);
virtual double extract(val_t type, double s, double k, double o);
virtual double get(val_t type) const;
virtual std::unique_ptr<Rod> clone() const { return std::make_unique<Rod>(*this); }
virtual double get_energy_output() const { return 0; }
virtual int get_id() const { return 0; }
virtual glm::vec4 get_colour() const { return {0, 0, 0, 0}; }
glm::vec4 get_heat_colour() const;
virtual bool has_sensors(val_t t) const { return false; }
virtual bool should_display() const { return false; }
virtual bool should_select() const { return false; }
virtual void update_selected(double a) { }
double get_flux() const;
double get_side_area() const;
double get_volume() const;
constexpr void toggle_selected() { selected = !selected; }
friend std::ostream& operator<<(std::ostream& o, const Rod& r)
{
if(!r.should_display()) return o;
o << r.get_name() << "\n";
r.display(o);
o << "Heat: " << r.get(val_t::HEAT) << " C\n";
o << "Fast: " << r.get(val_t::N_FAST) << " mol\n";
o << "Slow: " << r.get(val_t::N_SLOW) << " mol\n";
return o;
}
virtual Json::Value serialize() const;
protected:
double vals[VAL_N] = {0};
double vals_n[VAL_N] = {0};
virtual void display(std::ostream& o) const { };
virtual double get_k(val_t type) const { return 0; }
virtual const char* get_name() const { return "Empty"; }
void update_rod(double secs);
};
}