Kinect-Project/Sources/Lib/DepthImageStream.cs

using Microsoft.Kinect;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
using System.Windows.Media.Imaging;
using System.Windows.Media;
using System.Windows;

namespace Lib
{
    class DepthImageStream : KinectStream
    {
        private DepthFrameReader reader;
        public DepthFrameReader Reader
        {
            get { return reader; }
            set { reader = value; }
        }
        private byte[] depthPixels;
        public byte[] DepthPixels
        {
            get { return depthPixels; }
            set { depthPixels = value; }
        }

        public DepthImageStream(KinectManager kinectManager)
            : base(kinectManager)
        {
            var frameDescription = KinectManager.Sensor.DepthFrameSource.FrameDescription;
            this.Bitmap = new WriteableBitmap(
                frameDescription.Width,
                frameDescription.Height,
                96.0,
                96.0,
                PixelFormats.Gray8,
                null
            );
            reader = KinectManager.Sensor.DepthFrameSource.OpenReader();
            reader.FrameArrived += Reader_DepthFrameArrived;
            DepthPixels = new byte[frameDescription.Width * frameDescription.Height];
        }

        private void Reader_DepthFrameArrived(object sender, DepthFrameArrivedEventArgs e)
        {
            using (DepthFrame depthFrame = e.FrameReference.AcquireFrame())
            {
                if (depthFrame != null)
                {
                    FrameDescription depthFrameDescription = depthFrame.FrameDescription;

                    // Créez un tableau pour stocker les données de profondeur
                    ushort[] depthData = new ushort[depthFrameDescription.LengthInPixels];
                    depthFrame.CopyFrameDataToArray(depthData);

                    // Traitez les données de profondeur
                    ProcessDepthFrameData(
                        depthData,
                        depthFrameDescription.LengthInPixels,
                        depthFrame.DepthMinReliableDistance,
                        depthFrame.DepthMaxReliableDistance
                    );

                    // Mettez à jour le bitmap de profondeur
                    this.Bitmap.WritePixels(
                        new Int32Rect(
                            0,
                            0,
                            depthFrameDescription.Width,
                            depthFrameDescription.Height
                        ),
                        this.depthPixels,
                        depthFrameDescription.Width,
                        0
                    );
                }
            }
        }

        public override void Stop()
        {
            if (reader != null)
            {
                reader.Dispose();
                reader = null;
            }
        }

        private void ProcessDepthFrameData(
            ushort[] depthData,
            uint depthFrameDataSize,
            ushort minDepth,
            ushort maxDepth
        )
        {
            // Convertir les données de profondeur en niveaux de gris
            for (int i = 0; i < depthFrameDataSize; ++i)
            {
                ushort depth = depthData[i];
                DepthPixels[i] = (byte)(depth >= minDepth && depth <= maxDepth ? (depth % 256) : 0);
            }
        }
    }
}