aboutsummaryrefslogtreecommitdiffstatsci
path: root/vulkan/ExtensionMethods.cs
blob: d14625e579f0a67ba8345f3e7861b4efd3ab9182 (plain)
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
// Copyright (c) 2019  Jean-Philippe Bruyère <jp_bruyere@hotmail.com>
//
// This code is licensed under the MIT license (MIT) (http://opensource.org/licenses/MIT)
using System;
using System.Collections.Generic;
using System.Diagnostics;
using System.Linq;
using System.Runtime.InteropServices;

using static Vulkan.Vk;

namespace Vulkan {
	public static class ExtensionMethods {
		#region pinning
		/// <summary>
		/// list of pinned GCHandles used to pass value from managed to unmanaged code.
		/// </summary>
		public static Dictionary<object, GCHandle> handles = new Dictionary<object, GCHandle>();

		/// <summary>
		/// Unpin the specified object and free the GCHandle associated.
		/// </summary>
		public static void Unpin (this object obj) {
			if (!handles.ContainsKey (obj)) {
				Debug.WriteLine ("Trying to unpin unpinned object: {0}.", obj);
				return;
			}
			handles[obj].Free ();
			handles.Remove (obj);
		}

		/// <summary>
		/// Pin the specified object and return a pointer. MUST be Unpined as soon as possible.
		/// </summary>
		public static IntPtr Pin (this object obj) {
			if (handles.ContainsKey (obj)) {
				Debug.WriteLine ("Trying to pin already pinned object: {0}", obj);
				return handles[obj].AddrOfPinnedObject ();
			}

			GCHandle hnd = GCHandle.Alloc (obj, GCHandleType.Pinned);
			handles.Add (obj, hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr Pin<T> (this IEnumerable<T> obj) {
			if (handles.ContainsKey (obj))
				Debug.WriteLine ("Pinning already pinned object: {0}", obj);

			GCHandle hnd = GCHandle.Alloc (obj.ToArray(), GCHandleType.Pinned);
			handles.Add (obj, hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr Pin<T> (this T[] obj) {
			if (handles.ContainsKey (obj))
				Debug.WriteLine ("Pinning already pinned object: {0}", obj);

			GCHandle hnd = GCHandle.Alloc (obj, GCHandleType.Pinned);
			handles.Add (obj, hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr Pin (this string obj) {
			if (handles.ContainsKey (obj)) {
				Debug.WriteLine ("Trying to pin already pinned object: {0}", obj);
				return handles[obj].AddrOfPinnedObject ();
			}
			byte[] n = System.Text.Encoding.UTF8.GetBytes(obj +'\0');
			GCHandle hnd = GCHandle.Alloc (n, GCHandleType.Pinned);
			handles.Add (obj, hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr PinPointer (this string obj) {
            byte[] n = System.Text.Encoding.UTF8.GetBytes(obj +'\0');
			GCHandle hnd = GCHandle.Alloc (n, GCHandleType.Pinned);
            handles.Add (hnd.AddrOfPinnedObject (), hnd);
            return hnd.AddrOfPinnedObject ();
        }
		public static IntPtr PinPointer (this object obj) {
			#if DEBUG
			if (obj == null)
				Debug.WriteLine ("Trying to pin null object: {0}", obj);
			#endif
			GCHandle hnd = GCHandle.Alloc (obj, GCHandleType.Pinned);
			handles.Add (hnd.AddrOfPinnedObject (), hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr PinPointer<T> (this IEnumerable<T> obj) {
			#if DEBUG
			if (obj == null)
				Debug.WriteLine ("Trying to pin null object: {0}", obj);
			#endif
			GCHandle hnd = GCHandle.Alloc (obj.ToArray(), GCHandleType.Pinned);
			handles.Add (hnd.AddrOfPinnedObject (), hnd);
			return hnd.AddrOfPinnedObject ();
		}

		//pin with pinning context
		public static IntPtr Pin (this object obj, PinnedObjects ctx) {
			GCHandle hnd = GCHandle.Alloc (obj, GCHandleType.Pinned);
			ctx.Handles.Add (hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr Pin<T> (this IList<T> obj, PinnedObjects ctx) {
			GCHandle hnd = GCHandle.Alloc (obj.ToArray (), GCHandleType.Pinned);
			ctx.Handles.Add (hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr Pin<T> (this T[] obj, PinnedObjects ctx) {
			GCHandle hnd = GCHandle.Alloc (obj, GCHandleType.Pinned);
			ctx.Handles.Add (hnd);
			return hnd.AddrOfPinnedObject ();
		}
		public static IntPtr Pin (this string obj, PinnedObjects ctx) {
			byte[] n = System.Text.Encoding.UTF8.GetBytes (obj + '\0');
			GCHandle hnd = GCHandle.Alloc (n, GCHandleType.Pinned);
			ctx.Handles.Add (hnd);
			return hnd.AddrOfPinnedObject ();
		}

		#endregion

	}
}