mirror of
https://github.com/GreemDev/Ryujinx.git
synced 2024-12-15 03:21:15 +01:00
201 lines
5.9 KiB
C#
201 lines
5.9 KiB
C#
using ChocolArm64.Decoder;
|
|
using ChocolArm64.Events;
|
|
using ChocolArm64.Instruction;
|
|
using ChocolArm64.Memory;
|
|
using ChocolArm64.Translation;
|
|
using System;
|
|
using System.Collections.Concurrent;
|
|
using System.Collections.Generic;
|
|
using System.Reflection.Emit;
|
|
|
|
namespace ChocolArm64
|
|
{
|
|
public class ATranslator
|
|
{
|
|
private HashSet<long> SubBlocks;
|
|
|
|
private ConcurrentDictionary<long, ATranslatedSub> CachedSubs;
|
|
|
|
private ConcurrentDictionary<long, string> SymbolTable;
|
|
|
|
public event EventHandler<ACpuTraceEventArgs> CpuTrace;
|
|
|
|
public bool EnableCpuTrace { get; set; }
|
|
|
|
public ATranslator(IReadOnlyDictionary<long, string> SymbolTable = null)
|
|
{
|
|
SubBlocks = new HashSet<long>();
|
|
|
|
CachedSubs = new ConcurrentDictionary<long, ATranslatedSub>();
|
|
|
|
if (SymbolTable != null)
|
|
{
|
|
this.SymbolTable = new ConcurrentDictionary<long, string>(SymbolTable);
|
|
}
|
|
else
|
|
{
|
|
this.SymbolTable = new ConcurrentDictionary<long, string>();
|
|
}
|
|
}
|
|
|
|
internal void ExecuteSubroutine(AThread Thread, long Position)
|
|
{
|
|
do
|
|
{
|
|
if (EnableCpuTrace)
|
|
{
|
|
if (!SymbolTable.TryGetValue(Position, out string SubName))
|
|
{
|
|
SubName = string.Empty;
|
|
}
|
|
|
|
CpuTrace?.Invoke(this, new ACpuTraceEventArgs(Position, SubName));
|
|
}
|
|
|
|
if (!CachedSubs.TryGetValue(Position, out ATranslatedSub Sub))
|
|
{
|
|
Sub = TranslateTier0(Thread.Memory, Position);
|
|
}
|
|
|
|
if (Sub.ShouldReJit())
|
|
{
|
|
TranslateTier1(Thread.Memory, Position);
|
|
}
|
|
|
|
Position = Sub.Execute(Thread.ThreadState, Thread.Memory);
|
|
}
|
|
while (Position != 0 && Thread.ThreadState.Running);
|
|
}
|
|
|
|
internal bool TryGetCachedSub(AOpCode OpCode, out ATranslatedSub Sub)
|
|
{
|
|
if (OpCode.Emitter != AInstEmit.Bl)
|
|
{
|
|
Sub = null;
|
|
|
|
return false;
|
|
}
|
|
|
|
return TryGetCachedSub(((AOpCodeBImmAl)OpCode).Imm, out Sub);
|
|
}
|
|
|
|
internal bool TryGetCachedSub(long Position, out ATranslatedSub Sub)
|
|
{
|
|
return CachedSubs.TryGetValue(Position, out Sub);
|
|
}
|
|
|
|
internal bool HasCachedSub(long Position)
|
|
{
|
|
return CachedSubs.ContainsKey(Position);
|
|
}
|
|
|
|
private ATranslatedSub TranslateTier0(AMemory Memory, long Position)
|
|
{
|
|
ABlock Block = ADecoder.DecodeBasicBlock(this, Memory, Position);
|
|
|
|
ABlock[] Graph = new ABlock[] { Block };
|
|
|
|
string SubName = GetSubName(Position);
|
|
|
|
AILEmitterCtx Context = new AILEmitterCtx(this, Graph, Block, SubName);
|
|
|
|
do
|
|
{
|
|
Context.EmitOpCode();
|
|
}
|
|
while (Context.AdvanceOpCode());
|
|
|
|
ATranslatedSub Subroutine = Context.GetSubroutine();
|
|
|
|
lock (SubBlocks)
|
|
{
|
|
if (SubBlocks.Contains(Position))
|
|
{
|
|
SubBlocks.Remove(Position);
|
|
|
|
Subroutine.SetType(ATranslatedSubType.SubBlock);
|
|
}
|
|
else
|
|
{
|
|
Subroutine.SetType(ATranslatedSubType.SubTier0);
|
|
}
|
|
}
|
|
|
|
CachedSubs.AddOrUpdate(Position, Subroutine, (Key, OldVal) => Subroutine);
|
|
|
|
AOpCode LastOp = Block.GetLastOp();
|
|
|
|
lock (SubBlocks)
|
|
{
|
|
if (LastOp.Emitter != AInstEmit.Ret &&
|
|
LastOp.Emitter != AInstEmit.Br)
|
|
{
|
|
SubBlocks.Add(LastOp.Position + 4);
|
|
}
|
|
}
|
|
|
|
return Subroutine;
|
|
}
|
|
|
|
private void TranslateTier1(AMemory Memory, long Position)
|
|
{
|
|
(ABlock[] Graph, ABlock Root) Cfg = ADecoder.DecodeSubroutine(this, Memory, Position);
|
|
|
|
string SubName = GetSubName(Position);
|
|
|
|
PropagateName(Cfg.Graph, SubName);
|
|
|
|
AILEmitterCtx Context = new AILEmitterCtx(this, Cfg.Graph, Cfg.Root, SubName);
|
|
|
|
if (Context.CurrBlock.Position != Position)
|
|
{
|
|
Context.Emit(OpCodes.Br, Context.GetLabel(Position));
|
|
}
|
|
|
|
do
|
|
{
|
|
Context.EmitOpCode();
|
|
}
|
|
while (Context.AdvanceOpCode());
|
|
|
|
//Mark all methods that calls this method for ReJiting,
|
|
//since we can now call it directly which is faster.
|
|
if (CachedSubs.TryGetValue(Position, out ATranslatedSub OldSub))
|
|
{
|
|
foreach (long CallerPos in OldSub.GetCallerPositions())
|
|
{
|
|
if (CachedSubs.TryGetValue(Position, out ATranslatedSub CallerSub))
|
|
{
|
|
CallerSub.MarkForReJit();
|
|
}
|
|
}
|
|
}
|
|
|
|
ATranslatedSub Subroutine = Context.GetSubroutine();
|
|
|
|
Subroutine.SetType(ATranslatedSubType.SubTier1);
|
|
|
|
CachedSubs.AddOrUpdate(Position, Subroutine, (Key, OldVal) => Subroutine);
|
|
}
|
|
|
|
private string GetSubName(long Position)
|
|
{
|
|
return SymbolTable.GetOrAdd(Position, $"Sub{Position:x16}");
|
|
}
|
|
|
|
private void PropagateName(ABlock[] Graph, string Name)
|
|
{
|
|
foreach (ABlock Block in Graph)
|
|
{
|
|
AOpCode LastOp = Block.GetLastOp();
|
|
|
|
if (LastOp != null &&
|
|
(LastOp.Emitter == AInstEmit.Bl ||
|
|
LastOp.Emitter == AInstEmit.Blr))
|
|
{
|
|
SymbolTable.TryAdd(LastOp.Position + 4, Name);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |