AstroBlockFactory.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstCore/BlockDerivative.hpp"
#include "AstCore/BlockMotion.hpp"
#include "AstCore/BlockMotionPartial.hpp"
#include "AstCore/BlockTwoBody.hpp"
#include "AstCore/BlockTwoBodyPartial.hpp"
#include "AstCore/BlockGravity.hpp"
#include "AstCore/BlockGravityPartial.hpp"
#include "AstCore/BlockMass.hpp"
#include "AstCore/BlockDrag.hpp"
#include "AstCore/BlockDragPartial.hpp"
#include "AstCore/BlockSRP.hpp"
#include "AstCore/BlockSRPPartial.hpp"
#include "AstCore/BlockThirdBodyGravity.hpp"
#include "AstCore/BlockThirdBodyGravityPartial.hpp"
#include "AstCore/BlockThirdBodyPointMass.hpp"
#include "AstCore/BlockThirdBodyPointMassPartial.hpp"
#include "AstCore/BlockStateTransitionMatrix.hpp"
#include "AstCore/BlockDragSensitivity.hpp"
#include "AstCore/BlockSRPSensitivity.hpp"
#include "AstCore/Axes.hpp"
#include "AstCore/Atmosphere.hpp"
#include "AstCore/EclipseCalculator.hpp"
#include "AstMath/Matrix.hpp"
#include <vector>
 
AST_NAMESPACE_BEGIN
 
class AstroBlockFactory
{
public:
    explicit AstroBlockFactory(bool useSTM = false) : useSTM_(useSTM) {}
 
    // ── 运动学 ──
 
    BlockMotion* createMotionBlock() { 
        if (useSTM_)
            return new BlockMotionPartial();
        else
            return new BlockMotion(); 
    }
 
    // ── 中心天体 ──
 
    BlockTwoBody* createTwoBodyBlock(double GM) { 
        if (useSTM_)
            return new BlockTwoBodyPartial(GM);
        else
            return new BlockTwoBody(GM); 
    }
 
    BlockGravity* createGravityBlock(GravityField&& gf, int degree, int order,
                                         Axes* gravityAxes, Axes* propAxes)
    {
        if (useSTM_)
            return new BlockGravityPartial(std::move(gf), degree, order, gravityAxes, propAxes);
        else
            return new BlockGravity(std::move(gf), degree, order, gravityAxes, propAxes);
    }
 
    // ── 质量 ──
 
    BlockMass* createMassBlock(double mass) { return new BlockMass(mass); }
 
    // ── 阻力 / SRP ──
 
    BlockDrag* createDragBlock(Atmosphere* atm, double cd, double area, Frame* frame)
    {
        if (useSTM_)
            return new BlockDragPartial(atm, cd, area, frame);
        else
            return new BlockDrag(atm, cd, area, frame);
    }
 
    BlockSRP* createSRPBlock(EclipseCalculator* ec, double cr, double area, Frame* frame)
    {
        if (useSTM_)
            return new BlockSRPPartial(ec, cr, area, frame);
        else
            return new BlockSRP(ec, cr, area, frame);
    }
 
    // ── 三体 ──
 
    BlockThirdBodyGravity* createThirdBodyGravityBlock(Point* ephemeris, GravityField&& gf,
                                                   int degree, int order,
                                                   Axes* gravityAxes, Frame* propFrame)
    {
        if (useSTM_)
            return new BlockThirdBodyGravityPartial(ephemeris, std::move(gf), degree, order, gravityAxes, propFrame);
        else
            return new BlockThirdBodyGravity(ephemeris, std::move(gf), degree, order, gravityAxes, propFrame);
    }
 
    BlockThirdBodyPointMass* createThirdBodyPointMassBlock(Point* ephemeris, double GM, Frame* frame)
    {
        if (useSTM_)
            return new BlockThirdBodyPointMassPartial(ephemeris, GM, frame);
        else
            return new BlockThirdBodyPointMass(ephemeris, GM, frame);
    }
    BlockStateTransitionMatrix* createStateTransitionMatrixBlock() { return new BlockStateTransitionMatrix(); }
 
    // ── 参数敏感度传播 ──
 
    BlockDragSensitivity* createDragSensitivityBlock() { return new BlockDragSensitivity(); }
    BlockSRPSensitivity* createSRPSensitivityBlock() { return new BlockSRPSensitivity(); }
private:
    bool useSTM_;
};
 
AST_NAMESPACE_END