motor model selection working with RSE files

QtRocket.cpp

@@ -13,9 +13,6 @@
 // qtrocket headers
 #include "QtRocket.h"
 #include "gui/MainWindow.h"
-#include "sim/ConstantAtmosphere.h"
-#include "sim/ConstantGravityModel.h"
-
 
 // Initialize static member data
 QtRocket* QtRocket::instance = nullptr;

gui/MainWindow.cpp

@@ -90,9 +90,8 @@ void MainWindow::on_testButton2_clicked()
 
    double initialVelocityX = initialVelocity * std::cos(initialAngle / 57.2958);
    double initialVelocityZ = initialVelocity * std::sin(initialAngle / 57.2958);
-   std::shared_ptr<Rocket> rocket = std::make_shared<Rocket>();
-   QtRocket::getInstance()->addRocket(rocket);
    std::vector<double> initialState = {0.0, 0.0, 0.0, initialVelocityX, 0.0, initialVelocityZ, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0};
+   auto rocket = QtRocket::getInstance()->getRocket();
    rocket->setInitialState(initialState);
    rocket->setMass(mass);
    rocket->setDragCoefficient(dragCoeff);
@@ -111,14 +110,14 @@ void MainWindow::on_loadRSE_button_clicked()
                                                   "/home",
                                                   tr("Rocksim Engine Files (*.rse)"));
 
-   utils::RSEDatabaseLoader loader(rseFile.toStdString());
+   rseDatabase.reset(new utils::RSEDatabaseLoader(rseFile.toStdString()));
 
    ui->rocketPartButtons->findChild<QLineEdit*>(QString("databaseFileLine"))->setText(rseFile);
 
    QComboBox* engineSelector =
          ui->rocketPartButtons->findChild<QComboBox*>(QString("engineSelectorComboBox"));
 
-   const std::vector<model::MotorModel>& motors = loader.getMotors();
+   const std::vector<model::MotorModel>& motors = rseDatabase->getMotors();
    for(const auto& motor : motors)
    {
       std::cout << "Adding: " << motor.data.commonName << std::endl;
@@ -146,3 +145,13 @@ void MainWindow::on_actionSimulation_Options_triggered()
 
 }
 
+
+void MainWindow::on_setMotor_clicked()
+{
+   QString motorName = ui->engineSelectorComboBox->currentText();
+   model::MotorModel mm = rseDatabase->getMotorModelByName(motorName.toStdString());
+   QtRocket::getInstance()->getRocket()->setMotorModel(mm);
+
+
+}
+

gui/MainWindow.h

@@ -4,6 +4,7 @@
 /// \cond
 // C headers
 // C++ headers
+#include <memory>
 // 3rd Party headers
 #include <QMainWindow>
 /// \endcond
@@ -12,6 +13,7 @@
 #include "QtRocket.h"
 
 #include "gui/SimOptionsWindow.h"
+#include "utils/RSEDatabaseLoader.h"
 
 
 QT_BEGIN_NAMESPACE
@@ -47,10 +49,13 @@ private slots:
 
    void on_actionSimulation_Options_triggered();
 
+   void on_setMotor_clicked();
+
    private:
    Ui::MainWindow* ui;
    QtRocket* qtRocket;
 
    SimOptionsWindow* simOptionsWindow{nullptr};
+   std::unique_ptr<utils::RSEDatabaseLoader> rseDatabase;
 };
 #endif // MAINWINDOW_H

gui/MainWindow.ui

@@ -211,6 +211,13 @@
         <item row="1" column="0">
          <widget class="QComboBox" name="engineSelectorComboBox"/>
         </item>
+        <item row="1" column="1">
+         <widget class="QPushButton" name="setMotor">
+          <property name="text">
+           <string>Set Motor</string>
+          </property>
+         </widget>
+        </item>
        </layout>
       </widget>
       <widget class="QPushButton" name="getTCMotorData">

model/MotorModel.cpp

@@ -9,6 +9,7 @@
 #include "model/MotorModel.h"
 #include "utils/math/Constants.h"
 #include "utils/math/UtilityMathFunctions.h"
+#include "utils/Logger.h"
 
 namespace model
 {
@@ -69,6 +70,22 @@ double MotorModel::getMass(double simTime) const
    }
 }
 
+double MotorModel::getThrust(double simTime)
+{
+
+   if(simTime > thrust.getMaxTime() + ignitionTime)
+   {
+      if(!burnOutOccurred)
+      {
+         utils::Logger::getInstance()->info("motor burnout occurred: " + std::to_string(simTime));
+         burnOutOccurred = true;
+      }
+      return 0.0;
+   }
+   utils::Logger::getInstance()->info("simTime: " + std::to_string(simTime) + ": thrust: " + std::to_string(thrust.getThrust(simTime)));
+   return thrust.getThrust(simTime);
+}
+
 void MotorModel::setMetaData(const MetaData& md)
 {
    data = md;

model/MotorModel.h

@@ -396,11 +396,13 @@ public:
 
    void startMotor(double startTime) { ignitionOccurred = true; ignitionTime = startTime; }
 
+   void addThrustCurve(const ThrustCurve& tc) { thrust = tc; }
 
    // Thrust parameters
    MetaData data;
 private:
    bool ignitionOccurred{false};
+    bool burnOutOccurred{false};
    double emptyMass;
    double isp;
    double maxTime;

model/Rocket.cpp

@@ -9,6 +9,8 @@ Rocket::Rocket() : propagator(this)
 void Rocket::launch()
 {
    propagator.setTimeStep(QtRocket::getInstance()->getTimeStep());
+    propagator.clearStates();
+    mm.startMotor(0.0);
    propagator.runUntilTerminate();
 }
 
@@ -28,10 +30,5 @@ bool Rocket::terminateCondition(const std::pair<double, std::vector<double>>& co
 
 double Rocket::getThrust(double t)
 {
-   return tc.getThrust(t);
+   return mm.getThrust(t);
-}
-
-void Rocket::setThrustCurve(const ThrustCurve& curve)
-{
-   tc = curve;
 }

model/Rocket.h

@@ -114,7 +114,6 @@ private:
    double mass; /// @todo get rid of this, should be dynamically computed, but is the current rocket mass
 
    model::MotorModel mm; /// Current Motor Model
-   ThrustCurve tc; /// @todo get rid of this, should be returned from the MotorModel
 
 };
 

model/ThrustCurve.cpp

@@ -55,12 +55,6 @@ void ThrustCurve::setIgnitionTime(double t)
 double ThrustCurve::getThrust(double t)
 {
    // calculate t relative to the start time of the motor
-   static bool burnout{false};
-   if(!burnout && t >= (maxTime + ignitionTime))
-   {
-      burnout = true;
-      utils::Logger::getInstance()->info("Motor burnout at time: " + std::to_string(t));
-   }
    t -= ignitionTime;
    if(t < 0.0 || t > maxTime)
    {

sim/Propagator.cpp

@@ -2,6 +2,7 @@
 /// \cond
 // C headers
 // C++ headers
+#include <cmath>
 #include <chrono>
 #include <iostream>
 #include <sstream>
@@ -97,20 +98,20 @@ double Propagator::getMass()
 double Propagator::getForceX()
 {
     QtRocket* qtrocket = QtRocket::getInstance();
-    return - qtrocket->getAtmosphereModel()->getDensity(currentState[3])/ 2.0 * 0.008107 * rocket->getDragCoefficient() * currentState[3]* currentState[3];
+    return (currentState[3] >= 0 ? -1.0 : 1.0) *  qtrocket->getAtmosphereModel()->getDensity(currentState[2])/ 2.0 * 0.008107 * rocket->getDragCoefficient() * currentState[3]* currentState[3];
 }
 
 double Propagator::getForceY()
 {
     QtRocket* qtrocket = QtRocket::getInstance();
-    return -qtrocket->getAtmosphereModel()->getDensity(currentState[3]) / 2.0 * 0.008107 * rocket->getDragCoefficient() * currentState[4]* currentState[4];
+    return (currentState[4] >= 0 ? -1.0 : 1.0) * qtrocket->getAtmosphereModel()->getDensity(currentState[2]) / 2.0 * 0.008107 * rocket->getDragCoefficient() * currentState[4]* currentState[4];
 }
 
 double Propagator::getForceZ()
 {
     QtRocket* qtrocket = QtRocket::getInstance();
     double gravity = (qtrocket->getGravityModel()->getAccel(currentState[0], currentState[1], currentState[2])).x3;
-    double airDrag = -qtrocket->getAtmosphereModel()->getDensity(currentState[3]) / 2.0 * 0.008107 * rocket->getDragCoefficient() * currentState[5]* currentState[5];
+    double airDrag = (currentState[5] >= 0 ? -1.0 : 1.0) * qtrocket->getAtmosphereModel()->getDensity(currentState[2]) / 2.0 * 0.008107 * rocket->getDragCoefficient() * currentState[5]* currentState[5];
     double thrust  = rocket->getThrust(currentTime);
     return gravity + airDrag + thrust;
 }

sim/Propagator.h

@@ -51,6 +51,8 @@ public:
 
     const std::vector<std::pair<double, std::vector<double>>>& getStates() const { return states; }
 
+    void clearStates() { states.clear(); }
+
     void setTimeStep(double ts) { timeStep = ts; }
 
     void setSaveStats(bool s) { saveStates = s; }

sim/SimulationOptions.h

@@ -33,9 +33,9 @@ class SimulationOptions
 public:
     SimulationOptions()
     {
-        //setTimeStep(0.01);
+        setTimeStep(0.01);
-        //setGravityModel("Constant Gravity");
+        setGravityModel("Constant Gravity");
-        //setAtmosphereModel("Constant Atmosphere");
+        setAtmosphereModel("Constant Atmosphere");
     }
     ~SimulationOptions() = default;
     SimulationOptions(const SimulationOptions&) = delete;
@@ -79,16 +79,20 @@ public:
         if(model == "Constant Atmosphere")
         {
             atmosphereModel = model;
-            atmosphereModels[gravityModel].reset(new sim::ConstantAtmosphere);
+            atmosphereModels[atmosphereModel].reset(new sim::ConstantAtmosphere);
         }
         else if(model == "US Standard 1976")
         {
             atmosphereModel = model;
-            atmosphereModels[gravityModel].reset(new sim::USStandardAtmosphere);
+            atmosphereModels[atmosphereModel].reset(new sim::USStandardAtmosphere);
         }
     }
 
-    std::shared_ptr<sim::AtmosphericModel> getAtmosphericModel() { return atmosphereModels[atmosphereModel]; }
+    std::shared_ptr<sim::AtmosphericModel> getAtmosphericModel()
+    {
+        auto retVal = atmosphereModels[atmosphereModel];
+        return retVal;
+    }
     std::shared_ptr<sim::GravityModel> getGravityModel() { return gravityModels[gravityModel]; }
     double getTimeStep() { return timeStep; }
 

utils/RSEDatabaseLoader.cpp

@@ -3,6 +3,7 @@
 // C headers
 // C++ headers
 #include <iostream>
+#include <algorithm>
 #include <cmath>
 
 // 3rd party headers
@@ -12,6 +13,7 @@
 // qtrocket headers
 #include "utils/RSEDatabaseLoader.h"
 #include "QtRocket.h"
+#include "Logger.h"
 
 namespace utils {
 
@@ -33,6 +35,19 @@ RSEDatabaseLoader::RSEDatabaseLoader(const std::string& filename)
 RSEDatabaseLoader::~RSEDatabaseLoader()
 {}
 
+model::MotorModel RSEDatabaseLoader::getMotorModelByName(const std::string &name)
+{
+
+   auto mm = std::find_if(motors.begin(), motors.end(),
+                                        [&name](const auto& i) { return name == i.data.commonName; });
+   if(mm == motors.end())
+   {
+      Logger::getInstance()->error("Unable to locate " + name + " in RSE database");
+      return model::MotorModel();
+   }
+   return *mm;
+}
+
 void RSEDatabaseLoader::buildAndAppendMotorModel(boost::property_tree::ptree& v)
 {
    model::MotorModel::MetaData mm;
@@ -70,7 +85,9 @@ void RSEDatabaseLoader::buildAndAppendMotorModel(boost::property_tree::ptree& v)
       thrustData.push_back(std::make_pair(tdata, fdata));
    }
 
+   ThrustCurve tc(thrustData);
    model::MotorModel motorModel;
+   motorModel.addThrustCurve(tc);
    motorModel.moveMetaData(std::move(mm));
    motors.emplace_back(std::move(motorModel));
 }

utils/RSEDatabaseLoader.h

@@ -23,7 +23,9 @@ public:
    RSEDatabaseLoader(const std::string& filename);
    ~RSEDatabaseLoader();
 
-   const std::vector<model::MotorModel>& getMotors() const { return motors; }
+   std::vector<model::MotorModel>& getMotors() { return motors; }
+
+   model::MotorModel getMotorModelByName(const std::string& name);
 private:
 
    std::vector<model::MotorModel> motors;