forked/ustreamer
1
0
Fork 0
mirror of https://github.com/pikvm/ustreamer.git synced 2026-03-17 04:53:42 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactoring and test recorder

Browse Source
This commit is contained in:
Devaev Maxim
2021-01-01 15:04:12 +03:00
parent 5375781086
commit 0eed7f1b89
37 changed files with 149 additions and 86 deletions
Download Patch File Download Diff File Expand all files Collapse all files

88
Makefile
View File

@@ -1,6 +1,7 @@
-include config.mk -include config.mk
PROG ?= ustreamer USTR ?= ustreamer
REC ?= ustreamer-recorder
DESTDIR ?= DESTDIR ?=
PREFIX ?= /usr/local PREFIX ?= /usr/local
MANPREFIX ?= $(PREFIX)/share/man MANPREFIX ?= $(PREFIX)/share/man
@@ -14,14 +15,17 @@ RPI_VC_LIBS ?= /opt/vc/lib
BUILD ?= build BUILD ?= build
LINTERS_IMAGE ?= $(PROG)-linters LINTERS_IMAGE ?= $(USTR)-linters
# ===== # =====
_PROG_LIBS = -lm -ljpeg -pthread -levent -levent_pthreads -luuid
override CFLAGS += -c -std=c11 -Wall -Wextra -D_GNU_SOURCE override CFLAGS += -c -std=c11 -Wall -Wextra -D_GNU_SOURCE
_PROG_SRCS = $(shell ls \
src/common/*.c \ _COMMON_LIBS = -lm -ljpeg -pthread
_USTR_LIBS = $(_COMMON_LIBS) -levent -levent_pthreads -luuid
_USTR_SRCS = $(shell ls \
src/libs/common/*.c \
src/ustreamer/*.c \ src/ustreamer/*.c \
src/ustreamer/http/*.c \ src/ustreamer/http/*.c \
src/ustreamer/data/*.c \ src/ustreamer/data/*.c \
@@ -29,6 +33,14 @@ _PROG_SRCS = $(shell ls \
src/ustreamer/encoders/hw/*.c \ src/ustreamer/encoders/hw/*.c \
) )
_REC_LIBS = $(_COMMON_LIBS) -lrt -lbcm_host -lvcos -lmmal -lmmal_core -lmmal_util -lmmal_vc_client -lmmal_components -L$(RPI_VC_LIBS)
_REC_SRCS = $(shell ls \
src/libs/common/*.c \
src/libs/rawsink/*.c \
src/libs/h264/*.c \
src/recorder/*.c \
)
define optbool define optbool
$(filter $(shell echo $(1) | tr A-Z a-z), yes on 1) $(filter $(shell echo $(1) | tr A-Z a-z), yes on 1)
@@ -36,23 +48,23 @@ endef
ifneq ($(call optbool,$(WITH_RAWSINK)),) ifneq ($(call optbool,$(WITH_RAWSINK)),)
_PROG_LIBS += -lrt _USTR_LIBS += -lrt
override CFLAGS += -DWITH_RAWSINK override CFLAGS += -DWITH_RAWSINK
_PROG_SRCS += $(shell ls src/rawsink/*.c) _USTR_SRCS += $(shell ls src/libs/rawsink/*.c)
endif endif
ifneq ($(call optbool,$(WITH_OMX)),) ifneq ($(call optbool,$(WITH_OMX)),)
_PROG_LIBS += -lbcm_host -lvcos -lopenmaxil -L$(RPI_VC_LIBS) _USTR_LIBS += -lbcm_host -lvcos -lopenmaxil -L$(RPI_VC_LIBS)
override CFLAGS += -DWITH_OMX -DOMX_SKIP64BIT -I$(RPI_VC_HEADERS) override CFLAGS += -DWITH_OMX -DOMX_SKIP64BIT -I$(RPI_VC_HEADERS)
_PROG_SRCS += $(shell ls src/ustreamer/encoders/omx/*.c) _USTR_SRCS += $(shell ls src/ustreamer/encoders/omx/*.c)
endif endif
ifneq ($(call optbool,$(WITH_GPIO)),) ifneq ($(call optbool,$(WITH_GPIO)),)
_PROG_LIBS += -lgpiod _USTR_LIBS += -lgpiod
override CFLAGS += -DWITH_GPIO override CFLAGS += -DWITH_GPIO
_PROG_SRCS += $(shell ls src/ustreamer/gpio/*.c) _USTR_SRCS += $(shell ls src/ustreamer/gpio/*.c)
endif endif
@@ -65,28 +77,43 @@ endif
WITH_SETPROCTITLE ?= 1 WITH_SETPROCTITLE ?= 1
ifneq ($(call optbool,$(WITH_SETPROCTITLE)),) ifneq ($(call optbool,$(WITH_SETPROCTITLE)),)
ifeq ($(shell uname -s | tr A-Z a-z),linux) ifeq ($(shell uname -s | tr A-Z a-z),linux)
_PROG_LIBS += -lbsd _USTR_LIBS += -lbsd
endif endif
override CFLAGS += -DWITH_SETPROCTITLE override CFLAGS += -DWITH_SETPROCTITLE
endif endif
ifneq ($(call optbool,$(WITH_RAWSINK)),)
ifneq ($(call optbool,$(WITH_OMX)),)
_ENABLE_REC = 1
endif
endif
# ===== # =====
all: $(PROG) all: $(USTR) $(REC)
install: $(PROG) install: $(USTR) $(REC)
install -Dm755 $(PROG) $(DESTDIR)$(PREFIX)/bin/$(PROG) install -Dm755 $(USTR) $(DESTDIR)$(PREFIX)/bin/$(USTR)
install -Dm644 $(PROG).1 $(DESTDIR)$(MANPREFIX)/man1/$(PROG).1 install -Dm644 $(USTR).1 $(DESTDIR)$(MANPREFIX)/man1/$(USTR).1
gzip $(DESTDIR)$(MANPREFIX)/man1/$(PROG).1 gzip $(DESTDIR)$(MANPREFIX)/man1/$(USTR).1
ifneq ($(_ENABLE_REC),)
install -Dm755 $(DESTDIR)$(PREFIX)/bin/$(REC)
endif
install-strip: install install-strip: install
strip $(DESTDIR)$(PREFIX)/bin/$(PROG) strip $(DESTDIR)$(PREFIX)/bin/$(USTR)
ifneq ($(_ENABLE_REC),)
strip $(DESTDIR)$(PREFIX)/bin/$(REC)
endif
uninstall: uninstall:
rm $(DESTDIR)$(PREFIX)/bin/$(PROG) rm -f $(DESTDIR)$(PREFIX)/bin/$(USTR) \
$(DESTDIR)$(PREFIX)/bin/$(REC) \
$(DESTDIR)$(MANPREFIX)/man1/$(USTR).1
regen: regen:
@@ -94,16 +121,29 @@ regen:
tools/make-html-h.py src/ustreamer/data/index.html src/ustreamer/data/index_html.c INDEX tools/make-html-h.py src/ustreamer/data/index.html src/ustreamer/data/index_html.c INDEX
$(PROG): $(_PROG_SRCS:%.c=$(BUILD)/%.o) $(USTR): $(_USTR_SRCS:%.c=$(BUILD)/%.o)
$(info -- LD $@) $(info -- LD $@)
@ $(CC) $^ -o $@ $(LDFLAGS) $(_PROG_LIBS) @ $(CC) $^ -o $@ $(LDFLAGS) $(_USTR_LIBS)
$(info ===== Build complete =====)
$(info == CC = $(CC)) $(info == CC = $(CC))
$(info == LIBS = $(_PROG_LIBS)) $(info == LIBS = $(_USTR_LIBS))
$(info == CFLAGS = $(CFLAGS)) $(info == CFLAGS = $(CFLAGS))
$(info == LDFLAGS = $(LDFLAGS)) $(info == LDFLAGS = $(LDFLAGS))
ifneq ($(_ENABLE_REC),)
$(REC): $(_REC_SRCS:%.c=$(BUILD)/%.o)
$(info -- LD $@)
@ $(CC) $^ -o $@ $(LDFLAGS) $(_REC_LIBS)
$(info == CC = $(CC))
$(info == LIBS = $(_REC_LIBS))
$(info == CFLAGS = $(CFLAGS))
$(info == LDFLAGS = $(LDFLAGS))
else
$(REC):
@ true
endif
$(BUILD)/%.o: %.c $(BUILD)/%.o: %.c
$(info -- CC $<) $(info -- CC $<)
@ mkdir -p $(dir $@) || true @ mkdir -p $(dir $@) || true
@@ -152,6 +192,6 @@ clean-all: linters clean
-it $(LINTERS_IMAGE) bash -c "cd src && rm -rf linters/{.tox,.mypy_cache}" -it $(LINTERS_IMAGE) bash -c "cd src && rm -rf linters/{.tox,.mypy_cache}"
clean: clean:
rm -rf pkg/arch/pkg pkg/arch/src pkg/arch/v*.tar.gz pkg/arch/ustreamer-*.pkg.tar.{xz,zst} rm -rf pkg/arch/pkg pkg/arch/src pkg/arch/v*.tar.gz pkg/arch/ustreamer-*.pkg.tar.{xz,zst}
rm -rf $(PROG) $(BUILD) vgcore.* *.sock rm -rf $(USTR) $(REC) $(BUILD) vgcore.* *.sock
.PHONY: linters .PHONY: linters

0
src/common/config.h → src/libs/common/config.h
View File

0
src/common/frame.c → src/libs/common/frame.c
View File

0
src/common/frame.h → src/libs/common/frame.h
View File

0
src/common/logging.c → src/libs/common/logging.c
View File

0
src/common/logging.h → src/libs/common/logging.h
View File

0
src/common/process.h → src/libs/common/process.h
View File

0
src/common/threading.h → src/libs/common/threading.h
View File

0
src/common/tools.h → src/libs/common/tools.h
View File

0
src/h264/encoder.c → src/libs/h264/encoder.c
View File

0
src/h264/encoder.h → src/libs/h264/encoder.h
View File

0
src/h264/unjpeg.c → src/libs/h264/unjpeg.c
View File

0
src/h264/unjpeg.h → src/libs/h264/unjpeg.h
View File

0
src/rawsink/rawsink.c → src/libs/rawsink/rawsink.c
View File

0
src/rawsink/rawsink.h → src/libs/rawsink/rawsink.h
View File

30
src/recorder/main.c Normal file
View File

@@ -0,0 +1,30 @@
#include <stdio.h>
#include "../common/logging.h"
#include "../common/frame.h"
#include "../rawsink/rawsink.h"
#include "../h264/encoder.h"
int main(void) {
LOGGING_INIT;
log_level = 3;
frame_s *src = frame_init("src");
frame_s *dest = frame_init("dest");
h264_encoder_s *encoder = h264_encoder_init();
rawsink_s *rawsink = rawsink_init("test", false, 0, 0, (long double)encoder->fps / (long double)encoder->gop);
FILE *fp = fopen("test.h264", "wb");
if (rawsink) {
int error = 0;
while ((error = rawsink_client_get(rawsink, src)) != -1) {
if (error == 0 /*|| (error == -2 && src->used > 0)*/) {
if (!h264_encoder_compress(encoder, src, dest, false)) {
LOG_INFO("frame %Lf", get_now_monotonic() - dest->grab_ts);
/*fwrite(dest->data, 1, dest->used, fp);
fflush(fp);*/
}
}
}
}
return 0;
}

