BlockDynamicSystem.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "BlockAstro.hpp"
#include "AstCore/BlockSystem.hpp"
#include "AstCore/BlockDerivative.hpp"
#include <unordered_map>
 
AST_NAMESPACE_BEGIN
 
class BlockDerivative;
 
class AST_CORE_API BlockDynamicSystem : public BlockSystem
{
public:
    BlockDynamicSystem() = default;
    ~BlockDynamicSystem() = default;
 
    size_t size() const { return state_.size(); }
 
    int getDimension() const { return static_cast<int>(this->size()); }
 
    const double* getStateData() const { return state_.data(); }
 
    const double* getDerivativeData() const { return derivative_.data(); }
 
    void setStateData(const double* y) { std::copy(y, y + this->size(), state_.begin()); }
 
    void getDerivativeData(double* dydt) const { std::copy(derivative_.begin(), derivative_.end(), dydt); }
 
    void fillDerivativeData(double value) { std::fill(derivative_.begin(), derivative_.end(), value); }
 
    errc_t initialize();
 
    errc_t sortBlocks();
 
    errc_t createStateMap();
 
    errc_t connectSignalsByNames();
 
    void addBlock(BlockDerivative* block);
 
    void addDerivativeBlock(BlockDerivative* block);
 
    using BlockSystem::addBlock;
 
    void clearBlocks();
 
    void reset();
 
protected:
    using StateMap = std::unordered_map<Identifier*, double*>;
 
    std::vector<BlockDerivative*>   derivativeBlocks_;  // 状态量导数函数块
    std::vector<double>             state_;             // 状态量
    std::vector<double>             derivative_;        // 状态量导数
    StateMap                        stateMap_;          // 状态量映射表
    StateMap                        derivativeMap_;     // 状态量导数映射表
};
 
 
AST_NAMESPACE_END