MRCudaAccessor.h

Go to the documentation of this file.

#pragma once
 
#include "exports.h"
#include "MRMesh/MRMeshFwd.h"
#include "MRMesh/MRExpected.h"
#include <functional>
#include <memory>
 
namespace MR
{
 
struct PointsToDistanceVolumeParams;
class MRVIEWER_CLASS CudaAccessor
{
public:
    using CudaFreeMemoryFunc = std::function<size_t()>;
    using CudaFwnConstructor = std::function<std::unique_ptr<IFastWindingNumber>( const Mesh& )>;
    using CudaMeshProjectorConstructor = std::function<std::unique_ptr<IPointsToMeshProjector>()>;
 
    using CudaPointsToDistanceVolumeCallback = std::function<Expected<SimpleVolume>( const PointCloud& cloud, const PointsToDistanceVolumeParams& params )>;
 
    // setup functions
    MRVIEWER_API static void setCudaAvailable( bool val, int maxDriverVersion, int runtimeVersion, int computeMajor, int computeMinor );
    MRVIEWER_API static void setCudaFreeMemoryFunc( CudaFreeMemoryFunc freeMemFunc );
    MRVIEWER_API static void setCudaFastWindingNumberConstructor( CudaFwnConstructor fwnCtor );
    MRVIEWER_API static void setCudaMeshProjectorConstructor( CudaMeshProjectorConstructor mpCtor );
    MRVIEWER_API static void setCudaPointsToDistanceVolumeCallback( CudaPointsToDistanceVolumeCallback callback );
 
    // Returns true if CUDA is available on this computer
    [[nodiscard]] MRVIEWER_API static bool isCudaAvailable();
 
    // Returns maximum supported by driver version
    [[nodiscard]] MRVIEWER_API static int getCudaMaxDriverSupportedVersion();
 
    // Returns version of current runtime
    [[nodiscard]] MRVIEWER_API static int getCudaRuntimeVersion();
 
    // Returns Compute Capability major version of current device
    [[nodiscard]] MRVIEWER_API static int getComputeCapabilityMajor();
 
    // Returns Compute Capability minor version of current device
    [[nodiscard]] MRVIEWER_API static int getComputeCapabilityMinor();
 
    // Returns number of free bytes on cuda
    [[nodiscard]] MRVIEWER_API static size_t getCudaFreeMemory();
 
    // Returns cuda implementation of IFastWindingNumber
    [[nodiscard]] MRVIEWER_API static std::unique_ptr<IFastWindingNumber> getCudaFastWindingNumber( const Mesh& mesh );
 
    // Returns cuda implementation of IPointsToMeshProjector
    [[nodiscard]] MRVIEWER_API static std::unique_ptr<IPointsToMeshProjector> getCudaPointsToMeshProjector();
 
    // Returns cuda implementation of PointsToDistanceVolumeCallback
    [[nodiscard]] MRVIEWER_API static CudaPointsToDistanceVolumeCallback getCudaPointsToDistanceVolumeCallback();
 
    [[nodiscard]] MRVIEWER_API static size_t fastWindingNumberMeshMemory( const Mesh& mesh );
 
    [[nodiscard]] MRVIEWER_API static size_t fromGridMemory( const Mesh& mesh, const Vector3i& dims );
 
    [[nodiscard]] MRVIEWER_API static size_t fromVectorMemory( const Mesh& mesh, size_t inputSize );
 
    [[nodiscard]] MRVIEWER_API static size_t selfIntersectionsMemory( const Mesh& mesh );
private:
    CudaAccessor() = default;
    ~CudaAccessor() = default;
 
    static CudaAccessor& instance_();
 
    bool isCudaAvailable_ = false;
    int maxDriverVersion_ = 0;
    int runtimeVersion_ = 0;
    int computeMajor_ = 0;
    int computeMinor_ = 0;
    CudaFreeMemoryFunc freeMemFunc_;
    CudaFwnConstructor fwnCtor_;
    CudaMeshProjectorConstructor mpCtor_;
    CudaPointsToDistanceVolumeCallback pointsToDistanceVolumeCallback_;
};
 
} //namespace MR