SHCommon.bslinc 5.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273
  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 SHVector5RGB v)
  44. {
  45. SHZero(v.R);
  46. SHZero(v.G);
  47. SHZero(v.B);
  48. }
  49. void SHZero(inout SHVector3 v)
  50. {
  51. v.v0 = 0;
  52. v.v1 = 0;
  53. v.v2 = 0;
  54. }
  55. void SHZero(inout SHVector3RGB v)
  56. {
  57. SHZero(v.R);
  58. SHZero(v.G);
  59. SHZero(v.B);
  60. }
  61. void SHMultiplyAdd(inout SHVector5 lhs, SHVector5 rhs, float c)
  62. {
  63. lhs.v0 += rhs.v0 * c;
  64. lhs.v1 += rhs.v1 * c;
  65. lhs.v2 += rhs.v2 * c;
  66. lhs.v3 += rhs.v3 * c;
  67. lhs.v4 += rhs.v4 * c;
  68. lhs.v5 += rhs.v5 * c;
  69. lhs.v6 += rhs.v6 * c;
  70. }
  71. void SHMultiplyAdd(inout SHVector3 lhs, SHVector3 rhs, float c)
  72. {
  73. lhs.v0 += rhs.v0 * c;
  74. lhs.v1 += rhs.v1 * c;
  75. lhs.v2 += rhs.v2 * c;
  76. }
  77. void SHMultiplyAdd(inout SHVector5RGB lhs, SHVector5RGB rhs, float c)
  78. {
  79. SHMultiplyAdd(lhs.R, rhs.R, c);
  80. SHMultiplyAdd(lhs.G, rhs.G, c);
  81. SHMultiplyAdd(lhs.B, rhs.B, c);
  82. }
  83. void SHMultiplyAdd(inout SHVector3RGB lhs, SHVector3RGB rhs, float c)
  84. {
  85. SHMultiplyAdd(lhs.R, rhs.R, c);
  86. SHMultiplyAdd(lhs.G, rhs.G, c);
  87. SHMultiplyAdd(lhs.B, rhs.B, c);
  88. }
  89. void SHAdd(inout SHVector5 lhs, SHVector5 rhs)
  90. {
  91. lhs.v0 += rhs.v0;
  92. lhs.v1 += rhs.v1;
  93. lhs.v2 += rhs.v2;
  94. lhs.v3 += rhs.v3;
  95. lhs.v4 += rhs.v4;
  96. lhs.v5 += rhs.v5;
  97. lhs.v6 += rhs.v6;
  98. }
  99. void SHAdd(inout SHVector3 lhs, SHVector3 rhs)
  100. {
  101. lhs.v0 += rhs.v0;
  102. lhs.v1 += rhs.v1;
  103. lhs.v2 += rhs.v2;
  104. }
  105. void SHAdd(inout SHVector5RGB lhs, SHVector5RGB rhs)
  106. {
  107. SHAdd(lhs.R, rhs.R);
  108. SHAdd(lhs.G, rhs.G);
  109. SHAdd(lhs.B, rhs.B);
  110. }
  111. void SHAdd(inout SHVector3RGB lhs, SHVector3RGB rhs)
  112. {
  113. SHAdd(lhs.R, rhs.R);
  114. SHAdd(lhs.G, rhs.G);
  115. SHAdd(lhs.B, rhs.B);
  116. }
  117. void SHMultiply(inout SHVector5 lhs, SHVector5 rhs)
  118. {
  119. lhs.v0 *= rhs.v0;
  120. lhs.v1 *= rhs.v1;
  121. lhs.v2 *= rhs.v2;
  122. lhs.v3 *= rhs.v3;
  123. lhs.v4 *= rhs.v4;
  124. lhs.v5 *= rhs.v5;
  125. lhs.v6 *= rhs.v6;
  126. }
  127. void SHMultiply(inout SHVector3 lhs, SHVector3 rhs)
  128. {
  129. lhs.v0 *= rhs.v0;
  130. lhs.v1 *= rhs.v1;
  131. lhs.v2 *= rhs.v2;
  132. }
  133. void SHMultiply(inout SHVector5RGB lhs, SHVector5RGB rhs)
  134. {
  135. SHMultiply(lhs.R, rhs.R);
  136. SHMultiply(lhs.G, rhs.G);
  137. SHMultiply(lhs.B, rhs.B);
  138. }
  139. void SHMultiply(inout SHVector3RGB lhs, SHVector3RGB rhs)
  140. {
  141. SHMultiply(lhs.R, rhs.R);
  142. SHMultiply(lhs.G, rhs.G);
  143. SHMultiply(lhs.B, rhs.B);
  144. }
  145. void SHMultiply(inout SHVector5 lhs, float rhs)
  146. {
  147. lhs.v0 *= rhs;
  148. lhs.v1 *= rhs;
  149. lhs.v2 *= rhs;
  150. lhs.v3 *= rhs;
  151. lhs.v4 *= rhs;
  152. lhs.v5 *= rhs;
  153. lhs.v6 *= rhs;
  154. }
  155. void SHMultiply(inout SHVector3 lhs, float rhs)
  156. {
  157. lhs.v0 *= rhs;
  158. lhs.v1 *= rhs;
  159. lhs.v2 *= rhs;
  160. }
  161. void SHMultiply(inout SHVector5RGB lhs, float rhs)
  162. {
  163. SHMultiply(lhs.R, rhs);
  164. SHMultiply(lhs.G, rhs);
  165. SHMultiply(lhs.B, rhs);
  166. }
  167. void SHMultiply(inout SHVector3RGB lhs, float rhs)
  168. {
  169. SHMultiply(lhs.R, rhs);
  170. SHMultiply(lhs.G, rhs);
  171. SHMultiply(lhs.B, rhs);
  172. }
  173. SHVector5 SHBasis5(float3 dir)
  174. {
  175. float x = dir.x;
  176. float y = dir.y;
  177. float z = dir.z;
  178. float x2 = x*x;
  179. float y2 = y*y;
  180. float z2 = z*z;
  181. float z3 = z2 * z;
  182. float x4 = x2 * x2;
  183. float y4 = y2 * y2;
  184. float z4 = z2 * z2;
  185. SHVector5 o;
  186. o.v0[0] = 0.282095f;
  187. o.v0[1] = -0.488603f * y;
  188. o.v0[2] = 0.488603f * z;
  189. o.v0[3] = -0.488603f * x;
  190. o.v1[0] = 1.092548f * x * y;
  191. o.v1[1] = -1.092548f * y * z;
  192. o.v1[2] = 0.315392f * (3.0f * z2 - 1.0f);
  193. o.v1[3] = -1.092548f * x * z;
  194. o.v2[0] = 0.546274f * (x2 - y2);
  195. o.v2[1] = -0.590043f * y * (3.0f * x2 - y2);
  196. o.v2[2] = 2.890611f * y * x * z;
  197. o.v2[3] = -0.646360f * y * (-1.0f + 5.0f * z2);
  198. o.v3[0] = 0.373176f *(5.0f * z3 - 3.0f * z);
  199. o.v3[1] = -0.457045f * x * (-1.0f + 5.0f * z2);
  200. o.v3[2] = 1.445306f *(x2 - y2) * z;
  201. o.v3[3] = -0.590043f * x * (x2 - 3.0f * y2);
  202. o.v4[0] = 2.503340f * x * y * (x2 - y2);
  203. o.v4[1] = -1.770130f * y * z * (3.0f * x2 - y2);
  204. o.v4[2] = 0.946175f * y * x * (-1.0f + 7.0f * z2);
  205. o.v4[3] = -0.669046f * y * z * (-3.0f + 7.0f * z2);
  206. o.v5[0] = (105.0f * z4 - 90.0f * z2 + 9.0f) / 28.359261f;
  207. o.v5[1] = -0.669046f * x * z * (-3.0f + 7.0f * z2);
  208. o.v5[2] = 0.473087f * (x2 - y2) * (-1.0f + 7.0f * z2);
  209. o.v5[3] = -1.770130f * x * z * (x2 - 3.0f * y2);
  210. o.v6 = 0.625836f * (x4 - 6.0f * y2 * x2 + y4);
  211. return o;
  212. }
  213. SHVector3 SHBasis3(float3 dir)
  214. {
  215. float x = dir.x;
  216. float y = dir.y;
  217. float z = dir.z;
  218. float x2 = x*x;
  219. float y2 = y*y;
  220. float z2 = z*z;
  221. SHVector3 o;
  222. o.v0[0] = 0.282095f;
  223. o.v0[1] = -0.488603f * y;
  224. o.v0[2] = 0.488603f * z;
  225. o.v0[3] = -0.488603f * x;
  226. o.v1[0] = 1.092548f * x * y;
  227. o.v1[1] = -1.092548f * y * z;
  228. o.v1[2] = 0.315392f * (3.0f * z2 - 1.0f);
  229. o.v1[3] = -1.092548f * x * z;
  230. o.v2 = 0.546274f * (x2 - y2);
  231. return o;
  232. }
  233. };
  234. };