check point
This commit is contained in:
@@ -0,0 +1,100 @@
|
||||
using OpenCvSharp;
|
||||
using System.Drawing;
|
||||
using Ninject;
|
||||
using VisionBuilder.UI.Common.RecipeProcessing;
|
||||
|
||||
namespace Hawkeye.VisionBuilder.UI.Sources.Emulation
|
||||
{
|
||||
public class EmulationCameraImageSource:IImageSource, IInitializable
|
||||
{
|
||||
private readonly EmulationSettings _settings;
|
||||
private readonly string _name;
|
||||
private string _currentFolder;
|
||||
private string[] _selectedFiles;
|
||||
|
||||
public EmulationCameraImageSource(EmulationSettings settings)
|
||||
{
|
||||
_settings = settings;
|
||||
}
|
||||
|
||||
public string GetName()
|
||||
{
|
||||
return _name;
|
||||
}
|
||||
|
||||
public async Task<Mat> GetImage(CancellationToken token)
|
||||
{
|
||||
if (_selectedFiles.Length == 0) return null;
|
||||
var id = (_index) % _selectedFiles.Length;
|
||||
if (id == 0)
|
||||
{
|
||||
if (_index > 0)
|
||||
OnLoopOver();
|
||||
if (_settings.SingleRun && _index > 0)
|
||||
{
|
||||
// infinite sleep
|
||||
await Task.Delay(Timeout.Infinite, token);
|
||||
}
|
||||
Console.WriteLine("starting new emulation cycle");
|
||||
await Task.Delay(_settings.CycleDelay, token);
|
||||
}
|
||||
else
|
||||
{
|
||||
await Task.Delay(_settings.PerImageDelay, token);
|
||||
}
|
||||
|
||||
var path = _selectedFiles[id];
|
||||
_index++;
|
||||
|
||||
|
||||
|
||||
Console.WriteLine($"Getting image {path}");
|
||||
|
||||
return Cv2.ImRead(path);
|
||||
}
|
||||
public void SetFolder(string imagesFolder)
|
||||
{
|
||||
try
|
||||
{
|
||||
_currentFolder = imagesFolder;
|
||||
var path = Path.Combine(@"..\Data\Emulation", imagesFolder);
|
||||
_selectedFiles = Directory.GetFiles(path, "*.bmp")
|
||||
.Concat(Directory.GetFiles(path, "*.png"))
|
||||
.Concat(Directory.GetFiles(path, "*.jpg"))
|
||||
.Where(x => !Path.GetFileNameWithoutExtension(x).Contains("_analysis"))
|
||||
.ToArray();
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
}
|
||||
|
||||
public void SetFolderAbsolute(string imagesFolder)
|
||||
{
|
||||
try
|
||||
{
|
||||
_currentFolder = imagesFolder;
|
||||
var path = imagesFolder;
|
||||
_selectedFiles = Directory.GetFiles(path, "*.bmp")
|
||||
.Concat(Directory.GetFiles(path, "*.png"))
|
||||
.Concat(Directory.GetFiles(path, "*.jpg"))
|
||||
.Where(x => !Path.GetFileNameWithoutExtension(x).Contains("_analysis"))
|
||||
.ToArray();
|
||||
}
|
||||
catch
|
||||
{
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
public event Action OnLoopOver = delegate { };
|
||||
private int _index = 0;
|
||||
|
||||
|
||||
public void Initialize()
|
||||
{
|
||||
if(string.IsNullOrEmpty(_settings.EmulationPath)) return;
|
||||
SetFolderAbsolute(_settings.EmulationPath);
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user