6
src/ustreamer/blank.h
View File

@@ -29,9 +29,9 @@
#include <jpeglib.h> #include <jpeglib.h>
#include <linux/videodev2.h> #include <linux/videodev2.h>
#include "../common/tools.h" #include "../libs/common/tools.h"
#include "../common/logging.h" #include "../libs/common/logging.h"
#include "../common/frame.h" #include "../libs/common/frame.h"
#include "data/blank_jpeg.h" #include "data/blank_jpeg.h"

2
src/ustreamer/data/index_html.h
View File

@@ -24,7 +24,7 @@
#include <sys/types.h> #include <sys/types.h>
#include "../../common/config.h" #include "../../libs/common/config.h"
extern const char *const HTML_INDEX_PAGE; extern const char *const HTML_INDEX_PAGE;

10
src/ustreamer/device.h
View File

@@ -41,15 +41,15 @@
#include <linux/videodev2.h> #include <linux/videodev2.h>
#include <linux/v4l2-controls.h> #include <linux/v4l2-controls.h>
#include "../common/tools.h" #include "../libs/common/tools.h"
#include "../common/logging.h" #include "../libs/common/logging.h"
#include "../common/threading.h" #include "../libs/common/threading.h"
#include "../common/frame.h" #include "../libs/common/frame.h"
#include "xioctl.h" #include "xioctl.h"
#ifdef WITH_RAWSINK #ifdef WITH_RAWSINK
# include "../rawsink/rawsink.h" # include "../libs/rawsink/rawsink.h"
#endif #endif

