TrackerManager.cs

/*==============================================================================
Copyright 2017 Maxst, Inc. All Rights Reserved.
==============================================================================*/

using UnityEngine;
using System;
using System.Linq;
using System.Runtime.InteropServices;
using System.Text;

namespace maxstAR
{
    public class TrackerManager
    {
        public const int TRACKER_TYPE_CODE_SCANNER = 0X01;

        public const int TRACKER_TYPE_IMAGE = 0X02;

        public const int TRACKER_TYPE_MARKER = 0X04;

        public const int TRACKER_TYPE_OBJECT = 0X08;

        public const int TRACKER_TYPE_SLAM = 0x10;

        public const int TRACKER_TYPE_INSTANT = 0x20;

        private static TrackerManager instance = null;
        private float[] glPoseMatrix = new float[16];

        public enum TrackingOption
        {
            NORMAL_TRACKING = 1,

            EXTEND_TRACKING = 2,

            MULTI_TRACKING = 4,

            JITTER_REDUCTION_ACTIVATION = 8,

            JITTER_REDUCTION_DEACTIVATION = 16
        }

        public static TrackerManager GetInstance()
        {
            if (instance == null)
            {
                instance = new TrackerManager();
            }
            return instance;
        }

        //private SurfaceMesh surfaceMesh = null;
        private GuideInfo guideInfo = null;
        private TrackingState trackingState = null;

        private TrackerManager()
        {
            trackingState = new TrackingState();

            // To avoid null pointer error
            UpdateTrackingState();
        }

