utils: ipu3: Add test capture script

The script captures raw frames from cameras based on the Intel IPU3. It
takes the sensor name as an argument and isn't meant to depend on a
particular platform.

Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
Acked-by: Kieran Bingham <kieran.bingham@ideasonboard.com>

1 files changed, 185 insertions, 0 deletions

diff --git a/utils/ipu3/ipu3-capture.sh b/utils/ipu3/ipu3-capture.sh
new file mode 100755
index 00000000..af189da3
--- /dev/null
+++ b/utils/ipu3/ipu3-capture.sh
@@ -0,0 +1,185 @@
+#!/bin/bash
+# SPDX-License-Identifier: GPL-2.0-or-later
+# Copyright (C) 2018, Google Inc.
+#
+# Author: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
+#
+# ipu3-capture.sh - Capture raw frames from cameras based on the Intel IPU3
+#
+# The scripts makes use of the following tools, which are expected to be
+# executable from the system-wide path or from the local directory:
+#
+# - 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 sensor entity
+find_sensor() {
+	local bus
+	local sensor_name=$1
+
+	bus=$(grep "$sensor_name" /sys/class/video4linux/v4l-subdev*/name | cut -d ' ' -f 2)
+	if [[ -z $bus ]]; then
+		return
+	fi
+
+	echo "$sensor_name $bus"
+}
+
+# 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-cio2$" && break
+		mdev=
+	done
+
+	if [[ -z $mdev ]] ; then
+	       echo "IPU3 media device not found." >&2
+	       exit 1
+	fi
+
+	echo $mdev
+}
+
+# Locate the CSI2 and CIO2 and get the sensor format
+parse_pipeline() {
+	local cio2_queue
+	local resolution
+	local sensor=$1
+
+	read cio2_queue bus_format sensor_size <<< $($mediactl -p | awk "
+	/^- entity [0-9]*:/ {
+		sensor=0;
+	}
+
+	/^- entity [0-9]*: $sensor/ {
+		sensor=1;
+	}
+
+	/^[ \t]*(stream:0)?\[fmt:/ {
+		if (sensor) {
+			gsub(\".*fmt:\", \"\");
+			gsub(\"[] ].*\", \"\");
+			gsub(\"/\", \" \");
+			format=\$0;
+		}
+	}
+
+	/^[ \t]*->/ {
+		if (sensor)
+			cio2=substr(\$3, 0, 1);
+	}
+
+	END {
+		print cio2 \" \" format;
+	}
+	")
+
+	ipu3_csi2="ipu3-csi2 $cio2_queue"
+	ipu3_capture="ipu3-cio2 $cio2_queue"
+
+	sensor_width=$(echo $sensor_size | cut -d 'x' -f 1)
+	sensor_height=$(echo $sensor_size | cut -d 'x' -f 2)
+
+	echo "Using device $mdev with IPU3 CIO2 queue $cio2_queue"
+}
+
+# Configure the pipeline
+configure_pipeline() {
+	local format="fmt:$bus_format/$sensor_size"
+
+	echo "Configuring pipeline for $sensor in $format"
+
+	$mediactl -r
+
+	$mediactl -l "\"$sensor\":0 -> \"$ipu3_csi2\":0[1]"
+	$mediactl -l "\"$ipu3_csi2\":1 -> \"$ipu3_capture\":0[1]"
+
+	$mediactl -V "\"$sensor\":0 [$format]"
+	$mediactl -V "\"$ipu3_csi2\":1 [$format]"
+}
+
+# Capture frames
+capture_frames() {
+	local file_op
+	local frame_count=$1
+	local ipu3_format=IPU3_${bus_format/_1X10/}
+	local save_file=$2
+
+	if [[ $save_file -eq 1 ]]; then
+		file_op="--file=/tmp/frame-#.bin"
+	fi
+
+	yavta -c$frame_count -n5 -I -f $ipu3_format -s $sensor_size $file_op \
+		$($mediactl -e "$ipu3_capture")
+}
+
+# Convert captured files to ppm
+convert_files() {
+	local frame_count=$1
+	local format=${bus_format/_1X10/}
+	local padded_width=$(expr \( $sensor_width + 49 \) / 50 \* 50)
+
+	echo "Converting ${sensor_width}x${sensor_height} (${padded_width}x${sensor_height})"
+
+	for i in `seq -f '%06.0f' 0 $(($frame_count - 1))`; do
+		ipu3-unpack /tmp/frame-$i.bin /tmp/frame-$i.raw
+		raw2pnm -x$padded_width -y$sensor_height -f$format /tmp/frame-$i.raw /tmp/frame-$i.ppm
+	done
+}
+
+# Print usage message
+usage() {
+	echo "Usage: $1 [options] sensor-name"
+	echo "Supported options:"
+	echo "-c,--count n      Number of frame to capture"
+	echo "--no-save         Do not save captured frames to disk"
+}
+
+# Parse command line arguments
+frame_count=10
+save_file=1
+
+while (( "$#" )) ; do
+	case $1 in
+	-c|--count)
+		frame_count=$2
+		shift 2
+		;;
+	--no-save)
+		save_file=0
+		shift
+		;;
+	-*)
+		echo "Unsupported option $1" >&2
+		usage $0
+		exit 1
+		;;
+	*)
+		break
+		;;
+	esac
+done
+
+if [[ $# -ne 1 ]] ; then
+	usage $0
+	exit 1
+fi
+
+sensor_name=$1
+
+sensor=$(find_sensor $sensor_name)
+if [[ -z $sensor ]] ; then
+	echo "Sensor '$sensor_name' not found." >&2
+	exit 1
+fi
+
+mdev=$(find_media_device)
+mediactl="media-ctl -d $mdev"
+
+parse_pipeline $sensor
+configure_pipeline
+capture_frames $frame_count $save_file
+[[ $save_file -eq 1 ]] && convert_files $frame_count