// Copyright 2021 Pose AI Ltd. All rights reserved

using UnityEngine;

using System.Collections.Generic;

namespace PoseAI

{

/// <summary>

/// Component to load

/// </summary>

public class PoseAIRuntimeLoader

{

// the path must be relative to a 'Resources' folder in the project structure. If necessary move files into a Resources folder

public static string DefaultAnimationControllerPath = "StarterAssets/ThirdPersonController/Character/Animations/StarterAssetsThirdPerson";

// specify which type of rig you use

public static PoseAI_Rigs rigType = PoseAI_Rigs.Unity;

// specify the rig remapping here, if any, to make sure it is set for each new avatar

public static string Remapping = "";

/*

* pass in

* 1. gameCharacter to be initialized with PoseAIComponents

* 2. a configured avatar from a template character

* 3. either a new port for a new source, or an exsiting PoseAISource (i.e. if player is already connected and we are animating a new character or scene).

* 4. optionally pass in a different animation controller if you want to use different animation controllers in different parts of the game

*/

public static bool RuntimeLoad(GameObject gameCharacter, Avatar avatar, int port = 0, PoseAISourceDirect poseAISourceIn = null, RuntimeAnimatorController animationController = null)

{

Debug.Assert(poseAISourceIn != null || port != 0);

/*

* adds an animator component if it doesn't exist. Then configures key required properties:

* 1. runtimeAnimatorController - this should be a controller with IK pass enabled and the state machine logic referred to in the PoseAICharacterController

* 2. avatar - this should be a configured humanoid avatar with the correct mappings for the Mecanim system. Difficult to create at Runtime, easier to pass in from a template character in the project.

*/

Animator animator = gameCharacter.GetComponent<Animator>();

if (animator == null)

animator = gameCharacter.AddComponent<Animator>() as Animator;

if (animationController == null)

{

animator.runtimeAnimatorController = Resources.Load(DefaultAnimationControllerPath) as RuntimeAnimatorController;

if (animator.runtimeAnimatorController == null)

{

Debug.LogWarning("Failed to load animator at " + DefaultAnimationControllerPath);

return false;

}

}

else

{

animator.runtimeAnimatorController = animationController;

}

animator.avatar = avatar;

/*

* adds an default character controller and configures the capsule. may need to be tweaked to suit game and rig

*/

CharacterController characterController = gameCharacter.GetComponent<CharacterController>();

if (characterController == null)

characterController = gameCharacter.AddComponent<CharacterController>() as CharacterController;

characterController.radius = 0.28f;

characterController.height = 1.8f;

characterController.center = new Vector3(0.0f, 0.93f, 0.0f);

/*

* if an established Source is not passed to loader, this sets up a new poseAI source to listen on the specified port.

*/

PoseAISource poseAISource;

if (poseAISourceIn != null)

{

PoseAISourceShared poseAISourceShared = gameCharacter.AddComponent<PoseAISourceShared>() as PoseAISourceShared;

poseAISourceShared.Source = poseAISourceIn;

poseAISource = poseAISourceShared;

}

else

{

PoseAISourceDirect poseAISourceDirect = gameCharacter.GetComponent<PoseAISourceDirect>();

if (poseAISourceDirect == null)

poseAISourceDirect = gameCharacter.AddComponent<PoseAISourceDirect>() as PoseAISourceDirect;

poseAISourceDirect.Mode = PoseAI_Modes.Room;

poseAISourceDirect.RigType = rigType;

poseAISourceDirect.Port = port;

poseAISourceDirect.MirrorCamera = false;

poseAISource = poseAISourceDirect;

}

/*

* if an established Source is not passed to loader, this sets up a new poseAI source to listen on the specified port.

*/

PoseAICharacterAnimator poseAICharacterAnimator = gameCharacter.GetComponent<PoseAICharacterAnimator>();

if (poseAICharacterAnimator == null)

poseAICharacterAnimator = gameCharacter.AddComponent<PoseAICharacterAnimator>() as PoseAICharacterAnimator;

poseAICharacterAnimator.SetSource(poseAISource);

poseAICharacterAnimator.SetRemapping(Remapping);

poseAICharacterAnimator.OverrideRootName = "";

poseAICharacterAnimator.JointNamePrefix = "";

PoseAICharacterController poseAICharacterController = gameCharacter.GetComponent<PoseAICharacterController>();

if (poseAICharacterController == null)

poseAICharacterController = gameCharacter.AddComponent<PoseAICharacterController>() as PoseAICharacterController;

poseAICharacterController.SetSource(poseAISource);

poseAICharacterController.CinemachineCameraTarget = GameObject.Find("PlayerFollowCamera");

if (animator == null)

{

Debug.LogWarning("Failed to load animator");

return false;

}

if (animator.avatar == null)

{

Debug.LogWarning("Failed to load animator's avatar");

return false;

}

if (characterController == null)

{

Debug.LogWarning("Failed to load character controller");

return false;

}

if (poseAISource == null)

{

Debug.LogWarning("Failed to load source");

return false;

}

if (poseAICharacterAnimator == null)

{

Debug.LogWarning("Failed to load poseaicharacteranimator");

return false;

}

if (poseAICharacterController == null)

{

Debug.LogWarning("Failed to load poseaicharactercontroller");

return false;

}

return true;

}

public void EnableOrDisablePoseAI(GameObject gameCharacter, bool enabled)

{

PoseAISourceDirect poseAISource = gameCharacter.GetComponent<PoseAISourceDirect>();

if (poseAISource != null)

poseAISource.enabled = enabled;

PoseAICharacterAnimator poseAICharacterAnimator = gameCharacter.GetComponent<PoseAICharacterAnimator>();

if (poseAICharacterAnimator != null)

poseAICharacterAnimator.enabled = enabled;

PoseAICharacterController poseAICharacterController = gameCharacter.GetComponent<PoseAICharacterController>();

if (poseAICharacterController != null)

poseAICharacterController.enabled = enabled;

}

}

public static class PoseAI_Decoder

{

static readonly uint[] reverse_map = { 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 62, 63, 62, 62, 63, 52, 53, 54, 55, 56, 57, 58, 59, 60, 61, 0, 0, 0, 0, 0, 0, 0, 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 0, 0, 0, 0, 63, 0, 26, 27, 28, 29, 30, 31, 32, 33, 34, 35, 36, 37, 38, 39, 40, 41, 42, 43, 44, 45, 46, 47, 48, 49, 50, 51 };

static readonly float[] firstByte = { 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.9384465070835368f, 0.9697117733268197f, 0.9384465070835368f, 0.9384465070835368f, 0.9697117733268197f, 0.6257938446507083f, 0.6570591108939912f, 0.688324377137274f, 0.7195896433805569f, 0.7508549096238397f, 0.7821201758671226f, 0.8133854421104054f, 0.8446507083536883f, 0.8759159745969711f, 0.907181240840254f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, -1.0f, -0.9687347337567171f, -0.9374694675134343f, -0.9062042012701514f, -0.8749389350268686f, -0.8436736687835857f, -0.8124084025403029f, -0.78114313629702f, -0.7498778700537372f, -0.7186126038104543f, -0.6873473375671715f, -0.6560820713238886f, -0.6248168050806058f, -0.5935515388373229f, -0.5622862725940401f, -0.5310210063507572f, -0.49975574010747437f, -0.4684904738641915f, -0.43722520762090866f, -0.4059599413776258f, -0.37469467513434296f, -0.3434294088910601f, -0.31216414264777725f, -0.2808988764044944f, -0.24963361016121155f, -0.2183683439179287f, 0.0f, 0.0f, 0.0f, 0.0f, 0.9697117733268197f, 0.0f, -0.18710307767464585f, -0.155837811431363f, -0.12457254518808014f, -0.09330727894479729f, -0.06204201270151444f, -0.030776746458231585f, 0.0004885197850512668f, 0.03175378602833412f, 0.06301905227161697f, 0.09428431851489982f, 0.12554958475818268f, 0.15681485100146553f, 0.18808011724474838f, 0.21934538348803123f, 0.2506106497313141f, 0.28187591597459694f, 0.3131411822178798f, 0.34440644846116264f, 0.3756717147044455f, 0.40693698094772834f, 0.4382022471910112f, 0.46946751343429405f, 0.5007327796775769f, 0.5319980459208598f, 0.5632633121641426f, 0.5945285784074255f };

static readonly float[] secondByte = { 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.030288226673180263f, 0.030776746458231558f, 0.030288226673180263f, 0.030288226673180263f, 0.030776746458231558f, 0.025403028822667317f, 0.025891548607718612f, 0.026380068392769906f, 0.0268685881778212f, 0.027357107962872496f, 0.02784562774792379f, 0.028334147532975085f, 0.02882266731802638f, 0.029311187103077674f, 0.02979970688812897f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0f, 0.0004885197850512946f, 0.0009770395701025891f, 0.0014655593551538837f, 0.0019540791402051783f, 0.002442598925256473f, 0.0029311187103077674f, 0.003419638495359062f, 0.0039081582804103565f, 0.004396678065461651f, 0.004885197850512946f, 0.00537371763556424f, 0.005862237420615535f, 0.006350757205666829f, 0.006839276990718124f, 0.0073277967757694185f, 0.007816316560820713f, 0.008304836345872008f, 0.008793356130923302f, 0.009281875915974597f, 0.009770395701025891f, 0.010258915486077186f, 0.01074743527112848f, 0.011235955056179775f, 0.01172447484123107f, 0.012212994626282364f, 0.0f, 0.0f, 0.0f, 0.0f, 0.030776746458231558f, 0.0f, 0.012701514411333659f, 0.013190034196384953f, 0.013678553981436248f, 0.014167073766487542f, 0.014655593551538837f, 0.015144113336590131f, 0.015632633121641426f, 0.01612115290669272f, 0.016609672691744015f, 0.01709819247679531f, 0.017586712261846604f, 0.0180752320468979f, 0.018563751831949193f, 0.019052271617000488f, 0.019540791402051783f, 0.020029311187103077f, 0.02051783097215437f, 0.021006350757205666f, 0.02149487054225696f, 0.021983390327308255f, 0.02247191011235955f, 0.022960429897410845f, 0.02344894968246214f, 0.023937469467513434f, 0.024425989252564728f, 0.024914509037616023f };

static int CharToInt(char c)

{

return (int)(c);

}

public static uint UintB64ToUint(char a, char b)

{

return reverse_map[CharToInt(a)] * 64 + reverse_map[CharToInt(b)];

}

public static uint UintB64ToUint(char a, char b, char c)

{

return reverse_map[CharToInt(a)] * 4096 + reverse_map[CharToInt(b)] * 64 + reverse_map[CharToInt(c)];

}

public static float FixedB64pairToFloat(char a, char b)

{

return firstByte[CharToInt(a)] + secondByte[CharToInt(b)];

}

public static void FStringFixed12ToFloat(ref string data, ref List<float> flatArray)

{

flatArray.Capacity = flatArray.Count + data.Length / 2;

for (int i = 0; i < data.Length - 1; i += 2)

flatArray.Add(FixedB64pairToFloat(data[i], data[i + 1]));

}

public static void FStringFixed12ToFloat(ref string data, ref List<float?> flatArray)

{

flatArray.Capacity = flatArray.Count + data.Length / 2;

for (int i = 0; i < data.Length - 1; i += 2)

{

if(data[i]=='=' && data[i+1]=='=')

flatArray.Add(null);

else

flatArray.Add(FixedB64pairToFloat(data[i], data[i + 1]));

}

}

public static void FlatArrayToQuats(ref List<float> flatArray, ref List<Quaternion> quatArray)

{

quatArray.Capacity = quatArray.Count + flatArray.Count / 4;

for (int i = 0; i < flatArray.Count - 3; i += 4)

{

//reordered Quats for Unity coordinate system.

var quat = new Quaternion(flatArray[i + 1], -flatArray[i], flatArray[i + 3], -flatArray[i + 2]);

quat.Normalize();

quatArray.Add(quat);

}

}

}

public class UITouch

{

public int index = 0;

public float x = 0.0f;

public float y = 0.0f;

public TouchState state = TouchState.Ended;

public override string ToString()

{

return "Touch_" + index + ":[" + x.ToString("0.000") + "," + y.ToString("0.000") + "] " + state;

}

public static void DecodeMultiTouchString(ref string data, ref Queue<UITouch> touchQueue)

{

for (int i = 0; i < data.Length - 4; i += 5)

{

UITouch touch = new UITouch() { index = data[i] - '0'};

if (data[i + 1] == '=' && data[i + 2] == '=')

touch.state = TouchState.Ended;

else if (data[i + 1] == '=' && data[i + 2] == '0')

touch.state = TouchState.Cancelled;

else

{

touch.x = PoseAI_Decoder.FixedB64pairToFloat(data[i + 1], data[i + 2]);

touch.y = PoseAI_Decoder.FixedB64pairToFloat(data[i + 3], data[i + 4]);

touch.state = TouchState.Touching;

}

touchQueue.Enqueue(touch);

}

}

public static void DecodeMultiTouchStateString(ref string data, ref List<UITouch> touchArray)

{

for (int i = 0; i < data.Length - 3; i += 4)

{

int idx = i / 4;

touchArray[idx].index = idx;

if (data[i] == '=' && data[i + 1] == '=')

touchArray[idx].state = TouchState.Ended;

else if (data[i + 1] == '=' && data[i + 2] == '0')

touchArray[idx].state = TouchState.Cancelled;

else

{

touchArray[idx].x = PoseAI_Decoder.FixedB64pairToFloat(data[i], data[i + 1]);

touchArray[idx].y = PoseAI_Decoder.FixedB64pairToFloat(data[i + 2], data[i + 3]);

if (touchArray[idx].state == TouchState.Touching || touchArray[idx].state == TouchState.Begun)

touchArray[idx].state = TouchState.Touching;

else

touchArray[idx].state = TouchState.Begun;

}

}

}

public static List<UITouch> MakeList(int number) {

var retValue = new List<UITouch>(number);

for (int i = 0; i < number; ++i)

retValue.Add(new UITouch() { index = i }) ;

return retValue;

}

}

}