Mu - linux/tile/grapheme-stack.mu

https://github.com/akkartik/mu/blob/main/linux/tile/grapheme-stack.mu

  1 type grapheme-stack {
  2   data: (handle array grapheme)
  3   top: int
  4 }
  5 
  6 fn initialize-grapheme-stack _self: (addr grapheme-stack), n: int {
  7   var self/esi: (addr grapheme-stack) <- copy _self
  8   var d/edi: (addr handle array grapheme) <- get self, data
  9   populate d, n
 10   var top/eax: (addr int) <- get self, top
 11   copy-to *top, 0
 12 }
 13 
 14 fn clear-grapheme-stack _self: (addr grapheme-stack) {
 15   var self/esi: (addr grapheme-stack) <- copy _self
 16   var top/eax: (addr int) <- get self, top
 17   copy-to *top, 0
 18 pre { line-height: 125%; }
td.linenos .normal { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
span.linenos { color: inherit; background-color: transparent; padding-left: 5px; padding-right: 5px; }
td.linenos .special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
span.linenos.special { color: #000000; background-color: #ffffc0; padding-left: 5px; padding-right: 5px; }
.highlight .hll { background-color: #ffffcc }
.highlight .c { color: #888888 } /* Comment */
.highlight .err { color: #a61717; background-color: #e3d2d2 } /* Error */
.highlight .k { color: #008800; font-weight: bold } /* Keyword */
.highlight .ch { color: #888888 } /* Comment.Hashbang */
.highlight .cm { color: #888888 } /* Comment.Multiline */
.highlight .cp { color: #cc0000; font-weight: bold } /* Comment.Preproc */
.highlight .cpf { color: #888888 } /* Comment.PreprocFile */
.highlight .c1 { color: #888888 } /* Comment.Single */
.highlight .cs { color: #cc0000; font-weight: bold; background-color: #fff0f0 } /* Comment.Special */
.highlight .gd { color: #000000; background-color: #ffdddd } /* Generic.Deleted */
.highlight .ge { font-style: italic } /* Generic.Emph */
.highlight .ges { font-weight: bold; font-style: italic } /* Generic.EmphStrong */
.highlight .gr { color: #aa0000 } /* Generic.Error */
.highlight .gh { color: #333333 } /* Generic.Heading */
.highlight .gi { color: #000000; background-color: #ddffdd } /* Generic.Inserted */
.highlight .go { color: #888888 } /* Generic.Output */
.highlight .gp { color: #555555 } /* Generic.Prompt */
.highlight .gs { font-weight: bold } /* Generic.Strong */
.highlight .gu { color: #666666 } /* Generic.Subheading */
.highlight .gt { color: #aa0000 } /* Generic.Traceback */
.highlight .kc { color: #008800; font-weight: bold } /* Keyword.Constant */
.highlight .kd { color: #008800; font-weight: bold } /* Keyword.Declaration */
.highlight .kn { color: #008800; font-weight: bold } /* Keyword.Namespace */
.highlight .kp { color: #008800 } /* Keyword.Pseudo */
.highlight .kr { color: #008800; font-weight: bold } /* Keyword.Reserved */
.highlight .kt { color: #888888; font-weight: bold } /* Keyword.Type */
.highlight .m { color: #0000DD; font-weight: bold } /* Literal.Number */
.highlight .s { color: #dd2200; background-color: #fff0f0 } /* Literal.String */
.highlight .na { color: #336699 } /* Name.Attribute */
.highlight .nb { color: #003388 } /* Name.Builtin */
.highlight .nc { color: #bb0066; font-weight: bold } /* Name.Class */
.highlight .no { color: #003366; font-weight: bold } /* Name.Constant */
.highlight .nd { color: #555555 } /* Name.Decorator */
.highlight .ne { color: #bb0066; font-weight: bold } /* Name.Exception */
.highlight .nf { color: #0066bb; font-weight: bold } /* Name.Function */
.highlight .nl { color: #336699; font-style: italic } /* Name.Label */
.highlight .nn { color: #bb0066; font-weight: bold } /* Name.Namespace */
.highlight .py { color: #336699; font-weight: bold } /* Name.Property */
.highlight .nt { color: #bb0066; font-weight: bold } /* Name.Tag */
.highlight .nv { color: #336699 } /* Name.Variable */
.highlight .ow { color: #008800 } /* Operator.Word */
.highlight .w { color: #bbbbbb } /* Text.Whitespace */
.highlight .mb { color: #0000DD; font-weight: bold } /* Literal.Number.Bin */
.highlight .mf { color: #0000DD; font-weight: bold } /* Literal.Number.Float */
.highlight .mh { color: #0000DD; font-weight: bold } /* Literal.Number.Hex */
.highlight .mi { color: #0000DD; font-weight: bold } /* Literal.Number.Integer */
.highlight .mo { color: #0000DD; font-weight: bold } /* Literal.Number.Oct */
.highlight .sa { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Affix */
.highlight .sb { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Backtick */
.highlight .sc { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Char */
.highlight .dl { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Delimiter */
.highlight .sd { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Doc */
.highlight .s2 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Double */
.highlight .se { color: #0044dd; background-color: #fff0f0 } /* Literal.String.Escape */
.highlight .sh { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Heredoc */
.highlight .si { color: #3333bb; background-color: #fff0f0 } /* Literal.String.Interpol */
.highlight .sx { color: #22bb22; background-color: #f0fff0 } /* Literal.String.Other */
.highlight .sr { color: #008800; background-color: #fff0ff } /* Literal.String.Regex */
.highlight .s1 { color: #dd2200; background-color: #fff0f0 } /* Literal.String.Single */
.highlight .ss { color: #aa6600; background-color: #fff0f0 } /* Literal.String.Symbol */
.highlight .bp { color: #003388 } /* Name.Builtin.Pseudo */
.highlight .fm { color: #0066bb; font-weight: bold } /* Name.Function.Magic */
.highlight .vc { color: #336699 } /* Name.Variable.Class */
.highlight .vg { color: #dd7700 } /* Name.Variable.Global */
.highlight .vi { color: #3333bb } /* Name.Variable.Instance */
.highlight .vm { color: #336699 } /* Name.Variable.Magic */
.highlight .il { color: #0000DD; font-weight: bold } /* Literal.Number.Integer.Long */discard """
  output: '''@[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 12, 18, 20, 21, 24, 27, 30, 36, 40, 42]
1002'''
"""

import strutils

proc slice[T](iter: iterator(): T {.closure.}, sl: auto): seq[T] =
  var res: seq[int64] = @[]
  var i = 0
  for n in iter():
    if i > sl.b:
      break
    if i >= sl.a:
      res.add(n)
    inc i
  res

iterator harshad(): int64 {.closure.} =
  for n in 1 .. < int64.high:
    var sum = 0
    for ch in string($n):
      sum += parseInt("" & ch)
    if n mod sum == 0:
      yield n

echo harshad.slice 0 .. <20

for n in harshad():
  if n > 1000:
    echo n
    break


# bug #3499 last snippet fixed
# bug 705  last snippet fixed

an id="L97" class="LineNr"> 97 fn render-stack-from-top _self: (addr grapheme-stack), screen: (addr screen) { 98 var self/esi: (addr grapheme-stack) <- copy _self 99 var data-ah/edi: (addr handle array grapheme) <- get self, data 100 var _data/eax: (addr array grapheme) <- lookup *data-ah 101 var data/edi: (addr array grapheme) <- copy _data 102 var top-addr/ecx: (addr int) <- get self, top 103 var i/eax: int <- copy *top-addr 104 i <- decrement 105 { 106 compare i, 0 107 break-if-< 108 var g/edx: (addr grapheme) <- index data, i 109 print-grapheme screen, *g 110 i <- decrement 111 loop 112 } 113 } 114 115 # compare from bottom 116 # beware: modifies 'stream', which must be disposed of after a false result 117 fn prefix-match? _self: (addr grapheme-stack), s: (addr stream byte) -> _/eax: boolean { 118 var self/esi: (addr grapheme-stack) <- copy _self 119 var data-ah/edi: (addr handle array grapheme) <- get self, data 120 var _data/eax: (addr array grapheme) <- lookup *data-ah 121 var data/edi: (addr array grapheme) <- copy _data 122 var top-addr/ecx: (addr int) <- get self, top 123 var i/ebx: int <- copy 0 124 { 125 compare i, *top-addr 126 break-if->= 127 # if curr != expected, return false 128 { 129 var curr-a/edx: (addr grapheme) <- index data, i 130 var expected/eax: grapheme <- read-grapheme s 131 { 132 compare expected, *curr-a 133 break-if-= 134 return 0/false 135 } 136 } 137 i <- increment 138 loop 139 } 140 return 1 # true 141 } 142 143 # compare from bottom 144 # beware: modifies 'stream', which must be disposed of after a false result 145 fn suffix-match? _self: (addr grapheme-stack), s: (addr stream byte) -> _/eax: boolean { 146 var self/esi: (addr grapheme-stack) <- copy _self 147 var data-ah/edi: (addr handle array grapheme) <- get self, data 148 var _data/eax: (addr array grapheme) <- lookup *data-ah 149 var data/edi: (addr array grapheme) <- copy _data 150 var top-addr/eax: (addr int) <- get self, top 151 var i/ebx: int <- copy *top-addr 152 i <- decrement 153 { 154 compare i, 0 155 break-if-< 156 { 157 var curr-a/edx: (addr grapheme) <- index data, i 158 var expected/eax: grapheme <- read-grapheme s 159 # if curr != expected, return false 160 { 161 compare expected, *curr-a 162 break-if-= 163 return 0/false 164 } 165 } 166 i <- decrement 167 loop 168 } 169 return 1 # true 170 } 171 172 fn grapheme-stack-is-decimal-integer? _self: (addr grapheme-stack) -> _/eax: boolean { 173 var self/esi: (addr grapheme-stack) <- copy _self 174 var data-ah/eax: (addr handle array grapheme) <- get self, data 175 var _data/eax: (addr array grapheme) <- lookup *data-ah 176 var data/edx: (addr array grapheme) <- copy _data 177 var top-addr/ecx: (addr int) <- get self, top 178 var i/ebx: int <- copy 0 179 var result/eax: boolean <- copy 1/true 180 $grapheme-stack-is-integer?:loop: { 181 compare i, *top-addr 182 break-if->= 183 var g/edx: (addr grapheme) <- index data, i 184 result <- decimal-digit? *g 185 compare result, 0/false 186 break-if-= 187 i <- increment 188 loop 189 } 190 return result 191 }