| FazBrowse GitHub Viewer | Trending | | Home |
| Tools: [Download Repo ZIP] [Original HTTPS Page] |
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,4 +1,5 @@ | ||
| using System; | ||
| using System.Diagnostics; | ||
| using NUnit.Framework; | ||
| using Python.Runtime; | ||
|
|
||
| Expand Down | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,18 @@ | ||
| using NUnit.Framework; | ||
| using Python.Runtime; | ||
|
|
||
| namespace Python.EmbeddingTest | ||
| { | ||
| [SetUpFixture] | ||
| public class TestsSuite | ||
| { | ||
| [OneTimeTearDown] | ||
| public void FinalCleanup() | ||
| { | ||
| if (PythonEngine.IsInitialized) | ||
| { | ||
| PythonEngine.Shutdown(); | ||
| } | ||
| } | ||
| } | ||
| } |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| Expand Up | @@ -18,7 +18,7 @@ public object MarshalNativeToManaged(IntPtr pNativeData) | |
|
|
||
| public abstract IntPtr MarshalManagedToNative(object managedObj); | ||
|
|
||
| public void CleanUpNativeData(IntPtr pNativeData) | ||
| public virtual void CleanUpNativeData(IntPtr pNativeData) | ||
| { | ||
| Marshal.FreeHGlobal(pNativeData); | ||
| } | ||
| Expand All | @@ -44,7 +44,12 @@ internal class UcsMarshaler : MarshalerBase | |
| private static readonly MarshalerBase Instance = new UcsMarshaler(); | ||
| private static readonly Encoding PyEncoding = Runtime.PyEncoding; | ||
|
|
||
| public override IntPtr MarshalManagedToNative(object managedObj) | ||
| private const int MaxStringLength = 100; | ||
| private const int MaxItemSize = 4 * (MaxStringLength + 1); | ||
| private static readonly EncodedStringsFifoDictionary EncodedStringsDictionary = | ||
| new EncodedStringsFifoDictionary(10000, MaxItemSize); | ||
|
|
||
| public override unsafe IntPtr MarshalManagedToNative(object managedObj) | ||
| { | ||
| var s = managedObj as string; | ||
|
|
||
| Expand All | @@ -53,16 +58,36 @@ public override IntPtr MarshalManagedToNative(object managedObj) | |
| return IntPtr.Zero; | ||
| } | ||
|
|
||
| byte[] bStr = PyEncoding.GetBytes(s + "\0"); | ||
| IntPtr mem = Marshal.AllocHGlobal(bStr.Length); | ||
| try | ||
| IntPtr mem; | ||
| int stringBytesCount; | ||
| if (s.Length <= MaxStringLength) | ||
| { | ||
| Marshal.Copy(bStr, 0, mem, bStr.Length); | ||
| if (EncodedStringsDictionary.TryGetValue(s, out mem)) | ||
| { | ||
| return mem; | ||
| } | ||
|
|
||
| stringBytesCount = PyEncoding.GetByteCount(s); | ||
| mem = EncodedStringsDictionary.AddUnsafe(s); | ||
| } | ||
| catch (Exception) | ||
| else | ||
| { | ||
| Marshal.FreeHGlobal(mem); | ||
| throw; | ||
| stringBytesCount = PyEncoding.GetByteCount(s); | ||
| mem = Marshal.AllocHGlobal(stringBytesCount + 4); | ||
| } | ||
|
|
||
| fixed (char* str = s) | ||
| { | ||
| try | ||
| { | ||
| PyEncoding.GetBytes(str, s.Length, (byte*)mem, stringBytesCount); | ||
| } | ||
| catch | ||
| { | ||
| // Do nothing with this. Very strange problem. | ||
|
Comment thread
Copy link
Copy Markdown
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low QualityWhen does this happen?
Sorry, something went wrong.
All reactions
|
||
| } | ||
|
|
||
| *(int*)(mem + stringBytesCount) = 0; | ||
| } | ||
|
|
||
| return mem; | ||
| Expand Down Expand Up | @@ -106,6 +131,14 @@ public static int GetUnicodeByteLength(IntPtr p) | |
| } | ||
| } | ||
|
|
||
| public override void CleanUpNativeData(IntPtr pNativeData) | ||
| { | ||
| if (!EncodedStringsDictionary.IsKnownPtr(pNativeData)) | ||
| { | ||
| base.CleanUpNativeData(pNativeData); | ||
| } | ||
| } | ||
|
|
||
| /// <summary> | ||
| /// Utility function for Marshaling Unicode on PY3 and AnsiStr on PY2. | ||
| /// Use on functions whose Input signatures changed between PY2/PY3. | ||
| Expand All | @@ -118,11 +151,29 @@ public static int GetUnicodeByteLength(IntPtr p) | |
| /// <remarks> | ||
| /// You MUST deallocate the IntPtr of the Return when done with it. | ||
| /// </remarks> | ||
| public static IntPtr Py3UnicodePy2StringtoPtr(string s) | ||
| public unsafe static IntPtr Py3UnicodePy2StringtoPtr(string s) | ||
| { | ||
| return Runtime.IsPython3 | ||
| ? Instance.MarshalManagedToNative(s) | ||
| : Marshal.StringToHGlobalAnsi(s); | ||
| if (Runtime.IsPython3) | ||
|
Comment thread
Copy link
Copy Markdown
Member
There was a problem hiding this comment. Choose a reason for hiding this commentThe reason will be displayed to describe this comment to others. Learn more. Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low QualityThis means that your optimisation will only work at all for Python 3, is that necessary?
Sorry, something went wrong.
All reactions
|
||
| { | ||
| int stringBytesCount = PyEncoding.GetByteCount(s); | ||
| IntPtr mem = Marshal.AllocHGlobal(stringBytesCount + 4); | ||
| fixed (char* str = s) | ||
| { | ||
| try | ||
| { | ||
| PyEncoding.GetBytes(str, s.Length, (byte*)mem, stringBytesCount); | ||
| } | ||
| catch | ||
| { | ||
| // Do nothing with this. Very strange problem. | ||
| } | ||
|
|
||
| *(int*)(mem + stringBytesCount) = 0; | ||
| } | ||
| return mem; | ||
| } | ||
|
|
||
| return Marshal.StringToHGlobalAnsi(s); | ||
| } | ||
|
|
||
| /// <summary> | ||
| Expand Down Expand Up | @@ -208,7 +259,12 @@ internal class Utf8Marshaler : MarshalerBase | |
| private static readonly MarshalerBase Instance = new Utf8Marshaler(); | ||
| private static readonly Encoding PyEncoding = Encoding.UTF8; | ||
|
|
||
| public override IntPtr MarshalManagedToNative(object managedObj) | ||
| private const int MaxStringLength = 100; | ||
|
|
||
| private static readonly EncodedStringsFifoDictionary EncodedStringsDictionary = | ||
| new EncodedStringsFifoDictionary(10000, 4 * (MaxStringLength + 1)); | ||
|
|
||
| public override unsafe IntPtr MarshalManagedToNative(object managedObj) | ||
| { | ||
| var s = managedObj as string; | ||
|
|
||
| Expand All | @@ -217,21 +273,49 @@ public override IntPtr MarshalManagedToNative(object managedObj) | |
| return IntPtr.Zero; | ||
| } | ||
|
|
||
| byte[] bStr = PyEncoding.GetBytes(s + "\0"); | ||
| IntPtr mem = Marshal.AllocHGlobal(bStr.Length); | ||
| try | ||
| IntPtr mem; | ||
| int stringBytesCount; | ||
| if (s.Length <= MaxStringLength) | ||
| { | ||
| Marshal.Copy(bStr, 0, mem, bStr.Length); | ||
| if (EncodedStringsDictionary.TryGetValue(s, out mem)) | ||
| { | ||
| return mem; | ||
| } | ||
|
|
||
| stringBytesCount = PyEncoding.GetByteCount(s); | ||
| mem = EncodedStringsDictionary.AddUnsafe(s); | ||
| } | ||
| catch (Exception) | ||
| else | ||
| { | ||
| Marshal.FreeHGlobal(mem); | ||
| throw; | ||
| stringBytesCount = PyEncoding.GetByteCount(s); | ||
| mem = Marshal.AllocHGlobal(stringBytesCount + 1); | ||
| } | ||
|
|
||
| fixed (char* str = s) | ||
| { | ||
| try | ||
| { | ||
| PyEncoding.GetBytes(str, s.Length, (byte*)mem, stringBytesCount); | ||
| } | ||
| catch | ||
| { | ||
| // Do nothing with this. Very strange problem. | ||
| } | ||
|
|
||
| ((byte*)mem)[stringBytesCount] = 0; | ||
| } | ||
|
|
||
| return mem; | ||
| } | ||
|
|
||
| public override void CleanUpNativeData(IntPtr pNativeData) | ||
| { | ||
| if (!EncodedStringsDictionary.IsKnownPtr(pNativeData)) | ||
| { | ||
| base.CleanUpNativeData(pNativeData); | ||
| } | ||
| } | ||
|
|
||
| public static ICustomMarshaler GetInstance(string cookie) | ||
| { | ||
| return Instance; | ||
| Expand Down | ||
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,73 @@ | ||
| using System; | ||
|
|
||
| namespace Python.Runtime | ||
| { | ||
| using System.Runtime.InteropServices; | ||
|
|
||
| public class EncodedStringsFifoDictionary: IDisposable | ||
| { | ||
| private readonly FifoDictionary<string, IntPtr> _innerDictionary; | ||
|
|
||
| private readonly IntPtr _rawMemory; | ||
|
|
||
| private readonly int _allocatedSize; | ||
|
|
||
| public EncodedStringsFifoDictionary(int capacity, int maxItemSize) | ||
| { | ||
| if (maxItemSize < 1) | ||
| { | ||
| throw new ArgumentOutOfRangeException( | ||
| nameof(maxItemSize), | ||
| "Maximum item size should be non-zero positive."); | ||
| } | ||
|
|
||
| _innerDictionary = new FifoDictionary<string, IntPtr>(capacity); | ||
| _allocatedSize = maxItemSize * capacity; | ||
| _rawMemory = Marshal.AllocHGlobal(_allocatedSize); | ||
|
|
||
| MaxItemSize = maxItemSize; | ||
| } | ||
|
|
||
| public int MaxItemSize { get; } | ||
|
|
||
| public bool TryGetValue(string key, out IntPtr value) | ||
| { | ||
| return _innerDictionary.TryGetValue(key, out value); | ||
| } | ||
|
|
||
| public IntPtr AddUnsafe(string key) | ||
| { | ||
| int nextSlot = _innerDictionary.NextSlotToAdd; | ||
| IntPtr ptr = _rawMemory + (MaxItemSize * nextSlot); | ||
| _innerDictionary.AddUnsafe(key, ptr); | ||
| return ptr; | ||
| } | ||
|
|
||
| public bool IsKnownPtr(IntPtr ptr) | ||
| { | ||
| var uptr = (ulong)ptr; | ||
| var umem = (ulong)_rawMemory; | ||
|
|
||
| return uptr >= umem && uptr < umem + (ulong)_allocatedSize; | ||
| } | ||
|
|
||
| private void ReleaseUnmanagedResources() | ||
| { | ||
| if (_rawMemory != IntPtr.Zero) | ||
| { | ||
| Marshal.FreeHGlobal(_rawMemory); | ||
| } | ||
| } | ||
|
|
||
| public void Dispose() | ||
| { | ||
| ReleaseUnmanagedResources(); | ||
| GC.SuppressFinalize(this); | ||
| } | ||
|
|
||
| ~EncodedStringsFifoDictionary() | ||
| { | ||
| ReleaseUnmanagedResources(); | ||
| } | ||
| } | ||
| } |
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,53 @@ | ||
| using System; | ||
| using System.Collections.Generic; | ||
| using System.Reflection; | ||
| using System.Runtime.InteropServices; | ||
| using System.Text; | ||
|
|
||
| namespace Python.Runtime | ||
| { | ||
| #if !NETSTANDARD | ||
| /// <summary> | ||
| /// This polyfill is thread unsafe. | ||
| /// </summary> | ||
| [CLSCompliant(false)] | ||
| public static class EncodingGetStringPolyfill | ||
| { | ||
| private static readonly MethodInfo PlatformGetStringMethodInfo = | ||
| typeof(Encoding).GetMethod( | ||
| "GetString", | ||
| BindingFlags.Instance | BindingFlags.Public | BindingFlags.NonPublic, null, | ||
| new[] | ||
| { | ||
| typeof(byte*), typeof(int) | ||
| }, null); | ||
|
|
||
| private static readonly byte[] StdDecodeBuffer = PlatformGetStringMethodInfo == null ? new byte[1024 * 1024] : null; | ||
|
|
||
| private static Dictionary<Encoding, EncodingGetStringUnsafeDelegate> PlatformGetStringMethodsDelegatesCache = new Dictionary<Encoding, EncodingGetStringUnsafeDelegate>(); | ||
|
|
||
| private unsafe delegate string EncodingGetStringUnsafeDelegate(byte* pstr, int size); | ||
|
|
||
| public unsafe static string GetString(this Encoding encoding, byte* pstr, int size) | ||
| { | ||
| if (PlatformGetStringMethodInfo != null) | ||
| { | ||
| EncodingGetStringUnsafeDelegate getStringDelegate; | ||
| if (!PlatformGetStringMethodsDelegatesCache.TryGetValue(encoding, out getStringDelegate)) | ||
| { | ||
| getStringDelegate = | ||
| (EncodingGetStringUnsafeDelegate)Delegate.CreateDelegate( | ||
| typeof(EncodingGetStringUnsafeDelegate), encoding, PlatformGetStringMethodInfo); | ||
| PlatformGetStringMethodsDelegatesCache.Add(encoding, getStringDelegate); | ||
| } | ||
| return getStringDelegate(pstr, size); | ||
| } | ||
|
|
||
| byte[] buffer = size <= StdDecodeBuffer.Length ? StdDecodeBuffer : new byte[size]; | ||
| Marshal.Copy((IntPtr)pstr, buffer, 0, size); | ||
| return encoding.GetString(buffer, 0, size); | ||
| } | ||
| } | ||
| #endif | ||
|
|
||
| } |
| Back | FazBrowse Home | New Git URL |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Choose a reason Spam Abuse Off Topic Outdated Duplicate Resolved Low QualityWhy +4, wouldn't one additional byte be enough?
Sorry, something went wrong.
Uh oh!
There was an error while loading. Please reload this page.