reflect.odin 30 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345
  1. package reflect
  2. import "core:runtime"
  3. import "core:mem"
  4. import "intrinsics"
  5. _ :: intrinsics;
  6. Type_Info :: runtime.Type_Info;
  7. Type_Info_Named :: runtime.Type_Info_Named;
  8. Type_Info_Integer :: runtime.Type_Info_Integer;
  9. Type_Info_Rune :: runtime.Type_Info_Rune;
  10. Type_Info_Float :: runtime.Type_Info_Float;
  11. Type_Info_Complex :: runtime.Type_Info_Complex;
  12. Type_Info_Quaternion :: runtime.Type_Info_Quaternion;
  13. Type_Info_String :: runtime.Type_Info_String;
  14. Type_Info_Boolean :: runtime.Type_Info_Boolean;
  15. Type_Info_Any :: runtime.Type_Info_Any;
  16. Type_Info_Type_Id :: runtime.Type_Info_Type_Id;
  17. Type_Info_Pointer :: runtime.Type_Info_Pointer;
  18. Type_Info_Procedure :: runtime.Type_Info_Procedure;
  19. Type_Info_Array :: runtime.Type_Info_Array;
  20. Type_Info_Enumerated_Array :: runtime.Type_Info_Enumerated_Array;
  21. Type_Info_Dynamic_Array :: runtime.Type_Info_Dynamic_Array;
  22. Type_Info_Slice :: runtime.Type_Info_Slice;
  23. Type_Info_Tuple :: runtime.Type_Info_Tuple;
  24. Type_Info_Struct :: runtime.Type_Info_Struct;
  25. Type_Info_Union :: runtime.Type_Info_Union;
  26. Type_Info_Enum :: runtime.Type_Info_Enum;
  27. Type_Info_Map :: runtime.Type_Info_Map;
  28. Type_Info_Bit_Set :: runtime.Type_Info_Bit_Set;
  29. Type_Info_Simd_Vector :: runtime.Type_Info_Simd_Vector;
  30. Type_Info_Relative_Pointer :: runtime.Type_Info_Relative_Pointer;
  31. Type_Info_Relative_Slice :: runtime.Type_Info_Relative_Slice;
  32. Type_Info_Enum_Value :: runtime.Type_Info_Enum_Value;
  33. Type_Kind :: enum {
  34. Invalid,
  35. Named,
  36. Integer,
  37. Rune,
  38. Float,
  39. Complex,
  40. Quaternion,
  41. String,
  42. Boolean,
  43. Any,
  44. Type_Id,
  45. Pointer,
  46. Procedure,
  47. Array,
  48. Enumerated_Array,
  49. Dynamic_Array,
  50. Slice,
  51. Tuple,
  52. Struct,
  53. Union,
  54. Enum,
  55. Map,
  56. Bit_Set,
  57. Simd_Vector,
  58. Relative_Pointer,
  59. Relative_Slice,
  60. }
  61. type_kind :: proc(T: typeid) -> Type_Kind {
  62. ti := type_info_of(T);
  63. if ti != nil {
  64. switch _ in ti.variant {
  65. case Type_Info_Named: return .Named;
  66. case Type_Info_Integer: return .Integer;
  67. case Type_Info_Rune: return .Rune;
  68. case Type_Info_Float: return .Float;
  69. case Type_Info_Complex: return .Complex;
  70. case Type_Info_Quaternion: return .Quaternion;
  71. case Type_Info_String: return .String;
  72. case Type_Info_Boolean: return .Boolean;
  73. case Type_Info_Any: return .Any;
  74. case Type_Info_Type_Id: return .Type_Id;
  75. case Type_Info_Pointer: return .Pointer;
  76. case Type_Info_Procedure: return .Procedure;
  77. case Type_Info_Array: return .Array;
  78. case Type_Info_Enumerated_Array: return .Enumerated_Array;
  79. case Type_Info_Dynamic_Array: return .Dynamic_Array;
  80. case Type_Info_Slice: return .Slice;
  81. case Type_Info_Tuple: return .Tuple;
  82. case Type_Info_Struct: return .Struct;
  83. case Type_Info_Union: return .Union;
  84. case Type_Info_Enum: return .Enum;
  85. case Type_Info_Map: return .Map;
  86. case Type_Info_Bit_Set: return .Bit_Set;
  87. case Type_Info_Simd_Vector: return .Simd_Vector;
  88. case Type_Info_Relative_Pointer: return .Relative_Pointer;
  89. case Type_Info_Relative_Slice: return .Relative_Slice;
  90. }
  91. }
  92. return .Invalid;
  93. }
  94. // TODO(bill): Better name
  95. underlying_type_kind :: proc(T: typeid) -> Type_Kind {
  96. return type_kind(runtime.typeid_base(T));
  97. }
  98. // TODO(bill): Better name
  99. backing_type_kind :: proc(T: typeid) -> Type_Kind {
  100. return type_kind(runtime.typeid_core(T));
  101. }
  102. type_info_base :: proc(info: ^Type_Info) -> ^Type_Info {
  103. if info == nil { return nil; }
  104. base := info;
  105. loop: for {
  106. #partial switch i in base.variant {
  107. case Type_Info_Named: base = i.base;
  108. case: break loop;
  109. }
  110. }
  111. return base;
  112. }
  113. type_info_core :: proc(info: ^Type_Info) -> ^Type_Info {
  114. if info == nil { return nil; }
  115. base := info;
  116. loop: for {
  117. #partial switch i in base.variant {
  118. case Type_Info_Named: base = i.base;
  119. case Type_Info_Enum: base = i.base;
  120. case: break loop;
  121. }
  122. }
  123. return base;
  124. }
  125. type_info_base_without_enum :: type_info_core;
  126. typeid_base :: proc(id: typeid) -> typeid {
  127. ti := type_info_of(id);
  128. ti = type_info_base(ti);
  129. return ti.id;
  130. }
  131. typeid_core :: proc(id: typeid) -> typeid {
  132. ti := type_info_base_without_enum(type_info_of(id));
  133. return ti.id;
  134. }
  135. typeid_base_without_enum :: typeid_core;
  136. typeid_elem :: proc(id: typeid) -> typeid {
  137. ti := type_info_of(id);
  138. if ti == nil { return nil; }
  139. bits := 8*ti.size;
  140. #partial switch v in ti.variant {
  141. case Type_Info_Complex:
  142. switch bits {
  143. case 64: return f32;
  144. case 128: return f64;
  145. }
  146. case Type_Info_Quaternion:
  147. switch bits {
  148. case 128: return f32;
  149. case 256: return f64;
  150. }
  151. case Type_Info_Pointer: return v.elem.id;
  152. case Type_Info_Array: return v.elem.id;
  153. case Type_Info_Enumerated_Array: return v.elem.id;
  154. case Type_Info_Slice: return v.elem.id;
  155. case Type_Info_Dynamic_Array: return v.elem.id;
  156. }
  157. return id;
  158. }
  159. size_of_typeid :: proc(T: typeid) -> int {
  160. if ti := type_info_of(T); ti != nil {
  161. return ti.size;
  162. }
  163. return 0;
  164. }
  165. align_of_typeid :: proc(T: typeid) -> int {
  166. if ti := type_info_of(T); ti != nil {
  167. return ti.align;
  168. }
  169. return 1;
  170. }
  171. as_bytes :: proc(v: any) -> []byte {
  172. if v != nil {
  173. sz := size_of_typeid(v.id);
  174. return mem.slice_ptr((^byte)(v.data), sz);
  175. }
  176. return nil;
  177. }
  178. any_data :: #force_inline proc(v: any) -> (data: rawptr, id: typeid) {
  179. return v.data, v.id;
  180. }
  181. is_nil :: proc(v: any) -> bool {
  182. if v == nil {
  183. return true;
  184. }
  185. data := as_bytes(v);
  186. if data != nil {
  187. return true;
  188. }
  189. for v in data {
  190. if v != 0 {
  191. return false;
  192. }
  193. }
  194. return true;
  195. }
  196. length :: proc(val: any) -> int {
  197. if val == nil { return 0; }
  198. #partial switch a in type_info_of(val.id).variant {
  199. case Type_Info_Named:
  200. return length({val.data, a.base.id});
  201. case Type_Info_Pointer:
  202. return length({val.data, a.elem.id});
  203. case Type_Info_Array:
  204. return a.count;
  205. case Type_Info_Enumerated_Array:
  206. return a.count;
  207. case Type_Info_Slice:
  208. return (^mem.Raw_Slice)(val.data).len;
  209. case Type_Info_Dynamic_Array:
  210. return (^mem.Raw_Dynamic_Array)(val.data).len;
  211. case Type_Info_Map:
  212. return (^mem.Raw_Map)(val.data).entries.len;
  213. case Type_Info_String:
  214. if a.is_cstring {
  215. return len((^cstring)(val.data)^);
  216. } else {
  217. return (^mem.Raw_String)(val.data).len;
  218. }
  219. }
  220. return 0;
  221. }
  222. capacity :: proc(val: any) -> int {
  223. if val == nil { return 0; }
  224. #partial switch a in type_info_of(val.id).variant {
  225. case Type_Info_Named:
  226. return capacity({val.data, a.base.id});
  227. case Type_Info_Pointer:
  228. return capacity({val.data, a.elem.id});
  229. case Type_Info_Array:
  230. return a.count;
  231. case Type_Info_Enumerated_Array:
  232. return a.count;
  233. case Type_Info_Dynamic_Array:
  234. return (^mem.Raw_Dynamic_Array)(val.data).cap;
  235. case Type_Info_Map:
  236. return (^mem.Raw_Map)(val.data).entries.cap;
  237. }
  238. return 0;
  239. }
  240. index :: proc(val: any, i: int, loc := #caller_location) -> any {
  241. if val == nil { return nil; }
  242. #partial switch a in type_info_of(val.id).variant {
  243. case Type_Info_Named:
  244. return index({val.data, a.base.id}, i, loc);
  245. case Type_Info_Pointer:
  246. ptr := (^rawptr)(val.data)^;
  247. if ptr == nil {
  248. return nil;
  249. }
  250. return index({ptr, a.elem.id}, i, loc);
  251. case Type_Info_Array:
  252. runtime.bounds_check_error_loc(loc, i, a.count);
  253. offset := uintptr(a.elem.size * i);
  254. data := rawptr(uintptr(val.data) + offset);
  255. return any{data, a.elem.id};
  256. case Type_Info_Enumerated_Array:
  257. runtime.bounds_check_error_loc(loc, i, a.count);
  258. offset := uintptr(a.elem.size * i);
  259. data := rawptr(uintptr(val.data) + offset);
  260. return any{data, a.elem.id};
  261. case Type_Info_Slice:
  262. raw := (^mem.Raw_Slice)(val.data);
  263. runtime.bounds_check_error_loc(loc, i, raw.len);
  264. offset := uintptr(a.elem.size * i);
  265. data := rawptr(uintptr(raw.data) + offset);
  266. return any{data, a.elem.id};
  267. case Type_Info_Dynamic_Array:
  268. raw := (^mem.Raw_Dynamic_Array)(val.data);
  269. runtime.bounds_check_error_loc(loc, i, raw.len);
  270. offset := uintptr(a.elem.size * i);
  271. data := rawptr(uintptr(raw.data) + offset);
  272. return any{data, a.elem.id};
  273. case Type_Info_String:
  274. if a.is_cstring { return nil; }
  275. raw := (^mem.Raw_String)(val.data);
  276. runtime.bounds_check_error_loc(loc, i, raw.len);
  277. offset := uintptr(size_of(u8) * i);
  278. data := rawptr(uintptr(raw.data) + offset);
  279. return any{data, typeid_of(u8)};
  280. }
  281. return nil;
  282. }
  283. // Struct_Tag represents the type of the string of a struct field
  284. //
  285. // Through convention, tags are the concatenation of optionally space separationed key:"value" pairs.
  286. // Each key is a non-empty string which contains no control characters other than space, quotes, and colon.
  287. Struct_Tag :: distinct string;
  288. Struct_Field :: struct {
  289. name: string,
  290. type: ^Type_Info,
  291. tag: Struct_Tag,
  292. offset: uintptr,
  293. is_using: bool,
  294. }
  295. struct_field_at :: proc(T: typeid, i: int) -> (field: Struct_Field) {
  296. ti := runtime.type_info_base(type_info_of(T));
  297. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  298. if 0 <= i && i < len(s.names) {
  299. field.name = s.names[i];
  300. field.type = s.types[i];
  301. field.tag = Struct_Tag(s.tags[i]);
  302. field.offset = s.offsets[i];
  303. field.is_using = s.usings[i];
  304. }
  305. }
  306. return;
  307. }
  308. struct_field_by_name :: proc(T: typeid, name: string) -> (field: Struct_Field) {
  309. ti := runtime.type_info_base(type_info_of(T));
  310. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  311. for fname, i in s.names {
  312. if fname == name {
  313. field.name = s.names[i];
  314. field.type = s.types[i];
  315. field.tag = Struct_Tag(s.tags[i]);
  316. field.offset = s.offsets[i];
  317. field.is_using = s.usings[i];
  318. break;
  319. }
  320. }
  321. }
  322. return;
  323. }
  324. struct_field_value_by_name :: proc(a: any, field: string, allow_using := false) -> any {
  325. if a == nil { return nil; }
  326. ti := runtime.type_info_base(type_info_of(a.id));
  327. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  328. for name, i in s.names {
  329. if name == field {
  330. return any{
  331. rawptr(uintptr(a.data) + s.offsets[i]),
  332. s.types[i].id,
  333. };
  334. }
  335. if allow_using && s.usings[i] {
  336. f := any{
  337. rawptr(uintptr(a.data) + s.offsets[i]),
  338. s.types[i].id,
  339. };
  340. if res := struct_field_value_by_name(f, field, allow_using); res != nil {
  341. return res;
  342. }
  343. }
  344. }
  345. }
  346. return nil;
  347. }
  348. struct_field_names :: proc(T: typeid) -> []string {
  349. ti := runtime.type_info_base(type_info_of(T));
  350. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  351. return s.names;
  352. }
  353. return nil;
  354. }
  355. struct_field_types :: proc(T: typeid) -> []^Type_Info {
  356. ti := runtime.type_info_base(type_info_of(T));
  357. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  358. return s.types;
  359. }
  360. return nil;
  361. }
  362. struct_field_tags :: proc(T: typeid) -> []Struct_Tag {
  363. ti := runtime.type_info_base(type_info_of(T));
  364. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  365. return transmute([]Struct_Tag)s.tags;
  366. }
  367. return nil;
  368. }
  369. struct_field_offsets :: proc(T: typeid) -> []uintptr {
  370. ti := runtime.type_info_base(type_info_of(T));
  371. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  372. return s.offsets;
  373. }
  374. return nil;
  375. }
  376. struct_fields_zipped :: proc(T: typeid) -> (fields: #soa[]Struct_Field) {
  377. ti := runtime.type_info_base(type_info_of(T));
  378. if s, ok := ti.variant.(runtime.Type_Info_Struct); ok {
  379. return soa_zip(
  380. name = s.names,
  381. type = s.types,
  382. tag = transmute([]Struct_Tag)s.tags,
  383. offset = s.offsets,
  384. is_using = s.usings,
  385. );
  386. }
  387. return nil;
  388. }
  389. struct_tag_get :: proc(tag: Struct_Tag, key: string) -> (value: Struct_Tag) {
  390. value, _ = struct_tag_lookup(tag, key);
  391. return;
  392. }
  393. struct_tag_lookup :: proc(tag: Struct_Tag, key: string) -> (value: Struct_Tag, ok: bool) {
  394. for t := tag; t != ""; /**/ {
  395. i := 0;
  396. for i < len(t) && t[i] == ' ' { // Skip whitespace
  397. i += 1;
  398. }
  399. t = t[i:];
  400. if len(t) == 0 {
  401. break;
  402. }
  403. i = 0;
  404. loop: for i < len(t) {
  405. switch t[i] {
  406. case ':', '"':
  407. break loop;
  408. case 0x00 ..< ' ', 0x7f ..= 0x9f: // break if control character is found
  409. break loop;
  410. }
  411. i += 1;
  412. }
  413. if i == 0 {
  414. break;
  415. }
  416. if i+1 >= len(t) {
  417. break;
  418. }
  419. if t[i] != ':' || t[i+1] != '"' {
  420. break;
  421. }
  422. name := string(t[:i]);
  423. t = t[i+1:];
  424. i = 1;
  425. for i < len(t) && t[i] != '"' { // find closing quote
  426. if t[i] == '\\' {
  427. i += 1; // Skip escaped characters
  428. }
  429. i += 1;
  430. }
  431. if i >= len(t) {
  432. break;
  433. }
  434. val := string(t[:i+1]);
  435. t = t[i+1:];
  436. if key == name {
  437. return Struct_Tag(val[1:i]), true;
  438. }
  439. }
  440. return;
  441. }
  442. enum_string :: proc(a: any) -> string {
  443. if a == nil { return ""; }
  444. ti := runtime.type_info_base(type_info_of(a.id));
  445. if e, ok := ti.variant.(runtime.Type_Info_Enum); ok {
  446. v, _ := as_i64(a);
  447. for value, i in e.values {
  448. if value == Type_Info_Enum_Value(v) {
  449. return e.names[i];
  450. }
  451. }
  452. } else {
  453. panic("expected an enum to reflect.enum_string");
  454. }
  455. return "";
  456. }
  457. // Given a enum type and a value name, get the enum value.
  458. enum_from_name :: proc($Enum_Type: typeid, name: string) -> (value: Enum_Type, ok: bool) {
  459. ti := type_info_base(type_info_of(Enum_Type));
  460. if eti, eti_ok := ti.variant.(runtime.Type_Info_Enum); eti_ok {
  461. for value_name, i in eti.names {
  462. if value_name != name {
  463. continue;
  464. }
  465. v := eti.values[i];
  466. value = Enum_Type(v);
  467. ok = true;
  468. return;
  469. }
  470. }
  471. return;
  472. }
  473. enum_from_name_any :: proc(Enum_Type: typeid, name: string) -> (value: Type_Info_Enum_Value, ok: bool) {
  474. ti := runtime.type_info_base(type_info_of(Enum_Type));
  475. if eti, eti_ok := ti.variant.(runtime.Type_Info_Enum); eti_ok {
  476. for value_name, i in eti.names {
  477. if value_name != name {
  478. continue;
  479. }
  480. value = eti.values[i];
  481. ok = true;
  482. return;
  483. }
  484. }
  485. return;
  486. }
  487. enum_field_names :: proc(Enum_Type: typeid) -> []string {
  488. ti := runtime.type_info_base(type_info_of(Enum_Type));
  489. if eti, eti_ok := ti.variant.(runtime.Type_Info_Enum); eti_ok {
  490. return eti.names;
  491. }
  492. return nil;
  493. }
  494. enum_field_values :: proc(Enum_Type: typeid) -> []Type_Info_Enum_Value {
  495. ti := runtime.type_info_base(type_info_of(Enum_Type));
  496. if eti, eti_ok := ti.variant.(runtime.Type_Info_Enum); eti_ok {
  497. return eti.values;
  498. }
  499. return nil;
  500. }
  501. Enum_Field :: struct {
  502. name: string,
  503. value: Type_Info_Enum_Value,
  504. }
  505. enum_fields_zipped :: proc(Enum_Type: typeid) -> (fields: #soa[]Enum_Field) {
  506. ti := runtime.type_info_base(type_info_of(Enum_Type));
  507. if eti, eti_ok := ti.variant.(runtime.Type_Info_Enum); eti_ok {
  508. return soa_zip(name=eti.names, value=eti.values);
  509. }
  510. return nil;
  511. }
  512. union_variant_type_info :: proc(a: any) -> ^Type_Info {
  513. id := union_variant_typeid(a);
  514. return type_info_of(id);
  515. }
  516. type_info_union_is_pure_maybe :: proc(info: runtime.Type_Info_Union) -> bool {
  517. return info.maybe && len(info.variants) == 1 && is_pointer(info.variants[0]);
  518. }
  519. union_variant_typeid :: proc(a: any) -> typeid {
  520. if a == nil { return nil; }
  521. ti := runtime.type_info_base(type_info_of(a.id));
  522. if info, ok := ti.variant.(runtime.Type_Info_Union); ok {
  523. if type_info_union_is_pure_maybe(info) {
  524. if a.data != nil {
  525. return info.variants[0].id;
  526. }
  527. return nil;
  528. }
  529. tag_ptr := uintptr(a.data) + info.tag_offset;
  530. tag_any := any{rawptr(tag_ptr), info.tag_type.id};
  531. tag: i64 = ---;
  532. switch i in tag_any {
  533. case u8: tag = i64(i);
  534. case i8: tag = i64(i);
  535. case u16: tag = i64(i);
  536. case i16: tag = i64(i);
  537. case u32: tag = i64(i);
  538. case i32: tag = i64(i);
  539. case u64: tag = i64(i);
  540. case i64: tag = i;
  541. case: unimplemented();
  542. }
  543. if a.data != nil && tag != 0 {
  544. i := tag if info.no_nil else tag-1;
  545. return info.variants[i].id;
  546. }
  547. return nil;
  548. }
  549. panic("expected a union to reflect.union_variant_typeid");
  550. }
  551. get_union_variant_raw_tag :: proc(a: any) -> i64 {
  552. if a == nil { return -1; }
  553. ti := runtime.type_info_base(type_info_of(a.id));
  554. if info, ok := ti.variant.(runtime.Type_Info_Union); ok {
  555. if type_info_union_is_pure_maybe(info) {
  556. return 1 if a.data != nil else 0;
  557. }
  558. tag_ptr := uintptr(a.data) + info.tag_offset;
  559. tag_any := any{rawptr(tag_ptr), info.tag_type.id};
  560. tag: i64 = ---;
  561. switch i in tag_any {
  562. case u8: tag = i64(i);
  563. case i8: tag = i64(i);
  564. case u16: tag = i64(i);
  565. case i16: tag = i64(i);
  566. case u32: tag = i64(i);
  567. case i32: tag = i64(i);
  568. case u64: tag = i64(i);
  569. case i64: tag = i;
  570. case: unimplemented();
  571. }
  572. return tag;
  573. }
  574. panic("expected a union to reflect.get_union_variant_raw_tag");
  575. }
  576. set_union_variant_raw_tag :: proc(a: any, tag: i64) {
  577. if a == nil { return; }
  578. ti := runtime.type_info_base(type_info_of(a.id));
  579. if info, ok := ti.variant.(runtime.Type_Info_Union); ok {
  580. if type_info_union_is_pure_maybe(info) {
  581. // Cannot do anything
  582. return;
  583. }
  584. tag_ptr := uintptr(a.data) + info.tag_offset;
  585. tag_any := any{rawptr(tag_ptr), info.tag_type.id};
  586. switch i in &tag_any {
  587. case u8: i = u8(tag);
  588. case i8: i = i8(tag);
  589. case u16: i = u16(tag);
  590. case i16: i = i16(tag);
  591. case u32: i = u32(tag);
  592. case i32: i = i32(tag);
  593. case u64: i = u64(tag);
  594. case i64: i = tag;
  595. case: unimplemented();
  596. }
  597. return;
  598. }
  599. panic("expected a union to reflect.set_union_variant_raw_tag");
  600. }
  601. set_union_variant_typeid :: proc(a: any, id: typeid) {
  602. if a == nil { return; }
  603. ti := runtime.type_info_base(type_info_of(a.id));
  604. if info, ok := ti.variant.(runtime.Type_Info_Union); ok {
  605. if type_info_union_is_pure_maybe(info) {
  606. // Cannot do anything
  607. return;
  608. }
  609. if id == nil && !info.no_nil {
  610. set_union_variant_raw_tag(a, 0);
  611. return;
  612. }
  613. for variant, i in info.variants {
  614. if variant.id == id {
  615. tag := i64(i);
  616. if !info.no_nil {
  617. tag += 1;
  618. }
  619. set_union_variant_raw_tag(a, tag);
  620. return;
  621. }
  622. }
  623. return;
  624. }
  625. panic("expected a union to reflect.set_union_variant_typeid");
  626. }
  627. set_union_variant_type_info :: proc(a: any, tag_ti: ^Type_Info) {
  628. if a == nil { return; }
  629. ti := runtime.type_info_base(type_info_of(a.id));
  630. if info, ok := ti.variant.(runtime.Type_Info_Union); ok {
  631. if type_info_union_is_pure_maybe(info) {
  632. // Cannot do anything
  633. return;
  634. }
  635. if tag_ti == nil && !info.no_nil {
  636. set_union_variant_raw_tag(a, 0);
  637. return;
  638. }
  639. for variant, i in info.variants {
  640. if variant == tag_ti {
  641. tag := i64(i);
  642. if !info.no_nil {
  643. tag += 1;
  644. }
  645. set_union_variant_raw_tag(a, tag);
  646. return;
  647. }
  648. }
  649. return;
  650. }
  651. panic("expected a union to reflect.set_union_variant_type_info");
  652. }
  653. as_bool :: proc(a: any) -> (value: bool, valid: bool) {
  654. if a == nil { return; }
  655. a := a;
  656. ti := runtime.type_info_core(type_info_of(a.id));
  657. a.id = ti.id;
  658. #partial switch info in ti.variant {
  659. case Type_Info_Boolean:
  660. valid = true;
  661. switch v in a {
  662. case bool: value = v;
  663. case b8: value = bool(v);
  664. case b16: value = bool(v);
  665. case b32: value = bool(v);
  666. case b64: value = bool(v);
  667. case: valid = false;
  668. }
  669. }
  670. return;
  671. }
  672. as_int :: proc(a: any) -> (value: int, valid: bool) {
  673. v: i64;
  674. v, valid = as_i64(a);
  675. value = int(v);
  676. return;
  677. }
  678. as_uint :: proc(a: any) -> (value: uint, valid: bool) {
  679. v: u64;
  680. v, valid = as_u64(a);
  681. value = uint(v);
  682. return;
  683. }
  684. as_i64 :: proc(a: any) -> (value: i64, valid: bool) {
  685. if a == nil { return; }
  686. a := a;
  687. ti := runtime.type_info_core(type_info_of(a.id));
  688. a.id = ti.id;
  689. #partial switch info in ti.variant {
  690. case Type_Info_Integer:
  691. valid = true;
  692. switch v in a {
  693. case i8: value = i64(v);
  694. case i16: value = i64(v);
  695. case i32: value = i64(v);
  696. case i64: value = v;
  697. case i128: value = i64(v);
  698. case int: value = i64(v);
  699. case u8: value = i64(v);
  700. case u16: value = i64(v);
  701. case u32: value = i64(v);
  702. case u64: value = i64(v);
  703. case u128: value = i64(v);
  704. case uint: value = i64(v);
  705. case uintptr: value = i64(v);
  706. case u16le: value = i64(v);
  707. case u32le: value = i64(v);
  708. case u64le: value = i64(v);
  709. case u128le: value = i64(v);
  710. case i16le: value = i64(v);
  711. case i32le: value = i64(v);
  712. case i64le: value = i64(v);
  713. case i128le: value = i64(v);
  714. case u16be: value = i64(v);
  715. case u32be: value = i64(v);
  716. case u64be: value = i64(v);
  717. case u128be: value = i64(v);
  718. case i16be: value = i64(v);
  719. case i32be: value = i64(v);
  720. case i64be: value = i64(v);
  721. case i128be: value = i64(v);
  722. case: valid = false;
  723. }
  724. case Type_Info_Rune:
  725. r := a.(rune);
  726. value = i64(r);
  727. valid = true;
  728. case Type_Info_Float:
  729. valid = true;
  730. switch v in a {
  731. case f32: value = i64(v);
  732. case f64: value = i64(v);
  733. case f32le: value = i64(v);
  734. case f64le: value = i64(v);
  735. case f32be: value = i64(v);
  736. case f64be: value = i64(v);
  737. case: valid = false;
  738. }
  739. case Type_Info_Boolean:
  740. valid = true;
  741. switch v in a {
  742. case bool: value = i64(v);
  743. case b8: value = i64(v);
  744. case b16: value = i64(v);
  745. case b32: value = i64(v);
  746. case b64: value = i64(v);
  747. case: valid = false;
  748. }
  749. case Type_Info_Complex:
  750. switch v in a {
  751. case complex64:
  752. if imag(v) == 0 {
  753. value = i64(real(v));
  754. valid = true;
  755. }
  756. case complex128:
  757. if imag(v) == 0 {
  758. value = i64(real(v));
  759. valid = true;
  760. }
  761. }
  762. case Type_Info_Quaternion:
  763. switch v in a {
  764. case quaternion128:
  765. if imag(v) == 0 && jmag(v) == 0 && kmag(v) == 0 {
  766. value = i64(real(v));
  767. valid = true;
  768. }
  769. case quaternion256:
  770. if imag(v) == 0 && jmag(v) == 0 && kmag(v) == 0 {
  771. value = i64(real(v));
  772. valid = true;
  773. }
  774. }
  775. }
  776. return;
  777. }
  778. as_u64 :: proc(a: any) -> (value: u64, valid: bool) {
  779. if a == nil { return; }
  780. a := a;
  781. ti := runtime.type_info_core(type_info_of(a.id));
  782. a.id = ti.id;
  783. #partial switch info in ti.variant {
  784. case Type_Info_Integer:
  785. valid = true;
  786. switch v in a {
  787. case i8: value = u64(v);
  788. case i16: value = u64(v);
  789. case i32: value = u64(v);
  790. case i64: value = u64(v);
  791. case i128: value = u64(v);
  792. case int: value = u64(v);
  793. case u8: value = u64(v);
  794. case u16: value = u64(v);
  795. case u32: value = u64(v);
  796. case u64: value = (v);
  797. case u128: value = u64(v);
  798. case uint: value = u64(v);
  799. case uintptr:value = u64(v);
  800. case u16le: value = u64(v);
  801. case u32le: value = u64(v);
  802. case u64le: value = u64(v);
  803. case u128le: value = u64(v);
  804. case i16le: value = u64(v);
  805. case i32le: value = u64(v);
  806. case i64le: value = u64(v);
  807. case i128le: value = u64(v);
  808. case u16be: value = u64(v);
  809. case u32be: value = u64(v);
  810. case u64be: value = u64(v);
  811. case u128be: value = u64(v);
  812. case i16be: value = u64(v);
  813. case i32be: value = u64(v);
  814. case i64be: value = u64(v);
  815. case i128be: value = u64(v);
  816. case: valid = false;
  817. }
  818. case Type_Info_Rune:
  819. r := a.(rune);
  820. value = u64(r);
  821. valid = true;
  822. case Type_Info_Float:
  823. valid = true;
  824. switch v in a {
  825. case f32: value = u64(v);
  826. case f64: value = u64(v);
  827. case f32le: value = u64(v);
  828. case f64le: value = u64(v);
  829. case f32be: value = u64(v);
  830. case f64be: value = u64(v);
  831. case: valid = false;
  832. }
  833. case Type_Info_Boolean:
  834. valid = true;
  835. switch v in a {
  836. case bool: value = u64(v);
  837. case b8: value = u64(v);
  838. case b16: value = u64(v);
  839. case b32: value = u64(v);
  840. case b64: value = u64(v);
  841. case: valid = false;
  842. }
  843. case Type_Info_Complex:
  844. switch v in a {
  845. case complex64:
  846. if imag(v) == 0 {
  847. value = u64(real(v));
  848. valid = true;
  849. }
  850. case complex128:
  851. if imag(v) == 0 {
  852. value = u64(real(v));
  853. valid = true;
  854. }
  855. }
  856. case Type_Info_Quaternion:
  857. switch v in a {
  858. case quaternion128:
  859. if imag(v) == 0 && jmag(v) == 0 && kmag(v) == 0 {
  860. value = u64(real(v));
  861. valid = true;
  862. }
  863. case quaternion256:
  864. if imag(v) == 0 && jmag(v) == 0 && kmag(v) == 0 {
  865. value = u64(real(v));
  866. valid = true;
  867. }
  868. }
  869. }
  870. return;
  871. }
  872. as_f64 :: proc(a: any) -> (value: f64, valid: bool) {
  873. if a == nil { return; }
  874. a := a;
  875. ti := runtime.type_info_core(type_info_of(a.id));
  876. a.id = ti.id;
  877. #partial switch info in ti.variant {
  878. case Type_Info_Integer:
  879. valid = true;
  880. switch v in a {
  881. case i8: value = f64(v);
  882. case i16: value = f64(v);
  883. case i32: value = f64(v);
  884. case i64: value = f64(v);
  885. case i128: value = f64(v);
  886. case u8: value = f64(v);
  887. case u16: value = f64(v);
  888. case u32: value = f64(v);
  889. case u64: value = f64(v);
  890. case u128: value = f64(v);
  891. case u16le: value = f64(v);
  892. case u32le: value = f64(v);
  893. case u64le: value = f64(v);
  894. case u128le:value = f64(v);
  895. case i16le: value = f64(v);
  896. case i32le: value = f64(v);
  897. case i64le: value = f64(v);
  898. case i128le:value = f64(v);
  899. case u16be: value = f64(v);
  900. case u32be: value = f64(v);
  901. case u64be: value = f64(v);
  902. case u128be:value = f64(v);
  903. case i16be: value = f64(v);
  904. case i32be: value = f64(v);
  905. case i64be: value = f64(v);
  906. case i128be:value = f64(v);
  907. case: valid = false;
  908. }
  909. case Type_Info_Rune:
  910. r := a.(rune);
  911. value = f64(i32(r));
  912. valid = true;
  913. case Type_Info_Float:
  914. valid = true;
  915. switch v in a {
  916. case f32: value = f64(v);
  917. case f64: value = (v);
  918. case f32le: value = f64(v);
  919. case f64le: value = f64(v);
  920. case f32be: value = f64(v);
  921. case f64be: value = f64(v);
  922. case: valid = false;
  923. }
  924. case Type_Info_Boolean:
  925. valid = true;
  926. switch v in a {
  927. case bool: value = f64(i32(v));
  928. case b8: value = f64(i32(v));
  929. case b16: value = f64(i32(v));
  930. case b32: value = f64(i32(v));
  931. case b64: value = f64(i32(v));
  932. case: valid = false;
  933. }
  934. case Type_Info_Complex:
  935. switch v in a {
  936. case complex64:
  937. if imag(v) == 0 {
  938. value = f64(real(v));
  939. valid = true;
  940. }
  941. case complex128:
  942. if imag(v) == 0 {
  943. value = real(v);
  944. valid = true;
  945. }
  946. }
  947. case Type_Info_Quaternion:
  948. switch v in a {
  949. case quaternion128:
  950. if imag(v) == 0 && jmag(v) == 0 && kmag(v) == 0 {
  951. value = f64(real(v));
  952. valid = true;
  953. }
  954. case quaternion256:
  955. if imag(v) == 0 && jmag(v) == 0 && kmag(v) == 0 {
  956. value = real(v);
  957. valid = true;
  958. }
  959. }
  960. }
  961. return;
  962. }
  963. as_string :: proc(a: any) -> (value: string, valid: bool) {
  964. if a == nil { return; }
  965. a := a;
  966. ti := runtime.type_info_core(type_info_of(a.id));
  967. a.id = ti.id;
  968. #partial switch info in ti.variant {
  969. case Type_Info_String:
  970. valid = true;
  971. switch v in a {
  972. case string: value = v;
  973. case cstring: value = string(v);
  974. case: valid = false;
  975. }
  976. }
  977. return;
  978. }
  979. relative_pointer_to_absolute :: proc(a: any) -> rawptr {
  980. if a == nil { return nil; }
  981. a := a;
  982. ti := runtime.type_info_core(type_info_of(a.id));
  983. a.id = ti.id;
  984. #partial switch info in ti.variant {
  985. case Type_Info_Relative_Pointer:
  986. return relative_pointer_to_absolute_raw(a.data, info.base_integer.id);
  987. }
  988. return nil;
  989. }
  990. relative_pointer_to_absolute_raw :: proc(data: rawptr, base_integer_id: typeid) -> rawptr {
  991. _handle :: proc(ptr: ^$T) -> rawptr where intrinsics.type_is_integer(T) {
  992. if ptr^ == 0 {
  993. return nil;
  994. }
  995. when intrinsics.type_is_unsigned(T) {
  996. return rawptr(uintptr(ptr) + uintptr(ptr^));
  997. } else {
  998. return rawptr(uintptr(ptr) + uintptr(i64(ptr^)));
  999. }
  1000. }
  1001. ptr_any := any{data, base_integer_id};
  1002. ptr: rawptr;
  1003. switch i in &ptr_any {
  1004. case u8: ptr = _handle(&i);
  1005. case u16: ptr = _handle(&i);
  1006. case u32: ptr = _handle(&i);
  1007. case u64: ptr = _handle(&i);
  1008. case i8: ptr = _handle(&i);
  1009. case i16: ptr = _handle(&i);
  1010. case i32: ptr = _handle(&i);
  1011. case i64: ptr = _handle(&i);
  1012. case u16le: ptr = _handle(&i);
  1013. case u32le: ptr = _handle(&i);
  1014. case u64le: ptr = _handle(&i);
  1015. case i16le: ptr = _handle(&i);
  1016. case i32le: ptr = _handle(&i);
  1017. case i64le: ptr = _handle(&i);
  1018. case u16be: ptr = _handle(&i);
  1019. case u32be: ptr = _handle(&i);
  1020. case u64be: ptr = _handle(&i);
  1021. case i16be: ptr = _handle(&i);
  1022. case i32be: ptr = _handle(&i);
  1023. case i64be: ptr = _handle(&i);
  1024. }
  1025. return ptr;
  1026. }
  1027. as_pointer :: proc(a: any) -> (value: rawptr, valid: bool) {
  1028. if a == nil { return; }
  1029. a := a;
  1030. ti := runtime.type_info_core(type_info_of(a.id));
  1031. a.id = ti.id;
  1032. #partial switch info in ti.variant {
  1033. case Type_Info_Pointer:
  1034. valid = true;
  1035. value = a.data;
  1036. case Type_Info_String:
  1037. valid = true;
  1038. switch v in a {
  1039. case cstring: value = rawptr(v);
  1040. case: valid = false;
  1041. }
  1042. case Type_Info_Relative_Pointer:
  1043. valid = true;
  1044. value = relative_pointer_to_absolute_raw(a.data, info.base_integer.id);
  1045. }
  1046. return;
  1047. }
  1048. as_raw_data :: proc(a: any) -> (value: rawptr, valid: bool) {
  1049. if a == nil { return; }
  1050. a := a;
  1051. ti := runtime.type_info_core(type_info_of(a.id));
  1052. a.id = ti.id;
  1053. #partial switch info in ti.variant {
  1054. case Type_Info_String:
  1055. valid = true;
  1056. switch v in a {
  1057. case string: value = raw_data(v);
  1058. case cstring: value = rawptr(v); // just in case
  1059. case: valid = false;
  1060. }
  1061. case Type_Info_Array:
  1062. valid = true;
  1063. value = a.data;
  1064. case Type_Info_Slice:
  1065. valid = true;
  1066. value = (^mem.Raw_Slice)(a.data).data;
  1067. case Type_Info_Dynamic_Array:
  1068. valid = true;
  1069. value = (^mem.Raw_Dynamic_Array)(a.data).data;
  1070. }
  1071. return;
  1072. }
  1073. /*
  1074. not_equal :: proc(a, b: any) -> bool {
  1075. return !equal(a, b);
  1076. }
  1077. equal :: proc(a, b: any) -> bool {
  1078. if a == nil && b == nil {
  1079. return true;
  1080. }
  1081. if a.id != b.id {
  1082. return false;
  1083. }
  1084. if a.data == b.data {
  1085. return true;
  1086. }
  1087. t := type_info_of(a.id);
  1088. if .Comparable not_in t.flags {
  1089. return false;
  1090. }
  1091. if t.size == 0 {
  1092. return true;
  1093. }
  1094. if .Simple_Compare in t.flags {
  1095. return mem.compare_byte_ptrs((^byte)(a.data), (^byte)(b.data), t.size) == 0;
  1096. }
  1097. t = runtime.type_info_core(t);
  1098. #partial switch v in t.variant {
  1099. case Type_Info_String:
  1100. if v.is_cstring {
  1101. x := string((^cstring)(a.data)^);
  1102. y := string((^cstring)(b.data)^);
  1103. return x == y;
  1104. } else {
  1105. x := (^string)(a.data)^;
  1106. y := (^string)(b.data)^;
  1107. return x == y;
  1108. }
  1109. case Type_Info_Array:
  1110. for i in 0..<v.count {
  1111. x := rawptr(uintptr(a.data) + uintptr(v.elem_size*i));
  1112. y := rawptr(uintptr(b.data) + uintptr(v.elem_size*i));
  1113. if !equal(any{x, v.elem.id}, any{y, v.elem.id}) {
  1114. return false;
  1115. }
  1116. }
  1117. case Type_Info_Enumerated_Array:
  1118. for i in 0..<v.count {
  1119. x := rawptr(uintptr(a.data) + uintptr(v.elem_size*i));
  1120. y := rawptr(uintptr(b.data) + uintptr(v.elem_size*i));
  1121. if !equal(any{x, v.elem.id}, any{y, v.elem.id}) {
  1122. return false;
  1123. }
  1124. }
  1125. case Type_Info_Struct:
  1126. if v.equal != nil {
  1127. return v.equal(a.data, b.data);
  1128. } else {
  1129. for offset, i in v.offsets {
  1130. x := rawptr(uintptr(a.data) + offset);
  1131. y := rawptr(uintptr(b.data) + offset);
  1132. id := v.types[i].id;
  1133. if !equal(any{x, id}, any{y, id}) {
  1134. return false;
  1135. }
  1136. }
  1137. }
  1138. }
  1139. return true;
  1140. }
  1141. */