diff options
Diffstat (limited to 'scripts/vsp-lib.sh')
| -rwxr-xr-x | scripts/vsp-lib.sh | 441 |
1 files changed, 441 insertions, 0 deletions
diff --git a/scripts/vsp-lib.sh b/scripts/vsp-lib.sh new file mode 100755 index 0000000..4d3d166 --- /dev/null +++ b/scripts/vsp-lib.sh @@ -0,0 +1,441 @@ +#!/bin/sh + +mediactl='media-ctl' +yavta='yavta' + +# ------------------------------------------------------------------------------ +# Miscellaneous +# + +vsp1_device() { + $mediactl -d $mdev -p | grep 'bus info' | sed 's/.*platform://' +} + +vsp1_has_feature() { + feature=$1 + + $mediactl -d $mdev -p | grep -q -- "- entity.*$feature" +} + +vsp1_count_rpfs() { + $mediactl -d $mdev -p | grep -- '- entity.*rpf.[0-9] [^i]' | wc -l +} + +vsp1_count_wpfs() { + $mediactl -d $mdev -p | grep -- '- entity.*wpf.[0-9] [^o]' | wc -l +} + +vsp1_count_bru_inputs() { + num_pads=`media-ctl -p | grep 'entity.*bru' | sed 's/.*(\([0-9]\) pads.*/\1/'` + echo $((num_pads-1)) +} + +vsp1_entity_get_size() { + entity=$1 + pad=$2 + + $mediactl -d $mdev --get-v4l2 "'$dev $entity':$pad" | grep fmt | \ + sed 's/.*\/\([0-9x]*\).*/\1/' +} + +# ------------------------------------------------------------------------------ +# Image and histogram comparison +# + +# +# Compare the two frames for exact match. +# +compare_frame_exact() { + img_a=$3 + img_b=$4 + + match='fail' + diff -q $img_a $img_b > /dev/null && match='pass' + + echo "Compared $img_a and $img_b: $match" | ./logger.sh check >> $logfile + + if [ $match = 'pass' ] ; then + return 0 + else + return 1 + fi +} + +# +# Compare the two frames using a fuzzy match algorithm to account for errors +# introduced by the YUV packing. Accept a maximum 1% mean average error over +# the whole frame with no more than 5% of the pixels differing. +# +compare_frame_fuzzy() { + fmt=$(echo $1 | sed 's/M$//') + size=$2 + img_a=$3 + img_b=$4 + + png_a=${img_a/bin/png} + png_b=${img_b/bin/png} + + raw2rgbpnm -f $fmt -s $size $img_a $png_a > /dev/null + raw2rgbpnm -f $fmt -s $size $img_b $png_b > /dev/null + + ae=$(compare -metric ae $png_a $png_b /dev/null 2>&1) + mae=$(compare -metric mae $png_a $png_b /dev/null 2>&1 | sed 's/.*(\(.*\))/\1/') + + rm $png_a + rm $png_b + + width=$(echo $size | cut -d 'x' -f 1) + height=$(echo $size | cut -d 'x' -f 2) + + ae_match=$(echo $ae $width $height | awk '{ if ($1 / $2 / $3 < 0.05) { print "pass" } else { print "fail" } }') + mae_match=$(echo $mae | awk '{ if ($1 < 0.01) { print "pass" } else { print "fail" } }') + + echo "Compared $img_a and $img_b: ae $ae ($ae_match) mae $mae ($mae_match)" | ./logger.sh check >> $logfile + + if [ $ae_match = 'pass' -a $mae_match = 'pass' ] ; then + return 0 + else + return 1 + fi +} + +compare_frames() { + method=$1 + reftype=$2 + format=$3 + wpf=$4 + + fmt=$(echo $format | tr '[:upper:]' '[:lower:]') + size=$(vsp1_entity_get_size wpf.$wpf 1) + + case $format in + ARGB555) + reference="frame-$reftype-$fmt-$size-alpha255.bin" + ;; + ABGR32 | ARGB32) + reference="frame-$reftype-$fmt-$size-alpha200.bin" + ;; + XRGB555) + # XRGB555 has the X bit hardcoded to 0 + reference="frame-$reftype-$fmt-$size-alpha0.bin" + ;; + XBGR32 | XRGB32) + # The X bits are configurable with a default value of 255 + reference="frame-$reftype-$fmt-$size-alpha255.bin" + ;; + *) + reference="frame-$reftype-$fmt-$size.bin" + ;; + esac + + result="pass" + for frame in frame-*.bin ; do + (compare_frame_$method $format $size $frame frames/$reference) || { + mv $frame ${0/.sh/}-${frame/.bin/-$reftype-$fmt-$size.bin} ; + result="fail" + } + done + + echo $result +} + +compare_histogram() { + histo_a=$1 + histo_b=$2 + + match='fail' + diff -q $histo_a $histo_b > /dev/null && match='pass' + + echo "Compared $histo_a and $histo_b: $match" | ./logger.sh check >> $logfile + + if [ $match = 'pass' ] ; then + return 0 + else + return 1 + fi +} + +compare_histograms() { + format=$1 + wpf=$2 + + fmt=$(echo $format | tr '[:upper:]' '[:lower:]') + size=$(vsp1_entity_get_size wpf.$wpf 1) + reference="histo-reference-$fmt-$size.bin" + + result="pass" + for histo in histo-*.bin ; do + (compare_histogram $histo frames/$reference) || { + mv $histo ${0/.sh/}-${histo/.bin/-$fmt.bin} ; + result="fail" + } + done + + echo $result +} + +# ------------------------------------------------------------------------------ +# Pipeline configuration +# + +pipe_none() { + # Nothing to be done + return +} + +pipe_rpf_bru() { + ninputs=$1 + + bru_output=$(vsp1_count_bru_inputs) + + for input in `seq 0 1 $((ninputs-1))` ; do + $mediactl -d $mdev -l "'$dev rpf.$input':1 -> '$dev bru':$input [1]" + done + $mediactl -d $mdev -l "'$dev bru':$bru_output -> '$dev wpf.0':0 [1]" + $mediactl -d $mdev -l "'$dev wpf.0':1 -> '$dev wpf.0 output':0 [1]" +} + +pipe_rpf_bru_uds() { + bru_output=$(vsp1_count_bru_inputs) + + $mediactl -d $mdev -l "'$dev rpf.0':1 -> '$dev bru':0 [1]" + $mediactl -d $mdev -l "'$dev bru':$bru_output -> '$dev uds.0':0 [1]" + $mediactl -d $mdev -l "'$dev uds.0':1 -> '$dev wpf.0':0 [1]" + $mediactl -d $mdev -l "'$dev wpf.0':1 -> '$dev wpf.0 output':0 [1]" +} + +pipe_rpf_hgo() { + $mediactl -d $mdev -l "'$dev rpf.0':1 -> '$dev wpf.0':0 [1]" + $mediactl -d $mdev -l "'$dev rpf.0':1 -> '$dev hgo':0 [1]" + $mediactl -d $mdev -l "'$dev wpf.0':1 -> '$dev wpf.0 output':0 [1]" +} + +pipe_rpf_uds() { + $mediactl -d $mdev -l "'$dev rpf.0':1 -> '$dev uds.0':0 [1]" + $mediactl -d $mdev -l "'$dev uds.0':1 -> '$dev wpf.0':0 [1]" + $mediactl -d $mdev -l "'$dev wpf.0':1 -> '$dev wpf.0 output':0 [1]" +} + +pipe_rpf_uds_bru() { + bru_output=$(vsp1_count_bru_inputs) + + $mediactl -d $mdev -l "'$dev rpf.0':1 -> '$dev uds.0':0 [1]" + $mediactl -d $mdev -l "'$dev uds.0':1 -> '$dev bru':0 [1]" + $mediactl -d $mdev -l "'$dev bru':$bru_output -> '$dev wpf.0':0 [1]" + $mediactl -d $mdev -l "'$dev wpf.0':1 -> '$dev wpf.0 output':0 [1]" +} + +pipe_rpf_wpf() { + rpf=$1 + wpf=$2 + + $mediactl -d $mdev -l "'$dev rpf.$rpf':1 -> '$dev wpf.$wpf':0 [1]" + $mediactl -d $mdev -l "'$dev wpf.$wpf':1 -> '$dev wpf.$wpf output':0 [1]" +} + +pipe_reset() { + $mediactl -d $mdev -r +} + +pipe_configure() { + pipe=${1//-/_} + shift 1 + + pipe_reset + pipe_$pipe $* +} + +# ------------------------------------------------------------------------------ +# Format Configuration +# + +format_v4l2_to_mbus() { + case $1 in + RGB332 | ARGB555 | XRGB555 | RGB565 | BGR24 | RGB24 | XBGR32 | XRGB32 | ABGR32 | ARGB32) + echo "ARGB32"; + ;; + + NV12M | NV16M | NV21M | NV61M | UYVY | VYUY | YUV420M | YUYV | YVYU) + echo "AYUV32" + ;; + + *) + echo "Invalid format $1" >&2 + echo -e "Valid formats are +\tRGB332, ARGB555, XRGB555, RGB565, BGR24, RGB24, +\tXBGR32, XRGB32, ABGR32, ARGB32, +\tNV12M, NV16M, NV21M, NV61M, UYVY, VYUY, YUV420M, YUYV, YVYU" >&2 + exit 1 + esac +} + +format_rpf() { + format=$(format_v4l2_to_mbus $1) + size=$2 + rpf=$3 + + $mediactl -d $mdev -V "'$dev rpf.$rpf':0 [fmt:$format/$size]" +} + +format_rpf_bru() { + format=$(format_v4l2_to_mbus $1) + size=$2 + ninputs=$3 + offset=0 + + bru_output=$(vsp1_count_bru_inputs) + + for input in `seq 0 1 $((ninputs-1))` ; do + offset=$((offset+50)) + $mediactl -d $mdev -V "'$dev rpf.$input':0 [fmt:$format/$size]" + $mediactl -d $mdev -V "'$dev bru':$input [fmt:$format/$size compose:($offset,$offset)/$size]" + done + + $mediactl -d $mdev -V "'$dev bru':$bru_output [fmt:$format/$size]" + $mediactl -d $mdev -V "'$dev wpf.0':0 [fmt:$format/$size]" + $mediactl -d $mdev -V "'$dev wpf.0':1 [fmt:$format/$size]" +} + +format_rpf_bru_uds() { + infmt=$(format_v4l2_to_mbus $1) + insize=$2 + outfmt=$(format_v4l2_to_mbus $3) + outsize=$4 + + bru_output=$(vsp1_count_bru_inputs) + + $mediactl -d $mdev -V "'$dev rpf.0':0 [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev bru':0 [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev bru':$bru_output [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev uds.0':0 [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev uds.0':1 [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev wpf.0':0 [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev wpf.0':1 [fmt:$outfmt/$outsize]" +} + +format_rpf_hgo() { + format=$(format_v4l2_to_mbus $1) + size=$2 + crop=${3:+crop:$3} + compose=${4:+compose:$4} + + $mediactl -d $mdev -V "'$dev rpf.0':0 [fmt:$format/$size]" + $mediactl -d $mdev -V "'$dev wpf.0':0 [fmt:$format/$size]" + $mediactl -d $mdev -V "'$dev wpf.0':1 [fmt:$format/$size]" + $mediactl -d $mdev -V "'$dev hgo':0 [fmt:$format/$size $crop $compose]" +} + +format_rpf_uds() { + infmt=$(format_v4l2_to_mbus $1) + insize=$2 + outfmt=$(format_v4l2_to_mbus $3) + outsize=$4 + + $mediactl -d $mdev -V "'$dev rpf.0':0 [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev uds.0':0 [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev uds.0':1 [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev wpf.0':0 [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev wpf.0':1 [fmt:$outfmt/$outsize]" +} + +format_rpf_uds_bru() { + infmt=$(format_v4l2_to_mbus $1) + insize=$2 + outfmt=$(format_v4l2_to_mbus $3) + outsize=$4 + + bru_output=$(vsp1_count_bru_inputs) + + $mediactl -d $mdev -V "'$dev rpf.0':0 [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev uds.0':0 [fmt:$infmt/$insize]" + $mediactl -d $mdev -V "'$dev uds.0':1 [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev bru':0 [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev bru':$bru_output [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev wpf.0':0 [fmt:$infmt/$outsize]" + $mediactl -d $mdev -V "'$dev wpf.0':1 [fmt:$outfmt/$outsize]" +} + +format_rpf_wpf() { + rpf=$1 + wpf=$2 + infmt=$(format_v4l2_to_mbus $3) + size=$4 + outfmt=$(format_v4l2_to_mbus $5) + crop=$6 + + if [ x$crop != 'x' ] ; then + crop="crop:$crop" + outsize=$(echo $crop | sed 's/.*\///') + else + outsize=$size + fi + + $mediactl -d $mdev -V "'$dev rpf.$rpf':0 [fmt:$infmt/$size]" + $mediactl -d $mdev -V "'$dev wpf.$wpf':0 [fmt:$infmt/$size $crop]" + $mediactl -d $mdev -V "'$dev wpf.$wpf':1 [fmt:$outfmt/$outsize]" +} + +format_wpf() { + format=$(format_v4l2_to_mbus $1) + size=$2 + wpf=$3 + + $mediactl -d $mdev -V "'$dev wpf.$wpf':0 [fmt:$format/$size]" + $mediactl -d $mdev -V "'$dev wpf.$wpf':1 [fmt:$format/$size]" +} + +format_configure() { + pipe=${1//-/_} + shift 1 + + format_$pipe $* +} + +# ------------------------------------------------------------------------------ +# Test run +# + +test_init() { + logfile=${1/sh/log} + features=$2 + + rm -f $logfile + rm -f *.bin + + for mdev in /dev/media* ; do + match='true' + for feature in $features ; do + $(vsp1_has_feature $feature) || { + match='false'; + break; + } + done + + if [ $match == 'true' ] ; then + break + fi + done + + if [ $match == 'false' ] ; then + echo "No device found with feature set $features" | ./logger.sh config >> $logfile + exit 1 + fi + + dev=$(vsp1_device $mdev) + echo "Using device $mdev ($dev)" | ./logger.sh config >> $logfile + + vsp_runner=./vsp-runner.sh +} + +test_start() { + echo "Testing $1" | ./logger.sh >> $logfile + echo -n "Testing $1: " >&2 +} + +test_complete() { + echo "Done: $1" | ./logger.sh >> $logfile + echo $1 >&2 + + rm -f frame-*.bin + rm -f histo-*.bin +} |