HPOPForceModel.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstCore/CelestialBody.hpp"
#include "AstCore/GravityForce.hpp"
#include "AstCore/PointMassForce.hpp"
#include "AstCore/DragForce.hpp"
#include "AstCore/ThirdBodyForce.hpp"
#include "AstCore/SolarRadiationPressure.hpp"
#include "AstCore/RelativityCorrection.hpp"
#include "AstUtil/Object.hpp"
#include "AstUtil/ObjectNamed.hpp"
#include "AstUtil/Constants.hpp"
#include "AstUtil/ScopedPtr.hpp"
#include "AstUtil/ClonePtr.hpp"
#include <vector>
 
AST_NAMESPACE_BEGIN
 
class AST_CORE_API HPOPForceModel: public ObjectNamed
{
public:
    AST_OBJECT(HPOPForceModel)
    HPOPForceModel();
    ~HPOPForceModel();
 
    using ThirdBodyList = std::vector<ThirdBodyForce>;
 
    bool useCentralBodyAttraction() const{return useCentralBodyAttraction_;}
    
    bool useDrag() const{return useDrag_;}
    void useDrag(bool useDrag){useDrag_ = useDrag;}
 
    bool useSRP() const{return useSRP_;}
    void useSRP(bool useSRP){useSRP_ = useSRP;}
 
    bool useRelativityCorrection() const{return useRelativityCorrection_;}
    void useRelativityCorrection(bool useRelativityCorrection){useRelativityCorrection_ = useRelativityCorrection;}
 
    bool useSTM() const{return useSTM_;}
    void useSTM(bool useSTM){useSTM_ = useSTM;}
 
    bool useDragSensitivity() const{return useDragSensitivity_;}
    void useDragSensitivity(bool v){useDragSensitivity_ = v;}
 
    bool useSRPSensitivity() const{return useSRPSensitivity_;}
    void useSRPSensitivity(bool v){useSRPSensitivity_ = v;}
 
    Body* centralBody() const;
    void setCentralBody(Body* body){centralBody_ = body;}
 
    DragForce& drag(){return drag_;}
    const DragForce& drag() const {return drag_;}
 
    SolarRadiationPressure& srp(){return srp_;}
    const SolarRadiationPressure& srp() const {return srp_;}
 
    GravityForce& gravity();
 
    PointMassForce& pointMass();
    
    BodyAttraction& bodyAttraction();
 
    const BodyAttraction& bodyAttraction() const;
 
    EBodyAttractionType bodyAttractionType() const;
 
    ThirdBodyForce* addThirdBody(StringView bodyName);
 
    ThirdBodyForce* addThirdBody(Body* body);
 
    ThirdBodyForce* addThirdBody(ThirdBodyForce& force);
 
    ThirdBodyForce* getThirdBody(Body* body);
 
    ThirdBodyForce* getThirdBody(StringView bodyName);
 
    const ThirdBodyList& getThirdBodies() const{return thirdBodies_;}
    ThirdBodyList& thirdBodies(){return thirdBodies_;}
private:
    bool                        useDrag_{false};                    
    bool                        useSRP_{false};                     
    bool                        useRelativityCorrection_{false};    
    bool                        useCentralBodyAttraction_{true};    
    bool                        useSTM_{false};                     
    bool                        useDragSensitivity_{false};         
    bool                        useSRPSensitivity_{false};          
    SharedPtr<Body>             centralBody_{};                      
    DragForce                   drag_{};                             
    SolarRadiationPressure      srp_{};                              
    ThirdBodyList               thirdBodies_{};                      
    ClonePtr<BodyAttraction>    bodyAttraction_{};                   
 
    bool useMoonGravity_{false};
    double moonGravity_{kMoonGrav};
public: // [月球特定的接口，待废弃]
    bool useMoonGravity() const{return useMoonGravity_;}
    void useMoonGravity(bool useMoonGravity){useMoonGravity_ = useMoonGravity;}
    double moonGravity() const{return moonGravity_;}
    void setMoonGravity(double moonGravity){moonGravity_ = moonGravity;}
};
 
 
AST_NAMESPACE_END