        public void StartTracker(int trackerType)
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_startTracker(trackerType);
            }
            else
            {
                NativeSharedAPI.TrackerManager_startTracker(trackerType);
            }
        }

        public void StopTracker()
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_stopTracker();
            }
            else
            {
                NativeSharedAPI.TrackerManager_stopTracker();
            }
        }

        public void DestroyTracker()
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_destroyTracker();
            }
            else
            {
                NativeSharedAPI.TrackerManager_destroyTracker();
            }
        }

        public void AddTrackerData(string trackingFileName, bool isAndroidAssetFile = false)
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_addTrackerData(trackingFileName, isAndroidAssetFile);
            }
            else
            {
                NativeSharedAPI.TrackerManager_addTrackerData(trackingFileName, isAndroidAssetFile);
            }
        }

        public void RemoveTrackerData(string trackingFileName = "")
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_removeTrackerData(trackingFileName);
            }
            else
            {
                NativeSharedAPI.TrackerManager_removeTrackerData(trackingFileName);
            }
        }

        public void LoadTrackerData()
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_loadTrackerData();
            }
            else
            {
                NativeSharedAPI.TrackerManager_loadTrackerData();
            }
        }

        public void SetTrackingOption(TrackingOption trackingOption)
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_setTrackingOption((int)trackingOption);
            }
            else
            {
                NativeSharedAPI.TrackerManager_setTrackingOption((int)trackingOption);
            }

            if (trackingOption == TrackingOption.MULTI_TRACKING)
            {
                AbstractARManager.Instance.SetWorldCenterMode(AbstractARManager.WorldCenterMode.CAMERA);
            }
        }

        public bool IsTrackerDataLoadCompleted()
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                return NativeStaticAPI.TrackerManager_isTrackerDataLoadCompleted();
            }
            else
            {
                return NativeSharedAPI.TrackerManager_isTrackerDataLoadCompleted();
            }
        }

        public TrackingState UpdateTrackingState()
        {
            ulong trackingStateCPtr = 0;

            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                trackingStateCPtr = NativeStaticAPI.TrackerManager_updateTrackingState();
            }
            else
            {
                trackingStateCPtr = NativeSharedAPI.TrackerManager_updateTrackingState();
            }

            trackingState.SetTrackingStateCPtr(trackingStateCPtr);
            return trackingState;
        }

        public TrackingState GetTrackingState()
        {
            return trackingState;
        }

        //public string GetCodeScanResult()
        //{
        //  if (Application.platform == RuntimePlatform.IPhonePlayer)
        //  {
        //      int length = NativeStaticAPI.TrackerManager_getCodeScanResultLength();
        //      if (length > 0)
        //      {
        //          byte[] result = new byte[length];
        //          NativeStaticAPI.TrackerManager_getCodeScanResult(result, length);
        //          return Encoding.Default.GetString(result);
        //      }
        //      else
        //      {
        //          return "";
        //      }
        //  }
        //  else
        //  {
        //      int length = NativeSharedAPI.TrackerManager_getCodeScanResultLength();
        //      if (length > 0)
        //      {
        //          byte[] result = new byte[length];
        //          NativeSharedAPI.TrackerManager_getCodeScanResult(result, length);
        //          return Encoding.Default.GetString(result);
        //      }
        //      else
        //      {
        //          return "";
        //      }
        //  }
        //}

        public Vector3 GetWorldPositionFromScreenCoordinate(Vector2 screen)
        {
            float[] s = new float[2];
            float[] w = new float[3];

            s[0] = screen.x;
            s[1] = Screen.height - screen.y;

            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_getWorldPositionFromScreenCoordinate(s, w);
            }
            else
            {
                NativeSharedAPI.TrackerManager_getWorldPositionFromScreenCoordinate(s, w);
            }

            //Vector3 world = new Vector3(w[0], w[1], w[2]);
            Vector3 world = new Vector3(w[0], w[2], w[1]);

            switch (Screen.orientation)
            {
                case ScreenOrientation.Portrait:
                    world.z *= -1.0f;
                    break;

                case ScreenOrientation.PortraitUpsideDown:
                    world.x *= -1.0f;
                    break;

                case ScreenOrientation.Landscape:
                    world.z *= -1.0f;
                    break;

                case ScreenOrientation.LandscapeRight:
                    world.x *= -1.0f;
                    break;
            }

            return world;
        }

        //public Vector2 GetScreenCoordinateFromWorldPosition(Vector3 world)
        //{
        //  switch (Screen.orientation)
        //  {
        //      case ScreenOrientation.Portrait:
        //          world.z *= -1.0f;
        //          break;

        //      case ScreenOrientation.PortraitUpsideDown:
        //          world.x *= -1.0f;
        //          break;

        //      case ScreenOrientation.Landscape:
        //          world.z *= -1.0f;
        //          break;

        //      case ScreenOrientation.LandscapeRight:
        //          world.x *= -1.0f;
        //          break;
        //  }

        //  float[] s = new float[2];
        //  float[] w = new float[3];

        //  w[0] = world.x;
        //  w[1] = world.z;
        //  w[2] = world.y;

        //  if (Application.platform == RuntimePlatform.IPhonePlayer)
        //  {
        //      NativeStaticAPI.getScreenCoordinateFromWorldPosition(w, s);
        //  }
        //  else
        //  {
        //      NativeSharedAPI.getScreenCoordinateFromWorldPosition(w, s);
        //  }

        //  return new Vector2(s[0], Screen.height - s[1]);
        //}

        //public int GetKeyframeCount()
        //{
        //    if (Application.platform == RuntimePlatform.IPhonePlayer)
        //    {
        //        return NativeStaticAPI.TrackerManager_getKeyframeCount();
        //    }
        //    else
        //    {
        //        return NativeSharedAPI.TrackerManager_getKeyframeCount();
        //    }
        //}

        //public int GetFeatureCount()
        //{
        //    if (Application.platform == RuntimePlatform.IPhonePlayer)
        //    {
        //        return NativeStaticAPI.TrackerManager_getFeatureCount();
        //    }
        //    else
        //    {
        //        return NativeSharedAPI.TrackerManager_getFeatureCount();
        //    }
        //}

        //public void GetFeatureVertex(float[] vertex)
        //{
        //    if (Application.platform == RuntimePlatform.IPhonePlayer)
        //    {
        //        NativeStaticAPI.TrackerManager_getFeatureVertex(vertex);
        //    }
        //    else
        //    {
        //        NativeSharedAPI.TrackerManager_getFeatureVertex(vertex);
        //    }
        //}

        //public Matrix4x4 GetFeaturePose()
        //{
        //    if (Application.platform == RuntimePlatform.IPhonePlayer)
        //    {
        //        NativeStaticAPI.TrackerManager_getFeatureTrackingPose(glPoseMatrix);
        //    }
        //    else
        //    {
        //        NativeSharedAPI.TrackerManager_getFeatureTrackingPose(glPoseMatrix);
        //    }

        //    Matrix4x4 poseMatrix = MatrixUtils.ConvertGLMatrixToUnityMatrix4x4(glPoseMatrix);

        //    return poseMatrix;
        //}            

        public void FindSurface()
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_findSurface();
            }
            else
            {
                NativeSharedAPI.TrackerManager_findSurface();
            }
        }

        public void QuitFindingSurface()
        {
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                NativeStaticAPI.TrackerManager_quitFindingSurface();
            }
            else
            {
                NativeSharedAPI.TrackerManager_quitFindingSurface();
            }
        }

        //public SurfaceMesh GetSurfaceMesh()
        //{
        //    if (surfaceMesh == null)
        //    {
        //        surfaceMesh = new SurfaceMesh();
        //    }

        //    surfaceMesh.UpdateSurfaceMesh();
        //    return surfaceMesh;
        //}

        public GuideInfo GetGuideInfo()
        {
            if (guideInfo == null)
            {
                guideInfo = new GuideInfo();
            }

            guideInfo.UpdateGuideInfo();
            return guideInfo;
        }

        public SurfaceThumbnail SaveSurfaceData(string outputFileName)
        {
            SurfaceThumbnail surfaceThumbnail = null;
            ulong SurfaceThumbnail_cPtr = 0;
            if (Application.platform == RuntimePlatform.IPhonePlayer)
            {
                SurfaceThumbnail_cPtr = NativeStaticAPI.TrackerManager_saveSurfaceData(outputFileName);
            }
            else
            {
                SurfaceThumbnail_cPtr = NativeSharedAPI.TrackerManager_saveSurfaceData(outputFileName);
            }

            if (SurfaceThumbnail_cPtr == 0)
            {
                return null;
            }
            else
            {
                surfaceThumbnail = new SurfaceThumbnail(SurfaceThumbnail_cPtr);
            }

            return surfaceThumbnail;
        }
    }
}