SHCommon.bslinc 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203
  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 SHVector3
  16. {
  17. float4 v0;
  18. float4 v1;
  19. float v2;
  20. };
  21. struct SHVector5RGB
  22. {
  23. SHVector5 R;
  24. SHVector5 G;
  25. SHVector5 B;
  26. };
  27. struct SHVector3RGB
  28. {
  29. SHVector3 R;
  30. SHVector3 G;
  31. SHVector3 B;
  32. };
  33. void SHZero(inout SHVector5 v)
  34. {
  35. v.v0 = 0;
  36. v.v1 = 0;
  37. v.v2 = 0;
  38. v.v3 = 0;
  39. v.v4 = 0;
  40. v.v5 = 0;
  41. v.v6 = 0;
  42. }
  43. void SHZero(inout SHVector3 v)
  44. {
  45. v.v0 = 0;
  46. v.v1 = 0;
  47. v.v2 = 0;
  48. }
  49. void SHMultiplyAdd(inout SHVector5 lhs, SHVector5 rhs, float c)
  50. {
  51. lhs.v0 += rhs.v0 * c;
  52. lhs.v1 += rhs.v1 * c;
  53. lhs.v2 += rhs.v2 * c;
  54. lhs.v3 += rhs.v3 * c;
  55. lhs.v4 += rhs.v4 * c;
  56. lhs.v5 += rhs.v5 * c;
  57. lhs.v6 += rhs.v6 * c;
  58. }
  59. void SHMultiplyAdd(inout SHVector3 lhs, SHVector3 rhs, float c)
  60. {
  61. lhs.v0 += rhs.v0 * c;
  62. lhs.v1 += rhs.v1 * c;
  63. lhs.v2 += rhs.v2 * c;
  64. }
  65. void SHAdd(inout SHVector5 lhs, SHVector5 rhs)
  66. {
  67. lhs.v0 += rhs.v0;
  68. lhs.v1 += rhs.v1;
  69. lhs.v2 += rhs.v2;
  70. lhs.v3 += rhs.v3;
  71. lhs.v4 += rhs.v4;
  72. lhs.v5 += rhs.v5;
  73. lhs.v6 += rhs.v6;
  74. }
  75. void SHAdd(inout SHVector3 lhs, SHVector3 rhs)
  76. {
  77. lhs.v0 += rhs.v0;
  78. lhs.v1 += rhs.v1;
  79. lhs.v2 += rhs.v2;
  80. }
  81. void SHMultiply(inout SHVector5 lhs, SHVector5 rhs)
  82. {
  83. lhs.v0 *= rhs.v0;
  84. lhs.v1 *= rhs.v1;
  85. lhs.v2 *= rhs.v2;
  86. lhs.v3 *= rhs.v3;
  87. lhs.v4 *= rhs.v4;
  88. lhs.v5 *= rhs.v5;
  89. lhs.v6 *= rhs.v6;
  90. }
  91. void SHMultiply(inout SHVector3 lhs, SHVector3 rhs)
  92. {
  93. lhs.v0 *= rhs.v0;
  94. lhs.v1 *= rhs.v1;
  95. lhs.v2 *= rhs.v2;
  96. }
  97. void SHMultiply(inout SHVector5 lhs, float rhs)
  98. {
  99. lhs.v0 *= rhs;
  100. lhs.v1 *= rhs;
  101. lhs.v2 *= rhs;
  102. lhs.v3 *= rhs;
  103. lhs.v4 *= rhs;
  104. lhs.v5 *= rhs;
  105. lhs.v6 *= rhs;
  106. }
  107. void SHMultiply(inout SHVector3 lhs, float rhs)
  108. {
  109. lhs.v0 *= rhs;
  110. lhs.v1 *= rhs;
  111. lhs.v2 *= rhs;
  112. }
  113. SHVector5 SHBasis5(float3 dir)
  114. {
  115. float x = dir.x;
  116. float y = dir.y;
  117. float z = dir.z;
  118. float x2 = x*x;
  119. float y2 = y*y;
  120. float z2 = z*z;
  121. float z3 = z2 * z;
  122. float x4 = x2 * x2;
  123. float y4 = y2 * y2;
  124. float z4 = z2 * z2;
  125. SHVector5 o;
  126. o.v0[0] = 0.282095f;
  127. o.v0[1] = -0.488603f * y;
  128. o.v0[2] = 0.488603f * z;
  129. o.v0[3] = -0.488603f * x;
  130. o.v1[0] = 1.092548f * x * y;
  131. o.v1[1] = -1.092548f * y * z;
  132. o.v1[2] = 0.315392f * (3.0f * z2 - 1.0f);
  133. o.v1[3] = -1.092548f * x * z;
  134. o.v2[0] = 0.546274f * (x2 - y2);
  135. o.v2[1] = -0.590043f * y * (3.0f * x2 - y2);
  136. o.v2[2] = 2.890611f * y * x * z;
  137. o.v2[3] = -0.646360f * y * (-1.0f + 5.0f * z2);
  138. o.v3[0] = 0.373176f *(5.0f * z3 - 3.0f * z);
  139. o.v3[1] = -0.457045f * x * (-1.0f + 5.0f * z2);
  140. o.v3[2] = 1.445306f *(x2 - y2) * z;
  141. o.v3[3] = -0.590043f * x * (x2 - 3.0f * y2);
  142. o.v4[0] = 2.503340f * x * y * (x2 - y2);
  143. o.v4[1] = -1.770130f * y * z * (3.0f * x2 - y2);
  144. o.v4[2] = 0.946175f * y * x * (-1.0f + 7.0f * z2);
  145. o.v4[3] = -0.669046f * y * z * (-3.0f + 7.0f * z2);
  146. o.v5[0] = (105.0f * z4 - 90.0f * z2 + 9.0f) / 28.359261f;
  147. o.v5[1] = -0.669046f * x * z * (-3.0f + 7.0f * z2);
  148. o.v5[2] = 0.473087f * (x2 - y2) * (-1.0f + 7.0f * z2);
  149. o.v5[3] = -1.770130f * x * z * (x2 - 3.0f * y2);
  150. o.v6 = 0.625836f * (x4 - 6.0f * y2 * x2 + y4);
  151. return o;
  152. }
  153. SHVector3 SHBasis3(float3 dir)
  154. {
  155. float x = dir.x;
  156. float y = dir.y;
  157. float z = dir.z;
  158. float x2 = x*x;
  159. float y2 = y*y;
  160. float z2 = z*z;
  161. SHVector3 o;
  162. o.v0[0] = 0.282095f;
  163. o.v0[1] = -0.488603f * y;
  164. o.v0[2] = 0.488603f * z;
  165. o.v0[3] = -0.488603f * x;
  166. o.v1[0] = 1.092548f * x * y;
  167. o.v1[1] = -1.092548f * y * z;
  168. o.v1[2] = 0.315392f * (3.0f * z2 - 1.0f);
  169. o.v1[3] = -1.092548f * x * z;
  170. o.v2 = 0.546274f * (x2 - y2);
  171. return o;
  172. }
  173. };
  174. };