blob: f2e76e6bb1491d1d47253252d66433386446ffc1 [file] [log] [blame]
khenaidooab1f7bd2019-11-14 14:00:27 -05001package reflect2
2
3import (
4 "reflect"
5 "unsafe"
6)
7
8type UnsafeMapType struct {
9 unsafeType
10 pKeyRType unsafe.Pointer
11 pElemRType unsafe.Pointer
12}
13
14func newUnsafeMapType(cfg *frozenConfig, type1 reflect.Type) MapType {
15 return &UnsafeMapType{
16 unsafeType: *newUnsafeType(cfg, type1),
17 pKeyRType: unpackEFace(reflect.PtrTo(type1.Key())).data,
18 pElemRType: unpackEFace(reflect.PtrTo(type1.Elem())).data,
19 }
20}
21
22func (type2 *UnsafeMapType) IsNil(obj interface{}) bool {
23 if obj == nil {
24 return true
25 }
26 objEFace := unpackEFace(obj)
27 assertType("Type.IsNil argument 1", type2.ptrRType, objEFace.rtype)
28 return type2.UnsafeIsNil(objEFace.data)
29}
30
31func (type2 *UnsafeMapType) UnsafeIsNil(ptr unsafe.Pointer) bool {
32 if ptr == nil {
33 return true
34 }
35 return *(*unsafe.Pointer)(ptr) == nil
36}
37
38func (type2 *UnsafeMapType) LikePtr() bool {
39 return true
40}
41
42func (type2 *UnsafeMapType) Indirect(obj interface{}) interface{} {
43 objEFace := unpackEFace(obj)
44 assertType("MapType.Indirect argument 1", type2.ptrRType, objEFace.rtype)
45 return type2.UnsafeIndirect(objEFace.data)
46}
47
48func (type2 *UnsafeMapType) UnsafeIndirect(ptr unsafe.Pointer) interface{} {
49 return packEFace(type2.rtype, *(*unsafe.Pointer)(ptr))
50}
51
52func (type2 *UnsafeMapType) Key() Type {
53 return type2.cfg.Type2(type2.Type.Key())
54}
55
56func (type2 *UnsafeMapType) MakeMap(cap int) interface{} {
57 return packEFace(type2.ptrRType, type2.UnsafeMakeMap(cap))
58}
59
60func (type2 *UnsafeMapType) UnsafeMakeMap(cap int) unsafe.Pointer {
61 m := makeMapWithSize(type2.rtype, cap)
62 return unsafe.Pointer(&m)
63}
64
65func (type2 *UnsafeMapType) SetIndex(obj interface{}, key interface{}, elem interface{}) {
66 objEFace := unpackEFace(obj)
67 assertType("MapType.SetIndex argument 1", type2.ptrRType, objEFace.rtype)
68 keyEFace := unpackEFace(key)
69 assertType("MapType.SetIndex argument 2", type2.pKeyRType, keyEFace.rtype)
70 elemEFace := unpackEFace(elem)
71 assertType("MapType.SetIndex argument 3", type2.pElemRType, elemEFace.rtype)
72 type2.UnsafeSetIndex(objEFace.data, keyEFace.data, elemEFace.data)
73}
74
75func (type2 *UnsafeMapType) UnsafeSetIndex(obj unsafe.Pointer, key unsafe.Pointer, elem unsafe.Pointer) {
76 mapassign(type2.rtype, *(*unsafe.Pointer)(obj), key, elem)
77}
78
79func (type2 *UnsafeMapType) TryGetIndex(obj interface{}, key interface{}) (interface{}, bool) {
80 objEFace := unpackEFace(obj)
81 assertType("MapType.TryGetIndex argument 1", type2.ptrRType, objEFace.rtype)
82 keyEFace := unpackEFace(key)
83 assertType("MapType.TryGetIndex argument 2", type2.pKeyRType, keyEFace.rtype)
84 elemPtr := type2.UnsafeGetIndex(objEFace.data, keyEFace.data)
85 if elemPtr == nil {
86 return nil, false
87 }
88 return packEFace(type2.pElemRType, elemPtr), true
89}
90
91func (type2 *UnsafeMapType) GetIndex(obj interface{}, key interface{}) interface{} {
92 objEFace := unpackEFace(obj)
93 assertType("MapType.GetIndex argument 1", type2.ptrRType, objEFace.rtype)
94 keyEFace := unpackEFace(key)
95 assertType("MapType.GetIndex argument 2", type2.pKeyRType, keyEFace.rtype)
96 elemPtr := type2.UnsafeGetIndex(objEFace.data, keyEFace.data)
97 return packEFace(type2.pElemRType, elemPtr)
98}
99
100func (type2 *UnsafeMapType) UnsafeGetIndex(obj unsafe.Pointer, key unsafe.Pointer) unsafe.Pointer {
101 return mapaccess(type2.rtype, *(*unsafe.Pointer)(obj), key)
102}
103
104func (type2 *UnsafeMapType) Iterate(obj interface{}) MapIterator {
105 objEFace := unpackEFace(obj)
106 assertType("MapType.Iterate argument 1", type2.ptrRType, objEFace.rtype)
107 return type2.UnsafeIterate(objEFace.data)
108}
109
110func (type2 *UnsafeMapType) UnsafeIterate(obj unsafe.Pointer) MapIterator {
111 return &UnsafeMapIterator{
112 hiter: mapiterinit(type2.rtype, *(*unsafe.Pointer)(obj)),
113 pKeyRType: type2.pKeyRType,
114 pElemRType: type2.pElemRType,
115 }
116}
117
118type UnsafeMapIterator struct {
119 *hiter
120 pKeyRType unsafe.Pointer
121 pElemRType unsafe.Pointer
122}
123
124func (iter *UnsafeMapIterator) HasNext() bool {
125 return iter.key != nil
126}
127
128func (iter *UnsafeMapIterator) Next() (interface{}, interface{}) {
129 key, elem := iter.UnsafeNext()
130 return packEFace(iter.pKeyRType, key), packEFace(iter.pElemRType, elem)
131}
132
133func (iter *UnsafeMapIterator) UnsafeNext() (unsafe.Pointer, unsafe.Pointer) {
134 key := iter.key
135 elem := iter.value
136 mapiternext(iter.hiter)
137 return key, elem
138}