utils: ipu3: Add test process script

The script processes raw frames through the Intel IPU3 IMGU.

Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>

1 files changed, 183 insertions, 0 deletions

diff --git a/utils/ipu3/ipu3-process.sh b/utils/ipu3/ipu3-process.sh
new file mode 100755
index 00000000..6412aea4
--- /dev/null
+++ b/utils/ipu3/ipu3-process.sh
@@ -0,0 +1,183 @@
+#!/bin/bash
+# SPDX-License-Identifier: GPL-2.0-or-later
+# Copyright (C) 2018, Google Inc.
+#
+# Author: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
+#
+# ipu3-process.sh - Process raw frames with the Intel IPU3
+#
+# The scripts makes use of the following tools, which are expected to be
+# found in $PATH:
+#
+# - media-ctl (from v4l-utils git://linuxtv.org/v4l-utils.git)
+# - raw2pnm (from nvt https://github.com/intel/nvt.git)
+# - yavta (from git://git.ideasonboard.org/yavta.git)
+
+# Locate the media device
+find_media_device() {
+	local mdev
+
+	for mdev in /dev/media* ; do
+		media-ctl -d $mdev -p | grep -q "^driver[ \t]*ipu3-imgu$" && break
+		mdev=
+	done
+
+	if [[ -z $mdev ]] ; then
+	       echo "IPU3 media device not found." >&2
+	       exit 1
+	fi
+
+	echo $mdev
+}
+
+# Configure the pipeline
+configure_pipeline() {
+	local enable_3a=1
+	local enable_out=1
+	local enable_vf=1
+
+	$mediactl -r
+	$mediactl -l "\"ipu3-imgu 0 input\":0 -> \"ipu3-imgu 0\":0[1]"
+	$mediactl -l "\"ipu3-imgu 0\":2 -> \"ipu3-imgu 0 output\":0[$enable_out]"
+	$mediactl -l "\"ipu3-imgu 0\":3 -> \"ipu3-imgu 0 viewfinder\":0[$enable_vf]"
+	$mediactl -l "\"ipu3-imgu 0\":4 -> \"ipu3-imgu 0 3a stat\":0[$enable_3a]"
+}
+
+# Perform frame processing through the IMGU
+process_frames() {
+	configure_pipeline
+
+	local yavta="yavta -n $nbufs -c$frame_count"
+
+	# Save the main and viewfinder outputs to disk, capture and drop 3A
+	# statistics. Sleep 500ms between each execution of yavta to keep the
+	# stdout messages readable.
+	$yavta -f $IMGU_OUT_PIXELFORMAT -s $out_size "-F$ouput_dir/frame-out-#.bin" \
+		$($mediactl -e "ipu3-imgu 0 output") &
+	sleep 0.5
+	$yavta -f $IMGU_VF_PIXELFORMAT -s $vf_size "-F$output_dir/frame-vf-#.bin" \
+		$($mediactl -e "ipu3-imgu 0 viewfinder") &
+	sleep 0.5
+	$yavta $($mediactl -e "ipu3-imgu 0 3a stat") &
+	sleep 0.5
+
+	# Feed the IMGU input.
+	$yavta -f $IMGU_IN_PIXELFORMAT -s $in_size "-F$in_file" \
+		$($mediactl -e "ipu3-imgu 0 input")
+}
+
+# Convert captured files to ppm
+convert_files() {
+	local index=$1
+	local type=$2
+	local size=$3
+	local format=$4
+
+	local width=$(echo $size | awk -F 'x' '{print $1}')
+	local height=$(echo $size | awk -F 'x' '{print $2}')
+	local padded_width=$(expr $(expr $width + 63) / 64 \* 64)
+
+	raw2pnm -x$padded_width -y$height -f$format \
+		$output_dir/frame-$type-$index.bin \
+		$output_dir/frame-$type-$index.ppm
+}
+
+run_test() {
+	IMGU_IN_PIXELFORMAT=IPU3_SGRBG10
+	IMGU_OUT_PIXELFORMAT=NV12
+	IMGU_VF_PIXELFORMAT=NV12
+
+	echo "==== Test ===="
+	echo "input:  $in_file"
+	echo "output: $IMGU_OUT_PIXELFORMAT/$out_size"
+	echo "vf:     $IMGU_VF_PIXELFORMAT/$vf_size"
+
+	process_frames
+
+	for i in `seq -f '%06.0f' 0 $(($frame_count - 1))`; do
+		convert_files $i out $out_size $IMGU_OUT_PIXELFORMAT
+		convert_files $i vf $vf_size $IMGU_VF_PIXELFORMAT
+	done
+}
+
+validate_size() {
+	local size=$1
+	local width=$(echo $size | awk -F 'x' '{print $1}')
+	local height=$(echo $size | awk -F 'x' '{print $2}')
+
+	[[ "x${size}" == "x${width}x${height}" ]]
+}
+
+# Print usage message
+usage() {
+	echo "Usage: $(basename $1) [options] <input-file>"
+	echo "Supported options:"
+	echo "--out size        output frame size (defaults to input size)"
+	echo "--vf size         viewfinder frame size (defaults to input size)"
+	echo ""
+	echo "Where the input file name and size are"
+	echo ""
+	echo "input-file = prefix '-' width 'x' height '.' extension"
+	echo "size = width 'x' height"
+}
+
+# Parse command line arguments
+while (( "$#" )) ; do
+	case $1 in
+	--out)
+		out_size=$2
+		if ! validate_size $out_size ; then
+			echo "Invalid size '$out_size'"
+			usage $0
+			exit 1
+		fi
+		shift 2
+		;;
+	--vf)
+		vf_size=$2
+		if ! validate_size $vf_size ; then
+			echo "Invalid size '$vf_size'"
+			usage $0
+			exit 1
+		fi
+		shift 2
+		;;
+	-*)
+		echo "Unsupported option $1" >&2
+		usage $0
+		exit 1
+		;;
+	*)
+		break
+		;;
+	esac
+done
+
+if [ $# != 1 ] ; then
+	usage $0
+	exit 1
+fi
+
+in_file=$1
+
+# Parse the size from the input file name and perform minimal sanity
+# checks.
+in_size=$(echo $in_file | sed 's/.*-\([0-9]*\)x\([0-9]*\)\.[a-z0-9]*$/\1x\2/')
+validate_size $in_size
+if [[ $? != 0 ]] ; then
+	echo "Invalid input file name $in_file" >&2
+	usage $0
+	exit 1
+fi
+
+out_size=${out_size:-$in_size}
+vf_size=${vf_size:-$in_size}
+
+mdev=$(find_media_device)
+mediactl="media-ctl -d $mdev"
+echo "Using device $mdev"
+
+output_dir="/tmp"
+frame_count=5
+nbufs=7
+run_test