mediamtx/internal/rpicamera/exe/parameters.c

#include <stdlib.h>
#include <string.h>
#include <stdbool.h>
#include <stdarg.h>
#include <stdio.h>

#include <linux/videodev2.h>

#include "parameters.h"

static char errbuf[256];

static void set_error(const char *format, ...) {
    va_list args;
    va_start(args, format);
    vsnprintf(errbuf, 256, format, args);
}

const char *parameters_get_error() {
    return errbuf;
}

bool parameters_unserialize(parameters_t *params, const uint8_t *buf, size_t buf_size) {
    char *tmp = malloc(buf_size + 1);
    memcpy(tmp, buf, buf_size);
    tmp[buf_size] = 0x00;

    char *entry;

    while ((entry = strsep(&tmp, " ")) != NULL)  {
        char *key = strsep(&entry, "=");
        char *val = strsep(&entry, "=");

        if (strcmp(key, "CameraID") == 0) {
            params->camera_id = atoi(val);
        } else if (strcmp(key, "Width") == 0) {
            params->width = atoi(val);
        } else if (strcmp(key, "Height") == 0) {
            params->height = atoi(val);
        } else if (strcmp(key, "HFlip") == 0) {
            params->h_flip = (strcmp(val, "1") == 0);
        } else if (strcmp(key, "VFlip") == 0) {
            params->v_flip = (strcmp(val, "1") == 0);
        } else if (strcmp(key, "Brightness") == 0) {
            params->brightness = atof(val);
        } else if (strcmp(key, "Contrast") == 0) {
           params->contrast = atof(val);
        } else if (strcmp(key, "Saturation") == 0) {
            params->saturation = atof(val);
        } else if (strcmp(key, "Sharpness") == 0) {
            params->sharpness = atof(val);
        } else if (strcmp(key, "Exposure") == 0) {
            params->exposure = strdup(val);
        } else if (strcmp(key, "AWB") == 0) {
            params->awb = strdup(val);
        } else if (strcmp(key, "Denoise") == 0) {
            params->denoise = strdup(val);
        } else if (strcmp(key, "Shutter") == 0) {
            params->shutter = atoi(val);
        } else if (strcmp(key, "Metering") == 0) {
            params->metering = strdup(val);
        } else if (strcmp(key, "Gain") == 0) {
            params->gain = atof(val);
        } else if (strcmp(key, "EV") == 0) {
            params->ev = atof(val);
        } else if (strcmp(key, "ROI") == 0) {
            if (strlen(val) != 0) {
                params->roi = malloc(sizeof(window_t));
                bool ok = window_load(val, params->roi);
                if (!ok) {
                    set_error("invalid ROI");
                    return false;
                }
            } else {
                params->roi = NULL;
            }
        } else if (strcmp(key, "TuningFile") == 0) {
            params->tuning_file = strdup(val);
        } else if (strcmp(key, "Mode") == 0) {
            if (strlen(val) != 0) {
                params->mode = malloc(sizeof(sensor_mode_t));
                bool ok = sensor_mode_load(val, params->mode);
                if (!ok) {
                    set_error("invalid sensor mode");
                    return false;
                }
            } else {
                params->mode = NULL;
            }
        } else if (strcmp(key, "FPS") == 0) {
            params->fps = atoi(val);
        } else if (strcmp(key, "IDRPeriod") == 0) {
            params->idr_period = atoi(val);
        } else if (strcmp(key, "Bitrate") == 0) {
            params->bitrate = atoi(val);
        } else if (strcmp(key, "Profile") == 0) {
            if (strcmp(val, "baseline") == 0) {
                params->profile = V4L2_MPEG_VIDEO_H264_PROFILE_BASELINE;
            } else if (strcmp(val, "main") == 0) {
                params->profile = V4L2_MPEG_VIDEO_H264_PROFILE_MAIN;
            } else {
                params->profile = V4L2_MPEG_VIDEO_H264_PROFILE_HIGH;
            }
        } else if (strcmp(key, "Level") == 0) {
            if (strcmp(val, "4.0") == 0) {
                params->level = V4L2_MPEG_VIDEO_H264_LEVEL_4_0;
            } else if (strcmp(val, "4.1") == 0) {
                params->level = V4L2_MPEG_VIDEO_H264_LEVEL_4_1;
            } else {
                params->level = V4L2_MPEG_VIDEO_H264_LEVEL_4_2;
            }
        } else if (strcmp(key, "AfMode") == 0) {
            params->af_mode = strdup(val);
        } else if (strcmp(key, "AfRange") == 0) {
            params->af_range = strdup(val);
        } else if (strcmp(key, "AfSpeed") == 0) {
            params->af_speed = strdup(val);
        } else if (strcmp(key, "LensPosition") == 0) {
            params->lens_position = atof(val);
        } else if (strcmp(key, "AfWindow") == 0) {
            if (strlen(val) != 0) {
                params->af_window = malloc(sizeof(window_t));
                bool ok = window_load(val, params->af_window);
                if (!ok) {
                    set_error("invalid AfWindow");
                    return false;
                }
            } else {
                params->af_window = NULL;
            }
        }
    }

    free(tmp);

    params->buffer_count = 6;
    params->capture_buffer_count = params->buffer_count * 2;

    return true;
}

void parameters_destroy(parameters_t *params) {
    free(params->exposure);
    free(params->awb);
    free(params->denoise);
    free(params->metering);
    free(params->tuning_file);
    free(params->af_mode);
    free(params->af_range);
    free(params->af_speed);

    if (params->roi != NULL) {
        free(params->roi);
    }
    if (params->mode != NULL) {
        free(params->mode);
    }
    if (params->af_window != NULL) {
        free(params->af_window);
    }
}