stream_encoder_framing.c
上传用户:wstnjxml
上传日期:2014-04-03
资源大小:7248k
文件大小:21k
源码类别:

Windows CE

开发平台:

C/C++

  1. /* libFLAC - Free Lossless Audio Codec library
  2.  * Copyright (C) 2000,2001,2002,2003,2004,2005  Josh Coalson
  3.  *
  4.  * Redistribution and use in source and binary forms, with or without
  5.  * modification, are permitted provided that the following conditions
  6.  * are met:
  7.  *
  8.  * - Redistributions of source code must retain the above copyright
  9.  * notice, this list of conditions and the following disclaimer.
  10.  *
  11.  * - Redistributions in binary form must reproduce the above copyright
  12.  * notice, this list of conditions and the following disclaimer in the
  13.  * documentation and/or other materials provided with the distribution.
  14.  *
  15.  * - Neither the name of the Xiph.org Foundation nor the names of its
  16.  * contributors may be used to endorse or promote products derived from
  17.  * this software without specific prior written permission.
  18.  *
  19.  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  20.  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  21.  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
  22.  * A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
  23.  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  24.  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  25.  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  26.  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  27.  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  28.  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  29.  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  30.  */
  31. #include <stdio.h>
  32. #include <string.h> /* for strlen() */
  33. #include "private/stream_encoder_framing.h"
  34. #include "private/crc.h"
  35. #include "FLAC/assert.h"
  36. #ifdef max
  37. #undef max
  38. #endif
  39. #define max(x,y) ((x)>(y)?(x):(y))
  40. static FLAC__bool add_entropy_coding_method_(FLAC__BitBuffer *bb, const FLAC__EntropyCodingMethod *method);
  41. static FLAC__bool add_residual_partitioned_rice_(FLAC__BitBuffer *bb, const FLAC__int32 residual[], const unsigned residual_samples, const unsigned predictor_order, const unsigned rice_parameters[], const unsigned raw_bits[], const unsigned partition_order);
  42. FLAC__bool FLAC__add_metadata_block(const FLAC__StreamMetadata *metadata, FLAC__BitBuffer *bb)
  43. {
  44. unsigned i, j;
  45. const unsigned vendor_string_length = (unsigned)strlen(FLAC__VENDOR_STRING);
  46. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->is_last, FLAC__STREAM_METADATA_IS_LAST_LEN))
  47. return false;
  48. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->type, FLAC__STREAM_METADATA_TYPE_LEN))
  49. return false;
  50. /*
  51.  * First, for VORBIS_COMMENTs, adjust the length to reflect our vendor string
  52.  */
  53. i = metadata->length;
  54. if(metadata->type == FLAC__METADATA_TYPE_VORBIS_COMMENT) {
  55. FLAC__ASSERT(metadata->data.vorbis_comment.vendor_string.length == 0 || 0 != metadata->data.vorbis_comment.vendor_string.entry);
  56. i -= metadata->data.vorbis_comment.vendor_string.length;
  57. i += vendor_string_length;
  58. }
  59. FLAC__ASSERT(i < (1u << FLAC__STREAM_METADATA_LENGTH_LEN));
  60. if(!FLAC__bitbuffer_write_raw_uint32(bb, i, FLAC__STREAM_METADATA_LENGTH_LEN))
  61. return false;
  62. switch(metadata->type) {
  63. case FLAC__METADATA_TYPE_STREAMINFO:
  64. FLAC__ASSERT(metadata->data.stream_info.min_blocksize < (1u << FLAC__STREAM_METADATA_STREAMINFO_MIN_BLOCK_SIZE_LEN));
  65. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.stream_info.min_blocksize, FLAC__STREAM_METADATA_STREAMINFO_MIN_BLOCK_SIZE_LEN))
  66. return false;
  67. FLAC__ASSERT(metadata->data.stream_info.max_blocksize < (1u << FLAC__STREAM_METADATA_STREAMINFO_MAX_BLOCK_SIZE_LEN));
  68. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.stream_info.max_blocksize, FLAC__STREAM_METADATA_STREAMINFO_MAX_BLOCK_SIZE_LEN))
  69. return false;
  70. FLAC__ASSERT(metadata->data.stream_info.min_framesize < (1u << FLAC__STREAM_METADATA_STREAMINFO_MIN_FRAME_SIZE_LEN));
  71. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.stream_info.min_framesize, FLAC__STREAM_METADATA_STREAMINFO_MIN_FRAME_SIZE_LEN))
  72. return false;
  73. FLAC__ASSERT(metadata->data.stream_info.max_framesize < (1u << FLAC__STREAM_METADATA_STREAMINFO_MAX_FRAME_SIZE_LEN));
  74. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.stream_info.max_framesize, FLAC__STREAM_METADATA_STREAMINFO_MAX_FRAME_SIZE_LEN))
  75. return false;
  76. FLAC__ASSERT(FLAC__format_sample_rate_is_valid(metadata->data.stream_info.sample_rate));
  77. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.stream_info.sample_rate, FLAC__STREAM_METADATA_STREAMINFO_SAMPLE_RATE_LEN))
  78. return false;
  79. FLAC__ASSERT(metadata->data.stream_info.channels > 0);
  80. FLAC__ASSERT(metadata->data.stream_info.channels <= (1u << FLAC__STREAM_METADATA_STREAMINFO_CHANNELS_LEN));
  81. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.stream_info.channels-1, FLAC__STREAM_METADATA_STREAMINFO_CHANNELS_LEN))
  82. return false;
  83. FLAC__ASSERT(metadata->data.stream_info.bits_per_sample > 0);
  84. FLAC__ASSERT(metadata->data.stream_info.bits_per_sample <= (1u << FLAC__STREAM_METADATA_STREAMINFO_BITS_PER_SAMPLE_LEN));
  85. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.stream_info.bits_per_sample-1, FLAC__STREAM_METADATA_STREAMINFO_BITS_PER_SAMPLE_LEN))
  86. return false;
  87. if(!FLAC__bitbuffer_write_raw_uint64(bb, metadata->data.stream_info.total_samples, FLAC__STREAM_METADATA_STREAMINFO_TOTAL_SAMPLES_LEN))
  88. return false;
  89. if(!FLAC__bitbuffer_write_byte_block(bb, metadata->data.stream_info.md5sum, 16))
  90. return false;
  91. break;
  92. case FLAC__METADATA_TYPE_PADDING:
  93. if(!FLAC__bitbuffer_write_zeroes(bb, metadata->length * 8))
  94. return false;
  95. break;
  96. case FLAC__METADATA_TYPE_APPLICATION:
  97. if(!FLAC__bitbuffer_write_byte_block(bb, metadata->data.application.id, FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8))
  98. return false;
  99. if(!FLAC__bitbuffer_write_byte_block(bb, metadata->data.application.data, metadata->length - (FLAC__STREAM_METADATA_APPLICATION_ID_LEN / 8)))
  100. return false;
  101. break;
  102. case FLAC__METADATA_TYPE_SEEKTABLE:
  103. for(i = 0; i < metadata->data.seek_table.num_points; i++) {
  104. if(!FLAC__bitbuffer_write_raw_uint64(bb, metadata->data.seek_table.points[i].sample_number, FLAC__STREAM_METADATA_SEEKPOINT_SAMPLE_NUMBER_LEN))
  105. return false;
  106. if(!FLAC__bitbuffer_write_raw_uint64(bb, metadata->data.seek_table.points[i].stream_offset, FLAC__STREAM_METADATA_SEEKPOINT_STREAM_OFFSET_LEN))
  107. return false;
  108. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.seek_table.points[i].frame_samples, FLAC__STREAM_METADATA_SEEKPOINT_FRAME_SAMPLES_LEN))
  109. return false;
  110. }
  111. break;
  112. case FLAC__METADATA_TYPE_VORBIS_COMMENT:
  113. if(!FLAC__bitbuffer_write_raw_uint32_little_endian(bb, vendor_string_length))
  114. return false;
  115. if(!FLAC__bitbuffer_write_byte_block(bb, (const FLAC__byte*)FLAC__VENDOR_STRING, vendor_string_length))
  116. return false;
  117. if(!FLAC__bitbuffer_write_raw_uint32_little_endian(bb, metadata->data.vorbis_comment.num_comments))
  118. return false;
  119. for(i = 0; i < metadata->data.vorbis_comment.num_comments; i++) {
  120. if(!FLAC__bitbuffer_write_raw_uint32_little_endian(bb, metadata->data.vorbis_comment.comments[i].length))
  121. return false;
  122. if(!FLAC__bitbuffer_write_byte_block(bb, metadata->data.vorbis_comment.comments[i].entry, metadata->data.vorbis_comment.comments[i].length))
  123. return false;
  124. }
  125. break;
  126. case FLAC__METADATA_TYPE_CUESHEET:
  127. FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN % 8 == 0);
  128. if(!FLAC__bitbuffer_write_byte_block(bb, (const FLAC__byte*)metadata->data.cue_sheet.media_catalog_number, FLAC__STREAM_METADATA_CUESHEET_MEDIA_CATALOG_NUMBER_LEN/8))
  129. return false;
  130. if(!FLAC__bitbuffer_write_raw_uint64(bb, metadata->data.cue_sheet.lead_in, FLAC__STREAM_METADATA_CUESHEET_LEAD_IN_LEN))
  131. return false;
  132. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.cue_sheet.is_cd? 1 : 0, FLAC__STREAM_METADATA_CUESHEET_IS_CD_LEN))
  133. return false;
  134. if(!FLAC__bitbuffer_write_zeroes(bb, FLAC__STREAM_METADATA_CUESHEET_RESERVED_LEN))
  135. return false;
  136. if(!FLAC__bitbuffer_write_raw_uint32(bb, metadata->data.cue_sheet.num_tracks, FLAC__STREAM_METADATA_CUESHEET_NUM_TRACKS_LEN))
  137. return false;
  138. for(i = 0; i < metadata->data.cue_sheet.num_tracks; i++) {
  139. const FLAC__StreamMetadata_CueSheet_Track *track = metadata->data.cue_sheet.tracks + i;
  140. if(!FLAC__bitbuffer_write_raw_uint64(bb, track->offset, FLAC__STREAM_METADATA_CUESHEET_TRACK_OFFSET_LEN))
  141. return false;
  142. if(!FLAC__bitbuffer_write_raw_uint32(bb, track->number, FLAC__STREAM_METADATA_CUESHEET_TRACK_NUMBER_LEN))
  143. return false;
  144. FLAC__ASSERT(FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN % 8 == 0);
  145. if(!FLAC__bitbuffer_write_byte_block(bb, (const FLAC__byte*)track->isrc, FLAC__STREAM_METADATA_CUESHEET_TRACK_ISRC_LEN/8))
  146. return false;
  147. if(!FLAC__bitbuffer_write_raw_uint32(bb, track->type, FLAC__STREAM_METADATA_CUESHEET_TRACK_TYPE_LEN))
  148. return false;
  149. if(!FLAC__bitbuffer_write_raw_uint32(bb, track->pre_emphasis, FLAC__STREAM_METADATA_CUESHEET_TRACK_PRE_EMPHASIS_LEN))
  150. return false;
  151. if(!FLAC__bitbuffer_write_zeroes(bb, FLAC__STREAM_METADATA_CUESHEET_TRACK_RESERVED_LEN))
  152. return false;
  153. if(!FLAC__bitbuffer_write_raw_uint32(bb, track->num_indices, FLAC__STREAM_METADATA_CUESHEET_TRACK_NUM_INDICES_LEN))
  154. return false;
  155. for(j = 0; j < track->num_indices; j++) {
  156. const FLAC__StreamMetadata_CueSheet_Index *index = track->indices + j;
  157. if(!FLAC__bitbuffer_write_raw_uint64(bb, index->offset, FLAC__STREAM_METADATA_CUESHEET_INDEX_OFFSET_LEN))
  158. return false;
  159. if(!FLAC__bitbuffer_write_raw_uint32(bb, index->number, FLAC__STREAM_METADATA_CUESHEET_INDEX_NUMBER_LEN))
  160. return false;
  161. if(!FLAC__bitbuffer_write_zeroes(bb, FLAC__STREAM_METADATA_CUESHEET_INDEX_RESERVED_LEN))
  162. return false;
  163. }
  164. }
  165. break;
  166. default:
  167. if(!FLAC__bitbuffer_write_byte_block(bb, metadata->data.unknown.data, metadata->length))
  168. return false;
  169. break;
  170. }
  171. FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(bb));
  172. return true;
  173. }
  174. FLAC__bool FLAC__frame_add_header(const FLAC__FrameHeader *header, FLAC__bool streamable_subset, FLAC__BitBuffer *bb)
  175. {
  176. unsigned u, blocksize_hint, sample_rate_hint;
  177. FLAC__ASSERT(FLAC__bitbuffer_is_byte_aligned(bb));
  178. if(!FLAC__bitbuffer_write_raw_uint32(bb, FLAC__FRAME_HEADER_SYNC, FLAC__FRAME_HEADER_SYNC_LEN))
  179. return false;
  180. if(!FLAC__bitbuffer_write_raw_uint32(bb, 0, FLAC__FRAME_HEADER_RESERVED_LEN))
  181. return false;
  182. FLAC__ASSERT(header->blocksize > 0 && header->blocksize <= FLAC__MAX_BLOCK_SIZE);
  183. /* when this assertion holds true, any legal blocksize can be expressed in the frame header */
  184. FLAC__ASSERT(FLAC__MAX_BLOCK_SIZE <= 65535u);
  185. blocksize_hint = 0;
  186. switch(header->blocksize) {
  187. case   192: u = 1; break;
  188. case   576: u = 2; break;
  189. case  1152: u = 3; break;
  190. case  2304: u = 4; break;
  191. case  4608: u = 5; break;
  192. case   256: u = 8; break;
  193. case   512: u = 9; break;
  194. case  1024: u = 10; break;
  195. case  2048: u = 11; break;
  196. case  4096: u = 12; break;
  197. case  8192: u = 13; break;
  198. case 16384: u = 14; break;
  199. case 32768: u = 15; break;
  200. default:
  201. if(header->blocksize <= 0x100)
  202. blocksize_hint = u = 6;
  203. else if(header->blocksize <= 0x10000)
  204. blocksize_hint = u = 7;
  205. else {
  206. FLAC__ASSERT(0);
  207. return false;
  208. }
  209. break;
  210. }
  211. if(!FLAC__bitbuffer_write_raw_uint32(bb, u, FLAC__FRAME_HEADER_BLOCK_SIZE_LEN))
  212. return false;
  213. FLAC__ASSERT(FLAC__format_sample_rate_is_valid(header->sample_rate));
  214. sample_rate_hint = 0;
  215. switch(header->sample_rate) {
  216. case  8000: u = 4; break;
  217. case 16000: u = 5; break;
  218. case 22050: u = 6; break;
  219. case 24000: u = 7; break;
  220. case 32000: u = 8; break;
  221. case 44100: u = 9; break;
  222. case 48000: u = 10; break;
  223. case 96000: u = 11; break;
  224. default:
  225. if(header->sample_rate <= 255000 && header->sample_rate % 1000 == 0)
  226. sample_rate_hint = u = 12;
  227. else if(header->sample_rate % 10 == 0)
  228. sample_rate_hint = u = 14;
  229. else if(header->sample_rate <= 0xffff)
  230. sample_rate_hint = u = 13;
  231. else if(streamable_subset) {
  232. FLAC__ASSERT(0);
  233. return false;
  234. }
  235. else
  236. u = 0;
  237. break;
  238. }
  239. if(!FLAC__bitbuffer_write_raw_uint32(bb, u, FLAC__FRAME_HEADER_SAMPLE_RATE_LEN))
  240. return false;
  241. FLAC__ASSERT(header->channels > 0 && header->channels <= (1u << FLAC__STREAM_METADATA_STREAMINFO_CHANNELS_LEN) && header->channels <= FLAC__MAX_CHANNELS);
  242. switch(header->channel_assignment) {
  243. case FLAC__CHANNEL_ASSIGNMENT_INDEPENDENT:
  244. u = header->channels - 1;
  245. break;
  246. case FLAC__CHANNEL_ASSIGNMENT_LEFT_SIDE:
  247. FLAC__ASSERT(header->channels == 2);
  248. u = 8;
  249. break;
  250. case FLAC__CHANNEL_ASSIGNMENT_RIGHT_SIDE:
  251. FLAC__ASSERT(header->channels == 2);
  252. u = 9;
  253. break;
  254. case FLAC__CHANNEL_ASSIGNMENT_MID_SIDE:
  255. FLAC__ASSERT(header->channels == 2);
  256. u = 10;
  257. break;
  258. default:
  259. FLAC__ASSERT(0);
  260. }
  261. if(!FLAC__bitbuffer_write_raw_uint32(bb, u, FLAC__FRAME_HEADER_CHANNEL_ASSIGNMENT_LEN))
  262. return false;
  263. FLAC__ASSERT(header->bits_per_sample > 0 && header->bits_per_sample <= (1u << FLAC__STREAM_METADATA_STREAMINFO_BITS_PER_SAMPLE_LEN));
  264. switch(header->bits_per_sample) {
  265. case 8 : u = 1; break;
  266. case 12: u = 2; break;
  267. case 16: u = 4; break;
  268. case 20: u = 5; break;
  269. case 24: u = 6; break;
  270. default: u = 0; break;
  271. }
  272. if(!FLAC__bitbuffer_write_raw_uint32(bb, u, FLAC__FRAME_HEADER_BITS_PER_SAMPLE_LEN))
  273. return false;
  274. if(!FLAC__bitbuffer_write_raw_uint32(bb, 0, FLAC__FRAME_HEADER_ZERO_PAD_LEN))
  275. return false;
  276. FLAC__ASSERT(header->number_type == FLAC__FRAME_NUMBER_TYPE_FRAME_NUMBER);
  277. if(!FLAC__bitbuffer_write_utf8_uint32(bb, header->number.frame_number))
  278. return false;
  279. if(blocksize_hint)
  280. if(!FLAC__bitbuffer_write_raw_uint32(bb, header->blocksize-1, (blocksize_hint==6)? 8:16))
  281. return false;
  282. switch(sample_rate_hint) {
  283. case 12:
  284. if(!FLAC__bitbuffer_write_raw_uint32(bb, header->sample_rate / 1000, 8))
  285. return false;
  286. break;
  287. case 13:
  288. if(!FLAC__bitbuffer_write_raw_uint32(bb, header->sample_rate, 16))
  289. return false;
  290. break;
  291. case 14:
  292. if(!FLAC__bitbuffer_write_raw_uint32(bb, header->sample_rate / 10, 16))
  293. return false;
  294. break;
  295. }
  296. /* write the CRC */
  297. if(!FLAC__bitbuffer_write_raw_uint32(bb, FLAC__bitbuffer_get_write_crc8(bb), FLAC__FRAME_HEADER_CRC_LEN))
  298. return false;
  299. return true;
  300. }
  301. FLAC__bool FLAC__subframe_add_constant(const FLAC__Subframe_Constant *subframe, unsigned subframe_bps, unsigned wasted_bits, FLAC__BitBuffer *bb)
  302. {
  303. FLAC__bool ok;
  304. ok =
  305. FLAC__bitbuffer_write_raw_uint32(bb, FLAC__SUBFRAME_TYPE_CONSTANT_BYTE_ALIGNED_MASK | (wasted_bits? 1:0), FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN) &&
  306. (wasted_bits? FLAC__bitbuffer_write_unary_unsigned(bb, wasted_bits-1) : true) &&
  307. FLAC__bitbuffer_write_raw_int32(bb, subframe->value, subframe_bps)
  308. ;
  309. return ok;
  310. }
  311. FLAC__bool FLAC__subframe_add_fixed(const FLAC__Subframe_Fixed *subframe, unsigned residual_samples, unsigned subframe_bps, unsigned wasted_bits, FLAC__BitBuffer *bb)
  312. {
  313. unsigned i;
  314. if(!FLAC__bitbuffer_write_raw_uint32(bb, FLAC__SUBFRAME_TYPE_FIXED_BYTE_ALIGNED_MASK | (subframe->order<<1) | (wasted_bits? 1:0), FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN))
  315. return false;
  316. if(wasted_bits)
  317. if(!FLAC__bitbuffer_write_unary_unsigned(bb, wasted_bits-1))
  318. return false;
  319. for(i = 0; i < subframe->order; i++)
  320. if(!FLAC__bitbuffer_write_raw_int32(bb, subframe->warmup[i], subframe_bps))
  321. return false;
  322. if(!add_entropy_coding_method_(bb, &subframe->entropy_coding_method))
  323. return false;
  324. switch(subframe->entropy_coding_method.type) {
  325. case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE:
  326. if(!add_residual_partitioned_rice_(bb, subframe->residual, residual_samples, subframe->order, subframe->entropy_coding_method.data.partitioned_rice.contents->parameters, subframe->entropy_coding_method.data.partitioned_rice.contents->raw_bits, subframe->entropy_coding_method.data.partitioned_rice.order))
  327. return false;
  328. break;
  329. default:
  330. FLAC__ASSERT(0);
  331. }
  332. return true;
  333. }
  334. FLAC__bool FLAC__subframe_add_lpc(const FLAC__Subframe_LPC *subframe, unsigned residual_samples, unsigned subframe_bps, unsigned wasted_bits, FLAC__BitBuffer *bb)
  335. {
  336. unsigned i;
  337. if(!FLAC__bitbuffer_write_raw_uint32(bb, FLAC__SUBFRAME_TYPE_LPC_BYTE_ALIGNED_MASK | ((subframe->order-1)<<1) | (wasted_bits? 1:0), FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN))
  338. return false;
  339. if(wasted_bits)
  340. if(!FLAC__bitbuffer_write_unary_unsigned(bb, wasted_bits-1))
  341. return false;
  342. for(i = 0; i < subframe->order; i++)
  343. if(!FLAC__bitbuffer_write_raw_int32(bb, subframe->warmup[i], subframe_bps))
  344. return false;
  345. if(!FLAC__bitbuffer_write_raw_uint32(bb, subframe->qlp_coeff_precision-1, FLAC__SUBFRAME_LPC_QLP_COEFF_PRECISION_LEN))
  346. return false;
  347. if(!FLAC__bitbuffer_write_raw_int32(bb, subframe->quantization_level, FLAC__SUBFRAME_LPC_QLP_SHIFT_LEN))
  348. return false;
  349. for(i = 0; i < subframe->order; i++)
  350. if(!FLAC__bitbuffer_write_raw_int32(bb, subframe->qlp_coeff[i], subframe->qlp_coeff_precision))
  351. return false;
  352. if(!add_entropy_coding_method_(bb, &subframe->entropy_coding_method))
  353. return false;
  354. switch(subframe->entropy_coding_method.type) {
  355. case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE:
  356. if(!add_residual_partitioned_rice_(bb, subframe->residual, residual_samples, subframe->order, subframe->entropy_coding_method.data.partitioned_rice.contents->parameters, subframe->entropy_coding_method.data.partitioned_rice.contents->raw_bits, subframe->entropy_coding_method.data.partitioned_rice.order))
  357. return false;
  358. break;
  359. default:
  360. FLAC__ASSERT(0);
  361. }
  362. return true;
  363. }
  364. FLAC__bool FLAC__subframe_add_verbatim(const FLAC__Subframe_Verbatim *subframe, unsigned samples, unsigned subframe_bps, unsigned wasted_bits, FLAC__BitBuffer *bb)
  365. {
  366. unsigned i;
  367. const FLAC__int32 *signal = subframe->data;
  368. if(!FLAC__bitbuffer_write_raw_uint32(bb, FLAC__SUBFRAME_TYPE_VERBATIM_BYTE_ALIGNED_MASK | (wasted_bits? 1:0), FLAC__SUBFRAME_ZERO_PAD_LEN + FLAC__SUBFRAME_TYPE_LEN + FLAC__SUBFRAME_WASTED_BITS_FLAG_LEN))
  369. return false;
  370. if(wasted_bits)
  371. if(!FLAC__bitbuffer_write_unary_unsigned(bb, wasted_bits-1))
  372. return false;
  373. for(i = 0; i < samples; i++)
  374. if(!FLAC__bitbuffer_write_raw_int32(bb, signal[i], subframe_bps))
  375. return false;
  376. return true;
  377. }
  378. FLAC__bool add_entropy_coding_method_(FLAC__BitBuffer *bb, const FLAC__EntropyCodingMethod *method)
  379. {
  380. if(!FLAC__bitbuffer_write_raw_uint32(bb, method->type, FLAC__ENTROPY_CODING_METHOD_TYPE_LEN))
  381. return false;
  382. switch(method->type) {
  383. case FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE:
  384. if(!FLAC__bitbuffer_write_raw_uint32(bb, method->data.partitioned_rice.order, FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ORDER_LEN))
  385. return false;
  386. break;
  387. default:
  388. FLAC__ASSERT(0);
  389. }
  390. return true;
  391. }
  392. FLAC__bool add_residual_partitioned_rice_(FLAC__BitBuffer *bb, const FLAC__int32 residual[], const unsigned residual_samples, const unsigned predictor_order, const unsigned rice_parameters[], const unsigned raw_bits[], const unsigned partition_order)
  393. {
  394. if(partition_order == 0) {
  395. unsigned i;
  396. if(!FLAC__bitbuffer_write_raw_uint32(bb, rice_parameters[0], FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN))
  397. return false;
  398. if(rice_parameters[0] < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
  399. for(i = 0; i < residual_samples; i++) {
  400. #ifdef FLAC__SYMMETRIC_RICE
  401. if(!FLAC__bitbuffer_write_symmetric_rice_signed(bb, residual[i], rice_parameters[0]))
  402. return false;
  403. #else
  404. if(!FLAC__bitbuffer_write_rice_signed(bb, residual[i], rice_parameters[0]))
  405. return false;
  406. #endif
  407. }
  408. }
  409. else {
  410. if(!FLAC__bitbuffer_write_raw_uint32(bb, raw_bits[0], FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_RAW_LEN))
  411. return false;
  412. for(i = 0; i < residual_samples; i++) {
  413. if(!FLAC__bitbuffer_write_raw_int32(bb, residual[i], raw_bits[0]))
  414. return false;
  415. }
  416. }
  417. return true;
  418. }
  419. else {
  420. unsigned i, j, k = 0, k_last = 0;
  421. unsigned partition_samples;
  422. const unsigned default_partition_samples = (residual_samples+predictor_order) >> partition_order;
  423. for(i = 0; i < (1u<<partition_order); i++) {
  424. if(!FLAC__bitbuffer_write_raw_uint32(bb, rice_parameters[i], FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_PARAMETER_LEN))
  425. return false;
  426. partition_samples = default_partition_samples;
  427. if(i == 0)
  428. partition_samples -= predictor_order;
  429. k += partition_samples;
  430. if(rice_parameters[i] < FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_ESCAPE_PARAMETER) {
  431. for(j = k_last; j < k; j++) {
  432. #ifdef FLAC__SYMMETRIC_RICE
  433. if(!FLAC__bitbuffer_write_symmetric_rice_signed(bb, residual[j], rice_parameters[i]))
  434. return false;
  435. #else
  436. if(!FLAC__bitbuffer_write_rice_signed(bb, residual[j], rice_parameters[i]))
  437. return false;
  438. #endif
  439. }
  440. }
  441. else {
  442. if(!FLAC__bitbuffer_write_raw_uint32(bb, raw_bits[i], FLAC__ENTROPY_CODING_METHOD_PARTITIONED_RICE_RAW_LEN))
  443. return false;
  444. for(j = k_last; j < k; j++) {
  445. if(!FLAC__bitbuffer_write_raw_int32(bb, residual[j], raw_bits[i]))
  446. return false;
  447. }
  448. }
  449. k_last = k;
  450. }
  451. return true;
  452. }
  453. }