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 micro_asm import MicroAssembler, Combinational_Macroop, Rom_Macroop, Rom
  30
  31 class Bah(object):
  32     def __init__(self):
  33         self.mnemonic = "bah"
  34
  35 class Bah_Tweaked(object):
  36     def __init__(self):
  37         self.mnemonic = "bah_tweaked"
  38
  39 class Hoop(object):
  40     def __init__(self, first_param, second_param):
  41         self.mnemonic = "hoop_%s_%s" % (first_param, second_param)
  42     def __str__(self):
  43         return "%s" % self.mnemonic
  44
  45 class Dah(object):
  46     def __init__(self):
  47         self.mnemonic = "dah"
  48
  49 microops = {
  50     "bah": Bah,
  51     "hoop": Hoop,
  52     "dah": Dah
  53 }
  54
  55 class TestMacroop(Combinational_Macroop):
  56     def tweak(self):
  57         microops["bah"] = Bah_Tweaked
  58     def untweak(self):
  59         microops["bah"] = Bah
  60     def print_debug(self, message):
  61         print message
  62
  63     def __init__(self, name):
  64         super(TestMacroop, self).__init__(name)
  65         self.directives = {
  66             "tweak": self.tweak,
  67             "untweak": self.untweak,
  68             "print": self.print_debug
  69         }
  70
  71 assembler = MicroAssembler(TestMacroop, microops, Rom('main ROM'), Rom_Macroop)
  72
  73 testAssembly = '''
  74 # Single line comment
  75
  76 def rom {
  77     goo: bah
  78     extern la: hoop 4*8, "a"
  79 }; /* multiline comment on one line */
  80
  81 /* multi line comment across lines
  82    to make sure they work */
  83
  84 def macroop squishy {
  85     .tweak
  86     bah
  87     .untweak
  88     .print "In the midst"
  89     bah
  90     dah # single line comment after something
  91     .tweak
  92 };
  93
  94 #Extending the rom...
  95 def rom
  96 {
  97     #Here's more stuff for the rom
  98     bah
  99 };
 100
 101 def macroop squashy {
 102     bah
 103 };
 104
 105 def macroop jumper (bar);
 106 '''
 107 assembler.assemble(testAssembly)