Michael Fabian 'Xaymar' Dirks
83 lines
2.4 KiB
C++
83 lines
2.4 KiB
C++
/*
|
|
* AV1 Encoder for Open Broadcaster Software Studio
|
|
* Copyright (C) 2017 Michael Fabian Dirks
|
|
*
|
|
* This program is free software; you can redistribute it and/or modify
|
|
* it under the terms of the GNU General Public License as published by
|
|
* the Free Software Foundation; either version 2 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* This program is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
* GNU General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#pragma once
|
|
extern "C" {
|
|
#pragma warning(push)
|
|
#pragma warning(disable:4201)
|
|
#include "libobs/obs-module.h"
|
|
#include <aom/aom.h>
|
|
#include <aom/aom_encoder.h>
|
|
#pragma warning(pop)
|
|
}
|
|
|
|
class AV1Encoder {
|
|
public:
|
|
static const char *get_name(void *);
|
|
|
|
static void *create(obs_data_t *, obs_encoder_t *);
|
|
AV1Encoder(obs_data_t *, obs_encoder_t *);
|
|
|
|
static void destroy(void *);
|
|
~AV1Encoder();
|
|
|
|
static void get_defaults(obs_data_t *);
|
|
|
|
static obs_properties_t *get_properties(void *);
|
|
void get_properties(obs_properties_t *);
|
|
|
|
static bool update(void *, obs_data_t *);
|
|
bool update(obs_data_t *);
|
|
|
|
/// Called after encoder create, but before encode.
|
|
static void get_video_info(void *, struct video_scale_info *);
|
|
void get_video_info(struct video_scale_info *);
|
|
|
|
/// Encode a frame.
|
|
static bool encode(void *, struct encoder_frame *, struct encoder_packet *, bool *);
|
|
bool encode(struct encoder_frame *, struct encoder_packet *, bool *);
|
|
|
|
/// Retrieve encoder extra data.
|
|
static bool get_extra_data(void *, uint8_t **, size_t *);
|
|
bool get_extra_data(uint8_t **, size_t *);
|
|
|
|
private:
|
|
obs_encoder_t* m_self;
|
|
|
|
//AV1
|
|
aom_img_fmt m_imageFormat;
|
|
aom_image_t m_image;
|
|
aom_codec_enc_cfg_t m_configuration;
|
|
aom_codec_ctx_t m_codec;
|
|
|
|
uint32_t width, height;
|
|
uint32_t maxencodetime;
|
|
};
|
|
|
|
// Taken from tools_common.h
|
|
typedef struct AvxInterface {
|
|
const char *const name;
|
|
const uint32_t fourcc;
|
|
aom_codec_iface_t *(*const codec_interface)();
|
|
} AvxInterface;
|
|
|
|
int get_aom_encoder_count(void);
|
|
const AvxInterface *get_aom_encoder_by_index(int i);
|
|
const AvxInterface *get_aom_encoder_by_name(const char *name);
|