From ffef90a1c1f1990581d8fbc22193bfc766d23d3b Mon Sep 17 00:00:00 2001
From: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
Date: Tue, 19 Mar 2019 18:41:58 +0200
Subject: utils: rkisp1: Add test capture script

The script captures raw or processed frames from cameras based on the
Rockchip ISP1. 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>
---
 utils/rkisp1/rkisp1-capture.sh | 182 +++++++++++++++++++++++++++++++++++++++++
 1 file changed, 182 insertions(+)
 create mode 100755 utils/rkisp1/rkisp1-capture.sh

(limited to 'utils')

diff --git a/utils/rkisp1/rkisp1-capture.sh b/utils/rkisp1/rkisp1-capture.sh
new file mode 100755
index 00000000..cffe9fed
--- /dev/null
+++ b/utils/rkisp1/rkisp1-capture.sh
@@ -0,0 +1,182 @@
+#!/bin/sh
+# SPDX-License-Identifier: GPL-2.0-or-later
+# Copyright (C) 2019, Google Inc.
+#
+# Author: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
+#
+# rkisp-capture.sh - Capture processed frames from cameras based on the
+# Rockchip ISP1
+#
+# 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)
+# - raw2rgbpnm (from git://git.retiisi.org.uk/~sailus/raw2rgbpnm.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
+		echo "Sensor '$sensor_name' not found." >&2
+		exit 1
+	fi
+
+	echo "$sensor_name $bus"
+}
+
+# Locate the media device
+find_media_device() {
+	local mdev
+	local name=$1
+
+	for mdev in /dev/media* ; do
+		media-ctl -d $mdev -p | grep -q "^driver[ \t]*$name$" && break
+		mdev=
+	done
+
+	if [[ -z $mdev ]] ; then
+		echo "$name media device not found." >&2
+		exit 1
+	fi
+
+	echo $mdev
+}
+
+# Get the sensor format
+get_sensor_format() {
+	local format
+	local sensor=$1
+
+	format=$($mediactl --get-v4l2 "'$sensor':0" | sed 's/\[\([^ ]*\).*/\1/')
+	sensor_mbus_code=$(echo $format | sed 's/fmt:\([A-Z0-9_]*\).*/\1/')
+	sensor_size=$(echo $format | sed 's/[^\/]*\/\([0-9x]*\).*/\1/')
+
+	echo "Capturing ${sensor_size} from sensor $sensor in ${sensor_mbus_code}"
+}
+
+# Configure the pipeline
+configure_pipeline() {
+	local format="fmt:$sensor_mbus_code/$sensor_size"
+	local capture_mbus_code=$1
+	local capture_size=$2
+
+	echo "Configuring pipeline for $sensor in $format"
+
+	$mediactl -r
+
+	$mediactl -l "'$sensor':0 -> 'rockchip-sy-mipi-dphy':0 [1]"
+	$mediactl -l "'rockchip-sy-mipi-dphy':1 -> 'rkisp1-isp-subdev':0 [1]"
+	$mediactl -l "'rkisp1-isp-subdev':2 -> 'rkisp1_mainpath':0 [1]"
+
+	$mediactl -V "\"$sensor\":0 [$format]"
+	$mediactl -V "'rockchip-sy-mipi-dphy':1 [$format]"
+	$mediactl -V "'rkisp1-isp-subdev':0 [$format crop:(0,0)/$sensor_size]"
+	$mediactl -V "'rkisp1-isp-subdev':2 [fmt:$capture_mbus_code/$capture_size crop:(0,0)/$capture_size]"
+}
+
+# Capture frames
+capture_frames() {
+	local file_op
+	local capture_format=$1
+	local capture_size=$2
+	local frame_count=$3
+	local save_file=$4
+
+	if [[ $save_file -eq 1 ]]; then
+		file_op="--file=/tmp/frame-#.bin"
+	fi
+
+	yavta -c$frame_count -n5 -I -f $capture_format -s $capture_size \
+		$file_op $($mediactl -e "rkisp1_mainpath")
+}
+
+# Convert captured files to ppm
+convert_files() {
+	local format=$1
+	local size=$2
+	local frame_count=$3
+
+	echo "Converting ${frame_count} frames (${size})"
+
+	for i in `seq 0 $(($frame_count - 1))`; do
+		i=$(printf %06u $i)
+		raw2rgbpnm -f $format -s $size /tmp/frame-$i.bin /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"
+	echo "-r, --raw         Capture RAW frames"
+	echo "-s, --size wxh    Frame size"
+}
+
+# Parse command line arguments
+capture_size=1024x768
+frame_count=10
+raw=false
+save_file=1
+
+while [[ $# -ne 0 ]] ; do
+	case $1 in
+	-c|--count)
+		frame_count=$2
+		shift 2
+		;;
+	--no-save)
+		save_file=0
+		shift
+		;;
+
+	-r|--raw)
+		raw=true
+		shift
+		;;
+	-s|--size)
+		capture_size=$2
+		shift 2
+		;;
+	-*)
+		echo "Unsupported option $1" >&2
+		usage $0
+		exit 1
+		;;
+	*)
+		break
+		;;
+	esac
+done
+
+if [[ $# -ne 1 ]] ; then
+	usage $0
+	exit 1
+fi
+
+sensor_name=$1
+
+modprobe mipi_dphy_sy
+modprobe video_rkisp1
+
+sensor=$(find_sensor $sensor_name) || exit
+mdev=$(find_media_device rkisp1) || exit
+mediactl="media-ctl -d $mdev"
+
+get_sensor_format "$sensor"
+if [[ $raw == true ]] ; then
+	capture_format=$(echo $sensor_mbus_code | sed 's/_[0-9X]$//')
+	capture_mbus_code=$sensor_mbus_code
+else
+	capture_format=YUYV
+	capture_mbus_code=YUYV8_2X8
+fi
+
+configure_pipeline $capture_mbus_code $capture_size
+capture_frames $capture_format $capture_size $frame_count $save_file
+[[ $save_file -eq 1 ]] && convert_files $capture_format $capture_size $frame_count
-- 
cgit v1.2.1