BC7Encode_TryMode02CS.hlsl 44 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077
  1. // RUN: %dxc -E main -T cs_6_0 %s | FileCheck %s
  2. // CHECK: flattenedThreadIdInGroup
  3. // CHECK: groupId
  4. // CHECK: bufferLoad
  5. // CHECK: textureLoad
  6. // CHECK: UMax
  7. // CHECK: UMin
  8. // CHECK: barrier
  9. // CHECK: IMad
  10. // CHECK: barrier
  11. // CHECK: bufferStore
  12. //--------------------------------------------------------------------------------------
  13. // File: BC7Encode.hlsl
  14. //
  15. // The Compute Shader for BC7 Encoder
  16. //
  17. // Copyright (c) Microsoft Corporation. All rights reserved.
  18. //--------------------------------------------------------------------------------------
  19. #define REF_DEVICE
  20. #define CHAR_LENGTH 8
  21. #define NCHANNELS 4
  22. #define BC7_UNORM 98
  23. #define MAX_UINT 0xFFFFFFFF
  24. #define MIN_UINT 0
  25. static const uint candidateSectionBit[64] = //Associated to partition 0-63
  26. {
  27. 0xCCCC, 0x8888, 0xEEEE, 0xECC8,
  28. 0xC880, 0xFEEC, 0xFEC8, 0xEC80,
  29. 0xC800, 0xFFEC, 0xFE80, 0xE800,
  30. 0xFFE8, 0xFF00, 0xFFF0, 0xF000,
  31. 0xF710, 0x008E, 0x7100, 0x08CE,
  32. 0x008C, 0x7310, 0x3100, 0x8CCE,
  33. 0x088C, 0x3110, 0x6666, 0x366C,
  34. 0x17E8, 0x0FF0, 0x718E, 0x399C,
  35. 0xaaaa, 0xf0f0, 0x5a5a, 0x33cc,
  36. 0x3c3c, 0x55aa, 0x9696, 0xa55a,
  37. 0x73ce, 0x13c8, 0x324c, 0x3bdc,
  38. 0x6996, 0xc33c, 0x9966, 0x660,
  39. 0x272, 0x4e4, 0x4e40, 0x2720,
  40. 0xc936, 0x936c, 0x39c6, 0x639c,
  41. 0x9336, 0x9cc6, 0x817e, 0xe718,
  42. 0xccf0, 0xfcc, 0x7744, 0xee22,
  43. };
  44. static const uint candidateSectionBit2[64] = //Associated to partition 64-127
  45. {
  46. 0xf60008cc, 0x73008cc8, 0x3310cc80, 0xceec00,
  47. 0xcc003300, 0xcc0000cc, 0xccff00, 0x3300cccc,
  48. 0xf0000f00, 0xf0000ff0, 0xff0000f0, 0x88884444,
  49. 0x88886666, 0xcccc2222, 0xec80136c, 0x7310008c,
  50. 0xc80036c8, 0x310008ce, 0xccc03330, 0xcccf000,
  51. 0xee0000ee, 0x77008888, 0xcc0022c0, 0x33004430,
  52. 0xcc0c22, 0xfc880344, 0x6606996, 0x66009960,
  53. 0xc88c0330, 0xf9000066, 0xcc0c22c, 0x73108c00,
  54. 0xec801300, 0x8cec400, 0xec80004c, 0x44442222,
  55. 0xf0000f0, 0x49242492, 0x42942942, 0xc30c30c,
  56. 0x3c0c03c, 0xff0000aa, 0x5500aa00, 0xcccc3030,
  57. 0xc0cc0c0, 0x66669090, 0xff0a00a, 0x5550aaa0,
  58. 0xf0000aaa, 0xe0ee0e0, 0x88887070, 0x99906660,
  59. 0xe00e0ee0, 0x88880770, 0xf0000666, 0x99006600,
  60. 0xff000066, 0xc00c0cc0, 0xcccc0330, 0x90006000,
  61. 0x8088080, 0xeeee1010, 0xfff0000a, 0x731008ce,
  62. };
  63. static const uint2 candidateFixUpIndex1D[128] =
  64. {
  65. {15, 0},{15, 0},{15, 0},{15, 0},
  66. {15, 0},{15, 0},{15, 0},{15, 0},
  67. {15, 0},{15, 0},{15, 0},{15, 0},
  68. {15, 0},{15, 0},{15, 0},{15, 0},
  69. {15, 0},{ 2, 0},{ 8, 0},{ 2, 0},
  70. { 2, 0},{ 8, 0},{ 8, 0},{15, 0},
  71. { 2, 0},{ 8, 0},{ 2, 0},{ 2, 0},
  72. { 8, 0},{ 8, 0},{ 2, 0},{ 2, 0},
  73. {15, 0},{15, 0},{ 6, 0},{ 8, 0},
  74. { 2, 0},{ 8, 0},{15, 0},{15, 0},
  75. { 2, 0},{ 8, 0},{ 2, 0},{ 2, 0},
  76. { 2, 0},{15, 0},{15, 0},{ 6, 0},
  77. { 6, 0},{ 2, 0},{ 6, 0},{ 8, 0},
  78. {15, 0},{15, 0},{ 2, 0},{ 2, 0},
  79. {15, 0},{15, 0},{15, 0},{15, 0},
  80. {15, 0},{ 2, 0},{ 2, 0},{15, 0},
  81. //candidateFixUpIndex1D[i][1], i < 64 should not be used
  82. { 3,15},{ 3, 8},{15, 8},{15, 3},
  83. { 8,15},{ 3,15},{15, 3},{15, 8},
  84. { 8,15},{ 8,15},{ 6,15},{ 6,15},
  85. { 6,15},{ 5,15},{ 3,15},{ 3, 8},
  86. { 3,15},{ 3, 8},{ 8,15},{15, 3},
  87. { 3,15},{ 3, 8},{ 6,15},{10, 8},
  88. { 5, 3},{ 8,15},{ 8, 6},{ 6,10},
  89. { 8,15},{ 5,15},{15,10},{15, 8},
  90. { 8,15},{15, 3},{ 3,15},{ 5,10},
  91. { 6,10},{10, 8},{ 8, 9},{15,10},
  92. {15, 6},{ 3,15},{15, 8},{ 5,15},
  93. {15, 3},{15, 6},{15, 6},{15, 8}, //The Spec doesn't mark the first fixed up index in this row, so I apply 15 for them, and seems correct
  94. { 3,15},{15, 3},{ 5,15},{ 5,15},
  95. { 5,15},{ 8,15},{ 5,15},{10,15},
  96. { 5,15},{10,15},{ 8,15},{13,15},
  97. {15, 3},{12,15},{ 3,15},{ 3, 8},
  98. };
  99. static const uint2 candidateFixUpIndex1DOrdered[128] = //Same with candidateFixUpIndex1D but order the result when i >= 64
  100. {
  101. {15, 0},{15, 0},{15, 0},{15, 0},
  102. {15, 0},{15, 0},{15, 0},{15, 0},
  103. {15, 0},{15, 0},{15, 0},{15, 0},
  104. {15, 0},{15, 0},{15, 0},{15, 0},
  105. {15, 0},{ 2, 0},{ 8, 0},{ 2, 0},
  106. { 2, 0},{ 8, 0},{ 8, 0},{15, 0},
  107. { 2, 0},{ 8, 0},{ 2, 0},{ 2, 0},
  108. { 8, 0},{ 8, 0},{ 2, 0},{ 2, 0},
  109. {15, 0},{15, 0},{ 6, 0},{ 8, 0},
  110. { 2, 0},{ 8, 0},{15, 0},{15, 0},
  111. { 2, 0},{ 8, 0},{ 2, 0},{ 2, 0},
  112. { 2, 0},{15, 0},{15, 0},{ 6, 0},
  113. { 6, 0},{ 2, 0},{ 6, 0},{ 8, 0},
  114. {15, 0},{15, 0},{ 2, 0},{ 2, 0},
  115. {15, 0},{15, 0},{15, 0},{15, 0},
  116. {15, 0},{ 2, 0},{ 2, 0},{15, 0},
  117. //candidateFixUpIndex1DOrdered[i][1], i < 64 should not be used
  118. { 3,15},{ 3, 8},{ 8,15},{ 3,15},
  119. { 8,15},{ 3,15},{ 3,15},{ 8,15},
  120. { 8,15},{ 8,15},{ 6,15},{ 6,15},
  121. { 6,15},{ 5,15},{ 3,15},{ 3, 8},
  122. { 3,15},{ 3, 8},{ 8,15},{ 3,15},
  123. { 3,15},{ 3, 8},{ 6,15},{ 8,10},
  124. { 3, 5},{ 8,15},{ 6, 8},{ 6,10},
  125. { 8,15},{ 5,15},{10,15},{ 8,15},
  126. { 8,15},{ 3,15},{ 3,15},{ 5,10},
  127. { 6,10},{ 8,10},{ 8, 9},{10,15},
  128. { 6,15},{ 3,15},{ 8,15},{ 5,15},
  129. { 3,15},{ 6,15},{ 6,15},{ 8,15}, //The Spec doesn't mark the first fixed up index in this row, so I apply 15 for them, and seems correct
  130. { 3,15},{ 3,15},{ 5,15},{ 5,15},
  131. { 5,15},{ 8,15},{ 5,15},{10,15},
  132. { 5,15},{10,15},{ 8,15},{13,15},
  133. { 3,15},{12,15},{ 3,15},{ 3, 8},
  134. };
  135. static const uint4x4 candidateRotation[4] =
  136. {
  137. {1,0,0,0},{0,1,0,0},{0,0,1,0},{0,0,0,1},
  138. {0,0,0,1},{0,1,0,0},{0,0,1,0},{1,0,0,0},
  139. {1,0,0,0},{0,0,0,1},{0,0,1,0},{0,1,0,0},
  140. {1,0,0,0},{0,1,0,0},{0,0,0,1},{0,0,1,0}
  141. };
  142. static const uint2 candidateIndexPrec[8] = {{3,0},{3,0},{2,0},{2,0},
  143. {2,3}, //color index and alpha index can exchange
  144. {2,2},{4,4},{2,2}};
  145. static const uint aWeight[3][16] = { {0, 4, 9, 13, 17, 21, 26, 30, 34, 38, 43, 47, 51, 55, 60, 64},
  146. {0, 9, 18, 27, 37, 46, 55, 64, 0, 0, 0, 0, 0, 0, 0, 0},
  147. {0, 21, 43, 64, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0} };
  148. //0, 4, 9, 13, 17, 21, 26, 30, 34, 38, 43, 47, 51, 55, 60, 64
  149. static const uint aStep[3][64] = { { 0, 0, 0, 1, 1, 1, 1, 2,
  150. 2, 2, 2, 2, 3, 3, 3, 3,
  151. 4, 4, 4, 4, 5, 5, 5, 5,
  152. 6, 6, 6, 6, 6, 7, 7, 7,
  153. 7, 8, 8, 8, 8, 9, 9, 9,
  154. 9,10,10,10,10,10,11,11,
  155. 11,11,12,12,12,12,13,13,
  156. 13,13,14,14,14,14,15,15 },
  157. //0, 9, 18, 27, 37, 46, 55, 64
  158. { 0,0,0,0,0,1,1,1,
  159. 1,1,1,1,1,1,2,2,
  160. 2,2,2,2,2,2,2,3,
  161. 3,3,3,3,3,3,3,3,
  162. 3,4,4,4,4,4,4,4,
  163. 4,4,5,5,5,5,5,5,
  164. 5,5,5,6,6,6,6,6,
  165. 6,6,6,6,7,7,7,7 },
  166. //0, 21, 43, 64
  167. { 0,0,0,0,0,0,0,0,
  168. 0,0,0,1,1,1,1,1,
  169. 1,1,1,1,1,1,1,1,
  170. 1,1,1,1,1,1,1,1,
  171. 1,2,2,2,2,2,2,2,
  172. 2,2,2,2,2,2,2,2,
  173. 2,2,2,2,2,2,3,3,
  174. 3,3,3,3,3,3,3,3 } };
  175. cbuffer cbCS : register( b0 )
  176. {
  177. uint g_tex_width;
  178. uint g_num_block_x;
  179. uint g_format;
  180. uint g_mode_id;
  181. uint g_start_block_id;
  182. uint g_num_total_blocks;
  183. };
  184. //Forward declaration
  185. void compress_endpoints0( inout uint2x4 endPoint ); //Mode = 0
  186. void compress_endpoints1( inout uint2x4 endPoint ); //Mode = 1
  187. void compress_endpoints2( inout uint2x4 endPoint ); //Mode = 2
  188. void compress_endpoints3( inout uint2x4 endPoint ); //Mode = 3
  189. void compress_endpoints7( inout uint2x4 endPoint ); //Mode = 7
  190. void compress_endpoints6( inout uint2x4 endPoint ); //Mode = 6
  191. void compress_endpoints4( inout uint2x4 endPoint ); //Mode = 4
  192. void compress_endpoints5( inout uint2x4 endPoint ); //Mode = 5
  193. void block_package0( out uint4 block, uint partition, uint threadBase ); //Mode0
  194. void block_package1( out uint4 block, uint partition, uint threadBase ); //Mode1
  195. void block_package2( out uint4 block, uint partition, uint threadBase ); //Mode2
  196. void block_package3( out uint4 block, uint partition, uint threadBase ); //Mode3
  197. void block_package4( out uint4 block, uint rotation, uint index_selector, uint threadBase ); //Mode4
  198. void block_package5( out uint4 block, uint rotation, uint threadBase ); //Mode5
  199. void block_package6( out uint4 block, uint threadBase ); //Mode6
  200. void block_package7( out uint4 block, uint partition, uint threadBase ); //Mode7
  201. void swap(inout uint4 lhs, inout uint4 rhs)
  202. {
  203. int4 tmp = lhs;
  204. lhs = rhs;
  205. rhs = tmp;
  206. }
  207. void swap(inout uint3 lhs, inout uint3 rhs)
  208. {
  209. int3 tmp = lhs;
  210. lhs = rhs;
  211. rhs = tmp;
  212. }
  213. void swap(inout uint lhs, inout uint rhs)
  214. {
  215. int tmp = lhs;
  216. lhs = rhs;
  217. rhs = tmp;
  218. }
  219. Texture2D g_Input : register( t0 );
  220. StructuredBuffer<uint4> g_InBuff : register( t1 );
  221. RWStructuredBuffer<uint4> g_OutBuff : register( u0 );
  222. #define THREAD_GROUP_SIZE 64
  223. #define BLOCK_SIZE_Y 4
  224. #define BLOCK_SIZE_X 4
  225. #define BLOCK_SIZE (BLOCK_SIZE_Y * BLOCK_SIZE_X)
  226. struct BufferShared
  227. {
  228. uint4 pixel;
  229. uint error;
  230. uint mode;
  231. uint partition;
  232. uint index_selector;
  233. uint rotation;
  234. uint4 endPoint_low;
  235. uint4 endPoint_high;
  236. };
  237. groupshared BufferShared shared_temp[THREAD_GROUP_SIZE];
  238. [numthreads( THREAD_GROUP_SIZE, 1, 1 )]
  239. void TryMode456CS( uint GI : SV_GroupIndex, uint3 groupID : SV_GroupID )
  240. {
  241. const uint MAX_USED_THREAD = 16;
  242. uint BLOCK_IN_GROUP = THREAD_GROUP_SIZE / MAX_USED_THREAD;
  243. uint blockInGroup = GI / MAX_USED_THREAD;
  244. uint blockID = g_start_block_id + groupID.x * BLOCK_IN_GROUP + blockInGroup;
  245. uint threadBase = blockInGroup * MAX_USED_THREAD;
  246. uint threadInBlock = GI - threadBase;
  247. #ifndef REF_DEVICE
  248. if (blockID >= g_num_total_blocks)
  249. {
  250. return;
  251. }
  252. #endif
  253. uint block_y = blockID / g_num_block_x;
  254. uint block_x = blockID - block_y * g_num_block_x;
  255. uint base_x = block_x * BLOCK_SIZE_X;
  256. uint base_y = block_y * BLOCK_SIZE_Y;
  257. if (threadInBlock < 16)
  258. {
  259. shared_temp[GI].pixel = clamp(uint4(g_Input.Load( uint3( base_x + threadInBlock % 4, base_y + threadInBlock / 4, 0 ) ) * 255), 0, 255);
  260. shared_temp[GI].endPoint_low = shared_temp[GI].pixel;
  261. shared_temp[GI].endPoint_high = shared_temp[GI].pixel;
  262. }
  263. #ifdef REF_DEVICE
  264. GroupMemoryBarrierWithGroupSync();
  265. #endif
  266. if (threadInBlock < 8)
  267. {
  268. shared_temp[GI].endPoint_low = min(shared_temp[GI].endPoint_low, shared_temp[GI + 8].endPoint_low);
  269. shared_temp[GI].endPoint_high = max(shared_temp[GI].endPoint_high, shared_temp[GI + 8].endPoint_high);
  270. }
  271. #ifdef REF_DEVICE
  272. GroupMemoryBarrierWithGroupSync();
  273. #endif
  274. if (threadInBlock < 4)
  275. {
  276. shared_temp[GI].endPoint_low = min(shared_temp[GI].endPoint_low, shared_temp[GI + 4].endPoint_low);
  277. shared_temp[GI].endPoint_high = max(shared_temp[GI].endPoint_high, shared_temp[GI + 4].endPoint_high);
  278. }
  279. #ifdef REF_DEVICE
  280. GroupMemoryBarrierWithGroupSync();
  281. #endif
  282. if (threadInBlock < 2)
  283. {
  284. shared_temp[GI].endPoint_low = min(shared_temp[GI].endPoint_low, shared_temp[GI + 2].endPoint_low);
  285. shared_temp[GI].endPoint_high = max(shared_temp[GI].endPoint_high, shared_temp[GI + 2].endPoint_high);
  286. }
  287. #ifdef REF_DEVICE
  288. GroupMemoryBarrierWithGroupSync();
  289. #endif
  290. if (threadInBlock < 1)
  291. {
  292. shared_temp[GI].endPoint_low = min(shared_temp[GI].endPoint_low, shared_temp[GI + 1].endPoint_low);
  293. shared_temp[GI].endPoint_high = max(shared_temp[GI].endPoint_high, shared_temp[GI + 1].endPoint_high);
  294. }
  295. #ifdef REF_DEVICE
  296. GroupMemoryBarrierWithGroupSync();
  297. #endif
  298. uint2x4 endPoint;
  299. endPoint[0] = shared_temp[threadBase].endPoint_low;
  300. endPoint[1] = shared_temp[threadBase].endPoint_high;
  301. uint error = 0xFFFFFFFF;
  302. uint mode = 0;
  303. uint index_selector = 0;
  304. uint rotation = 0;
  305. uint2 indexPrec;
  306. if (threadInBlock < 8)
  307. {
  308. if (0 == (threadInBlock & 1))
  309. {
  310. //2 represents 2bit index precision; 1 represents 3bit index precision
  311. indexPrec = uint2( 2, 1 );
  312. }
  313. else
  314. {
  315. //2 represents 2bit index precision; 1 represents 3bit index precision
  316. index_selector = 1;
  317. indexPrec = uint2( 1, 2 );
  318. }
  319. }
  320. else
  321. {
  322. //2 represents 2bit index precision
  323. indexPrec = uint2( 2, 2 );
  324. }
  325. uint4 pixel_r;
  326. uint color_index;
  327. uint alpha_index;
  328. int4 span;
  329. int2 span_norm_sqr;
  330. int2 dotProduct;
  331. if (threadInBlock < 12)
  332. {
  333. if ((threadInBlock < 2) || (8 == threadInBlock)) // rotation = 0
  334. {
  335. rotation = 0;
  336. }
  337. else if ((threadInBlock < 4) || (9 == threadInBlock)) // rotation = 1
  338. {
  339. endPoint[0].ra = endPoint[0].ar;
  340. endPoint[1].ra = endPoint[1].ar;
  341. rotation = 1;
  342. }
  343. else if ((threadInBlock < 6) || (10 == threadInBlock)) // rotation = 2
  344. {
  345. endPoint[0].ga = endPoint[0].ag;
  346. endPoint[1].ga = endPoint[1].ag;
  347. rotation = 2;
  348. }
  349. else if ((threadInBlock < 8) || (11 == threadInBlock)) // rotation = 3
  350. {
  351. endPoint[0].ba = endPoint[0].ab;
  352. endPoint[1].ba = endPoint[1].ab;
  353. rotation = 3;
  354. }
  355. if (threadInBlock < 8)
  356. {
  357. mode = 4;
  358. compress_endpoints4( endPoint );
  359. }
  360. else
  361. {
  362. mode = 5;
  363. compress_endpoints5( endPoint );
  364. }
  365. uint4 pixel = shared_temp[threadBase + 0].pixel;
  366. if (1 == rotation)
  367. {
  368. pixel.ra = pixel.ar;
  369. }
  370. else if (2 == rotation)
  371. {
  372. pixel.ga = pixel.ag;
  373. }
  374. else if (3 == rotation)
  375. {
  376. pixel.ba = pixel.ab;
  377. }
  378. span = endPoint[1] - endPoint[0];
  379. span_norm_sqr = uint2( dot( span.rgb, span.rgb ), span.a * span.a );
  380. dotProduct = int2( dot( span.rgb, pixel.rgb - endPoint[0].rgb ), span.a * ( pixel.a - endPoint[0].a ) );
  381. if ( span_norm_sqr.x > 0 && dotProduct.x > 0 && uint( dotProduct.x * 63.49999 ) > uint( 32 * span_norm_sqr.x ) )
  382. {
  383. span.rgb = -span.rgb;
  384. swap(endPoint[0].rgb, endPoint[1].rgb);
  385. }
  386. if ( span_norm_sqr.y > 0 && dotProduct.y > 0 && uint( dotProduct.y * 63.49999 ) > uint( 32 * span_norm_sqr.y ) )
  387. {
  388. span.a = -span.a;
  389. swap(endPoint[0].a, endPoint[1].a);
  390. }
  391. error = 0;
  392. for ( uint i = 0; i < 16; i ++ )
  393. {
  394. pixel = shared_temp[threadBase + i].pixel;
  395. if (1 == rotation)
  396. {
  397. pixel.ra = pixel.ar;
  398. }
  399. else if (2 == rotation)
  400. {
  401. pixel.ga = pixel.ag;
  402. }
  403. else if (3 == rotation)
  404. {
  405. pixel.ba = pixel.ab;
  406. }
  407. dotProduct.x = dot( span.rgb, pixel.rgb - endPoint[0].rgb );
  408. color_index = ( span_norm_sqr.x <= 0 || dotProduct.x <= 0 ) ? 0
  409. : ( ( dotProduct.x < span_norm_sqr.x ) ? aStep[indexPrec.x][ uint( dotProduct.x * 63.49999 / span_norm_sqr.x ) ] : aStep[indexPrec.x][63] );
  410. dotProduct.y = dot( span.a, pixel.a - endPoint[0].a );
  411. alpha_index = ( span_norm_sqr.y <= 0 || dotProduct.y <= 0 ) ? 0
  412. : ( ( dotProduct.y < span_norm_sqr.y ) ? aStep[indexPrec.y][ uint( dotProduct.y * 63.49999 / span_norm_sqr.y ) ] : aStep[indexPrec.y][63] );
  413. if (index_selector)
  414. {
  415. swap(color_index, alpha_index);
  416. }
  417. pixel_r.rgb = ( ( 64 - aWeight[indexPrec.x][color_index] ) * endPoint[0].rgb
  418. + aWeight[indexPrec.x][color_index] * endPoint[1] + 32 ) >> 6;
  419. pixel_r.a = ( ( 64 - aWeight[indexPrec.y][alpha_index] ) * endPoint[0].a
  420. + aWeight[indexPrec.y][alpha_index] * endPoint[1] + 32 ) >> 6;
  421. pixel_r -= pixel;
  422. error += dot(pixel_r, pixel_r);
  423. }
  424. }
  425. else if (12 == threadInBlock)//Mode6
  426. {
  427. compress_endpoints6( endPoint );
  428. uint4 pixel = shared_temp[threadBase + 0].pixel;
  429. span = endPoint[1] - endPoint[0];
  430. span_norm_sqr = dot( span, span );
  431. dotProduct = dot( span, pixel - endPoint[0] );
  432. if ( span_norm_sqr.x > 0 && dotProduct.x >= 0 && uint( dotProduct.x * 63.49999 ) > uint( 32 * span_norm_sqr.x ) )
  433. {
  434. span = -span;
  435. swap(endPoint[0], endPoint[1]);
  436. }
  437. error = 0;
  438. for ( uint i = 0; i < 16; i ++ )
  439. {
  440. pixel = shared_temp[threadBase + i].pixel;
  441. dotProduct.x = dot( span, pixel - endPoint[0] );
  442. color_index = ( span_norm_sqr.x <= 0 || dotProduct.x <= 0 ) ? 0
  443. : ( ( dotProduct.x < span_norm_sqr.x ) ? aStep[0][ uint( dotProduct.x * 63.49999 / span_norm_sqr.x ) ] : aStep[0][63] );
  444. pixel_r = ( ( 64 - aWeight[0][color_index] ) * endPoint[0]
  445. + aWeight[0][color_index] * endPoint[1] + 32 ) >> 6;
  446. pixel_r -= pixel;
  447. error += dot(pixel_r, pixel_r);
  448. }
  449. mode = 6;
  450. rotation = 0;
  451. }
  452. shared_temp[GI].error = error;
  453. shared_temp[GI].mode = mode;
  454. shared_temp[GI].index_selector = index_selector;
  455. shared_temp[GI].rotation = rotation;
  456. #ifdef REF_DEVICE
  457. GroupMemoryBarrierWithGroupSync();
  458. #endif
  459. if (threadInBlock < 8)
  460. {
  461. if ( shared_temp[GI].error > shared_temp[GI + 8].error )
  462. {
  463. shared_temp[GI].error = shared_temp[GI + 8].error;
  464. shared_temp[GI].mode = shared_temp[GI + 8].mode;
  465. shared_temp[GI].index_selector = shared_temp[GI + 8].index_selector;
  466. shared_temp[GI].rotation = shared_temp[GI + 8].rotation;
  467. }
  468. }
  469. #ifdef REF_DEVICE
  470. GroupMemoryBarrierWithGroupSync();
  471. #endif
  472. if (threadInBlock < 4)
  473. {
  474. if ( shared_temp[GI].error > shared_temp[GI + 4].error )
  475. {
  476. shared_temp[GI].error = shared_temp[GI + 4].error;
  477. shared_temp[GI].mode = shared_temp[GI + 4].mode;
  478. shared_temp[GI].index_selector = shared_temp[GI + 4].index_selector;
  479. shared_temp[GI].rotation = shared_temp[GI + 4].rotation;
  480. }
  481. }
  482. #ifdef REF_DEVICE
  483. GroupMemoryBarrierWithGroupSync();
  484. #endif
  485. if (threadInBlock < 2)
  486. {
  487. if ( shared_temp[GI].error > shared_temp[GI + 2].error )
  488. {
  489. shared_temp[GI].error = shared_temp[GI + 2].error;
  490. shared_temp[GI].mode = shared_temp[GI + 2].mode;
  491. shared_temp[GI].index_selector = shared_temp[GI + 2].index_selector;
  492. shared_temp[GI].rotation = shared_temp[GI + 2].rotation;
  493. }
  494. }
  495. #ifdef REF_DEVICE
  496. GroupMemoryBarrierWithGroupSync();
  497. #endif
  498. if (threadInBlock < 1)
  499. {
  500. if ( shared_temp[GI].error > shared_temp[GI + 1].error )
  501. {
  502. shared_temp[GI].error = shared_temp[GI + 1].error;
  503. shared_temp[GI].mode = shared_temp[GI + 1].mode;
  504. shared_temp[GI].index_selector = shared_temp[GI + 1].index_selector;
  505. shared_temp[GI].rotation = shared_temp[GI + 1].rotation;
  506. }
  507. g_OutBuff[blockID] = uint4(shared_temp[GI].error, (shared_temp[GI].index_selector << 31) | shared_temp[GI].mode,
  508. 0, shared_temp[GI].rotation);
  509. }
  510. }
  511. [numthreads( THREAD_GROUP_SIZE, 1, 1 )]
  512. void main( uint GI : SV_GroupIndex, uint3 groupID : SV_GroupID )
  513. {
  514. const uint MAX_USED_THREAD = 64;
  515. uint BLOCK_IN_GROUP = THREAD_GROUP_SIZE / MAX_USED_THREAD;
  516. uint blockInGroup = GI / MAX_USED_THREAD;
  517. uint blockID = g_start_block_id + groupID.x * BLOCK_IN_GROUP + blockInGroup;
  518. uint threadBase = blockInGroup * MAX_USED_THREAD;
  519. uint threadInBlock = GI - threadBase;
  520. uint block_y = blockID / g_num_block_x;
  521. uint block_x = blockID - block_y * g_num_block_x;
  522. uint base_x = block_x * BLOCK_SIZE_X;
  523. uint base_y = block_y * BLOCK_SIZE_Y;
  524. if (threadInBlock < 16)
  525. {
  526. shared_temp[GI].pixel = clamp(uint4(g_Input.Load( uint3( base_x + threadInBlock % 4, base_y + threadInBlock / 4, 0 ) ) * 255), 0, 255);
  527. }
  528. GroupMemoryBarrierWithGroupSync();
  529. shared_temp[GI].error = 0xFFFFFFFF;
  530. uint num_partitions;
  531. if (0 == g_mode_id)
  532. {
  533. num_partitions = 16;
  534. }
  535. else
  536. {
  537. num_partitions = 64;
  538. }
  539. uint4 pixel_r;
  540. uint2x4 endPoint[3];
  541. uint color_index[16];
  542. if (threadInBlock < num_partitions)
  543. {
  544. uint partition = threadInBlock + 64;
  545. endPoint[0][0] = MAX_UINT;
  546. endPoint[0][1] = MIN_UINT;
  547. endPoint[1][0] = MAX_UINT;
  548. endPoint[1][1] = MIN_UINT;
  549. endPoint[2][0] = MAX_UINT;
  550. endPoint[2][1] = MIN_UINT;
  551. uint bits2 = candidateSectionBit2[partition - 64];
  552. for ( uint i = 0; i < 16; i ++ )
  553. {
  554. uint4 pixel = shared_temp[threadBase + i].pixel;
  555. if ( (( bits2 >> ( i + 15 ) ) & 0x02) == 2 ) //It gets error when using "candidateSectionCompressed" as "endPoint" index
  556. {
  557. endPoint[2][0] = min( endPoint[2][0], pixel );
  558. endPoint[2][1] = max( endPoint[2][1], pixel );
  559. }
  560. else if ( (( bits2 >> i ) & 0x01) == 1 )
  561. {
  562. endPoint[1][0] = min( endPoint[1][0], pixel );
  563. endPoint[1][1] = max( endPoint[1][1], pixel );
  564. }
  565. else
  566. {
  567. endPoint[0][0] = min( endPoint[0][0], pixel );
  568. endPoint[0][1] = max( endPoint[0][1], pixel );
  569. }
  570. }
  571. for ( uint i = 0; i < 3; i ++ )
  572. {
  573. if (0 == g_mode_id)
  574. {
  575. compress_endpoints0( endPoint[i] );
  576. }
  577. else
  578. {
  579. compress_endpoints2( endPoint[i] );
  580. }
  581. }
  582. uint step_selector = 1 + (2 == g_mode_id);
  583. int4 span[3];
  584. span[0] = endPoint[0][1] - endPoint[0][0];
  585. span[1] = endPoint[1][1] - endPoint[1][0];
  586. span[2] = endPoint[2][1] - endPoint[2][0];
  587. span[0].w = span[1].w = span[2].w = 0;
  588. int span_norm_sqr[3];
  589. span_norm_sqr[0] = dot( span[0], span[0] );
  590. span_norm_sqr[1] = dot( span[1], span[1] );
  591. span_norm_sqr[2] = dot( span[2], span[2] );
  592. int dotProduct = dot( span[0], shared_temp[threadBase + 0].pixel - endPoint[0][0] );
  593. if ( span_norm_sqr[0] > 0 && dotProduct > 0 && uint( dotProduct * 63.49999 ) > uint( 32 * span_norm_sqr[0] ) )
  594. {
  595. span[0] = -span[0];
  596. swap(endPoint[0][0], endPoint[0][1]);
  597. }
  598. dotProduct = dot( span[1], shared_temp[threadBase + candidateFixUpIndex1D[partition].x].pixel - endPoint[1][0] );
  599. if ( span_norm_sqr[1] > 0 && dotProduct > 0 && uint( dotProduct * 63.49999 ) > uint( 32 * span_norm_sqr[1] ) )
  600. {
  601. span[1] = -span[1];
  602. swap(endPoint[1][0], endPoint[1][1]);
  603. }
  604. dotProduct = dot( span[2], shared_temp[threadBase + candidateFixUpIndex1D[partition].y].pixel - endPoint[2][0] );
  605. if ( span_norm_sqr[2] > 0 && dotProduct > 0 && uint( dotProduct * 63.49999 ) > uint( 32 * span_norm_sqr[2] ) )
  606. {
  607. span[2] = -span[2];
  608. swap(endPoint[2][0], endPoint[2][1]);
  609. }
  610. uint error = 0;
  611. for ( uint i = 0; i < 16; i ++ )
  612. {
  613. if ( (( bits2 >> ( i + 15 ) ) & 0x02) == 2 )
  614. {
  615. dotProduct = dot( span[2], shared_temp[threadBase + i].pixel - endPoint[2][0] );
  616. color_index[i] = ( span_norm_sqr[1] <= 0 || dotProduct <= 0 ) ? 0
  617. : ( ( dotProduct < span_norm_sqr[2] ) ? aStep[step_selector][ uint( dotProduct * 63.49999 / span_norm_sqr[2] ) ] : aStep[step_selector][63] );
  618. }
  619. else if ( (( bits2 >> i ) & 0x01) == 1 )
  620. {
  621. dotProduct = dot( span[1], shared_temp[threadBase + i].pixel - endPoint[1][0] );
  622. color_index[i] = ( span_norm_sqr[1] <= 0 || dotProduct <= 0 ) ? 0
  623. : ( ( dotProduct < span_norm_sqr[1] ) ? aStep[step_selector][ uint( dotProduct * 63.49999 / span_norm_sqr[1] ) ] : aStep[step_selector][63] );
  624. }
  625. else
  626. {
  627. dotProduct = dot( span[0], shared_temp[threadBase + i].pixel - endPoint[0][0] );
  628. color_index[i] = ( span_norm_sqr[0] <= 0 || dotProduct <= 0 ) ? 0
  629. : ( ( dotProduct < span_norm_sqr[0] ) ? aStep[step_selector][ uint( dotProduct * 63.49999 / span_norm_sqr[0] ) ] : aStep[step_selector][63] );
  630. }
  631. uint subset_index = ((bits2 >> (i + 15)) & 0x02) | ((bits2 >> i) & 0x01);
  632. pixel_r = ( ( 64 - aWeight[step_selector][color_index[i]] ) * endPoint[subset_index][0]
  633. + aWeight[step_selector][color_index[i]] * endPoint[subset_index][1] + 32 ) >> 6;
  634. pixel_r -= shared_temp[threadBase + i].pixel;
  635. pixel_r.w = 0;
  636. error += dot(pixel_r, pixel_r);
  637. }
  638. shared_temp[GI].error = error;
  639. shared_temp[GI].partition = partition;
  640. }
  641. GroupMemoryBarrierWithGroupSync();
  642. if (threadInBlock < 32)
  643. {
  644. if ( shared_temp[GI].error > shared_temp[GI + 32].error )
  645. {
  646. shared_temp[GI].error = shared_temp[GI + 32].error;
  647. shared_temp[GI].partition = shared_temp[GI + 32].partition;
  648. }
  649. }
  650. #ifdef REF_DEVICE
  651. GroupMemoryBarrierWithGroupSync();
  652. #endif
  653. if (threadInBlock < 16)
  654. {
  655. if ( shared_temp[GI].error > shared_temp[GI + 16].error )
  656. {
  657. shared_temp[GI].error = shared_temp[GI + 16].error;
  658. shared_temp[GI].partition = shared_temp[GI + 16].partition;
  659. }
  660. }
  661. #ifdef REF_DEVICE
  662. GroupMemoryBarrierWithGroupSync();
  663. #endif
  664. if (threadInBlock < 8)
  665. {
  666. if ( shared_temp[GI].error > shared_temp[GI + 8].error )
  667. {
  668. shared_temp[GI].error = shared_temp[GI + 8].error;
  669. shared_temp[GI].partition = shared_temp[GI + 8].partition;
  670. }
  671. }
  672. #ifdef REF_DEVICE
  673. GroupMemoryBarrierWithGroupSync();
  674. #endif
  675. if (threadInBlock < 4)
  676. {
  677. if ( shared_temp[GI].error > shared_temp[GI + 4].error )
  678. {
  679. shared_temp[GI].error = shared_temp[GI + 4].error;
  680. shared_temp[GI].partition = shared_temp[GI + 4].partition;
  681. }
  682. }
  683. #ifdef REF_DEVICE
  684. GroupMemoryBarrierWithGroupSync();
  685. #endif
  686. if (threadInBlock < 2)
  687. {
  688. if ( shared_temp[GI].error > shared_temp[GI + 2].error )
  689. {
  690. shared_temp[GI].error = shared_temp[GI + 2].error;
  691. shared_temp[GI].partition = shared_temp[GI + 2].partition;
  692. }
  693. }
  694. #ifdef REF_DEVICE
  695. GroupMemoryBarrierWithGroupSync();
  696. #endif
  697. if (threadInBlock < 1)
  698. {
  699. if ( shared_temp[GI].error > shared_temp[GI + 1].error )
  700. {
  701. shared_temp[GI].error = shared_temp[GI + 1].error;
  702. shared_temp[GI].partition = shared_temp[GI + 1].partition;
  703. }
  704. if (g_InBuff[blockID].x > shared_temp[GI].error)
  705. {
  706. g_OutBuff[blockID] = uint4(shared_temp[GI].error, g_mode_id, shared_temp[GI].partition, 0);
  707. }
  708. else
  709. {
  710. g_OutBuff[blockID] = g_InBuff[blockID];
  711. }
  712. }
  713. }
  714. void compress_endpoints0( inout uint2x4 endPoint )
  715. {
  716. uint3 tmp;
  717. for ( uint j = 0; j < 2; j ++ )
  718. {
  719. tmp = endPoint[j].rgb & 0x0F;
  720. tmp.x += tmp.y + tmp.z;
  721. endPoint[j].rgb = ( endPoint[j].rgb & 0xF0 ) | ( ( tmp.x / 3 ) & 0x08 );
  722. }
  723. }
  724. void compress_endpoints1( inout uint2x4 endPoint )
  725. {
  726. uint3 tmp;
  727. tmp = ( endPoint[0].rgb & 0x03 ) + ( endPoint[1].rgb & 0x03 );
  728. tmp.x += tmp.y + tmp.z;
  729. tmp.x = ( tmp.x / 6 ) & 0x02;
  730. for ( uint j = 0; j < 2; j ++ )
  731. {
  732. endPoint[j].rgb = ( endPoint[j].rgb & 0xFC ) | tmp.x;
  733. }
  734. }
  735. void compress_endpoints2( inout uint2x4 endPoint )
  736. {
  737. for ( uint j = 0; j < 2; j ++ )
  738. {
  739. endPoint[j].rgb = min(255, ( endPoint[j].rgb + 0x04 ) ) & 0xF8;
  740. }
  741. }
  742. void compress_endpoints3( inout uint2x4 endPoint )
  743. {
  744. uint3 tmp;
  745. for ( uint j = 0; j < 2; j ++ )
  746. {
  747. tmp = endPoint[j].rgb & 0x01;
  748. tmp.x += tmp.y + tmp.z;
  749. endPoint[j].rgb = ( endPoint[j].rgb & 0xFE ) | ( tmp.x / 3 );
  750. }
  751. }
  752. void compress_endpoints4( inout uint2x4 endPoint )
  753. {
  754. for ( uint j = 0; j < 2; j ++ )
  755. {
  756. endPoint[j] = min(255, ( endPoint[j] + uint4(0x04.xxx, 0x02) ) ) & uint4(0xF8.xxx, 0xFC);
  757. }
  758. }
  759. void compress_endpoints5( inout uint2x4 endPoint )
  760. {
  761. for ( uint j = 0; j < 2; j ++ )
  762. {
  763. endPoint[j].rgb = min(255, ( endPoint[j].rgb + 0x01 ) ) & 0xFE;
  764. }
  765. }
  766. void compress_endpoints6( inout uint2x4 endPoint )
  767. {
  768. uint4 tmp;
  769. for ( uint j = 0; j < 2; j ++ )
  770. {
  771. tmp = endPoint[j] & 0x01;
  772. tmp.x += tmp.y + tmp.z + tmp.w;
  773. endPoint[j] = ( endPoint[j] & 0xFE ) | ( ( tmp.x >> 2 ) & 0x01 );
  774. }
  775. }
  776. void compress_endpoints7( inout uint2x4 endPoint )
  777. {
  778. uint4 tmp;
  779. for ( uint j = 0; j < 2; j ++ )
  780. {
  781. tmp = endPoint[j] & 0x07;
  782. tmp.x += tmp.y + tmp.z + tmp.w;
  783. endPoint[j] = ( endPoint[j] & 0xF8 ) | ( ( tmp.x >> 2 ) & 0x04 );
  784. }
  785. }
  786. #define get_end_point_l(subset) shared_temp[threadBase + subset].endPoint_low
  787. #define get_end_point_h(subset) shared_temp[threadBase + subset].endPoint_high
  788. #define get_color_index(index) shared_temp[threadBase + index].error
  789. #define get_alpha_index(index) shared_temp[threadBase + index].mode
  790. void block_package0( out uint4 block, uint partition, uint threadBase )
  791. {
  792. block.x = 0x01 | ( (partition - 64) << 1 )
  793. | ( ( get_end_point_l(0).r & 0xF0 ) << 1 ) | ( ( get_end_point_h(0).r & 0xF0 ) << 5 )
  794. | ( ( get_end_point_l(1).r & 0xF0 ) << 9 ) | ( ( get_end_point_h(1).r & 0xF0 ) << 13 )
  795. | ( ( get_end_point_l(2).r & 0xF0 ) << 17 ) | ( ( get_end_point_h(2).r & 0xF0 ) << 21 )
  796. | ( ( get_end_point_l(0).g & 0xF0 ) << 25 );
  797. block.y = ( ( get_end_point_l(0).g & 0xF0 ) >> 7 ) | ( ( get_end_point_h(0).g & 0xF0 ) >> 3 )
  798. | ( ( get_end_point_l(1).g & 0xF0 ) << 1 ) | ( ( get_end_point_h(1).g & 0xF0 ) << 5 )
  799. | ( ( get_end_point_l(2).g & 0xF0 ) << 9 ) | ( ( get_end_point_h(2).g & 0xF0 ) << 13 )
  800. | ( ( get_end_point_l(0).b & 0xF0 ) << 17 ) | ( ( get_end_point_h(0).b & 0xF0 ) << 21 )
  801. | ( ( get_end_point_l(1).b & 0xF0 ) << 25 );
  802. block.z = ( ( get_end_point_l(1).b & 0xF0 ) >> 7 ) | ( ( get_end_point_h(1).b & 0xF0 ) >> 3 )
  803. | ( ( get_end_point_l(2).b & 0xF0 ) << 1 ) | ( ( get_end_point_h(2).b & 0xF0 ) << 5 )
  804. | ( ( get_end_point_l(0).r & 0x08 ) << 10 ) | ( ( get_end_point_h(0).r & 0x08 ) << 11 )
  805. | ( ( get_end_point_l(1).r & 0x08 ) << 12 ) | ( ( get_end_point_h(1).r & 0x08 ) << 13 )
  806. | ( ( get_end_point_l(2).r & 0x08 ) << 14 ) | ( ( get_end_point_h(2).r & 0x08 ) << 15 )
  807. | ( get_color_index(0) << 19 );
  808. block.w = 0;
  809. uint i = 1;
  810. for ( ; i <= min( candidateFixUpIndex1DOrdered[partition][0], 4 ); i ++ )
  811. {
  812. block.z |= get_color_index(i) << ( i * 3 + 18 );
  813. }
  814. if ( candidateFixUpIndex1DOrdered[partition][0] < 4 ) //i = 4
  815. {
  816. block.z |= get_color_index(4) << 29;
  817. i += 1;
  818. }
  819. else //i = 5
  820. {
  821. block.w |= ( get_color_index(4) & 0x04 ) >> 2;
  822. for ( ; i <= candidateFixUpIndex1DOrdered[partition][0]; i ++ )
  823. block.w |= get_color_index(i) << ( i * 3 - 14 );
  824. }
  825. for ( ; i <= candidateFixUpIndex1DOrdered[partition][1]; i ++ )
  826. {
  827. block.w |= get_color_index(i) << ( i * 3 - 15 );
  828. }
  829. for ( ; i < 16; i ++ )
  830. {
  831. block.w |= get_color_index(i) << ( i * 3 - 16 );
  832. }
  833. }
  834. void block_package1( out uint4 block, uint partition, uint threadBase )
  835. {
  836. block.x = 0x02 | ( partition << 2 )
  837. | ( ( get_end_point_l(0).r & 0xFC ) << 6 ) | ( ( get_end_point_h(0).r & 0xFC ) << 12 )
  838. | ( ( get_end_point_l(1).r & 0xFC ) << 18 ) | ( ( get_end_point_h(1).r & 0xFC ) << 24 );
  839. block.y = ( ( get_end_point_l(0).g & 0xFC ) >> 2 ) | ( ( get_end_point_h(0).g & 0xFC ) << 4 )
  840. | ( ( get_end_point_l(1).g & 0xFC ) << 10 ) | ( ( get_end_point_h(1).g & 0xFC ) << 16 )
  841. | ( ( get_end_point_l(0).b & 0xFC ) << 22 ) | ( ( get_end_point_h(0).b & 0xFC ) << 28 );
  842. block.z = ( ( get_end_point_h(0).b & 0xFC ) >> 4 ) | ( ( get_end_point_l(1).b & 0xFC ) << 2 )
  843. | ( ( get_end_point_h(1).b & 0xFC ) << 8 )
  844. | ( ( get_end_point_l(0).r & 0x02 ) << 15 ) | ( ( get_end_point_l(1).r & 0x02 ) << 16 )
  845. | ( get_color_index(0) << 18 );
  846. if ( candidateFixUpIndex1DOrdered[partition][0] == 15 )
  847. {
  848. block.w = (get_color_index(15) << 30) | (get_color_index(14) << 27) | (get_color_index(13) << 24) | (get_color_index(12) << 21) | (get_color_index(11) << 18) | (get_color_index(10) << 15)
  849. | (get_color_index(9) << 12) | (get_color_index(8) << 9) | (get_color_index(7) << 6) | (get_color_index(6) << 3) | get_color_index(5);
  850. block.z |= (get_color_index(4) << 29) | (get_color_index(3) << 26) | (get_color_index(2) << 23) | (get_color_index(1) << 20) | (get_color_index(0) << 18);
  851. }
  852. else if ( candidateFixUpIndex1DOrdered[partition][0] == 2 )
  853. {
  854. block.w = (get_color_index(15) << 29) | (get_color_index(14) << 26) | (get_color_index(13) << 23) | (get_color_index(12) << 20) | (get_color_index(11) << 17) | (get_color_index(10) << 14)
  855. | (get_color_index(9) << 11) | (get_color_index(8) << 8) | (get_color_index(7) << 5) | (get_color_index(6) << 2) | (get_color_index(5) >> 1);
  856. block.z |= (get_color_index(5) << 31) | (get_color_index(4) << 28) | (get_color_index(3) << 25) | (get_color_index(2) << 23) | (get_color_index(1) << 20) | (get_color_index(0) << 18);
  857. }
  858. else if ( candidateFixUpIndex1DOrdered[partition][0] == 8 )
  859. {
  860. block.w = (get_color_index(15) << 29) | (get_color_index(14) << 26) | (get_color_index(13) << 23) | (get_color_index(12) << 20) | (get_color_index(11) << 17) | (get_color_index(10) << 14)
  861. | (get_color_index(9) << 11) | (get_color_index(8) << 9) | (get_color_index(7) << 6) | (get_color_index(6) << 3) | get_color_index(5);
  862. block.z |= (get_color_index(4) << 29) | (get_color_index(3) << 26) | (get_color_index(2) << 23) | (get_color_index(1) << 20) | (get_color_index(0) << 18);
  863. }
  864. else //candidateFixUpIndex1DOrdered[partition] == 6
  865. {
  866. block.w = (get_color_index(15) << 29) | (get_color_index(14) << 26) | (get_color_index(13) << 23) | (get_color_index(12) << 20) | (get_color_index(11) << 17) | (get_color_index(10) << 14)
  867. | (get_color_index(9) << 11) | (get_color_index(8) << 8) | (get_color_index(7) << 6) | (get_color_index(6) << 4) | get_color_index(5);
  868. block.z |= (get_color_index(4) << 29) | (get_color_index(3) << 26) | (get_color_index(2) << 23) | (get_color_index(1) << 20) | (get_color_index(0) << 18);
  869. }
  870. }
  871. void block_package2( out uint4 block, uint partition, uint threadBase )
  872. {
  873. block.x = 0x04 | ( (partition - 64) << 3 )
  874. | ( ( get_end_point_l(0).r & 0xF8 ) << 6 ) | ( ( get_end_point_h(0).r & 0xF8 ) << 11 )
  875. | ( ( get_end_point_l(1).r & 0xF8 ) << 16 ) | ( ( get_end_point_h(1).r & 0xF8 ) << 21 )
  876. | ( ( get_end_point_l(2).r & 0xF8 ) << 26 );
  877. block.y = ( ( get_end_point_l(2).r & 0xF8 ) >> 6 ) | ( ( get_end_point_h(2).r & 0xF8 ) >> 1 )
  878. | ( ( get_end_point_l(0).g & 0xF8 ) << 4 ) | ( ( get_end_point_h(0).g & 0xF8 ) << 9 )
  879. | ( ( get_end_point_l(1).g & 0xF8 ) << 14 ) | ( ( get_end_point_h(1).g & 0xF8 ) << 19 )
  880. | ( ( get_end_point_l(2).g & 0xF8 ) << 24 );
  881. block.z = ( ( get_end_point_h(2).g & 0xF8 ) >> 3 ) | ( ( get_end_point_l(0).b & 0xF8 ) << 2 )
  882. | ( ( get_end_point_h(0).b & 0xF8 ) << 7 ) | ( ( get_end_point_l(1).b & 0xF8 ) << 12 )
  883. | ( ( get_end_point_h(1).b & 0xF8 ) << 17 ) | ( ( get_end_point_l(2).b & 0xF8 ) << 22 )
  884. | ( ( get_end_point_h(2).b & 0xF8 ) << 27 );
  885. block.w = ( ( get_end_point_h(2).b & 0xF8 ) >> 5 )
  886. | ( get_color_index(0) << 3 );
  887. uint i = 1;
  888. for ( ; i <= candidateFixUpIndex1DOrdered[partition][0]; i ++ )
  889. {
  890. block.w |= get_color_index(i) << ( i * 2 + 2 );
  891. }
  892. for ( ; i <= candidateFixUpIndex1DOrdered[partition][1]; i ++ )
  893. {
  894. block.w |= get_color_index(i) << ( i * 2 + 1 );
  895. }
  896. for ( ; i < 16; i ++ )
  897. {
  898. block.w |= get_color_index(i) << ( i * 2 );
  899. }
  900. }
  901. void block_package3( out uint4 block, uint partition, uint threadBase )
  902. {
  903. block.x = 0x08 | ( partition << 4 )
  904. | ( ( get_end_point_l(0).r & 0xFE ) << 9 ) | ( ( get_end_point_h(0).r & 0xFE ) << 16 )
  905. | ( ( get_end_point_l(1).r & 0xFE ) << 23 ) | ( ( get_end_point_h(1).r & 0xFE ) << 30 );
  906. block.y = ( ( get_end_point_h(1).r & 0xFE ) >> 2 ) | ( ( get_end_point_l(0).g & 0xFE ) << 5 )
  907. | ( ( get_end_point_h(0).g & 0xFE ) << 12 ) | ( ( get_end_point_l(1).g & 0xFE ) << 19 )
  908. | ( ( get_end_point_h(1).g & 0xFE ) << 26 );
  909. block.z = ( ( get_end_point_h(1).g & 0xFE ) >> 6 ) | ( ( get_end_point_l(0).b & 0xFE ) << 1 )
  910. | ( ( get_end_point_h(0).b & 0xFE ) << 8 ) | ( ( get_end_point_l(1).b & 0xFE ) << 15 )
  911. | ( ( get_end_point_h(1).b & 0xFE ) << 22 )
  912. | ( ( get_end_point_l(0).r & 0x01 ) << 30 ) | ( ( get_end_point_h(0).r & 0x01 ) << 31 );
  913. block.w = ( ( get_end_point_l(1).r & 0x01 ) << 0 ) | ( ( get_end_point_h(1).r & 0x01 ) << 1 )
  914. | ( get_color_index(0) << 2 );
  915. uint i = 1;
  916. for ( ; i <= candidateFixUpIndex1DOrdered[partition][0]; i ++ )
  917. {
  918. block.w |= get_color_index(i) << ( i * 2 + 1 );
  919. }
  920. for ( ; i < 16; i ++ )
  921. {
  922. block.w |= get_color_index(i) << ( i * 2 );
  923. }
  924. }
  925. void block_package4( out uint4 block, uint rotation, uint index_selector, uint threadBase )
  926. {
  927. block.x = 0x10 | ( rotation << 5 ) | ( index_selector << 7 )
  928. | ( ( get_end_point_l(0).r & 0xF8 ) << 5 ) | ( ( get_end_point_h(0).r & 0xF8 ) << 10 )
  929. | ( ( get_end_point_l(0).g & 0xF8 ) << 15 ) | ( ( get_end_point_h(0).g & 0xF8 ) << 20 )
  930. | ( ( get_end_point_l(0).b & 0xF8 ) << 25 );
  931. block.y = ( ( get_end_point_l(0).b & 0xF8 ) >> 7 ) | ( ( get_end_point_h(0).b & 0xF8 ) >> 2 )
  932. | ( ( get_end_point_l(0).a & 0xFC ) << 4 ) | ( ( get_end_point_h(0).a & 0xFC ) << 10 )
  933. | ( get_color_index(0) << 18 ) | ( get_color_index(1) << 19 ) | ( get_color_index(2) << 21 ) | ( get_color_index(3) << 23 )
  934. | ( get_color_index(4) << 25 ) | ( get_color_index(5) << 27 ) | ( get_color_index(6) << 29 ) | ( get_color_index(7) << 31 );
  935. block.z = ( get_color_index(7) >> 1 ) | ( get_color_index(8) << 1 ) | ( get_color_index(9) << 3 ) | ( get_color_index(10)<< 5 )
  936. | ( get_color_index(11)<< 7 ) | ( get_color_index(12)<< 9 ) | ( get_color_index(13)<< 11 ) | ( get_color_index(14)<< 13 )
  937. | ( get_color_index(15)<< 15 ) | ( get_alpha_index(0) << 17 ) | ( get_alpha_index(1) << 19 ) | ( get_alpha_index(2) << 22 )
  938. | ( get_alpha_index(3) << 25 ) | ( get_alpha_index(4) << 28 ) | ( get_alpha_index(5) << 31 );
  939. block.w = ( get_alpha_index(5) >> 1 ) | ( get_alpha_index(6) << 2 ) | ( get_alpha_index(7) << 5 ) | ( get_alpha_index(8) << 8 )
  940. | ( get_alpha_index(9) << 11 ) | ( get_alpha_index(10)<< 14 ) | ( get_alpha_index(11)<< 17 ) | ( get_alpha_index(12)<< 20 )
  941. | ( get_alpha_index(13)<< 23 ) | ( get_alpha_index(14)<< 26 ) | ( get_alpha_index(15)<< 29 );
  942. }
  943. void block_package5( out uint4 block, uint rotation, uint threadBase )
  944. {
  945. block.x = 0x20 | ( rotation << 6 )
  946. | ( ( get_end_point_l(0).r & 0xFE ) << 7 ) | ( ( get_end_point_h(0).r & 0xFE ) << 14 )
  947. | ( ( get_end_point_l(0).g & 0xFE ) << 21 ) | ( ( get_end_point_h(0).g & 0xFE ) << 28 );
  948. block.y = ( ( get_end_point_h(0).g & 0xFE ) >> 4 ) | ( ( get_end_point_l(0).b & 0xFE ) << 3 )
  949. | ( ( get_end_point_h(0).b & 0xFE ) << 10 ) | ( get_end_point_l(0).a << 18 ) | ( get_end_point_h(0).a << 26 );
  950. block.z = ( get_end_point_h(0).a >> 6 )
  951. | ( get_color_index(0) << 2 ) | ( get_color_index(1) << 3 ) | ( get_color_index(2) << 5 ) | ( get_color_index(3) << 7 )
  952. | ( get_color_index(4) << 9 ) | ( get_color_index(5) << 11 ) | ( get_color_index(6) << 13 ) | ( get_color_index(7) << 15 )
  953. | ( get_color_index(8) << 17 ) | ( get_color_index(9) << 19 ) | ( get_color_index(10)<< 21 ) | ( get_color_index(11)<< 23 )
  954. | ( get_color_index(12)<< 25 ) | ( get_color_index(13)<< 27 ) | ( get_color_index(14)<< 29 ) | ( get_color_index(15)<< 31 );
  955. block.w = ( get_color_index(15)>> 1 ) | ( get_alpha_index(0) << 0 ) | ( get_alpha_index(1) << 2 ) | ( get_alpha_index(2) << 4 )
  956. | ( get_alpha_index(3) << 6 ) | ( get_alpha_index(4) << 8 ) | ( get_alpha_index(5) << 10 ) | ( get_alpha_index(6) << 12 )
  957. | ( get_alpha_index(7) << 14 ) | ( get_alpha_index(8) << 16 ) | ( get_alpha_index(9) << 18 ) | ( get_alpha_index(10)<< 20 )
  958. | ( get_alpha_index(11)<< 22 ) | ( get_alpha_index(12)<< 24 ) | ( get_alpha_index(13)<< 26 ) | ( get_alpha_index(14)<< 28 )
  959. | ( get_alpha_index(15)<< 30 );
  960. }
  961. void block_package6( out uint4 block, uint threadBase )
  962. {
  963. block.x = 0x40
  964. | ( ( get_end_point_l(0).r & 0xFE ) << 6 ) | ( ( get_end_point_h(0).r & 0xFE ) << 13 )
  965. | ( ( get_end_point_l(0).g & 0xFE ) << 20 ) | ( ( get_end_point_h(0).g & 0xFE ) << 27 );
  966. block.y = ( ( get_end_point_h(0).g & 0xFE ) >> 5 ) | ( ( get_end_point_l(0).b & 0xFE ) << 2 )
  967. | ( ( get_end_point_h(0).b & 0xFE ) << 9 ) | ( ( get_end_point_l(0).a & 0xFE ) << 16 )
  968. | ( ( get_end_point_h(0).a & 0xFE ) << 23 )
  969. | ( get_end_point_l(0).r & 0x01 ) << 31;
  970. block.z = ( get_end_point_h(0).r & 0x01 )
  971. | ( get_color_index(0) << 1 ) | ( get_color_index(1) << 4 ) | ( get_color_index(2) << 8 ) | ( get_color_index(3) << 12 )
  972. | ( get_color_index(4) << 16 ) | ( get_color_index(5) << 20 ) | ( get_color_index(6) << 24 ) | ( get_color_index(7) << 28 );
  973. block.w = ( get_color_index(8) << 0 ) | ( get_color_index(9) << 4 ) | ( get_color_index(10)<< 8 ) | ( get_color_index(11)<< 12 )
  974. | ( get_color_index(12)<< 16 ) | ( get_color_index(13)<< 20 ) | ( get_color_index(14)<< 24 ) | ( get_color_index(15)<< 28 );
  975. }
  976. void block_package7( out uint4 block, uint partition, uint threadBase )
  977. {
  978. block.x = 0x80 | ( partition << 8 )
  979. | ( ( get_end_point_l(0).r & 0xF8 ) << 11 ) | ( ( get_end_point_h(0).r & 0xF8 ) << 16 )
  980. | ( ( get_end_point_l(1).r & 0xF8 ) << 21 ) | ( ( get_end_point_h(1).r & 0xF8 ) << 26 );
  981. block.y = ( ( get_end_point_h(1).r & 0xF8 ) >> 6 ) | ( ( get_end_point_l(0).g & 0xF8 ) >> 1 )
  982. | ( ( get_end_point_h(0).g & 0xF8 ) << 4 ) | ( ( get_end_point_l(1).g & 0xF8 ) << 9 )
  983. | ( ( get_end_point_h(1).g & 0xF8 ) << 14 ) | ( ( get_end_point_l(0).b & 0xF8 ) << 19 )
  984. | ( ( get_end_point_h(0).b & 0xF8 ) << 24 );
  985. block.z = ( ( get_end_point_l(1).b & 0xF8 ) >> 3 ) | ( ( get_end_point_h(1).b & 0xF8 ) << 2 )
  986. | ( ( get_end_point_l(0).a & 0xF8 ) << 7 ) | ( ( get_end_point_h(0).a & 0xF8 ) << 12 )
  987. | ( ( get_end_point_l(1).a & 0xF8 ) << 17 ) | ( ( get_end_point_h(1).a & 0xF8 ) << 22 )
  988. | ( ( get_end_point_l(0).r & 0x04 ) << 27 ) | ( ( get_end_point_h(0).r & 0x04 ) << 28 );
  989. block.w = ( ( get_end_point_l(1).r & 0x04 ) >> 2 ) | ( ( get_end_point_h(1).r & 0x04 ) >> 1 )
  990. | ( get_color_index(0) << 2 );
  991. uint i = 1;
  992. for ( ; i <= candidateFixUpIndex1DOrdered[partition][0]; i ++ )
  993. {
  994. block.w |= get_color_index(i) << ( i * 2 + 1 );
  995. }
  996. for ( ; i < 16; i ++ )
  997. {
  998. block.w |= get_color_index(i) << ( i * 2 );
  999. }
  1000. }