public static string Enclose(string expr, IAstNode node, Instruction pInst, InstInfo pInfo, bool isLhs = false)
{
if (NeedsParenthesis(node, pInst, pInfo, isLhs))
{
expr = "(" + expr + ")";
}
return(expr);
}
public static string Negate(CodeGenContext context, AstOperation operation, InstInfo info)
{
IAstNode src = operation.GetSource(0);
VariableType type = GetSrcVarType(operation.Inst, 0);
string srcExpr = GetSoureExpr(context, src, type);
NumberFormatter.TryFormat(0, type, out string zero);
// Starting in the 496.13 NVIDIA driver, there's an issue with assigning variables to negated expressions.
// (-expr) does not work, but (0.0 - expr) does. This should be removed once the issue is resolved.
return($"{zero} - {Enclose(srcExpr, src, operation.Inst, info, false)}");
}
public static bool NeedsParenthesis(IAstNode node, Instruction pInst, InstInfo pInfo, bool isLhs)
{
// If the node isn't a operation, then it can only be a operand,
// and those never needs to be surrounded in parenthesis.
if (!(node is AstOperation operation))
{
// This is sort of a special case, if this is a negative constant,
// and it is consumed by a unary operation, we need to put on the parenthesis,
// as in GLSL a sequence like --2 or ~-1 is not valid.
if (IsNegativeConst(node) && pInfo.Type == InstType.OpUnary)
{
return(true);
}
return(false);
}
if ((pInfo.Type & (InstType.Call | InstType.Special)) != 0)
{
return(false);
}
InstInfo info = _infoTbl[(int)(operation.Inst & Instruction.Mask)];
if ((info.Type & (InstType.Call | InstType.Special)) != 0)
{
return(false);
}
if (info.Precedence < pInfo.Precedence)
{
return(false);
}
if (info.Precedence == pInfo.Precedence && isLhs)
{
return(false);
}
if (pInst == operation.Inst && info.Type == InstType.OpBinaryCom)
{
return(false);
}
return(true);
}
public static string Enclose(string expr, IAstNode node, Instruction pInst, bool isLhs)
{
InstInfo pInfo = GetInstructionInfo(pInst);
return(Enclose(expr, node, pInst, pInfo, isLhs));
}
private static void Add(Instruction inst, InstType flags, string opName = null, int precedence = 0)
{
_infoTbl[(int)inst] = new InstInfo(flags, opName, precedence);
}
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++)
{
// For shared memory access, the second argument is unused and should be ignored.
// It is there to make both storage and shared access have the same number of arguments.
// For storage, both inputs are consumed when the argument index is 0, so we should skip it here.
if (argIndex == 1 && (atomic || (inst & Instruction.MrMask) == Instruction.MrShared))
{
continue;
}
if (argIndex != 0)
{
args += ", ";
}
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}\".");
}
}
else
{
VariableType dstType = GetSrcVarType(inst, argIndex);
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}\".");
}
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;
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.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}\".");
}
private static string GetExpression(CodeGenContext context, AstOperation operation)
{
Instruction inst = operation.Inst;
InstInfo info = GetInstructionInfo(inst);
if ((info.Type & InstType.Call) != 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);
args += GetSoureExpr(context, operation.GetSource(argIndex), dstType);
}
return(info.OpName + "(" + args + ")");
}
else if ((info.Type & InstType.Op) != 0)
{
string op = info.OpName;
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.LoadConstant:
return(InstGenMemory.LoadConstant(context, operation));
case Instruction.PackHalf2x16:
return(InstGenPacking.PackHalf2x16(context, operation));
case Instruction.TextureSample:
return(InstGenMemory.TextureSample(context, operation));
case Instruction.TextureSize:
return(InstGenMemory.TextureSize(context, operation));
case Instruction.UnpackHalf2x16:
return(InstGenPacking.UnpackHalf2x16(context, operation));
}
}
throw new InvalidOperationException($"Unexpected instruction type \"{info.Type}\".");
}
