From patchwork Fri Aug 5 18:18:49 2022 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Umang Jain X-Patchwork-Id: 16991 X-Patchwork-Delegate: umang.jain@ideasonboard.com Return-Path: X-Original-To: parsemail@patchwork.libcamera.org Delivered-To: parsemail@patchwork.libcamera.org Received: from lancelot.ideasonboard.com (lancelot.ideasonboard.com [92.243.16.209]) by patchwork.libcamera.org (Postfix) with ESMTPS id E4DF8BE173 for ; Fri, 5 Aug 2022 18:19:06 +0000 (UTC) Received: from lancelot.ideasonboard.com (localhost [IPv6:::1]) by lancelot.ideasonboard.com (Postfix) with ESMTP id A4CD963338; Fri, 5 Aug 2022 20:19:06 +0200 (CEST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=libcamera.org; s=mail; t=1659723546; bh=fGrdO2TuEep+O+72BRzStN55XplH5FweOMO9CGFaOEk=; h=To:Date:In-Reply-To:References:Subject:List-Id:List-Unsubscribe: List-Archive:List-Post:List-Help:List-Subscribe:From:Reply-To: From; b=erFMzaWUWt6NoJESgtlimtVKVQpl4T3/XigjYRULK/i9wfFEAYAK9zjyRQdnx9wjh OjGzdaPXVatEebY6liYC765DXhF2u3qstHZ4Bvy6OwyD1WtvUWyZD5hEXD3lKpo2aX xM8yBPv9mtIDkQZ4rJkzhY2yJuaW2cGAtUO5mYi/bebIOZzLsTKUb6fEXovy8oytd+ EO9QB6oYrx9OS8usXoAaJ12Cw0ImHU/LAxaPhKHsz+k6s6Z6/yTf622JiPH/BsxVJP PRIXrzk+yl4Q3AMQt6L3gpLUiD9V/bNo5kun1X7tR8xaalDFhVQKkwcZ78NZPi/+uE a4JxMdP2e5VWw== Received: from perceval.ideasonboard.com (perceval.ideasonboard.com [IPv6:2001:4b98:dc2:55:216:3eff:fef7:d647]) by lancelot.ideasonboard.com (Postfix) with ESMTPS id 6A6E663338 for ; Fri, 5 Aug 2022 20:19:05 +0200 (CEST) Authentication-Results: lancelot.ideasonboard.com; dkim=pass (1024-bit key; unprotected) header.d=ideasonboard.com header.i=@ideasonboard.com header.b="uEWCf5So"; dkim-atps=neutral Received: from perceval.ideasonboard.com (unknown [IPv6:2401:4900:1f3e:69e0:a479:4ccc:ab2a:bf48]) by perceval.ideasonboard.com (Postfix) with ESMTPSA id 61D9A480; Fri, 5 Aug 2022 20:19:04 +0200 (CEST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/simple; d=ideasonboard.com; s=mail; t=1659723545; bh=fGrdO2TuEep+O+72BRzStN55XplH5FweOMO9CGFaOEk=; h=From:To:Cc:Subject:Date:In-Reply-To:References:From; b=uEWCf5SoQE8DuL//UVHlSPxaDjd9XG+uDOYsdMkJEz5vuXy8IE2wvEPKP75uRKnMf dVbdJCdUu3J8K5pWWdmQ6bIM8nXWFryykToWYhfwDjzdK7PtuEcefXygE1uEbZI+cr RijytFdvSnvkGy1hjQoRDtzDEbC7SveriT0uH0Ss= To: libcamera-devel@lists.libcamera.org Date: Fri, 5 Aug 2022 23:48:49 +0530 Message-Id: <20220805181850.848886-5-umang.jain@ideasonboard.com> X-Mailer: git-send-email 2.31.1 In-Reply-To: <20220805181850.848886-1-umang.jain@ideasonboard.com> References: <20220805181850.848886-1-umang.jain@ideasonboard.com> MIME-Version: 1.0 Subject: [libcamera-devel] [PATCH 4/5] utils: ipu3: Executable helper to read statistics X-BeenThere: libcamera-devel@lists.libcamera.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , X-Patchwork-Original-From: Umang Jain via libcamera-devel From: Umang Jain Reply-To: Umang Jain Errors-To: libcamera-devel-bounces@lists.libcamera.org Sender: "libcamera-devel" Usage: ipu3-read-stats input-file output-file [Currently only reads R,G,B averages out of AWB] To be extended to dump more meaningful stats. Signed-off-by: Umang Jain --- utils/ipu3/ipu3-read-stats.c | 168 +++++++++++++++++++++++++++++++++++ utils/ipu3/meson.build | 2 + 2 files changed, 170 insertions(+) create mode 100644 utils/ipu3/ipu3-read-stats.c diff --git a/utils/ipu3/ipu3-read-stats.c b/utils/ipu3/ipu3-read-stats.c new file mode 100644 index 00000000..cd342a63 --- /dev/null +++ b/utils/ipu3/ipu3-read-stats.c @@ -0,0 +1,168 @@ +/* SPDX-License-Identifier: GPL-2.0-or-later */ +/* + * ipu3-read-stats - Dump IPU3 statistics + * + * Copyright 2022 Umang Jain + */ +#define _GNU_SOURCE + +#include +#include +#include +#include +#include +#include +#include +#include + +#include "../../include/linux/intel-ipu3.h" + +static void usage(const char *argv0) +{ + printf("Utility for dumping IPU3 statistics\n"); + printf("Usage: %s --if --bds --gdc input-file output-file\n", + basename(argv0)); + printf(" widthxheight\n"); + printf("If the output-file '-', output data will be written to standard output\n"); +} + +void readSize(const char *size, unsigned int *w, unsigned int *h) +{ + char *sz = strdup(size); + const char *delim = "x"; + char *token = strtok(sz, delim); + if (token) + *w = atoi(token); + + token = strtok(NULL, delim); + if (token) + *h = atoi(token); +} + +int main(int argc, char *argv[]) +{ + int in_fd; + FILE *fout; + int ret; + unsigned int bds_w, bds_h = 0; + unsigned int iif_w, iif_h = 0; + unsigned int gdc_w, gdc_h = 0; + + if (argc != 9) { + usage(argv[0]); + return 1; + } + + if (strcmp(argv[1], "--if") == 0) { + readSize(argv[2], &iif_w, &iif_h); + } + + if (strcmp(argv[3], "--bds") == 0) { + readSize(argv[4], &bds_w, &bds_h); + } + + if (strcmp(argv[5], "--gdc") == 0) { + readSize(argv[6], &gdc_w, &gdc_h); + } + + if (!bds_w || !bds_h || !iif_w || !iif_h || !gdc_w || !gdc_h) { + usage(argv[0]); + return 1; + } + + in_fd = open(argv[7], O_RDONLY); + if (in_fd == -1) { + fprintf(stderr, "Failed to open statistics file '%s': %s\n", + argv[1], strerror(errno)); + return 1; + } + + if (strcmp(argv[8], "-") == 0) { + fout = stdout; + } else { + fout = fopen(argv[8], "w"); + if (!fout) { + fprintf(stderr, "Failed to open output file '%s': %s\n", + argv[4], strerror(errno)); + close(in_fd); + return 1; + } + } + + int frame = 0; + struct ipu3_uapi_stats_3a *stats = (struct ipu3_uapi_stats_3a *) malloc(sizeof(struct ipu3_uapi_stats_3a)); + + while (1) { + ret = read(in_fd, stats, sizeof(struct ipu3_uapi_stats_3a)); + if (ret < 0) { + fprintf(stderr, "Failed to read stats data: %s\n", + strerror(errno)); + goto done; + } + + if ((unsigned)ret < sizeof(struct ipu3_uapi_stats_3a)) { + if (ret != 0) + fprintf(stderr, "%u bytes of stray data at end of input\n", + ret); + goto done; + } + + /* ================ Dump AWB stats ================ */ + + uint32_t k_min_grid_w = 16; + uint32_t k_max_grid_w = 80; + uint32_t minError = UINT_MAX; + uint32_t best_log2 = 0; + uint32_t stride = 0; + + struct ipu3_uapi_grid_config grid = stats->stats_4a_config.awb_config.grid; + + for (uint32_t shift = grid.block_width_log2; shift <= 6; ++shift) { + uint32_t width = bds_w >> shift; + if (width < k_min_grid_w) + width = k_min_grid_w; + else if (width > k_max_grid_w) + width = k_max_grid_w; + + width = width << shift; + uint32_t error = bds_w > width ? bds_w - width : width - bds_w; + if (error >= minError) + continue; + + minError = error; + stride = width; + best_log2 = shift; + } + + stride = stride >> best_log2; + + /* Sum the per-channel averages */ + double red_sum = 0, green_sum = 0, blue_sum = 0; + for (unsigned int cellY = 0; cellY < grid.height; cellY++) { + for (unsigned int cellX = 0; cellX < grid.width; cellX++) { + uint32_t cell_pos = cellY * stride + cellX; + + const struct ipu3_uapi_awb_set_item *cell = + (const struct ipu3_uapi_awb_set_item *)( + &stats->awb_raw_buffer.meta_data[cell_pos] + ); + const uint8_t G_avg = (cell->Gr_avg + cell->Gb_avg) / 2; + + red_sum += cell->R_avg; + green_sum += G_avg; + blue_sum += cell->B_avg; + } + } + + fprintf(fout, "Frame: %d, redSum: %.1f, greenSum: %.1f, blueSum: %.1f\n", + frame, red_sum, green_sum, blue_sum); + frame++; + } + +done: + free(stats); + close(in_fd); + fclose(fout); + + return ret ? 1 : 0; +} diff --git a/utils/ipu3/meson.build b/utils/ipu3/meson.build index c92cc658..88fc2a1b 100644 --- a/utils/ipu3/meson.build +++ b/utils/ipu3/meson.build @@ -2,3 +2,5 @@ ipu3_pack = executable('ipu3-pack', 'ipu3-pack.c') ipu3_unpack = executable('ipu3-unpack', 'ipu3-unpack.c') + +ipu3_read_stats = executable('ipu3-read-stats', 'ipu3-read-stats.c')