summaryrefslogtreecommitdiff
path: root/utils/rkisp1
diff options
context:
space:
mode:
authorLaurent Pinchart <laurent.pinchart@ideasonboard.com>2019-03-19 18:41:58 +0200
committerLaurent Pinchart <laurent.pinchart@ideasonboard.com>2019-03-29 16:39:10 +0200
commitffef90a1c1f1990581d8fbc22193bfc766d23d3b (patch)
tree85a2b5567739ef459d645214fae7e46f6a86d787 /utils/rkisp1
parentf8386836dfda04c626e63ba9d64c1a4f63867dfa (diff)
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>
Diffstat (limited to 'utils/rkisp1')
-rwxr-xr-xutils/rkisp1/rkisp1-capture.sh182
1 files changed, 182 insertions, 0 deletions
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