ARKitImageTrackingProvider.cs
5.3 KB
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
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
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
using System;
using System.Runtime.InteropServices;
using Unity.Collections;
using UnityEngine.Scripting;
using UnityEngine.XR.ARSubsystems;
namespace UnityEngine.XR.ARKit
{
/// <summary>
/// ARKit implementation of the <c>XRImageTrackingSubsystem</c>.
/// </summary>
[Preserve]
public sealed class ARKitImageTrackingSubsystem : XRImageTrackingSubsystem
{
class ARKitProvider : Provider
{
#if UNITY_2020_2_OR_NEWER
public override void Start() { }
public override void Stop() { }
#endif
public unsafe override RuntimeReferenceImageLibrary CreateRuntimeLibrary(
XRReferenceImageLibrary serializedLibrary)
{
return new ARKitImageDatabase(serializedLibrary);
}
public override RuntimeReferenceImageLibrary imageLibrary
{
set
{
if (value == null)
{
UnityARKit_ImageTracking_Stop();
}
else if (value is ARKitImageDatabase database)
{
UnityARKit_ImageTracking_SetDatabase(database.nativePtr);
}
else
{
throw new ArgumentException($"{value.GetType().Name} is not a valid ARKit image library.");
}
}
}
public unsafe override TrackableChanges<XRTrackedImage> GetChanges(
XRTrackedImage defaultTrackedImage,
Allocator allocator)
{
void* addedPtr, updatedPtr, removedPtr;
int addedLength, updatedLength, removedLength, stride;
var context = UnityARKit_ImageTracking_AcquireChanges(
out addedPtr, out addedLength,
out updatedPtr, out updatedLength,
out removedPtr, out removedLength,
out stride);
try
{
return new TrackableChanges<XRTrackedImage>(
addedPtr, addedLength,
updatedPtr, updatedLength,
removedPtr, removedLength,
defaultTrackedImage, stride,
allocator);
}
finally
{
UnityARKit_ImageTracking_ReleaseChanges(context);
}
}
public override void Destroy() => UnityARKit_ImageTracking_Destroy();
public override int requestedMaxNumberOfMovingImages
{
get => UnityARKit_ImageTracking_GetRequestedMaximumNumberOfTrackedImages();
set => UnityARKit_ImageTracking_SetRequestedMaximumNumberOfTrackedImages(value);
}
public override int currentMaxNumberOfMovingImages => UnityARKit_ImageTracking_GetCurrentMaximumNumberOfTrackedImages();
}
[DllImport("__Internal")]
static extern int UnityARKit_ImageTracking_GetRequestedMaximumNumberOfTrackedImages();
[DllImport("__Internal")]
static extern void UnityARKit_ImageTracking_SetRequestedMaximumNumberOfTrackedImages(int maxNumTrackedImages);
[DllImport("__Internal")]
static extern int UnityARKit_ImageTracking_GetCurrentMaximumNumberOfTrackedImages();
[DllImport("__Internal")]
static extern void UnityARKit_ImageTracking_SetDatabase(IntPtr database);
[DllImport("__Internal")]
static extern void UnityARKit_ImageTracking_Stop();
[DllImport("__Internal")]
static extern void UnityARKit_ImageTracking_Destroy();
[DllImport("__Internal")]
static extern unsafe void* UnityARKit_ImageTracking_AcquireChanges(
out void* addedPtr, out int addedLength,
out void* updatedPtr, out int updatedLength,
out void* removedPtr, out int removedLength,
out int stride);
[DllImport("__Internal")]
static extern unsafe void UnityARKit_ImageTracking_ReleaseChanges(void* changes);
[RuntimeInitializeOnLoadMethod(RuntimeInitializeLoadType.SubsystemRegistration)]
static void RegisterDescriptor()
{
// No support before iOS 11.3
if (!Api.AtLeast11_3())
return;
XRImageTrackingSubsystemDescriptor.Create(new XRImageTrackingSubsystemDescriptor.Cinfo
{
id = "ARKit-ImageTracking",
#if UNITY_2020_2_OR_NEWER
providerType = typeof(ARKitImageTrackingSubsystem.ARKitProvider),
subsystemTypeOverride = typeof(ARKitImageTrackingSubsystem),
#else
subsystemImplementationType = typeof(ARKitImageTrackingSubsystem),
#endif
supportsMovingImages = Api.AtLeast12_0(),
supportsMutableLibrary = true,
requiresPhysicalImageDimensions = true
});
}
#if !UNITY_2020_2_OR_NEWER
/// <summary>
/// Creates the ARKit-specific implementation which will service the `XRImageTrackingSubsystem`.
/// </summary>
/// <returns>A new instance of the `Provider` specific to ARKit.</returns>
protected override Provider CreateProvider() => new ARKitProvider();
#endif
}
}