8
src/ustreamer/encoder.h
View File

@@ -30,10 +30,10 @@
#include <pthread.h> #include <pthread.h>
#include <linux/videodev2.h> #include <linux/videodev2.h>
#include "../common/tools.h" #include "../libs/common/tools.h"
#include "../common/threading.h" #include "../libs/common/threading.h"
#include "../common/logging.h" #include "../libs/common/logging.h"
#include "../common/frame.h" #include "../libs/common/frame.h"
#include "device.h" #include "device.h"

4
src/ustreamer/encoders/cpu/encoder.h
View File

@@ -31,8 +31,8 @@
#include <linux/videodev2.h> #include <linux/videodev2.h>
#include "../../../common/tools.h" #include "../../../libs/common/tools.h"
#include "../../../common/frame.h" #include "../../../libs/common/frame.h"
void cpu_encoder_compress(frame_s *src, frame_s *dest, unsigned quality); void cpu_encoder_compress(frame_s *src, frame_s *dest, unsigned quality);

6
src/ustreamer/encoders/hw/encoder.h
View File

@@ -29,9 +29,9 @@
#include <linux/videodev2.h> #include <linux/videodev2.h>
#include "../../../common/tools.h" #include "../../../libs/common/tools.h"
#include "../../../common/logging.h" #include "../../../libs/common/logging.h"
#include "../../../common/frame.h" #include "../../../libs/common/frame.h"
#include "../../xioctl.h" #include "../../xioctl.h"
#include "../../device.h" #include "../../device.h"

