RetinaDetect.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using sb_cs;
using Tutorials_Common;
 
namespace Tutorial3_RetinaDetect
{
    class RetinaDetect
    {
        static SbError execute()
        {
            SbProject prj = null;
            SbImage img = null;
            SbRoi roi = null;
            SbRes res = 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);
                goto FnExit;
            }
 
            String project_file = "../../../images/wheels." + 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("GetProjectsInfo, 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_INSUFFICIENT_FREE_MEMORY : prj.Error();
                Console.WriteLine("SbProject.Load failed with error " + err);
                goto FnExit;
            }
 
            SbFolder folder = SbFolder.Load("../../../images", "ppm", true, 0);
            if (folder != null)
            {
                for (int i = 0; i < folder.files.Length; i++)
                {
                    SbRect rect = new SbRect();
 
                    img = SbImage.Load(folder.files[i]);
                    if (img == null || img.Error() != SbError.SB_ERR_NONE)
                    {
                        Console.WriteLine("SbImage.Load " + folder.files[i] + " failed");
                        err = img == null ? SbError.SB_ERR_INTERNAL : img.Error();
                        break;
                    }
 
                    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();
                        break;
                    }
 
                    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);
                        break;
                    }
 
                    err = prj.Detection(img, roi);
                    if (err != SbError.SB_ERR_NONE)
                    {
                        Console.WriteLine("SbProject.Detection failed with error " + err);
                        break;
                    }
 
                    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);
                        break;
                    }
 
                    Console.WriteLine("image: ", folder.files[i]);
                    for (int j = 0; j < res.samples.Length; ++j)
                        Console.WriteLine("    centre=(" + res.samples[j].centre.x + "," + res.samples[j].centre.y + ") weight=" + res.samples[j].weight);
               
                    if (img != null)
                    {
                        img.Dispose();
                        img = null;
                    }                        
                    if (roi != null)
                    {
                        roi.Dispose();
                        roi = null;
                    }
                    if (res != null)
                    {
                        res.Dispose();
                        res = null;
                    }
                }
            }
 
        FnExit:
            if (prj != null)
                prj.Dispose();
            if (img != null)
                img.Dispose();
            if (roi != null)
                roi.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();
        }
    }
}