FilesExpand file tree

 master

/

InternString.cs

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

82 lines (73 loc) · 2.74 KB

 master

/

InternString.cs

Copy path

Top

File metadata and controls

• Code
• Blame

82 lines (73 loc) · 2.74 KB

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

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

using System;

using System.Collections.Generic;

using System.Diagnostics;

using System.Linq;

using System.Reflection;

namespace Python.Runtime

{

static partial class InternString

{

private static readonly Dictionary<string, PyString> _string2interns = new();

private static readonly Dictionary<IntPtr, string> _intern2strings = new();

const BindingFlags PyIdentifierFieldFlags = BindingFlags.Static | BindingFlags.NonPublic;

static InternString()

{

var identifierNames = typeof(PyIdentifier).GetFields(PyIdentifierFieldFlags)

.Select(fi => fi.Name.Substring(1));

var validNames = new HashSet<string>(identifierNames);

if (validNames.Count != _builtinNames.Length)

{

throw new InvalidOperationException("Identifiers args not matching");

}

foreach (var name in _builtinNames)

{

if (!validNames.Contains(name))

{

throw new InvalidOperationException($"{name} is not declared");

}

}

}

public static void Initialize()

{

Debug.Assert(_string2interns.Count == 0);

Type type = typeof(PyIdentifier);

foreach (string name in _builtinNames)

{

NewReference pyStr = Runtime.PyUnicode_InternFromString(name);

var op = new PyString(pyStr.StealOrThrow());

Debug.Assert(name == op.As<string>());

SetIntern(name, op);

var field = type.GetField("f" + name, PyIdentifierFieldFlags)!;

field.SetValue(null, op.DangerousGetAddressOrNull());

}

}

public static void Shutdown()

{

foreach (var entry in _string2interns)

{

var field = typeof(PyIdentifier).GetField("f" + entry.Value, PyIdentifierFieldFlags)!;

entry.Value.Dispose();

field.SetValue(null, IntPtr.Zero);

}

_string2interns.Clear();

_intern2strings.Clear();

}

public static string? GetManagedString(BorrowedReference op)

{

if (TryGetInterned(op, out string s))

{

return s;

}

return Runtime.GetManagedString(op);

}

public static bool TryGetInterned(BorrowedReference op, out string s)

{

return _intern2strings.TryGetValue(op.DangerousGetAddress(), out s);

}

private static void SetIntern(string s, PyString op)

{

_string2interns.Add(s, op);

_intern2strings.Add(op.Reference.DangerousGetAddress(), s);

}

}

}