dwelling act 4 (live motion cap w/ kinect azure)
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

2777 lines
102 KiB

1 year ago
using System;
using System.Collections;
using System.Collections.Generic;
using System.IO;
using System.Reflection;
using System.Runtime.InteropServices;
using System.Threading;
using UnityEngine;
namespace com.rfilkov.kinect
{
/// <summary>
/// KinectInterop is a class containing utility and interop functions, to call the proper sensor interface.
/// </summary>
public class KinectInterop
{
// whether to persist the space tables to disk and then load them from there, or not.
internal const bool SAVE_SPACE_TABLES_TO_DISK = true;
// multi-camera config file name
internal const string MULTI_CAM_CONFIG_FILE_NAME = "multicam_config.json";
// thread sleep time in milliseconds (to balance between cpu usage and performance)
internal const int THREAD_SLEEP_TIME_MS = 5;
/// <summary>
/// Constants used by this class and other K2-components
/// </summary>
public static class Constants
{
public const int MaxBodyCount = 50;
public const float MinTimeBetweenSameGestures = 0.0f;
public const float PoseCompleteDuration = 1.0f;
public const float ClickMaxDistance = 0.05f;
public const float ClickStayDuration = 2.0f;
}
/// <summary>
/// Depth sensor platforms.
/// </summary>
public enum DepthSensorPlatform : int
{
None = 0,
KinectV1 = 1,
KinectV2 = 2,
RealSense = 3,
Kinect4Azure = 4,
ARKit = 5,
WebCam = 50,
DummyK2 = 102,
DummyK4A = 104,
NetSensor = 110
}
/// <summary>
/// Device streaming modes.
/// </summary>
public enum DeviceStreamingMode
{
Disabled = 0,
ConnectedSensor = 1,
PlayRecording = 2,
SaveRecording = 3
}
// Data structures for interfacing C# with the native wrappers
/// <summary>
/// Frame-source flags.
/// </summary>
[Flags]
public enum FrameSource : uint
{
TypeNone = 0x0,
TypeColor = 0x1,
TypeInfrared = 0x2,
TypeDepth = 0x8,
TypeBodyIndex = 0x10,
TypeBody = 0x20,
TypeAudio = 0x40,
TypePose = 0x80,
TypeAll = 0xFF
}
/// <summary>
/// Body joint types (with new joints added)
/// </summary>
public enum JointType : int
{
Pelvis = 0,
SpineNaval = 1,
SpineChest = 2,
Neck = 3,
Head = 4,
ClavicleLeft = 5,
ShoulderLeft = 6,
ElbowLeft = 7,
WristLeft = 8,
HandLeft = 9,
ClavicleRight = 10,
ShoulderRight = 11,
ElbowRight = 12,
WristRight = 13,
HandRight = 14,
HipLeft = 15,
KneeLeft = 16,
AnkleLeft = 17,
FootLeft = 18,
HipRight = 19,
KneeRight = 20,
AnkleRight = 21,
FootRight = 22,
Nose = 23,
EyeLeft = 24,
EarLeft = 25,
EyeRight = 26,
EarRight = 27,
HandtipLeft = 28,
ThumbLeft = 29,
HandtipRight = 30,
ThumbRight = 31,
Count = 32
}
/// <summary>
/// Joint tracking state.
/// </summary>
public enum TrackingState
{
NotTracked = 0,
Inferred = 1,
Tracked = 2,
HighConf = 3 // reserved for future use
}
/// <summary>
/// Hand state.
/// </summary>
public enum HandState
{
Unknown = 0,
NotTracked = 1,
Open = 2,
Closed = 3,
Lasso = 4
}
/// <summary>
/// Container for the body-joint data.
/// </summary>
//[System.Serializable]
[StructLayout(LayoutKind.Sequential)]
public struct JointData
{
// parameters filled in by the sensor interface
public JointType jointType;
public TrackingState trackingState;
public Vector3 kinectPos;
public Vector3 position;
public Quaternion orientation;
public Vector3 posPrev;
public Vector3 posRel;
public Vector3 posVel;
// KM calculated parameters
public Vector3 direction;
public Quaternion normalRotation;
public Quaternion mirroredRotation;
public float lastAngle; // used by the constraints filter
public override string ToString()
{
return $"{jointType} @ {position}, rot: {normalRotation.eulerAngles.ToString("F0")}";
}
public void CopyTo(ref JointData toJoint)
{
toJoint.jointType = jointType;
toJoint.trackingState = trackingState;
toJoint.kinectPos.Set(kinectPos.x, kinectPos.y, kinectPos.z);
toJoint.position.Set(position.x, position.y, position.z);
toJoint.orientation.Set(orientation.x, orientation.y, orientation.z, orientation.w);
toJoint.posPrev.Set(posPrev.x, posPrev.y, posPrev.z);
toJoint.posRel.Set(posRel.x, posRel.y, posRel.z);
toJoint.posVel.Set(posVel.x, posVel.y, posVel.z);
toJoint.direction.Set(direction.x, direction.y, direction.z);
toJoint.normalRotation.Set(normalRotation.x, normalRotation.y, normalRotation.z, normalRotation.w);
toJoint.mirroredRotation.Set(mirroredRotation.x, mirroredRotation.y, mirroredRotation.z, mirroredRotation.w);
}
public void AverageTo(ref JointData toJoint)
{
//toJoint.jointType = jointType;
//toJoint.trackingState = trackingState;
//toJoint.kinectPos = (kinectPos + toJoint.kinectPos) * 0.5f;
//toJoint.position = (position + toJoint.position) * 0.5f;
toJoint.orientation = Quaternion.Slerp(orientation, toJoint.orientation, 0.5f);
//toJoint.posPrev = (posPrev + toJoint.posPrev) * 0.5f;
//toJoint.posRel = (posRel + toJoint.posRel) * 0.5f;
//toJoint.posVel = (posVel + toJoint.posVel) * 0.5f;
//toJoint.direction = (direction + toJoint.direction) * 0.5f;
toJoint.normalRotation = Quaternion.Slerp(normalRotation, toJoint.normalRotation, 0.5f);
toJoint.mirroredRotation = Quaternion.Slerp(mirroredRotation, toJoint.mirroredRotation, 0.5f);
}
public bool IsTracked()
{
return trackingState >= TrackingState.Tracked;
}
}
/// <summary>
/// Container for the body data.
/// </summary>
//[System.Serializable]
[StructLayout(LayoutKind.Sequential)]
public struct BodyData
{
// parameters filled in by the sensor interface
public ulong liTrackingID;
public int iBodyIndex;
public bool bIsTracked;
public Vector3 kinectPos;
public Vector3 position;
public Quaternion orientation;
[MarshalAs(UnmanagedType.ByValArray, SizeConst = (int)JointType.Count)]
public JointData[] joint;
// KM calculated parameters
public Quaternion normalRotation;
public Quaternion mirroredRotation;
// body merger parameter
public int sensorIndex;
// last body timestamp
public ulong bodyTimestamp;
// hand state parameters
public HandState leftHandState;
//public TrackingConfidence leftHandConfidence;
public HandState rightHandState;
//public TrackingConfidence rightHandConfidence;
public BodyData(int jointCount)
{
liTrackingID = 0;
iBodyIndex = 0;
bIsTracked = false;
kinectPos = Vector3.zero;
position = Vector3.zero;
orientation = Quaternion.identity;
normalRotation = Quaternion.identity;
mirroredRotation = Quaternion.identity;
sensorIndex = 0;
bodyTimestamp = 0;
leftHandState = HandState.NotTracked;
rightHandState = HandState.NotTracked;
joint = new JointData[jointCount];
for(int j = 0; j < jointCount; j++)
{
joint[j].jointType = (JointType)j;
joint[j].trackingState = TrackingState.NotTracked;
joint[j].kinectPos = Vector3.zero;
joint[j].position = Vector3.zero;
joint[j].orientation = Quaternion.identity;
joint[j].normalRotation = Quaternion.identity;
joint[j].mirroredRotation = Quaternion.identity;
}
}
public override string ToString()
{
return $"Body{iBodyIndex} id: {liTrackingID} @ {position.ToString("F2")}, rot: {normalRotation.eulerAngles.ToString("F0")}";
}
public void CopyTo(ref BodyData toBody)
{
toBody.liTrackingID = liTrackingID;
toBody.iBodyIndex = iBodyIndex;
toBody.bIsTracked = bIsTracked;
toBody.kinectPos.Set(kinectPos.x, kinectPos.y, kinectPos.z);
toBody.position.Set(position.x, position.y, position.z);
toBody.orientation.Set(orientation.x, orientation.y, orientation.z, orientation.w);
if (toBody.joint == null || toBody.joint.Length != joint.Length)
{
toBody.joint = new JointData[joint.Length];
}
for (int j = 0; j < joint.Length; j++)
{
joint[j].CopyTo(ref toBody.joint[j]);
}
toBody.normalRotation.Set(normalRotation.x, normalRotation.y, normalRotation.z, normalRotation.w);
toBody.mirroredRotation.Set(mirroredRotation.x, mirroredRotation.y, mirroredRotation.z, mirroredRotation.w);
toBody.sensorIndex = sensorIndex;
toBody.bodyTimestamp = bodyTimestamp;
toBody.leftHandState = leftHandState;
toBody.rightHandState = rightHandState;
}
}
/// <summary>
/// Sensor device info.
/// </summary>
public class SensorDeviceInfo
{
public string sensorId;
public string sensorName;
public FrameSource sensorCaps;
}
/// <summary>
/// Camera distortion type
/// </summary>
public enum DistortionType
{
None = 0,
ModifiedBrownConrady = 1,
InverseBrownConrady = 2,
Theta = 3,
BrownConrady = 4,
Polynomial3K = 5,
Rational6KT = 6
}
/// <summary>
/// Camera intrinsics
/// </summary>
[System.Serializable]
public class CameraIntrinsics
{
public int cameraType; // camera type (0=depth, 1=color)
public int width; // Camera image width
public int height; // Camera image height
public float ppx; // Principal point in image, x
public float ppy; // Principal point in image, y
public float fx; // Focal length x
public float fy; // Focal length y
public DistortionType distType; // distortion type
public float[] distCoeffs; // radial distortion coefficient
public float codx; // Center of distortion in Z=1 plane, x (only used for Rational6KT)
public float cody; // Center of distortion in Z=1 plane, y (only used for Rational6KT)
public float p2; // Tangential distortion coefficient 2
public float p1; // Tangential distortion coefficient 1
public float maxRadius; // Metric radius
public float hFOV, vFOV;
}
/// <summary>
/// Camera extrinsics.
/// </summary>
[System.Serializable]
public class CameraExtrinsics
{
public float[] rotation;
public float[] translation;
}
/// <summary>
/// Intrinsics & extrinsics of the sensor
/// </summary>
[System.Serializable]
public class SensorIntrExtr
{
public CameraIntrinsics depthCamIntr;
public CameraIntrinsics colorCamIntr;
public CameraExtrinsics depth2ColorExtr;
public CameraExtrinsics color2DepthExtr;
}
/// <summary>
/// Container for the sensor data, including color, depth, ir and body frames.
/// </summary>
public class SensorData
{
public DepthSensorInterface sensorInterface;
public DepthSensorPlatform sensorIntPlatform;
public string sensorId;
public string sensorName;
public FrameSource sensorCaps;
public long startTimeOffset;
public int sensorIndex;
public Thread pollFramesThread = null;
public AutoResetEvent threadStopEvent = null;
public CameraIntrinsics depthCamIntr = null;
public CameraIntrinsics colorCamIntr = null;
public CameraExtrinsics depth2ColorExtr = null;
public CameraExtrinsics color2DepthExtr = null;
public Vector3 colorImageScale = Vector3.one;
public Vector3 depthImageScale = Vector3.one;
public Vector3 infraredImageScale = Vector3.one;
public Vector3 sensorSpaceScale = Vector3.one;
public float unitToMeterFactor = 1f;
public int colorImageWidth;
public int colorImageHeight;
public TextureFormat colorImageFormat = TextureFormat.RGBA32;
public int colorImageStride = 4;
//public byte[] colorImage;
public Texture colorImageTexture = null;
public ulong lastColorFrameTime = 0;
public ulong prevColorFrameTime = 0;
public int depthImageWidth;
public int depthImageHeight;
public ushort[] depthImage;
public ulong lastDepthFrameTime = 0;
public ulong prevDepthFrameTime = 0;
public ushort[] infraredImage;
public ulong lastInfraredFrameTime = 0;
public ulong prevInfraredFrameTime = 0;
public Vector3 sensorPosePosition;
public Quaternion sensorPoseRotation;
public ulong lastSensorPoseFrameTime = 0;
//public bool sensorTransformUpdated = false;
public byte[] bodyIndexImage;
public ulong lastBodyIndexFrameTime = 0;
public uint trackedBodiesCount = 0;
public BodyData[] alTrackedBodies;
public ulong lastBodyFrameTime = 0;
//public int firstUserIndex = 255;
//public int[] depthHistBufferData;
//public int depthHistTotalPoints;
//public ulong lastDepthHistTime = 0;
public RenderTexture depthImageTexture;
public Material depthImageMaterial;
public ComputeBuffer depthImageBuffer;
public ComputeBuffer depthHistBuffer;
public ulong lastDepthImageTime = 0;
public RenderTexture infraredImageTexture;
public Material infraredImageMaterial;
public ComputeBuffer infraredImageBuffer;
public ulong lastInfraredImageTime = 0;
//public int[] bodyHistBufferData;
//public int bodyHistTotalPoints;
//public ulong lastBodyHistTime = 0;
public RenderTexture bodyImageTexture;
public Material bodyImageMaterial;
public ComputeBuffer bodyIndexBuffer;
public ComputeBuffer bodyHistBuffer;
public ulong lastBodyImageTime = 0;
// public Vector3[] depth2SpaceFrame = null;
public ulong lastDepth2SpaceFrameTime = 0;
// public Vector2[] depth2ColorFrame = null;
public Texture depthCamColorImageTexture = null;
public ulong lastDepthCamColorFrameTime = 0;
// public Vector2[] color2DepthFrame = null;
public ushort[] colorCamDepthImage;
public ulong lastColorCamDepthFrameTime = 0;
public ushort[] colorCamInfraredImage;
public ulong lastColorCamInfraredFrameTime = 0;
public byte[] colorCamBodyIndexImage;
public ulong lastColorCamBodyIndexFrameTime = 0;
public ComputeBuffer colorDepthBuffer;
public RenderTexture colorDepthTexture;
public ulong lastColorDepthBufferTime = 0;
//public ulong usedColorDepthBufferTime = 0;
public ComputeBuffer colorInfraredBuffer;
public RenderTexture colorInfraredTexture;
public ulong lastColorInfraredBufferTime = 0;
//public ulong usedColorInfraredBufferTime = 0;
public Material colorInfraredMaterial;
public ComputeBuffer colorBodyIndexBuffer;
public RenderTexture colorBodyIndexTexture;
public ulong lastColorBodyIndexBufferTime = 0;
//public ulong usedColorBodyIndexBufferTime = 0;
public RenderTexture depthTexTexture;
public Material depthTexMaterial;
public ulong lastDepthTexTime = 0;
public RenderTexture infraredTexTexture;
public Material infraredTexMaterial;
public ulong lastInfraredTexTime = 0;
}
[System.Serializable]
public class NetSensorData
{
public int sensorType;
public string sensorId;
public int colorImageWidth;
public int colorImageHeight;
public int depthImageWidth;
public int depthImageHeight;
public CameraIntrinsics depthCamIntr = null;
public CameraIntrinsics colorCamIntr = null;
public CameraExtrinsics depth2ColorExtr = null;
public CameraExtrinsics color2DepthExtr = null;
public Vector3 colorImageScale = Vector3.one;
public Vector3 depthImageScale = Vector3.one;
public Vector3 infraredImageScale = Vector3.one;
public Vector3 sensorSpaceScale = Vector3.one;
public float unitToMeterFactor = 1f;
public Vector3 sensorPosition = Vector3.zero;
public Vector3 sensorRotation = Vector3.zero;
public Vector3 sensorRotOffset = Vector3.zero;
public bool sensorRotFlipZ = false;
public bool sensorRotIgnoreY = false;
}
[System.Serializable]
public class NetPoseData
{
public Vector3 sensorPosition;
public Vector3 sensorRotation;
//public Vector3 localPos;
//public Vector3 localRot;
public ulong sensorPoseTime;
}
[System.Serializable]
public class CameraPose
{
public int sensorType;
public int sensorIndex;
public string sensorId;
public Vector3 position;
public Vector3 rotation;
}
[System.Serializable]
public class MultiCameraPose
{
public int version;
public string[] settings;
public CameraPose[] camPose;
public long estimatedAtTime;
public string estimatedDateTime;
}
// graphics shader level
private static int graphicsShaderLevel = 0;
// persistent data path
private static string persistentDataPath = string.Empty;
// sets the graphics shader level
public static void SetGraphicsShaderLevel(int shaderLevel)
{
graphicsShaderLevel = shaderLevel;
persistentDataPath = Application.persistentDataPath;
}
// checks if DirectX11/Direct3D-11 is turned on or not
public static bool IsDirectX11Available()
{
return (graphicsShaderLevel >= 50);
}
// returns true if the project is running on 64-bit architecture, false if 32-bit
public static bool Is64bitArchitecture()
{
int sizeOfPtr = Marshal.SizeOf(typeof(IntPtr));
return (sizeOfPtr > 4);
}
// returns the text contained in the given resource asset, or null if not found
public static string GetResourceText(string resFileName)
{
TextAsset textRes = Resources.Load(resFileName, typeof(TextAsset)) as TextAsset;
if (textRes == null)
{
Debug.LogWarning("Resource not found: " + resFileName);
return null;
}
return textRes.text;
}
// returns the binary data contained in the given resource asset, or null if not found
public static byte[] GetResourceBytes(string resFileName)
{
TextAsset textRes = Resources.Load(resFileName, typeof(TextAsset)) as TextAsset;
if (textRes == null)
{
Debug.LogWarning("Resource not found: " + resFileName);
return null;
}
return textRes.bytes;
}
// copy resource asset to the target file
public static bool CopyResourceFile(string resFileName, string targetFilePath)
{
TextAsset textRes = Resources.Load(resFileName, typeof(TextAsset)) as TextAsset;
if (textRes == null)
{
Debug.LogWarning("Resource not found: " + resFileName);
return false;
}
FileInfo targetFile = new FileInfo(targetFilePath);
if (!targetFile.Directory.Exists)
{
targetFile.Directory.Create();
}
if (!targetFile.Exists || targetFile.Length != textRes.bytes.Length)
{
Debug.Log("Copying '" + resFileName + "' to: " + targetFilePath);
using (Stream resStream = new MemoryStream(textRes.bytes))
{
BinaryReader resReader = new BinaryReader(resStream);
byte[] buffer = new byte[32768]; //set the size of your buffer (chunk)
using (FileStream fileStream = new FileStream(targetFilePath, FileMode.Create, FileAccess.Write, FileShare.Read))
{
while (true) //loop to the end of the file
{
int read = resReader.Read(buffer, 0, buffer.Length);
if (read <= 0) //check for end of file
break;
fileStream.Write(buffer, 0, read);
}
}
}
bool bFileCopied = File.Exists(targetFilePath);
return bFileCopied;
}
return true;
}
// checks if the file exists
public static bool IsFolderExist(string folderPath)
{
return Directory.Exists(folderPath);
}
// checks if the file exists
public static bool IsFileExist(string filePath)
{
return File.Exists(filePath);
}
// copy file from the source folder to the target folder
public static bool CopyFolderFile(string sourceFolderPath, string fileName, string targetFolderPath, bool hideWarning = false)
{
//if(!Directory.Exists(sourceFolderPath))
//{
// //Debug.Log("Folder not found: " + folderPath);
// return false;
//}
if(!string.IsNullOrEmpty(sourceFolderPath) && sourceFolderPath[sourceFolderPath.Length - 1] != '/' && sourceFolderPath[sourceFolderPath.Length - 1] != '\\')
{
sourceFolderPath += "/";
}
string sourceFilePath = sourceFolderPath + fileName;
FileInfo sourceFile = new FileInfo(sourceFilePath);
if (!sourceFile.Exists)
{
if(!hideWarning)
Debug.LogWarning("File not found: " + sourceFilePath /**+ ". Trying to find it in resources..."*/);
return false;
}
if (!string.IsNullOrEmpty(targetFolderPath) && sourceFolderPath[targetFolderPath.Length - 1] != '/' && sourceFolderPath[targetFolderPath.Length - 1] != '\\')
{
targetFolderPath += "/";
}
string targetFilePath = targetFolderPath + fileName;
FileInfo targetFile = new FileInfo(targetFilePath);
if (!Directory.Exists(targetFolderPath))
{
Directory.CreateDirectory(targetFolderPath);
}
if (!targetFile.Exists || targetFile.Length != sourceFile.Length || targetFile.LastWriteTime != sourceFile.LastWriteTime)
{
Debug.Log("Copying '" + fileName + "' from " + sourceFolderPath + " to '" + targetFolderPath + "'");
File.Copy(sourceFilePath, targetFilePath, true);
bool bFileCopied = File.Exists(targetFilePath);
return bFileCopied;
}
return true;
}
// creates new render texture with the given dimensions and format
public static RenderTexture CreateRenderTexture(RenderTexture currentTex, int width, int height, RenderTextureFormat texFormat = RenderTextureFormat.Default)
{
if(currentTex != null)
{
currentTex.Release();
//UnityEngine.Object.Destroy(currentTex);
}
RenderTexture renderTex = new RenderTexture(width, height, 0, texFormat);
renderTex.wrapMode = TextureWrapMode.Clamp;
renderTex.filterMode = FilterMode.Point;
renderTex.enableRandomWrite = true;
return renderTex;
}
// creates new compute buffer with the given length and stride
public static ComputeBuffer CreateComputeBuffer(ComputeBuffer currentBuf, int bufLen, int bufStride)
{
if(currentBuf != null)
{
currentBuf.Release();
currentBuf.Dispose();
}
ComputeBuffer computeBuf = new ComputeBuffer(bufLen, bufStride);
return computeBuf;
}
// initializes the secondary sensor data, after sensor initialization
public static void InitSensorData(SensorData sensorData, KinectManager kinectManager)
{
// init depth texture
if (IsDirectX11Available() && sensorData.depthImage != null &&
kinectManager.getDepthFrames == KinectManager.DepthTextureType.DepthTexture)
{
Shader depthImageShader = Shader.Find("Kinect/DepthHistImageShader");
if (depthImageShader != null)
{
if (sensorData.depthImageTexture == null || sensorData.depthImageTexture.width != sensorData.depthImageWidth || sensorData.depthImageTexture.height != sensorData.depthImageHeight)
{
sensorData.depthImageTexture = CreateRenderTexture(sensorData.depthImageTexture, sensorData.depthImageWidth, sensorData.depthImageHeight);
}
sensorData.depthImageMaterial = new Material(depthImageShader);
if(sensorData.depthImageBuffer == null)
{
int depthBufferLength = sensorData.depthImageWidth * sensorData.depthImageHeight / 2;
sensorData.depthImageBuffer = CreateComputeBuffer(sensorData.depthImageBuffer, depthBufferLength, sizeof(uint));
}
if(sensorData.depthHistBuffer == null)
{
sensorData.depthHistBuffer = CreateComputeBuffer(sensorData.depthHistBuffer, DepthSensorBase.MAX_DEPTH_DISTANCE_MM + 1, sizeof(int));
}
}
}
// init infrared texture
if (IsDirectX11Available() && sensorData.infraredImage != null &&
kinectManager.getInfraredFrames == KinectManager.InfraredTextureType.InfraredTexture)
{
Shader infraredImageShader = Shader.Find("Kinect/InfraredImageShader");
if (infraredImageShader != null)
{
if (sensorData.infraredImageTexture == null || sensorData.infraredImageTexture.width != sensorData.depthImageWidth || sensorData.infraredImageTexture.height != sensorData.depthImageHeight)
{
sensorData.infraredImageTexture = CreateRenderTexture(sensorData.infraredImageTexture, sensorData.depthImageWidth, sensorData.depthImageHeight);
}
sensorData.infraredImageMaterial = new Material(infraredImageShader);
if (sensorData.infraredImageBuffer == null)
{
int infraredBufferLength = sensorData.depthImageWidth * sensorData.depthImageHeight / 2;
sensorData.infraredImageBuffer = CreateComputeBuffer(sensorData.infraredImageBuffer, infraredBufferLength, sizeof(uint));
}
}
}
// init body texture
if (IsDirectX11Available() && sensorData.bodyIndexImage != null &&
(kinectManager.getBodyFrames == KinectManager.BodyTextureType.BodyTexture ||
kinectManager.getBodyFrames == KinectManager.BodyTextureType.UserTexture))
{
Shader bodyImageShader = null;
switch(kinectManager.getBodyFrames)
{
case KinectManager.BodyTextureType.BodyTexture:
bodyImageShader = Shader.Find("Kinect/UserBodyImageShader");
break;
case KinectManager.BodyTextureType.UserTexture:
bodyImageShader = Shader.Find("Kinect/UserHistImageShader");
break;
}
if (bodyImageShader != null)
{
if (sensorData.bodyImageTexture == null || sensorData.bodyImageTexture.width != sensorData.depthImageWidth || sensorData.bodyImageTexture.height != sensorData.depthImageHeight)
{
sensorData.bodyImageTexture = CreateRenderTexture(sensorData.bodyImageTexture, sensorData.depthImageWidth, sensorData.depthImageHeight);
}
sensorData.bodyImageMaterial = new Material(bodyImageShader);
if (sensorData.bodyIndexBuffer == null)
{
int bodyIndexBufferLength = sensorData.depthImageWidth * sensorData.depthImageHeight / 4;
sensorData.bodyIndexBuffer = CreateComputeBuffer(sensorData.bodyIndexBuffer, bodyIndexBufferLength, sizeof(uint));
}
if (sensorData.depthImageBuffer == null)
{
int depthBufferLength = sensorData.depthImageWidth * sensorData.depthImageHeight / 2;
sensorData.depthImageBuffer = CreateComputeBuffer(sensorData.depthImageBuffer, depthBufferLength, sizeof(uint));
}
if (sensorData.bodyHistBuffer == null)
{
sensorData.bodyHistBuffer = CreateComputeBuffer(sensorData.bodyHistBuffer, DepthSensorBase.MAX_DEPTH_DISTANCE_MM + 1, sizeof(int));
}
}
}
// invoke the sensor interface to init its proprietary data
if (sensorData.sensorInterface != null)
{
sensorData.sensorInterface.InitSensorData(sensorData, kinectManager);
}
}
// closes the sensor and releases the related buffers
public static void CloseSensor(SensorData sensorData)
{
if (sensorData == null)
return;
//FinishBackgroundRemoval(sensorData);
if (sensorData.sensorInterface != null)
{
sensorData.sensorInterface.CloseSensor(sensorData);
}
if (sensorData.depthImageTexture)
{
sensorData.depthImageTexture.Release();
sensorData.depthImageTexture = null;
}
if (sensorData.depthImageBuffer != null)
{
sensorData.depthImageBuffer.Release();
sensorData.depthImageBuffer.Dispose();
sensorData.depthImageBuffer = null;
}
if (sensorData.depthHistBuffer != null)
{
sensorData.depthHistBuffer.Release();
sensorData.depthHistBuffer.Dispose();
sensorData.depthHistBuffer = null;
}
if (sensorData.infraredImageTexture)
{
sensorData.infraredImageTexture.Release();
sensorData.infraredImageTexture = null;
}
if (sensorData.infraredImageBuffer != null)
{
sensorData.infraredImageBuffer.Release();
sensorData.infraredImageBuffer.Dispose();
sensorData.infraredImageBuffer = null;
}
if (sensorData.bodyImageTexture)
{
sensorData.bodyImageTexture.Release();
sensorData.bodyImageTexture = null;
}
if (sensorData.bodyIndexBuffer != null)
{
sensorData.bodyIndexBuffer.Release();
sensorData.bodyIndexBuffer.Dispose();
sensorData.bodyIndexBuffer = null;
}
if (sensorData.bodyHistBuffer != null)
{
sensorData.bodyHistBuffer.Release();
sensorData.bodyHistBuffer.Dispose();
sensorData.bodyHistBuffer = null;
}
}
// infoked by the sensor thread to poll for frames
public static void PollSensorFrames(SensorData sensorData)
{
if (sensorData != null && sensorData.sensorInterface != null)
{
sensorData.sensorInterface.PollSensorFrames(sensorData);
sensorData.sensorInterface.PollCoordTransformFrames(sensorData);
sensorData.sensorInterface.PollSensorFrameTimes(sensorData);
}
}
// invoked periodically to update sensor data, if needed
public static bool UpdateSensorData(SensorData sensorData, KinectManager kinectManager, bool isPlayMode)
{
bool bResult = false;
if (sensorData != null && sensorData.sensorInterface != null)
{
sensorData.prevDepthFrameTime = sensorData.lastDepthFrameTime;
sensorData.prevInfraredFrameTime = sensorData.lastInfraredFrameTime;
sensorData.prevColorFrameTime = sensorData.lastColorFrameTime;
bResult = sensorData.sensorInterface.UpdateSensorData(sensorData, kinectManager, isPlayMode);
if((sensorData.lastDepthFrameTime != sensorData.prevDepthFrameTime ||
sensorData.lastColorFrameTime != sensorData.prevColorFrameTime) && !isPlayMode)
{
sensorData.sensorInterface.UpdateTransformedFrameTextures(sensorData, kinectManager);
}
}
return bResult;
}
// invoked periodically to update sensor textures, as needed
public static bool UpdateSensorTextures(SensorData sensorData, KinectManager kinectManager)
{
bool bResult = false;
if (sensorData != null && sensorData.sensorInterface != null)
{
bResult = sensorData.sensorInterface.UpdateSensorTextures(sensorData, kinectManager,
sensorData.prevDepthFrameTime, sensorData.prevInfraredFrameTime);
}
return bResult;
}
// displays the given texture on the screen
public static void DisplayGuiTexture(int displayIndex, float screenWidthPercent, Vector2 imageScale, Texture imageTex)
{
if (imageTex == null || imageTex.width == 0 || imageTex.height == 0)
return;
// get the screen width & height
float screenW = (float)Screen.width;
float screenH = (float)Screen.height;
float rectWidthPercent = screenWidthPercent;
float rectHeightPercent = rectWidthPercent * (float)imageTex.height / (float)imageTex.width;
float rectWidth = screenW * rectWidthPercent;
float rectHeight = screenW * rectHeightPercent;
float rectX = screenW - (displayIndex + 1) * rectWidth;
float rectY = screenH - rectHeight;
if (rectX < 0 || rectY < 0)
return;
if (imageScale.x < 0)
{
rectX = screenW - displayIndex * rectWidth;
rectWidth = -rectWidth;
}
if (imageScale.y < 0)
{
rectY = screenH;
rectHeight = -rectHeight;
}
Rect imageRect = new Rect(rectX, rectY, rectWidth, rectHeight);
if (imageTex != null)
{
GUI.DrawTexture(imageRect, imageTex);
}
}
// returns the respective sensor-to-world matrix
public static Matrix4x4 GetSensorToWorldMatrix(SensorData sensorData)
{
if (sensorData != null && sensorData.sensorInterface != null)
{
return sensorData.sensorInterface.GetSensorToWorldMatrix();
}
return Matrix4x4.identity;
}
// returns sensor transform. Please note transform updates depend on the getPoseFrames-KM setting.
public static Transform GetSensorTransform(SensorData sensorData)
{
if (sensorData != null && sensorData.sensorInterface != null)
{
return sensorData.sensorInterface.GetSensorTransform();
}
return null;
}
// returns depth camera space coordinates for the given depth-image point
public static Vector3 MapDepthPointToSpaceCoords(SensorData sensorData, Vector2 depthPos, ushort depthVal)
{
Vector3 vPoint = Vector3.zero;
if (sensorData != null && sensorData.sensorInterface != null)
{
vPoint = sensorData.sensorInterface.MapDepthPointToSpaceCoords(sensorData, depthPos, depthVal);
}
return vPoint;
}
// returns depth image coordinates for the given depth camera space point
public static Vector2 MapSpacePointToDepthCoords(SensorData sensorData, Vector3 spacePos)
{
Vector2 vPoint = Vector2.zero;
if (sensorData != null && sensorData.sensorInterface != null)
{
vPoint = sensorData.sensorInterface.MapSpacePointToDepthCoords(sensorData, spacePos);
}
return vPoint;
}
// returns color camera space coordinates for the given color-image point
public static Vector3 MapColorPointToSpaceCoords(SensorData sensorData, Vector2 colorPos, ushort distance)
{
Vector3 vPoint = Vector3.zero;
if (sensorData != null && sensorData.sensorInterface != null)
{
vPoint = sensorData.sensorInterface.MapColorPointToSpaceCoords(sensorData, colorPos, distance);
}
return vPoint;
}
// returns color image coordinates for the given color camera space point
public static Vector2 MapSpacePointToColorCoords(SensorData sensorData, Vector3 spacePos)
{
Vector2 vPoint = Vector2.zero;
if (sensorData != null && sensorData.sensorInterface != null)
{
vPoint = sensorData.sensorInterface.MapSpacePointToColorCoords(sensorData, spacePos);
}
return vPoint;
}
// returns color-image coordinates for the given depth-image point
public static Vector2 MapDepthPointToColorCoords(SensorData sensorData, Vector2 depthPos, ushort depthVal)
{
Vector2 vPoint = Vector2.zero;
if (sensorData != null && sensorData.sensorInterface != null)
{
vPoint = sensorData.sensorInterface.MapDepthPointToColorCoords(sensorData, depthPos, depthVal);
}
return vPoint;
}
// returns depth-image coordinates for the given color-image point
public static Vector2 MapColorPointToDepthCoords(SensorData sensorData, Vector2 colorPos, int minDist, int maxDist)
{
Vector2 vPoint = Vector2.zero;
if (sensorData != null && sensorData.sensorInterface != null)
{
vPoint = sensorData.sensorInterface.MapColorPointToDepthCoords(sensorData, colorPos, minDist, maxDist);
}
return vPoint;
}
//// estimates depth-map coordinates for the given color coords
//public static Vector2 MapColorPointToDepthCoords(SensorData sensorData, Vector2 colorPos, bool bReadDepthCoordsIfNeeded)
//{
// Vector2 vPoint = Vector2.zero;
// if (sensorData != null && sensorData.sensorInterface != null && !float.IsInfinity(colorPos.x) && !float.IsInfinity(colorPos.y))
// {
// int cIndex = (int)colorPos.y * sensorData.colorImageWidth + (int)colorPos.x;
// //if (sensorData.color2DepthFrame != null)
// //{
// // if (cIndex >= 0 && cIndex < sensorData.color2DepthFrame.Length)
// // {
// // vPoint = sensorData.color2DepthFrame[cIndex];
// // }
// //}
// //else
// if (bReadDepthCoordsIfNeeded)
// {
// Vector2[] vDepthCoords = new Vector2[sensorData.colorImageWidth * sensorData.colorImageHeight];
// if (MapColorFrameToDepthCoords(sensorData, ref vDepthCoords))
// {
// if (cIndex >= 0 && cIndex < vDepthCoords.Length)
// {
// vPoint = vDepthCoords[cIndex];
// }
// }
// vDepthCoords = null;
// }
// }
// return vPoint;
//}
//// estimates space coordinates for the current depth frame
//public static bool MapDepthFrameToSpaceCoords(SensorData sensorData, ref Vector3[] vSpaceCoords)
//{
// bool bResult = false;
// if (sensorData != null && sensorData.sensorInterface != null)
// {
// bResult = sensorData.sensorInterface.MapDepthFrameToSpaceCoords(sensorData, ref vSpaceCoords);
// }
// return bResult;
//}
//// estimates color-map coordinates for the current depth frame
//public static bool MapDepthFrameToColorCoords(SensorData sensorData, ref Vector2[] vColorCoords)
//{
// bool bResult = false;
// if (sensorData != null && sensorData.sensorInterface != null)
// {
// bResult = sensorData.sensorInterface.MapDepthFrameToColorCoords(sensorData, ref vColorCoords);
// }
// return bResult;
//}
//// estimates depth-map coordinates for the current color frame
//public static bool MapColorFrameToDepthCoords(SensorData sensorData, ref Vector2[] vDepthCoords)
//{
// bool bResult = false;
// if (sensorData != null && sensorData.sensorInterface != null)
// {
// bResult = sensorData.sensorInterface.MapColorFrameToDepthCoords(sensorData, ref vDepthCoords);
// }
// return bResult;
//}
private static readonly Dictionary<JointType, JointType> dictParentJoint = new Dictionary<KinectInterop.JointType, JointType>
{
{JointType.Pelvis, JointType.Pelvis},
{JointType.SpineNaval, JointType.Pelvis},
{JointType.SpineChest, JointType.SpineNaval},
{JointType.Neck, JointType.SpineChest},
{JointType.Head, JointType.Neck},
{JointType.ClavicleLeft, JointType.SpineChest},
{JointType.ShoulderLeft, JointType.ClavicleLeft},
{JointType.ElbowLeft, JointType.ShoulderLeft},
{JointType.WristLeft, JointType.ElbowLeft},
{JointType.HandLeft, JointType.WristLeft},
{JointType.HandtipLeft, JointType.HandLeft},
{JointType.ThumbLeft, JointType.WristLeft},
{JointType.ClavicleRight, JointType.SpineChest},
{JointType.ShoulderRight, JointType.ClavicleRight},
{JointType.ElbowRight, JointType.ShoulderRight},
{JointType.WristRight, JointType.ElbowRight},
{JointType.HandRight, JointType.WristRight},
{JointType.HandtipRight, JointType.HandRight},
{JointType.ThumbRight, JointType.WristRight},
{JointType.HipLeft, JointType.Pelvis},
{JointType.KneeLeft, JointType.HipLeft},
{JointType.AnkleLeft, JointType.KneeLeft},
{JointType.FootLeft, JointType.AnkleLeft},
{JointType.HipRight, JointType.Pelvis},
{JointType.KneeRight, JointType.HipRight},
{JointType.AnkleRight, JointType.KneeRight},
{JointType.FootRight, JointType.AnkleRight},
{JointType.Nose, JointType.Head},
{JointType.EyeLeft, JointType.Nose},
{JointType.EarLeft, JointType.EyeLeft},
{JointType.EyeRight, JointType.Nose},
{JointType.EarRight, JointType.EyeRight},
};
// returns the parent joint of the given joint
public static JointType GetParentJoint(JointType joint)
{
return dictParentJoint[joint];
}
private static readonly Dictionary<JointType, JointType> dictNextJoint = new Dictionary<KinectInterop.JointType, JointType>
{
{JointType.Pelvis, JointType.SpineNaval},
{JointType.SpineNaval, JointType.SpineChest},
{JointType.SpineChest, JointType.Neck},
{JointType.Neck, JointType.Head},
{JointType.Head, JointType.Nose},
{JointType.ClavicleLeft, JointType.ShoulderLeft},
{JointType.ShoulderLeft, JointType.ElbowLeft},
{JointType.ElbowLeft, JointType.WristLeft},
{JointType.WristLeft, JointType.HandLeft},
{JointType.HandLeft, JointType.HandtipLeft},
{JointType.HandtipLeft, JointType.HandtipLeft},
{JointType.ThumbLeft, JointType.ThumbLeft},
{JointType.ClavicleRight, JointType.ShoulderRight},
{JointType.ShoulderRight, JointType.ElbowRight},
{JointType.ElbowRight, JointType.WristRight},
{JointType.WristRight, JointType.HandRight},
{JointType.HandRight, JointType.HandtipRight},
{JointType.HandtipRight, JointType.HandtipRight},
{JointType.ThumbRight, JointType.ThumbRight},
{JointType.HipLeft, JointType.KneeLeft},
{JointType.KneeLeft, JointType.AnkleLeft},
{JointType.AnkleLeft, JointType.FootLeft},
{JointType.FootLeft, JointType.FootLeft},
{JointType.HipRight, JointType.KneeRight},
{JointType.KneeRight, JointType.AnkleRight},
{JointType.AnkleRight, JointType.FootRight},
{JointType.FootRight, JointType.FootRight},
{JointType.Nose, JointType.Nose},
{JointType.EyeLeft, JointType.EarLeft},
{JointType.EarLeft, JointType.EarLeft},
{JointType.EyeRight, JointType.EarRight},
{JointType.EarRight, JointType.EarRight},
};
// returns the next joint in the hierarchy, as to the given joint
public static JointType GetNextJoint(JointType joint)
{
return dictNextJoint[joint];
}
private static readonly Dictionary<JointType, JointType> dictMirrorJoint = new Dictionary<KinectInterop.JointType, JointType>
{
{JointType.Pelvis, JointType.Pelvis},
{JointType.SpineNaval, JointType.SpineNaval},
{JointType.SpineChest, JointType.SpineChest},
{JointType.Neck, JointType.Neck},
{JointType.Head, JointType.Head},
{JointType.ClavicleLeft, JointType.ClavicleRight},
{JointType.ShoulderLeft, JointType.ShoulderRight},
{JointType.ElbowLeft, JointType.ElbowRight},
{JointType.WristLeft, JointType.WristRight},
{JointType.HandLeft, JointType.HandRight},
{JointType.HandtipLeft, JointType.HandtipRight},
{JointType.ThumbLeft, JointType.ThumbRight},
{JointType.ClavicleRight, JointType.ClavicleLeft},
{JointType.ShoulderRight, JointType.ShoulderLeft},
{JointType.ElbowRight, JointType.ElbowLeft},
{JointType.WristRight, JointType.WristLeft},
{JointType.HandRight, JointType.HandLeft},
{JointType.HandtipRight, JointType.HandtipLeft},
{JointType.ThumbRight, JointType.ThumbLeft},
{JointType.HipLeft, JointType.HipRight},
{JointType.KneeLeft, JointType.KneeRight},
{JointType.AnkleLeft, JointType.AnkleRight},
{JointType.FootLeft, JointType.FootRight},
{JointType.HipRight, JointType.HipLeft},
{JointType.KneeRight, JointType.KneeLeft},
{JointType.AnkleRight, JointType.AnkleLeft},
{JointType.FootRight, JointType.FootLeft},
{JointType.Nose, JointType.Nose},
{JointType.EyeLeft, JointType.EyeRight},
{JointType.EarLeft, JointType.EarRight},
{JointType.EyeRight, JointType.EyeLeft},
{JointType.EarRight, JointType.EarLeft},
};
// returns the mirror joint of the given joint
public static JointType GetMirrorJoint(JointType joint)
{
return dictMirrorJoint[joint];
}
// memcpy declaration
[UnmanagedFunctionPointer(CallingConvention.Cdecl)]
public delegate IntPtr MemCpyDelegate(IntPtr dest, IntPtr src, int count);
internal static readonly MemCpyDelegate Memcpy = MemCpy.GetMethod();
[DllImport("libc", EntryPoint = "memcpy", CallingConvention = CallingConvention.Cdecl, SetLastError = false)]
internal static extern IntPtr unix_memcpy(IntPtr dest, IntPtr src, int count);
[DllImport("msvcrt.dll", EntryPoint = "memcpy", CallingConvention = CallingConvention.Cdecl, SetLastError = false)]
internal static extern IntPtr win_memcpy(IntPtr dest, IntPtr src, int count);
[System.Security.SuppressUnmanagedCodeSecurity]
internal static class MemCpy
{
internal static MemCpyDelegate GetMethod()
{
switch (System.Environment.OSVersion.Platform)
{
case PlatformID.Win32NT:
return win_memcpy;
case PlatformID.Unix:
case PlatformID.MacOSX:
return unix_memcpy;
default:
throw new PlatformNotSupportedException(System.Environment.OSVersion.ToString());
}
}
}
// copies the given number of bytes from source to destination
public static void CopyBytes(IntPtr srcPtr, IntPtr dstPtr, int byteCount)
{
if (srcPtr == IntPtr.Zero)
throw new Exception("srcPtr should not be null.");
if (dstPtr == IntPtr.Zero)
throw new Exception("dstPtr should not be null.");
if (byteCount == 0)
throw new Exception("byteCount (" + byteCount + ") should be a positive number.");
Memcpy(dstPtr, srcPtr, byteCount);
}
// copies the source array to the destination array
public static void CopyBytes(Array srcArray, int copyElemCount, int srcElemSize, IntPtr dstPtr, int dstBlockSize)
{
if (srcArray == null)
throw new Exception("srcArray should not be null.");
if (dstPtr == IntPtr.Zero)
throw new Exception("dstPtr should not be null.");
int copyBytesCount = copyElemCount * srcElemSize;
if (copyBytesCount > dstBlockSize)
throw new Exception("Copied " + copyBytesCount + " bytes will overflow the destination memory. MemBlock size: " + dstBlockSize + " bytes");
if (copyBytesCount > 0)
{
var pSrcData = GCHandle.Alloc(srcArray, GCHandleType.Pinned);
Memcpy(dstPtr, pSrcData.AddrOfPinnedObject(), copyBytesCount);
pSrcData.Free();
}
}
// copies the source array to the destination array
public static void CopyBytes(IntPtr srcPtr, int srcBlockSize, Array dstArray, int copyElemCount, int dstElemSize)
{
if (srcPtr == IntPtr.Zero)
throw new Exception("srcPtr should not be null.");
if (dstArray == null)
throw new Exception("dstArray should not be null.");
int copyBytesCount = copyElemCount * dstElemSize;
if (srcBlockSize < copyBytesCount)
throw new Exception(copyBytesCount + " bytes are not available in the source memory. Size: " + srcBlockSize);
if (copyBytesCount > 0)
{
var pDstData = GCHandle.Alloc(dstArray, GCHandleType.Pinned);
Memcpy(pDstData.AddrOfPinnedObject(), srcPtr, copyBytesCount);
pDstData.Free();
}
}
// copies the source array to the destination array
public static void CopyBytes(Array srcArray, int srcElemSize, Array dstArray, int dstElemSize)
{
if (srcArray == null)
throw new Exception("srcArray should not be null.");
if (dstArray == null)
throw new Exception("dstArray should not be null.");
if ((srcArray.Length * srcElemSize) != (dstArray.Length * dstElemSize))
throw new Exception(string.Format("srcArray (L:{0}*{1}) and dstArray (L:{2}*{3}) should have the same byte length.", srcArray.Length, srcElemSize, dstArray.Length, dstElemSize));
int copyBytesCount = dstArray.Length * dstElemSize;
if (copyBytesCount > 0)
{
var pSrcData = GCHandle.Alloc(srcArray, GCHandleType.Pinned);
var pDstData = GCHandle.Alloc(dstArray, GCHandleType.Pinned);
Memcpy(pDstData.AddrOfPinnedObject(), pSrcData.AddrOfPinnedObject(), copyBytesCount);
pDstData.Free();
pSrcData.Free();
}
}
//// copies the source array to the destination array
//public static void CopyBytes<T>(T srcStruct, ref T dstStruct)
//{
// if (srcStruct == null)
// throw new Exception("srcStruct should not be null.");
// if (dstStruct == null)
// throw new Exception("dstStruct should not be null.");
// int copyBytesCount = Marshal.SizeOf(typeof(T));
// Debug.Log("Copy " + copyBytesCount + " bytes");
// if (copyBytesCount > 0)
// {
// var pSrcData = GCHandle.Alloc(srcStruct, GCHandleType.Pinned);
// var pDstData = GCHandle.Alloc(dstStruct, GCHandleType.Pinned);
// CopyBytes(pSrcData.AddrOfPinnedObject(), pDstData.AddrOfPinnedObject(), copyBytesCount);
// pDstData.Free();
// pSrcData.Free();
// }
//}
// borrowed from @keijiro
private static MethodInfo _setNativeDataMethod;
private static object[] _setNativeDataArgs = new object[5];
// sets compute buffer data
public static void SetComputeBufferData(ComputeBuffer computeBuffer, IntPtr dataPointer, int elemCount, int elemSize)
{
if (_setNativeDataMethod == null)
{
_setNativeDataMethod = typeof(ComputeBuffer).GetMethod("InternalSetNativeData",
BindingFlags.InvokeMethod | BindingFlags.NonPublic | BindingFlags.Instance);
}
_setNativeDataArgs[0] = dataPointer;
_setNativeDataArgs[1] = 0; // source offset
_setNativeDataArgs[2] = 0; // buffer offset
_setNativeDataArgs[3] = elemCount;
_setNativeDataArgs[4] = elemSize;
_setNativeDataMethod.Invoke(computeBuffer, _setNativeDataArgs);
}
// sets compute buffer data
public static void SetComputeBufferData(ComputeBuffer computeBuffer, Array data, int elemCount, int elemSize)
{
var pData = GCHandle.Alloc(data, GCHandleType.Pinned);
SetComputeBufferData(computeBuffer, pData.AddrOfPinnedObject(), elemCount, elemSize);
pData.Free();
}
private static int[] _csIntArgs2 = new int[2];
// sets compute shader int2 params
public static void SetComputeShaderInt2(ComputeShader computeShader, string name, int x, int y)
{
_csIntArgs2[0] = x;
_csIntArgs2[1] = y;
computeShader.SetInts(name, _csIntArgs2);
}
private static float[] _shFloatArgs2 = new float[2];
// sets compute shader float2 params
public static void SetComputeShaderFloat2(ComputeShader computeShader, string name, float x, float y)
{
_shFloatArgs2[0] = x;
_shFloatArgs2[1] = y;
computeShader.SetFloats(name, _shFloatArgs2);
}
// sets compute shader float2 params
public static void SetComputeShaderFloat2(ComputeShader computeShader, string name, Vector2 v)
{
_shFloatArgs2[0] = v.x;
_shFloatArgs2[1] = v.y;
computeShader.SetFloats(name, _shFloatArgs2);
}
private static float[] _shFloatArgs3 = new float[3];
// sets compute shader float3 params
public static void SetComputeShaderFloat3(ComputeShader computeShader, string name, float x, float y, float z)
{
_shFloatArgs3[0] = x;
_shFloatArgs3[1] = y;
_shFloatArgs3[2] = z;
computeShader.SetFloats(name, _shFloatArgs3);
}
// sets compute shader float3 params
public static void SetComputeShaderFloat3(ComputeShader computeShader, string name, Vector3 v)
{
_shFloatArgs3[0] = v.x;
_shFloatArgs3[1] = v.y;
_shFloatArgs3[2] = v.z;
computeShader.SetFloats(name, _shFloatArgs3);
}
// sets compute shader float2 params
public static void SetMatShaderFloat2(Material mat, string name, Vector2 vFloat2)
{
mat.SetVector(name, vFloat2);
}
// sets compute shader float2 params
public static void SetMatShaderFloat2(Material mat, string name, float x, float y)
{
Vector2 vFloat2 = new Vector2(x, y);
mat.SetVector(name, vFloat2);
}
// reads render texture contents into tex2d (it must have the same width and height).
public static bool RenderTex2Tex2D(RenderTexture rt, ref Texture2D tex)
{
if (!rt || !tex || rt.width != tex.width || rt.height != tex.height)
return false;
RenderTexture currentActiveRT = RenderTexture.active;
RenderTexture.active = rt;
tex.ReadPixels(new Rect(0, 0, tex.width, tex.height), 0, 0);
tex.Apply();
RenderTexture.active = currentActiveRT;
return true;
}
// reads render texture contents into tex2d (it must have the same width and height).
public static bool RenderTex2Tex2D(RenderTexture rt, int rtX, int rtY, int rtW, int rtH, ref Texture2D tex)
{
if (!rt || !tex || rtW != tex.width || rtH != tex.height)
return false;
RenderTexture currentActiveRT = RenderTexture.active;
RenderTexture.active = rt;
tex.ReadPixels(new Rect(rtX, rtY, rtW, rtH), 0, 0);
tex.Apply();
RenderTexture.active = currentActiveRT;
return true;
}
private static Material matRender = null;
// sets up the render material, if needed
private static void SetRenderMat()
{
if (!matRender)
{
Shader shader = Shader.Find("Hidden/Internal-Colored");
matRender = new Material(shader);
matRender.hideFlags = HideFlags.HideAndDontSave;
matRender.SetInt("_SrcBlend", (int)UnityEngine.Rendering.BlendMode.SrcAlpha);
matRender.SetInt("_DstBlend", (int)UnityEngine.Rendering.BlendMode.OneMinusSrcAlpha);
matRender.SetInt("_Cull", (int)UnityEngine.Rendering.CullMode.Off);
matRender.SetInt("_ZWrite", 0);
}
}
// draws point with the given size and color
public static void DrawPoint(int x, int y, float size, Color color)
{
Vector3 vPoint = new Vector3(x, y, 0);
DrawPoint(vPoint, size, color);
}
// draws point with the given size and color
public static void DrawPoint(Vector3 vPoint, float quadSize, Color color)
{
if (!matRender)
{
SetRenderMat();
}
GL.PushMatrix();
matRender.SetPass(0);
GL.LoadPixelMatrix();
GL.Begin(GL.QUADS);
GL.Color(color);
_DrawPoint(vPoint, quadSize);
GL.End();
GL.PopMatrix();
}
// draws list of points with the given size and color
public static void DrawPoints(List<Vector3> alPoints, float quadSize, Color color)
{
if (alPoints == null)
return;
if (!matRender)
{
SetRenderMat();
}
GL.PushMatrix();
matRender.SetPass(0);
GL.LoadPixelMatrix();
GL.Begin(GL.QUADS);
GL.Color(color);
foreach (Vector3 v in alPoints)
{
_DrawPoint(v, quadSize);
}
GL.End();
GL.PopMatrix();
}
// draws point with given size
private static void _DrawPoint(Vector3 v, float quadSize)
{
float q2 = quadSize / 2f;
GL.Vertex3(v.x - q2, v.y - q2, 0f);
GL.Vertex3(v.x - q2, v.y + q2, 0f);
GL.Vertex3(v.x + q2, v.y + q2, 0f);
GL.Vertex3(v.x + q2, v.y - q2, 0f);
}
// draws a line with the given width and color
public static void DrawLine(int x0, int y0, int x1, int y1, float width, Color color)
{
Vector3 v0 = new Vector3(x0, y0, 0);
Vector3 v1 = new Vector3(x1, y1, 0);
DrawLine(v0, v1, width, color);
}
// draws a line with the given width and color
public static void DrawLine(Vector3 v0, Vector3 v1, float lineWidth, Color color)
{
if (!matRender)
{
SetRenderMat();
}
GL.PushMatrix();
matRender.SetPass(0);
GL.LoadPixelMatrix();
GL.Begin(GL.QUADS);
GL.Color(color);
_DrawLine(v0, v1, lineWidth);
GL.End();
GL.PopMatrix();
}
// draws list of lines with the given width and color
public static void DrawLines(List<Vector3> alLinePoints, float lineWidth, Color color)
{
if (alLinePoints == null)
return;
if (!matRender)
{
SetRenderMat();
}
GL.PushMatrix();
matRender.SetPass(0);
GL.LoadPixelMatrix();
GL.Begin(GL.QUADS);
GL.Color(color);
for (int i = 0; i < alLinePoints.Count; i += 2)
{
Vector3 v0 = alLinePoints[i];
Vector3 v1 = alLinePoints[i + 1];
_DrawLine(v0, v1, lineWidth);
}
GL.End();
GL.PopMatrix();
}
// draws rectangle with the given width and color
public static void DrawRect(Rect rect, float width, Color color)
{
Vector3 topLeft = new Vector3(rect.xMin, rect.yMin, 0);
Vector3 bottomRight = new Vector3(rect.xMax, rect.yMax, 0);
DrawRect(topLeft, bottomRight, width, color);
}
// draws rectangle with the given width and color
public static void DrawRect(Vector3 topLeft, Vector3 bottomRight, float lineWidth, Color color)
{
if (!matRender)
{
SetRenderMat();
}
GL.PushMatrix();
matRender.SetPass(0);
GL.LoadPixelMatrix();
GL.Begin(GL.QUADS);
GL.Color(color);
// top
Vector3 v0 = topLeft;
Vector3 v1 = topLeft; v1.x = bottomRight.x;
_DrawLine(v0, v1, lineWidth);
// right
v0 = v1;
v1 = bottomRight;
_DrawLine(v0, v1, lineWidth);
// bottom
v0 = v1;
v1 = topLeft; v1.y = bottomRight.y;
_DrawLine(v0, v1, lineWidth);
// left
v0 = v1;
v1 = topLeft;
_DrawLine(v0, v1, lineWidth);
GL.End();
GL.PopMatrix();
}
// draws line from v0 to v1 with the given width
private static void _DrawLine(Vector3 v0, Vector3 v1, float lineWidth)
{
Vector3 n = ((new Vector3(v1.y, v0.x, 0f)) - (new Vector3(v0.y, v1.x, 0f))).normalized * lineWidth;
GL.Vertex3(v0.x - n.x, v0.y - n.y, 0f);
GL.Vertex3(v0.x + n.x, v0.y + n.y, 0f);
GL.Vertex3(v1.x + n.x, v1.y + n.y, 0f);
GL.Vertex3(v1.x - n.x, v1.y - n.y, 0f);
}
// converts current body frame to a single csv line. returns empty string if there is no body data
public static string GetBodyFrameAsCsv(ref BodyData[] bodyData, uint bodyCount, ulong liRelTime, Vector3 spaceScale, ref float fUnityTime, char delimiter)
{
// check for invalid sensor data and if the frame is still the same
if (bodyData == null)
return string.Empty;
// create the output string
System.Text.StringBuilder sbBuf = new System.Text.StringBuilder();
System.Globalization.CultureInfo invCulture = System.Globalization.CultureInfo.InvariantCulture;
sbBuf.Append("k4b").Append(delimiter);
sbBuf.Append(liRelTime).Append(delimiter);
fUnityTime = Time.time;
int jointCount = (int)JointType.Count;
sbBuf.Append(bodyCount).Append(delimiter);
sbBuf.Append(jointCount).Append(delimiter);
sbBuf.Append((int)spaceScale.x).Append(delimiter);
sbBuf.Append((int)spaceScale.y).Append(delimiter);
sbBuf.Append((int)spaceScale.z).Append(delimiter);
// add information for all bodies
for (int i = 0; i < bodyCount; i++)
{
sbBuf.Append(bodyData[i].bIsTracked ? "1" : "0").Append(delimiter);
if (bodyData[i].bIsTracked)
{
// add information for the tracked body - body-id and joints
sbBuf.Append(bodyData[i].liTrackingID).Append(delimiter);
for (int j = 0; j < jointCount; j++)
{
JointData jointData = bodyData[i].joint[j];
sbBuf.Append((int)jointData.trackingState).Append(delimiter);
if (jointData.trackingState != TrackingState.NotTracked)
{
Vector3 jointPos = jointData.kinectPos;
Vector3 jointRot = jointData.normalRotation.eulerAngles;
sbBuf.AppendFormat(invCulture, "{0:F3}", jointPos.x).Append(delimiter);
sbBuf.AppendFormat(invCulture, "{0:F3}", jointPos.y).Append(delimiter);
sbBuf.AppendFormat(invCulture, "{0:F3}", jointPos.z).Append(delimiter);
sbBuf.AppendFormat(invCulture, "{0:F3}", jointRot.x).Append(delimiter);
sbBuf.AppendFormat(invCulture, "{0:F3}", jointRot.y).Append(delimiter);
sbBuf.AppendFormat(invCulture, "{0:F3}", jointRot.z).Append(delimiter);
}
}
}
}
// remove the last delimiter
if (sbBuf.Length > 0 && sbBuf[sbBuf.Length - 1] == delimiter)
{
sbBuf.Remove(sbBuf.Length - 1, 1);
}
return sbBuf.ToString();
}
// sets current body frame from the given csv line. returns the number of bodies
public static uint SetBodyFrameFromCsv(string sCsvLine, string sDelim, SensorData sensorData, ref BodyData[] bodyData,
ref Matrix4x4 sensorToWorld, bool bIgnoreZCoords, out ulong liRelTime)
{
liRelTime = 0;
// check for invalid sensor data and for same frame time
if (sCsvLine.Length == 0 || sensorData == null)
return 0;
// split the csv line in parts
char[] delimiters = sDelim.ToCharArray();
if (delimiters == null || delimiters.Length == 0)
return 0;
string[] alCsvParts = sCsvLine.Split(delimiters);
if (alCsvParts.Length < 7)
return 0;
System.Globalization.CultureInfo invCulture = System.Globalization.CultureInfo.InvariantCulture;
System.Globalization.NumberStyles numFloat = System.Globalization.NumberStyles.Float;
// body count & joint count
int.TryParse(alCsvParts[2], out int bodyCount);
int.TryParse(alCsvParts[3], out int jointCount);
// space scale
int.TryParse(alCsvParts[4], out int iSX);
int.TryParse(alCsvParts[5], out int iSY);
int.TryParse(alCsvParts[6], out int iSZ);
Vector3 sensorSpaceScale = new Vector3(iSX, iSY, iSZ);
sensorData.sensorSpaceScale = sensorSpaceScale;
ulong.TryParse(alCsvParts[1], out liRelTime);
if (alCsvParts[0] != "k4b" || bodyCount == 0 || jointCount == 0 || liRelTime == 0)
return 0;
//if (bodyCount != sensorData.bodyCount || jointCount != sensorData.jointCount)
// return 0;
//// update body frame data
//bodyFrame.liPreviousTime = bodyFrame.liRelativeTime;
//bodyFrame.liRelativeTime = liRelTime;
if (bodyData.Length < bodyCount)
{
Array.Resize<BodyData>(ref bodyData, bodyCount);
for (int i = 0; i < bodyCount; i++)
{
bodyData[i] = new BodyData((int)JointType.Count);
}
}
int iIndex = 7;
for (int i = 0; i < bodyCount; i++)
{
if (alCsvParts.Length < (iIndex + 1))
continue;
// update the tracked-flag and body id
short bIsTracked = 0;
ulong liTrackingID = 0;
short.TryParse(alCsvParts[iIndex], out bIsTracked);
iIndex++;
if (bIsTracked != 0 && alCsvParts.Length >= (iIndex + 1))
{
ulong.TryParse(alCsvParts[iIndex], out liTrackingID);
iIndex++;
if (liTrackingID == 0)
{
bIsTracked = 0;
}
}
bodyData[i].bIsTracked = (bIsTracked != 0);
bodyData[i].liTrackingID = liTrackingID;
if (bIsTracked != 0)
{
if (jointCount != (int)JointType.Count)
{
// set the other joints as not tracked
for (int j = jointCount; j < (int)JointType.Count; j++)
{
bodyData[i].joint[j].trackingState = TrackingState.NotTracked;
}
}
// update joints' data
for (int j = 0; j < jointCount; j++)
{
JointData jointData = bodyData[i].joint[j];
int iTrackingState = 0;
if (alCsvParts.Length >= (iIndex + 1))
{
int.TryParse(alCsvParts[iIndex], out iTrackingState);
iIndex++;
jointData.trackingState = (KinectInterop.TrackingState)iTrackingState;
if (iTrackingState != (int)TrackingState.NotTracked && alCsvParts.Length >= (iIndex + 6))
{
float.TryParse(alCsvParts[iIndex], numFloat, invCulture, out float x);
float.TryParse(alCsvParts[iIndex + 1], numFloat, invCulture, out float y);
float.TryParse(alCsvParts[iIndex + 2], numFloat, invCulture, out float z);
iIndex += 3;
jointData.kinectPos = new Vector3(x, y, z);
float.TryParse(alCsvParts[iIndex], numFloat, invCulture, out x);
float.TryParse(alCsvParts[iIndex + 1], numFloat, invCulture, out y);
float.TryParse(alCsvParts[iIndex + 2], numFloat, invCulture, out z);
iIndex += 3;
jointData.normalRotation = Quaternion.Euler(x, y, z);
}
else
{
jointData.kinectPos = Vector3.zero;
jointData.normalRotation = Quaternion.identity;
}
float jPosZ = (bIgnoreZCoords && j > 0) ? bodyData[i].joint[0].kinectPos.z : jointData.kinectPos.z;
jointData.position = sensorToWorld.MultiplyPoint3x4(new Vector3(jointData.kinectPos.x * sensorSpaceScale.x, jointData.kinectPos.y * sensorSpaceScale.y, jPosZ));
jointData.orientation = Quaternion.identity;
Vector3 mirroredAngles = jointData.normalRotation.eulerAngles;
mirroredAngles.y = -mirroredAngles.y;
mirroredAngles.z = -mirroredAngles.z;
jointData.mirroredRotation = Quaternion.Euler(mirroredAngles);
if (j == 0)
{
// set body position
bodyData[i].kinectPos = jointData.kinectPos;
bodyData[i].position = jointData.position;
bodyData[i].orientation = jointData.orientation;
bodyData[i].normalRotation = jointData.normalRotation;
bodyData[i].mirroredRotation = jointData.mirroredRotation;
//Debug.Log("User " + i + " kPos: " + jointData.kinectPos + ", pos: " + jointData.position);
}
}
bodyData[i].joint[j] = jointData;
}
// calculate bone directions
CalcBodyJointDirs(ref bodyData[i]);
}
}
return (uint)bodyCount;
}
// sets current body frame from the given csv line. returns true on success, false otherwise
public static uint SetBodyFrameFromK2b(string sCsvLine, SensorData sensorData, KinectManager kinectManager, ref BodyData[] bodyData,
ref Matrix4x4 kinectToWorld, bool bIgnoreZCoords, out ulong liRelTime)
{
liRelTime = 0;
// check for invalid sensor data and for same frame time
if (sensorData == null || kinectManager == null)
return 0;
if (sCsvLine.Length == 0)
return 0;
// split the csv line in parts
char[] delimiters = { ',' };
string[] alCsvParts = sCsvLine.Split(delimiters);
if (alCsvParts.Length < 4)
return 0;
System.Globalization.CultureInfo invCulture = System.Globalization.CultureInfo.InvariantCulture;
System.Globalization.NumberStyles numFloat = System.Globalization.NumberStyles.Float;
// check the id, body count & joint count
int bodyCount = 0, jointCount = 0;
int.TryParse(alCsvParts[2], out bodyCount);
int.TryParse(alCsvParts[3], out jointCount);
ulong.TryParse(alCsvParts[1], out liRelTime);
sensorData.sensorSpaceScale = Vector3.one; // the space scale of K2
if (alCsvParts[0] != "kb" || bodyCount == 0 || jointCount == 0 || liRelTime == 0)
return 0;
// if(bodyCount != sensorData.bodyCount || jointCount != sensorData.jointCount)
// return false;
//// update body frame data
//bodyFrame.liPreviousTime = bodyFrame.liRelativeTime;
//bodyFrame.liRelativeTime = liRelTime;
if (bodyData.Length < bodyCount)
{
Array.Resize<BodyData>(ref bodyData, bodyCount);
for (int i = 0; i < bodyCount; i++)
{
bodyData[i] = new BodyData((int)JointType.Count);
}
}
int iIndex = 4;
for (int i = 0; i < bodyCount; i++)
{
if (alCsvParts.Length < (iIndex + 1))
continue;
// update the tracked-flag and body id
short bIsTracked = 0;
ulong liTrackingID = 0;
short.TryParse(alCsvParts[iIndex], out bIsTracked);
iIndex++;
if (bIsTracked != 0 && alCsvParts.Length >= (iIndex + 1))
{
ulong.TryParse(alCsvParts[iIndex], out liTrackingID);
iIndex++;
if (liTrackingID == 0)
{
bIsTracked = 0;
}
}
bodyData[i].bIsTracked = (bIsTracked != 0);
bodyData[i].liTrackingID = liTrackingID;
if (bIsTracked != 0)
{
// set all joints as not tracked
for (int j = 0; j < (int)JointType.Count; j++)
{
bodyData[i].joint[j].trackingState = TrackingState.NotTracked;
}
// update joints' data
for (int j2b = 0; j2b < jointCount; j2b++)
{
int j = K2bJoint2JointType[j2b];
JointData jointData = bodyData[i].joint[j];
int iTrackingState = 0;
if (alCsvParts.Length >= (iIndex + 1))
{
int.TryParse(alCsvParts[iIndex], out iTrackingState);
iIndex++;
jointData.trackingState = (KinectInterop.TrackingState)iTrackingState;
if (iTrackingState != (int)TrackingState.NotTracked && alCsvParts.Length >= (iIndex + 3))
{
float x = 0f, y = 0f, z = 0f;
float.TryParse(alCsvParts[iIndex], numFloat, invCulture, out x);
float.TryParse(alCsvParts[iIndex + 1], numFloat, invCulture, out y);
float.TryParse(alCsvParts[iIndex + 2], numFloat, invCulture, out z);
iIndex += 3;
jointData.kinectPos = new Vector3(x, y, z);
}
else
{
jointData.kinectPos = Vector3.zero;
}
jointData.position = kinectToWorld.MultiplyPoint3x4(jointData.kinectPos);
jointData.kinectPos = new Vector3(jointData.kinectPos.x, jointData.kinectPos.y, jointData.kinectPos.z);
jointData.orientation = Quaternion.identity;
if (j == 0)
{
// set body position
bodyData[i].kinectPos = jointData.kinectPos;
bodyData[i].position = jointData.position;
bodyData[i].orientation = jointData.orientation;
}
}
bodyData[i].joint[j] = jointData;
if(j == (int)JointType.ClavicleLeft)
{
// clavicle right = clavicle left
int j2 = (int)JointType.ClavicleRight;
bodyData[i].joint[j2].trackingState = jointData.trackingState;
bodyData[i].joint[j2].kinectPos = jointData.kinectPos;
bodyData[i].joint[j2].position = jointData.position;
bodyData[i].joint[j2].orientation = jointData.orientation;
}
}
// spine naval
{
int p = (int)JointType.Pelvis;
int sc = (int)JointType.SpineChest;
int sn = (int)JointType.SpineNaval;
JointData jointData = bodyData[i].joint[sn];
jointData.trackingState = bodyData[i].joint[sc].trackingState;
jointData.orientation = bodyData[i].joint[sc].orientation;
Vector3 posChest = bodyData[i].joint[sc].kinectPos;
Vector3 posPelvis = bodyData[i].joint[p].kinectPos;
jointData.kinectPos = (posPelvis + posChest) * 0.5f;
posChest = bodyData[i].joint[sc].position;
posPelvis = bodyData[i].joint[p].position;
jointData.position = (posPelvis + posChest) * 0.5f;
bodyData[i].joint[sn] = jointData;
}
// calculate bone directions
CalcBodyJointDirs(ref bodyData[i]);
// calculate joint orients
if (kinectManager.ignoreZCoordinates)
{
DepthSensorBase.DoCalcBodyJointOrients2D(ref bodyData[i], kinectManager.ignoreInferredJoints);
}
else
{
DepthSensorBase.DoCalcBodyJointOrients(ref bodyData[i], kinectManager.ignoreInferredJoints);
}
// body orientation
bodyData[i].normalRotation = bodyData[i].joint[0].normalRotation;
bodyData[i].mirroredRotation = bodyData[i].joint[0].mirroredRotation;
}
}
return (uint)bodyCount;
}
// k2b joint to k4a joint
private static readonly int[] K2bJoint2JointType =
{
(int)JointType.Pelvis,
(int)JointType.SpineChest,
(int)JointType.Neck,
(int)JointType.Head,
(int)JointType.ShoulderLeft,
(int)JointType.ElbowLeft,
(int)JointType.WristLeft,
(int)JointType.HandLeft,
(int)JointType.ShoulderRight,
(int)JointType.ElbowRight,
(int)JointType.WristRight,
(int)JointType.HandRight,
(int)JointType.HipLeft,
(int)JointType.KneeLeft,
(int)JointType.AnkleLeft,
(int)JointType.FootLeft,
(int)JointType.HipRight,
(int)JointType.KneeRight,
(int)JointType.AnkleRight,
(int)JointType.FootRight,
(int)JointType.ClavicleLeft,
(int)JointType.HandtipLeft,
(int)JointType.ThumbLeft,
(int)JointType.HandtipRight,
(int)JointType.ThumbRight
};
// calculates all bone directions for the given body
public static void CalcBodyJointDirs(ref BodyData bodyData)
{
if (bodyData.bIsTracked)
{
for (int j = 0; j < (int)JointType.Count; j++)
{
if (j == 0)
{
bodyData.joint[j].direction = Vector3.zero;
}
else
{
int jParent = (int)GetParentJoint((JointType)j);
if (bodyData.joint[j].trackingState != TrackingState.NotTracked &&
bodyData.joint[jParent].trackingState != TrackingState.NotTracked)
{
bodyData.joint[j].direction = (bodyData.joint[j].position - bodyData.joint[jParent].position); //.normalized;
}
}
}
}
}
// calculates all joint velocities for the given body
public static void CalcBodyFrameJointVels(ref BodyData bodyData, long lastBodyFrameTcks, long prevBodyFrameTcks)
{
// calculate the inter-frame time
float frameTime = (float)(lastBodyFrameTcks - prevBodyFrameTcks) / 10000000f; // 100000000000f
//Debug.Log(" calcJointVels - bodyId: " + bodyData.liTrackingID + ", frameTime: " + frameTime);
if (bodyData.bIsTracked)
{
for (int j = 0; j < (int)JointType.Count; j++)
{
JointData jointData = bodyData.joint[j];
int p = (int)GetParentJoint((JointType)j);
Vector3 parentPos = bodyData.joint[p].position;
if(jointData.trackingState != TrackingState.NotTracked &&
bodyData.joint[p].trackingState != TrackingState.NotTracked)
{
jointData.posRel = jointData.position - parentPos;
jointData.posVel = frameTime > 0f ? (jointData.position - jointData.posPrev) / frameTime : Vector3.zero;
jointData.posPrev = jointData.position;
bodyData.joint[j] = jointData;
}
}
}
}
// tries to loads the space table from a file. returns the space table as array, or null if the file is not found
public static Vector3[] LoadSpaceTable(DepthSensorPlatform sensorPlatform, string sensorId, string camType, int frameWidth, int frameHeight)
{
Vector3[] spaceTable = null;
if (SAVE_SPACE_TABLES_TO_DISK)
{
string saveFolder = !string.IsNullOrEmpty(persistentDataPath) ? persistentDataPath : Application.persistentDataPath;
#if UNITY_EDITOR || UNITY_STANDALONE
saveFolder = ".";
#endif
if(saveFolder.Length > 0 && saveFolder[saveFolder.Length - 1] != '/' && saveFolder[saveFolder.Length - 1] != '\\')
{
saveFolder += "/";
}
string filePath = string.Format("{0}ST-{1}-{2}-{3}-{4}-{5}.bin", saveFolder, sensorPlatform.ToString(), sensorId, camType, frameWidth, frameHeight);
if (File.Exists(filePath))
{
byte[] btSpaceTable = File.ReadAllBytes(filePath);
spaceTable = new Vector3[frameWidth * frameHeight];
CopyBytes(btSpaceTable, 1, spaceTable, 3 * sizeof(float));
}
}
return spaceTable;
}
// saves the space file to a file
public static void SaveSpaceTable(Vector3[] spaceTable, DepthSensorPlatform sensorPlatform, string sensorId, string camType, int frameWidth, int frameHeight)
{
if (!SAVE_SPACE_TABLES_TO_DISK || spaceTable == null || spaceTable.Length == 0)
return;
string saveFolder = !string.IsNullOrEmpty(persistentDataPath) ? persistentDataPath : Application.persistentDataPath;
#if UNITY_EDITOR || UNITY_STANDALONE
saveFolder = ".";
#endif
if (saveFolder.Length > 0 && saveFolder[saveFolder.Length - 1] != '/' && saveFolder[saveFolder.Length - 1] != '\\')
{
saveFolder += "/";
}
string filePath = string.Format("{0}ST-{1}-{2}-{3}-{4}-{5}.bin", saveFolder, sensorPlatform.ToString(), sensorId, camType, frameWidth, frameHeight);
byte[] btSpaceTable = new byte[spaceTable.Length * 3 * sizeof(float)];
CopyBytes(spaceTable, 3 * sizeof(float), btSpaceTable, 1);
File.WriteAllBytes(filePath, btSpaceTable);
}
// save intrinsics & extrinsics of the sensor
public static bool LoadSensorIntrExtr(SensorData sensorData)
{
if (sensorData == null || sensorData.sensorInterface == null)
return false;
string saveFolder = !string.IsNullOrEmpty(persistentDataPath) ? persistentDataPath : Application.persistentDataPath;
#if UNITY_EDITOR || UNITY_STANDALONE
saveFolder = ".";
#endif
if (saveFolder.Length > 0 && saveFolder[saveFolder.Length - 1] != '/' && saveFolder[saveFolder.Length - 1] != '\\')
{
saveFolder += "/";
}
string filePath = string.Format("{0}IE-{1}-{2}-{3}-{4}-{5}-{6}.json", saveFolder, sensorData.sensorInterface.GetSensorPlatform().ToString(), sensorData.sensorInterface.GetSensorDeviceId(),
sensorData.depthImageWidth, sensorData.depthImageHeight, sensorData.colorImageWidth, sensorData.colorImageHeight);
if (File.Exists(filePath))
{
string sieJson = File.ReadAllText(filePath);
SensorIntrExtr sie = JsonUtility.FromJson<SensorIntrExtr>(sieJson);
if (sie != null)
{
sensorData.depthCamIntr = sie.depthCamIntr;
sensorData.colorCamIntr = sie.colorCamIntr;
sensorData.depth2ColorExtr = sie.depth2ColorExtr;
sensorData.color2DepthExtr = sie.color2DepthExtr;
return true;
}
}
return false;
}
// save intrinsics & extrinsics of the sensor
public static void SaveSensorIntrExtr(SensorData sensorData)
{
if (sensorData == null || sensorData.sensorInterface == null)
return;
SensorIntrExtr sie = new SensorIntrExtr();
sie.depthCamIntr = sensorData.depthCamIntr;
sie.colorCamIntr = sensorData.colorCamIntr;
sie.depth2ColorExtr = sensorData.depth2ColorExtr;
sie.color2DepthExtr = sensorData.color2DepthExtr;
string saveFolder = !string.IsNullOrEmpty(persistentDataPath) ? persistentDataPath : Application.persistentDataPath;
#if UNITY_EDITOR || UNITY_STANDALONE
saveFolder = ".";
#endif
if (saveFolder.Length > 0 && saveFolder[saveFolder.Length - 1] != '/' && saveFolder[saveFolder.Length - 1] != '\\')
{
saveFolder += "/";
}
string filePath = string.Format("{0}IE-{1}-{2}-{3}-{4}-{5}-{6}.json", saveFolder, sensorData.sensorInterface.GetSensorPlatform().ToString(), sensorData.sensorInterface.GetSensorDeviceId(),
sensorData.depthImageWidth, sensorData.depthImageHeight, sensorData.colorImageWidth, sensorData.colorImageHeight);
string sieJson = JsonUtility.ToJson(sie, true);
File.WriteAllText(filePath, sieJson);
//Debug.Log("Saved intr-extr to: " + filePath);
}
// returns the file name only, with or without extension
public static string GetFileName(string filePath, bool withExt)
{
if (string.IsNullOrEmpty(filePath))
return null;
int iP1 = filePath.LastIndexOf('/');
if(iP1 < 0)
iP1 = filePath.LastIndexOf('\\');
iP1++;
int iP2 = filePath.LastIndexOf('.');
if (withExt || iP2 < 0)
iP2 = filePath.Length;
string sFileName = filePath.Substring(iP1, iP2 - iP1);
return sFileName;
}
// returns the content of a text file, or null if the file does not exist
public static bool IsFileExists(string fileName)
{
if (string.IsNullOrEmpty(fileName))
return false;
string saveFolder = !string.IsNullOrEmpty(persistentDataPath) ? persistentDataPath : Application.persistentDataPath;
#if UNITY_EDITOR || UNITY_STANDALONE
saveFolder = ".";
#endif
if (saveFolder.Length > 0 && saveFolder[saveFolder.Length - 1] != '/' && saveFolder[saveFolder.Length - 1] != '\\')
{
saveFolder += "/";
}
string filePath = saveFolder + fileName;
bool isFileExists = File.Exists(filePath);
return isFileExists;
}
// returns the content of a text file, or null if the file does not exist
public static string LoadTextFile(string fileName)
{
string sContent = null;
if (string.IsNullOrEmpty(fileName))
return sContent;
string saveFolder = !string.IsNullOrEmpty(persistentDataPath) ? persistentDataPath : Application.persistentDataPath;
#if UNITY_EDITOR || UNITY_STANDALONE
saveFolder = ".";
#endif
if (saveFolder.Length > 0 && saveFolder[saveFolder.Length - 1] != '/' && saveFolder[saveFolder.Length - 1] != '\\')
{
saveFolder += "/";
}
string filePath = saveFolder + fileName;
if (File.Exists(filePath))
{
sContent = File.ReadAllText(filePath);
}
return sContent;
}
// saves the given content to a text file.
public static void SaveTextFile(string fileName, string content)
{
if (string.IsNullOrEmpty(fileName))
return;
string saveFolder = !string.IsNullOrEmpty(persistentDataPath) ? persistentDataPath : Application.persistentDataPath;
#if UNITY_EDITOR || UNITY_STANDALONE
saveFolder = ".";
#endif
if (saveFolder.Length > 0 && saveFolder[saveFolder.Length - 1] != '/' && saveFolder[saveFolder.Length - 1] != '\\')
{
saveFolder += "/";
}
string filePath = saveFolder + fileName;
File.WriteAllText(filePath, content);
}
//// calculates the average (mean) of multiple quaternions
//// note: this only works if all the quaternions are relatively close together.
//public static Quaternion AverageQuaternion(ref Vector4 cumulative, Quaternion newRotation, Quaternion firstRotation, int addAmount)
//{
// float w = 0.0f;
// float x = 0.0f;
// float y = 0.0f;
// float z = 0.0f;
// if (!AreQuaternionsClose(newRotation, firstRotation))
// {
// newRotation = InverseSignQuaternion(newRotation);
// }
// float addDet = 1f / (float)addAmount;
// cumulative.w += newRotation.w;
// w = cumulative.w * addDet;
// cumulative.x += newRotation.x;
// x = cumulative.x * addDet;
// cumulative.y += newRotation.y;
// y = cumulative.y * addDet;
// cumulative.z += newRotation.z;
// z = cumulative.z * addDet;
// return NormalizeQuaternion(x, y, z, w);
//}
// sums up multiple quaternions
public static void SumUpQuaternions(ref Vector4 cumulative, Quaternion newRotation, Quaternion firstRotation)
{
//Before we add the new rotation to the average (mean), we have to check whether the quaternion has to be inverted. Because
//q and -q are the same rotation, but cannot be averaged, we have to make sure they are all the same.
if (!AreQuaternionsClose(newRotation, firstRotation))
{
newRotation = InverseSignQuaternion(newRotation);
}
cumulative.w += newRotation.w;
cumulative.x += newRotation.x;
cumulative.y += newRotation.y;
cumulative.z += newRotation.z;
}
// calculates the average (mean) of multiple quaternions
// note: this only works if all the quaternions are relatively close together.
public static Quaternion AverageQuaternions(Vector4 cumulative, int addAmount)
{
float addDet = 1f / (float)addAmount;
float w = cumulative.w * addDet;
float x = cumulative.x * addDet;
float y = cumulative.y * addDet;
float z = cumulative.z * addDet;
return NormalizeQuaternion(x, y, z, w);
}
// normalizes a quaternion
private static Quaternion NormalizeQuaternion(float x, float y, float z, float w)
{
float lengthD = 1.0f / (w * w + x * x + y * y + z * z);
w *= lengthD;
x *= lengthD;
y *= lengthD;
z *= lengthD;
return new Quaternion(x, y, z, w);
}
// inverts the sign of a quaternion
private static Quaternion InverseSignQuaternion(Quaternion q)
{
return new Quaternion(-q.x, -q.y, -q.z, -q.w);
}
// checks if the two quaternions are close or not
private static bool AreQuaternionsClose(Quaternion q1, Quaternion q2)
{
float dot = Quaternion.Dot(q1, q2);
return dot >= 0f;
}
// transforms the source texture to target texture, by applying the needed resize, rotation and flip
public static void TransformTexture(Texture srcTex, RenderTexture tgtTex, int rotationAngle = 0, bool flipX = false, bool flipY = false)
{
if (srcTex == null || tgtTex == null)
return;
int srcWidth = srcTex.width;
int srcHeight = srcTex.height;
bool isPortraitMode = rotationAngle == 90 || rotationAngle == 270;
if (isPortraitMode)
{
// swap width and height
(srcWidth, srcHeight) = (srcHeight, srcWidth);
}
float camAspect = (float)srcWidth / srcHeight;
float tgtAspect = (float)tgtTex.width / tgtTex.height;
//// fix bug on Android
//if (Application.platform == RuntimePlatform.Android)
//{
// rotationAngle = -rotationAngle;
//}
Matrix4x4 transMat = Matrix4x4.identity;
Vector4 texST = Vector4.zero;
if (isPortraitMode)
{
transMat = GetTransformMat(rotationAngle, flipY, flipX);
texST = GetTexST(tgtAspect, camAspect);
}
else
{
transMat = GetTransformMat(rotationAngle, flipX, flipY);
texST = GetTexST(camAspect, tgtAspect);
}
if(_textureMat == null)
{
_textureMat = new Material(Shader.Find("Kinect/TransformTexShader"));
}
// blit the texture
if(_textureMat != null)
{
_textureMat.SetMatrix(_TransformMatParam, transMat);
_textureMat.SetVector(_TexSTParam, texST);
Graphics.Blit(srcTex, tgtTex, _textureMat, 0);
}
}
private static Material _textureMat = null;
private static readonly int _TransformMatParam = Shader.PropertyToID("_TransformMat");
private static readonly int _TexSTParam = Shader.PropertyToID("_TexST");
// returns the camera transform matrix
private static Matrix4x4 GetTransformMat(float rotation, bool mirrorHorizontal, bool mirrorVertical)
{
Vector3 scale = new Vector3(mirrorHorizontal ? -1f : 1f, mirrorVertical ? -1f : 1f, 1f);
Matrix4x4 mat = Matrix4x4.TRS(Vector3.zero, Quaternion.Euler(0, 0, rotation), scale);
return PUSH_MATRIX * mat * POP_MATRIX;
}
private static readonly Matrix4x4 PUSH_MATRIX = Matrix4x4.Translate(new Vector3(0.5f, 0.5f, 0));
private static readonly Matrix4x4 POP_MATRIX = Matrix4x4.Translate(new Vector3(-0.5f, -0.5f, 0));
// returns the camera texture offsets
private static Vector4 GetTexST(float srcAspect, float dstAspect)
{
if (srcAspect > dstAspect)
{
float s = dstAspect / srcAspect;
return new Vector4(s, 1, (1 - s) / 2, 0);
}
else
{
float s = srcAspect / dstAspect;
return new Vector4(1, s, 0, (1 - s) / 2);
}
}
// destroys the given object
public static void Destroy(UnityEngine.Object o)
{
if (o == null)
return;
if (o is RenderTexture)
{
((RenderTexture)o).Release();
}
if (Application.isPlaying)
UnityEngine.Object.Destroy(o);
else
UnityEngine.Object.DestroyImmediate(o);
}
}
}