demo.odin 46 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656165716581659166016611662166316641665166616671668166916701671167216731674167516761677167816791680168116821683168416851686168716881689169016911692169316941695169616971698169917001701170217031704170517061707170817091710171117121713171417151716171717181719172017211722172317241725172617271728172917301731173217331734173517361737173817391740174117421743174417451746174717481749175017511752175317541755175617571758175917601761176217631764176517661767176817691770177117721773177417751776177717781779178017811782178317841785178617871788178917901791179217931794179517961797179817991800180118021803180418051806180718081809181018111812181318141815181618171818181918201821182218231824182518261827182818291830183118321833183418351836183718381839184018411842184318441845184618471848184918501851185218531854185518561857185818591860186118621863186418651866186718681869187018711872187318741875187618771878187918801881188218831884188518861887188818891890189118921893189418951896189718981899190019011902190319041905190619071908190919101911191219131914191519161917191819191920192119221923192419251926192719281929193019311932193319341935193619371938193919401941194219431944194519461947194819491950195119521953195419551956195719581959196019611962196319641965196619671968196919701971197219731974197519761977197819791980198119821983198419851986198719881989199019911992199319941995199619971998199920002001200220032004200520062007200820092010201120122013201420152016201720182019202020212022202320242025202620272028202920302031203220332034
  1. package main
  2. import "core:fmt"
  3. import "core:mem"
  4. import "core:os"
  5. import "core:thread"
  6. import "core:time"
  7. import "core:reflect"
  8. import "core:runtime"
  9. import "intrinsics"
  10. /*
  11. The Odin programming language is fast, concise, readable, pragmatic and open sourced.
  12. It is designed with the intent of replacing C with the following goals:
  13. * simplicity
  14. * high performance
  15. * built for modern systems
  16. * joy of programming
  17. # Installing Odin
  18. Getting Started - https://odin-lang.org/docs/install/
  19. Instructions for downloading and install the Odin compiler and libraries.
  20. # Learning Odin
  21. Overview of Odin - https://odin-lang.org/docs/overview/
  22. An overview of the Odin programming language.
  23. Frequently Asked Questions (FAQ) - https://odin-lang.org/docs/faq/
  24. Answers to common questions about Odin.
  25. */
  26. the_basics :: proc() {
  27. fmt.println("\n# the basics");
  28. { // The Basics
  29. fmt.println("Hellope");
  30. // Lexical elements and literals
  31. // A comment
  32. my_integer_variable: int; // A comment for documentaton
  33. // Multi-line comments begin with /* and end with */. Multi-line comments can
  34. // also be nested (unlike in C):
  35. /*
  36. You can have any text or code here and
  37. have it be commented.
  38. /*
  39. NOTE: comments can be nested!
  40. */
  41. */
  42. // String literals are enclosed in double quotes and character literals in single quotes.
  43. // Special characters are escaped with a backslash \
  44. some_string := "This is a string";
  45. _ = 'A'; // unicode codepoint literal
  46. _ = '\n';
  47. _ = "C:\\Windows\\notepad.exe";
  48. // Raw string literals are enclosed with single back ticks
  49. _ = `C:\Windows\notepad.exe`;
  50. // The length of a string in bytes can be found using the built-in `len` procedure:
  51. _ = len("Foo");
  52. _ = len(some_string);
  53. // Numbers
  54. // Numerical literals are written similar to most other programming languages.
  55. // A useful feature in Odin is that underscores are allowed for better
  56. // readability: 1_000_000_000 (one billion). A number that contains a dot is a
  57. // floating point literal: 1.0e9 (one billion). If a number literal is suffixed
  58. // with i, is an imaginary number literal: 2i (2 multiply the square root of -1).
  59. // Binary literals are prefixed with 0b, octal literals with 0o, and hexadecimal
  60. // literals 0x. A leading zero does not produce an octal constant (unlike C).
  61. // In Odin, if a numeric constant can be represented by a type without
  62. // precision loss, it will automatically convert to that type.
  63. x: int = 1.0; // A float literal but it can be represented by an integer without precision loss
  64. // Constant literals are “untyped” which means that they can implicitly convert to a type.
  65. y: int; // `y` is typed of type `int`
  66. y = 1; // `1` is an untyped integer literal which can implicitly convert to `int`
  67. z: f64; // `z` is typed of type `f64` (64-bit floating point number)
  68. z = 1; // `1` is an untyped integer literal which can be implicitly converted to `f64`
  69. // No need for any suffixes or decimal places like in other languages
  70. // CONSTANTS JUST WORK!!!
  71. // Assignment statements
  72. h: int = 123; // declares a new variable `h` with type `int` and assigns a value to it
  73. h = 637; // assigns a new value to `h`
  74. // `=` is the assignment operator
  75. // You can assign multiple variables with it:
  76. a, b := 1, "hello"; // declares `a` and `b` and infers the types from the assignments
  77. b, a = "byte", 0;
  78. // Note: `:=` is two tokens, `:` and `=`. The following are equivalent,
  79. /*
  80. i: int = 123;
  81. i: = 123;
  82. i := 123;
  83. */
  84. // Constant declarations
  85. // Constants are entities (symbols) which have an assigned value.
  86. // The constant’s value cannot be changed.
  87. // The constant’s value must be able to be evaluated at compile time:
  88. X :: "what"; // constant `X` has the untyped string value "what"
  89. // Constants can be explicitly typed like a variable declaration:
  90. Y : int : 123;
  91. Z :: Y + 7; // constant computations are possible
  92. _ = my_integer_variable;
  93. _ = x;
  94. }
  95. }
  96. control_flow :: proc() {
  97. fmt.println("\n# control flow");
  98. { // Control flow
  99. // For loop
  100. // Odin has only one loop statement, the `for` loop
  101. // Basic for loop
  102. for i := 0; i < 10; i += 1 {
  103. fmt.println(i);
  104. }
  105. // NOTE: Unlike other languages like C, there are no parentheses `( )` surrounding the three components.
  106. // Braces `{ }` or a `do` are always required
  107. for i := 0; i < 10; i += 1 { }
  108. // for i := 0; i < 10; i += 1 do fmt.print();
  109. // The initial and post statements are optional
  110. i := 0;
  111. for ; i < 10; {
  112. i += 1;
  113. }
  114. // These semicolons can be dropped. This `for` loop is equivalent to C's `while` loop
  115. i = 0;
  116. for i < 10 {
  117. i += 1;
  118. }
  119. // If the condition is omitted, an infinite loop is produced:
  120. for {
  121. break;
  122. }
  123. // Range-based for loop
  124. // The basic for loop
  125. for j := 0; j < 10; j += 1 {
  126. fmt.println(j);
  127. }
  128. // can also be written
  129. for j in 0..<10 {
  130. fmt.println(j);
  131. }
  132. for j in 0..9 {
  133. fmt.println(j);
  134. }
  135. // Certain built-in types can be iterated over
  136. some_string := "Hello, 世界";
  137. for character in some_string { // Strings are assumed to be UTF-8
  138. fmt.println(character);
  139. }
  140. some_array := [3]int{1, 4, 9};
  141. for value in some_array {
  142. fmt.println(value);
  143. }
  144. some_slice := []int{1, 4, 9};
  145. for value in some_slice {
  146. fmt.println(value);
  147. }
  148. some_dynamic_array := [dynamic]int{1, 4, 9};
  149. defer delete(some_dynamic_array);
  150. for value in some_dynamic_array {
  151. fmt.println(value);
  152. }
  153. some_map := map[string]int{"A" = 1, "C" = 9, "B" = 4};
  154. defer delete(some_map);
  155. for key in some_map {
  156. fmt.println(key);
  157. }
  158. // Alternatively a second index value can be added
  159. for character, index in some_string {
  160. fmt.println(index, character);
  161. }
  162. for value, index in some_array {
  163. fmt.println(index, value);
  164. }
  165. for value, index in some_slice {
  166. fmt.println(index, value);
  167. }
  168. for value, index in some_dynamic_array {
  169. fmt.println(index, value);
  170. }
  171. for key, value in some_map {
  172. fmt.println(key, value);
  173. }
  174. // The iterated values are copies and cannot be written to.
  175. // The following idiom is useful for iterating over a container in a by-reference manner:
  176. for _, idx in some_slice {
  177. some_slice[idx] = (idx+1)*(idx+1);
  178. }
  179. // If statements
  180. x := 123;
  181. if x >= 0 {
  182. fmt.println("x is positive");
  183. }
  184. if y := -34; y < 0 {
  185. fmt.println("y is negative");
  186. }
  187. if y := 123; y < 0 {
  188. fmt.println("y is negative");
  189. } else if y == 0 {
  190. fmt.println("y is zero");
  191. } else {
  192. fmt.println("y is positive");
  193. }
  194. // Switch statement
  195. // A switch statement is another way to write a sequence of if-else statements.
  196. // In Odin, the default case is denoted as a case without any expression.
  197. switch arch := ODIN_ARCH; arch {
  198. case "386":
  199. fmt.println("32-bit");
  200. case "amd64":
  201. fmt.println("64-bit");
  202. case: // default
  203. fmt.println("Unsupported architecture");
  204. }
  205. // Odin’s `switch` is like one in C or C++, except that Odin only runs the selected case.
  206. // This means that a `break` statement is not needed at the end of each case.
  207. // Another important difference is that the case values need not be integers nor constants.
  208. // To achieve a C-like fall through into the next case block, the keyword `fallthrough` can be used.
  209. one_angry_dwarf :: proc() -> int {
  210. fmt.println("one_angry_dwarf was called");
  211. return 1;
  212. }
  213. switch j := 0; j {
  214. case 0:
  215. case one_angry_dwarf():
  216. }
  217. // A switch statement without a condition is the same as `switch true`.
  218. // This can be used to write a clean and long if-else chain and have the
  219. // ability to break if needed
  220. switch {
  221. case x < 0:
  222. fmt.println("x is negative");
  223. case x == 0:
  224. fmt.println("x is zero");
  225. case:
  226. fmt.println("x is positive");
  227. }
  228. // A `switch` statement can also use ranges like a range-based loop:
  229. switch c := 'j'; c {
  230. case 'A'..'Z', 'a'..'z', '0'..'9':
  231. fmt.println("c is alphanumeric");
  232. }
  233. switch x {
  234. case 0..<10:
  235. fmt.println("units");
  236. case 10..<13:
  237. fmt.println("pre-teens");
  238. case 13..<20:
  239. fmt.println("teens");
  240. case 20..<30:
  241. fmt.println("twenties");
  242. }
  243. }
  244. { // Defer statement
  245. // A defer statement defers the execution of a statement until the end of
  246. // the scope it is in.
  247. // The following will print 4 then 234:
  248. {
  249. x := 123;
  250. defer fmt.println(x);
  251. {
  252. defer x = 4;
  253. x = 2;
  254. }
  255. fmt.println(x);
  256. x = 234;
  257. }
  258. // You can defer an entire block too:
  259. {
  260. bar :: proc() {}
  261. defer {
  262. fmt.println("1");
  263. fmt.println("2");
  264. }
  265. cond := false;
  266. defer if cond {
  267. bar();
  268. }
  269. }
  270. // Defer statements are executed in the reverse order that they were declared:
  271. {
  272. defer fmt.println("1");
  273. defer fmt.println("2");
  274. defer fmt.println("3");
  275. }
  276. // Will print 3, 2, and then 1.
  277. if false {
  278. f, err := os.open("my_file.txt");
  279. if err != 0 {
  280. // handle error
  281. }
  282. defer os.close(f);
  283. // rest of code
  284. }
  285. }
  286. { // When statement
  287. /*
  288. The when statement is almost identical to the if statement but with some differences:
  289. * Each condition must be a constant expression as a when
  290. statement is evaluated at compile time.
  291. * The statements within a branch do not create a new scope
  292. * The compiler checks the semantics and code only for statements
  293. that belong to the first condition that is true
  294. * An initial statement is not allowed in a when statement
  295. * when statements are allowed at file scope
  296. */
  297. // Example
  298. when ODIN_ARCH == "386" {
  299. fmt.println("32 bit");
  300. } else when ODIN_ARCH == "amd64" {
  301. fmt.println("64 bit");
  302. } else {
  303. fmt.println("Unsupported architecture");
  304. }
  305. // The when statement is very useful for writing platform specific code.
  306. // This is akin to the #if construct in C’s preprocessor however, in Odin,
  307. // it is type checked.
  308. }
  309. { // Branch statements
  310. cond, cond1, cond2 := false, false, false;
  311. one_step :: proc() { fmt.println("one_step"); }
  312. beyond :: proc() { fmt.println("beyond"); }
  313. // Break statement
  314. for cond {
  315. switch {
  316. case:
  317. if cond {
  318. break; // break out of the `switch` statement
  319. }
  320. }
  321. break; // break out of the `for` statement
  322. }
  323. loop: for cond1 {
  324. for cond2 {
  325. break loop; // leaves both loops
  326. }
  327. }
  328. // Continue statement
  329. for cond {
  330. if cond2 {
  331. continue;
  332. }
  333. fmt.println("Hellope");
  334. }
  335. // Fallthrough statement
  336. // Odin’s switch is like one in C or C++, except that Odin only runs the selected
  337. // case. This means that a break statement is not needed at the end of each case.
  338. // Another important difference is that the case values need not be integers nor
  339. // constants.
  340. // fallthrough can be used to explicitly fall through into the next case block:
  341. switch i := 0; i {
  342. case 0:
  343. one_step();
  344. fallthrough;
  345. case 1:
  346. beyond();
  347. }
  348. }
  349. }
  350. named_proc_return_parameters :: proc() {
  351. fmt.println("\n# named proc return parameters");
  352. foo0 :: proc() -> int {
  353. return 123;
  354. }
  355. foo1 :: proc() -> (a: int) {
  356. a = 123;
  357. return;
  358. }
  359. foo2 :: proc() -> (a, b: int) {
  360. // Named return values act like variables within the scope
  361. a = 321;
  362. b = 567;
  363. return b, a;
  364. }
  365. fmt.println("foo0 =", foo0()); // 123
  366. fmt.println("foo1 =", foo1()); // 123
  367. fmt.println("foo2 =", foo2()); // 567 321
  368. }
  369. explicit_procedure_overloading :: proc() {
  370. fmt.println("\n# explicit procedure overloading");
  371. add_ints :: proc(a, b: int) -> int {
  372. x := a + b;
  373. fmt.println("add_ints", x);
  374. return x;
  375. }
  376. add_floats :: proc(a, b: f32) -> f32 {
  377. x := a + b;
  378. fmt.println("add_floats", x);
  379. return x;
  380. }
  381. add_numbers :: proc(a: int, b: f32, c: u8) -> int {
  382. x := int(a) + int(b) + int(c);
  383. fmt.println("add_numbers", x);
  384. return x;
  385. }
  386. add :: proc{add_ints, add_floats, add_numbers};
  387. add(int(1), int(2));
  388. add(f32(1), f32(2));
  389. add(int(1), f32(2), u8(3));
  390. add(1, 2); // untyped ints coerce to int tighter than f32
  391. add(1.0, 2.0); // untyped floats coerce to f32 tighter than int
  392. add(1, 2, 3); // three parameters
  393. // Ambiguous answers
  394. // add(1.0, 2);
  395. // add(1, 2.0);
  396. }
  397. struct_type :: proc() {
  398. fmt.println("\n# struct type");
  399. // A struct is a record type in Odin. It is a collection of fields.
  400. // Struct fields are accessed by using a dot:
  401. {
  402. Vector2 :: struct {
  403. x: f32,
  404. y: f32,
  405. };
  406. v := Vector2{1, 2};
  407. v.x = 4;
  408. fmt.println(v.x);
  409. // Struct fields can be accessed through a struct pointer:
  410. v = Vector2{1, 2};
  411. p := &v;
  412. p.x = 1335;
  413. fmt.println(v);
  414. // We could write p^.x, however, it is to nice abstract the ability
  415. // to not explicitly dereference the pointer. This is very useful when
  416. // refactoring code to use a pointer rather than a value, and vice versa.
  417. }
  418. {
  419. // A struct literal can be denoted by providing the struct’s type
  420. // followed by {}. A struct literal must either provide all the
  421. // arguments or none:
  422. Vector3 :: struct {
  423. x, y, z: f32,
  424. };
  425. v: Vector3;
  426. v = Vector3{}; // Zero value
  427. v = Vector3{1, 4, 9};
  428. // You can list just a subset of the fields if you specify the
  429. // field by name (the order of the named fields does not matter):
  430. v = Vector3{z=1, y=2};
  431. assert(v.x == 0);
  432. assert(v.y == 2);
  433. assert(v.z == 1);
  434. }
  435. {
  436. // Structs can tagged with different memory layout and alignment requirements:
  437. a :: struct #align 4 {}; // align to 4 bytes
  438. b :: struct #packed {}; // remove padding between fields
  439. c :: struct #raw_union {}; // all fields share the same offset (0). This is the same as C's union
  440. }
  441. }
  442. union_type :: proc() {
  443. fmt.println("\n# union type");
  444. {
  445. val: union{int, bool};
  446. val = 137;
  447. if i, ok := val.(int); ok {
  448. fmt.println(i);
  449. }
  450. val = true;
  451. fmt.println(val);
  452. val = nil;
  453. switch v in val {
  454. case int: fmt.println("int", v);
  455. case bool: fmt.println("bool", v);
  456. case: fmt.println("nil");
  457. }
  458. }
  459. {
  460. // There is a duality between `any` and `union`
  461. // An `any` has a pointer to the data and allows for any type (open)
  462. // A `union` has as binary blob to store the data and allows only certain types (closed)
  463. // The following code is with `any` but has the same syntax
  464. val: any;
  465. val = 137;
  466. if i, ok := val.(int); ok {
  467. fmt.println(i);
  468. }
  469. val = true;
  470. fmt.println(val);
  471. val = nil;
  472. switch v in val {
  473. case int: fmt.println("int", v);
  474. case bool: fmt.println("bool", v);
  475. case: fmt.println("nil");
  476. }
  477. }
  478. Vector3 :: distinct [3]f32;
  479. Quaternion :: distinct quaternion128;
  480. // More realistic examples
  481. {
  482. // NOTE(bill): For the above basic examples, you may not have any
  483. // particular use for it. However, my main use for them is not for these
  484. // simple cases. My main use is for hierarchical types. Many prefer
  485. // subtyping, embedding the base data into the derived types. Below is
  486. // an example of this for a basic game Entity.
  487. Entity :: struct {
  488. id: u64,
  489. name: string,
  490. position: Vector3,
  491. orientation: Quaternion,
  492. derived: any,
  493. };
  494. Frog :: struct {
  495. using entity: Entity,
  496. jump_height: f32,
  497. };
  498. Monster :: struct {
  499. using entity: Entity,
  500. is_robot: bool,
  501. is_zombie: bool,
  502. };
  503. // See `parametric_polymorphism` procedure for details
  504. new_entity :: proc($T: typeid) -> ^Entity {
  505. t := new(T);
  506. t.derived = t^;
  507. return t;
  508. }
  509. entity := new_entity(Monster);
  510. switch e in entity.derived {
  511. case Frog:
  512. fmt.println("Ribbit");
  513. case Monster:
  514. if e.is_robot { fmt.println("Robotic"); }
  515. if e.is_zombie { fmt.println("Grrrr!"); }
  516. fmt.println("I'm a monster");
  517. }
  518. }
  519. {
  520. // NOTE(bill): A union can be used to achieve something similar. Instead
  521. // of embedding the base data into the derived types, the derived data
  522. // in embedded into the base type. Below is the same example of the
  523. // basic game Entity but using an union.
  524. Entity :: struct {
  525. id: u64,
  526. name: string,
  527. position: Vector3,
  528. orientation: Quaternion,
  529. derived: union {Frog, Monster},
  530. };
  531. Frog :: struct {
  532. using entity: ^Entity,
  533. jump_height: f32,
  534. };
  535. Monster :: struct {
  536. using entity: ^Entity,
  537. is_robot: bool,
  538. is_zombie: bool,
  539. };
  540. // See `parametric_polymorphism` procedure for details
  541. new_entity :: proc($T: typeid) -> ^Entity {
  542. t := new(Entity);
  543. t.derived = T{entity = t};
  544. return t;
  545. }
  546. entity := new_entity(Monster);
  547. switch e in entity.derived {
  548. case Frog:
  549. fmt.println("Ribbit");
  550. case Monster:
  551. if e.is_robot { fmt.println("Robotic"); }
  552. if e.is_zombie { fmt.println("Grrrr!"); }
  553. }
  554. // NOTE(bill): As you can see, the usage code has not changed, only its
  555. // memory layout. Both approaches have their own advantages but they can
  556. // be used together to achieve different results. The subtyping approach
  557. // can allow for a greater control of the memory layout and memory
  558. // allocation, e.g. storing the derivatives together. However, this is
  559. // also its disadvantage. You must either preallocate arrays for each
  560. // derivative separation (which can be easily missed) or preallocate a
  561. // bunch of "raw" memory; determining the maximum size of the derived
  562. // types would require the aid of metaprogramming. Unions solve this
  563. // particular problem as the data is stored with the base data.
  564. // Therefore, it is possible to preallocate, e.g. [100]Entity.
  565. // It should be noted that the union approach can have the same memory
  566. // layout as the any and with the same type restrictions by using a
  567. // pointer type for the derivatives.
  568. /*
  569. Entity :: struct {
  570. ...
  571. derived: union{^Frog, ^Monster},
  572. }
  573. Frog :: struct {
  574. using entity: Entity,
  575. ...
  576. }
  577. Monster :: struct {
  578. using entity: Entity,
  579. ...
  580. }
  581. new_entity :: proc(T: type) -> ^Entity {
  582. t := new(T);
  583. t.derived = t;
  584. return t;
  585. }
  586. */
  587. }
  588. }
  589. using_statement :: proc() {
  590. fmt.println("\n# using statement");
  591. // using can used to bring entities declared in a scope/namespace
  592. // into the current scope. This can be applied to import names, struct
  593. // fields, procedure fields, and struct values.
  594. Vector3 :: struct{x, y, z: f32};
  595. {
  596. Entity :: struct {
  597. position: Vector3,
  598. orientation: quaternion128,
  599. };
  600. // It can used like this:
  601. foo0 :: proc(entity: ^Entity) {
  602. fmt.println(entity.position.x, entity.position.y, entity.position.z);
  603. }
  604. // The entity members can be brought into the procedure scope by using it:
  605. foo1 :: proc(entity: ^Entity) {
  606. using entity;
  607. fmt.println(position.x, position.y, position.z);
  608. }
  609. // The using can be applied to the parameter directly:
  610. foo2 :: proc(using entity: ^Entity) {
  611. fmt.println(position.x, position.y, position.z);
  612. }
  613. // It can also be applied to sub-fields:
  614. foo3 :: proc(entity: ^Entity) {
  615. using entity.position;
  616. fmt.println(x, y, z);
  617. }
  618. }
  619. {
  620. // We can also apply the using statement to the struct fields directly,
  621. // making all the fields of position appear as if they on Entity itself:
  622. Entity :: struct {
  623. using position: Vector3,
  624. orientation: quaternion128,
  625. };
  626. foo :: proc(entity: ^Entity) {
  627. fmt.println(entity.x, entity.y, entity.z);
  628. }
  629. // Subtype polymorphism
  630. // It is possible to get subtype polymorphism, similar to inheritance-like
  631. // functionality in C++, but without the requirement of vtables or unknown
  632. // struct layout:
  633. Colour :: struct {r, g, b, a: u8};
  634. Frog :: struct {
  635. ribbit_volume: f32,
  636. using entity: Entity,
  637. colour: Colour,
  638. };
  639. frog: Frog;
  640. // Both work
  641. foo(&frog.entity);
  642. foo(&frog);
  643. frog.x = 123;
  644. // Note: using can be applied to arbitrarily many things, which allows
  645. // the ability to have multiple subtype polymorphism (but also its issues).
  646. // Note: using’d fields can still be referred by name.
  647. }
  648. { // using on an enum declaration
  649. using Foo :: enum {A, B, C};
  650. f0 := A;
  651. f1 := B;
  652. f2 := C;
  653. fmt.println(f0, f1, f2);
  654. fmt.println(len(Foo));
  655. }
  656. }
  657. implicit_context_system :: proc() {
  658. fmt.println("\n# implicit context system");
  659. // In each scope, there is an implicit value named context. This
  660. // context variable is local to each scope and is implicitly passed
  661. // by pointer to any procedure call in that scope (if the procedure
  662. // has the Odin calling convention).
  663. // The main purpose of the implicit context system is for the ability
  664. // to intercept third-party code and libraries and modify their
  665. // functionality. One such case is modifying how a library allocates
  666. // something or logs something. In C, this was usually achieved with
  667. // the library defining macros which could be overridden so that the
  668. // user could define what he wanted. However, not many libraries
  669. // supported this in many languages by default which meant intercepting
  670. // third-party code to see what it does and to change how it does it is
  671. // not possible.
  672. c := context; // copy the current scope's context
  673. context.user_index = 456;
  674. {
  675. context.allocator = my_custom_allocator();
  676. context.user_index = 123;
  677. what_a_fool_believes(); // the `context` for this scope is implicitly passed to `what_a_fool_believes`
  678. }
  679. // `context` value is local to the scope it is in
  680. assert(context.user_index == 456);
  681. what_a_fool_believes :: proc() {
  682. c := context; // this `context` is the same as the parent procedure that it was called from
  683. // From this example, context.user_index == 123
  684. // An context.allocator is assigned to the return value of `my_custom_allocator()`
  685. assert(context.user_index == 123);
  686. // The memory management procedure use the `context.allocator` by
  687. // default unless explicitly specified otherwise
  688. china_grove := new(int);
  689. free(china_grove);
  690. _ = c;
  691. }
  692. my_custom_allocator :: mem.nil_allocator;
  693. _ = c;
  694. // By default, the context value has default values for its parameters which is
  695. // decided in the package runtime. What the defaults are are compiler specific.
  696. // To see what the implicit context value contains, please see the following
  697. // definition in package runtime.
  698. }
  699. parametric_polymorphism :: proc() {
  700. fmt.println("\n# parametric polymorphism");
  701. print_value :: proc(value: $T) {
  702. fmt.printf("print_value: %T %v\n", value, value);
  703. }
  704. v1: int = 1;
  705. v2: f32 = 2.1;
  706. v3: f64 = 3.14;
  707. v4: string = "message";
  708. print_value(v1);
  709. print_value(v2);
  710. print_value(v3);
  711. print_value(v4);
  712. fmt.println();
  713. add :: proc(p, q: $T) -> T {
  714. x: T = p + q;
  715. return x;
  716. }
  717. a := add(3, 4);
  718. fmt.printf("a: %T = %v\n", a, a);
  719. b := add(3.2, 4.3);
  720. fmt.printf("b: %T = %v\n", b, b);
  721. // This is how `new` is implemented
  722. alloc_type :: proc($T: typeid) -> ^T {
  723. t := cast(^T)alloc(size_of(T), align_of(T));
  724. t^ = T{}; // Use default initialization value
  725. return t;
  726. }
  727. copy_slice :: proc(dst, src: []$T) -> int {
  728. n := min(len(dst), len(src));
  729. if n > 0 {
  730. mem.copy(&dst[0], &src[0], n*size_of(T));
  731. }
  732. return n;
  733. }
  734. double_params :: proc(a: $A, b: $B) -> A {
  735. return a + A(b);
  736. }
  737. fmt.println(double_params(12, 1.345));
  738. { // Polymorphic Types and Type Specialization
  739. Table_Slot :: struct(Key, Value: typeid) {
  740. occupied: bool,
  741. hash: u32,
  742. key: Key,
  743. value: Value,
  744. };
  745. TABLE_SIZE_MIN :: 32;
  746. Table :: struct(Key, Value: typeid) {
  747. count: int,
  748. allocator: mem.Allocator,
  749. slots: []Table_Slot(Key, Value),
  750. };
  751. // Only allow types that are specializations of a (polymorphic) slice
  752. make_slice :: proc($T: typeid/[]$E, len: int) -> T {
  753. return make(T, len);
  754. }
  755. // Only allow types that are specializations of `Table`
  756. allocate :: proc(table: ^$T/Table, capacity: int) {
  757. c := context;
  758. if table.allocator.procedure != nil {
  759. c.allocator = table.allocator;
  760. }
  761. context = c;
  762. table.slots = make_slice(type_of(table.slots), max(capacity, TABLE_SIZE_MIN));
  763. }
  764. expand :: proc(table: ^$T/Table) {
  765. c := context;
  766. if table.allocator.procedure != nil {
  767. c.allocator = table.allocator;
  768. }
  769. context = c;
  770. old_slots := table.slots;
  771. defer delete(old_slots);
  772. cap := max(2*len(table.slots), TABLE_SIZE_MIN);
  773. allocate(table, cap);
  774. for s in old_slots {
  775. if s.occupied {
  776. put(table, s.key, s.value);
  777. }
  778. }
  779. }
  780. // Polymorphic determination of a polymorphic struct
  781. // put :: proc(table: ^$T/Table, key: T.Key, value: T.Value) {
  782. put :: proc(table: ^Table($Key, $Value), key: Key, value: Value) {
  783. hash := get_hash(key); // Ad-hoc method which would fail in a different scope
  784. index := find_index(table, key, hash);
  785. if index < 0 {
  786. if f64(table.count) >= 0.75*f64(len(table.slots)) {
  787. expand(table);
  788. }
  789. assert(table.count <= len(table.slots));
  790. index = int(hash % u32(len(table.slots)));
  791. for table.slots[index].occupied {
  792. if index += 1; index >= len(table.slots) {
  793. index = 0;
  794. }
  795. }
  796. table.count += 1;
  797. }
  798. slot := &table.slots[index];
  799. slot.occupied = true;
  800. slot.hash = hash;
  801. slot.key = key;
  802. slot.value = value;
  803. }
  804. // find :: proc(table: ^$T/Table, key: T.Key) -> (T.Value, bool) {
  805. find :: proc(table: ^Table($Key, $Value), key: Key) -> (Value, bool) {
  806. hash := get_hash(key);
  807. index := find_index(table, key, hash);
  808. if index < 0 {
  809. return Value{}, false;
  810. }
  811. return table.slots[index].value, true;
  812. }
  813. find_index :: proc(table: ^Table($Key, $Value), key: Key, hash: u32) -> int {
  814. if len(table.slots) <= 0 {
  815. return -1;
  816. }
  817. index := int(hash % u32(len(table.slots)));
  818. for table.slots[index].occupied {
  819. if table.slots[index].hash == hash {
  820. if table.slots[index].key == key {
  821. return index;
  822. }
  823. }
  824. if index += 1; index >= len(table.slots) {
  825. index = 0;
  826. }
  827. }
  828. return -1;
  829. }
  830. get_hash :: proc(s: string) -> u32 { // fnv32a
  831. h: u32 = 0x811c9dc5;
  832. for i in 0..<len(s) {
  833. h = (h ~ u32(s[i])) * 0x01000193;
  834. }
  835. return h;
  836. }
  837. table: Table(string, int);
  838. for i in 0..36 { put(&table, "Hellope", i); }
  839. for i in 0..42 { put(&table, "World!", i); }
  840. found, _ := find(&table, "Hellope");
  841. fmt.printf("`found` is %v\n", found);
  842. found, _ = find(&table, "World!");
  843. fmt.printf("`found` is %v\n", found);
  844. // I would not personally design a hash table like this in production
  845. // but this is a nice basic example
  846. // A better approach would either use a `u64` or equivalent for the key
  847. // and let the user specify the hashing function or make the user store
  848. // the hashing procedure with the table
  849. }
  850. { // Parametric polymorphic union
  851. Error :: enum {
  852. Foo0,
  853. Foo1,
  854. Foo2,
  855. Foo3,
  856. };
  857. Para_Union :: union(T: typeid) {T, Error};
  858. r: Para_Union(int);
  859. fmt.println(typeid_of(type_of(r)));
  860. fmt.println(r);
  861. r = 123;
  862. fmt.println(r);
  863. r = Error.Foo0; // r = .Foo0; is allow too, see implicit selector expressions below
  864. fmt.println(r);
  865. }
  866. { // Polymorphic names
  867. foo :: proc($N: $I, $T: typeid) -> (res: [N]T) {
  868. // `N` is the constant value passed
  869. // `I` is the type of N
  870. // `T` is the type passed
  871. fmt.printf("Generating an array of type %v from the value %v of type %v\n",
  872. typeid_of(type_of(res)), N, typeid_of(I));
  873. for i in 0..<N {
  874. res[i] = T(i*i);
  875. }
  876. return;
  877. }
  878. T :: int;
  879. array := foo(4, T);
  880. for v, i in array {
  881. assert(v == T(i*i));
  882. }
  883. // Matrix multiplication
  884. mul :: proc(a: [$M][$N]$T, b: [N][$P]T) -> (c: [M][P]T) {
  885. for i in 0..<M {
  886. for j in 0..<P {
  887. for k in 0..<N {
  888. c[i][j] += a[i][k] * b[k][j];
  889. }
  890. }
  891. }
  892. return;
  893. }
  894. x := [2][3]f32{
  895. {1, 2, 3},
  896. {3, 2, 1},
  897. };
  898. y := [3][2]f32{
  899. {0, 8},
  900. {6, 2},
  901. {8, 4},
  902. };
  903. z := mul(x, y);
  904. assert(z == {{36, 24}, {20, 32}});
  905. }
  906. }
  907. prefix_table := [?]string{
  908. "White",
  909. "Red",
  910. "Green",
  911. "Blue",
  912. "Octarine",
  913. "Black",
  914. };
  915. threading_example :: proc() {
  916. if ODIN_OS == "darwin" {
  917. // TODO: Fix threads on darwin/macOS
  918. return;
  919. }
  920. fmt.println("\n# threading_example");
  921. { // Basic Threads
  922. fmt.println("\n## Basic Threads");
  923. worker_proc :: proc(t: ^thread.Thread) {
  924. for iteration in 1..5 {
  925. fmt.printf("Thread %d is on iteration %d\n", t.user_index, iteration);
  926. fmt.printf("`%s`: iteration %d\n", prefix_table[t.user_index], iteration);
  927. time.sleep(1 * time.Millisecond);
  928. }
  929. }
  930. threads := make([dynamic]^thread.Thread, 0, len(prefix_table));
  931. defer delete(threads);
  932. for in prefix_table {
  933. if t := thread.create(worker_proc); t != nil {
  934. t.init_context = context;
  935. t.user_index = len(threads);
  936. append(&threads, t);
  937. thread.start(t);
  938. }
  939. }
  940. for len(threads) > 0 {
  941. for i := 0; i < len(threads); /**/ {
  942. if t := threads[i]; thread.is_done(t) {
  943. fmt.printf("Thread %d is done\n", t.user_index);
  944. thread.destroy(t);
  945. ordered_remove(&threads, i);
  946. } else {
  947. i += 1;
  948. }
  949. }
  950. }
  951. }
  952. { // Thread Pool
  953. fmt.println("\n## Thread Pool");
  954. task_proc :: proc(t: ^thread.Task) {
  955. index := t.user_index % len(prefix_table);
  956. for iteration in 1..5 {
  957. fmt.printf("Worker Task %d is on iteration %d\n", t.user_index, iteration);
  958. fmt.printf("`%s`: iteration %d\n", prefix_table[index], iteration);
  959. time.sleep(1 * time.Millisecond);
  960. }
  961. }
  962. pool: thread.Pool;
  963. thread.pool_init(pool=&pool, thread_count=3);
  964. defer thread.pool_destroy(&pool);
  965. for i in 0..<30 {
  966. thread.pool_add_task(pool=&pool, procedure=task_proc, data=nil, user_index=i);
  967. }
  968. thread.pool_start(&pool);
  969. thread.pool_wait_and_process(&pool);
  970. }
  971. }
  972. array_programming :: proc() {
  973. fmt.println("\n# array programming");
  974. {
  975. a := [3]f32{1, 2, 3};
  976. b := [3]f32{5, 6, 7};
  977. c := a * b;
  978. d := a + b;
  979. e := 1 + (c - d) / 2;
  980. fmt.printf("%.1f\n", e); // [0.5, 3.0, 6.5]
  981. }
  982. {
  983. a := [3]f32{1, 2, 3};
  984. b := swizzle(a, 2, 1, 0);
  985. assert(b == [3]f32{3, 2, 1});
  986. c := swizzle(a, 0, 0);
  987. assert(c == [2]f32{1, 1});
  988. assert(c == 1);
  989. }
  990. {
  991. Vector3 :: distinct [3]f32;
  992. a := Vector3{1, 2, 3};
  993. b := Vector3{5, 6, 7};
  994. c := (a * b)/2 + 1;
  995. d := c.x + c.y + c.z;
  996. fmt.printf("%.1f\n", d); // 22.0
  997. cross :: proc(a, b: Vector3) -> Vector3 {
  998. i := swizzle(a, 1, 2, 0) * swizzle(b, 2, 0, 1);
  999. j := swizzle(a, 2, 0, 1) * swizzle(b, 1, 2, 0);
  1000. return i - j;
  1001. }
  1002. blah :: proc(a: Vector3) -> f32 {
  1003. return a.x + a.y + a.z;
  1004. }
  1005. x := cross(a, b);
  1006. fmt.println(x);
  1007. fmt.println(blah(x));
  1008. }
  1009. }
  1010. map_type :: proc() {
  1011. fmt.println("\n# map type");
  1012. m := make(map[string]int);
  1013. defer delete(m);
  1014. m["Bob"] = 2;
  1015. m["Ted"] = 5;
  1016. fmt.println(m["Bob"]);
  1017. delete_key(&m, "Ted");
  1018. // If an element of a key does not exist, the zero value of the
  1019. // element will be returned. To check to see if an element exists
  1020. // can be done in two ways:
  1021. elem, ok := m["Bob"];
  1022. exists := "Bob" in m;
  1023. _, _ = elem, ok;
  1024. _ = exists;
  1025. }
  1026. implicit_selector_expression :: proc() {
  1027. fmt.println("\n# implicit selector expression");
  1028. Foo :: enum {A, B, C};
  1029. f: Foo;
  1030. f = Foo.A;
  1031. f = .A;
  1032. BAR :: bit_set[Foo]{.B, .C};
  1033. switch f {
  1034. case .A:
  1035. fmt.println("HITHER");
  1036. case .B:
  1037. fmt.println("NEVER");
  1038. case .C:
  1039. fmt.println("FOREVER");
  1040. }
  1041. my_map := make(map[Foo]int);
  1042. defer delete(my_map);
  1043. my_map[.A] = 123;
  1044. my_map[Foo.B] = 345;
  1045. fmt.println(my_map[.A] + my_map[Foo.B] + my_map[.C]);
  1046. }
  1047. partial_switch :: proc() {
  1048. fmt.println("\n# partial_switch");
  1049. { // enum
  1050. Foo :: enum {
  1051. A,
  1052. B,
  1053. C,
  1054. D,
  1055. };
  1056. f := Foo.A;
  1057. switch f {
  1058. case .A: fmt.println("A");
  1059. case .B: fmt.println("B");
  1060. case .C: fmt.println("C");
  1061. case .D: fmt.println("D");
  1062. case: fmt.println("?");
  1063. }
  1064. #partial switch f {
  1065. case .A: fmt.println("A");
  1066. case .D: fmt.println("D");
  1067. }
  1068. }
  1069. { // union
  1070. Foo :: union {int, bool};
  1071. f: Foo = 123;
  1072. switch in f {
  1073. case int: fmt.println("int");
  1074. case bool: fmt.println("bool");
  1075. case:
  1076. }
  1077. #partial switch in f {
  1078. case bool: fmt.println("bool");
  1079. }
  1080. }
  1081. }
  1082. cstring_example :: proc() {
  1083. fmt.println("\n# cstring_example");
  1084. W :: "Hellope";
  1085. X :: cstring(W);
  1086. Y :: string(X);
  1087. w := W;
  1088. _ = w;
  1089. x: cstring = X;
  1090. y: string = Y;
  1091. z := string(x);
  1092. fmt.println(x, y, z);
  1093. fmt.println(len(x), len(y), len(z));
  1094. fmt.println(len(W), len(X), len(Y));
  1095. // IMPORTANT NOTE for cstring variables
  1096. // len(cstring) is O(N)
  1097. // cast(string)cstring is O(N)
  1098. }
  1099. bit_set_type :: proc() {
  1100. fmt.println("\n# bit_set type");
  1101. {
  1102. using Day :: enum {
  1103. Sunday,
  1104. Monday,
  1105. Tuesday,
  1106. Wednesday,
  1107. Thursday,
  1108. Friday,
  1109. Saturday,
  1110. };
  1111. Days :: distinct bit_set[Day];
  1112. WEEKEND :: Days{Sunday, Saturday};
  1113. d: Days;
  1114. d = {Sunday, Monday};
  1115. e := d + WEEKEND;
  1116. e += {Monday};
  1117. fmt.println(d, e);
  1118. ok := Saturday in e; // `in` is only allowed for `map` and `bit_set` types
  1119. fmt.println(ok);
  1120. if Saturday in e {
  1121. fmt.println("Saturday in", e);
  1122. }
  1123. X :: Saturday in WEEKEND; // Constant evaluation
  1124. fmt.println(X);
  1125. fmt.println("Cardinality:", card(e));
  1126. }
  1127. {
  1128. x: bit_set['A'..'Z'];
  1129. #assert(size_of(x) == size_of(u32));
  1130. y: bit_set[0..8; u16];
  1131. fmt.println(typeid_of(type_of(x))); // bit_set[A..Z]
  1132. fmt.println(typeid_of(type_of(y))); // bit_set[0..8; u16]
  1133. x += {'F'};
  1134. assert('F' in x);
  1135. x -= {'F'};
  1136. assert('F' not_in x);
  1137. y += {1, 4, 2};
  1138. assert(2 in y);
  1139. }
  1140. {
  1141. Letters :: bit_set['A'..'Z'];
  1142. a := Letters{'A', 'B'};
  1143. b := Letters{'A', 'B', 'C', 'D', 'F'};
  1144. c := Letters{'A', 'B'};
  1145. assert(a <= b); // 'a' is a subset of 'b'
  1146. assert(b >= a); // 'b' is a superset of 'a'
  1147. assert(a < b); // 'a' is a strict subset of 'b'
  1148. assert(b > a); // 'b' is a strict superset of 'a'
  1149. assert(!(a < c)); // 'a' is a not strict subset of 'c'
  1150. assert(!(c > a)); // 'c' is a not strict superset of 'a'
  1151. }
  1152. }
  1153. deferred_procedure_associations :: proc() {
  1154. fmt.println("\n# deferred procedure associations");
  1155. @(deferred_out=closure)
  1156. open :: proc(s: string) -> bool {
  1157. fmt.println(s);
  1158. return true;
  1159. }
  1160. closure :: proc(ok: bool) {
  1161. fmt.println("Goodbye?", ok);
  1162. }
  1163. if open("Welcome") {
  1164. fmt.println("Something in the middle, mate.");
  1165. }
  1166. }
  1167. reflection :: proc() {
  1168. fmt.println("\n# reflection");
  1169. Foo :: struct {
  1170. x: int `tag1`,
  1171. y: string `json:"y_field"`,
  1172. z: bool, // no tag
  1173. };
  1174. id := typeid_of(Foo);
  1175. names := reflect.struct_field_names(id);
  1176. types := reflect.struct_field_types(id);
  1177. tags := reflect.struct_field_tags(id);
  1178. assert(len(names) == len(types) && len(names) == len(tags));
  1179. fmt.println("Foo :: struct {");
  1180. for tag, i in tags {
  1181. name, type := names[i], types[i];
  1182. if tag != "" {
  1183. fmt.printf("\t%s: %T `%s`,\n", name, type, tag);
  1184. } else {
  1185. fmt.printf("\t%s: %T,\n", name, type);
  1186. }
  1187. }
  1188. fmt.println("}");
  1189. for tag, i in tags {
  1190. if val, ok := reflect.struct_tag_lookup(tag, "json"); ok {
  1191. fmt.printf("json: %s -> %s\n", names[i], val);
  1192. }
  1193. }
  1194. }
  1195. quaternions :: proc() {
  1196. // Not just an April Fool's Joke any more, but a fully working thing!
  1197. fmt.println("\n# quaternions");
  1198. { // Quaternion operations
  1199. q := 1 + 2i + 3j + 4k;
  1200. r := quaternion(5, 6, 7, 8);
  1201. t := q * r;
  1202. fmt.printf("(%v) * (%v) = %v\n", q, r, t);
  1203. v := q / r;
  1204. fmt.printf("(%v) / (%v) = %v\n", q, r, v);
  1205. u := q + r;
  1206. fmt.printf("(%v) + (%v) = %v\n", q, r, u);
  1207. s := q - r;
  1208. fmt.printf("(%v) - (%v) = %v\n", q, r, s);
  1209. }
  1210. { // The quaternion types
  1211. q128: quaternion128; // 4xf32
  1212. q256: quaternion256; // 4xf64
  1213. q128 = quaternion(1, 0, 0, 0);
  1214. q256 = 1; // quaternion(1, 0, 0, 0);
  1215. }
  1216. { // Built-in procedures
  1217. q := 1 + 2i + 3j + 4k;
  1218. fmt.println("q =", q);
  1219. fmt.println("real(q) =", real(q));
  1220. fmt.println("imag(q) =", imag(q));
  1221. fmt.println("jmag(q) =", jmag(q));
  1222. fmt.println("kmag(q) =", kmag(q));
  1223. fmt.println("conj(q) =", conj(q));
  1224. fmt.println("abs(q) =", abs(q));
  1225. }
  1226. { // Conversion of a complex type to a quaternion type
  1227. c := 1 + 2i;
  1228. q := quaternion256(c);
  1229. fmt.println(c);
  1230. fmt.println(q);
  1231. }
  1232. { // Memory layout of Quaternions
  1233. q := 1 + 2i + 3j + 4k;
  1234. a := transmute([4]f64)q;
  1235. fmt.println("Quaternion memory layout: xyzw/(ijkr)");
  1236. fmt.println(q); // 1.000+2.000i+3.000j+4.000k
  1237. fmt.println(a); // [2.000, 3.000, 4.000, 1.000]
  1238. }
  1239. }
  1240. unroll_for_statement :: proc() {
  1241. fmt.println("\n#'#unroll for' statements");
  1242. // '#unroll for' works the same as if the 'inline' prefix did not
  1243. // exist but these ranged loops are explicitly unrolled which can
  1244. // be very very useful for certain optimizations
  1245. fmt.println("Ranges");
  1246. #unroll for x, i in 1..<4 {
  1247. fmt.println(x, i);
  1248. }
  1249. fmt.println("Strings");
  1250. #unroll for r, i in "Hello, 世界" {
  1251. fmt.println(r, i);
  1252. }
  1253. fmt.println("Arrays");
  1254. #unroll for elem, idx in ([4]int{1, 4, 9, 16}) {
  1255. fmt.println(elem, idx);
  1256. }
  1257. Foo_Enum :: enum {
  1258. A = 1,
  1259. B,
  1260. C = 6,
  1261. D,
  1262. };
  1263. fmt.println("Enum types");
  1264. #unroll for elem, idx in Foo_Enum {
  1265. fmt.println(elem, idx);
  1266. }
  1267. }
  1268. where_clauses :: proc() {
  1269. fmt.println("\n#procedure 'where' clauses");
  1270. { // Sanity checks
  1271. simple_sanity_check :: proc(x: [2]int)
  1272. where len(x) > 1,
  1273. type_of(x) == [2]int {
  1274. fmt.println(x);
  1275. }
  1276. }
  1277. { // Parametric polymorphism checks
  1278. cross_2d :: proc(a, b: $T/[2]$E) -> E
  1279. where intrinsics.type_is_numeric(E) {
  1280. return a.x*b.y - a.y*b.x;
  1281. }
  1282. cross_3d :: proc(a, b: $T/[3]$E) -> T
  1283. where intrinsics.type_is_numeric(E) {
  1284. x := a.y*b.z - a.z*b.y;
  1285. y := a.z*b.x - a.x*b.z;
  1286. z := a.x*b.y - a.y*b.z;
  1287. return T{x, y, z};
  1288. }
  1289. a := [2]int{1, 2};
  1290. b := [2]int{5, -3};
  1291. fmt.println(cross_2d(a, b));
  1292. x := [3]f32{1, 4, 9};
  1293. y := [3]f32{-5, 0, 3};
  1294. fmt.println(cross_3d(x, y));
  1295. // Failure case
  1296. // i := [2]bool{true, false};
  1297. // j := [2]bool{false, true};
  1298. // fmt.println(cross_2d(i, j));
  1299. }
  1300. { // Procedure groups usage
  1301. foo :: proc(x: [$N]int) -> bool
  1302. where N > 2 {
  1303. fmt.println(#procedure, "was called with the parameter", x);
  1304. return true;
  1305. }
  1306. bar :: proc(x: [$N]int) -> bool
  1307. where 0 < N,
  1308. N <= 2 {
  1309. fmt.println(#procedure, "was called with the parameter", x);
  1310. return false;
  1311. }
  1312. baz :: proc{foo, bar};
  1313. x := [3]int{1, 2, 3};
  1314. y := [2]int{4, 9};
  1315. ok_x := baz(x);
  1316. ok_y := baz(y);
  1317. assert(ok_x == true);
  1318. assert(ok_y == false);
  1319. }
  1320. { // Record types
  1321. Foo :: struct(T: typeid, N: int)
  1322. where intrinsics.type_is_integer(T),
  1323. N > 2 {
  1324. x: [N]T,
  1325. y: [N-2]T,
  1326. };
  1327. T :: i32;
  1328. N :: 5;
  1329. f: Foo(T, N);
  1330. #assert(size_of(f) == (N+N-2)*size_of(T));
  1331. }
  1332. }
  1333. when ODIN_OS == "windows" {
  1334. foreign import kernel32 "system:kernel32.lib"
  1335. }
  1336. foreign_system :: proc() {
  1337. fmt.println("\n#foreign system");
  1338. when ODIN_OS == "windows" {
  1339. // It is sometimes necessarily to interface with foreign code,
  1340. // such as a C library. In Odin, this is achieved through the
  1341. // foreign system. You can “import” a library into the code
  1342. // using the same semantics as a normal import declaration.
  1343. // This foreign import declaration will create a
  1344. // “foreign import name” which can then be used to associate
  1345. // entities within a foreign block.
  1346. foreign kernel32 {
  1347. ExitProcess :: proc "stdcall" (exit_code: u32) ---
  1348. }
  1349. // Foreign procedure declarations have the cdecl/c calling
  1350. // convention by default unless specified otherwise. Due to
  1351. // foreign procedures do not have a body declared within this
  1352. // code, you need append the --- symbol to the end to distinguish
  1353. // it as a procedure literal without a body and not a procedure type.
  1354. // The attributes system can be used to change specific properties
  1355. // of entities declared within a block:
  1356. @(default_calling_convention = "std")
  1357. foreign kernel32 {
  1358. @(link_name="GetLastError") get_last_error :: proc() -> i32 ---
  1359. }
  1360. // Example using the link_prefix attribute
  1361. @(default_calling_convention = "std")
  1362. @(link_prefix = "Get")
  1363. foreign kernel32 {
  1364. LastError :: proc() -> i32 ---
  1365. }
  1366. }
  1367. }
  1368. ranged_fields_for_array_compound_literals :: proc() {
  1369. fmt.println("\n#ranged fields for array compound literals");
  1370. { // Normal Array Literal
  1371. foo := [?]int{1, 4, 9, 16};
  1372. fmt.println(foo);
  1373. }
  1374. { // Indexed
  1375. foo := [?]int{
  1376. 3 = 16,
  1377. 1 = 4,
  1378. 2 = 9,
  1379. 0 = 1,
  1380. };
  1381. fmt.println(foo);
  1382. }
  1383. { // Ranges
  1384. i := 2;
  1385. foo := [?]int {
  1386. 0 = 123,
  1387. 5..9 = 54,
  1388. 10..<16 = i*3 + (i-1)*2,
  1389. };
  1390. #assert(len(foo) == 16);
  1391. fmt.println(foo); // [123, 0, 0, 0, 0, 54, 54, 54, 54, 54, 8, 8, 8, 8, 8]
  1392. }
  1393. { // Slice and Dynamic Array support
  1394. i := 2;
  1395. foo_slice := []int {
  1396. 0 = 123,
  1397. 5..9 = 54,
  1398. 10..<16 = i*3 + (i-1)*2,
  1399. };
  1400. assert(len(foo_slice) == 16);
  1401. fmt.println(foo_slice); // [123, 0, 0, 0, 0, 54, 54, 54, 54, 54, 8, 8, 8, 8, 8]
  1402. foo_dynamic_array := [dynamic]int {
  1403. 0 = 123,
  1404. 5..9 = 54,
  1405. 10..<16 = i*3 + (i-1)*2,
  1406. };
  1407. assert(len(foo_dynamic_array) == 16);
  1408. fmt.println(foo_dynamic_array); // [123, 0, 0, 0, 0, 54, 54, 54, 54, 54, 8, 8, 8, 8, 8]
  1409. }
  1410. }
  1411. deprecated_attribute :: proc() {
  1412. @(deprecated="Use foo_v2 instead")
  1413. foo_v1 :: proc(x: int) {
  1414. fmt.println("foo_v1");
  1415. }
  1416. foo_v2 :: proc(x: int) {
  1417. fmt.println("foo_v2");
  1418. }
  1419. // NOTE: Uncomment to see the warning messages
  1420. // foo_v1(1);
  1421. }
  1422. range_statements_with_multiple_return_values :: proc() {
  1423. // IMPORTANT NOTE(bill, 2019-11-02): This feature is subject to be changed/removed
  1424. fmt.println("\n#range statements with multiple return values");
  1425. My_Iterator :: struct {
  1426. index: int,
  1427. data: []i32,
  1428. };
  1429. make_my_iterator :: proc(data: []i32) -> My_Iterator {
  1430. return My_Iterator{data = data};
  1431. }
  1432. my_iterator :: proc(it: ^My_Iterator) -> (val: i32, idx: int, cond: bool) {
  1433. if cond = it.index < len(it.data); cond {
  1434. val = it.data[it.index];
  1435. idx = it.index;
  1436. it.index += 1;
  1437. }
  1438. return;
  1439. }
  1440. data := make([]i32, 6);
  1441. for _, i in data {
  1442. data[i] = i32(i*i);
  1443. }
  1444. {
  1445. it := make_my_iterator(data);
  1446. for val in my_iterator(&it) {
  1447. fmt.println(val);
  1448. }
  1449. }
  1450. {
  1451. it := make_my_iterator(data);
  1452. for val, idx in my_iterator(&it) {
  1453. fmt.println(val, idx);
  1454. }
  1455. }
  1456. {
  1457. it := make_my_iterator(data);
  1458. for {
  1459. val, _, cond := my_iterator(&it);
  1460. if !cond {
  1461. break;
  1462. }
  1463. fmt.println(val);
  1464. }
  1465. }
  1466. }
  1467. soa_struct_layout :: proc() {
  1468. fmt.println("\n#SOA Struct Layout");
  1469. {
  1470. Vector3 :: struct {x, y, z: f32};
  1471. N :: 2;
  1472. v_aos: [N]Vector3;
  1473. v_aos[0].x = 1;
  1474. v_aos[0].y = 4;
  1475. v_aos[0].z = 9;
  1476. fmt.println(len(v_aos));
  1477. fmt.println(v_aos[0]);
  1478. fmt.println(v_aos[0].x);
  1479. fmt.println(&v_aos[0].x);
  1480. v_aos[1] = {0, 3, 4};
  1481. v_aos[1].x = 2;
  1482. fmt.println(v_aos[1]);
  1483. fmt.println(v_aos);
  1484. v_soa: #soa[N]Vector3;
  1485. v_soa[0].x = 1;
  1486. v_soa[0].y = 4;
  1487. v_soa[0].z = 9;
  1488. // Same syntax as AOS and treat as if it was an array
  1489. fmt.println(len(v_soa));
  1490. fmt.println(v_soa[0]);
  1491. fmt.println(v_soa[0].x);
  1492. fmt.println(&v_soa[0].x);
  1493. v_soa[1] = {0, 3, 4};
  1494. v_soa[1].x = 2;
  1495. fmt.println(v_soa[1]);
  1496. // Can use SOA syntax if necessary
  1497. v_soa.x[0] = 1;
  1498. v_soa.y[0] = 4;
  1499. v_soa.z[0] = 9;
  1500. fmt.println(v_soa.x[0]);
  1501. // Same pointer addresses with both syntaxes
  1502. assert(&v_soa[0].x == &v_soa.x[0]);
  1503. // Same fmt printing
  1504. fmt.println(v_aos);
  1505. fmt.println(v_soa);
  1506. }
  1507. {
  1508. // Works with arrays of length <= 4 which have the implicit fields xyzw/rgba
  1509. Vector3 :: distinct [3]f32;
  1510. N :: 2;
  1511. v_aos: [N]Vector3;
  1512. v_aos[0].x = 1;
  1513. v_aos[0].y = 4;
  1514. v_aos[0].z = 9;
  1515. v_soa: #soa[N]Vector3;
  1516. v_soa[0].x = 1;
  1517. v_soa[0].y = 4;
  1518. v_soa[0].z = 9;
  1519. }
  1520. {
  1521. // SOA Slices
  1522. // Vector3 :: struct {x, y, z: f32};
  1523. Vector3 :: struct {x: i8, y: i16, z: f32};
  1524. N :: 3;
  1525. v: #soa[N]Vector3;
  1526. v[0].x = 1;
  1527. v[0].y = 4;
  1528. v[0].z = 9;
  1529. s: #soa[]Vector3;
  1530. s = v[:];
  1531. assert(len(s) == N);
  1532. fmt.println(s);
  1533. fmt.println(s[0].x);
  1534. a := s[1:2];
  1535. assert(len(a) == 1);
  1536. fmt.println(a);
  1537. d: #soa[dynamic]Vector3;
  1538. append_soa(&d, Vector3{1, 2, 3}, Vector3{4, 5, 9}, Vector3{-4, -4, 3});
  1539. fmt.println(d);
  1540. fmt.println(len(d));
  1541. fmt.println(cap(d));
  1542. fmt.println(d[:]);
  1543. }
  1544. { // soa_zip and soa_unzip
  1545. fmt.println("\nsoa_zip and soa_unzip");
  1546. x := []i32{1, 3, 9};
  1547. y := []f32{2, 4, 16};
  1548. z := []b32{true, false, true};
  1549. // produce an #soa slice the normal slices passed
  1550. s := soa_zip(a=x, b=y, c=z);
  1551. // iterate over the #soa slice
  1552. for v, i in s {
  1553. fmt.println(v, i); // exactly the same as s[i]
  1554. // NOTE: 'v' is NOT a temporary value but has a specialized addressing mode
  1555. // which means that when accessing v.a etc, it does the correct transformation
  1556. // internally:
  1557. // s[i].a === s.a[i]
  1558. fmt.println(v.a, v.b, v.c);
  1559. }
  1560. // Recover the slices from the #soa slice
  1561. a, b, c := soa_unzip(s);
  1562. fmt.println(a, b, c);
  1563. }
  1564. }
  1565. constant_literal_expressions :: proc() {
  1566. fmt.println("\n#constant literal expressions");
  1567. Bar :: struct {x, y: f32};
  1568. Foo :: struct {a, b: int, using c: Bar};
  1569. FOO_CONST :: Foo{b = 2, a = 1, c = {3, 4}};
  1570. fmt.println(FOO_CONST.a);
  1571. fmt.println(FOO_CONST.b);
  1572. fmt.println(FOO_CONST.c);
  1573. fmt.println(FOO_CONST.c.x);
  1574. fmt.println(FOO_CONST.c.y);
  1575. fmt.println(FOO_CONST.x); // using works as expected
  1576. fmt.println(FOO_CONST.y);
  1577. fmt.println("-------");
  1578. ARRAY_CONST :: [3]int{1 = 4, 2 = 9, 0 = 1};
  1579. fmt.println(ARRAY_CONST[0]);
  1580. fmt.println(ARRAY_CONST[1]);
  1581. fmt.println(ARRAY_CONST[2]);
  1582. fmt.println("-------");
  1583. FOO_ARRAY_DEFAULTS :: [3]Foo{{}, {}, {}};
  1584. fmt.println(FOO_ARRAY_DEFAULTS[2].x);
  1585. fmt.println("-------");
  1586. Baz :: enum{A=5, B, C, D};
  1587. ENUM_ARRAY_CONST :: [Baz]int{.A .. .C = 1, .D = 16};
  1588. fmt.println(ENUM_ARRAY_CONST[.A]);
  1589. fmt.println(ENUM_ARRAY_CONST[.B]);
  1590. fmt.println(ENUM_ARRAY_CONST[.C]);
  1591. fmt.println(ENUM_ARRAY_CONST[.D]);
  1592. fmt.println("-------");
  1593. Partial_Baz :: enum{A=5, B, C, D=16};
  1594. #assert(len(Partial_Baz) < len(#partial [Partial_Baz]int));
  1595. PARTIAL_ENUM_ARRAY_CONST :: #partial [Partial_Baz]int{.A .. .C = 1, .D = 16};
  1596. fmt.println(PARTIAL_ENUM_ARRAY_CONST[.A]);
  1597. fmt.println(PARTIAL_ENUM_ARRAY_CONST[.B]);
  1598. fmt.println(PARTIAL_ENUM_ARRAY_CONST[.C]);
  1599. fmt.println(PARTIAL_ENUM_ARRAY_CONST[.D]);
  1600. fmt.println("-------");
  1601. STRING_CONST :: "Hellope!";
  1602. fmt.println(STRING_CONST[0]);
  1603. fmt.println(STRING_CONST[2]);
  1604. fmt.println(STRING_CONST[3]);
  1605. fmt.println(STRING_CONST[0:5]);
  1606. fmt.println(STRING_CONST[3:][:4]);
  1607. }
  1608. union_maybe :: proc() {
  1609. fmt.println("\n#union #maybe");
  1610. Maybe :: union(T: typeid) #maybe {T};
  1611. i: Maybe(u8);
  1612. p: Maybe(^u8); // No tag is stored for pointers, nil is the sentinel value
  1613. #assert(size_of(i) == size_of(u8) + size_of(u8));
  1614. #assert(size_of(p) == size_of(^u8));
  1615. i = 123;
  1616. x := i.?;
  1617. y, y_ok := p.?;
  1618. p = &x;
  1619. z, z_ok := p.?;
  1620. fmt.println(i, p);
  1621. fmt.println(x, &x);
  1622. fmt.println(y, y_ok);
  1623. fmt.println(z, z_ok);
  1624. }
  1625. dummy_procedure :: proc() {
  1626. fmt.println("dummy_procedure");
  1627. }
  1628. explicit_context_definition :: proc "c" () {
  1629. // Try commenting the following statement out below
  1630. context = runtime.default_context();
  1631. fmt.println("\n#explicit context definition");
  1632. dummy_procedure();
  1633. }
  1634. relative_data_types :: proc() {
  1635. fmt.println("\n#relative data types");
  1636. x: int = 123;
  1637. ptr: #relative(i16) ^int;
  1638. ptr = &x;
  1639. fmt.println(ptr^);
  1640. arr := [3]int{1, 2, 3};
  1641. s := arr[:];
  1642. rel_slice: #relative(i16) []int;
  1643. rel_slice = s;
  1644. fmt.println(rel_slice);
  1645. fmt.println(rel_slice[:]);
  1646. fmt.println(rel_slice[1]);
  1647. }
  1648. main :: proc() {
  1649. when true {
  1650. the_basics();
  1651. control_flow();
  1652. named_proc_return_parameters();
  1653. explicit_procedure_overloading();
  1654. struct_type();
  1655. union_type();
  1656. using_statement();
  1657. implicit_context_system();
  1658. parametric_polymorphism();
  1659. array_programming();
  1660. map_type();
  1661. implicit_selector_expression();
  1662. partial_switch();
  1663. cstring_example();
  1664. bit_set_type();
  1665. deferred_procedure_associations();
  1666. reflection();
  1667. quaternions();
  1668. unroll_for_statement();
  1669. where_clauses();
  1670. foreign_system();
  1671. ranged_fields_for_array_compound_literals();
  1672. deprecated_attribute();
  1673. range_statements_with_multiple_return_values();
  1674. threading_example();
  1675. soa_struct_layout();
  1676. constant_literal_expressions();
  1677. union_maybe();
  1678. explicit_context_definition();
  1679. relative_data_types();
  1680. }
  1681. }