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>
This commit is contained in:
Laurent Pinchart
parent
b15e378851
commit
b4935e9a4d
1 changed files with 183 additions and 0 deletions
183
utils/ipu3/ipu3-process.sh
Executable file
183
utils/ipu3/ipu3-process.sh
Executable file
|
|
@ -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
|
||||
Loading…
Add table
Add a link
Reference in a new issue