imstkCapsule.h

/*
** This file is part of the Interactive Medical Simulation Toolkit (iMSTK)
** iMSTK is distributed under the Apache License, Version 2.0.
** See accompanying NOTICE for details.
*/

#pragma once

#include "imstkAnalyticalGeometry.h"
#include "imstkMacros.h"

namespace imstk
{
class Capsule : public AnalyticalGeometry
{
public:
    Capsule(const Vec3d& pos    = Vec3d(0.0, 0.0, 0.0), const double radius = 0.5,
            const double length = 1.0, const Quatd orientation = Quatd::Identity()) :
        AnalyticalGeometry()
    {
        setPosition(pos);
        setOrientation(orientation);
        setRadius(radius);
        setLength(length);
        updatePostTransformData();
    }

    ~Capsule() override = default;

    IMSTK_TYPE_NAME(Capsule)

    
    void print() const override;

    double getVolume() override { return PI * m_radius * m_radius * (m_length + 4.0 / 3.0 * m_radius); }

    double getRadius(DataType type = DataType::PostTransform);
    void setRadius(const double r);

    double getLength(DataType type = DataType::PostTransform);
    void setLength(const double l);

    double getFunctionValue(const Vec3d& x) const override;

    void computeBoundingBox(Vec3d& min, Vec3d& max, const double paddingPercent);

    void updatePostTransformData() const override;

    std::unique_ptr<Capsule> clone()
    {
        return std::unique_ptr<Capsule>(cloneImplementation());
    }

protected:
    void applyTransform(const Mat4d& m) override;

    double m_radius = 1.0;                      
    mutable double m_radiusPostTransform = 1.0; 
    double m_length = 1.0;                      
    mutable double m_lengthPostTransform = 1.0; 

private:
    Capsule* cloneImplementation() const
    {
        return new Capsule(*this);
    }
};
} // namespace imstk