imstkConnectiveStrandGenerator.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 "imstkGeometryAlgorithm.h"
#include "imstkDataArray.h"

#include <random>
namespace imstk
{
class PointSet;
class SurfaceMesh;
class LineMesh;

class ConnectiveStrandGenerator : public GeometryAlgorithm
{
public:
    ConnectiveStrandGenerator();
    ~ConnectiveStrandGenerator() override = default;

    void setInputMeshes(
        std::shared_ptr<SurfaceMesh> inputMeshA,
        std::shared_ptr<SurfaceMesh> inputMeshB);

    std::shared_ptr<LineMesh> getOutputMesh() const;

    void setSegmentsPerStrand(int numSegs) { m_segmentsPerStrand = numSegs; }
    int getSegmentsPerStrand() { return m_segmentsPerStrand; }

    void setStrandsPerFace(double density) { m_strandsPerFace = density; }
    double getStrandsPerFace() { return m_strandsPerFace; }

    void setAllowedAngleDeviation(double angle) { m_allowedAngleDeviation = angle; }
    double getAllowedAgnleDeviation() const { return m_allowedAngleDeviation; }

protected:
    void requestUpdate() override;

    std::vector<int> filterCells(SurfaceMesh* meshA, SurfaceMesh* meshB) const;

    std::shared_ptr<LineMesh> createStrands(
        SurfaceMesh*            meshA,
        const std::vector<int>& faces,
        SurfaceMesh*            meshB) const;

    const Vec3d generateRandomPointOnFace(SurfaceMesh* mesh, int face) const;

    int    m_segmentsPerStrand     = 3;
    double m_strandsPerFace        = 1;
    double m_allowedAngleDeviation = PI;
};
} // namespace imstk