From fb859712f36254ca9c34773b82618a09626c8cd9 Mon Sep 17 00:00:00 2001 From: Mikhail Burakov Date: Sun, 11 Jun 2023 20:36:29 +0200 Subject: Introduce communication proto for video stream --- encode.c | 9 ++++++++- proto.h | 41 +++++++++++++++++++++++++++++++++++++++++ 2 files changed, 49 insertions(+), 1 deletion(-) create mode 100644 proto.h diff --git a/encode.c b/encode.c index be920d1..1c45e63 100644 --- a/encode.c +++ b/encode.c @@ -32,6 +32,7 @@ #include "bitstream.h" #include "gpu.h" #include "hevc.h" +#include "proto.h" #include "toolbox/utils.h" struct EncodeContext { @@ -922,8 +923,14 @@ bool EncodeContextEncodeFrame(struct EncodeContext* encode_context, int fd) { abort(); } + struct Proto proto = { + .size = segment->size, + .type = PROTO_TYPE_VIDEO, + .flags = idr ? PROTO_FLAG_KEYFRAME : 0, + .latency = 0, + }; struct iovec iovec[] = { - {.iov_base = &segment->size, .iov_len = sizeof(segment->size)}, + {.iov_base = &proto, .iov_len = sizeof(proto)}, {.iov_base = segment->buf, .iov_len = segment->size}, }; if (!DrainBuffers(fd, iovec, LENGTH(iovec))) { diff --git a/proto.h b/proto.h new file mode 100644 index 0000000..c8c4d0b --- /dev/null +++ b/proto.h @@ -0,0 +1,41 @@ +/* + * Copyright (C) 2023 Mikhail Burakov. This file is part of streamer. + * + * streamer 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 3 of the License, or + * (at your option) any later version. + * + * streamer 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 streamer. If not, see . + */ + +#ifndef STREAMER_PROTO_H_ +#define STREAMER_PROTO_H_ + +#include +#include + +#define PROTO_TYPE_MISC 0 +#define PROTO_TYPE_VIDEO 1 +#define PROTO_TYPE_AUDIO 2 + +#define PROTO_FLAG_KEYFRAME 1 + +struct Proto { + uint32_t size; + uint8_t type; + uint8_t flags; + uint16_t latency; + uint8_t data[]; +}; + +static_assert(sizeof(struct Proto) == 8 * sizeof(uint8_t), + "Suspicious proto struct size"); + +#endif // STREAMER_PROTO_H_ -- cgit v1.2.3