SHCommon.bslinc 2.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128
  1. mixin SHCommon
  2. {
  3. code
  4. {
  5. struct SHVector5
  6. {
  7. float4 v0;
  8. float4 v1;
  9. float4 v2;
  10. float4 v3;
  11. float4 v4;
  12. float4 v5;
  13. float v6;
  14. };
  15. struct SHVector5RGB
  16. {
  17. SHVector5 R;
  18. SHVector5 G;
  19. SHVector5 B;
  20. };
  21. void SHZero(inout SHVector5 v)
  22. {
  23. v.v0 = 0;
  24. v.v1 = 0;
  25. v.v2 = 0;
  26. v.v3 = 0;
  27. v.v4 = 0;
  28. v.v5 = 0;
  29. v.v6 = 0;
  30. }
  31. void SHMultiplyAdd(inout SHVector5 lhs, SHVector5 rhs, float c)
  32. {
  33. lhs.v0 += rhs.v0 * c;
  34. lhs.v1 += rhs.v1 * c;
  35. lhs.v2 += rhs.v2 * c;
  36. lhs.v3 += rhs.v3 * c;
  37. lhs.v4 += rhs.v4 * c;
  38. lhs.v5 += rhs.v5 * c;
  39. lhs.v6 += rhs.v6 * c;
  40. }
  41. void SHAdd(inout SHVector5 lhs, SHVector5 rhs)
  42. {
  43. lhs.v0 += rhs.v0;
  44. lhs.v1 += rhs.v1;
  45. lhs.v2 += rhs.v2;
  46. lhs.v3 += rhs.v3;
  47. lhs.v4 += rhs.v4;
  48. lhs.v5 += rhs.v5;
  49. lhs.v6 += rhs.v6;
  50. }
  51. void SHMultiply(inout SHVector5 lhs, SHVector5 rhs)
  52. {
  53. lhs.v0 *= rhs.v0;
  54. lhs.v1 *= rhs.v1;
  55. lhs.v2 *= rhs.v2;
  56. lhs.v3 *= rhs.v3;
  57. lhs.v4 *= rhs.v4;
  58. lhs.v5 *= rhs.v5;
  59. lhs.v6 *= rhs.v6;
  60. }
  61. void SHMultiply(inout SHVector5 lhs, float rhs)
  62. {
  63. lhs.v0 *= rhs;
  64. lhs.v1 *= rhs;
  65. lhs.v2 *= rhs;
  66. lhs.v3 *= rhs;
  67. lhs.v4 *= rhs;
  68. lhs.v5 *= rhs;
  69. lhs.v6 *= rhs;
  70. }
  71. SHVector5 SHBasis5(float3 dir)
  72. {
  73. float x = dir.x;
  74. float y = dir.y;
  75. float z = dir.z;
  76. float x2 = x*x;
  77. float y2 = y*y;
  78. float z2 = z*z;
  79. float z3 = z2 * z;
  80. float x4 = x2 * x2;
  81. float y4 = y2 * y2;
  82. float z4 = z2 * z2;
  83. SHVector5 o;
  84. o.v0[0] = 0.282095f;
  85. o.v0[1] = -0.488603f * y;
  86. o.v0[2] = 0.488603f * z;
  87. o.v0[3] = -0.488603f * x;
  88. o.v1[0] = 1.092548f * x * y;
  89. o.v1[1] = -1.092548f * y * z;
  90. o.v1[2] = 0.315392f * (3.0f * z2 - 1.0f);
  91. o.v1[3] = -1.092548f * x * z;
  92. o.v2[0] = 0.546274f * (x2 - y2);
  93. o.v2[1] = -0.590043f * y * (3.0f * x2 - y2);
  94. o.v2[2] = 2.890611f * y * x * z;
  95. o.v2[3] = -0.646360f * y * (-1.0f + 5.0f * z2);
  96. o.v3[0] = 0.373176f *(5.0f * z3 - 3.0f * z);
  97. o.v3[1] = -0.457045f * x * (-1.0f + 5.0f * z2);
  98. o.v3[2] = 1.445306f *(x2 - y2) * z;
  99. o.v3[3] = -0.590043f * x * (x2 - 3.0f * y2);
  100. o.v4[0] = 2.503340f * x * y * (x2 - y2);
  101. o.v4[1] = -1.770130f * y * z * (3.0f * x2 - y2);
  102. o.v4[2] = 0.946175f * y * x * (-1.0f + 7.0f * z2);
  103. o.v4[3] = -0.669046f * y * z * (-3.0f + 7.0f * z2);
  104. o.v5[0] = (105.0f * z4 - 90.0f * z2 + 9.0f) / 28.359261f;
  105. o.v5[1] = -0.669046f * x * z * (-3.0f + 7.0f * z2);
  106. o.v5[2] = 0.473087f * (x2 - y2) * (-1.0f + 7.0f * z2);
  107. o.v5[3] = -1.770130f * x * z * (x2 - 3.0f * y2);
  108. o.v6 = 0.625836f * (x4 - 6.0f * y2 * x2 + y4);
  109. return o;
  110. }
  111. };
  112. };