<!DOCTYPE HTML PUBLIC "-//W3C//DTD HTML 4.01//EN" "http://www.w3.org/TR/html4/strict.dtd">
<html>
<head>
<meta http-equiv="content-type" content="text/html; charset=UTF-8">
<title>Mu - linux/advent2020/3b.mu</title>
<meta name="Generator" content="Vim/8.1">
<meta name="plugin-version" content="vim8.1_v1">
<meta name="syntax" content="none">
<meta name="settings" content="number_lines,use_css,pre_wrap,no_foldcolumn,expand_tabs,line_ids,prevent_copy=">
<meta name="colorscheme" content="minimal-light">
<style type="text/css">
<!--
pre { white-space: pre-wrap; font-family: monospace; color: #000000; background-color: #eeeeee; }
body { font-size:12pt; font-family: monospace; color: #000000; background-color: #eeeeee; }
a { color:inherit; }
* { font-size:12pt; font-size: 1em; }
.LineNr { }
.Delimiter { color: #c000c0; }
.CommentedCode { color: #8a8a8a; }
.muRegEdx { color: #878700; }
.muRegEbx { color: #8787af; }
.muRegEsi { color: #87d787; }
.muRegEdi { color: #87ffd7; }
.Constant { color: #008787; }
.Special { color: #ff6060; }
.PreProc { color: #c000c0; }
.muFunction { color: #af5f00; text-decoration: underline; }
.muComment { color: #005faf; }
.muRegEax { color: #875f00; }
.muRegEcx { color: #af875f; }
-->
</style>
<script type='text/javascript'>
<!--
/* function to open any folds containing a jumped-to line before jumping to it */
function JumpToLine()
{
var lineNum;
lineNum = window.location.hash;
lineNum = lineNum.substr(1); /* strip off '#' */
if (lineNum.indexOf('L') == -1) {
lineNum = 'L'+lineNum;
}
var lineElem = document.getElementById(lineNum);
/* Always jump to new location even if the line was hidden inside a fold, or
* we corrected the raw number to a line ID.
*/
if (lineElem) {
lineElem.scrollIntoView(true);
}
return true;
}
if ('onhashchange' in window) {
window.onhashchange = JumpToLine;
}
-->
</script>
</head>
<body onload='JumpToLine();'>
<a href='https://github.com/akkartik/mu/blob/main/linux/advent2020/3b.mu'>https://github.com/akkartik/mu/blob/main/linux/advent2020/3b.mu</a>
<pre id='vimCodeElement'>
<span id="L1" class="LineNr"> 1 </span><span class="muComment"># <a href="https://adventofcode.com/2020/day/3">https://adventofcode.com/2020/day/3</a></span>
<span id="L2" class="LineNr"> 2 </span><span class="muComment">#</span>
<span id="L3" class="LineNr"> 3 </span><span class="muComment"># To run (on Linux):</span>
<span id="L4" class="LineNr"> 4 </span><span class="muComment"># $ git clone <a href="https://github.com/akkartik/mu">https://github.com/akkartik/mu</a></span>
<span id="L5" class="LineNr"> 5 </span><span class="muComment"># $ cd mu</span>
<span id="L6" class="LineNr"> 6 </span><span class="muComment"># $ ./translate advent2020/3a.mu</span>
<span id="L7" class="LineNr"> 7 </span><span class="muComment"># $ ./a.elf < input</span>
<span id="L8" class="LineNr"> 8 </span><span class="muComment">#</span>
<span id="L9" class="LineNr"> 9 </span><span class="muComment"># You'll need to register to download the 'input' file for yourself.</span>
<span id="L10" class="LineNr"> 10 </span>
<span id="L11" class="LineNr"> 11 </span><span class="PreProc">fn</span> <span class="muFunction"><a href='3b.mu.html#L11'>main</a></span><span class="PreProc"> -> </span>_/<span class="muRegEbx">ebx</span>: int <span class="Delimiter">{</span>
<span id="L12" class="LineNr"> 12 </span> <span class="muComment"># represent trees in a 2D array of ints</span>
<span id="L13" class="LineNr"> 13 </span> <span class="muComment"># wasteful since each tree is just one bit</span>
<span id="L14" class="LineNr"> 14 </span> <span class="PreProc">var</span> trees-storage: (array int <span class="Constant">0x2800</span>) <span class="muComment"># 10k ints</span>
<span id="L15" class="LineNr"> 15 </span> <span class="PreProc">var</span> trees/<span class="muRegEsi">esi</span>: (addr array int) <span class="Special"><-</span> address trees-storage
<span id="L16" class="LineNr"> 16 </span> <span class="PreProc">var</span> trees-length/<span class="muRegEcx">ecx</span>: int <span class="Special"><-</span> copy <span class="Constant">0</span>
<span id="L17" class="LineNr"> 17 </span> <span class="PreProc">var</span> num-rows: int
<span id="L18" class="LineNr"> 18 </span> <span class="PreProc">var</span> width: int
<span id="L19" class="LineNr"> 19 </span> <span class="muComment"># phase 1: parse each row of trees from stdin</span>
<span id="L20" class="LineNr"> 20 </span> <span class="Delimiter">{</span>
<span id="L21" class="LineNr"> 21 </span> <span class="PreProc">var</span> line-storage: (stream byte <span class="Constant">0x40</span>) <span class="muComment"># 64 bytes</span>
<span id="L22" class="LineNr"> 22 </span> <span class="PreProc">var</span> line/<span class="muRegEdx">edx</span>: (addr stream byte) <span class="Special"><-</span> address line-storage
<span id="L23" class="LineNr"> 23 </span> <span class="Delimiter">{</span>
<span id="L24" class="LineNr"> 24 </span> <span class="muComment"># read line from stdin</span>
<span id="L25" class="LineNr"> 25 </span> <a href='../106stream.subx.html#L17'>clear-stream</a> line
<span id="L26" class="LineNr"> 26 </span> <a href='../305keyboard.subx.html#L158'>read-line-from-real-keyboard</a> line
<span id="L27" class="LineNr"> 27 </span> <span class="muComment"># if line is empty (not even a newline), quit</span>
<span id="L28" class="LineNr"> 28 </span> <span class="PreProc">var</span> done?/<span class="muRegEax">eax</span>: boolean <span class="Special"><-</span> <a href='../309stream.subx.html#L6'>stream-empty?</a> line
<span id="L29" class="LineNr"> 29 </span> compare done?, <span class="Constant">0</span>/false
<span id="L30" class="LineNr"> 30 </span> <span class="PreProc">break-if-!=</span>
<span id="L31" class="LineNr"> 31 </span> <span class="muComment"># wastefully recompute width on every line</span>
<span id="L32" class="LineNr"> 32 </span> <span class="muComment"># zero error-checking; we assume input lines are all equally long</span>
<span id="L33" class="LineNr"> 33 </span> copy-to width, <span class="Constant">0</span>
<span id="L34" class="LineNr"> 34 </span> <span class="muComment"># turn each byte into a tree and append it</span>
<span id="L35" class="LineNr"> 35 </span> $main:line-loop: <span class="Delimiter">{</span>
<span id="L36" class="LineNr"> 36 </span> <span class="PreProc">var</span> done?/<span class="muRegEax">eax</span>: boolean <span class="Special"><-</span> <a href='../309stream.subx.html#L6'>stream-empty?</a> line
<span id="L37" class="LineNr"> 37 </span> compare done?, <span class="Constant">0</span>/false
<span id="L38" class="LineNr"> 38 </span> <span class="PreProc">break-if-!=</span>
<span id="L39" class="LineNr"> 39 </span><span class="CommentedCode">#? print-int32-decimal 0, num-rows</span>
<span id="L40" class="LineNr"> 40 </span><span class="CommentedCode">#? print-string 0, " "</span>
<span id="L41" class="LineNr"> 41 </span><span class="CommentedCode">#? print-int32-decimal 0, width</span>
<span id="L42" class="LineNr"> 42 </span><span class="CommentedCode">#? print-string 0, "\n"</span>
<span id="L43" class="LineNr"> 43 </span> <span class="PreProc">var</span> dest/<span class="muRegEbx">ebx</span>: (addr int) <span class="Special"><-</span> index trees, trees-length
<span id="L44" class="LineNr"> 44 </span> <span class="PreProc">var</span> c/<span class="muRegEax">eax</span>: byte <span class="Special"><-</span> <a href='../112read-byte.subx.html#L273'>read-byte</a> line
<span id="L45" class="LineNr"> 45 </span> <span class="muComment"># newline comes only at end of line</span>
<span id="L46" class="LineNr"> 46 </span> compare c, <span class="Constant">0xa</span>/newline
<span id="L47" class="LineNr"> 47 </span> <span class="PreProc">break-if-=</span>
<span id="L48" class="LineNr"> 48 </span> <span class="muComment"># '#' = tree</span>
<span id="L49" class="LineNr"> 49 </span> compare c, <span class="Constant">0x23</span>/hash
<span id="L50" class="LineNr"> 50 </span> <span class="Delimiter">{</span>
<span id="L51" class="LineNr"> 51 </span> <span class="PreProc">break-if-!=</span>
<span id="L52" class="LineNr"> 52 </span> copy-to *dest, <span class="Constant">1</span>
<span id="L53" class="LineNr"> 53 </span> <span class="Delimiter">}</span>
<span id="L54" class="LineNr"> 54 </span> <span class="muComment"># anything else = no tree</span>
<span id="L55" class="LineNr"> 55 </span> <span class="Delimiter">{</span>
<span id="L56" class="LineNr"> 56 </span> <span class="PreProc">break-if-=</span>
<span id="L57" class="LineNr"> 57 </span> copy-to *dest, <span class="Constant">0</span>
<span id="L58" class="LineNr"> 58 </span> <span class="Delimiter">}</span>
<span id="L59" class="LineNr"> 59 </span> increment width
<span id="L60" class="LineNr"> 60 </span> trees-length <span class="Special"><-</span> increment
<span id="L61" class="LineNr"> 61 </span> <span class="PreProc">loop</span>
<span id="L62" class="LineNr"> 62 </span> <span class="Delimiter">}</span>
<span id="L63" class="LineNr"> 63 </span> increment num-rows
<span id="L64" class="LineNr"> 64 </span> <span class="PreProc">loop</span>
<span id="L65" class="LineNr"> 65 </span> <span class="Delimiter">}</span>
<span id="L66" class="LineNr"> 66 </span> <span class="Delimiter">}</span>
<span id="L67" class="LineNr"> 67 </span> <span class="muComment"># phase 2: compute</span>
<span id="L68" class="LineNr"> 68 </span> <span class="PreProc">var</span> product/<span class="muRegEdi">edi</span>: int <span class="Special"><-</span> copy <span class="Constant">1</span>
<span id="L69" class="LineNr"> 69 </span> <span class="PreProc">var</span> result/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> <a href='3b.mu.html#L94'>num-trees-hit</a> trees, width, num-rows, <span class="Constant">1</span>, <span class="Constant">1</span>
<span id="L70" class="LineNr"> 70 </span> <a href='../405screen.mu.html#L484'>print-int32-decimal</a> <span class="Constant">0</span>, result
<span id="L71" class="LineNr"> 71 </span> <a href='../405screen.mu.html#L169'>print-string</a> <span class="Constant">0</span>, <span class="Constant">" x "</span>
<span id="L72" class="LineNr"> 72 </span> product <span class="Special"><-</span> multiply result
<span id="L73" class="LineNr"> 73 </span> <span class="PreProc">var</span> result/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> <a href='3b.mu.html#L94'>num-trees-hit</a> trees, width, num-rows, <span class="Constant">3</span>, <span class="Constant">1</span>
<span id="L74" class="LineNr"> 74 </span> <a href='../405screen.mu.html#L484'>print-int32-decimal</a> <span class="Constant">0</span>, result
<span id="L75" class="LineNr"> 75 </span> <a href='../405screen.mu.html#L169'>print-string</a> <span class="Constant">0</span>, <span class="Constant">" x "</span>
<span id="L76" class="LineNr"> 76 </span> product <span class="Special"><-</span> multiply result
<span id="L77" class="LineNr"> 77 </span> <span class="PreProc">var</span> result/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> <a href='3b.mu.html#L94'>num-trees-hit</a> trees, width, num-rows, <span class="Constant">5</span>, <span class="Constant">1</span>
<span id="L78" class="LineNr"> 78 </span> <a href='../405screen.mu.html#L484'>print-int32-decimal</a> <span class="Constant">0</span>, result
<span id="L79" class="LineNr"> 79 </span> <a href='../405screen.mu.html#L169'>print-string</a> <span class="Constant">0</span>, <span class="Constant">" x "</span>
<span id="L80" class="LineNr"> 80 </span> product <span class="Special"><-</span> multiply result
<span id="L81" class="LineNr"> 81 </span> <span class="PreProc">var</span> result/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> <a href='3b.mu.html#L94'>num-trees-hit</a> trees, width, num-rows, <span class="Constant">7</span>, <span class="Constant">1</span>
<span id="L82" class="LineNr"> 82 </span> <a href='../405screen.mu.html#L484'>print-int32-decimal</a> <span class="Constant">0</span>, result
<span id="L83" class="LineNr"> 83 </span> <a href='../405screen.mu.html#L169'>print-string</a> <span class="Constant">0</span>, <span class="Constant">" x "</span>
<span id="L84" class="LineNr"> 84 </span> product <span class="Special"><-</span> multiply result
<span id="L85" class="LineNr"> 85 </span> <span class="PreProc">var</span> result/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> <a href='3b.mu.html#L94'>num-trees-hit</a> trees, width, num-rows, <span class="Constant">1</span>, <span class="Constant">2</span>
<span id="L86" class="LineNr"> 86 </span> <a href='../405screen.mu.html#L484'>print-int32-decimal</a> <span class="Constant">0</span>, result
<span id="L87" class="LineNr"> 87 </span> <a href='../405screen.mu.html#L169'>print-string</a> <span class="Constant">0</span>, <span class="Constant">" = "</span>
<span id="L88" class="LineNr"> 88 </span> product <span class="Special"><-</span> multiply result
<span id="L89" class="LineNr"> 89 </span> <a href='../405screen.mu.html#L440'>print-int32-hex</a> <span class="Constant">0</span>, product
<span id="L90" class="LineNr"> 90 </span> <a href='../405screen.mu.html#L169'>print-string</a> <span class="Constant">0</span>, <span class="Constant">"\n"</span>
<span id="L91" class="LineNr"> 91 </span> <span class="PreProc">return</span> <span class="Constant">0</span>
<span id="L92" class="LineNr"> 92 </span><span class="Delimiter">}</span>
<span id="L93" class="LineNr"> 93 </span>
<span id="L94" class="LineNr"> 94 </span><span class="PreProc">fn</span> <span class="muFunction"><a href='3b.mu.html#L94'>num-trees-hit</a></span> trees: (addr array int), width: int, num-rows: int, right: int, down: int<span class="PreProc"> -> </span>_/<span class="muRegEax">eax</span>: int <span class="Delimiter">{</span>
<span id="L95" class="LineNr"> 95 </span><span class="CommentedCode">#? print-string 0, "== "</span>
<span id="L96" class="LineNr"> 96 </span><span class="CommentedCode">#? print-int32-decimal 0, right</span>
<span id="L97" class="LineNr"> 97 </span><span class="CommentedCode">#? print-string 0, " "</span>
<span id="L98" class="LineNr"> 98 </span><span class="CommentedCode">#? print-int32-decimal 0, down</span>
<span id="L99" class="LineNr"> 99 </span><span class="CommentedCode">#? print-string 0, "\n"</span>
<span id="L100" class="LineNr">100 </span> <span class="PreProc">var</span> row/<span class="muRegEcx">ecx</span>: int <span class="Special"><-</span> copy <span class="Constant">0</span>
<span id="L101" class="LineNr">101 </span> <span class="PreProc">var</span> col/<span class="muRegEdx">edx</span>: int <span class="Special"><-</span> copy <span class="Constant">0</span>
<span id="L102" class="LineNr">102 </span> <span class="PreProc">var</span> <a href='3b.mu.html#L94'>num-trees-hit</a>/<span class="muRegEdi">edi</span>: int <span class="Special"><-</span> copy <span class="Constant">0</span>
<span id="L103" class="LineNr">103 </span> <span class="Delimiter">{</span>
<span id="L104" class="LineNr">104 </span> compare row, num-rows
<span id="L105" class="LineNr">105 </span> <span class="PreProc">break-if->=</span>
<span id="L106" class="LineNr">106 </span><span class="CommentedCode">#? print-int32-decimal 0, col</span>
<span id="L107" class="LineNr">107 </span><span class="CommentedCode">#? print-string 0, "\n"</span>
<span id="L108" class="LineNr">108 </span> <span class="PreProc">var</span> curr/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> <a href='3b.mu.html#L121'>index2d</a> trees, row, col, width
<span id="L109" class="LineNr">109 </span> compare curr, <span class="Constant">0</span>
<span id="L110" class="LineNr">110 </span> <span class="Delimiter">{</span>
<span id="L111" class="LineNr">111 </span> <span class="PreProc">break-if-=</span>
<span id="L112" class="LineNr">112 </span> <a href='3b.mu.html#L94'>num-trees-hit</a> <span class="Special"><-</span> increment
<span id="L113" class="LineNr">113 </span> <span class="Delimiter">}</span>
<span id="L114" class="LineNr">114 </span> col <span class="Special"><-</span> add right
<span id="L115" class="LineNr">115 </span> row <span class="Special"><-</span> add down
<span id="L116" class="LineNr">116 </span> <span class="PreProc">loop</span>
<span id="L117" class="LineNr">117 </span> <span class="Delimiter">}</span>
<span id="L118" class="LineNr">118 </span> <span class="PreProc">return</span> <a href='3b.mu.html#L94'>num-trees-hit</a>
<span id="L119" class="LineNr">119 </span><span class="Delimiter">}</span>
<span id="L120" class="LineNr">120 </span>
<span id="L121" class="LineNr">121 </span><span class="PreProc">fn</span> <span class="muFunction"><a href='3b.mu.html#L121'>index2d</a></span> _arr: (addr array int), _row: int, _col: int, width: int<span class="PreProc"> -> </span>_/<span class="muRegEax">eax</span>: int <span class="Delimiter">{</span>
<span id="L122" class="LineNr">122 </span> <span class="muComment"># handle repeating columns of trees</span>
<span id="L123" class="LineNr">123 </span> <span class="PreProc">var</span> dummy/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> copy <span class="Constant">0</span>
<span id="L124" class="LineNr">124 </span> <span class="PreProc">var</span> col/<span class="muRegEdx">edx</span>: int <span class="Special"><-</span> copy <span class="Constant">0</span>
<span id="L125" class="LineNr">125 </span> dummy, col <span class="Special"><-</span> <a href='../314divide.subx.html#L3'>integer-divide</a> _col, width
<span id="L126" class="LineNr">126 </span><span class="CommentedCode">#? print-string 0, " "</span>
<span id="L127" class="LineNr">127 </span><span class="CommentedCode">#? print-int32-decimal 0, col</span>
<span id="L128" class="LineNr">128 </span><span class="CommentedCode">#? print-string 0, "\n"</span>
<span id="L129" class="LineNr">129 </span> <span class="muComment"># compute index</span>
<span id="L130" class="LineNr">130 </span> <span class="PreProc">var</span> index/<span class="muRegEax">eax</span>: int <span class="Special"><-</span> copy _row
<span id="L131" class="LineNr">131 </span> index <span class="Special"><-</span> multiply width
<span id="L132" class="LineNr">132 </span> index <span class="Special"><-</span> add col
<span id="L133" class="LineNr">133 </span> <span class="muComment"># look up array</span>
<span id="L134" class="LineNr">134 </span> <span class="PreProc">var</span> arr/<span class="muRegEsi">esi</span>: (addr array int) <span class="Special"><-</span> copy _arr
<span id="L135" class="LineNr">135 </span> <span class="PreProc">var</span> src/<span class="muRegEax">eax</span>: (addr int) <span class="Special"><-</span> index arr, index
<span id="L136" class="LineNr">136 </span> <span class="PreProc">return</span> *src
<span id="L137" class="LineNr">137 </span><span class="Delimiter">}</span>
</pre>
</body>
</html>
<!-- vim: set foldmethod=manual : -->