transform_shape.h

Go to the documentation of this file.

#ifndef TRANSFORM_SHAPE_H
#define TRANSFORM_SHAPE_H

#include "base_data_package.h"
#include "base_geometry.h"

namespace SPH
{
    template <class BaseShapeType>
    class TransformShape : public BaseShapeType
    {

    public:
        template <typename... ConstructorArgs>
        explicit TransformShape(const Transformd &transformd, ConstructorArgs &&...args)
            : BaseShapeType(std::forward<ConstructorArgs>(args)...),
              transformd_(transformd){};
        virtual ~TransformShape(){};

        Transformd &getTransform() { return transformd_; };
        void setTransform(const Transformd &transformd) { transformd_ = transformd; };

        virtual bool checkContain(const Vecd &input_pnt, bool BOUNDARY_INCLUDED = true) override
        {
            Vecd input_pnt_origin = transformd_.shiftBaseStationToFrame(input_pnt);
            return BaseShapeType::checkContain(input_pnt_origin);
        };
        virtual Vecd findClosestPoint(const Vecd &input_pnt) override
        {
            Vecd input_pnt_origin = transformd_.shiftBaseStationToFrame(input_pnt);
            Vecd closest_point_origin = BaseShapeType::findClosestPoint(input_pnt_origin);
            return transformd_.shiftFrameStationToBase(closest_point_origin);
        };

    protected:
        Transformd transformd_;

        virtual BoundingBox findBounds() override
        {
            BoundingBox original_bound = BaseShapeType::findBounds();
            return BoundingBox(transformd_.shiftFrameStationToBase(original_bound.first),
                               transformd_.shiftFrameStationToBase(original_bound.second));
        };
    };
}

#endif // TRANSFORM_SHAPE_H