2
* Copyright (C) 2013 Savoir-Faire Linux Inc.
3
* Author: Guillaume Roguez <Guillaume.Roguez@savoirfairelinux.com>
5
* This program is free software; you can redistribute it and/or modify
6
* it under the terms of the GNU General Public License as published by
7
* the Free Software Foundation; either version 3 of the License, or
8
* (at your option) any later version.
10
* This program is distributed in the hope that it will be useful,
11
* but WITHOUT ANY WARRANTY; without even the implied warranty of
12
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13
* GNU General Public License for more details.
15
* You should have received a copy of the GNU General Public License
16
* along with this program; if not, write to the Free Software
17
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
20
* Additional permission under GNU GPL version 3 section 7:
22
* If you modify this program, or any covered work, by linking or
23
* combining it with the OpenSSL project's OpenSSL library (or a
24
* modified version of that library), containing parts covered by the
25
* terms of the OpenSSL or SSLeay licenses, Savoir-Faire Linux Inc.
26
* grants you additional permission to convey the resulting work.
27
* Corresponding Source for a non-source form of such a combination
28
* shall include the source code for the parts of OpenSSL used as well
29
* as that of the covered work.
32
#ifndef __VIDEO_ENCODER_H__
33
#define __VIDEO_ENCODER_H__
35
#include "video_base.h"
36
#include "video_scaler.h"
37
#include "noncopyable.h"
43
class AVFormatContext;
48
class VideoEncoder : public VideoCodec {
53
void setInterruptCallback(int (*cb)(void*), void *opaque);
54
void setIOContext(const std::unique_ptr<VideoIOHandle> &ioctx);
55
int openOutput(const char *enc_name, const char *short_name,
56
const char *filename, const char *mime_type);
58
int encode(VideoFrame &input, bool is_keyframe, int frame_number);
60
void print_sdp(std::string &sdp_);
62
/* getWidth and getHeight return size of the encoded frame.
63
* Values have meaning only after openOutput call.
65
int getWidth() const { return dstWidth_; }
66
int getHeight() const { return dstHeight_; }
69
NON_COPYABLE(VideoEncoder);
70
void setScaleDest(void *data, int width, int height, int pix_fmt);
71
void prepareEncoderContext();
72
void forcePresetX264();
73
void extractProfileLevelID(const std::string ¶meters, AVCodecContext *ctx);
75
AVCodec *outputEncoder_;
76
AVCodecContext *encoderCtx_;
77
AVFormatContext *outputCtx_;
80
VideoFrame scaledFrame_;
82
uint8_t *scaledFrameBuffer_;
83
int scaledFrameBufferSize_;
87
#if (LIBAVCODEC_VERSION_MAJOR < 54)
88
uint8_t *encoderBuffer_;
89
int encoderBufferSize_;
95
#endif // __VIDEO_ENCODER_H__