summaryrefslogtreecommitdiff
path: root/src/ipa/raspberrypi/md_parser.cpp
blob: d82c102c9e6516b392aa3792f597047559e2d305 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
/* SPDX-License-Identifier: BSD-2-Clause */
/*
 * Copyright (C) 2019, Raspberry Pi (Trading) Limited
 *
 * md_parser.cpp - image sensor metadata parsers
 */

#include <assert.h>
#include <map>
#include <string.h>

#include "md_parser.hpp"

using namespace RPiController;

// This function goes through the embedded data to find the offsets (not
// values!), in the data block, where the values of the given registers can
// subsequently be found.

// Embedded data tag bytes, from Sony IMX219 datasheet but general to all SMIA
// sensors, I think.

#define LINE_START 0x0a
#define LINE_END_TAG 0x07
#define REG_HI_BITS 0xaa
#define REG_LOW_BITS 0xa5
#define REG_VALUE 0x5a
#define REG_SKIP 0x55

MdParserSmia::ParseStatus MdParserSmia::findRegs(unsigned char *data,
						 uint32_t regs[], int offsets[],
						 unsigned int num_regs)
{
	assert(num_regs > 0);
	if (data[0] != LINE_START)
		return NO_LINE_START;

	unsigned int current_offset = 1; // after the LINE_START
	unsigned int current_line_start = 0, current_line = 0;
	unsigned int reg_num = 0, first_reg = 0;
	ParseStatus retcode = PARSE_OK;
	while (1) {
		int tag = data[current_offset++];
		if ((bits_per_pixel_ == 10 &&
		     (current_offset + 1 - current_line_start) % 5 == 0) ||
		    (bits_per_pixel_ == 12 &&
		     (current_offset + 1 - current_line_start) % 3 == 0)) {
			if (data[current_offset++] != REG_SKIP)
				return BAD_DUMMY;
		}
		int data_byte = data[current_offset++];
		//printf("Offset %u, tag 0x%02x data_byte 0x%02x\n", current_offset-1, tag, data_byte);
		if (tag == LINE_END_TAG) {
			if (data_byte != LINE_END_TAG)
				return BAD_LINE_END;
			if (num_lines_ && ++current_line == num_lines_)
				return MISSING_REGS;
			if (line_length_bytes_) {
				current_offset =
					current_line_start + line_length_bytes_;
				// Require whole line to be in the buffer (if buffer size set).
				if (buffer_size_bytes_ &&
				    current_offset + line_length_bytes_ >
					    buffer_size_bytes_)
					return MISSING_REGS;
				if (data[current_offset] != LINE_START)
					return NO_LINE_START;
			} else {
				// allow a zero line length to mean "hunt for the next line"
				while (data[current_offset] != LINE_START &&
				       current_offset < buffer_size_bytes_)
					current_offset++;
				if (current_offset == buffer_size_bytes_)
					return NO_LINE_START;
			}
			// inc current_offset to after LINE_START
			current_line_start =
				current_offset++;
		} else {
			if (tag == REG_HI_BITS)
				reg_num = (reg_num & 0xff) | (data_byte << 8);
			else if (tag == REG_LOW_BITS)
				reg_num = (reg_num & 0xff00) | data_byte;
			else if (tag == REG_SKIP)
				reg_num++;
			else if (tag == REG_VALUE) {
				while (reg_num >=
				       // assumes registers are in order...
				       regs[first_reg]) {
					if (reg_num == regs[first_reg])
						offsets[first_reg] =
							current_offset - 1;
					if (++first_reg == num_regs)
						return retcode;
				}
				reg_num++;
			} else
				return ILLEGAL_TAG;
		}
	}
}
crop:(0,0)/$sensor_size]" $mediactl -V "'rkisp1_isp':2 [fmt:$capture_mbus_code/$sensor_size crop:(0,0)/$sensor_size]" $mediactl -V "'rkisp1_resizer_mainpath':0 [fmt:$capture_mbus_code/$sensor_size crop:(0,0)/$sensor_size]" $mediactl -V "'rkisp1_resizer_mainpath':1 [fmt:$capture_mbus_code/$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 phy_rockchip_dphy_rx0 modprobe rockchip_isp1 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