summaryrefslogtreecommitdiff
path: root/src/ipa/raspberrypi/controller/rpi/focus.cpp
blob: 8c5029bd0e951e859d481e929cf2ffbecdfda995 (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
/* SPDX-License-Identifier: BSD-2-Clause */
/*
 * Copyright (C) 2020, Raspberry Pi Ltd
 *
 * focus.cpp - focus algorithm
 */
#include <stdint.h>

#include <libcamera/base/log.h>

#include "../focus_status.h"
#include "focus.h"

using namespace RPiController;
using namespace libcamera;

LOG_DEFINE_CATEGORY(RPiFocus)

#define NAME "rpi.focus"

Focus::Focus(Controller *controller)
	: Algorithm(controller)
{
}

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

void Focus::process(StatisticsPtr &stats, Metadata *imageMetadata)
{
	FocusStatus status;
	unsigned int i;
	for (i = 0; i < FOCUS_REGIONS; i++)
		status.focusMeasures[i] = stats->focus_stats[i].contrast_val[1][1] / 1000;
	status.num = i;
	imageMetadata->set("focus.status", status);

	LOG(RPiFocus, Debug)
		<< "Focus contrast measure: "
		<< (status.focusMeasures[5] + status.focusMeasures[6]) / 10;
}

/* Register algorithm with the system. */
static Algorithm *create(Controller *controller)
{
	return new Focus(controller);
}
static RegisterAlgorithm reg(NAME, &create);
"skipnewline": True, "need_indent": False, "need_space": False, } self.fout = fout def newline(self): if not self.state["skipnewline"]: self.fout.write('\n') self.state["need_indent"] = True self.state["need_space"] = False self.state["skipnewline"] = True def write(self, c): if self.state["need_indent"]: self.fout.write(' ' * self.state["indent"] * 4) self.state["need_indent"] = False if self.state["need_space"]: self.fout.write(' ') self.state["need_space"] = False self.fout.write(c) self.state["skipnewline"] = False def process_char(self, c): if c == '{': self.newline() self.write(c) self.state["indent"] += 1 self.newline() elif c == '}': self.state["indent"] -= 1 self.newline() self.write(c) elif c == '[': self.newline() self.write(c) self.state["indent"] += 1 self.newline() self.state["inarray"] = [True] + self.state["inarray"] self.state["arraycount"] = [0] + self.state["arraycount"] elif c == ']': self.state["indent"] -= 1 self.newline() self.state["inarray"].pop(0) self.state["arraycount"].pop(0) self.write(c) elif c == ':': self.write(c) self.state["need_space"] = True elif c == ',': if not self.state["inarray"][0]: self.write(c) self.newline() else: self.write(c) self.state["arraycount"][0] += 1 if self.state["arraycount"][0] == 16: self.state["arraycount"][0] = 0 self.newline() else: self.state["need_space"] = True elif c.isspace(): pass else: self.write(c) def print(self, string): for c in string: self.process_char(c) self.newline() def pretty_print_json(str_in, output_filename): with open(output_filename, "w") as fout: printer = JSONPrettyPrinter(fout) printer.print(str_in) if __name__ == '__main__': if len(sys.argv) != 2: print("Usage: %s filename" % sys.argv[0]) sys.exit(1) input_filename = sys.argv[1] with open(input_filename, "r") as fin: printer = JSONPrettyPrinter(sys.stdout) printer.print(fin.read())