milkymist/cif.py

   1 from operator import itemgetter
   2 import re
   3
   4 from migen.bank.description import CSRStatus
   5
   6 def get_macros(filename):
   7         f = open(filename, "r")
   8         r = {}
   9         for line in f:
  10                 match = re.match("\w*#define\s+(\w+)\s+(.*)", line, re.IGNORECASE)
  11                 if match:
  12                         r[match.group(1)] = match.group(2)
  13         return r
  14
  15 def _get_rw_functions(reg_name, reg_base, size, read_only):
  16         r = ""
  17         if size > 8:
  18                 raise NotImplementedError("Register too large")
  19         elif size > 4:
  20                 ctype = "unsigned long long int"
  21         elif size > 2:
  22                 ctype = "unsigned int"
  23         elif size > 1:
  24                 ctype = "unsigned short int"
  25         else:
  26                 ctype = "unsigned char"
  27
  28         r += "static inline "+ctype+" "+reg_name+"_read(void) {\n"
  29         if size > 1:
  30                 r += "\t"+ctype+" r = MMPTR("+hex(reg_base)+");\n"
  31                 for byte in range(1, size):
  32                         r += "\tr <<= 8;\n\tr |= MMPTR("+hex(reg_base+4*byte)+");\n"
  33                 r += "\treturn r;\n}\n"
  34         else:
  35                 r += "\treturn MMPTR("+hex(reg_base)+");\n}\n"
  36
  37         if not read_only:
  38                 r += "static inline void "+reg_name+"_write("+ctype+" value) {\n"
  39                 for byte in range(size):
  40                         shift = (size-byte-1)*8
  41                         if shift:
  42                                 value_shifted = "value >> "+str(shift)
  43                         else:
  44                                 value_shifted = "value"
  45                         r += "\tMMPTR("+hex(reg_base+4*byte)+") = "+value_shifted+";\n"
  46                 r += "}\n"
  47         return r
  48
  49 def get_csr_header(csr_base, bank_array, interrupt_map):
  50         r = "#ifndef __HW_CSR_H\n#define __HW_CSR_H\n#include <hw/common.h>\n"
  51         for name, csrs, mapaddr, rmap in bank_array.banks:
  52                 r += "\n/* "+name+" */\n"
  53                 reg_base = csr_base + 0x800*mapaddr
  54                 r += "#define "+name.upper()+"_BASE "+hex(reg_base)+"\n"
  55                 for csr in csrs:
  56                         nr = (csr.size + 7)//8
  57                         r += _get_rw_functions(name + "_" + csr.name, reg_base, nr, isinstance(csr, CSRStatus))
  58                         reg_base += 4*nr
  59                 try:
  60                         interrupt_nr = interrupt_map[name]
  61                 except KeyError:
  62                         pass
  63                 else:
  64                         r += "#define "+name.upper()+"_INTERRUPT "+str(interrupt_nr)+"\n"
  65         r += "\n#endif\n"
  66         return r