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.ImageStore:
return(InstGenMemory.ImageStore(context, operation));
case Instruction.LoadAttribute:
return(InstGenMemory.LoadAttribute(context, operation));
case Instruction.LoadConstant:
return(InstGenMemory.LoadConstant(context, operation));
case Instruction.LoadLocal:
return(InstGenMemory.LoadLocal(context, operation));
case Instruction.LoadShared:
return(InstGenMemory.LoadShared(context, operation));
case Instruction.LoadStorage:
return(InstGenMemory.LoadStorage(context, operation));
case Instruction.Lod:
return(InstGenMemory.Lod(context, operation));
case Instruction.PackDouble2x32:
return(InstGenPacking.PackDouble2x32(context, operation));
case Instruction.PackHalf2x16:
return(InstGenPacking.PackHalf2x16(context, operation));
case Instruction.StoreLocal:
return(InstGenMemory.StoreLocal(context, operation));
case Instruction.StoreShared:
return(InstGenMemory.StoreShared(context, operation));
case Instruction.StoreStorage:
return(InstGenMemory.StoreStorage(context, operation));
case Instruction.TextureSample:
return(InstGenMemory.TextureSample(context, operation));
case Instruction.TextureSize:
return(InstGenMemory.TextureSize(context, operation));
case Instruction.UnpackDouble2x32:
return(InstGenPacking.UnpackDouble2x32(context, operation));
case Instruction.UnpackHalf2x16:
return(InstGenPacking.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}\".");
}
