bbiwarg-original/bbiwarg/InputProviders/InputProvider.cs

using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using Iisu;

namespace bbiwarg.InputProviders
{
    class InputProvider
    {
        public int ImageWidth { get; private set; }
        public int ImageHeight { get; private set; }
        public virtual int CurrentFrameID { get { return device.FrameId; } }
        public InputFrame CurrentFrame { get; private set; }

        protected IHandle handle;
        protected IDevice device;

        protected IDataHandle<Iisu.Data.IImageData> depthImage;
        protected IDataHandle<Iisu.Data.IImageData> confidenceImage;

        public void initialize()
        {
            createDevice();
            registerHandles();
        }

        protected void createDevice()
        {
            handle = Iisu.Iisu.Context.CreateHandle();
            IDeviceConfiguration conf = createDeviceConfiguration();
            device = handle.InitializeDevice(conf);
        }

        protected virtual IDeviceConfiguration createDeviceConfiguration()
        {
            return handle.CreateDeviceConfiguration();
        }

        protected virtual void registerHandles()
        {
            depthImage = device.RegisterDataHandle<Iisu.Data.IImageData>("SOURCE.CAMERA.DEPTH.Image");
            confidenceImage = device.RegisterDataHandle<Iisu.Data.IImageData>("SOURCE.CAMERA.CONFIDENCE.Image");
            device.EventManager.RegisterEventListener("DEVICE.DataFrame", new onNewFrameDelegate(onNewFrame));
        }

        private delegate void onNewFrameDelegate(String name, uint frameID);

        private void onNewFrame(String name, uint frameID)
        {
            device.UpdateFrame(false);
            IntPtr rawDepthData = depthImage.Value.Raw;
            IntPtr rawConfidenceData = confidenceImage.Value.Raw;
            device.ReleaseFrame();

            CurrentFrame = new InputFrame(CurrentFrameID, ImageWidth, ImageHeight, rawDepthData, rawConfidenceData);
        }

        public void start()
        {
            device.Start();
            device.UpdateFrame(true);
            ImageWidth = (int)depthImage.Value.ImageInfos.Width;
            ImageHeight = (int)depthImage.Value.ImageInfos.Height;
            device.ReleaseFrame();
        }

        public void stop()
        {
            device.Stop(true);
        }
    }
}