src/openpower/decoder/power_insn.py

   1 import collections as _collections
   2 import csv as _csv
   3 import dataclasses as _dataclasses
   4 import enum as _enum
   5 import functools as _functools
   6 import itertools as _itertools
   7 import os as _os
   8 import operator as _operator
   9 import pathlib as _pathlib
  10 import re as _re
  11
  12 try:
  13     from functools import cached_property
  14 except ImportError:
  15     from cached_property import cached_property
  16
  17 from openpower.decoder.power_enums import (
  18     Function as _Function,
  19     MicrOp as _MicrOp,
  20     In1Sel as _In1Sel,
  21     In2Sel as _In2Sel,
  22     In3Sel as _In3Sel,
  23     OutSel as _OutSel,
  24     CRInSel as _CRInSel,
  25     CRIn2Sel as _CRIn2Sel,
  26     CROutSel as _CROutSel,
  27     LDSTLen as _LDSTLen,
  28     LDSTMode as _LDSTMode,
  29     RCOE as _RCOE,
  30     CryIn as _CryIn,
  31     Form as _Form,
  32     SVEtype as _SVEtype,
  33     SVmask_src as _SVmask_src,
  34     SVMode as _SVMode,
  35     SVPtype as _SVPtype,
  36     SVExtra as _SVExtra,
  37     RegType as _RegType,
  38     SVExtraRegType as _SVExtraRegType,
  39     SVExtraReg as _SVExtraReg,
  40 )
  41 from openpower.decoder.selectable_int import (
  42     SelectableInt as _SelectableInt,
  43     selectconcat as _selectconcat,
  44 )
  45 from openpower.decoder.power_fields import (
  46     Field as _Field,
  47     Mapping as _Mapping,
  48     DecodeFields as _DecodeFields,
  49 )
  50 from openpower.decoder.pseudo.pagereader import ISA as _ISA
  51
  52
  53 @_functools.total_ordering
  54 class Verbosity(_enum.Enum):
  55     SHORT = _enum.auto()
  56     NORMAL = _enum.auto()
  57     VERBOSE = _enum.auto()
  58
  59     def __lt__(self, other):
  60         if not isinstance(other, self.__class__):
  61             return NotImplemented
  62         return (self.value < other.value)
  63
  64
  65 def dataclass(cls, record, keymap=None, typemap=None):
  66     if keymap is None:
  67         keymap = {}
  68     if typemap is None:
  69         typemap = {field.name:field.type for field in _dataclasses.fields(cls)}
  70
  71     def transform(key_value):
  72         (key, value) = key_value
  73         key = keymap.get(key, key)
  74         hook = typemap.get(key, lambda value: value)
  75         if hook is bool and value in ("", "0"):
  76             value = False
  77         else:
  78             value = hook(value)
  79         return (key, value)
  80
  81     record = dict(map(transform, record.items()))
  82     for key in frozenset(record.keys()):
  83         if record[key] == "":
  84             record.pop(key)
  85
  86     return cls(**record)
  87
  88
  89 @_functools.total_ordering
  90 @_dataclasses.dataclass(eq=True, frozen=True)
  91 class Opcode:
  92     class Integer(int):
  93         def __new__(cls, value):
  94             if isinstance(value, str):
  95                 value = int(value, 0)
  96             if not isinstance(value, int):
  97                 raise ValueError(value)
  98
  99             if value.bit_length() > 64:
 100                 raise ValueError(value)
 101
 102             return super().__new__(cls, value)
 103
 104         def __str__(self):
 105             return super().__repr__()
 106
 107         def __repr__(self):
 108             return f"{self:0{self.bit_length()}b}"
 109
 110         def bit_length(self):
 111             if super().bit_length() > 32:
 112                 return 64
 113             return 32
 114
 115     class Value(Integer):
 116         pass
 117
 118     class Mask(Integer):
 119         pass
 120
 121     value: Value
 122     mask: Mask
 123
 124     def __lt__(self, other):
 125         if not isinstance(other, Opcode):
 126             return NotImplemented
 127         return ((self.value, self.mask) < (other.value, other.mask))
 128
 129     def __post_init__(self):
 130         if self.value.bit_length() != self.mask.bit_length():
 131             raise ValueError("bit length mismatch")
 132
 133     def __repr__(self):
 134         def pattern(value, mask, bit_length):
 135             for bit in range(bit_length):
 136                 if ((mask & (1 << (bit_length - bit - 1))) == 0):
 137                     yield "-"
 138                 elif (value & (1 << (bit_length - bit - 1))):
 139                     yield "1"
 140                 else:
 141                     yield "0"
 142
 143         return "".join(pattern(self.value, self.mask, self.value.bit_length()))
 144
 145
 146 class IntegerOpcode(Opcode):
 147     def __init__(self, value):
 148         if value.startswith("0b"):
 149            mask = int(("1" * len(value[2:])), 2)
 150         else:
 151             mask = 0b111111
 152
 153         value = Opcode.Value(value)
 154         mask = Opcode.Mask(mask)
 155
 156         return super().__init__(value=value, mask=mask)
 157
 158
 159 class PatternOpcode(Opcode):
 160     def __init__(self, pattern):
 161         if not isinstance(pattern, str):
 162             raise ValueError(pattern)
 163
 164         (value, mask) = (0, 0)
 165         for symbol in pattern:
 166             if symbol not in {"0", "1", "-"}:
 167                 raise ValueError(pattern)
 168             value |= (symbol == "1")
 169             mask |= (symbol != "-")
 170             value <<= 1
 171             mask <<= 1
 172         value >>= 1
 173         mask >>= 1
 174
 175         value = Opcode.Value(value)
 176         mask = Opcode.Mask(mask)
 177
 178         return super().__init__(value=value, mask=mask)
 179
 180
 181 @_dataclasses.dataclass(eq=True, frozen=True)
 182 class PPCRecord:
 183     class FlagsMeta(type):
 184         def __iter__(cls):
 185             yield from (
 186                 "inv A",
 187                 "inv out",
 188                 "cry out",
 189                 "BR",
 190                 "sgn ext",
 191                 "rsrv",
 192                 "32b",
 193                 "sgn",
 194                 "lk",
 195                 "sgl pipe",
 196             )
 197
 198     class Flags(frozenset, metaclass=FlagsMeta):
 199         def __new__(cls, flags=frozenset()):
 200             flags = frozenset(flags)
 201             diff = (flags - frozenset(cls))
 202             if diff:
 203                 raise ValueError(flags)
 204             return super().__new__(cls, flags)
 205
 206     opcode: Opcode
 207     comment: str
 208     flags: Flags = Flags()
 209     comment2: str = ""
 210     function: _Function = _Function.NONE
 211     intop: _MicrOp = _MicrOp.OP_ILLEGAL
 212     in1: _In1Sel = _In1Sel.RA
 213     in2: _In2Sel = _In2Sel.NONE
 214     in3: _In3Sel = _In3Sel.NONE
 215     out: _OutSel = _OutSel.NONE
 216     cr_in: _CRInSel = _CRInSel.NONE
 217     cr_in2: _CRIn2Sel = _CRIn2Sel.NONE
 218     cr_out: _CROutSel = _CROutSel.NONE
 219     cry_in: _CryIn = _CryIn.ZERO
 220     ldst_len: _LDSTLen = _LDSTLen.NONE
 221     upd: _LDSTMode = _LDSTMode.NONE
 222     Rc: _RCOE = _RCOE.NONE
 223     form: _Form = _Form.NONE
 224     conditions: str = ""
 225     unofficial: bool = False
 226
 227     __KEYMAP = {
 228         "unit": "function",
 229         "internal op": "intop",
 230         "CR in": "cr_in",
 231         "CR out": "cr_out",
 232         "cry in": "cry_in",
 233         "ldst len": "ldst_len",
 234         "rc": "Rc",
 235         "CONDITIONS": "conditions",
 236     }
 237
 238     @classmethod
 239     def CSV(cls, record, opcode_cls):
 240         typemap = {field.name:field.type for field in _dataclasses.fields(cls)}
 241         typemap["opcode"] = opcode_cls
 242
 243         if record["CR in"] == "BA_BB":
 244             record["cr_in"] = "BA"
 245             record["cr_in2"] = "BB"
 246             del record["CR in"]
 247
 248         flags = set()
 249         for flag in frozenset(PPCRecord.Flags):
 250             if bool(record.pop(flag, "")):
 251                 flags.add(flag)
 252         record["flags"] = PPCRecord.Flags(flags)
 253
 254         return dataclass(cls, record,
 255             keymap=PPCRecord.__KEYMAP,
 256             typemap=typemap)
 257
 258     @cached_property
 259     def names(self):
 260         return frozenset(self.comment.split("=")[-1].split("/"))
 261
 262
 263 class PPCMultiRecord(tuple):
 264     def __getattr__(self, attr):
 265         if attr == "opcode":
 266             raise AttributeError(attr)
 267         return getattr(self[0], attr)
 268
 269
 270 @_dataclasses.dataclass(eq=True, frozen=True)
 271 class SVP64Record:
 272     class ExtraMap(tuple):
 273         class Extra(tuple):
 274             @_dataclasses.dataclass(eq=True, frozen=True)
 275             class Entry:
 276                 regtype: _SVExtraRegType = _SVExtraRegType.NONE
 277                 reg: _SVExtraReg = _SVExtraReg.NONE
 278
 279                 def __repr__(self):
 280                     return f"{self.regtype.value}:{self.reg.name}"
 281
 282             def __new__(cls, value="0"):
 283                 if isinstance(value, str):
 284                     def transform(value):
 285                         (regtype, reg) = value.split(":")
 286                         regtype = _SVExtraRegType(regtype)
 287                         reg = _SVExtraReg(reg)
 288                         return cls.Entry(regtype=regtype, reg=reg)
 289
 290                     if value == "0":
 291                         value = tuple()
 292                     else:
 293                         value = map(transform, value.split(";"))
 294
 295                 return super().__new__(cls, value)
 296
 297             def __repr__(self):
 298                 return repr(list(self))
 299
 300         def __new__(cls, value=tuple()):
 301             value = tuple(value)
 302             if len(value) == 0:
 303                 value = (("0",) * 4)
 304             return super().__new__(cls, map(cls.Extra, value))
 305
 306         def __repr__(self):
 307             return repr({index:self[index] for index in range(0, 4)})
 308
 309     name: str
 310     ptype: _SVPtype = _SVPtype.NONE
 311     etype: _SVEtype = _SVEtype.NONE
 312     msrc: _SVmask_src = _SVmask_src.NO # MASK_SRC is active
 313     in1: _In1Sel = _In1Sel.NONE
 314     in2: _In2Sel = _In2Sel.NONE
 315     in3: _In3Sel = _In3Sel.NONE
 316     out: _OutSel = _OutSel.NONE
 317     out2: _OutSel = _OutSel.NONE
 318     cr_in: _CRInSel = _CRInSel.NONE
 319     cr_in2: _CRIn2Sel = _CRIn2Sel.NONE
 320     cr_out: _CROutSel = _CROutSel.NONE
 321     extra: ExtraMap = ExtraMap()
 322     conditions: str = ""
 323     mode: _SVMode = _SVMode.NORMAL
 324
 325     __KEYMAP = {
 326         "insn": "name",
 327         "CONDITIONS": "conditions",
 328         "Ptype": "ptype",
 329         "Etype": "etype",
 330         "SM": "msrc",
 331         "CR in": "cr_in",
 332         "CR out": "cr_out",
 333     }
 334
 335     @classmethod
 336     def CSV(cls, record):
 337         for key in frozenset({
 338                     "in1", "in2", "in3", "CR in",
 339                     "out", "out2", "CR out",
 340                 }):
 341             value = record[key]
 342             if value == "0":
 343                 record[key] = "NONE"
 344
 345         if record["CR in"] == "BA_BB":
 346             record["cr_in"] = "BA"
 347             record["cr_in2"] = "BB"
 348             del record["CR in"]
 349
 350         extra = []
 351         for idx in range(0, 4):
 352             extra.append(record.pop(f"{idx}"))
 353
 354         record["extra"] = cls.ExtraMap(extra)
 355
 356         return dataclass(cls, record, keymap=cls.__KEYMAP)
 357
 358     @_functools.lru_cache(maxsize=None)
 359     def extra_idx(self, key):
 360         extra_idx = (
 361             _SVExtra.Idx0,
 362             _SVExtra.Idx1,
 363             _SVExtra.Idx2,
 364             _SVExtra.Idx3,
 365         )
 366
 367         if key not in frozenset({
 368                     "in1", "in2", "in3", "cr_in", "cr_in2",
 369                     "out", "out2", "cr_out",
 370                 }):
 371             raise KeyError(key)
 372
 373         sel = getattr(self, key)
 374         if sel is _CRInSel.BA_BB:
 375             return _SVExtra.Idx_1_2
 376         reg = _SVExtraReg(sel)
 377         if reg is _SVExtraReg.NONE:
 378             return _SVExtra.NONE
 379
 380         extra_map = {
 381             _SVExtraRegType.SRC: {},
 382             _SVExtraRegType.DST: {},
 383         }
 384         for index in range(0, 4):
 385             for entry in self.extra[index]:
 386                 extra_map[entry.regtype][entry.reg] = extra_idx[index]
 387
 388         for regtype in (_SVExtraRegType.SRC, _SVExtraRegType.DST):
 389             extra = extra_map[regtype].get(reg, _SVExtra.NONE)
 390             if extra is not _SVExtra.NONE:
 391                 return extra
 392
 393         return _SVExtra.NONE
 394
 395     extra_idx_in1 = property(_functools.partial(extra_idx, key="in1"))
 396     extra_idx_in2 = property(_functools.partial(extra_idx, key="in2"))
 397     extra_idx_in3 = property(_functools.partial(extra_idx, key="in3"))
 398     extra_idx_out = property(_functools.partial(extra_idx, key="out"))
 399     extra_idx_out2 = property(_functools.partial(extra_idx, key="out2"))
 400     extra_idx_cr_in = property(_functools.partial(extra_idx, key="cr_in"))
 401     extra_idx_cr_out = property(_functools.partial(extra_idx, key="cr_out"))
 402
 403     @_functools.lru_cache(maxsize=None)
 404     def extra_reg(self, key):
 405         return _SVExtraReg(getattr(self, key))
 406
 407     extra_reg_in1 = property(_functools.partial(extra_reg, key="in1"))
 408     extra_reg_in2 = property(_functools.partial(extra_reg, key="in2"))
 409     extra_reg_in3 = property(_functools.partial(extra_reg, key="in3"))
 410     extra_reg_out = property(_functools.partial(extra_reg, key="out"))
 411     extra_reg_out2 = property(_functools.partial(extra_reg, key="out2"))
 412     extra_reg_cr_in = property(_functools.partial(extra_reg, key="cr_in"))
 413     extra_reg_cr_out = property(_functools.partial(extra_reg, key="cr_out"))
 414
 415
 416 class BitSel:
 417     def __init__(self, value=(0, 32)):
 418         if isinstance(value, str):
 419             (start, end) = map(int, value.split(":"))
 420         else:
 421             (start, end) = value
 422         if start < 0 or end < 0 or start >= end:
 423             raise ValueError(value)
 424
 425         self.__start = start
 426         self.__end = end
 427
 428         return super().__init__()
 429
 430     def __repr__(self):
 431         return f"[{self.__start}:{self.__end}]"
 432
 433     def __iter__(self):
 434         yield from range(self.start, (self.end + 1))
 435
 436     def __reversed__(self):
 437         return tuple(reversed(tuple(self)))
 438
 439     @property
 440     def start(self):
 441         return self.__start
 442
 443     @property
 444     def end(self):
 445         return self.__end
 446
 447
 448 @_dataclasses.dataclass(eq=True, frozen=True)
 449 class Section:
 450     class Mode(_enum.Enum):
 451         INTEGER = _enum.auto()
 452         PATTERN = _enum.auto()
 453
 454         @classmethod
 455         def _missing_(cls, value):
 456             if isinstance(value, str):
 457                 return cls[value.upper()]
 458             return super()._missing_(value)
 459
 460     class Suffix(int):
 461         def __new__(cls, value=None):
 462             if isinstance(value, str):
 463                 if value.upper() == "NONE":
 464                     value = None
 465                 else:
 466                     value = int(value, 0)
 467             if value is None:
 468                 value = 0
 469
 470             return super().__new__(cls, value)
 471
 472         def __str__(self):
 473             return repr(self)
 474
 475         def __repr__(self):
 476             return (bin(self) if self else "None")
 477
 478     path: _pathlib.Path
 479     bitsel: BitSel
 480     suffix: Suffix
 481     mode: Mode
 482     opcode: IntegerOpcode = None
 483
 484     @classmethod
 485     def CSV(cls, record):
 486         typemap = {field.name:field.type for field in _dataclasses.fields(cls)}
 487         if record["opcode"] == "NONE":
 488             typemap["opcode"] = lambda _: None
 489
 490         return dataclass(cls, record, typemap=typemap)
 491
 492
 493 class Fields:
 494     def __init__(self, items):
 495         if isinstance(items, dict):
 496             items = items.items()
 497
 498         def transform(item):
 499             (name, bitrange) = item
 500             return (name, tuple(bitrange.values()))
 501
 502         self.__mapping = dict(map(transform, items))
 503
 504         return super().__init__()
 505
 506     def __repr__(self):
 507         return repr(self.__mapping)
 508
 509     def __iter__(self):
 510         yield from self.__mapping.items()
 511
 512     def __contains__(self, key):
 513         return self.__mapping.__contains__(key)
 514
 515     def __getitem__(self, key):
 516         return self.__mapping.get(key, None)
 517
 518
 519 @_dataclasses.dataclass(eq=True, frozen=True)
 520 class Operand:
 521     name: str
 522
 523     def span(self, record):
 524         return record.fields[self.name]
 525
 526     def disassemble(self, insn, record,
 527             verbosity=Verbosity.NORMAL, indent=""):
 528         raise NotImplementedError
 529
 530
 531 class DynamicOperand(Operand):
 532     def disassemble(self, insn, record,
 533             verbosity=Verbosity.NORMAL, indent=""):
 534         span = self.span(record=record)
 535         if isinstance(insn, SVP64Instruction):
 536             span = tuple(map(lambda bit: (bit + 32), span))
 537         value = insn[span]
 538
 539         if verbosity >= Verbosity.VERBOSE:
 540             span = map(str, span)
 541             yield f"{indent}{self.name}"
 542             yield f"{indent}{indent}{int(value):0{value.bits}b}"
 543             yield f"{indent}{indent}{', '.join(span)}"
 544         else:
 545             yield str(int(value))
 546
 547
 548 class SignedOperand(DynamicOperand):
 549     def disassemble(self, insn, record,
 550             verbosity=Verbosity.NORMAL, indent=""):
 551         span = self.span(record=record)
 552         if isinstance(insn, SVP64Instruction):
 553             span = tuple(map(lambda bit: (bit + 32), span))
 554         value = insn[span]
 555
 556         if verbosity >= Verbosity.VERBOSE:
 557             span = map(str, span)
 558             yield f"{indent}{self.name}"
 559             yield f"{indent}{indent}{int(value):0{value.bits}b}"
 560             yield f"{indent}{indent}{', '.join(span)}"
 561         else:
 562             yield str(value.to_signed_int())
 563
 564
 565 @_dataclasses.dataclass(eq=True, frozen=True)
 566 class StaticOperand(Operand):
 567     value: int
 568
 569     def disassemble(self, insn, record,
 570             verbosity=Verbosity.NORMAL, indent=""):
 571         span = self.span(record=record)
 572         if isinstance(insn, SVP64Instruction):
 573             span = tuple(map(lambda bit: (bit + 32), span))
 574         value = insn[span]
 575
 576         if verbosity >= Verbosity.VERBOSE:
 577             span = map(str, span)
 578             yield f"{indent}{self.name}"
 579             yield f"{indent}{indent}{int(value):0{value.bits}b}"
 580             yield f"{indent}{indent}{', '.join(span)}"
 581         else:
 582             yield str(int(value))
 583
 584
 585 class ImmediateOperand(DynamicOperand):
 586     pass
 587
 588
 589 class NonZeroOperand(DynamicOperand):
 590     def disassemble(self, insn, record,
 591             verbosity=Verbosity.NORMAL, indent=""):
 592         span = self.span(record=record)
 593         if isinstance(insn, SVP64Instruction):
 594             span = tuple(map(lambda bit: (bit + 32), span))
 595         value = insn[span]
 596
 597         if verbosity >= Verbosity.VERBOSE:
 598             span = map(str, span)
 599             yield f"{indent}{self.name}"
 600             yield f"{indent}{indent}{int(value):0{value.bits}b}"
 601             yield f"{indent}{indent}{', '.join(span)}"
 602         else:
 603             yield str(int(value) + 1)
 604
 605
 606 class RegisterOperand(DynamicOperand):
 607     def sv_spec_enter(self, value, span):
 608         return (value, span)
 609
 610     def sv_spec_leave(self, value, span, origin_value, origin_span):
 611         return (value, span)
 612
 613     def spec(self, insn, record):
 614         vector = False
 615         span = self.span(record=record)
 616         if isinstance(insn, SVP64Instruction):
 617             span = tuple(map(lambda bit: (bit + 32), span))
 618         value = insn[span]
 619         span = tuple(map(str, span))
 620
 621         if isinstance(insn, SVP64Instruction):
 622             (origin_value, origin_span) = (value, span)
 623             (value, span) = self.sv_spec_enter(value=value, span=span)
 624
 625             extra_idx = self.extra_idx(record=record)
 626             if extra_idx is _SVExtra.NONE:
 627                 return (vector, value, span)
 628
 629             if record.etype is _SVEtype.EXTRA3:
 630                 spec = insn.prefix.rm.extra3[extra_idx]
 631             elif record.etype is _SVEtype.EXTRA2:
 632                 spec = insn.prefix.rm.extra2[extra_idx]
 633             else:
 634                 raise ValueError(record.etype)
 635
 636             if spec != 0:
 637                 vector = bool(spec[0])
 638                 spec_span = spec.__class__
 639                 if record.etype is _SVEtype.EXTRA3:
 640                     spec_span = tuple(map(str, spec_span[1, 2]))
 641                     spec = spec[1, 2]
 642                 elif record.etype is _SVEtype.EXTRA2:
 643                     spec_span = tuple(map(str, spec_span[1,]))
 644                     spec = _SelectableInt(value=spec[1].value, bits=2)
 645                     if vector:
 646                         spec <<= 1
 647                         spec_span = (spec_span + ("{0}",))
 648                     else:
 649                         spec_span = (("{0}",) + spec_span)
 650                 else:
 651                     raise ValueError(record.etype)
 652
 653                 vector_shift = (2 + (5 - value.bits))
 654                 scalar_shift = value.bits
 655                 spec_shift = (5 - value.bits)
 656
 657                 bits = (len(span) + len(spec_span))
 658                 value = _SelectableInt(value=value.value, bits=bits)
 659                 spec = _SelectableInt(value=spec.value, bits=bits)
 660                 if vector:
 661                     value = ((value << vector_shift) | (spec << spec_shift))
 662                     span = (span + spec_span + ((spec_shift * ('{0}',))))
 663                 else:
 664                     value = ((spec << scalar_shift) | value)
 665                     span = ((spec_shift * ('{0}',)) + spec_span + span)
 666
 667             (value, span) = self.sv_spec_leave(value=value, span=span,
 668                 origin_value=origin_value, origin_span=origin_span)
 669
 670         return (vector, value, span)
 671
 672     @property
 673     def extra_reg(self):
 674         return _SVExtraReg(self.name)
 675
 676     def extra_idx(self, record):
 677         for key in frozenset({
 678                     "in1", "in2", "in3", "cr_in", "cr_in2",
 679                     "out", "out2", "cr_out",
 680                 }):
 681             extra_reg = record.svp64.extra_reg(key=key)
 682             if extra_reg is self.extra_reg:
 683                 return record.extra_idx(key=key)
 684
 685         return _SVExtra.NONE
 686
 687     def disassemble(self, insn, record,
 688             verbosity=Verbosity.NORMAL, prefix="", indent=""):
 689         (vector, value, span) = self.spec(insn=insn, record=record)
 690
 691         if verbosity >= Verbosity.VERBOSE:
 692             mode = "vector" if vector else "scalar"
 693             yield f"{indent}{self.name} ({mode})"
 694             yield f"{indent}{indent}{int(value):0{value.bits}b}"
 695             yield f"{indent}{indent}{', '.join(span)}"
 696             if isinstance(insn, SVP64Instruction):
 697                 extra_idx = self.extra_idx(record)
 698                 if record.etype is _SVEtype.NONE:
 699                     yield f"{indent}{indent}extra[none]"
 700                 else:
 701                     etype = repr(record.etype).lower()
 702                     yield f"{indent}{indent}{etype}{extra_idx!r}"
 703         else:
 704             vector = "*" if vector else ""
 705             yield f"{vector}{prefix}{int(value)}"
 706
 707
 708 class GPROperand(RegisterOperand):
 709     def disassemble(self, insn, record,
 710             verbosity=Verbosity.NORMAL, indent=""):
 711         prefix = "" if (verbosity <= Verbosity.SHORT) else "r"
 712         yield from super().disassemble(prefix=prefix,
 713             insn=insn, record=record,
 714             verbosity=verbosity, indent=indent)
 715
 716
 717 class FPROperand(RegisterOperand):
 718     def disassemble(self, insn, record,
 719             verbosity=Verbosity.NORMAL, indent=""):
 720         prefix = "" if (verbosity <= Verbosity.SHORT) else "f"
 721         yield from super().disassemble(prefix=prefix,
 722             insn=insn, record=record,
 723             verbosity=verbosity, indent=indent)
 724
 725
 726 class CR3Operand(RegisterOperand):
 727     pass
 728
 729
 730 class CR5Operand(RegisterOperand):
 731     def sv_spec_enter(self, value, span):
 732         value = _SelectableInt(value=(value.value >> 2), bits=3)
 733         return (value, span)
 734
 735     def sv_spec_leave(self, value, span, origin_value, origin_span):
 736         value = _selectconcat(value, origin_value[3:5])
 737         span += origin_span
 738         return (value, span)
 739
 740
 741 class TargetAddrOperand(RegisterOperand):
 742     def disassemble(self, insn, record, field,
 743             verbosity=Verbosity.NORMAL, indent=""):
 744         span = self.span(record=record)
 745         if isinstance(insn, SVP64Instruction):
 746             span = tuple(map(lambda bit: (bit + 32), span))
 747         value = insn[span]
 748
 749         if verbosity >= Verbosity.VERBOSE:
 750             span = tuple(map(str, span))
 751             yield f"{indent}{self.name} = EXTS({field} || 0b00))"
 752             yield f"{indent}{indent}{field}"
 753             yield f"{indent}{indent}{indent}{int(value):0{value.bits}b}00"
 754             yield f"{indent}{indent}{indent}{', '.join(span + ('{0}', '{0}'))}"
 755         else:
 756             yield hex(_selectconcat(value,
 757                 _SelectableInt(value=0b00, bits=2)).to_signed_int())
 758
 759
 760 class TargetAddrOperandLI(TargetAddrOperand):
 761     def span(self, record):
 762         return record.fields["LI"]
 763
 764     def disassemble(self, insn, record,
 765             verbosity=Verbosity.NORMAL, indent=""):
 766         return super().disassemble(field="LI",
 767             insn=insn, record=record,
 768             verbosity=verbosity, indent=indent)
 769
 770
 771 class TargetAddrOperandBD(TargetAddrOperand):
 772     def span(self, record):
 773         return record.fields["BD"]
 774
 775     def disassemble(self, insn, record,
 776             verbosity=Verbosity.NORMAL, indent=""):
 777         return super().disassemble(field="BD",
 778             insn=insn, record=record,
 779             verbosity=verbosity, indent=indent)
 780
 781
 782 class DOperandDX(SignedOperand):
 783     def span(self, record):
 784         operands = map(DynamicOperand, ("d0", "d1", "d2"))
 785         spans = map(lambda operand: operand.span(record=record), operands)
 786         return sum(spans, tuple())
 787
 788     def disassemble(self, insn, record,
 789             verbosity=Verbosity.NORMAL, indent=""):
 790         span = self.span(record=record)
 791         if isinstance(insn, SVP64Instruction):
 792             span = tuple(map(lambda bit: (bit + 32), span))
 793         value = insn[span]
 794
 795         if verbosity >= Verbosity.VERBOSE:
 796             yield f"{indent}D"
 797             mapping = {
 798                 "d0": "[0:9]",
 799                 "d1": "[10:15]",
 800                 "d2": "[16]",
 801             }
 802             for (subname, subspan) in mapping.items():
 803                 operand = DynamicOperand(name=subname)
 804                 span = operand.span(record=record)
 805                 if isinstance(insn, SVP64Instruction):
 806                     span = tuple(map(lambda bit: (bit + 32), span))
 807                 value = insn[span]
 808                 span = map(str, span)
 809                 yield f"{indent}{indent}{operand.name} = D{subspan}"
 810                 yield f"{indent}{indent}{indent}{int(value):0{value.bits}b}"
 811                 yield f"{indent}{indent}{indent}{', '.join(span)}"
 812         else:
 813             yield str(value.to_signed_int())
 814
 815
 816 class Operands(tuple):
 817     def __new__(cls, insn, iterable):
 818         custom_insns = {
 819             "b": {"target_addr": TargetAddrOperandLI},
 820             "ba": {"target_addr": TargetAddrOperandLI},
 821             "bl": {"target_addr": TargetAddrOperandLI},
 822             "bla": {"target_addr": TargetAddrOperandLI},
 823             "bc": {"target_addr": TargetAddrOperandBD},
 824             "bca": {"target_addr": TargetAddrOperandBD},
 825             "bcl": {"target_addr": TargetAddrOperandBD},
 826             "bcla": {"target_addr": TargetAddrOperandBD},
 827             "addpcis": {"D": DOperandDX},
 828             "fishmv": {"D": DOperandDX},
 829             "fmvis": {"D": DOperandDX},
 830         }
 831         custom_fields = {
 832             "SVi": NonZeroOperand,
 833             "SVd": NonZeroOperand,
 834             "SVxd": NonZeroOperand,
 835             "SVyd": NonZeroOperand,
 836             "SVzd": NonZeroOperand,
 837             "BD": SignedOperand,
 838             "D": SignedOperand,
 839             "DQ": SignedOperand,
 840             "DS": SignedOperand,
 841             "SI": SignedOperand,
 842             "IB": SignedOperand,
 843             "LI": SignedOperand,
 844             "SIM": SignedOperand,
 845             "SVD": SignedOperand,
 846             "SVDS": SignedOperand,
 847         }
 848
 849         operands = []
 850         for operand in iterable:
 851             dynamic_cls = DynamicOperand
 852             static_cls = StaticOperand
 853
 854             if "=" in operand:
 855                 (name, value) = operand.split("=")
 856                 operand = static_cls(name=name, value=int(value))
 857                 operands.append(operand)
 858             else:
 859                 if operand.endswith(")"):
 860                     operand = operand.replace("(", " ").replace(")", "")
 861                     (immediate, _, operand) = operand.partition(" ")
 862                 else:
 863                     immediate = None
 864
 865                 if immediate is not None:
 866                     operands.append(ImmediateOperand(name=immediate))
 867
 868                 if operand in custom_fields:
 869                     dynamic_cls = custom_fields[operand]
 870                 if insn in custom_insns and operand in custom_insns[insn]:
 871                     dynamic_cls = custom_insns[insn][operand]
 872
 873                 if operand in _RegType.__members__:
 874                     regtype = _RegType[operand]
 875                     if regtype is _RegType.GPR:
 876                         dynamic_cls = GPROperand
 877                     elif regtype is _RegType.FPR:
 878                         dynamic_cls = FPROperand
 879                     if regtype is _RegType.CR_BIT: # 5-bit
 880                         dynamic_cls = CR5Operand
 881                     if regtype is _RegType.CR_REG: # actually CR Field, 3-bit
 882                         dynamic_cls = CR3Operand
 883
 884                 operand = dynamic_cls(name=operand)
 885                 operands.append(operand)
 886
 887         return super().__new__(cls, operands)
 888
 889     def __contains__(self, key):
 890         return self.__getitem__(key) is not None
 891
 892     def __getitem__(self, key):
 893         for operand in self:
 894             if operand.name == key:
 895                 return operand
 896
 897         return None
 898
 899     @property
 900     def dynamic(self):
 901         for operand in self:
 902             if isinstance(operand, DynamicOperand):
 903                 yield operand
 904
 905     @property
 906     def static(self):
 907         for operand in self:
 908             if isinstance(operand, StaticOperand):
 909                 yield operand
 910
 911
 912 class PCode:
 913     def __init__(self, iterable):
 914         self.__pcode = tuple(iterable)
 915         return super().__init__()
 916
 917     def __iter__(self):
 918         yield from self.__pcode
 919
 920     def __repr__(self):
 921         return self.__pcode.__repr__()
 922
 923
 924 @_dataclasses.dataclass(eq=True, frozen=True)
 925 class MarkdownRecord:
 926     pcode: PCode
 927     operands: Operands
 928
 929
 930 @_functools.total_ordering
 931 @_dataclasses.dataclass(eq=True, frozen=True)
 932 class Record:
 933     name: str
 934     section: Section
 935     ppc: PPCRecord
 936     fields: Fields
 937     mdwn: MarkdownRecord
 938     svp64: SVP64Record = None
 939
 940     def __lt__(self, other):
 941         if not isinstance(other, Record):
 942             return NotImplemented
 943         return (min(self.opcodes) < min(other.opcodes))
 944
 945     @property
 946     def opcodes(self):
 947         def opcode(ppc):
 948             value = ([0] * 32)
 949             mask = ([0] * 32)
 950
 951             PO = self.section.opcode
 952             if PO is not None:
 953                 for (src, dst) in enumerate(reversed(BitSel((0, 5)))):
 954                     value[dst] = int((PO.value & (1 << src)) != 0)
 955                     mask[dst] = int((PO.mask & (1 << src)) != 0)
 956
 957             XO = ppc.opcode
 958             for (src, dst) in enumerate(reversed(self.section.bitsel)):
 959                 value[dst] = int((XO.value & (1 << src)) != 0)
 960                 mask[dst] = int((XO.mask & (1 << src)) != 0)
 961
 962             for operand in self.mdwn.operands.static:
 963                 for (src, dst) in enumerate(reversed(operand.span(record=self))):
 964                     value[dst] = int((operand.value & (1 << src)) != 0)
 965                     mask[dst] = 1
 966
 967             value = Opcode.Value(int(("".join(map(str, value))), 2))
 968             mask = Opcode.Mask(int(("".join(map(str, mask))), 2))
 969
 970             return Opcode(value=value, mask=mask)
 971
 972         return tuple(sorted(map(opcode, self.ppc)))
 973
 974     def match(self, key):
 975         for opcode in self.opcodes:
 976             if ((opcode.value & opcode.mask) ==
 977                     (key & opcode.mask)):
 978                 return True
 979         return False
 980
 981     @property
 982     def function(self):
 983         return self.ppc.function
 984
 985     @property
 986     def in1(self):
 987         return self.ppc.in1
 988
 989     @property
 990     def in2(self):
 991         return self.ppc.in2
 992
 993     @property
 994     def in3(self):
 995         return self.ppc.in3
 996
 997     @property
 998     def out(self):
 999         return self.ppc.out
1000
1001     @property
1002     def out2(self):
1003         if self.svp64 is None:
1004             return _OutSel.NONE
1005         return self.ppc.out
1006
1007     @property
1008     def cr_in(self):
1009         return self.ppc.cr_in
1010
1011     @property
1012     def cr_in2(self):
1013         return self.ppc.cr_in2
1014
1015     @property
1016     def cr_out(self):
1017         return self.ppc.cr_out
1018
1019     ptype = property(lambda self: self.svp64.ptype)
1020     etype = property(lambda self: self.svp64.etype)
1021
1022     def extra_idx(self, key):
1023         return self.svp64.extra_idx(key)
1024
1025     extra_idx_in1 = property(lambda self: self.svp64.extra_idx_in1)
1026     extra_idx_in2 = property(lambda self: self.svp64.extra_idx_in2)
1027     extra_idx_in3 = property(lambda self: self.svp64.extra_idx_in3)
1028     extra_idx_out = property(lambda self: self.svp64.extra_idx_out)
1029     extra_idx_out2 = property(lambda self: self.svp64.extra_idx_out2)
1030     extra_idx_cr_in = property(lambda self: self.svp64.extra_idx_cr_in)
1031     extra_idx_cr_out = property(lambda self: self.svp64.extra_idx_cr_out)
1032
1033
1034 class Instruction(_Mapping):
1035     @classmethod
1036     def integer(cls, value=0, bits=None, byteorder="little"):
1037         if isinstance(value, (int, bytes)) and not isinstance(bits, int):
1038             raise ValueError(bits)
1039
1040         if isinstance(value, bytes):
1041             if ((len(value) * 8) != bits):
1042                 raise ValueError(f"bit length mismatch")
1043             value = int.from_bytes(value, byteorder=byteorder)
1044
1045         if isinstance(value, int):
1046             value = _SelectableInt(value=value, bits=bits)
1047         elif isinstance(value, Instruction):
1048             value = value.storage
1049
1050         if not isinstance(value, _SelectableInt):
1051             raise ValueError(value)
1052         if bits is None:
1053             bits = len(cls)
1054         if len(value) != bits:
1055             raise ValueError(value)
1056
1057         value = _SelectableInt(value=value, bits=bits)
1058
1059         return cls(storage=value)
1060
1061     def __hash__(self):
1062         return hash(int(self))
1063
1064     def __getitem__(self, key):
1065         return self.storage.__getitem__(key)
1066
1067     def __setitem__(self, key, value):
1068         return self.storage.__setitem__(key, value)
1069
1070     def bytes(self, byteorder="little"):
1071         nr_bytes = (self.storage.bits // 8)
1072         return int(self).to_bytes(nr_bytes, byteorder=byteorder)
1073
1074     def record(self, db):
1075         record = db[self]
1076         if record is None:
1077             raise KeyError(self)
1078         return record
1079
1080     def spec(self, db, prefix):
1081         record = self.record(db=db)
1082
1083         dynamic_operands = tuple(map(_operator.itemgetter(0),
1084             self.dynamic_operands(db=db)))
1085
1086         static_operands = []
1087         for (name, value) in self.static_operands(db=db):
1088             static_operands.append(f"{name}={value}")
1089
1090         operands = ""
1091         if dynamic_operands:
1092             operands += f" {','.join(dynamic_operands)}"
1093         if static_operands:
1094             operands += f" ({' '.join(static_operands)})"
1095
1096         return f"{prefix}{record.name}{operands}"
1097
1098     def dynamic_operands(self, db, verbosity=Verbosity.NORMAL):
1099         record = self.record(db=db)
1100
1101         imm = False
1102         imm_name = ""
1103         imm_value = ""
1104         for operand in record.mdwn.operands.dynamic:
1105             name = operand.name
1106             dis = operand.disassemble(insn=self, record=record,
1107                 verbosity=min(verbosity, Verbosity.NORMAL))
1108             value = " ".join(dis)
1109             if imm:
1110                 name = f"{imm_name}({name})"
1111                 value = f"{imm_value}({value})"
1112                 imm = False
1113             if isinstance(operand, ImmediateOperand):
1114                 imm_name = name
1115                 imm_value = value
1116                 imm = True
1117             if not imm:
1118                 yield (name, value)
1119
1120     def static_operands(self, db):
1121         record = self.record(db=db)
1122         for operand in record.mdwn.operands.static:
1123             yield (operand.name, operand.value)
1124
1125     def disassemble(self, db,
1126             byteorder="little",
1127             verbosity=Verbosity.NORMAL):
1128         raise NotImplementedError
1129
1130
1131 class WordInstruction(Instruction):
1132     _: _Field = range(0, 32)
1133     po: _Field = range(0, 6)
1134
1135     @classmethod
1136     def integer(cls, value, byteorder="little"):
1137         return super().integer(bits=32, value=value, byteorder=byteorder)
1138
1139     @property
1140     def binary(self):
1141         bits = []
1142         for idx in range(32):
1143             bit = int(self[idx])
1144             bits.append(bit)
1145         return "".join(map(str, bits))
1146
1147     def disassemble(self, db,
1148             byteorder="little",
1149             verbosity=Verbosity.NORMAL):
1150         integer = int(self)
1151         if verbosity <= Verbosity.SHORT:
1152             blob = ""
1153         else:
1154             blob = integer.to_bytes(length=4, byteorder=byteorder)
1155             blob = " ".join(map(lambda byte: f"{byte:02x}", blob))
1156             blob += "    "
1157
1158         record = db[self]
1159         if record is None:
1160             yield f"{blob}.long 0x{integer:08x}"
1161             return
1162
1163         operands = tuple(map(_operator.itemgetter(1),
1164             self.dynamic_operands(db=db, verbosity=verbosity)))
1165         if operands:
1166             yield f"{blob}{record.name} {','.join(operands)}"
1167         else:
1168             yield f"{blob}{record.name}"
1169
1170         if verbosity >= Verbosity.VERBOSE:
1171             indent = (" " * 4)
1172             binary = self.binary
1173             spec = self.spec(db=db, prefix="")
1174             yield f"{indent}spec"
1175             yield f"{indent}{indent}{spec}"
1176             yield f"{indent}pcode"
1177             for stmt in record.mdwn.pcode:
1178                 yield f"{indent}{indent}{stmt}"
1179             yield f"{indent}binary"
1180             yield f"{indent}{indent}[0:8]   {binary[0:8]}"
1181             yield f"{indent}{indent}[8:16]  {binary[8:16]}"
1182             yield f"{indent}{indent}[16:24] {binary[16:24]}"
1183             yield f"{indent}{indent}[24:32] {binary[24:32]}"
1184             yield f"{indent}opcodes"
1185             for opcode in record.opcodes:
1186                 yield f"{indent}{indent}{opcode!r}"
1187             for operand in record.mdwn.operands:
1188                 yield from operand.disassemble(insn=self, record=record,
1189                     verbosity=verbosity, indent=indent)
1190             yield ""
1191
1192
1193 class PrefixedInstruction(Instruction):
1194     class Prefix(WordInstruction.remap(range(0, 32))):
1195         pass
1196
1197     class Suffix(WordInstruction.remap(range(32, 64))):
1198         pass
1199
1200     _: _Field = range(64)
1201     prefix: Prefix
1202     suffix: Suffix
1203     po: Suffix.po
1204
1205     @classmethod
1206     def integer(cls, value, byteorder="little"):
1207         return super().integer(bits=64, value=value, byteorder=byteorder)
1208
1209     @classmethod
1210     def pair(cls, prefix=0, suffix=0, byteorder="little"):
1211         def transform(value):
1212             return WordInstruction.integer(value=value,
1213                 byteorder=byteorder)[0:32]
1214
1215         (prefix, suffix) = map(transform, (prefix, suffix))
1216         value = _selectconcat(prefix, suffix)
1217
1218         return super().integer(bits=64, value=value)
1219
1220
1221 class Mode(_Mapping):
1222     _: _Field = range(0, 5)
1223
1224
1225 class Extra(_Mapping):
1226     _: _Field = range(0, 9)
1227
1228
1229 class Extra2(Extra):
1230     idx0: _Field = range(0, 2)
1231     idx1: _Field = range(2, 4)
1232     idx2: _Field = range(4, 6)
1233     idx3: _Field = range(6, 8)
1234
1235     def __getitem__(self, key):
1236         return {
1237             0: self.idx0,
1238             1: self.idx1,
1239             2: self.idx2,
1240             3: self.idx3,
1241             _SVExtra.Idx0: self.idx0,
1242             _SVExtra.Idx1: self.idx1,
1243             _SVExtra.Idx2: self.idx2,
1244             _SVExtra.Idx3: self.idx3,
1245         }[key]
1246
1247     def __setitem__(self, key, value):
1248         self[key].assign(value)
1249
1250
1251 class Extra3(Extra):
1252     idx0: _Field = range(0, 3)
1253     idx1: _Field = range(3, 6)
1254     idx2: _Field = range(6, 9)
1255
1256     def __getitem__(self, key):
1257         return {
1258             0: self.idx0,
1259             1: self.idx1,
1260             2: self.idx2,
1261             _SVExtra.Idx0: self.idx0,
1262             _SVExtra.Idx1: self.idx1,
1263             _SVExtra.Idx2: self.idx2,
1264         }[key]
1265
1266     def __setitem__(self, key, value):
1267         self[key].assign(value)
1268
1269
1270 class BaseRM(_Mapping):
1271     _: _Field = range(24)
1272     mmode: _Field = (0,)
1273     mask: _Field = range(1, 4)
1274     elwidth: _Field = range(4, 6)
1275     ewsrc: _Field = range(6, 8)
1276     subvl: _Field = range(8, 10)
1277     mode: Mode.remap(range(19, 24))
1278     smask: _Field = range(16, 19)
1279     extra: Extra.remap(range(10, 19))
1280     extra2: Extra2.remap(range(10, 19))
1281     extra3: Extra3.remap(range(10, 19))
1282
1283     def specifiers(self, record):
1284         subvl = int(self.subvl)
1285         if subvl > 0:
1286             yield {
1287                 1: "vec2",
1288                 2: "vec3",
1289                 3: "vec4",
1290             }[subvl]
1291
1292     def disassemble(self, verbosity=Verbosity.NORMAL):
1293         if verbosity >= Verbosity.VERBOSE:
1294             indent = (" " * 4)
1295             for (name, value, members) in self.traverse(path="RM"):
1296                 yield f"{name}"
1297                 yield f"{indent}{int(value):0{value.bits}b}"
1298                 yield f"{indent}{', '.join(map(str, members))}"
1299
1300
1301 class FFPRRc1BaseRM(BaseRM):
1302     def specifiers(self, record, mode):
1303         inv = _SelectableInt(value=int(self.inv), bits=1)
1304         CR = _SelectableInt(value=int(self.CR), bits=2)
1305         mask = int(_selectconcat(inv, CR))
1306         predicate = {
1307             0b000: "lt",
1308             0b001: "ge",
1309             0b010: "gt",
1310             0b011: "le",
1311             0b100: "eq",
1312             0b101: "ne",
1313             0b110: "so",
1314             0b111: "ns",
1315         }[mask]
1316         yield f"{mode}={predicate}"
1317
1318         yield from super().specifiers(record=record)
1319
1320
1321 class FFPRRc0BaseRM(BaseRM):
1322     def specifiers(self, record, mode):
1323         if self.RC1:
1324             inv = "~" if self.inv else ""
1325             yield f"{mode}={inv}RC1"
1326
1327         yield from super().specifiers(record=record)
1328
1329
1330 class SatBaseRM(BaseRM):
1331     def specifiers(self, record):
1332         if self.N:
1333             yield "sats"
1334         else:
1335             yield "satu"
1336
1337         yield from super().specifiers(record=record)
1338
1339
1340 class ZZBaseRM(BaseRM):
1341     def specifiers(self, record):
1342         if self.zz:
1343             yield "zz"
1344
1345         yield from super().specifiers(record=record)
1346
1347
1348 class DZBaseRM(BaseRM):
1349     def specifiers(self, record):
1350         if self.dz:
1351             yield "dz"
1352
1353         yield from super().specifiers(record=record)
1354
1355
1356 class SZBaseRM(BaseRM):
1357     def specifiers(self, record):
1358         if self.sz:
1359             yield "sz"
1360
1361         yield from super().specifiers(record=record)
1362
1363
1364 class MRBaseRM(BaseRM):
1365     def specifiers(self, record):
1366         if self.RG:
1367             yield "mrr"
1368         else:
1369             yield "mr"
1370
1371         yield from super().specifiers(record=record)
1372
1373
1374 class NormalLDSTBaseRM(BaseRM):
1375     def specifiers(self, record):
1376         widths = {
1377             0b11: "8",
1378             0b10: "16",
1379             0b01: "32",
1380         }
1381         predicates = {
1382             # integer
1383             (0, 0b001): "1<<r3",
1384             (0, 0b010): "r3",
1385             (0, 0b011): "~r3",
1386             (0, 0b100): "r10",
1387             (0, 0b101): "~r10",
1388             (0, 0b110): "r30",
1389             (0, 0b111): "~r30",
1390             # CRs
1391             (1, 0b000): "lt",
1392             (1, 0b001): "ge",
1393             (1, 0b010): "gt",
1394             (1, 0b011): "le",
1395             (1, 0b100): "eq",
1396             (1, 0b101): "ne",
1397             (1, 0b110): "so",
1398             (1, 0b111): "ns",
1399         }
1400
1401         mmode = int(self.mmode)
1402         mask = int(self.mask)
1403         if record.svp64.ptype is _SVPtype.P2:
1404             (smask, dmask) = (int(self.smask), mask)
1405         else:
1406             (smask, dmask) = (mask, mask)
1407         if all((smask, dmask)) and (smask == dmask):
1408             yield f"m={predicates[(mmode, smask)]}"
1409         else:
1410             sw = predicates.get((mmode, smask))
1411             dw = predicates.get((mmode, dmask))
1412             if sw:
1413                 yield f"sm={sw}"
1414             if dw:
1415                 yield f"dm={dw}"
1416
1417         dw = int(self.elwidth)
1418         sw = int(self.ewsrc)
1419         if all((dw, sw)) and (dw == sw):
1420             yield f"w={widths[dw]}"
1421         else:
1422             if dw != 0b00:
1423                 yield f"dw={widths[dw]}"
1424             if sw != 0b00:
1425                 yield f"sw={widths[sw]}"
1426
1427         yield from super().specifiers(record=record)
1428
1429
1430 # ********************
1431 # Normal mode
1432 # https://libre-soc.org/openpower/sv/normal/
1433 class NormalBaseRM(NormalLDSTBaseRM):
1434     pass
1435
1436
1437 class NormalSimpleRM(DZBaseRM, SZBaseRM, NormalBaseRM):
1438     """normal: simple mode"""
1439     dz: BaseRM.mode[3]
1440     sz: BaseRM.mode[4]
1441
1442     def specifiers(self, record):
1443         yield from super().specifiers(record=record)
1444
1445
1446 class NormalSMRRM(MRBaseRM, NormalBaseRM):
1447     """normal: scalar reduce mode (mapreduce), SUBVL=1"""
1448     RG: BaseRM.mode[4]
1449
1450
1451 class NormalReservedRM(NormalBaseRM):
1452     """normal: reserved"""
1453     pass
1454
1455
1456 class NormalFFRc1RM(FFPRRc1BaseRM, NormalBaseRM):
1457     """normal: Rc=1: ffirst CR sel"""
1458     inv: BaseRM.mode[2]
1459     CR: BaseRM.mode[3, 4]
1460
1461     def specifiers(self, record):
1462         yield from super().specifiers(record=record, mode="ff")
1463
1464
1465 class NormalFFRc0RM(FFPRRc0BaseRM, NormalBaseRM):
1466     """normal: Rc=0: ffirst z/nonz"""
1467     inv: BaseRM.mode[2]
1468     VLi: BaseRM.mode[3]
1469     RC1: BaseRM.mode[4]
1470
1471     def specifiers(self, record):
1472         yield from super().specifiers(record=record, mode="ff")
1473
1474
1475 class NormalSatRM(SatBaseRM, DZBaseRM, SZBaseRM, NormalBaseRM):
1476     """normal: sat mode: N=0/1 u/s, SUBVL=1"""
1477     N: BaseRM.mode[2]
1478     dz: BaseRM.mode[3]
1479     sz: BaseRM.mode[4]
1480
1481
1482 class NormalPRRc1RM(FFPRRc1BaseRM, NormalBaseRM):
1483     """normal: Rc=1: pred-result CR sel"""
1484     inv: BaseRM.mode[2]
1485     CR: BaseRM.mode[3, 4]
1486
1487     def specifiers(self, record):
1488         yield from super().specifiers(record=record, mode="pr")
1489
1490
1491 class NormalPRRc0RM(FFPRRc0BaseRM, ZZBaseRM, NormalBaseRM):
1492     """normal: Rc=0: pred-result z/nonz"""
1493     inv: BaseRM.mode[2]
1494     zz: BaseRM.mode[3]
1495     RC1: BaseRM.mode[4]
1496     dz: BaseRM.mode[3]
1497     sz: BaseRM.mode[3]
1498
1499     def specifiers(self, record):
1500         yield from super().specifiers(record=record, mode="pr")
1501
1502
1503 class NormalRM(NormalBaseRM):
1504     simple: NormalSimpleRM
1505     smr: NormalSMRRM
1506     reserved: NormalReservedRM
1507     ffrc1: NormalFFRc1RM
1508     ffrc0: NormalFFRc0RM
1509     sat: NormalSatRM
1510     prrc1: NormalPRRc1RM
1511     prrc0: NormalPRRc0RM
1512
1513
1514 # ********************
1515 # LD/ST Immediate mode
1516 # https://libre-soc.org/openpower/sv/ldst/
1517
1518 class LDSTImmBaseRM(NormalLDSTBaseRM):
1519     pass
1520
1521
1522 class LDSTImmSimpleRM(ZZBaseRM, LDSTImmBaseRM):
1523     """ld/st immediate: simple mode"""
1524     zz: BaseRM.mode[3]
1525     els: BaseRM.mode[4]
1526     dz: BaseRM.mode[3]
1527     sz: BaseRM.mode[3]
1528
1529
1530 class LDSTImmReservedRM(LDSTImmBaseRM):
1531     """ld/st immediate: reserved"""
1532     pass
1533
1534
1535 class LDSTImmFFRc1RM(FFPRRc1BaseRM, LDSTImmBaseRM):
1536     """ld/st immediate: Rc=1: ffirst CR sel"""
1537     inv: BaseRM.mode[2]
1538     CR: BaseRM.mode[3, 4]
1539
1540     def specifiers(self, record):
1541         yield from super().specifiers(record=record, mode="ff")
1542
1543
1544 class LDSTImmFFRc0RM(FFPRRc0BaseRM, LDSTImmBaseRM):
1545     """ld/st immediate: Rc=0: ffirst z/nonz"""
1546     inv: BaseRM.mode[2]
1547     els: BaseRM.mode[3]
1548     RC1: BaseRM.mode[4]
1549
1550     def specifiers(self, record):
1551         yield from super().specifiers(record=record, mode="ff")
1552
1553
1554 class LDSTImmSatRM(SatBaseRM, ZZBaseRM, LDSTImmBaseRM):
1555     """ld/st immediate: sat mode: N=0/1 u/s"""
1556     N: BaseRM.mode[2]
1557     zz: BaseRM.mode[3]
1558     els: BaseRM.mode[4]
1559     dz: BaseRM.mode[3]
1560     sz: BaseRM.mode[3]
1561
1562
1563 class LDSTImmPRRc1RM(FFPRRc1BaseRM, LDSTImmBaseRM):
1564     """ld/st immediate: Rc=1: pred-result CR sel"""
1565     inv: BaseRM.mode[2]
1566     CR: BaseRM.mode[3, 4]
1567
1568     def specifiers(self, record):
1569         yield from super().specifiers(record=record, mode="pr")
1570
1571
1572 class LDSTImmPRRc0RM(FFPRRc0BaseRM, LDSTImmBaseRM):
1573     """ld/st immediate: Rc=0: pred-result z/nonz"""
1574     inv: BaseRM.mode[2]
1575     els: BaseRM.mode[3]
1576     RC1: BaseRM.mode[4]
1577
1578     def specifiers(self, record):
1579         yield from super().specifiers(record=record, mode="pr")
1580
1581
1582 class LDSTImmRM(LDSTImmBaseRM):
1583     simple: LDSTImmSimpleRM
1584     reserved: LDSTImmReservedRM
1585     ffrc1: LDSTImmFFRc1RM
1586     ffrc0: LDSTImmFFRc0RM
1587     sat: LDSTImmSatRM
1588     prrc1: LDSTImmPRRc1RM
1589     prrc0: LDSTImmPRRc0RM
1590
1591
1592 # ********************
1593 # LD/ST Indexed mode
1594 # https://libre-soc.org/openpower/sv/ldst/
1595
1596 class LDSTIdxBaseRM(NormalLDSTBaseRM):
1597     pass
1598
1599
1600 class LDSTIdxSimpleRM(DZBaseRM, SZBaseRM, LDSTIdxBaseRM):
1601     """ld/st index: simple mode"""
1602     SEA: BaseRM.mode[2]
1603     dz: BaseRM.mode[3]
1604     sz: BaseRM.mode[4]
1605
1606
1607 class LDSTIdxStrideRM(DZBaseRM, SZBaseRM, LDSTIdxBaseRM):
1608     """ld/st index: strided (scalar only source)"""
1609     SEA: BaseRM.mode[2]
1610     dz: BaseRM.mode[3]
1611     sz: BaseRM.mode[4]
1612
1613
1614 class LDSTIdxSatRM(SatBaseRM, DZBaseRM, SZBaseRM, LDSTIdxBaseRM):
1615     """ld/st index: sat mode: N=0/1 u/s"""
1616     N: BaseRM.mode[2]
1617     dz: BaseRM.mode[3]
1618     sz: BaseRM.mode[4]
1619
1620
1621 class LDSTIdxPRRc1RM(LDSTIdxBaseRM):
1622     """ld/st index: Rc=1: pred-result CR sel"""
1623     inv: BaseRM.mode[2]
1624     CR: BaseRM.mode[3, 4]
1625
1626     def specifiers(self, record):
1627         yield from super().specifiers(record=record, mode="pr")
1628
1629
1630 class LDSTIdxPRRc0RM(FFPRRc0BaseRM, ZZBaseRM, LDSTIdxBaseRM):
1631     """ld/st index: Rc=0: pred-result z/nonz"""
1632     inv: BaseRM.mode[2]
1633     zz: BaseRM.mode[3]
1634     RC1: BaseRM.mode[4]
1635     dz: BaseRM.mode[3]
1636     sz: BaseRM.mode[3]
1637
1638     def specifiers(self, record):
1639         yield from super().specifiers(record=record, mode="pr")
1640
1641
1642 class LDSTIdxRM(LDSTIdxBaseRM):
1643     simple: LDSTIdxSimpleRM
1644     stride: LDSTIdxStrideRM
1645     sat: LDSTIdxSatRM
1646     prrc1: LDSTIdxPRRc1RM
1647     prrc0: LDSTIdxPRRc0RM
1648
1649
1650
1651 # ********************
1652 # CR ops mode
1653 # https://libre-soc.org/openpower/sv/cr_ops/
1654 class CROpBaseRM(BaseRM):
1655     SNZ: BaseRM[7]
1656
1657
1658 class CROpSimpleRM(DZBaseRM, SZBaseRM, CROpBaseRM):
1659     """cr_op: simple mode"""
1660     RG: BaseRM[20]
1661     dz: BaseRM[22]
1662     sz: BaseRM[23]
1663
1664     def specifiers(self, record):
1665         if self.RG:
1666             yield "rg" # simple CR Mode reports /rg
1667
1668         yield from super().specifiers(record=record)
1669
1670 class CROpSMRRM(MRBaseRM, DZBaseRM, SZBaseRM, CROpBaseRM):
1671     """cr_op: scalar reduce mode (mapreduce), SUBVL=1"""
1672     RG: BaseRM[20]
1673     dz: BaseRM[22]
1674     sz: BaseRM[23]
1675
1676
1677 class CROpFF3RM(ZZBaseRM, CROpBaseRM):
1678     """cr_op: ffirst 3-bit mode"""
1679     VLI: BaseRM[20]
1680     inv: BaseRM[21]
1681     CR: BaseRM[22, 23]
1682     zz: BaseRM[6]
1683     sz: BaseRM[6]
1684     dz: BaseRM[6]
1685
1686     def specifiers(self, record):
1687         yield from super().specifiers(record=record, mode="ff")
1688
1689
1690 class CROpFF5RM(DZBaseRM, SZBaseRM, CROpBaseRM):
1691     """cr_op: ffirst 5-bit mode"""
1692     VLI: BaseRM[20]
1693     inv: BaseRM[21]
1694     dz: BaseRM[22]
1695     sz: BaseRM[23]
1696
1697     def specifiers(self, record):
1698         yield from super().specifiers(record=record)
1699
1700
1701 class CROpRM(CROpBaseRM):
1702     simple: CROpSimpleRM
1703     smr: CROpSMRRM
1704     ff3: CROpFF3RM
1705     ff5: CROpFF5RM
1706
1707
1708 # ********************
1709 # Branches mode
1710 # https://libre-soc.org/openpower/sv/branches/
1711 class BranchBaseRM(BaseRM):
1712     ALL: BaseRM[4]
1713     SNZ: BaseRM[5]
1714     SL: BaseRM[17]
1715     SLu: BaseRM[18]
1716     LRu: BaseRM[22]
1717     sz: BaseRM[23]
1718
1719
1720 class BranchSimpleRM(BranchBaseRM):
1721     """branch: simple mode"""
1722     pass
1723
1724
1725 class BranchVLSRM(BranchBaseRM):
1726     """branch: VLSET mode"""
1727     VSb: BaseRM[7]
1728     VLI: BaseRM[21]
1729
1730
1731 class BranchCTRRM(BranchBaseRM):
1732     """branch: CTR-test mode"""
1733     CTi: BaseRM[6]
1734
1735
1736 class BranchCTRVLSRM(BranchVLSRM, BranchCTRRM):
1737     """branch: CTR-test+VLSET mode"""
1738     pass
1739
1740
1741 class BranchRM(BranchBaseRM):
1742     simple: BranchSimpleRM
1743     vls: BranchVLSRM
1744     ctr: BranchCTRRM
1745     ctrvls: BranchCTRVLSRM
1746
1747
1748 class RM(BaseRM):
1749     normal: NormalRM
1750     ldst_imm: LDSTImmRM
1751     ldst_idx: LDSTIdxRM
1752     cr_op: CROpRM
1753
1754     def select(self, record, Rc):
1755         rm = self
1756
1757         # the idea behind these tables is that they are now literally
1758         # in identical format to insndb.csv and minor_xx.csv and can
1759         # be done precisely as that.  the only thing to watch out for
1760         # is the insertion of Rc=1 as a "mask/value" bit and likewise
1761         # regtype detection (3-bit BF/BFA, 5-bit BA/BB/BT) also inserted
1762         # as the LSB.
1763         table = None
1764         if record.svp64.mode is _SVMode.NORMAL:
1765             # concatenate mode 5-bit with Rc (LSB) then do a mask/map search
1766             #    mode  Rc  mask  Rc  member
1767             table = (
1768                 (0b000000, 0b111000, "simple"), # simple     (no Rc)
1769                 (0b001000, 0b111000, "smr"),    # mapreduce  (no Rc)
1770                 (0b010000, 0b110001, "ffrc0"),  # ffirst,     Rc=0
1771                 (0b010001, 0b110001, "ffrc1"),  # ffirst,     Rc=1
1772                 (0b100000, 0b110000, "sat"),    # saturation (no Rc)
1773                 (0b110000, 0b110001, "prrc0"),  # predicate,  Rc=0
1774                 (0b110001, 0b110001, "prrc1"),  # predicate,  Rc=1
1775             )
1776             rm = rm.normal
1777             search = ((int(rm.mode) << 1) | Rc)
1778
1779         elif record.svp64.mode is _SVMode.LDST_IMM:
1780             # concatenate mode 5-bit with Rc (LSB) then do a mask/map search
1781             #    mode  Rc  mask  Rc  member
1782             # ironically/coincidentally this table is identical to NORMAL
1783             # mode except reserved in place of smr
1784             table = (
1785                 (0b000000, 0b111000, "simple"),   # simple     (no Rc)
1786                 (0b001000, 0b111000, "reserved"), # rsvd       (no Rc)
1787                 (0b010000, 0b110001, "ffrc0"),    # ffirst,     Rc=0
1788                 (0b010001, 0b110001, "ffrc1"),    # ffirst,     Rc=1
1789                 (0b100000, 0b110000, "sat"),      # saturation (no Rc)
1790                 (0b110000, 0b110001, "prrc0"),    # predicate,  Rc=0
1791                 (0b110001, 0b110001, "prrc1"),    # predicate,  Rc=1
1792             )
1793             rm = rm.ldst_imm
1794             search = ((int(rm.mode) << 1) | Rc)
1795
1796         elif record.svp64.mode is _SVMode.LDST_IDX:
1797             # concatenate mode 5-bit with Rc (LSB) then do a mask/map search
1798             #    mode  Rc  mask  Rc  member
1799             table = (
1800                 (0b000000, 0b111000, "simple"), # simple     (no Rc)
1801                 (0b010000, 0b110000, "stride"), # strided,   (no Rc)
1802                 (0b100000, 0b110000, "sat"),    # saturation (no Rc)
1803                 (0b110000, 0b110001, "prrc0"),  # predicate,  Rc=0
1804                 (0b110001, 0b110001, "prrc1"),  # predicate,  Rc=1
1805             )
1806             rm = rm.ldst_idx
1807             search = ((int(rm.mode) << 1) | Rc)
1808
1809         elif record.svp64.mode is _SVMode.CROP:
1810             # concatenate mode 5-bit with regtype (LSB) then do mask/map search
1811             #    mode  3b  mask  3b  member
1812             table = (
1813                 (0b000000, 0b111000, "simple"), # simple
1814                 (0b001000, 0b111000, "smr"),    # mapreduce
1815                 (0b100000, 0b100000, "ff5"),    # failfirst, 5-bit CR
1816                 (0b100001, 0b100001, "ff3"),    # failfirst, 3-bit CR
1817             )
1818             # determine CR type, 5-bit (BA/BB/BT) or 3-bit Field (BF/BFA)
1819             regtype = None
1820             for idx in range(0, 4):
1821                 for entry in record.svp64.extra[idx]:
1822                     if entry.regtype is _SVExtraRegType.DST:
1823                         if regtype is not None:
1824                             raise ValueError(record.svp64)
1825                         regtype = _RegType(entry.reg)
1826             if regtype is _RegType.CR_REG:
1827                 regtype = 0 # 5-bit
1828             elif regtype is _RegType.CR_BIT:
1829                 regtype = 1 # 3-bit
1830             else:
1831                 raise ValueError(record.svp64)
1832             # finally provide info for search
1833             rm = rm.cr_op
1834             search = ((int(rm.mode) << 1) | (regtype or 0))
1835
1836         elif record.svp64.mode is _SVMode.BRANCH:
1837             # just mode 5-bit. could be reduced down to 2, oh well.
1838             #    mode     mask     member
1839             table = (
1840                 (0b00000, 0b11000, "simple"), # simple
1841                 (0b01000, 0b11000, "vls"),    # VLset
1842                 (0b10000, 0b11000, "ctr"),    # CTR mode
1843                 (0b11000, 0b11000, "ctrvls"), # CTR+VLset mode
1844             )
1845             # slightly weird: doesn't have a 5-bit "mode" field like others
1846             search = int(rm[19:23])
1847
1848         # look up in table
1849         if table is not None:
1850             for (value, mask, member) in table:
1851                 if ((value & search) == (mask & search)):
1852                     rm = getattr(rm, member)
1853                     break
1854
1855         if rm.__class__ is self.__class__:
1856             raise ValueError(self)
1857
1858         return rm
1859
1860
1861 class SVP64Instruction(PrefixedInstruction):
1862     """SVP64 instruction: https://libre-soc.org/openpower/sv/svp64/"""
1863     class Prefix(PrefixedInstruction.Prefix):
1864         id: _Field = (7, 9)
1865         rm: RM.remap((6, 8) + tuple(range(10, 32)))
1866
1867     prefix: Prefix
1868
1869     def record(self, db):
1870         record = db[self.suffix]
1871         if record is None:
1872             raise KeyError(self)
1873         return record
1874
1875     @property
1876     def binary(self):
1877         bits = []
1878         for idx in range(64):
1879             bit = int(self[idx])
1880             bits.append(bit)
1881         return "".join(map(str, bits))
1882
1883     def disassemble(self, db,
1884             byteorder="little",
1885             verbosity=Verbosity.NORMAL):
1886         def blob(integer):
1887             if verbosity <= Verbosity.SHORT:
1888                 return ""
1889             else:
1890                 blob = integer.to_bytes(length=4, byteorder=byteorder)
1891                 blob = " ".join(map(lambda byte: f"{byte:02x}", blob))
1892                 return f"{blob}    "
1893
1894         record = self.record(db=db)
1895         blob_prefix = blob(int(self.prefix))
1896         blob_suffix = blob(int(self.suffix))
1897         if record is None or record.svp64 is None:
1898             yield f"{blob_prefix}.long 0x{int(self.prefix):08x}"
1899             yield f"{blob_suffix}.long 0x{int(self.suffix):08x}"
1900             return
1901
1902         name = f"sv.{record.name}"
1903
1904         Rc = False
1905         if record.mdwn.operands["Rc"] is not None:
1906             Rc = bool(record.mdwn.operands["Rc"].value)
1907         rm = self.prefix.rm.select(record=record, Rc=Rc)
1908
1909         # convert specifiers to /x/y/z
1910         specifiers = list(rm.specifiers(record=record))
1911         if specifiers: # if any add one extra to get the extra "/"
1912             specifiers = ['']+specifiers
1913         specifiers = "/".join(specifiers)
1914
1915         # convert operands to " ,x,y,z"
1916         operands = tuple(map(_operator.itemgetter(1),
1917             self.dynamic_operands(db=db, verbosity=verbosity)))
1918         operands = ','.join(operands)
1919         if len(operands) > 0: # if any separate with a space
1920             operands = " " + operands
1921
1922         yield f"{blob_prefix}{name}{specifiers}{operands}"
1923         if blob_suffix:
1924             yield f"{blob_suffix}"
1925
1926         if verbosity >= Verbosity.VERBOSE:
1927             indent = (" " * 4)
1928             binary = self.binary
1929             spec = self.spec(db=db, prefix="sv.")
1930
1931             yield f"{indent}spec"
1932             yield f"{indent}{indent}{spec}"
1933             yield f"{indent}pcode"
1934             for stmt in record.mdwn.pcode:
1935                 yield f"{indent}{indent}{stmt}"
1936             yield f"{indent}binary"
1937             yield f"{indent}{indent}[0:8]   {binary[0:8]}"
1938             yield f"{indent}{indent}[8:16]  {binary[8:16]}"
1939             yield f"{indent}{indent}[16:24] {binary[16:24]}"
1940             yield f"{indent}{indent}[24:32] {binary[24:32]}"
1941             yield f"{indent}{indent}[32:40] {binary[32:40]}"
1942             yield f"{indent}{indent}[40:48] {binary[40:48]}"
1943             yield f"{indent}{indent}[48:56] {binary[48:56]}"
1944             yield f"{indent}{indent}[56:64] {binary[56:64]}"
1945             yield f"{indent}opcodes"
1946             for opcode in record.opcodes:
1947                 yield f"{indent}{indent}{opcode!r}"
1948             for operand in record.mdwn.operands:
1949                 yield from operand.disassemble(insn=self, record=record,
1950                     verbosity=verbosity, indent=indent)
1951             yield f"{indent}RM"
1952             yield f"{indent}{indent}{rm.__doc__}"
1953             for line in rm.disassemble(verbosity=verbosity):
1954                 yield f"{indent}{indent}{line}"
1955             yield ""
1956
1957
1958 def parse(stream, factory):
1959     def match(entry):
1960         return ("TODO" not in frozenset(entry.values()))
1961
1962     lines = filter(lambda line: not line.strip().startswith("#"), stream)
1963     entries = _csv.DictReader(lines)
1964     entries = filter(match, entries)
1965     return tuple(map(factory, entries))
1966
1967
1968 class MarkdownDatabase:
1969     def __init__(self):
1970         db = {}
1971         for (name, desc) in _ISA():
1972             operands = []
1973             if desc.regs:
1974                 (dynamic, *static) = desc.regs
1975                 operands.extend(dynamic)
1976                 operands.extend(static)
1977             pcode = PCode(iterable=desc.pcode)
1978             operands = Operands(insn=name, iterable=operands)
1979             db[name] = MarkdownRecord(pcode=pcode, operands=operands)
1980
1981         self.__db = db
1982
1983         return super().__init__()
1984
1985     def __iter__(self):
1986         yield from self.__db.items()
1987
1988     def __contains__(self, key):
1989         return self.__db.__contains__(key)
1990
1991     def __getitem__(self, key):
1992         return self.__db.__getitem__(key)
1993
1994
1995 class FieldsDatabase:
1996     def __init__(self):
1997         db = {}
1998         df = _DecodeFields()
1999         df.create_specs()
2000         for (form, fields) in df.instrs.items():
2001             if form in {"DQE", "TX"}:
2002                 continue
2003             if form == "all":
2004                 form = "NONE"
2005             db[_Form[form]] = Fields(fields)
2006
2007         self.__db = db
2008
2009         return super().__init__()
2010
2011     def __getitem__(self, key):
2012         return self.__db.__getitem__(key)
2013
2014
2015 class PPCDatabase:
2016     def __init__(self, root, mdwndb):
2017         # The code below groups the instructions by section:identifier.
2018         # We use the comment as an identifier, there's nothing better.
2019         # The point is to capture different opcodes for the same instruction.
2020         dd = _collections.defaultdict
2021         records = dd(lambda: dd(set))
2022         path = (root / "insndb.csv")
2023         with open(path, "r", encoding="UTF-8") as stream:
2024             for section in parse(stream, Section.CSV):
2025                 path = (root / section.path)
2026                 opcode_cls = {
2027                     section.Mode.INTEGER: IntegerOpcode,
2028                     section.Mode.PATTERN: PatternOpcode,
2029                 }[section.mode]
2030                 factory = _functools.partial(
2031                     PPCRecord.CSV, opcode_cls=opcode_cls)
2032                 with open(path, "r", encoding="UTF-8") as stream:
2033                     for insn in parse(stream, factory):
2034                         records[section][insn.comment].add(insn)
2035
2036         sections = dd(set)
2037         for (section, group) in records.items():
2038             for records in group.values():
2039                 sections[section].add(PPCMultiRecord(records))
2040
2041         db = {}
2042         for (section, records) in sections.items():
2043             for record in records:
2044                 def exact_match(names):
2045                     for name in names:
2046                         if name in mdwndb:
2047                             yield name
2048
2049                 def Rc_match(names):
2050                     for name in names:
2051                         if f"{name}." in mdwndb:
2052                             yield f"{name}."
2053                         yield name
2054
2055                 def LK_match(names):
2056                     if "lk" not in record.flags:
2057                         yield from names
2058                         return
2059
2060                     for name in names:
2061                         if f"{name}l" in mdwndb:
2062                             yield f"{name}l"
2063                         yield name
2064
2065                 def AA_match(names):
2066                     if record.intop not in {_MicrOp.OP_B, _MicrOp.OP_BC}:
2067                         yield from names
2068                         return
2069
2070                     for name in names:
2071                         operands = mdwndb[name].operands["AA"]
2072                         if ((operands is not None) and
2073                                 (f"{name}a" in mdwndb)):
2074                             yield f"{name}a"
2075                         yield name
2076
2077                 def reductor(names, match):
2078                     return match(names)
2079
2080                 matches = (exact_match, Rc_match, LK_match, AA_match)
2081
2082                 names = _functools.reduce(reductor, matches, record.names)
2083                 for name in names:
2084                     db[name] = (section, record)
2085
2086         self.__db = db
2087         self.__mdwndb = mdwndb
2088
2089         return super().__init__()
2090
2091     @_functools.lru_cache(maxsize=512, typed=False)
2092     def __getitem__(self, key):
2093         return self.__db.get(key, (None, None))
2094
2095
2096 class SVP64Database:
2097     def __init__(self, root, ppcdb):
2098         db = set()
2099         pattern = _re.compile(r"^(?:LDST)?RM-(1P|2P)-.*?\.csv$")
2100         for (prefix, _, names) in _os.walk(root):
2101             prefix = _pathlib.Path(prefix)
2102             for name in filter(lambda name: pattern.match(name), names):
2103                 path = (prefix / _pathlib.Path(name))
2104                 with open(path, "r", encoding="UTF-8") as stream:
2105                     db.update(parse(stream, SVP64Record.CSV))
2106
2107         self.__db = {record.name:record for record in db}
2108         self.__ppcdb = ppcdb
2109
2110         return super().__init__()
2111
2112     def __getitem__(self, key):
2113         (_, record) = self.__ppcdb[key]
2114         if record is None:
2115             return None
2116
2117         for name in record.names:
2118             record = self.__db.get(name, None)
2119             if record is not None:
2120                 return record
2121
2122         return None
2123
2124
2125 class Database:
2126     def __init__(self, root):
2127         root = _pathlib.Path(root)
2128         mdwndb = MarkdownDatabase()
2129         fieldsdb = FieldsDatabase()
2130         ppcdb = PPCDatabase(root=root, mdwndb=mdwndb)
2131         svp64db = SVP64Database(root=root, ppcdb=ppcdb)
2132
2133         db = set()
2134         names = {}
2135         opcodes = _collections.defaultdict(set)
2136
2137         for (name, mdwn) in mdwndb:
2138             (section, ppc) = ppcdb[name]
2139             if ppc is None:
2140                 continue
2141             svp64 = svp64db[name]
2142             fields = fieldsdb[ppc.form]
2143             record = Record(name=name,
2144                 section=section, ppc=ppc, svp64=svp64,
2145                 mdwn=mdwn, fields=fields)
2146             db.add(record)
2147             names[record.name] = record
2148             PO = section.opcode
2149             if PO is None:
2150                 PO = ppc[0].opcode
2151             opcodes[PO.value].add(record)
2152
2153         self.__db = db
2154         self.__names = names
2155         self.__opcodes = opcodes
2156
2157         return super().__init__()
2158
2159     def __repr__(self):
2160         return repr(self.__db)
2161
2162     def __iter__(self):
2163         yield from self.__db
2164
2165     @_functools.lru_cache(maxsize=None)
2166     def __contains__(self, key):
2167         return self.__getitem__(key) is not None
2168
2169     @_functools.lru_cache(maxsize=None)
2170     def __getitem__(self, key):
2171         if isinstance(key, (int, Instruction)):
2172             key = int(key)
2173             XO = int(_SelectableInt(value=int(key), bits=32)[0:6])
2174             for record in self.__opcodes[XO]:
2175                 if record.match(key=key):
2176                    return record
2177
2178         elif isinstance(key, str):
2179             return self.__names[key]
2180
2181         return None