2024-08-18

pedal to the metal——HLSL

HLSL 是与 DirectX 中的可编程着色器一起使用的类似 C 的高级着色器语言。

例如，可以使用 HLSL 编写 顶点着色器(Vertex shader)或 像素着色器，并在 Direct3D 应用程序中的呈现器实现中使用这些着色器。

或者，可以使用 HLSL 编写计算着色器，也许可以实现物理模拟。 但是，例如，如果你倾向于编写自己的卷积运算符 (，以便在计算着色器中将图像处理) 为 HLSL，则在这种情况下，如果改用 Direct Machine Learning (DirectML) ，你将获得更好的性能。

基本数据类型

注意：一些平台可能不支持int, half和double，如果出现这些情况将会使用float来模拟

向量（Vectors） in HLSL

向量类型可以支持2到4个同类元素

一种表示方式是使用类似模板的形式来描述

vector<float, 4> vec1;  // 向量vec1包含4个float元素

vector<int, 2> vec2; // 向量vec2包含2个int元素

另一种方式则是直接在基本类型后面加上数字

float4 vec1;    // 向量vec1包含4个float元素

int3 vec2; // 向量vec2包含3个int元素

当然，只使用vector本身则表示为一种包含4个float元素的类型

vector vec1;\t// 向量vec1包含4个float元素

向量的初始化方式

float2 vec0 = {0.0f, 1.0f};

float3 vec1 = float3(0.0f, 0.1f, 0.2f); float4 vec2 = float4(vec1, 1.0f);

向量的索引方式

在HLSL中，向量具有三种重要的索引方式，包括xyzw,rgba与数组下标三种手段，具体如下

float4 vec0 = {1.0f, 2.0f, 3.0f, 0.0f};

float f0 = vec0.x; // 1.0f float f1 = vec0.g; // 2.0f float f2 = vec0[2]; // 3.0f vec0.a = 4.0f;\t\t// 4.0f

我们还可以使用swizzles的方式来进行赋值，可以一次性提供多个分量进行赋值操作，这些分量的名称可以重复出现：

float4 vec0 = {1.0f, 2.0f, 3.0f, 4.0f}; 

float3 vec1 = vec0.xyz; // (1.0f, 2.0f, 3.0f) float2 vec2 = vec0.rg; // (1.0f, 2.0f) float4 vec3 = vec0.zzxy; // (3.0f, 3.0f, 1.0f, 2.0f) vec3.wxyz = vec3; // (3.0f, 1.0f, 2.0f, 3.0f) vec3.yw = vec1.yy; // (3.0f, 2.0f, 2.0f, 2.0f)

矩阵

矩阵的基本类型如下：

float1x1 float1x2 float1x3 float1x4
float2x1 float2x2 float2x3 float2x4
float3x1 float3x2 float3x3 float3x4
float4x1 float4x2 float4x3 float4x4

此外，我们也可以使用类似模板的形式来描述：

matrix<float, 2, 2> mat1;\t// float2x2

矩阵的初始化

