SurfaceDetect.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using sb_cs;
using Tutorials_Common;
 
namespace Tutorial9_SurfaceDetect
{
    class SurfaceDetect
    {
        static SbError execute()
        {
            SbProject prj = null;
            SbImage img = null;
            SbRoi roi = null;
            SbRes res = null;
            SbPar par = null;
            SbError err = SbError.SB_ERR_NONE;
 
            Console.WriteLine("Sb Init");
            err = Sb.Init("../../../sb.lic");
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("Sb.Init failed with error " + err);
                return err;
            }
 
 
            Console.WriteLine("Wait for license...");
            err = Common.WaitLicense();
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbLicense.WaitLicense failed with error " + err);
                return err;
            }
 
            String project_file = "../../solution/surface_tutorial." + SbDefines.SbSolutionExt;
            SbSolutionInfo solutionInfo = SbSolution.GetInfo(project_file);
            if (solutionInfo == null || solutionInfo.Error() != SbError.SB_ERR_NONE)
            {
                err = solutionInfo == null ? SbError.SB_ERR_INSUFFICIENT_FREE_MEMORY : solutionInfo.Error();
                Console.WriteLine("SbSolution.GetInfo failed");
                goto FnExit;
            }
            else if(solutionInfo.projects.Length == 0)
            {
                err = SbError.SB_ERR_PROJECT_NOT_FOUND;
                Console.WriteLine("SbSolution.GetInfo no projects found");
                goto FnExit;
 
            }
            Console.WriteLine("SbSolutionInfo.GetInfo, found " + solutionInfo.projects.Length + " projects");
 
            prj = SbProject.Load(project_file, solutionInfo.projects[0].uuid, SbProjectMode.SB_PROJECT_MODE_DETECTION_ONLY);
            if (prj == null || prj.Error() != SbError.SB_ERR_NONE)
            {
                err = prj == null ? SbError.SB_ERR_INTERNAL : prj.Error();
                Console.WriteLine("SbProject.Load failed with error " + err);
                goto FnExit;
            }
 
            img = SbImage.Load("../../dataset/surface_001.png");
            if (img == null || img.Error() != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbImage.Load failed");
                err = img == null ? SbError.SB_ERR_INTERNAL : img.Error();
                goto FnExit;
            }
 
            roi = SbRoi.Create(img.Width(), img.Height());
            if (roi == null || roi.Error() != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbRoi.Create failed");
                err = roi == null ? SbError.SB_ERR_INTERNAL : roi.Error();
                goto FnExit;
            }
 
            err = roi.SetRect(255, new SbRect(0, 0, img.Width(), img.Height()), false);
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbRoi.SetRect failed with error " + err);
                goto FnExit;
            }
 
            // Set the blob analysis par
            par = prj.GetPar();
            par.blob_par.area.min = 10;
            par.blob_par.merge_area.min = 20;
            par.blob_par.blob_rle = 1;
            par.blob_par.blob_contour = 1;
            par.blob_par.merge_distance = 10;
            par.surface_blob_analysis = 1;
 
            err = prj.SetPar(par);
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbProject.SetPar failed with error " + err);
                goto FnExit;
            }
 
            err = prj.Detection(img, roi);
            if (err != SbError.SB_ERR_NONE)
            {
                Console.WriteLine("SbProject.Detection failed with error " + err);
                goto FnExit;
            }            
 
            res = prj.GetRes(false);
            if (res == null || prj.Error() != SbError.SB_ERR_NONE)
            {
                err = res == null ? SbError.SB_ERR_INSUFFICIENT_FREE_MEMORY : prj.Error();
                Console.WriteLine("SbProject.GetRes failed with error " + err);
                goto FnExit;
            }
 
            // Print the results
            Console.WriteLine(res.global.surface.blobs.Length + " defect blobs found:", res.global.surface.blobs.Length);
            for (int j = 0; j < res.global.surface.blobs.Length; ++j)
                Console.WriteLine("    ul=(" + res.global.surface.blobs[j].rect.x + "," + res.global.surface.blobs[j].rect.y + "), size=(" + res.global.surface.blobs[j].rect.width + "," + res.global.surface.blobs[j].rect.height + "), area=" + res.global.surface.blobs[j].area);
 
FnExit:
            if (prj != null)
                prj.Dispose();
            if (img != null)
                img.Dispose();
            if (roi != null)
                roi.Dispose();
            if (par != null)
                par.Dispose();
            if (res != null)
                res.Dispose();
 
 
            Console.WriteLine("Release SqueezeBrains library");
            Sb.Release();
 
            Console.WriteLine("Press ENTER to terminate");
            Console.ReadKey();
 
            return err;
        }
 
        static void Main(string[] args)
        {
            execute();
        }
    }
}