Chris@1: /* libFLAC++ - Free Lossless Audio Codec library
Chris@1:  * Copyright (C) 2002,2003,2004,2005,2006,2007  Josh Coalson
Chris@1:  *
Chris@1:  * Redistribution and use in source and binary forms, with or without
Chris@1:  * modification, are permitted provided that the following conditions
Chris@1:  * are met:
Chris@1:  *
Chris@1:  * - Redistributions of source code must retain the above copyright
Chris@1:  * notice, this list of conditions and the following disclaimer.
Chris@1:  *
Chris@1:  * - Redistributions in binary form must reproduce the above copyright
Chris@1:  * notice, this list of conditions and the following disclaimer in the
Chris@1:  * documentation and/or other materials provided with the distribution.
Chris@1:  *
Chris@1:  * - Neither the name of the Xiph.org Foundation nor the names of its
Chris@1:  * contributors may be used to endorse or promote products derived from
Chris@1:  * this software without specific prior written permission.
Chris@1:  *
Chris@1:  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
Chris@1:  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
Chris@1:  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
Chris@1:  * A PARTICULAR PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR
Chris@1:  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
Chris@1:  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
Chris@1:  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
Chris@1:  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
Chris@1:  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
Chris@1:  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
Chris@1:  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
Chris@1:  */
Chris@1: 
Chris@1: #ifndef FLACPP__ENCODER_H
Chris@1: #define FLACPP__ENCODER_H
Chris@1: 
Chris@1: #include "export.h"
Chris@1: 
Chris@1: #include "FLAC/stream_encoder.h"
Chris@1: #include "decoder.h"
Chris@1: #include "metadata.h"
Chris@1: 
Chris@1: 
Chris@1: /** \file include/FLAC++/encoder.h
Chris@1:  *
Chris@1:  *  \brief
Chris@1:  *  This module contains the classes which implement the various
Chris@1:  *  encoders.
Chris@1:  *
Chris@1:  *  See the detailed documentation in the
Chris@1:  *  \link flacpp_encoder encoder \endlink module.
Chris@1:  */
Chris@1: 
Chris@1: /** \defgroup flacpp_encoder FLAC++/encoder.h: encoder classes
Chris@1:  *  \ingroup flacpp
Chris@1:  *
Chris@1:  *  \brief
Chris@1:  *  This module describes the encoder layers provided by libFLAC++.
Chris@1:  *
Chris@1:  * The libFLAC++ encoder classes are object wrappers around their
Chris@1:  * counterparts in libFLAC.  All encoding layers available in
Chris@1:  * libFLAC are also provided here.  The interface is very similar;
Chris@1:  * make sure to read the \link flac_encoder libFLAC encoder module \endlink.
Chris@1:  *
Chris@1:  * There are only two significant differences here.  First, instead of
Chris@1:  * passing in C function pointers for callbacks, you inherit from the
Chris@1:  * encoder class and provide implementations for the callbacks in your
Chris@1:  * derived class; because of this there is no need for a 'client_data'
Chris@1:  * property.
Chris@1:  *
Chris@1:  * Second, there are two stream encoder classes.  FLAC::Encoder::Stream
Chris@1:  * is used for the same cases that FLAC__stream_encoder_init_stream() /
Chris@1:  * FLAC__stream_encoder_init_ogg_stream() are used, and FLAC::Encoder::File
Chris@1:  * is used for the same cases that
Chris@1:  * FLAC__stream_encoder_init_FILE() and FLAC__stream_encoder_init_file() /
Chris@1:  * FLAC__stream_encoder_init_ogg_FILE() and FLAC__stream_encoder_init_ogg_file()
Chris@1:  * are used.
Chris@1:  */
Chris@1: 
Chris@1: namespace FLAC {
Chris@1: 	namespace Encoder {
Chris@1: 
Chris@1: 		/** \ingroup flacpp_encoder
Chris@1: 		 *  \brief
Chris@1: 		 *  This class wraps the ::FLAC__StreamEncoder.  If you are
Chris@1: 		 *  encoding to a file, FLAC::Encoder::File may be more
Chris@1: 		 *  convenient.
Chris@1: 		 *
Chris@1: 		 * The usage of this class is similar to FLAC__StreamEncoder,
Chris@1: 		 * except instead of providing callbacks to
Chris@1: 		 * FLAC__stream_encoder_init*_stream(), you will inherit from this
Chris@1: 		 * class and override the virtual callback functions with your
Chris@1: 		 * own implementations, then call init() or init_ogg().  The rest of
Chris@1: 		 * the calls work the same as in the C layer.
Chris@1: 		 *
Chris@1: 		 * Only the write callback is mandatory.  The others are
Chris@1: 		 * optional; this class provides default implementations that do
Chris@1: 		 * nothing.  In order for some STREAMINFO and SEEKTABLE data to
Chris@1: 		 * be written properly, you must overide seek_callback() and
Chris@1: 		 * tell_callback(); see FLAC__stream_encoder_init_stream() as to
Chris@1: 		 * why.
Chris@1: 		 */
Chris@1: 		class FLACPP_API Stream {
Chris@1: 		public:
Chris@1: 			/** This class is a wrapper around FLAC__StreamEncoderState.
Chris@1: 			 */
Chris@1: 			class FLACPP_API State {
Chris@1: 			public:
Chris@1: 				inline State(::FLAC__StreamEncoderState state): state_(state) { }
Chris@1: 				inline operator ::FLAC__StreamEncoderState() const { return state_; }
Chris@1: 				inline const char *as_cstring() const { return ::FLAC__StreamEncoderStateString[state_]; }
Chris@1: 				inline const char *resolved_as_cstring(const Stream &encoder) const { return ::FLAC__stream_encoder_get_resolved_state_string(encoder.encoder_); }
Chris@1: 			protected:
Chris@1: 				::FLAC__StreamEncoderState state_;
Chris@1: 			};
Chris@1: 
Chris@1: 			Stream();
Chris@1: 			virtual ~Stream();
Chris@1: 
Chris@1: 			//@{
Chris@1: 			/** Call after construction to check the that the object was created
Chris@1: 			 *  successfully.  If not, use get_state() to find out why not.
Chris@1: 			 *
Chris@1: 			 */
Chris@1: 			virtual bool is_valid() const;
Chris@1: 			inline operator bool() const { return is_valid(); } ///< See is_valid()
Chris@1: 			//@}
Chris@1: 
Chris@1: 			virtual bool set_ogg_serial_number(long value);                 ///< See FLAC__stream_encoder_set_ogg_serial_number()
Chris@1: 			virtual bool set_verify(bool value);                            ///< See FLAC__stream_encoder_set_verify()
Chris@1: 			virtual bool set_streamable_subset(bool value);                 ///< See FLAC__stream_encoder_set_streamable_subset()
Chris@1: 			virtual bool set_channels(unsigned value);                      ///< See FLAC__stream_encoder_set_channels()
Chris@1: 			virtual bool set_bits_per_sample(unsigned value);               ///< See FLAC__stream_encoder_set_bits_per_sample()
Chris@1: 			virtual bool set_sample_rate(unsigned value);                   ///< See FLAC__stream_encoder_set_sample_rate()
Chris@1: 			virtual bool set_compression_level(unsigned value);             ///< See FLAC__stream_encoder_set_compression_level()
Chris@1: 			virtual bool set_blocksize(unsigned value);                     ///< See FLAC__stream_encoder_set_blocksize()
Chris@1: 			virtual bool set_do_mid_side_stereo(bool value);                ///< See FLAC__stream_encoder_set_do_mid_side_stereo()
Chris@1: 			virtual bool set_loose_mid_side_stereo(bool value);             ///< See FLAC__stream_encoder_set_loose_mid_side_stereo()
Chris@1: 			virtual bool set_apodization(const char *specification);        ///< See FLAC__stream_encoder_set_apodization()
Chris@1: 			virtual bool set_max_lpc_order(unsigned value);                 ///< See FLAC__stream_encoder_set_max_lpc_order()
Chris@1: 			virtual bool set_qlp_coeff_precision(unsigned value);           ///< See FLAC__stream_encoder_set_qlp_coeff_precision()
Chris@1: 			virtual bool set_do_qlp_coeff_prec_search(bool value);          ///< See FLAC__stream_encoder_set_do_qlp_coeff_prec_search()
Chris@1: 			virtual bool set_do_escape_coding(bool value);                  ///< See FLAC__stream_encoder_set_do_escape_coding()
Chris@1: 			virtual bool set_do_exhaustive_model_search(bool value);        ///< See FLAC__stream_encoder_set_do_exhaustive_model_search()
Chris@1: 			virtual bool set_min_residual_partition_order(unsigned value);  ///< See FLAC__stream_encoder_set_min_residual_partition_order()
Chris@1: 			virtual bool set_max_residual_partition_order(unsigned value);  ///< See FLAC__stream_encoder_set_max_residual_partition_order()
Chris@1: 			virtual bool set_rice_parameter_search_dist(unsigned value);    ///< See FLAC__stream_encoder_set_rice_parameter_search_dist()
Chris@1: 			virtual bool set_total_samples_estimate(FLAC__uint64 value);    ///< See FLAC__stream_encoder_set_total_samples_estimate()
Chris@1: 			virtual bool set_metadata(::FLAC__StreamMetadata **metadata, unsigned num_blocks);    ///< See FLAC__stream_encoder_set_metadata()
Chris@1: 			virtual bool set_metadata(FLAC::Metadata::Prototype **metadata, unsigned num_blocks); ///< See FLAC__stream_encoder_set_metadata()
Chris@1: 
Chris@1: 			/* get_state() is not virtual since we want subclasses to be able to return their own state */
Chris@1: 			State get_state() const;                                   ///< See FLAC__stream_encoder_get_state()
Chris@1: 			virtual Decoder::Stream::State get_verify_decoder_state() const; ///< See FLAC__stream_encoder_get_verify_decoder_state()
Chris@1: 			virtual void get_verify_decoder_error_stats(FLAC__uint64 *absolute_sample, unsigned *frame_number, unsigned *channel, unsigned *sample, FLAC__int32 *expected, FLAC__int32 *got); ///< See FLAC__stream_encoder_get_verify_decoder_error_stats()
Chris@1: 			virtual bool     get_verify() const;                       ///< See FLAC__stream_encoder_get_verify()
Chris@1: 			virtual bool     get_streamable_subset() const;            ///< See FLAC__stream_encoder_get_streamable_subset()
Chris@1: 			virtual bool     get_do_mid_side_stereo() const;           ///< See FLAC__stream_encoder_get_do_mid_side_stereo()
Chris@1: 			virtual bool     get_loose_mid_side_stereo() const;        ///< See FLAC__stream_encoder_get_loose_mid_side_stereo()
Chris@1: 			virtual unsigned get_channels() const;                     ///< See FLAC__stream_encoder_get_channels()
Chris@1: 			virtual unsigned get_bits_per_sample() const;              ///< See FLAC__stream_encoder_get_bits_per_sample()
Chris@1: 			virtual unsigned get_sample_rate() const;                  ///< See FLAC__stream_encoder_get_sample_rate()
Chris@1: 			virtual unsigned get_blocksize() const;                    ///< See FLAC__stream_encoder_get_blocksize()
Chris@1: 			virtual unsigned get_max_lpc_order() const;                ///< See FLAC__stream_encoder_get_max_lpc_order()
Chris@1: 			virtual unsigned get_qlp_coeff_precision() const;          ///< See FLAC__stream_encoder_get_qlp_coeff_precision()
Chris@1: 			virtual bool     get_do_qlp_coeff_prec_search() const;     ///< See FLAC__stream_encoder_get_do_qlp_coeff_prec_search()
Chris@1: 			virtual bool     get_do_escape_coding() const;             ///< See FLAC__stream_encoder_get_do_escape_coding()
Chris@1: 			virtual bool     get_do_exhaustive_model_search() const;   ///< See FLAC__stream_encoder_get_do_exhaustive_model_search()
Chris@1: 			virtual unsigned get_min_residual_partition_order() const; ///< See FLAC__stream_encoder_get_min_residual_partition_order()
Chris@1: 			virtual unsigned get_max_residual_partition_order() const; ///< See FLAC__stream_encoder_get_max_residual_partition_order()
Chris@1: 			virtual unsigned get_rice_parameter_search_dist() const;   ///< See FLAC__stream_encoder_get_rice_parameter_search_dist()
Chris@1: 			virtual FLAC__uint64 get_total_samples_estimate() const;   ///< See FLAC__stream_encoder_get_total_samples_estimate()
Chris@1: 
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init();            ///< See FLAC__stream_encoder_init_stream()
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init_ogg();        ///< See FLAC__stream_encoder_init_ogg_stream()
Chris@1: 
Chris@1: 			virtual bool finish(); ///< See FLAC__stream_encoder_finish()
Chris@1: 
Chris@1: 			virtual bool process(const FLAC__int32 * const buffer[], unsigned samples);     ///< See FLAC__stream_encoder_process()
Chris@1: 			virtual bool process_interleaved(const FLAC__int32 buffer[], unsigned samples); ///< See FLAC__stream_encoder_process_interleaved()
Chris@1: 		protected:
Chris@1: 			/// See FLAC__StreamEncoderReadCallback
Chris@1: 			virtual ::FLAC__StreamEncoderReadStatus read_callback(FLAC__byte buffer[], size_t *bytes);
Chris@1: 
Chris@1: 			/// See FLAC__StreamEncoderWriteCallback
Chris@1: 			virtual ::FLAC__StreamEncoderWriteStatus write_callback(const FLAC__byte buffer[], size_t bytes, unsigned samples, unsigned current_frame) = 0;
Chris@1: 
Chris@1: 			/// See FLAC__StreamEncoderSeekCallback
Chris@1: 			virtual ::FLAC__StreamEncoderSeekStatus seek_callback(FLAC__uint64 absolute_byte_offset);
Chris@1: 
Chris@1: 			/// See FLAC__StreamEncoderTellCallback
Chris@1: 			virtual ::FLAC__StreamEncoderTellStatus tell_callback(FLAC__uint64 *absolute_byte_offset);
Chris@1: 
Chris@1: 			/// See FLAC__StreamEncoderMetadataCallback
Chris@1: 			virtual void metadata_callback(const ::FLAC__StreamMetadata *metadata);
Chris@1: 
Chris@1: #if (defined _MSC_VER) || (defined __BORLANDC__) || (defined __GNUG__ && (__GNUG__ < 2 || (__GNUG__ == 2 && __GNUC_MINOR__ < 96))) || (defined __SUNPRO_CC)
Chris@1: 			// lame hack: some MSVC/GCC versions can't see a protected encoder_ from nested State::resolved_as_cstring()
Chris@1: 			friend State;
Chris@1: #endif
Chris@1: 			::FLAC__StreamEncoder *encoder_;
Chris@1: 
Chris@1: 			static ::FLAC__StreamEncoderReadStatus read_callback_(const ::FLAC__StreamEncoder *encoder, FLAC__byte buffer[], size_t *bytes, void *client_data);
Chris@1: 			static ::FLAC__StreamEncoderWriteStatus write_callback_(const ::FLAC__StreamEncoder *encoder, const FLAC__byte buffer[], size_t bytes, unsigned samples, unsigned current_frame, void *client_data);
Chris@1: 			static ::FLAC__StreamEncoderSeekStatus seek_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 absolute_byte_offset, void *client_data);
Chris@1: 			static ::FLAC__StreamEncoderTellStatus tell_callback_(const FLAC__StreamEncoder *encoder, FLAC__uint64 *absolute_byte_offset, void *client_data);
Chris@1: 			static void metadata_callback_(const ::FLAC__StreamEncoder *encoder, const ::FLAC__StreamMetadata *metadata, void *client_data);
Chris@1: 		private:
Chris@1: 			// Private and undefined so you can't use them:
Chris@1: 			Stream(const Stream &);
Chris@1: 			void operator=(const Stream &);
Chris@1: 		};
Chris@1: 
Chris@1: 		/** \ingroup flacpp_encoder
Chris@1: 		 *  \brief
Chris@1: 		 *  This class wraps the ::FLAC__StreamEncoder.  If you are
Chris@1: 		 *  not encoding to a file, you may need to use
Chris@1: 		 *  FLAC::Encoder::Stream.
Chris@1: 		 *
Chris@1: 		 * The usage of this class is similar to FLAC__StreamEncoder,
Chris@1: 		 * except instead of providing callbacks to
Chris@1: 		 * FLAC__stream_encoder_init*_FILE() or
Chris@1: 		 * FLAC__stream_encoder_init*_file(), you will inherit from this
Chris@1: 		 * class and override the virtual callback functions with your
Chris@1: 		 * own implementations, then call init() or init_ogg().  The rest
Chris@1: 		 * of the calls work the same as in the C layer.
Chris@1: 		 *
Chris@1: 		 * There are no mandatory callbacks; all the callbacks from
Chris@1: 		 * FLAC::Encoder::Stream are implemented here fully and support
Chris@1: 		 * full post-encode STREAMINFO and SEEKTABLE updating.  There is
Chris@1: 		 * only an optional progress callback which you may override to
Chris@1: 		 * get periodic reports on the progress of the encode.
Chris@1: 		 */
Chris@1: 		class FLACPP_API File: public Stream {
Chris@1: 		public:
Chris@1: 			File();
Chris@1: 			virtual ~File();
Chris@1: 
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init(FILE *file);                      ///< See FLAC__stream_encoder_init_FILE()
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init(const char *filename);            ///< See FLAC__stream_encoder_init_file()
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init(const std::string &filename);     ///< See FLAC__stream_encoder_init_file()
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init_ogg(FILE *file);                  ///< See FLAC__stream_encoder_init_ogg_FILE()
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init_ogg(const char *filename);        ///< See FLAC__stream_encoder_init_ogg_file()
Chris@1: 			virtual ::FLAC__StreamEncoderInitStatus init_ogg(const std::string &filename); ///< See FLAC__stream_encoder_init_ogg_file()
Chris@1: 		protected:
Chris@1: 			/// See FLAC__StreamEncoderProgressCallback
Chris@1: 			virtual void progress_callback(FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate);
Chris@1: 
Chris@1: 			/// This is a dummy implementation to satisfy the pure virtual in Stream that is actually supplied internally by the C layer
Chris@1: 			virtual ::FLAC__StreamEncoderWriteStatus write_callback(const FLAC__byte buffer[], size_t bytes, unsigned samples, unsigned current_frame);
Chris@1: 		private:
Chris@1: 			static void progress_callback_(const ::FLAC__StreamEncoder *encoder, FLAC__uint64 bytes_written, FLAC__uint64 samples_written, unsigned frames_written, unsigned total_frames_estimate, void *client_data);
Chris@1: 
Chris@1: 			// Private and undefined so you can't use them:
Chris@1: 			File(const Stream &);
Chris@1: 			void operator=(const Stream &);
Chris@1: 		};
Chris@1: 
Chris@1: 	}
Chris@1: }
Chris@1: 
Chris@1: #endif