FMCN Kinectハンズオン ステップ② BodyIndexを表示しよう

KinectController_1.cs

using UnityEngine;
using System.Collections;

// 必要な名前空間を宣言
using UnityEngine.UI;
using Windows.Kinect;

public class KinectController : MonoBehaviour
{
    // Kinect全体
    private KinectSensor sensor;
    private MultiSourceFrameReader reader;

    // BodyIndex関連
    private FrameDescription bodyIndexDescription; // BodyIndexフレームの情報
    private byte[] bodyIndexData; // BodyIndexのデータを保持するバッファ

    // 画像表示関連
    public RawImage rawImage;   // RawImageオブジェクト（publicにしてInspectorで関連付け）
    private Texture2D texture;  // 動的に生成するテクスチャ
    private byte[] textureData; // テクスチャに書き込む色データのバッファ

    // はじめに一度だけ実行　初期化処理
    void Start ()
    {
        // Kinectセンサーを取得
        sensor = KinectSensor.GetDefault();

        // 複数データを扱うためのリーダーを開く
        reader = sensor.OpenMultiSourceFrameReader(FrameSourceTypes.BodyIndex);

        // BodyIndexのフレーム情報を取得し、データを受け取るバッファを生成
        bodyIndexDescription = sensor.BodyIndexFrameSource.FrameDescription;
        bodyIndexData = new byte[bodyIndexDescription.Width * bodyIndexDescription.Height];

        // BodyIndexの解像度に合わせてテクスチャ、データバッファを生成
        texture = new Texture2D(bodyIndexDescription.Width, bodyIndexDescription.Height, TextureFormat.RGBA32, false);
        textureData = new byte[texture.width * texture.height * 4];

        // 生成したテクスチャをRawImageにセット
        rawImage.texture = texture;

        // センサーをオープン
        if(!sensor.IsOpen)
        {
            sensor.Open();
        }
　　}

    // オブジェクトが破棄されるときに実行　終了処理
    void OnDestroy()
    {
        // リーダーを終了
        if (reader != null)
        {
            reader.Dispose();
            reader = null;
        }

        // センサーを終了
        if (sensor != null)
        {
            sensor.Close();
            sensor = null;
        }
    }

    // 毎フレーム実行される　更新処理
    void Update ()
    {
	    if(reader != null)
        {
            // 最新フレームを取得
            var frame = reader.AcquireLatestFrame();
            if(frame != null)
            {
                // BodyIndexの最新フレームを取得
                using (var bodyIndexFrame = frame.BodyIndexFrameReference.AcquireFrame())
                {
                    if(bodyIndexFrame != null)
                    {
                        // BodyIndexデータをバッファにコピー
                        bodyIndexFrame.CopyFrameDataToArray(bodyIndexData);
                    }
                }
            }
        }

        // 各画素ごとにテクスチャの色を決定する
        for(int i=0; i<bodyIndexData.Length; ++i)
        {
            byte r, g, b, a;

            // BodyInexデータの値が255以外であれば人物領域
            if(bodyIndexData[i] != 255)
            {
                // aを255にして不透明に（ここでは赤にしている）
                r = 255;
                g = 0;
                b = 0;
                a = 255;
            }
            else
            {
                // aを0にして透明に
                r = 0;
                g = 0;
                b = 0;
                a = 0;
            }

            // テクスチャデータを更新する
            textureData[i * 4 + 0] = r;
            textureData[i * 4 + 1] = g;
            textureData[i * 4 + 2] = b;
            textureData[i * 4 + 3] = a;
        }

        // テクスチャデータをロードして適用する
        texture.LoadRawTextureData(textureData);
        texture.Apply();
    }
}