using System;
using System.Runtime.InteropServices;
using UnityEngine;

unsafe public class GenerateCloud : MonoBehaviour
{
    ParticleSystem ps;

    [StructLayout(LayoutKind.Sequential)]
    public struct float4
    {
        public float x;
        public float y;
        public float z;
        public float w;
    }

    [DllImport("k4a_cuda_win.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    static extern void init_cpc(bool color, bool body_tracking);

    [DllImport("k4a_cuda_win.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    static extern void get_capture();

    [DllImport("k4a_cuda_win.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    static extern void setup_point_cloud(ref float4[] point_positions, ref uint[] point_colors);

    [DllImport("k4a_cuda_win.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    static extern void get_point_cloud();

    [DllImport("k4a_cuda_win.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    static extern uint get_point_count();

    [DllImport("k4a_cuda_win.dll", CharSet = CharSet.Unicode, SetLastError = true)]
    static extern int get_max_point_count();

    float4[] point_positions;
    uint[] point_colors;

    void Start()
    {
        ps = GetComponent<ParticleSystem>();

        init_cpc(true, false);

        int max_points = get_max_point_count();

        point_positions = new float4[max_points];
        point_colors = new uint[max_points];

        GC.KeepAlive(point_positions);
        GC.KeepAlive(point_colors);

        setup_point_cloud(ref point_positions, ref point_colors);
    }

    void Update()
    {
        get_capture();
        get_point_cloud();
        uint point_count = get_point_count();

        for (int i = 0; i < point_count; i  )
        {
            byte[] color_bits = BitConverter.GetBytes(point_colors[i]);
            ParticleSystem.EmitParams ep = new ParticleSystem.EmitParams
            {
                position = new Vector3(point_positions[i].x, point_positions[i].y, point_positions[i].z),
                startColor = new Color32(color_bits[2], color_bits[1], color_bits[0], 0xFF)
            };

            ps.Emit(ep, 1);
        }
    }
}