float2x2 mat1 = {

\t1.0f, 2.0f,\t// 第一行 \t3.0f, 4.0f // 第二行 }; float3x3 TBN = float3x3(T, B, N); // T, B, N都是float3

矩阵的取值则是如下操作：

matrix M;

// ... float f0 = M._m00; // 第一行第一列元素(索引从0开始) float f1 = M._12; // 第一行第二列元素(索引从1开始) float f2 = M[0][1]; // 第一行第二列元素(索引从0开始) float2 f3 = M._11_12; // Swizzles

默认的*法都是逐位相乘，所以需要自己做额外的实现才能进行x乘等复杂操作

float4 vec0 = 2.0f * float4(1.0f, 2.0f, 3.0f, 4.0f);    //(2.0f, 4.0f, 6.0f, 8.0f)

float4 vec1 = vec0 * float4(1.0f, 0.2f, 0.1f, 0.0f); //(2.0f, 0.8f, 0.6f, 0.0f)

若要进行向量与矩阵的乘法，则需要使用mul函数。

在C++代码层中，DirectXMath数学库创建的矩阵都是行矩阵，但当矩阵从C++传递给HLSL时，HLSL默认是列矩阵的，看起来就好像传递的过程中进行了一次转置那样。如果希望不发生转置操作的话，可以添加修饰关键字row_major：

row_major matrix M;

数组

和C++一样，我们可以声明数组：

float M[4][4];

int p[4]; float3 v[12];\t// 12个3D向量

结构体(struct)

HLSL的结构体和C/C++的十分相似，它可以存放任意数目的标量，向量和矩阵类型，除此之外，它还可以存放数组或者别的结构体类型。结构体的成员访问也和C/C++相似：

struct A

{ float4 vec; }; struct B { int scalar; float4 vec; float4x4 mat; float arr[8]; A a; }; // ... B b; b.vec = float4(1.0f, 2.0f, 3.0f, 4.0f);

变量的修饰符

HLSL有着极其灵活的类型转换机制。HLSL中的类型转换语法和C/C++的相同。下面是一些例子：

float f = 4.0f;

float4x4 m = (float4x4)f;\t// 将浮点数f复制到矩阵m的每一个元素当中 float3 n = float3(...); float3 v = 2.0f * n - 1.0f;\t// 这里1.0f将会隐式转换成(1.0f, 1.0f, 1.0f) float4x4 WInvT = float4x4(...); float3x3 mat = (float3x3)WInvT;\t// 只取4x4矩阵的前3行前3列

控制流语句

HLSL也支持if, else, continue, break, switch关键字，此外discard关键字用于像素着色阶段抛弃该像素。

条件的判断使用一个布尔值进行，通常由各种逻辑运算符或者比较运算符操作得到。注意向量之间的比较或者逻辑操作是得到一个存有布尔值的向量，不能够直接用于条件判断，也不能用于switch语句。

判断与动态分支

基于值的条件分支只有在程序执行的时候被编译好的着色器汇编成两种方式：判断(predication) 和动态分支(dynamic branching) 。

如果使用的是判断的形式，编译器会提前计算两个不同分支下表达式的值。然后使用比较指令来基于比较结果来"选择"正确的值。

而动态分支使用的是跳转指令来避免一些非必要的计算和内存访问。

着色器程序在同时执行的时候应当选择相同的分支，以防止硬件在分支的两边执行。通常情况下，硬件会同时将一系列连续的顶点数据传入到顶点着色器并行计算，或者是一系列连续的像素单元传入到像素着色器同时运算等。

动态分支会由于执行分支指令所带来的开销而导致一定的性能损失，因此要权衡动态分支的开销和可以跳过的指令数目。

通常情况下编译器会自行选择使用判断还是动态分支，但我们可以通过重写某些属性来修改编译器的行为。我们可以在条件语句前可以选择添加下面两个属性之一：

用法如下：

[flatten]

if (x) { x = sqrt(x); }

函数

函数的语法也和C/C++的十分类似，但它具有以下属性：

1. 参数只能按值传递

2. 不支持递归

3. 只有内联函数（避免产生调用的跳转来减小开销）

此外，HLSL函数的形参可以指定输入/输出类别：

例如：

bool foo(in bool b,\t\t\t// 输入的bool类型参数

\tout int r1,\t\t\t\t// 输出的int类型参数 \tinout float r2)\t\t\t// 具备输入/输出的float类型参数 { if (b) { f1 = 5; } else { r1 = 1; }

// 注意r1不能出现在等式的右边

// r2既可以被读入，也可以写出结果到外面的实参上
r2 = r2 * r2 * r2;

return true;

}

语义

语义通常是附加在着色器输入/输出参数上的字符串。它在着色器程序的用途如下：

1. 用于描述传递给着色器程序的变量参数的含义

2. 允许着色器程序接受由渲染管线生成的特殊系统值

3. 允许着色器程序传递由渲染管线解释的特殊系统值

顶点着色器语义

n是一个可选的整数，从0开始。比如POSITION0, TEXCOORD1等等。

像素着色器语义

系统值语义

所有的系统值都包含前缀SV_。这些系统值将用于某些着色器的特定用途（并未全部列出）

所有的可编程着色器阶段使用通用着色器核心来实现相同的基础功能。此外，顶点着色阶段、几何着色阶段和像素着色阶段则提供了独特的功能，例如几何着色阶段可以生成新的图元或删减图元，像素着色阶段可以决定当前像素是否被抛弃等。下图展示了数据是怎么流向一个着色阶段，以及通用着色器核心与着色器内存资源之间的关系：

Input Data ：顶点着色器从输入装配阶段获取数据；几何着色器则从上一个着色阶段的输出获取等等。通过给形参引入可以使用的系统值可以提供额外的输入

Output Data ：着色器生成输出的结果然后传递给管线的下一个阶段。有些输出会被通用着色器核心解释成特定用途（如顶点位置、渲染目标对应位置的值），另外一些输出则由应用程序来解释。

Shader Code ：着色器代码可以从内存读取，然后用于执行代码中所期望的内容。

Samplers ：采样器决定了如何对纹理进行采样和滤波。

Textures ：纹理可以使用采样器进行采样，也可以基于索引的方式按像素读取。

Buffers ：缓冲区可以使用读取相关的内置函数，在内存中按元素直接读取。

Constant Buffers ：常量缓冲区对常量值的读取有所优化。他们被设计用于CPU对这些数据的频繁更新，因此他们有额外的大小、布局和访问限制。

着色器常量

着色器常量存在内存中的一个或多个缓冲区资源当中。他们可以被组织成两种类型的缓冲区：常量缓冲区（cbuffers）和纹理缓冲区（tbuffers）。关于纹理缓冲区，我们不在这讨论。

常量缓冲区(Constant Buffer)

常量缓冲区允许C++端将数据传递给HLSL中使用，在HLSL端，这些传递过来的数据不可更改，因而是常量。常量缓冲区对这种使用方式有所优化，表现为低延迟的访问和允许来自CPU的频繁更新，因此他们有额外的大小、布局和访问限制。

声明方式如下：

cbuffer VSConstants

{ float4x4 g_WorldViewProj; fioat3 g_Color; uint g_EnableFog; float2 g_ViewportXY; float2 g_ViewportWH; }

由于我们写的是原生HLSL，当我们在HLSL中声明常量缓冲区时，还需要在HLSL的声明中使用关键字register手动指定对应的寄存器索引，然后编译器会为对应的着色器阶段自动将其映射到15个常量缓冲寄存器的其中一个位置。这些寄存器的名字为b0到b14：

cbuffer VSConstants : register(b0)

{ float4x4 g_WorldViewProj; fioat3 g_Color; uint g_EnableFog; float2 g_ViewportXY; float2 g_ViewportWH; }

在C++端是通过ID3D11DeviceContext::*SSetConstantBuffers指定特定的槽(slot)来给某一着色器阶段对应的寄存器索引提供常量缓冲区的数据。

如果是存在多个不同的着色器阶段使用同一个常量缓冲区，那就需要分别给这两个着色器阶段设置好相同的数据。

综合前面几节内容，下面演示了顶点着色器和常量缓冲区的用法：

cbuffer ConstantBuffer : register(b0)

{ float4x4 g_WorldViewProj; }

void VS_Main( in float4 inPos : POSITION, // 绑定变量到输入装配器 in uint VID : SV_VertexID, // 绑定变量到系统生成值 out float4 outPos : SV_Position) // 告诉管线将该值解释为输出的顶点位置 { outPos = mul(inPos, g_WorldViewProj); }

上面的代码也可以写成：

cbuffer ConstantBuffer : register(b0)

{ float4x4 g_WorldViewProj; } struct VertexIn { \tfloat4 inPos : POSITION;\t// 源自输入装配器 \tuint VID : SV_VertexID;\t\t// 源自系统生成值 }; float4 VS_Main(VertexIn vIn) : SV_Position { return mul(vIn.inPos, g_WorldViewProj); }

有关常量缓冲区的打包规则，建议在阅读到时索引缓冲区、常量缓冲区一章时，再来参考杂项篇的HLSL常量缓冲区的打包规则。