using Ryujinx.Graphics.Shader.IntermediateRepresentation; using Ryujinx.Graphics.Shader.StructuredIr; using System; using static Ryujinx.Graphics.Shader.CodeGen.Glsl.Instructions.InstGenCall; using static Ryujinx.Graphics.Shader.CodeGen.Glsl.Instructions.InstGenHelper; using static Ryujinx.Graphics.Shader.CodeGen.Glsl.Instructions.InstGenMemory; using static Ryujinx.Graphics.Shader.CodeGen.Glsl.Instructions.InstGenPacking; using static Ryujinx.Graphics.Shader.StructuredIr.InstructionInfo; namespace Ryujinx.Graphics.Shader.CodeGen.Glsl.Instructions { static class InstGen { public static string GetExpression(CodeGenContext context, IAstNode node) { if (node is AstOperation operation) { return GetExpression(context, operation); } else if (node is AstOperand operand) { return context.OperandManager.GetExpression(operand, context.Config, context.CbIndexable); } throw new ArgumentException($"Invalid node type \"{node?.GetType().Name ?? "null"}\"."); } private static string GetExpression(CodeGenContext context, AstOperation operation) { Instruction inst = operation.Inst; InstInfo info = GetInstructionInfo(inst); if ((info.Type & InstType.Call) != 0) { bool atomic = (info.Type & InstType.Atomic) != 0; int arity = (int)(info.Type & InstType.ArityMask); string args = string.Empty; for (int argIndex = 0; argIndex < arity; argIndex++) { if (argIndex != 0) { args += ", "; } VariableType dstType = GetSrcVarType(inst, argIndex); if (argIndex == 0 && atomic) { Instruction memRegion = inst & Instruction.MrMask; switch (memRegion) { case Instruction.MrShared: args += LoadShared (context, operation); break; case Instruction.MrStorage: args += LoadStorage(context, operation); break; default: throw new InvalidOperationException($"Invalid memory region \"{memRegion}\"."); } // We use the first 2 operands above. argIndex++; } else { args += GetSoureExpr(context, operation.GetSource(argIndex), dstType); } } if (inst == Instruction.Ballot) { return $"unpackUint2x32({info.OpName}({args})).x"; } else { return info.OpName + "(" + args + ")"; } } else if ((info.Type & InstType.Op) != 0) { string op = info.OpName; // Return may optionally have a return value (and in this case it is unary). if (inst == Instruction.Return && operation.SourcesCount != 0) { return $"{op} {GetSoureExpr(context, operation.GetSource(0), context.CurrentFunction.ReturnType)}"; } int arity = (int)(info.Type & InstType.ArityMask); string[] expr = new string[arity]; for (int index = 0; index < arity; index++) { IAstNode src = operation.GetSource(index); string srcExpr = GetSoureExpr(context, src, GetSrcVarType(inst, index)); bool isLhs = arity == 2 && index == 0; expr[index] = Enclose(srcExpr, src, inst, info, isLhs); } switch (arity) { case 0: return op; case 1: return op + expr[0]; case 2: return $"{expr[0]} {op} {expr[1]}"; case 3: return $"{expr[0]} {op[0]} {expr[1]} {op[1]} {expr[2]}"; } } else if ((info.Type & InstType.Special) != 0) { switch (inst) { case Instruction.Call: return Call(context, operation); case Instruction.ImageLoad: return ImageLoadOrStore(context, operation); case Instruction.ImageStore: return ImageLoadOrStore(context, operation); case Instruction.LoadAttribute: return LoadAttribute(context, operation); case Instruction.LoadConstant: return LoadConstant(context, operation); case Instruction.LoadLocal: return LoadLocal(context, operation); case Instruction.LoadShared: return LoadShared(context, operation); case Instruction.LoadStorage: return LoadStorage(context, operation); case Instruction.Lod: return Lod(context, operation); case Instruction.PackDouble2x32: return PackDouble2x32(context, operation); case Instruction.PackHalf2x16: return PackHalf2x16(context, operation); case Instruction.StoreLocal: return StoreLocal(context, operation); case Instruction.StoreShared: return StoreShared(context, operation); case Instruction.StoreStorage: return StoreStorage(context, operation); case Instruction.TextureSample: return TextureSample(context, operation); case Instruction.TextureSize: return TextureSize(context, operation); case Instruction.UnpackDouble2x32: return UnpackDouble2x32(context, operation); case Instruction.UnpackHalf2x16: return UnpackHalf2x16(context, operation); } } throw new InvalidOperationException($"Unexpected instruction type \"{info.Type}\"."); } } }