2
src/ustreamer/encoders/omx/component.h
View File

@@ -28,7 +28,7 @@
#include <IL/OMX_Core.h> #include <IL/OMX_Core.h>
#include <IL/OMX_Component.h> #include <IL/OMX_Component.h>
#include "../../../common/logging.h" #include "../../../libs/common/logging.h"
#include "formatters.h" #include "formatters.h"

6
src/ustreamer/encoders/omx/encoder.h
View File

@@ -36,9 +36,9 @@
#include <IL/OMX_Broadcom.h> #include <IL/OMX_Broadcom.h>
#include <interface/vcos/vcos_semaphore.h> #include <interface/vcos/vcos_semaphore.h>
#include "../../../common/tools.h" #include "../../../libs/common/tools.h"
#include "../../../common/logging.h" #include "../../../libs/common/logging.h"
#include "../../../common/frame.h" #include "../../../libs/common/frame.h"
#include "../../device.h" #include "../../device.h"
#include "formatters.h" #include "formatters.h"

9
src/ustreamer/encoders/omx/formatters.c
View File

@@ -26,12 +26,6 @@
#define CASE_TO_STRING(_value) \ #define CASE_TO_STRING(_value) \
case _value: { return #_value; } case _value: { return #_value; }
#define CASE_ASSERT(_msg, _value) default: { \
char *_assert_buf; A_CALLOC(_assert_buf, 128); \
sprintf(_assert_buf, _msg ": 0x%08x", _value); \
assert(0 && _assert_buf); \
}
const char *omx_error_to_string(OMX_ERRORTYPE error) { const char *omx_error_to_string(OMX_ERRORTYPE error) {
switch (error) { switch (error) {
CASE_TO_STRING(OMX_ErrorNone); CASE_TO_STRING(OMX_ErrorNone);
@@ -93,9 +87,8 @@ const char *omx_state_to_string(OMX_STATETYPE state) {
CASE_TO_STRING(OMX_StateWaitForResources); CASE_TO_STRING(OMX_StateWaitForResources);
// cppcheck-suppress constArgument // cppcheck-suppress constArgument
// cppcheck-suppress knownArgument // cppcheck-suppress knownArgument
CASE_ASSERT("Unsupported OMX state", state);
} }
assert(0 && "Unsupported OMX state");
} }
#undef CASE_ASSERT
#undef CASE_TO_STRING #undef CASE_TO_STRING

4
src/ustreamer/encoders/omx/formatters.h
View File

@@ -29,8 +29,8 @@
#include <IL/OMX_Core.h> #include <IL/OMX_Core.h>
#include <IL/OMX_Image.h> #include <IL/OMX_Image.h>
#include "../../../common/tools.h" #include "../../../libs/common/tools.h"
#include "../../../common/logging.h" #include "../../../libs/common/logging.h"
#define LOG_ERROR_OMX(_error, _msg, ...) { \ #define LOG_ERROR_OMX(_error, _msg, ...) { \

6
src/ustreamer/gpio/gpio.h
View File

@@ -30,9 +30,9 @@
#include <pthread.h> #include <pthread.h>
#include <gpiod.h> #include <gpiod.h>
#include "../../common/tools.h" #include "../../libs/common/tools.h"
#include "../../common/logging.h" #include "../../libs/common/logging.h"
#include "../../common/threading.h" #include "../../libs/common/threading.h"
typedef struct { typedef struct {

2
src/ustreamer/http/base64.h
View File

@@ -26,7 +26,7 @@
#include <stdint.h> #include <stdint.h>
#include <string.h> #include <string.h>
#include "../../common/tools.h" #include "../../libs/common/tools.h"
char *base64_encode(const uint8_t *str); char *base64_encode(const uint8_t *str);

2
src/ustreamer/http/mime.h
View File

@@ -26,7 +26,7 @@
#include <event2/util.h> #include <event2/util.h>
#include "../../common/tools.h" #include "../../libs/common/tools.h"
const char *guess_mime_type(const char *str); const char *guess_mime_type(const char *str);

2
src/ustreamer/http/path.h
View File

@@ -28,7 +28,7 @@
#endif #endif
#include <string.h> #include <string.h>
#include "../../common/tools.h" #include "../../libs/common/tools.h"
char *simplify_request_path(const char *str); char *simplify_request_path(const char *str);

10
src/ustreamer/http/server.h
View File

@@ -53,11 +53,11 @@
# error Required libevent-pthreads support # error Required libevent-pthreads support
#endif #endif
#include "../../common/tools.h" #include "../../libs/common/tools.h"
#include "../../common/threading.h" #include "../../libs/common/threading.h"
#include "../../common/logging.h" #include "../../libs/common/logging.h"
#include "../../common/process.h" #include "../../libs/common/process.h"
#include "../../common/frame.h" #include "../../libs/common/frame.h"
#include "../data/index_html.h" #include "../data/index_html.h"
#include "../encoder.h" #include "../encoder.h"
#include "../stream.h" #include "../stream.h"

4
src/ustreamer/http/static.h
View File

@@ -29,8 +29,8 @@
#include <sys/stat.h> #include <sys/stat.h>
#include "../../common/tools.h" #include "../../libs/common/tools.h"
#include "../../common/logging.h" #include "../../libs/common/logging.h"
#include "path.h" #include "path.h"

4
src/ustreamer/http/unix.h
View File

@@ -35,8 +35,8 @@
#include <event2/http.h> #include <event2/http.h>
#include <event2/util.h> #include <event2/util.h>
#include "../../common/tools.h" #include "../../libs/common/tools.h"
#include "../../common/logging.h" #include "../../libs/common/logging.h"
evutil_socket_t evhttp_my_bind_unix(struct evhttp *http, const char *path, bool rm, mode_t mode); evutil_socket_t evhttp_my_bind_unix(struct evhttp *http, const char *path, bool rm, mode_t mode);

6
src/ustreamer/main.c
View File

@@ -36,9 +36,9 @@
#include <pthread.h> #include <pthread.h>
#include "../common/tools.h" #include "../libs/common/tools.h"
#include "../common/threading.h" #include "../libs/common/threading.h"
#include "../common/logging.h" #include "../libs/common/logging.h"
#include "options.h" #include "options.h"
#include "device.h" #include "device.h"

10
src/ustreamer/options.h
View File

@@ -33,12 +33,12 @@
#include <errno.h> #include <errno.h>
#include <assert.h> #include <assert.h>
#include "../common/config.h" #include "../libs/common/config.h"
#include "../common/logging.h" #include "../libs/common/logging.h"
#include "../common/process.h" #include "../libs/common/process.h"
#include "../common/frame.h" #include "../libs/common/frame.h"
#ifdef WITH_RAWSINK #ifdef WITH_RAWSINK
# include "../rawsink/rawsink.h" # include "../libs/rawsink/rawsink.h"
#endif #endif
#include "device.h" #include "device.h"

10
src/ustreamer/stream.h
View File

@@ -32,16 +32,16 @@
#include <pthread.h> #include <pthread.h>
#include <linux/videodev2.h> #include <linux/videodev2.h>
#include "../common/tools.h" #include "../libs/common/tools.h"
#include "../common/threading.h" #include "../libs/common/threading.h"
#include "../common/logging.h" #include "../libs/common/logging.h"
#include "../common/frame.h" #include "../libs/common/frame.h"
#include "blank.h" #include "blank.h"
#include "device.h" #include "device.h"
#include "encoder.h" #include "encoder.h"
#ifdef WITH_RAWSINK #ifdef WITH_RAWSINK
# include "../rawsink/rawsink.h" # include "../libs/rawsink/rawsink.h"
#endif #endif
#ifdef WITH_GPIO #ifdef WITH_GPIO
# include "gpio/gpio.h" # include "gpio/gpio.h"

4
src/ustreamer/xioctl.h
View File

@@ -26,8 +26,8 @@
#include <sys/ioctl.h> #include <sys/ioctl.h>
#include "../common/tools.h" #include "../libs/common/tools.h"
#include "../common/logging.h" #include "../libs/common/logging.h"
#ifndef CFG_XIOCTL_RETRIES #ifndef CFG_XIOCTL_RETRIES