mirror of
https://github.com/Ryujinx/Ryujinx.git
synced 2024-10-01 12:30:00 +02:00
188 lines
5 KiB
C#
188 lines
5 KiB
C#
|
using ChocolArm64.Decoders;
|
||
|
using ChocolArm64.State;
|
||
|
using System;
|
||
|
using System.Collections.Generic;
|
||
|
using System.Reflection.Emit;
|
||
|
using System.Runtime.Intrinsics;
|
||
|
|
||
|
namespace ChocolArm64.Translation
|
||
|
{
|
||
|
class ILEmitter
|
||
|
{
|
||
|
public LocalAlloc LocalAlloc { get; private set; }
|
||
|
|
||
|
public ILGenerator Generator { get; private set; }
|
||
|
|
||
|
private Dictionary<Register, int> _locals;
|
||
|
|
||
|
private ILBlock[] _ilBlocks;
|
||
|
|
||
|
private ILBlock _root;
|
||
|
|
||
|
private TranslatedSub _subroutine;
|
||
|
|
||
|
private string _subName;
|
||
|
|
||
|
private int _localsCount;
|
||
|
|
||
|
public ILEmitter(Block[] graph, Block root, string subName)
|
||
|
{
|
||
|
_subName = subName;
|
||
|
|
||
|
_locals = new Dictionary<Register, int>();
|
||
|
|
||
|
_ilBlocks = new ILBlock[graph.Length];
|
||
|
|
||
|
ILBlock GetBlock(int index)
|
||
|
{
|
||
|
if (index < 0 || index >= _ilBlocks.Length)
|
||
|
{
|
||
|
return null;
|
||
|
}
|
||
|
|
||
|
if (_ilBlocks[index] == null)
|
||
|
{
|
||
|
_ilBlocks[index] = new ILBlock();
|
||
|
}
|
||
|
|
||
|
return _ilBlocks[index];
|
||
|
}
|
||
|
|
||
|
for (int index = 0; index < _ilBlocks.Length; index++)
|
||
|
{
|
||
|
ILBlock block = GetBlock(index);
|
||
|
|
||
|
block.Next = GetBlock(Array.IndexOf(graph, graph[index].Next));
|
||
|
block.Branch = GetBlock(Array.IndexOf(graph, graph[index].Branch));
|
||
|
}
|
||
|
|
||
|
_root = _ilBlocks[Array.IndexOf(graph, root)];
|
||
|
}
|
||
|
|
||
|
public ILBlock GetIlBlock(int index) => _ilBlocks[index];
|
||
|
|
||
|
public TranslatedSub GetSubroutine()
|
||
|
{
|
||
|
LocalAlloc = new LocalAlloc(_ilBlocks, _root);
|
||
|
|
||
|
InitSubroutine();
|
||
|
InitLocals();
|
||
|
|
||
|
foreach (ILBlock ilBlock in _ilBlocks)
|
||
|
{
|
||
|
ilBlock.Emit(this);
|
||
|
}
|
||
|
|
||
|
return _subroutine;
|
||
|
}
|
||
|
|
||
|
private void InitSubroutine()
|
||
|
{
|
||
|
List<Register> Params = new List<Register>();
|
||
|
|
||
|
void SetParams(long inputs, RegisterType baseType)
|
||
|
{
|
||
|
for (int bit = 0; bit < 64; bit++)
|
||
|
{
|
||
|
long mask = 1L << bit;
|
||
|
|
||
|
if ((inputs & mask) != 0)
|
||
|
{
|
||
|
Params.Add(GetRegFromBit(bit, baseType));
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
SetParams(LocalAlloc.GetIntInputs(_root), RegisterType.Int);
|
||
|
SetParams(LocalAlloc.GetVecInputs(_root), RegisterType.Vector);
|
||
|
|
||
|
DynamicMethod mthd = new DynamicMethod(_subName, typeof(long), GetParamTypes(Params));
|
||
|
|
||
|
Generator = mthd.GetILGenerator();
|
||
|
|
||
|
_subroutine = new TranslatedSub(mthd, Params);
|
||
|
}
|
||
|
|
||
|
private void InitLocals()
|
||
|
{
|
||
|
int paramsStart = TranslatedSub.FixedArgTypes.Length;
|
||
|
|
||
|
_locals = new Dictionary<Register, int>();
|
||
|
|
||
|
for (int index = 0; index < _subroutine.Params.Count; index++)
|
||
|
{
|
||
|
Register reg = _subroutine.Params[index];
|
||
|
|
||
|
Generator.EmitLdarg(index + paramsStart);
|
||
|
Generator.EmitStloc(GetLocalIndex(reg));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
private Type[] GetParamTypes(IList<Register> Params)
|
||
|
{
|
||
|
Type[] fixedArgs = TranslatedSub.FixedArgTypes;
|
||
|
|
||
|
Type[] output = new Type[Params.Count + fixedArgs.Length];
|
||
|
|
||
|
fixedArgs.CopyTo(output, 0);
|
||
|
|
||
|
int typeIdx = fixedArgs.Length;
|
||
|
|
||
|
for (int index = 0; index < Params.Count; index++)
|
||
|
{
|
||
|
output[typeIdx++] = GetFieldType(Params[index].Type);
|
||
|
}
|
||
|
|
||
|
return output;
|
||
|
}
|
||
|
|
||
|
public int GetLocalIndex(Register reg)
|
||
|
{
|
||
|
if (!_locals.TryGetValue(reg, out int index))
|
||
|
{
|
||
|
Generator.DeclareLocal(GetLocalType(reg));
|
||
|
|
||
|
index = _localsCount++;
|
||
|
|
||
|
_locals.Add(reg, index);
|
||
|
}
|
||
|
|
||
|
return index;
|
||
|
}
|
||
|
|
||
|
public Type GetLocalType(Register reg) => GetFieldType(reg.Type);
|
||
|
|
||
|
public Type GetFieldType(RegisterType regType)
|
||
|
{
|
||
|
switch (regType)
|
||
|
{
|
||
|
case RegisterType.Flag: return typeof(bool);
|
||
|
case RegisterType.Int: return typeof(ulong);
|
||
|
case RegisterType.Vector: return typeof(Vector128<float>);
|
||
|
}
|
||
|
|
||
|
throw new ArgumentException(nameof(regType));
|
||
|
}
|
||
|
|
||
|
public static Register GetRegFromBit(int bit, RegisterType baseType)
|
||
|
{
|
||
|
if (bit < 32)
|
||
|
{
|
||
|
return new Register(bit, baseType);
|
||
|
}
|
||
|
else if (baseType == RegisterType.Int)
|
||
|
{
|
||
|
return new Register(bit & 0x1f, RegisterType.Flag);
|
||
|
}
|
||
|
else
|
||
|
{
|
||
|
throw new ArgumentOutOfRangeException(nameof(bit));
|
||
|
}
|
||
|
}
|
||
|
|
||
|
public static bool IsRegIndex(int index)
|
||
|
{
|
||
|
return index >= 0 && index < 32;
|
||
|
}
|
||
|
}
|
||
|
}
|