HPOPEquation.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstMath/OrdinaryDifferentialEquation.hpp"
#include "AstCore/BlockDynamicSystem.hpp"
#include "AstCore/SimTime.hpp"
#include "AstCore/HPOPForceModel.hpp"
#include "AstCore/Frame.hpp"
#include "AstCore/SpacecraftParam.hpp"
#include <memory>
 
 
AST_NAMESPACE_BEGIN
 
class FuncBlock;
class BlockDerivative;
class HPOPForceModel;
 
class AST_CORE_API HPOPEquation : public OrdinaryDifferentialEquation
{
public:
    HPOPEquation();
    explicit HPOPEquation(HPOPForceModel&& forceModel);
    ~HPOPEquation();
 
    int getDimension() const final;
 
    errc_t evaluate(const double* y, double* dy, double t) final;
    
    void setEpoch(const TimePoint& epoch){ epoch_ = epoch; }
 
    errc_t setForceModel(HPOPForceModel&& forceModel);
    errc_t setForceModel(const HPOPForceModel& forceModel);
 
    HPOPForceModel& forceModel() { return forceModel_; }
    const HPOPForceModel& forceModel() const { return forceModel_; }
 
    const BlockDynamicSystem& dynamicSystem() const { return dynamicSystem_; }
    BlockDynamicSystem& dynamicSystem() { return dynamicSystem_; }
 
    void setSpacecraftParam(const SpacecraftParam& spacecraftParam);
 
    const SpacecraftParam& spacecraftParam() const { return spacecraftParam_; }
 
    errc_t setPropagationFrame(Frame* frame);
 
    Frame* getPropagationFrame() const;
 
    Body* getCentralBody() const;
protected:
    void addBlock(FuncBlock* block);
    void addBlock(BlockDerivative* block);
    
    void clearBlocks();
    void reset();
public:
    errc_t initialize();
protected:
    errc_t initializeFromForceModel(const HPOPForceModel& forceModel, const SpacecraftParam& spacecraftParam);
    errc_t initBlocks(const HPOPForceModel& forceModel, const SpacecraftParam &spacecraftParam);
private:
    BlockDynamicSystem      dynamicSystem_;     
    TimePoint               epoch_{};           
    HPOPForceModel          forceModel_{};      
    SpacecraftParam         spacecraftParam_{}; 
    HFrame                  propFrame_;         
};
 
AST_NAMESPACE_END