<?php
namespace Std\Lib\Syntax;

/**
 * RISC-V（RV32/RV64、GAS）
 */
class AsmRiscv extends Assembly {
  protected array $comment = ['#'];
  protected ?array $multiline = ['/*', '*/'];
  protected bool $dollarHex = false;
  protected bool $percentBinary = false;
  protected string $immediatePrefix = '';
  // fcvt.w.s → fcvt.w → fcvt、lr.w.aq → lr、c.addi → addi
  protected array $suffixes = ['/\.[a-z0-9]+$/', '/\.[a-z0-9]+$/', '/^c\./'];

  protected array $tokens = [
    'register' => [
      'zero', 'ra', 'sp', 'gp', 'tp', 'fp', 'pc',
    ],
    'instruction' => [
      // RV32I/RV64I
      'lui', 'auipc', 'jal', 'jalr', 'beq', 'bne', 'blt', 'bge', 'bltu', 'bgeu', 'lb', 'lh', 'lw', 'lbu', 'lhu', 'lwu', 'ld', 'sb', 'sh', 'sw', 'sd',
      'addi', 'slti', 'sltiu', 'xori', 'ori', 'andi', 'slli', 'srli', 'srai', 'add', 'sub', 'sll', 'slt', 'sltu', 'xor', 'srl', 'sra', 'or', 'and',
      'addiw', 'slliw', 'srliw', 'sraiw', 'addw', 'subw', 'sllw', 'srlw', 'sraw', 'fence', 'ecall', 'ebreak',
      // M・A
      'mul', 'mulh', 'mulhsu', 'mulhu', 'div', 'divu', 'rem', 'remu', 'mulw', 'divw', 'divuw', 'remw', 'remuw',
      'lr', 'sc', 'amoswap', 'amoadd', 'amoxor', 'amoand', 'amoor', 'amomin', 'amomax', 'amominu', 'amomaxu',
      // F・D
      'flw', 'fsw', 'fld', 'fsd', 'fadd', 'fsub', 'fmul', 'fdiv', 'fsqrt', 'fmin', 'fmax', 'fmadd', 'fmsub', 'fnmadd', 'fnmsub',
      'fsgnj', 'fsgnjn', 'fsgnjx', 'fcvt', 'fmv', 'feq', 'flt', 'fle', 'fclass',
      // Zicsr・特権
      'csrrw', 'csrrs', 'csrrc', 'csrrwi', 'csrrsi', 'csrrci', 'csrr', 'csrw', 'csrs', 'csrc', 'mret', 'sret', 'wfi', 'sfence',
      // B・V（よく使う物）
      'andn', 'orn', 'xnor', 'clz', 'ctz', 'cpop', 'min', 'max', 'minu', 'maxu', 'rol', 'ror', 'rev8', 'sext', 'zext',
      'vsetvli', 'vsetivli', 'vsetvl', 'vle8', 'vle16', 'vle32', 'vle64', 'vse8', 'vse16', 'vse32', 'vse64', 'vadd', 'vsub', 'vmul', 'vfadd', 'vfmul',
      'vfmacc', 'vredsum', 'vfredusum', 'vmv',
      // 疑似命令
      'li', 'la', 'lla', 'mv', 'not', 'neg', 'negw', 'nop', 'j', 'jr', 'ret', 'call', 'tail', 'seqz', 'snez', 'sltz', 'sgtz',
      'beqz', 'bnez', 'blez', 'bgez', 'bltz', 'bgtz', 'bgt', 'ble', 'bgtu', 'bleu', 'fabs', 'fneg',
    ],
  ];

  public function prepare(string $code): void {
    $tokens = $this->tokens;
    $names = array_merge(
      self::numbered('x', 0, 31), self::numbered('f', 0, 31), self::numbered('v', 0, 31),
      self::numbered('t', 0, 6), self::numbered('s', 0, 11), self::numbered('a', 0, 7),
      self::numbered('ft', 0, 11), self::numbered('fs', 0, 11), self::numbered('fa', 0, 7),
    );
    foreach ($names as $r) $tokens['register'][] = $r;
    $this->words = $this->buildWords($tokens);
  }
}