summaryrefslogtreecommitdiff
path: root/src/lc-compliance/simple_capture.cpp
blob: 64e862a08e3a8221f24677a5a14f9cbea44aa936 (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
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright (C) 2020, Google Inc.
 *
 * simple_capture.cpp - Simple capture helper
 */

#include "simple_capture.h"

using namespace libcamera;

SimpleCapture::SimpleCapture(std::shared_ptr<Camera> camera)
	: loop_(nullptr), camera_(camera),
	  allocator_(std::make_unique<FrameBufferAllocator>(camera))
{
}

SimpleCapture::~SimpleCapture()
{
}

Results::Result SimpleCapture::configure(StreamRole role)
{
	config_ = camera_->generateConfiguration({ role });

	if (!config_)
		return { Results::Skip, "Role not supported by camera" };

	if (config_->validate() != CameraConfiguration::Valid) {
		config_.reset();
		return { Results::Fail, "Configuration not valid" };
	}

	if (camera_->configure(config_.get())) {
		config_.reset();
		return { Results::Fail, "Failed to configure camera" };
	}

	return { Results::Pass, "Configure camera" };
}

Results::Result SimpleCapture::start()
{
	Stream *stream = config_->at(0).stream();
	if (allocator_->allocate(stream) < 0)
		return { Results::Fail, "Failed to allocate buffers" };

	if (camera_->start())
		return { Results::Fail, "Failed to start camera" };

	camera_->requestCompleted.connect(this, &SimpleCapture::requestComplete);

	return { Results::Pass, "Started camera" };
}

void SimpleCapture::stop()
{
	Stream *stream = config_->at(0).stream();

	camera_->stop();

	camera_->requestCompleted.disconnect(this, &SimpleCapture::requestComplete);

	allocator_->free(stream);
}

/* SimpleCaptureBalanced */

SimpleCaptureBalanced::SimpleCaptureBalanced(std::shared_ptr<Camera> camera)
	: SimpleCapture(camera)
{
}

Results::Result SimpleCaptureBalanced::capture(unsigned int numRequests)
{
	Results::Result ret = start();
	if (ret.first != Results::Pass)
		return ret;

	Stream *stream = config_->at(0).stream();
	const std::vector<std::unique_ptr<FrameBuffer>> &buffers = allocator_->buffers(stream);

	/* No point in testing less requests then the camera depth. */
	if (buffers.size() > numRequests) {
		/* Cache buffers.size() before we destroy it in stop() */
		int buffers_size = buffers.size();
		stop();

		return { Results::Skip, "Camera needs " + std::to_string(buffers_size)
			+ " requests, can't test only " + std::to_string(numRequests) };
	}

	queueCount_ = 0;
	captureCount_ = 0;
	captureLimit_ = numRequests;

	/* Queue the recommended number of reqeuests. */
	std::vector<std::unique_ptr<libcamera::Request>> requests;
	for (const std::unique_ptr<FrameBuffer> &buffer : buffers) {
		std::unique_ptr<Request> request = camera_->createRequest();
		if (!request) {
			stop();
			return { Results::Fail, "Can't create request" };
		}

		if (request->addBuffer(stream, buffer.get())) {
			stop();
			return { Results::Fail, "Can't set buffer for request" };
		}

		if (queueRequest(request.get()) < 0) {
			stop();
			return { Results::Fail, "Failed to queue request" };
		}

		requests.push_back(std::move(request));
	}

	/* Run capture session. */
	loop_ = new EventLoop();
	loop_->exec();
	stop();
	delete loop_;

	if (captureCount_ != captureLimit_)
		return { Results::Fail, "Got " + std::to_string(captureCount_) +
			" request, wanted " + std::to_string(captureLimit_) };

	return { Results::Pass, "Balanced capture of " +
		std::to_string(numRequests) + " requests" };
}

int SimpleCaptureBalanced::queueRequest(Request *request)
{
	queueCount_++;
	if (queueCount_ > captureLimit_)
		return 0;

	return camera_->queueRequest(request);
}

void SimpleCaptureBalanced::requestComplete(Request *request)
{
	captureCount_++;
	if (captureCount_ >= captureLimit_) {
		loop_->exit(0);
		return;
	}

	request->reuse(Request::ReuseBuffers);
	if (queueRequest(request))
		loop_->exit(-EINVAL);
}

/* SimpleCaptureUnbalanced */

SimpleCaptureUnbalanced::SimpleCaptureUnbalanced(std::shared_ptr<Camera> camera)
	: SimpleCapture(camera)
{
}

Results::Result SimpleCaptureUnbalanced::capture(unsigned int numRequests)
{
	Results::Result ret = start();
	if (ret.first != Results::Pass)
		return ret;

	Stream *stream = config_->at(0).stream();
	const std::vector<std::unique_ptr<FrameBuffer>> &buffers = allocator_->buffers(stream);

	captureCount_ = 0;
	captureLimit_ = numRequests;

	/* Queue the recommended number of reqeuests. */
	std::vector<std::unique_ptr<libcamera::Request>> requests;
	for (const std::unique_ptr<FrameBuffer> &buffer : buffers) {
		std::unique_ptr<Request> request = camera_->createRequest();
		if (!request) {
			stop();
			return { Results::Fail, "Can't create request" };
		}

		if (request->addBuffer(stream, buffer.get())) {
			stop();
			return { Results::Fail, "Can't set buffer for request" };
		}

		if (camera_->queueRequest(request.get()) < 0) {
			stop();
			return { Results::Fail, "Failed to queue request" };
		}

		requests.push_back(std::move(request));
	}

	/* Run capture session. */
	loop_ = new EventLoop();
	int status = loop_->exec();
	stop();
	delete loop_;

	return { status ? Results::Fail : Results::Pass, "Unbalanced capture of " + std::to_string(numRequests) + " requests" };
}

void SimpleCaptureUnbalanced::requestComplete(Request *request)
{
	captureCount_++;
	if (captureCount_ >= captureLimit_) {
		loop_->exit(0);
		return;
	}

	request->reuse(Request::ReuseBuffers);
	if (camera_->queueRequest(request))
		loop_->exit(-EINVAL);
}
l opt">< 0) return -errno; return ret; } /** * \brief Read data from the file * \param[in] data Memory to read data into * * Read at most \a data.size() bytes from the file into \a data.data(), and * return the number of bytes read. If less data than requested is available, * the returned byte count may be smaller than the requested size. If no more * data is available, 0 is returned. * * The position of the file as returned by pos() is advanced by the number of * bytes read. If an error occurs, the position isn't modified. * * \return The number of bytes read on success, or a negative error code * otherwise */ ssize_t File::read(const Span<uint8_t> &data) { if (!isOpen()) return -EINVAL; size_t readBytes = 0; ssize_t ret = 0; /* Retry in case of interrupted system calls. */ while (readBytes < data.size()) { ret = ::read(fd_, data.data() + readBytes, data.size() - readBytes); if (ret <= 0) break; readBytes += ret; } if (ret < 0 && !readBytes) return -errno; return readBytes; } /** * \brief Write data to the file * \param[in] data Memory containing data to be written * * Write at most \a data.size() bytes from \a data.data() to the file, and * return the number of bytes written. If the file system doesn't have enough * space for the data, the returned byte count may be less than requested. * * The position of the file as returned by pos() is advanced by the number of * bytes written. If an error occurs, the position isn't modified. * * \return The number of bytes written on success, or a negative error code * otherwise */ ssize_t File::write(const Span<const uint8_t> &data) { if (!isOpen()) return -EINVAL; size_t writtenBytes = 0; /* Retry in case of interrupted system calls. */ while (writtenBytes < data.size()) { ssize_t ret = ::write(fd_, data.data() + writtenBytes, data.size() - writtenBytes); if (ret <= 0) break; writtenBytes += ret; } if (data.size() && !writtenBytes) return -errno; return writtenBytes; } /** * \brief Map a region of the file in the process memory * \param[in] offset The region offset within the file * \param[in] size The region sise * \param[in] flags The mapping flags * * This function maps a region of \a size bytes of the file starting at \a * offset into the process memory. The File instance shall be open, but may be * closed after mapping the region. Mappings stay valid when the File is * closed, and are destroyed automatically when the File is deleted. * * If \a size is a negative value, this function maps the region starting at \a * offset until the end of the file. * * The mapping memory protection is controlled by the file open mode, unless \a * flags contains MapPrivate in which case the region is mapped in read/write * mode. * * The error() status is updated. * * \return The mapped memory on success, or an empty span otherwise */ Span<uint8_t> File::map(off_t offset, ssize_t size, enum File::MapFlag flags) { if (!isOpen()) { error_ = -EBADF; return {}; } if (size < 0) { size = File::size(); if (size < 0) { error_ = size; return {}; } size -= offset; } int mmapFlags = flags & MapPrivate ? MAP_PRIVATE : MAP_SHARED; int prot = 0; if (mode_ & ReadOnly) prot |= PROT_READ; if (mode_ & WriteOnly) prot |= PROT_WRITE; if (flags & MapPrivate) prot |= PROT_WRITE; void *map = mmap(NULL, size, prot, mmapFlags, fd_, offset); if (map == MAP_FAILED) { error_ = -errno; return {}; } maps_.emplace(map, size); error_ = 0; return { static_cast<uint8_t *>(map), static_cast<size_t>(size) }; } /** * \brief Unmap a region mapped with map() * \param[in] addr The region address * * The error() status is updated. * * \return True on success, or false if an error occurs */ bool File::unmap(uint8_t *addr) { auto iter = maps_.find(static_cast<void *>(addr)); if (iter == maps_.end()) { error_ = -ENOENT; return false; } int ret = munmap(addr, iter->second); if (ret < 0) { error_ = -errno; return false; } maps_.erase(iter); error_ = 0; return true; } void File::unmapAll() { for (const auto &map : maps_) munmap(map.first, map.second); maps_.clear(); } /** * \brief Check if the file specified by \a name exists * \param[in] name The file name * \return True if the file exists, false otherwise */ bool File::exists(const std::string &name) { struct stat st; int ret = stat(name.c_str(), &st); if (ret < 0) return false; /* Directories can not be handled here, even if they exist. */ return !S_ISDIR(st.st_mode); } } /* namespace libcamera */