about summary refs log blame commit diff stats
path: root/html/linux/advent2020/3b.mu.html
blob: 153d86da0f1445ba75ea4a13275a4ec40feb9e85 (plain) (tree)
1
2
3
4
5
6
7
8
9



                                                                                          
                                          



                                                                                                               
                                                 

                       

                                                                                                 

                                     
           

                                  



                             
                             

                            
                                                           
                              

                             
































                                                                                 
                                                                                                                                             





                                                                                                                                                                  
                                                                                                         



                                                                                                                                           
                                                                                                                                                                                                                                                       


                                                                                                                                                                                      

                                                                                                                                                                                                       




                                                                                                                                                                                       
                                                                                                                                                                                                

                                                                                                    

                                                                                                                            
                                                                                                                           
                                                                                                                                                                                                                              
                                                                                                   





                                                                                                                                               
                                                                                                                                                                                                                                
                                                                                                     




                                                                                                                      

                                                                                                                                                                                                                          
                                                                                                                   
                                                                                                     

                                                                                            
                                                                                                   









                                                                                                         
                                                                                                          






                                                                                            

                                                                                                                                                                                                                                                                                                


                                                                                                                                                                              
                                                                                                                                                                                                                                                                                                


                                                                                                                                                                              
                                                                                                                                                                                                                                                                                                


                                                                                                                                                                              
                                                                                                                                                                                                                                                                                                


                                                                                                                                                                              
                                                                                                                                                                                                                                                                                                




                                                                                                                                                                              


                                                                                                              
                                                                                                                                                                                                                                                                                                                                          




                                                                                                                


                                                                                                                                                                                                                                      




                                                                                                                  
                                                                                                                                                                                                                          


                                                                                           
                                                                                                                                       
                                                                              

                                                                                              




                                                                                                                          
                                                                                                                                                                                                                                                                                                                     
                                                                                                              

                                                                                                                                                                                                 
                                                                                                                                                            



                                                                                                                
                                                                                                                                                                      

                                                                                                   
                                                                                          

                                                                                                                                                                                  





                                                                                    
<!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 &lt; 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"> -&gt; </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">&lt;-</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">&lt;-</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">&lt;-</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">&lt;-</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">&lt;-</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, &quot; &quot;</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, &quot;\n&quot;</span>
<span id="L43" class="LineNr"> 43 </span>        <span class="PreProc">var</span> dest/<span class="muRegEbx">ebx</span>: (addr int) <span class="Special">&lt;-</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">&lt;-</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">&lt;-</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">&lt;-</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">&lt;-</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">&quot; x &quot;</span>
<span id="L72" class="LineNr"> 72 </span>  product <span class="Special">&lt;-</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">&lt;-</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">&quot; x &quot;</span>
<span id="L76" class="LineNr"> 76 </span>  product <span class="Special">&lt;-</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">&lt;-</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">&quot; x &quot;</span>
<span id="L80" class="LineNr"> 80 </span>  product <span class="Special">&lt;-</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">&lt;-</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">&quot; x &quot;</span>
<span id="L84" class="LineNr"> 84 </span>  product <span class="Special">&lt;-</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">&lt;-</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">&quot; = &quot;</span>
<span id="L88" class="LineNr"> 88 </span>  product <span class="Special">&lt;-</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">&quot;\n&quot;</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"> -&gt; </span>_/<span class="muRegEax">eax</span>: int <span class="Delimiter">{</span>
<span id="L95" class="LineNr"> 95 </span><span class="CommentedCode">#?   print-string 0, &quot;== &quot;</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, &quot; &quot;</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, &quot;\n&quot;</span>
<span id="L100" class="LineNr">100 </span>  <span class="PreProc">var</span> row/<span class="muRegEcx">ecx</span>: int <span class="Special">&lt;-</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">&lt;-</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">&lt;-</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-&gt;=</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, &quot;\n&quot;</span>
<span id="L108" class="LineNr">108 </span>    <span class="PreProc">var</span> curr/<span class="muRegEax">eax</span>: int <span class="Special">&lt;-</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">&lt;-</span> increment
<span id="L113" class="LineNr">113 </span>    <span class="Delimiter">}</span>
<span id="L114" class="LineNr">114 </span>    col <span class="Special">&lt;-</span> add right
<span id="L115" class="LineNr">115 </span>    row <span class="Special">&lt;-</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"> -&gt; </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">&lt;-</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">&lt;-</span> copy <span class="Constant">0</span>
<span id="L125" class="LineNr">125 </span>  dummy, col <span class="Special">&lt;-</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, &quot;  &quot;</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, &quot;\n&quot;</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">&lt;-</span> copy _row
<span id="L131" class="LineNr">131 </span>  index <span class="Special">&lt;-</span> multiply width
<span id="L132" class="LineNr">132 </span>  index <span class="Special">&lt;-</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">&lt;-</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">&lt;-</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 : -->