summaryrefslogtreecommitdiff
path: root/src/ipa/raspberrypi/controller/rpi/noise.cpp
blob: bcd8b9edaebee845efa8e0dfa121dd2ecdc9c69a (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
/* SPDX-License-Identifier: BSD-2-Clause */
/*
 * Copyright (C) 2019, Raspberry Pi Ltd
 *
 * noise.cpp - Noise control algorithm
 */

#include <math.h>

#include <libcamera/base/log.h>

#include "../device_status.h"
#include "../noise_status.h"

#include "noise.h"

using namespace RPiController;
using namespace libcamera;

LOG_DEFINE_CATEGORY(RPiNoise)

#define NAME "rpi.noise"

Noise::Noise(Controller *controller)
	: Algorithm(controller), modeFactor_(1.0)
{
}

char const *Noise::name() const
{
	return NAME;
}

void Noise::switchMode(CameraMode const &cameraMode,
		       [[maybe_unused]] Metadata *metadata)
{
	/*
	 * For example, we would expect a 2x2 binned mode to have a "noise
	 * factor" of sqrt(2x2) = 2. (can't be less than one, right?)
	 */
	modeFactor_ = std::max(1.0, cameraMode.noiseFactor);
}

int Noise::read(const libcamera::YamlObject &params)
{
	auto value = params["reference_constant"].get<double>();
	if (!value)
		return -EINVAL;
	referenceConstant_ = *value;

	value = params["reference_slope"].get<double>();
	if (!value)
		return -EINVAL;
	referenceSlope_ = *value;

	return 0;
}

void Noise::prepare(Metadata *imageMetadata)
{
	struct DeviceStatus deviceStatus;
	deviceStatus.analogueGain = 1.0; /* keep compiler calm */
	if (imageMetadata->get("device.status", deviceStatus) == 0) {
		/*
		 * There is a slight question as to exactly how the noise
		 * profile, specifically the constant part of it, scales. For
		 * now we assume it all scales the same, and we'll revisit this
		 * if it proves substantially wrong.  NOTE: we may also want to
		 * make some adjustments based on the camera mode (such as
		 * binning), if we knew how to discover it...
		 */
		double factor = sqrt(deviceStatus.analogueGain) / modeFactor_;
		struct NoiseStatus status;
		status.noiseConstant = referenceConstant_ * factor;
		status.noiseSlope = referenceSlope_ * factor;
		imageMetadata->set("noise.status", status);
		LOG(RPiNoise, Debug)
			<< "constant " << status.noiseConstant
			<< " slope " << status.noiseSlope;
	} else
		LOG(RPiNoise, Warning) << " no metadata";
}

/* Register algorithm with the system. */
static Algorithm *create(Controller *controller)
{
	return new Noise(controller);
}
static RegisterAlgorithm reg(NAME, &create);
s="hl opt">.join([ '"' + line.replace('\\', r'\\').replace('"', r'\"') + ' "' for line in description if line ]).rstrip() # Custom filter to allow indenting by a string prior to Jinja version 3.0 # # This function can be removed and the calls to indent_str() replaced by the # built-in indent() filter when dropping Jinja versions older than 3.0 def indent_str(s, indention): s += '\n' lines = s.splitlines() rv = lines.pop(0) if lines: rv += '\n' + '\n'.join( indention + line if line else line for line in lines ) return rv def snake_case(s): return ''.join([ c.isupper() and ('_' + c.lower()) or c for c in s ]).strip('_') def kebab_case(s): return snake_case(s).replace('_', '-') def extend_control(ctrl): if ctrl.vendor != 'libcamera': ctrl.namespace = f'{ctrl.vendor}::' ctrl.vendor_prefix = f'{ctrl.vendor}-' else: ctrl.namespace = '' ctrl.vendor_prefix = '' ctrl.is_array = ctrl.size is not None if ctrl.is_enum: # Remove common prefix from enum variant names prefix = find_common_prefix([enum.name for enum in ctrl.enum_values]) for enum in ctrl.enum_values: enum.gst_name = kebab_case(enum.name.removeprefix(prefix)) ctrl.gtype = 'enum' ctrl.default = '0' elif ctrl.element_type == 'bool': ctrl.gtype = 'boolean' ctrl.default = 'false' elif ctrl.element_type == 'float': ctrl.gtype = 'float' ctrl.default = '0' ctrl.min = '-G_MAXFLOAT' ctrl.max = 'G_MAXFLOAT' elif ctrl.element_type == 'int32_t': ctrl.gtype = 'int' ctrl.default = '0' ctrl.min = 'G_MININT' ctrl.max = 'G_MAXINT' elif ctrl.element_type == 'int64_t': ctrl.gtype = 'int64' ctrl.default = '0' ctrl.min = 'G_MININT64' ctrl.max = 'G_MAXINT64' elif ctrl.element_type == 'uint8_t': ctrl.gtype = 'uchar' ctrl.default = '0' ctrl.min = '0' ctrl.max = 'G_MAXUINT8' elif ctrl.element_type == 'Rectangle': ctrl.is_rectangle = True ctrl.default = '0' ctrl.min = '0' ctrl.max = 'G_MAXINT' else: raise RuntimeError(f'The type `{ctrl.element_type}` is unknown') return ctrl def main(argv): # Parse command line arguments parser = argparse.ArgumentParser() parser.add_argument('--output', '-o', metavar='file', type=str, help='Output file name. Defaults to standard output if not specified.') parser.add_argument('--template', '-t', dest='template', type=str, required=True, help='Template file name.') parser.add_argument('input', type=str, nargs='+', help='Input file name.') args = parser.parse_args(argv[1:]) controls = {} for input in args.input: data = yaml.safe_load(open(input, 'rb').read()) vendor = data['vendor'] ctrls = controls.setdefault(vendor, []) for ctrl in data['controls']: ctrl = Control(*ctrl.popitem(), vendor, mode='controls') if ctrl.name in exposed_controls: ctrls.append(extend_control(ctrl)) data = {'controls': list(controls.items())} env = jinja2.Environment() env.filters['format_description'] = format_description env.filters['indent_str'] = indent_str env.filters['snake_case'] = snake_case env.filters['kebab_case'] = kebab_case template = env.from_string(open(args.template, 'r', encoding='utf-8').read()) string = template.render(data) if args.output: with open(args.output, 'w', encoding='utf-8') as output: output.write(string) else: sys.stdout.write(string) return 0 if __name__ == '__main__': sys.exit(main(sys.argv))