MassSpringPlyReaderDelegate.h

// This file is a part of the OpenSurgSim project.
// Copyright 2020, SimQuest Solutions Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef SURGSIM_PHYSICS_MASSSPRINGPLYREADERDELEGATE_H
#define SURGSIM_PHYSICS_MASSSPRINGPLYREADERDELEGATE_H

#include <array>
#include <limits>
#include <memory>
#include <string>

#include "SurgSim/DataStructures/PlyReaderDelegate.h"

namespace SurgSim
{
namespace Physics
{
class MassSpringModel;

class MassSpringPlyReaderDelegate : public SurgSim::DataStructures::PlyReaderDelegate
{
public:
    MassSpringPlyReaderDelegate();

    explicit MassSpringPlyReaderDelegate(std::shared_ptr<MassSpringModel> mesh);

protected:
    bool registerDelegate(SurgSim::DataStructures::PlyReader* reader) override;
    bool fileIsAcceptable(const SurgSim::DataStructures::PlyReader& reader) override;

    virtual void endParseFile();

    virtual void* beginVertices(const std::string& elementName, size_t vertexCount);

    virtual void processVertex(const std::string& elementName);

    virtual void endVertices(const std::string& elementName);

    void* beginElements(const std::string& elementName, size_t elementCount);

    virtual void processElement(const std::string& elementName);

    void endElements(const std::string& elementName);

    void* beginSprings(const std::string& elementName, size_t elementCount);

    virtual void processSpring(const std::string& elementName);

    void endSprings(const std::string& elementName);

    void* beginBoundaryConditions(const std::string& elementName, size_t boundaryConditionCount);

    virtual void processBoundaryCondition(const std::string& elementName);

    void* beginRadius(const std::string& elementName, size_t radiusCount);

    void endRadius(const std::string& elementName);

    void* beginThickness(const std::string& elementName, size_t thicknessCount);

    void endThickness(const std::string& elementName);

protected:
    struct MassData
    {
        double x = std::numeric_limits<double>::signaling_NaN();
        double y = std::numeric_limits<double>::signaling_NaN();
        double z = std::numeric_limits<double>::signaling_NaN();
        double mass = std::numeric_limits<double>::signaling_NaN();
        int64_t overrun1 = -1; 
    } m_massData;

    bool m_hasBoundaryConditions = false;

    unsigned int m_boundaryConditionData = 0;

    bool m_has1dElement = false;
    
    bool m_has2dElement = false;

    bool m_has3dElement = false;

    struct ElementData
    {
        int64_t overrun1 = -1; 
        unsigned int* indices = nullptr;
        unsigned int nodeCount = 0;
        int64_t overrun2 = -1; 
    } m_elementData;

    struct SpringData
    {
        int64_t overrun1 = -1; 
        unsigned int* indices = nullptr;
        unsigned int nodeCount = 0;
        int64_t overrun2 = -1; 
        double stiffness = std::numeric_limits<double>::signaling_NaN();
        double damping = std::numeric_limits<double>::signaling_NaN();
    } m_springData;

    bool m_hasRadius = false;
    double m_radius = std::numeric_limits<double>::signaling_NaN();

    bool m_hasThickness = false;
    double m_thickness = std::numeric_limits<double>::signaling_NaN();

    std::shared_ptr<MassSpringModel> m_mesh;
};

} // namespace Physics
} // namespace SurgSim

#endif // SURGSIM_PHYSICS_MASSSPRINGPLYREADERDELEGATE_H