82 lines
2.0 KiB
C++
82 lines
2.0 KiB
C++
#ifndef QTROCKET_H
|
|
#define QTROCKET_H
|
|
|
|
/// \cond
|
|
// C headers
|
|
// C++ headers
|
|
#include <atomic>
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <utility>
|
|
|
|
// 3rd party headers
|
|
/// \endcond
|
|
|
|
// qtrocket headers
|
|
#include "model/MotorModel.h"
|
|
#include "model/Rocket.h"
|
|
#include "sim/AtmosphericModel.h"
|
|
#include "sim/GravityModel.h"
|
|
#include "sim/Environment.h"
|
|
#include "sim/Propagator.h"
|
|
#include "utils/Logger.h"
|
|
#include "utils/MotorModelDatabase.h"
|
|
|
|
/**
|
|
* @brief The QtRocket class is the master controller for the QtRocket application.
|
|
* It is the singleton that controls the interaction of the various components of
|
|
* the QtRocket program
|
|
*/
|
|
class QtRocket
|
|
{
|
|
public:
|
|
static QtRocket* getInstance();
|
|
|
|
utils::Logger* getLogger() { return logger; }
|
|
|
|
// This will return when the main window returns;
|
|
// If called multiple times, subsequent calls, will simply
|
|
// immediately return with value 0
|
|
int run(int argc, char* argv[]);
|
|
|
|
void runSim();
|
|
|
|
|
|
std::shared_ptr<sim::Environment> getEnvironment() { return environment; }
|
|
void setTimeStep(double t) { rocket.second->setTimeStep(t); }
|
|
std::shared_ptr<Rocket> getRocket() { return rocket.first; }
|
|
|
|
std::shared_ptr<utils::MotorModelDatabase> getMotorDatabase() { return motorDatabase; }
|
|
|
|
void addMotorModels(std::vector<model::MotorModel>& m);
|
|
|
|
const std::vector<model::MotorModel>& getMotorModels() const { return motorModels; }
|
|
|
|
void addRocket(std::shared_ptr<Rocket> r) { rocket.first = r; }
|
|
|
|
void setEnvironment(std::shared_ptr<sim::Environment> e) { environment = e; }
|
|
|
|
private:
|
|
QtRocket();
|
|
|
|
static void init();
|
|
|
|
std::atomic_bool running;
|
|
static bool initialized;
|
|
static std::mutex mtx;
|
|
static QtRocket* instance;
|
|
|
|
// Motor "database(s)"
|
|
std::vector<model::MotorModel> motorModels;
|
|
|
|
utils::Logger* logger;
|
|
|
|
std::pair<std::shared_ptr<Rocket>, std::shared_ptr<sim::Propagator>> rocket;
|
|
|
|
std::shared_ptr<sim::Environment> environment;
|
|
std::shared_ptr<utils::MotorModelDatabase> motorDatabase;
|
|
|
|
};
|
|
|
|
#endif // QTROCKET_H
|