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

Windows CE

开发平台:

C/C++

  1. /* libFLAC++ - Free Lossless Audio Codec library
  2.  * Copyright (C) 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 "FLAC++/encoder.h"
  32. #include "FLAC++/metadata.h"
  33. #include "FLAC/assert.h"
  34. #ifdef _MSC_VER
  35. // warning C4800: 'int' : forcing to bool 'true' or 'false' (performance warning)
  36. #pragma warning ( disable : 4800 )
  37. #endif
  38. namespace FLAC {
  39. namespace Encoder {
  40. Stream::Stream():
  41. encoder_(::FLAC__stream_encoder_new())
  42. { }
  43. Stream::~Stream()
  44. {
  45. if(0 != encoder_) {
  46. ::FLAC__stream_encoder_finish(encoder_);
  47. ::FLAC__stream_encoder_delete(encoder_);
  48. }
  49. }
  50. bool Stream::is_valid() const
  51. {
  52. return 0 != encoder_;
  53. }
  54. bool Stream::set_verify(bool value)
  55. {
  56. FLAC__ASSERT(is_valid());
  57. return (bool)::FLAC__stream_encoder_set_verify(encoder_, value);
  58. }
  59. bool Stream::set_streamable_subset(bool value)
  60. {
  61. FLAC__ASSERT(is_valid());
  62. return (bool)::FLAC__stream_encoder_set_streamable_subset(encoder_, value);
  63. }
  64. bool Stream::set_do_mid_side_stereo(bool value)
  65. {
  66. FLAC__ASSERT(is_valid());
  67. return (bool)::FLAC__stream_encoder_set_do_mid_side_stereo(encoder_, value);
  68. }
  69. bool Stream::set_loose_mid_side_stereo(bool value)
  70. {
  71. FLAC__ASSERT(is_valid());
  72. return (bool)::FLAC__stream_encoder_set_loose_mid_side_stereo(encoder_, value);
  73. }
  74. bool Stream::set_channels(unsigned value)
  75. {
  76. FLAC__ASSERT(is_valid());
  77. return (bool)::FLAC__stream_encoder_set_channels(encoder_, value);
  78. }
  79. bool Stream::set_bits_per_sample(unsigned value)
  80. {
  81. FLAC__ASSERT(is_valid());
  82. return (bool)::FLAC__stream_encoder_set_bits_per_sample(encoder_, value);
  83. }
  84. bool Stream::set_sample_rate(unsigned value)
  85. {
  86. FLAC__ASSERT(is_valid());
  87. return (bool)::FLAC__stream_encoder_set_sample_rate(encoder_, value);
  88. }
  89. bool Stream::set_blocksize(unsigned value)
  90. {
  91. FLAC__ASSERT(is_valid());
  92. return (bool)::FLAC__stream_encoder_set_blocksize(encoder_, value);
  93. }
  94. bool Stream::set_max_lpc_order(unsigned value)
  95. {
  96. FLAC__ASSERT(is_valid());
  97. return (bool)::FLAC__stream_encoder_set_max_lpc_order(encoder_, value);
  98. }
  99. bool Stream::set_qlp_coeff_precision(unsigned value)
  100. {
  101. FLAC__ASSERT(is_valid());
  102. return (bool)::FLAC__stream_encoder_set_qlp_coeff_precision(encoder_, value);
  103. }
  104. bool Stream::set_do_qlp_coeff_prec_search(bool value)
  105. {
  106. FLAC__ASSERT(is_valid());
  107. return (bool)::FLAC__stream_encoder_set_do_qlp_coeff_prec_search(encoder_, value);
  108. }
  109. bool Stream::set_do_escape_coding(bool value)
  110. {
  111. FLAC__ASSERT(is_valid());
  112. return (bool)::FLAC__stream_encoder_set_do_escape_coding(encoder_, value);
  113. }
  114. bool Stream::set_do_exhaustive_model_search(bool value)
  115. {
  116. FLAC__ASSERT(is_valid());
  117. return (bool)::FLAC__stream_encoder_set_do_exhaustive_model_search(encoder_, value);
  118. }
  119. bool Stream::set_min_residual_partition_order(unsigned value)
  120. {
  121. FLAC__ASSERT(is_valid());
  122. return (bool)::FLAC__stream_encoder_set_min_residual_partition_order(encoder_, value);
  123. }
  124. bool Stream::set_max_residual_partition_order(unsigned value)
  125. {
  126. FLAC__ASSERT(is_valid());
  127. return (bool)::FLAC__stream_encoder_set_max_residual_partition_order(encoder_, value);
  128. }
  129. bool Stream::set_rice_parameter_search_dist(unsigned value)
  130. {
  131. FLAC__ASSERT(is_valid());
  132. return (bool)::FLAC__stream_encoder_set_rice_parameter_search_dist(encoder_, value);
  133. }
  134. bool Stream::set_total_samples_estimate(FLAC__uint64 value)
  135. {
  136. FLAC__ASSERT(is_valid());
  137. return (bool)::FLAC__stream_encoder_set_total_samples_estimate(encoder_, value);
  138. }
  139. bool Stream::set_metadata(::FLAC__StreamMetadata **metadata, unsigned num_blocks)
  140. {
  141. FLAC__ASSERT(is_valid());
  142. return (bool)::FLAC__stream_encoder_set_metadata(encoder_, metadata, num_blocks);
  143. }
  144. bool Stream::set_metadata(FLAC::Metadata::Prototype **metadata, unsigned num_blocks)
  145. {
  146. FLAC__ASSERT(is_valid());
  147. #if (defined _MSC_VER) || (defined __SUNPRO_CC)
  148. // MSVC++ can't handle:
  149. // ::FLAC__StreamMetadata *m[num_blocks];
  150. // so we do this ugly workaround
  151. ::FLAC__StreamMetadata **m = new ::FLAC__StreamMetadata*[num_blocks];
  152. #else
  153. ::FLAC__StreamMetadata *m[num_blocks];
  154. #endif
  155. for(unsigned i = 0; i < num_blocks; i++) {
  156. // we can get away with this since we know the encoder will only correct the is_last flags
  157. m[i] = const_cast< ::FLAC__StreamMetadata*>((::FLAC__StreamMetadata*)metadata[i]);
  158. }
  159. #if (defined _MSC_VER) || (defined __SUNPRO_CC)
  160. // complete the hack
  161. const bool ok = (bool)::FLAC__stream_encoder_set_metadata(encoder_, m, num_blocks);
  162. delete [] m;
  163. return ok;
  164. #else
  165. return (bool)::FLAC__stream_encoder_set_metadata(encoder_, m, num_blocks);
  166. #endif
  167. }
  168. Stream::State Stream::get_state() const
  169. {
  170. FLAC__ASSERT(is_valid());
  171. return State(::FLAC__stream_encoder_get_state(encoder_));
  172. }
  173. Decoder::Stream::State Stream::get_verify_decoder_state() const
  174. {
  175. FLAC__ASSERT(is_valid());
  176. return Decoder::Stream::State(::FLAC__stream_encoder_get_verify_decoder_state(encoder_));
  177. }
  178. void Stream::get_verify_decoder_error_stats(FLAC__uint64 *absolute_sample, unsigned *frame_number, unsigned *channel, unsigned *sample, FLAC__int32 *expected, FLAC__int32 *got)
  179. {
  180. FLAC__ASSERT(is_valid());
  181. ::FLAC__stream_encoder_get_verify_decoder_error_stats(encoder_, absolute_sample, frame_number, channel, sample, expected, got);
  182. }
  183. bool Stream::get_verify() const
  184. {
  185. FLAC__ASSERT(is_valid());
  186. return (bool)::FLAC__stream_encoder_get_verify(encoder_);
  187. }
  188. bool Stream::get_streamable_subset() const
  189. {
  190. FLAC__ASSERT(is_valid());
  191. return (bool)::FLAC__stream_encoder_get_streamable_subset(encoder_);
  192. }
  193. bool Stream::get_do_mid_side_stereo() const
  194. {
  195. FLAC__ASSERT(is_valid());
  196. return (bool)::FLAC__stream_encoder_get_do_mid_side_stereo(encoder_);
  197. }
  198. bool Stream::get_loose_mid_side_stereo() const
  199. {
  200. FLAC__ASSERT(is_valid());
  201. return (bool)::FLAC__stream_encoder_get_loose_mid_side_stereo(encoder_);
  202. }
  203. unsigned Stream::get_channels() const
  204. {
  205. FLAC__ASSERT(is_valid());
  206. return ::FLAC__stream_encoder_get_channels(encoder_);
  207. }
  208. unsigned Stream::get_bits_per_sample() const
  209. {
  210. FLAC__ASSERT(is_valid());
  211. return ::FLAC__stream_encoder_get_bits_per_sample(encoder_);
  212. }
  213. unsigned Stream::get_sample_rate() const
  214. {
  215. FLAC__ASSERT(is_valid());
  216. return ::FLAC__stream_encoder_get_sample_rate(encoder_);
  217. }
  218. unsigned Stream::get_blocksize() const
  219. {
  220. FLAC__ASSERT(is_valid());
  221. return ::FLAC__stream_encoder_get_blocksize(encoder_);
  222. }
  223. unsigned Stream::get_max_lpc_order() const
  224. {
  225. FLAC__ASSERT(is_valid());
  226. return ::FLAC__stream_encoder_get_max_lpc_order(encoder_);
  227. }
  228. unsigned Stream::get_qlp_coeff_precision() const
  229. {
  230. FLAC__ASSERT(is_valid());
  231. return ::FLAC__stream_encoder_get_qlp_coeff_precision(encoder_);
  232. }
  233. bool Stream::get_do_qlp_coeff_prec_search() const
  234. {
  235. FLAC__ASSERT(is_valid());
  236. return (bool)::FLAC__stream_encoder_get_do_qlp_coeff_prec_search(encoder_);
  237. }
  238. bool Stream::get_do_escape_coding() const
  239. {
  240. FLAC__ASSERT(is_valid());
  241. return (bool)::FLAC__stream_encoder_get_do_escape_coding(encoder_);
  242. }
  243. bool Stream::get_do_exhaustive_model_search() const
  244. {
  245. FLAC__ASSERT(is_valid());
  246. return (bool)::FLAC__stream_encoder_get_do_exhaustive_model_search(encoder_);
  247. }
  248. unsigned Stream::get_min_residual_partition_order() const
  249. {
  250. FLAC__ASSERT(is_valid());
  251. return ::FLAC__stream_encoder_get_min_residual_partition_order(encoder_);
  252. }
  253. unsigned Stream::get_max_residual_partition_order() const
  254. {
  255. FLAC__ASSERT(is_valid());
  256. return ::FLAC__stream_encoder_get_max_residual_partition_order(encoder_);
  257. }
  258. unsigned Stream::get_rice_parameter_search_dist() const
  259. {
  260. FLAC__ASSERT(is_valid());
  261. return ::FLAC__stream_encoder_get_rice_parameter_search_dist(encoder_);
  262. }
  263. FLAC__uint64 Stream::get_total_samples_estimate() const
  264. {
  265. FLAC__ASSERT(is_valid());
  266. return ::FLAC__stream_encoder_get_total_samples_estimate(encoder_);
  267. }
  268. Stream::State Stream::init()
  269. {
  270. FLAC__ASSERT(is_valid());
  271. ::FLAC__stream_encoder_set_write_callback(encoder_, write_callback_);
  272. ::FLAC__stream_encoder_set_metadata_callback(encoder_, metadata_callback_);
  273. ::FLAC__stream_encoder_set_client_data(encoder_, (void*)this);
  274. return State(::FLAC__stream_encoder_init(encoder_));
  275. }
  276. void Stream::finish()
  277. {
  278. FLAC__ASSERT(is_valid());
  279. ::FLAC__stream_encoder_finish(encoder_);
  280. }
  281. bool Stream::process(const FLAC__int32 * const buffer[], unsigned samples)
  282. {
  283. FLAC__ASSERT(is_valid());
  284. return (bool)::FLAC__stream_encoder_process(encoder_, buffer, samples);
  285. }
  286. bool Stream::process_interleaved(const FLAC__int32 buffer[], unsigned samples)
  287. {
  288. FLAC__ASSERT(is_valid());
  289. return (bool)::FLAC__stream_encoder_process_interleaved(encoder_, buffer, samples);
  290. }
  291. ::FLAC__StreamEncoderWriteStatus Stream::write_callback_(const ::FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], unsigned bytes, unsigned samples, unsigned current_frame, void *client_data)
  292. {
  293. (void)encoder;
  294. FLAC__ASSERT(0 != client_data);
  295. Stream *instance = reinterpret_cast<Stream *>(client_data);
  296. FLAC__ASSERT(0 != instance);
  297. return instance->write_callback(buffer, bytes, samples, current_frame);
  298. }
  299. void Stream::metadata_callback_(const ::FLAC__StreamEncoder *encoder, const ::FLAC__StreamMetadata *metadata, void *client_data)
  300. {
  301. (void)encoder;
  302. FLAC__ASSERT(0 != client_data);
  303. Stream *instance = reinterpret_cast<Stream *>(client_data);
  304. FLAC__ASSERT(0 != instance);
  305. instance->metadata_callback(metadata);
  306. }
  307. }
  308. }