OctreeNodePlyReaderDelegate.h

// This file is a part of the OpenSurgSim project.
// Copyright 2013, 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_DATASTRUCTURES_OCTREENODEPLYREADERDELEGATE_H
#define SURGSIM_DATASTRUCTURES_OCTREENODEPLYREADERDELEGATE_H

#include "SurgSim/DataStructures/PlyReaderDelegate.h"
#include "SurgSim/DataStructures/OctreeNode.h"
#include "SurgSim/DataStructures/EmptyData.h"

namespace SurgSim
{
namespace DataStructures
{

class OctreeNodePlyReaderDelegateBase : public PlyReaderDelegate
{
public:

    OctreeNodePlyReaderDelegateBase();

    virtual ~OctreeNodePlyReaderDelegateBase();

    bool registerDelegate(PlyReader* reader) override;

    bool fileIsAcceptable(const PlyReader& reader) override;

    void* beginBounds(const std::string& elementName, size_t count);

    void* beginDimension(const std::string& elementName, size_t count);

    void* beginSpacing(const std::string& elementName, size_t count);

    virtual void* beginVoxel(const std::string& elementName, size_t count);

    virtual void processVoxel(const std::string& elementName) = 0;

protected:
    virtual void initializeOctree() = 0;

    struct BoundsData
    {
        double xMin = std::numeric_limits<double>::signaling_NaN();
        double yMin = std::numeric_limits<double>::signaling_NaN();
        double zMin = std::numeric_limits<double>::signaling_NaN();
        double xMax = std::numeric_limits<double>::signaling_NaN();
        double yMax = std::numeric_limits<double>::signaling_NaN();
        double zMax = std::numeric_limits<double>::signaling_NaN();
    } m_bounds;

    Eigen::AlignedBox<double, 3> m_boundingBox;

    struct SpacingData
    {
        double x = std::numeric_limits<double>::signaling_NaN();
        double y = std::numeric_limits<double>::signaling_NaN();
        double z = std::numeric_limits<double>::signaling_NaN();
    } m_spacing;

    struct DimensionData
    {
        unsigned int x = 0;
        unsigned int y = 0;
        unsigned int z = 0;
    } m_dimension;

    struct VoxelData
    {
        double x = std::numeric_limits<double>::signaling_NaN();
        double y = std::numeric_limits<double>::signaling_NaN();
        double z = std::numeric_limits<double>::signaling_NaN();
    } m_voxel;

    int m_numLevels = -1;

    bool m_haveSpacing;
    bool m_haveDimensions;
    bool m_haveBounds;

};

template <typename Data>
class OctreeNodePlyReaderDelegate : public OctreeNodePlyReaderDelegateBase
{
public:

    OctreeNodePlyReaderDelegate();

    explicit OctreeNodePlyReaderDelegate(std::shared_ptr<OctreeNode<Data>> octree);

    virtual ~OctreeNodePlyReaderDelegate();

    std::shared_ptr<OctreeNode<Data>> getOctree();

    void processVoxel(const std::string& elementName) override;

    void initializeOctree() override;

private:

    std::shared_ptr<OctreeNode<Data>> m_octree;
};

}
}

#include "SurgSim/DataStructures/OctreeNodePlyReaderDelegate-inl.h"

#endif