MRCudaAccessor.h

Go to the documentation of this file.

#pragma once
 
#include "MRViewerFwd.h"
#include "MRMesh/MRExpected.h"
 
#include <functional>
#include <memory>
 
#ifndef MRVIEWER_NO_VOXELS
#include "MRVoxels/MRVoxelsFwd.h"
#include "MRVoxels/MRSweptVolume.h"
#endif
 
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 CudaPointsProjectorConstructor = std::function<std::unique_ptr<IPointsProjector>()>;
 
#ifndef MRVIEWER_NO_VOXELS
    using CudaPointsToDistanceVolumeCallback = std::function<Expected<SimpleVolumeMinMax>( const PointCloud& cloud, const PointsToDistanceVolumeParams& params )>;
    using CudaPointsToDistanceVolumeByPartsCallback = std::function<Expected<void>( const PointCloud& cloud, const PointsToDistanceVolumeParams& params, std::function<Expected<void> ( const SimpleVolumeMinMax& volume, int zOffset )> addPart, int layerOverlap )>;
    using CudaComputeToolDistanceConstructor = std::function<std::unique_ptr<IComputeToolDistance>()>;
#endif
 
    // 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 setCudaPointsProjectorConstructor( CudaPointsProjectorConstructor ppCtor );
 
#ifndef MRVIEWER_NO_VOXELS
    MRVIEWER_API static void setCudaPointsToDistanceVolumeCallback( CudaPointsToDistanceVolumeCallback callback );
    MRVIEWER_API static void setCudaPointsToDistanceVolumeByPartsCallback( CudaPointsToDistanceVolumeByPartsCallback callback );
    MRVIEWER_API static void setCudaComputeToolDistanceConstructor( CudaComputeToolDistanceConstructor ctdCtor );
#endif
 
    // 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 IPointsProjector
    [[nodiscard]] MRVIEWER_API static std::unique_ptr<IPointsProjector> getCudaPointsProjector();
 
#ifndef MRVIEWER_NO_VOXELS
    // Returns cuda implementation of PointsToDistanceVolumeCallback
    [[nodiscard]] MRVIEWER_API static CudaPointsToDistanceVolumeCallback getCudaPointsToDistanceVolumeCallback();
 
    // Returns cuda implementation of PointsToDistanceVolumeByPartsCallback
    [[nodiscard]] MRVIEWER_API static CudaPointsToDistanceVolumeByPartsCallback getCudaPointsToDistanceVolumeByPartsCallback();
 
    [[nodiscard]] MRVIEWER_API static std::unique_ptr<IComputeToolDistance> getCudaComputeToolDistance();
#endif
 
    [[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 );
 
    [[nodiscard]] MRVIEWER_API static size_t pointsToDistanceVolumeMemory( const PointCloud& pointCloud, const Vector3i& dims, const VertNormals* ptNormals );
 
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_;
    CudaPointsProjectorConstructor ppCtor_;
#ifndef MRVIEWER_NO_VOXELS
    CudaPointsToDistanceVolumeCallback pointsToDistanceVolumeCallback_;
    CudaPointsToDistanceVolumeByPartsCallback pointsToDistanceVolumeByPartsCallback_;
    CudaComputeToolDistanceConstructor ctdCtor_;
#endif
};
 
} //namespace MR