src/arch/micro_asm_test.py

   1 # Copyright (c) 2007 The Regents of The University of Michigan
   2 # All rights reserved.
   3 #
   4 # Redistribution and use in source and binary forms, with or without
   5 # modification, are permitted provided that the following conditions are
   6 # met: redistributions of source code must retain the above copyright
   7 # notice, this list of conditions and the following disclaimer;
   8 # redistributions in binary form must reproduce the above copyright
   9 # notice, this list of conditions and the following disclaimer in the
  10 # documentation and/or other materials provided with the distribution;
  11 # neither the name of the copyright holders nor the names of its
  12 # contributors may be used to endorse or promote products derived from
  13 # this software without specific prior written permission.
  14 #
  15 # THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  16 # "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  17 # LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  18 # A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
  19 # OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
  20 # SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
  21 # LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  22 # DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  23 # THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  24 # (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
  25 # OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  26 #
  27 # Authors: Gabe Black
  28
  29 from __future__ import print_function
  30
  31 from micro_asm import MicroAssembler, Combinational_Macroop, Rom_Macroop, Rom
  32
  33 class Bah(object):
  34     def __init__(self):
  35         self.mnemonic = "bah"
  36
  37 class Bah_Tweaked(object):
  38     def __init__(self):
  39         self.mnemonic = "bah_tweaked"
  40
  41 class Hoop(object):
  42     def __init__(self, first_param, second_param):
  43         self.mnemonic = "hoop_%s_%s" % (first_param, second_param)
  44     def __str__(self):
  45         return "%s" % self.mnemonic
  46
  47 class Dah(object):
  48     def __init__(self):
  49         self.mnemonic = "dah"
  50
  51 microops = {
  52     "bah": Bah,
  53     "hoop": Hoop,
  54     "dah": Dah
  55 }
  56
  57 class TestMacroop(Combinational_Macroop):
  58     def tweak(self):
  59         microops["bah"] = Bah_Tweaked
  60     def untweak(self):
  61         microops["bah"] = Bah
  62     def print_debug(self, message):
  63         print(message)
  64
  65     def __init__(self, name):
  66         super(TestMacroop, self).__init__(name)
  67         self.directives = {
  68             "tweak": self.tweak,
  69             "untweak": self.untweak,
  70             "print": self.print_debug
  71         }
  72
  73 assembler = MicroAssembler(TestMacroop, microops, Rom('main ROM'), Rom_Macroop)
  74
  75 testAssembly = '''
  76 # Single line comment
  77
  78 def rom {
  79     goo: bah
  80     extern la: hoop 4*8, "a"
  81 }; /* multiline comment on one line */
  82
  83 /* multi line comment across lines
  84    to make sure they work */
  85
  86 def macroop squishy {
  87     .tweak
  88     bah
  89     .untweak
  90     .print "In the midst"
  91     bah
  92     dah # single line comment after something
  93     .tweak
  94 };
  95
  96 #Extending the rom...
  97 def rom
  98 {
  99     #Here's more stuff for the rom
 100     bah
 101 };
 102
 103 def macroop squashy {
 104     bah
 105 };
 106
 107 def macroop jumper (bar);
 108 '''
 109 assembler.assemble(testAssembly)