InstructionInfo.cs 14 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201
  1. using Ryujinx.Graphics.Shader.IntermediateRepresentation;
  2. using System;
  3. namespace Ryujinx.Graphics.Shader.StructuredIr
  4. {
  5. static class InstructionInfo
  6. {
  7. private struct InstInfo
  8. {
  9. public VariableType DestType { get; }
  10. public VariableType[] SrcTypes { get; }
  11. public InstInfo(VariableType destType, params VariableType[] srcTypes)
  12. {
  13. DestType = destType;
  14. SrcTypes = srcTypes;
  15. }
  16. }
  17. private static InstInfo[] _infoTbl;
  18. static InstructionInfo()
  19. {
  20. _infoTbl = new InstInfo[(int)Instruction.Count];
  21. // Inst Destination type Source 1 type Source 2 type Source 3 type Source 4 type
  22. Add(Instruction.AtomicAdd, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32);
  23. Add(Instruction.AtomicAnd, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32);
  24. Add(Instruction.AtomicCompareAndSwap, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32, VariableType.U32);
  25. Add(Instruction.AtomicMaxS32, VariableType.S32, VariableType.S32, VariableType.S32, VariableType.S32);
  26. Add(Instruction.AtomicMaxU32, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32);
  27. Add(Instruction.AtomicMinS32, VariableType.S32, VariableType.S32, VariableType.S32, VariableType.S32);
  28. Add(Instruction.AtomicMinU32, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32);
  29. Add(Instruction.AtomicOr, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32);
  30. Add(Instruction.AtomicSwap, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32);
  31. Add(Instruction.AtomicXor, VariableType.U32, VariableType.S32, VariableType.S32, VariableType.U32);
  32. Add(Instruction.Absolute, VariableType.Scalar, VariableType.Scalar);
  33. Add(Instruction.Add, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  34. Add(Instruction.Ballot, VariableType.U32, VariableType.Bool);
  35. Add(Instruction.BitCount, VariableType.Int, VariableType.Int);
  36. Add(Instruction.BitfieldExtractS32, VariableType.S32, VariableType.S32, VariableType.S32, VariableType.S32);
  37. Add(Instruction.BitfieldExtractU32, VariableType.U32, VariableType.U32, VariableType.S32, VariableType.S32);
  38. Add(Instruction.BitfieldInsert, VariableType.Int, VariableType.Int, VariableType.Int, VariableType.S32, VariableType.S32);
  39. Add(Instruction.BitfieldReverse, VariableType.Int, VariableType.Int);
  40. Add(Instruction.BitwiseAnd, VariableType.Int, VariableType.Int, VariableType.Int);
  41. Add(Instruction.BitwiseExclusiveOr, VariableType.Int, VariableType.Int, VariableType.Int);
  42. Add(Instruction.BitwiseNot, VariableType.Int, VariableType.Int);
  43. Add(Instruction.BitwiseOr, VariableType.Int, VariableType.Int, VariableType.Int);
  44. Add(Instruction.BranchIfTrue, VariableType.None, VariableType.Bool);
  45. Add(Instruction.BranchIfFalse, VariableType.None, VariableType.Bool);
  46. Add(Instruction.Ceiling, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  47. Add(Instruction.Clamp, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  48. Add(Instruction.ClampU32, VariableType.U32, VariableType.U32, VariableType.U32, VariableType.U32);
  49. Add(Instruction.CompareEqual, VariableType.Bool, VariableType.Scalar, VariableType.Scalar);
  50. Add(Instruction.CompareGreater, VariableType.Bool, VariableType.Scalar, VariableType.Scalar);
  51. Add(Instruction.CompareGreaterOrEqual, VariableType.Bool, VariableType.Scalar, VariableType.Scalar);
  52. Add(Instruction.CompareGreaterOrEqualU32, VariableType.Bool, VariableType.U32, VariableType.U32);
  53. Add(Instruction.CompareGreaterU32, VariableType.Bool, VariableType.U32, VariableType.U32);
  54. Add(Instruction.CompareLess, VariableType.Bool, VariableType.Scalar, VariableType.Scalar);
  55. Add(Instruction.CompareLessOrEqual, VariableType.Bool, VariableType.Scalar, VariableType.Scalar);
  56. Add(Instruction.CompareLessOrEqualU32, VariableType.Bool, VariableType.U32, VariableType.U32);
  57. Add(Instruction.CompareLessU32, VariableType.Bool, VariableType.U32, VariableType.U32);
  58. Add(Instruction.CompareNotEqual, VariableType.Bool, VariableType.Scalar, VariableType.Scalar);
  59. Add(Instruction.ConditionalSelect, VariableType.Scalar, VariableType.Bool, VariableType.Scalar, VariableType.Scalar);
  60. Add(Instruction.ConvertFP32ToFP64, VariableType.F64, VariableType.F32);
  61. Add(Instruction.ConvertFP64ToFP32, VariableType.F32, VariableType.F64);
  62. Add(Instruction.ConvertFPToS32, VariableType.S32, VariableType.F32);
  63. Add(Instruction.ConvertFPToU32, VariableType.U32, VariableType.F32);
  64. Add(Instruction.ConvertS32ToFP, VariableType.F32, VariableType.S32);
  65. Add(Instruction.ConvertU32ToFP, VariableType.F32, VariableType.U32);
  66. Add(Instruction.Cosine, VariableType.Scalar, VariableType.Scalar);
  67. Add(Instruction.Ddx, VariableType.F32, VariableType.F32);
  68. Add(Instruction.Ddy, VariableType.F32, VariableType.F32);
  69. Add(Instruction.Divide, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  70. Add(Instruction.ExponentB2, VariableType.Scalar, VariableType.Scalar);
  71. Add(Instruction.FindFirstSetS32, VariableType.S32, VariableType.S32);
  72. Add(Instruction.FindFirstSetU32, VariableType.S32, VariableType.U32);
  73. Add(Instruction.Floor, VariableType.Scalar, VariableType.Scalar);
  74. Add(Instruction.FusedMultiplyAdd, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  75. Add(Instruction.ImageLoad, VariableType.F32);
  76. Add(Instruction.ImageStore, VariableType.None);
  77. Add(Instruction.IsNan, VariableType.Bool, VariableType.F32);
  78. Add(Instruction.LoadAttribute, VariableType.F32, VariableType.S32, VariableType.S32);
  79. Add(Instruction.LoadConstant, VariableType.F32, VariableType.S32, VariableType.S32);
  80. Add(Instruction.LoadGlobal, VariableType.U32, VariableType.S32, VariableType.S32);
  81. Add(Instruction.LoadLocal, VariableType.U32, VariableType.S32);
  82. Add(Instruction.LoadShared, VariableType.U32, VariableType.S32);
  83. Add(Instruction.LoadStorage, VariableType.U32, VariableType.S32, VariableType.S32);
  84. Add(Instruction.Lod, VariableType.F32);
  85. Add(Instruction.LogarithmB2, VariableType.Scalar, VariableType.Scalar);
  86. Add(Instruction.LogicalAnd, VariableType.Bool, VariableType.Bool, VariableType.Bool);
  87. Add(Instruction.LogicalExclusiveOr, VariableType.Bool, VariableType.Bool, VariableType.Bool);
  88. Add(Instruction.LogicalNot, VariableType.Bool, VariableType.Bool);
  89. Add(Instruction.LogicalOr, VariableType.Bool, VariableType.Bool, VariableType.Bool);
  90. Add(Instruction.ShiftLeft, VariableType.Int, VariableType.Int, VariableType.Int);
  91. Add(Instruction.ShiftRightS32, VariableType.S32, VariableType.S32, VariableType.Int);
  92. Add(Instruction.ShiftRightU32, VariableType.U32, VariableType.U32, VariableType.Int);
  93. Add(Instruction.Shuffle, VariableType.F32, VariableType.F32, VariableType.U32, VariableType.U32);
  94. Add(Instruction.ShuffleDown, VariableType.F32, VariableType.F32, VariableType.U32, VariableType.U32);
  95. Add(Instruction.ShuffleUp, VariableType.F32, VariableType.F32, VariableType.U32, VariableType.U32);
  96. Add(Instruction.ShuffleXor, VariableType.F32, VariableType.F32, VariableType.U32, VariableType.U32);
  97. Add(Instruction.Maximum, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  98. Add(Instruction.MaximumU32, VariableType.U32, VariableType.U32, VariableType.U32);
  99. Add(Instruction.Minimum, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  100. Add(Instruction.MinimumU32, VariableType.U32, VariableType.U32, VariableType.U32);
  101. Add(Instruction.Multiply, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  102. Add(Instruction.MultiplyHighS32, VariableType.S32, VariableType.S32, VariableType.S32);
  103. Add(Instruction.MultiplyHighU32, VariableType.U32, VariableType.U32, VariableType.U32);
  104. Add(Instruction.Negate, VariableType.Scalar, VariableType.Scalar);
  105. Add(Instruction.PackDouble2x32, VariableType.F64, VariableType.U32, VariableType.U32);
  106. Add(Instruction.PackHalf2x16, VariableType.U32, VariableType.F32, VariableType.F32);
  107. Add(Instruction.ReciprocalSquareRoot, VariableType.Scalar, VariableType.Scalar);
  108. Add(Instruction.Round, VariableType.Scalar, VariableType.Scalar);
  109. Add(Instruction.Sine, VariableType.Scalar, VariableType.Scalar);
  110. Add(Instruction.SquareRoot, VariableType.Scalar, VariableType.Scalar);
  111. Add(Instruction.StoreGlobal, VariableType.None, VariableType.S32, VariableType.S32, VariableType.U32);
  112. Add(Instruction.StoreLocal, VariableType.None, VariableType.S32, VariableType.U32);
  113. Add(Instruction.StoreShared, VariableType.None, VariableType.S32, VariableType.U32);
  114. Add(Instruction.StoreStorage, VariableType.None, VariableType.S32, VariableType.S32, VariableType.U32);
  115. Add(Instruction.Subtract, VariableType.Scalar, VariableType.Scalar, VariableType.Scalar);
  116. Add(Instruction.SwizzleAdd, VariableType.F32, VariableType.F32, VariableType.F32, VariableType.S32);
  117. Add(Instruction.TextureSample, VariableType.F32);
  118. Add(Instruction.TextureSize, VariableType.S32, VariableType.S32, VariableType.S32);
  119. Add(Instruction.Truncate, VariableType.Scalar, VariableType.Scalar);
  120. Add(Instruction.UnpackDouble2x32, VariableType.U32, VariableType.F64);
  121. Add(Instruction.UnpackHalf2x16, VariableType.F32, VariableType.U32);
  122. Add(Instruction.VoteAll, VariableType.Bool, VariableType.Bool);
  123. Add(Instruction.VoteAllEqual, VariableType.Bool, VariableType.Bool);
  124. Add(Instruction.VoteAny, VariableType.Bool, VariableType.Bool);
  125. }
  126. private static void Add(Instruction inst, VariableType destType, params VariableType[] srcTypes)
  127. {
  128. _infoTbl[(int)inst] = new InstInfo(destType, srcTypes);
  129. }
  130. public static VariableType GetDestVarType(Instruction inst)
  131. {
  132. return GetFinalVarType(_infoTbl[(int)(inst & Instruction.Mask)].DestType, inst);
  133. }
  134. public static VariableType GetSrcVarType(Instruction inst, int index)
  135. {
  136. // TODO: Return correct type depending on source index,
  137. // that can improve the decompiler output.
  138. if (inst == Instruction.ImageLoad ||
  139. inst == Instruction.ImageStore ||
  140. inst == Instruction.Lod ||
  141. inst == Instruction.TextureSample)
  142. {
  143. return VariableType.F32;
  144. }
  145. return GetFinalVarType(_infoTbl[(int)(inst & Instruction.Mask)].SrcTypes[index], inst);
  146. }
  147. private static VariableType GetFinalVarType(VariableType type, Instruction inst)
  148. {
  149. if (type == VariableType.Scalar)
  150. {
  151. if ((inst & Instruction.FP32) != 0)
  152. {
  153. return VariableType.F32;
  154. }
  155. else if ((inst & Instruction.FP64) != 0)
  156. {
  157. return VariableType.F64;
  158. }
  159. else
  160. {
  161. return VariableType.S32;
  162. }
  163. }
  164. else if (type == VariableType.Int)
  165. {
  166. return VariableType.S32;
  167. }
  168. else if (type == VariableType.None)
  169. {
  170. throw new ArgumentException($"Invalid operand for instruction \"{inst}\".");
  171. }
  172. return type;
  173. }
  174. public static bool IsUnary(Instruction inst)
  175. {
  176. if (inst == Instruction.Copy)
  177. {
  178. return true;
  179. }
  180. else if (inst == Instruction.TextureSample)
  181. {
  182. return false;
  183. }
  184. return _infoTbl[(int)(inst & Instruction.Mask)].SrcTypes.Length == 1;
  185. }
  186. }
  187. }