Rocket.h

Go to the documentation of this file.

#ifndef ROCKET_H
#define ROCKET_H
 
#include <string>
#include <vector>
#include <memory> // for smart pointers
 
// Forward declarations to keep compile times fast
class Stage;
class FlightState;
class Environment;

class Rocket {
public:
    Rocket(const std::string& name);

    ~Rocket() = default;

    double getTotalMass() const;

    double getTotalPropellantMass() const;

    double getCenterOfGravity() const;

    double getCenterOfPressure() const;

    double getStabilityMargin() const;

    void addStage(std::unique_ptr<Stage> stage);

    const std::vector<std::unique_ptr<Stage>>& getStages() const;

    void updateMassProperties();

    void prepareForFlight(const Environment& env);

    void applyFlightState(const FlightState& state);

    const std::string& getName() const;

    void setName(const std::string& name);
 
private:
    std::string name_; 
    std::vector<std::unique_ptr<Stage>> stages_; 
 
    // Cached mass properties (updated by updateMassProperties())
    double totalMass_; 
    double totalPropellantMass_; 
    double centerOfGravity_; 
    double centerOfPressure_; 
};
 
#endif // ROCKET_H