blob: 9e88d2ce9778ba13abcf50b9fd3cf35b6ecdd6ef (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
|
//TEST_IGNORE_FILE:
__exported import IMaterial;
__exported import IMaterialInstance;
import MxLayeredMaterialInstance;
import MxWeights;
public struct LayeredData_mixedLobes : IMxLayeredMaterialData
{
public static const int bsdfCount = 3;
public static const int layerCount = 1;
public int getBsdfCount() { return bsdfCount; }
public int getLayerCount() { return layerCount; }
}
#define TMxLayeredMaterialData LayeredData_mixedLobes
public struct Layered_mixedLobes_WeightsCalculator : IMxLayeredWeightCalculator
{
public void calculateWeights<let TBsdfCount : int, let TLayerCount : int, TLayeredMaterialData : IMxLayeredMaterialData>(
const TLayeredMaterialData data, out MxWeights<TBsdfCount> weights)
{
for (uint i = 0; i < TBsdfCount; i++) weights.weights[i] = float3(0.f);
}
}
public struct MxLayeredMaterial : IMaterial
{
public float3 baseColor;
public typealias UsedMaterialInstance = MxLayeredMaterialInstance<3, 1, TMxLayeredMaterialData, Layered_mixedLobes_WeightsCalculator>;
public typedef UsedMaterialInstance MaterialInstance;
public float3 getAlbedo(float3 albedo)
{
return albedo;
}
[ForwardDerivativeOf(getAlbedo)]
[TreatAsDifferentiable]
public DifferentialPair<float3> __fwd_d_getAlbedo(DifferentialPair<float3> dpAlbedo)
{
return diffPair(dpAlbedo.p, float3(1.f));
}
[BackwardDerivativeOf(getAlbedo)]
[TreatAsDifferentiable]
public void __bwd_d_getAlbedo(inout DifferentialPair<float3> dpAlbedo, float3 dOut)
{
[unroll]
for (int j = 0; j < 3; j++) outputBuffer[j + 6] += dOut[j];
}
[Differentiable]
[NoDiffThis]
public UsedMaterialInstance setupMaterialInstance(out MaterialInstanceData miData)
{
float3 albedo = getAlbedo(baseColor);
UsedMaterialInstance mi;
for (uint i = 0; i < 3; i++) miData.data[i] = albedo[i];
// unused
miData.data[3] = 0.f;
miData.data[4] = 0.f;
return mi;
}
}
|
