16#ifndef OPENSHOT_FFMPEG_WRITER_H
17#define OPENSHOT_FFMPEG_WRITER_H
122 int64_t video_timestamp;
123 int64_t audio_timestamp;
125 bool prepare_streams;
130 AVStream *audio_st, *video_st;
131 AVCodecContext *video_codec_ctx;
132 AVCodecContext *audio_codec_ctx;
133 SwsContext *img_convert_ctx;
135 uint8_t *audio_outbuf;
136 uint8_t *audio_encoder_buffer;
138 int num_of_rescalers;
139 int rescaler_position;
140 std::vector<SwsContext *> image_rescalers;
142 int audio_outbuf_size;
143 int audio_input_frame_size;
144 int initial_audio_input_frame_size;
145 int audio_input_position;
146 int audio_encoder_buffer_size;
151 int original_sample_rate;
152 int original_channels;
154 std::shared_ptr<openshot::Frame> last_frame;
155 std::deque<std::shared_ptr<openshot::Frame> > spooled_audio_frames;
156 std::deque<std::shared_ptr<openshot::Frame> > spooled_video_frames;
158 std::deque<std::shared_ptr<openshot::Frame> > queued_audio_frames;
159 std::deque<std::shared_ptr<openshot::Frame> > queued_video_frames;
161 std::deque<std::shared_ptr<openshot::Frame> > processed_frames;
162 std::deque<std::shared_ptr<openshot::Frame> > deallocate_frames;
164 std::map<std::shared_ptr<openshot::Frame>, AVFrame *> av_frames;
167 void add_avframe(std::shared_ptr<openshot::Frame> frame, AVFrame *av_frame);
170 AVStream *add_audio_stream();
173 AVStream *add_video_stream();
176 AVFrame *allocate_avframe(
PixelFormat pix_fmt,
int width,
int height,
int *buffer_size, uint8_t *new_buffer);
179 void auto_detect_format();
182 void close_audio(AVFormatContext *oc, AVStream *st);
185 void close_video(AVFormatContext *oc, AVStream *st);
188 void flush_encoders();
191 void initialize_streams();
196 void InitScalers(
int source_width,
int source_height);
199 void open_audio(AVFormatContext *oc, AVStream *st);
202 void open_video(AVFormatContext *oc, AVStream *st);
205 void process_video_packet(std::shared_ptr<openshot::Frame> frame);
208 void write_audio_packets(
bool is_final);
211 bool write_video_packet(std::shared_ptr<openshot::Frame> frame, AVFrame *frame_final);
214 void write_queued_frames();
324 void WriteFrame(std::shared_ptr<openshot::Frame> frame);
Header file for FFmpegUtilities.
Header file for ReaderBase class.
Header file for WriterBase class.
This class uses the FFmpeg libraries, to write and encode video files and audio files.
void Close()
Close the writer.
void SetAudioOptions(bool has_audio, std::string codec, int sample_rate, int channels, openshot::ChannelLayout channel_layout, int bit_rate)
Set audio export options.
void SetOption(openshot::StreamType stream, std::string name, std::string value)
Set custom options (some codecs accept additional params). This must be called after the PrepareStrea...
void PrepareStreams()
Prepare & initialize streams and open codecs. This method is called automatically by the Open() metho...
void SetVideoOptions(bool has_video, std::string codec, openshot::Fraction fps, int width, int height, openshot::Fraction pixel_ratio, bool interlaced, bool top_field_first, int bit_rate)
Set video export options.
bool IsOpen()
Determine if writer is open or closed.
void SetCacheSize(int new_size)
Set the cache size.
void ResampleAudio(int sample_rate, int channels)
Set audio resample options.
int GetCacheSize()
Get the cache size (number of frames to queue before writing)
void WriteHeader()
Write the file header (after the options are set). This method is called automatically by the Open() ...
static bool IsValidCodec(std::string codec_name)
Determine if codec name is valid.
void OutputStreamInfo()
Output the ffmpeg info about this format, streams, and codecs (i.e. dump format)
void WriteFrame(std::shared_ptr< openshot::Frame > frame)
Add a frame to the stack waiting to be encoded.
void WriteTrailer()
Write the file trailer (after all frames are written). This is called automatically by the Close() me...
void RemoveScalers()
Remove & deallocate all software scalers.
This class represents a fraction.
This abstract class is the base class, used by all readers in libopenshot.
This abstract class is the base class, used by writers. Writers are types of classes that encode vide...
This namespace is the default namespace for all code in the openshot library.
ChannelLayout
This enumeration determines the audio channel layout (such as stereo, mono, 5 point surround,...
StreamType
This enumeration designates the type of stream when encoding (video or audio)
@ AUDIO_STREAM
An audio stream (used to determine which type of stream)
@ VIDEO_STREAM
A video stream (used to determine which type of stream)