summaryrefslogtreecommitdiff
path: root/src/ipa/raspberrypi/controller/algorithm.hpp
blob: 5123c87bab344ee40c900cc012a655ed62f8c62e (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
/* SPDX-License-Identifier: BSD-2-Clause */
/*
 * Copyright (C) 2019, Raspberry Pi (Trading) Limited
 *
 * algorithm.hpp - ISP control algorithm interface
 */
#pragma once

// All algorithms should be derived from this class and made available to the
// Controller.

#include <string>
#include <memory>
#include <map>

#include "controller.hpp"

#include <boost/property_tree/ptree.hpp>

namespace RPiController {

// This defines the basic interface for all control algorithms.

class Algorithm
{
public:
	Algorithm(Controller *controller)
		: controller_(controller), paused_(false)
	{
	}
	virtual ~Algorithm() = default;
	virtual char const *Name() const = 0;
	virtual bool IsPaused() const { return paused_; }
	virtual void Pause() { paused_ = true; }
	virtual void Resume() { paused_ = false; }
	virtual void Read(boost::property_tree::ptree const &params);
	virtual void Initialise();
	virtual void SwitchMode(CameraMode const &camera_mode, Metadata *metadata);
	virtual void Prepare(Metadata *image_metadata);
	virtual void Process(StatisticsPtr &stats, Metadata *image_metadata);
	Metadata &GetGlobalMetadata() const
	{
		return controller_->GetGlobalMetadata();
	}

private:
	Controller *controller_;
	bool paused_;
};

// This code is for automatic registration of Front End algorithms with the
// system.

typedef Algorithm *(*AlgoCreateFunc)(Controller *controller);
struct RegisterAlgorithm {
	RegisterAlgorithm(char const *name, AlgoCreateFunc create_func);
};
std::map<std::string, AlgoCreateFunc> const &GetAlgorithms();

} // namespace RPiController
opt">(buffer->metadata().status != FrameMetadata::FrameSuccess) return; completeBuffersCount_++; } void requestComplete(Request *request) { if (request->status() != Request::RequestComplete) return; const Request::BufferMap &buffers = request->buffers(); completeRequestsCount_++; /* Create a new request. */ const Stream *stream = buffers.begin()->first; FrameBuffer *buffer = buffers.begin()->second; request->reuse(); request->addBuffer(stream, buffer); camera_->queueRequest(request); } int init() override { if (status_ != TestPass) return status_; config_ = camera_->generateConfiguration({ StreamRole::VideoRecording }); if (!config_ || config_->size() != 1) { std::cout << "Failed to generate default configuration" << std::endl; return TestFail; } return TestPass; } int run() override { StreamConfiguration &cfg = config_->at(0); if (camera_->acquire()) { std::cout << "Failed to acquire the camera" << std::endl; return TestFail; } if (camera_->configure(config_.get())) { std::cout << "Failed to set default configuration" << std::endl; return TestFail; } Stream *stream = cfg.stream(); BufferSource source; int ret = source.allocate(cfg); if (ret != TestPass) return ret; for (const std::unique_ptr<FrameBuffer> &buffer : source.buffers()) { std::unique_ptr<Request> request = camera_->createRequest(); if (!request) { std::cout << "Failed to create request" << std::endl; return TestFail; } if (request->addBuffer(stream, buffer.get())) { std::cout << "Failed to associating buffer with request" << std::endl; return TestFail; } requests_.push_back(std::move(request)); } completeRequestsCount_ = 0; completeBuffersCount_ = 0; camera_->bufferCompleted.connect(this, &BufferImportTest::bufferComplete); camera_->requestCompleted.connect(this, &BufferImportTest::requestComplete); if (camera_->start()) { std::cout << "Failed to start camera" << std::endl; return TestFail; } for (std::unique_ptr<Request> &request : requests_) { if (camera_->queueRequest(request.get())) { std::cout << "Failed to queue request" << std::endl; return TestFail; } } EventDispatcher *dispatcher = Thread::current()->eventDispatcher(); Timer timer; timer.start(1000ms); while (timer.isRunning()) dispatcher->processEvents(); if (completeRequestsCount_ < cfg.bufferCount * 2) { std::cout << "Failed to capture enough frames (got " << completeRequestsCount_ << " expected at least " << cfg.bufferCount * 2 << ")" << std::endl; return TestFail; } if (completeRequestsCount_ != completeBuffersCount_) { std::cout << "Number of completed buffers and requests differ" << std::endl; return TestFail; } if (camera_->stop()) { std::cout << "Failed to stop camera" << std::endl; return TestFail; } return TestPass; } private: std::vector<std::unique_ptr<Request>> requests_; unsigned int completeBuffersCount_; unsigned int completeRequestsCount_; std::unique_ptr<CameraConfiguration> config_; }; } /* namespace */ TEST_REGISTER(BufferImportTest)