Cutlery/calibre
1
0
Fork 0
mirror of https://github.com/kovidgoyal/calibre.git synced 2025-07-09 03:04:10 -04:00
Code Issues Packages Projects Releases Wiki Activity

Move python i/o init to python wrapper function

Browse Source
This commit is contained in:
Kovid Goyal 2024-10-12 11:52:19 +05:30
parent f587af8c56
commit 6799054cf8
No known key found for this signature in database
GPG Key ID: 06BC317B515ACE7C
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/calibre/utils/ffmpeg.c
View File

@ -136,9 +136,7 @@ set_seek_pointers(PyObject *file, PyObject **seek) {
} }
static PyObject* static PyObject*
open_input_file(Transcoder *t, PyObject *input_file) { open_input_file(Transcoder *t) {
if (!set_seek_pointers(input_file, &t->seek_in_input)) return NULL;
if (!(t->read_input = PyObject_GetAttrString(input_file, "read"))) return NULL;
if (!(t->ifmt_ctx = avformat_alloc_context())) { PyErr_NoMemory(); return NULL; } if (!(t->ifmt_ctx = avformat_alloc_context())) { PyErr_NoMemory(); return NULL; }
static const size_t io_bufsize = 8192; static const size_t io_bufsize = 8192;
uint8_t *input_buf; uint8_t *input_buf;
@ -165,9 +163,7 @@ open_input_file(Transcoder *t, PyObject *input_file) {
} }
static PyObject* static PyObject*
open_output_file(Transcoder *t, PyObject *output_file) { open_output_file(Transcoder *t) {
if (!set_seek_pointers(output_file, &t->seek_in_output)) return NULL;
if (!(t->write_output = PyObject_GetAttrString(output_file, "write"))) return NULL;
if (!(t->ofmt_ctx = avformat_alloc_context())) { PyErr_NoMemory(); return NULL; } if (!(t->ofmt_ctx = avformat_alloc_context())) { PyErr_NoMemory(); return NULL; }
static const size_t io_bufsize = 8192; static const size_t io_bufsize = 8192;
uint8_t *input_buf; uint8_t *input_buf;
@ -403,9 +399,13 @@ transcode_single_audio_stream(PyObject *self, PyObject *args, PyObject *kw) {
Transcoder t = {.container_format = "mp4", .output_codec_name = ""}; Transcoder t = {.container_format = "mp4", .output_codec_name = ""};
PyObject *input_file, *output_file; PyObject *input_file, *output_file;
if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|Iss", kwds, &input_file, &output_file, &t.write_output, &t.seek_in_output, &t.output_bitrate, &t.container_format, &t.output_codec_name)) return NULL; if (!PyArg_ParseTupleAndKeywords(args, kw, "OO|Iss", kwds, &input_file, &output_file, &t.write_output, &t.seek_in_output, &t.output_bitrate, &t.container_format, &t.output_codec_name)) return NULL;
if (!set_seek_pointers(input_file, &t.seek_in_input)) return NULL;
if (!(t.read_input = PyObject_GetAttrString(input_file, "read"))) return NULL;
if (!set_seek_pointers(output_file, &t.seek_in_output)) return NULL;
if (!(t.write_output = PyObject_GetAttrString(output_file, "write"))) return NULL;
if (!open_input_file(&t, input_file)) goto cleanup; if (!open_input_file(&t)) goto cleanup;
if (!open_output_file(&t, output_file)) goto cleanup; if (!open_output_file(&t)) goto cleanup;
if (!(t.fifo = av_audio_fifo_alloc(t.enc_ctx->sample_fmt, t.enc_ctx->ch_layout.nb_channels, 1))) { if (!(t.fifo = av_audio_fifo_alloc(t.enc_ctx->sample_fmt, t.enc_ctx->ch_layout.nb_channels, 1))) {
PyErr_NoMemory(); goto cleanup; PyErr_NoMemory(); goto cleanup;
} }