BodyAttraction.hpp

浏览该文件的文档.

 
 
#pragma once
 
#include "AstGlobal.h"
#include "AstCore/ForceModel.hpp"
 
AST_NAMESPACE_BEGIN
 
class GravityForce;
class PointMassForce;
class ThirdBodyForce;
 
enum class EBodyAttractionType
{
    eGravity,       
    ePointMass,     
    eUnknown,       
};
 
 
class AST_CORE_API BodyAttraction: public ForceModel
{
public:
    AST_OBJECT(BodyAttraction)
    
    
    virtual EBodyAttractionType getBodyAttractionType() const = 0;
 
    virtual BodyAttraction* clone() const = 0;
 
    bool isAttractionType(EBodyAttractionType type) const{
        return getBodyAttractionType() == type;
    }
    
    const GravityForce* asGravityForce() const;
    GravityForce* asGravityForce();
 
    const PointMassForce* asPointMassForce() const;
    PointMassForce* asPointMassForce();
};
 
 
 
AST_NAMESPACE_END