parser.cpp

#pragma once
#include <iostream>
#include <cstring>
const int MEM_MAX=1e6;
static unsigned char mem[MEM_MAX];
enum commands {LUI,AUIPC,JAL,JALR,BEQ,BNE,BLT,BGE,BLTU,BGEU,LB,LH,LW,LBU,LHU,SB,SH,SW,ADDI,SLTI,SLTIU,XORI,ORI,ANDI,SLLI,SRLI,SRAI,ADD,SUB,SLL,SLT,
SLTU,XOR,SRL,SRA,OR,AND};
enum operators {none=-1,eq,neq,le,geq,leu,gequ,add,xori,ori,andi,slli,srli,srai,sub};
const int ops[37]={none,none,none,add,eq,neq,le,geq,leu,gequ,add,add,add,add,add,add,add,add,add,le,leu,xori,ori,andi,slli,srli,srai,add,sub,slli,
le,leu,xori,srli,srai,ori,andi};
unsigned int getValxx(const unsigned char *s,int l,int r)
{
    unsigned int ret=0;
    for(int i=l;i<=r;i++) ret|=s[i]<<((i-l)*8);
    return ret; 
}
unsigned int getPart(unsigned int v,int l,int r)
{
    return (v>>r)&((1u<<(l-r+1))-1);
}
int sext(unsigned int num,int len)
{
    if(num&(1u<<(len-1))) return num|((-1u)>>len<<len);
    else return num;
}
unsigned int getValx(const char *s,int l,int r)
{
    unsigned int ret=0;
    for(int i=l;i<=r;i++)
    {
        int x=('0'<=s[i]&&s[i]<='9')?s[i]-'0':s[i]-'A'+10;
        ret=ret<<4|x;
    }
    return ret; 
}
char getType1(int l)
{
    int s=getPart(getValxx(mem,l,l+3),6,0);
    if(s==0b0110111||s==0b0010111) return 'U';
    if(s==0b1101111) return 'J';
    if(s==0b1100011) return 'B';
    if(s==0b1100111||s==0b0000011||s==0b0010011) return 'I';
    if(s==0b0100011) return 'S';
    if(s==0b0110011) return 'R';
    return '?';
}
int getType2(int l)
{
    unsigned int in=getValxx(mem,l,l+3);
    int s=getPart(in,6,0);
    if(s==0b0110111) return LUI;
    if(s==0b0010111) return AUIPC;
    if(s==0b1101111) return JAL;
    if(s==0b1100111) return JALR;
    if(s==0b1100011)
    {
        int t=getPart(in,14,12);
        if(t==0b000) return BEQ;
        if(t==0b001) return BNE;
        if(t==0b100) return BLT;
        if(t==0b101) return BGE;
        if(t==0b110) return BLTU;
        if(t==0b111) return BGEU;
    }
    if(s==0b0000011)
    {
        int t=getPart(in,14,12);
        if(t==0b000) return LB;
        if(t==0b001) return LH;
        if(t==0b010) return LW;
        if(t==0b100) return LBU;
        if(t==0b101) return LHU;
    }
    if(s==0b0100011)
    {
        int t=getPart(in,14,12);
        if(t==0b000) return SB;
        if(t==0b001) return SH;
        if(t==0b010) return SW;
    }
    if(s==0b0010011)
    {
        int t=getPart(in,14,12);
        if(t==0b000) return ADDI;
        if(t==0b010) return SLTI;
        if(t==0b011) return SLTIU;
        if(t==0b100) return XORI;
        if(t==0b110) return ORI;
        if(t==0b111) return ANDI;
        if(t==0b001) return SLLI;
        if(t==0b101)
        {
            if(getPart(in,30,30)==0) return SRLI;
            else return SRAI;
        }
    }
    if(s==0b0110011)
    {
        int t=getPart(in,14,12);
        if(t==0b000)
        {
            if(getPart(in,30,30)==0) return ADD;
            else return SUB;
        }
        if(t==0b001) return SLL;
        if(t==0b010) return SLT;
        if(t==0b011) return SLTU;
        if(t==0b100) return XOR;
        if(t==0b101)
        {
            if(getPart(in,30,30)==0) return SRL;
            else return SRA;
        }
        if(t==0b110) return OR;
        if(t==0b111) return AND;
    }
    return -1;
}
int getOp(int l)
{
    int type=getType2(l);
    return ops[type];
}
int getImm(int l)
{
    int in=getValxx(mem,l,l+3);
    char typ=getType1(l);
    if(typ=='I') return sext(getPart(in,31,20),12);
    if(typ=='S') return sext(getPart(in,31,25)<<5|getPart(in,11,7),12);
    if(typ=='B') return sext(getPart(in,31,31)<<12|getPart(in,30,25)<<5|getPart(in,11,8)<<1|getPart(in,7,7)<<11,13);
    if(typ=='U') return getPart(in,31,12)<<12;
    if(typ=='J') return sext(getPart(in,31,31)<<20|getPart(in,30,21)<<1|getPart(in,20,20)<<11|getPart(in,19,12)<<12,21);
    return 114514;
}
int getRd(int l)
{
    return getPart(getValxx(mem,l,l+3),11,7);
}
int getRs1(int l)
{
    if(getType1(l)=='U'||getType1(l)=='J') return -1;
    return getPart(getValxx(mem,l,l+3),19,15);
}
int getRs2(int l)
{
    if(getType1(l)=='U'||getType1(l)=='J'||getType1(l)=='I') return -1;
    return getPart(getValxx(mem,l,l+3),24,20);
}
int getDest(int l)
{
    char t1=getType1(l);
    if(t1=='B'||t1=='S') return 0;
    else return getRd(l);
}
void getAllCommands()
{
    std::ios::sync_with_stdio(false);
    int tp=0;
    char ts[20];
    while(std::cin>>ts)
    {
        if(ts[0]=='@') tp=getValx(ts,1,strlen(ts)-1);
        else
        {
            int v1=('0'<=ts[0]&&ts[0]<='9')?ts[0]-'0':ts[0]-'A'+10;
            int v2=('0'<=ts[1]&&ts[1]<='9')?ts[1]-'0':ts[1]-'A'+10;
            mem[tp++]=v1<<4|v2;
        }
    }
}