ExampleSegErrorHandling.cpp

This is an example of how to handle Segmentation's error code. More details about this example.

 
#include "SegmentationTestDLL.h"
 
#include <iostream>
#include <string>
 
 // User's error code
int SRErrorReturnCode = -1;
 
#define CheckSRSegErrors(status, handle, testResult)\
{                                                   \
    eErrorType e = status;                          \
    if (e != NO_ERROR_OUT)                          \
    {                                               \
        PrintErrorType(e);                          \
        SR_DestroySegTestHandle(handle);            \
        SR_DeleteSegTestResult(testResult);         \
        return SRErrorReturnCode;                   \
    }                                               \
}
 
int main()
{
    std::wstring data_dir = L"../../../examples/data";
 
    std::wstring model_path = data_dir + L"/models/pcb_sample_model.srSeg";
 
    SegTestHandle handle = SR_CreateSegTestHandleByWideCharPath(model_path.c_str());
    if (handle == nullptr)
    {
        std::cerr << "SaigeVision Segmentation Error: invalid file path" << std::endl;
        std::cin.get();
        return -1;
    }
 
    int gpu_index = -1;
 
    CheckSRSegErrors(SR_LoadSegTestNetworkInGPU(handle, gpu_index), handle, nullptr);
 
    SR_DestroySegTestHandle(handle);
 
    std::cout << "SaigeVision successfully completed, please press enter key to terminate" << std::endl;
    std::cin.get();
 
    return 0;
}