1 files changed, 62 insertions, 23 deletions

diff --git a/src/apps/lc-compliance/helpers/capture.cpp b/src/apps/lc-compliance/helpers/capture.cpp
index 5aab973f..bb95af3d 100644
--- a/src/apps/lc-compliance/helpers/capture.cpp
+++ b/src/apps/lc-compliance/helpers/capture.cpp
@@ -7,6 +7,8 @@
 
 #include "capture.h"
 
+#include <algorithm>
+
 #include <gtest/gtest.h>
 
 using namespace libcamera;
@@ -22,15 +24,27 @@ Capture::~Capture()
 	stop();
 }
 
-void Capture::configure(StreamRole role)
+void Capture::configure(Span<const StreamRole> roles)
 {
-	config_ = camera_->generateConfiguration({ role });
+	config_ = camera_->generateConfiguration(roles);
 
 	if (!config_) {
-		std::cout << "Role not supported by camera" << std::endl;
+		std::cout << "Roles not supported by camera" << std::endl;
 		GTEST_SKIP();
 	}
 
+	/*
+	 * Set the buffers count to the largest value across all streams.
+	 * \todo: Should all streams from a Camera have the same buffer count ?
+	 * */
+	auto largest =
+		std::max_element(config_->begin(), config_->end(),
+				 [](StreamConfiguration &l, StreamConfiguration &r)
+				 { return l.bufferCount < r.bufferCount; });
+
+	for (auto &stream : *config_)
+		stream.bufferCount = largest->bufferCount;
+
 	if (config_->validate() != CameraConfiguration::Valid) {
 		config_.reset();
 		FAIL() << "Configuration not valid";
@@ -44,11 +58,17 @@ void Capture::configure(StreamRole role)
 
 void Capture::start()
 {
-	Stream *stream = config_->at(0).stream();
-	int count = allocator_->allocate(stream);
+	unsigned int i = 0;
+	for (auto const &config : *config_) {
+		Stream *stream = config.stream();
+		int count = allocator_->allocate(stream);
 
-	ASSERT_GE(count, 0) << "Failed to allocate buffers";
-	EXPECT_EQ(count, config_->at(0).bufferCount) << "Allocated less buffers than expected";
+		ASSERT_GE(count, 0) << "Failed to allocate buffers for stream " << i;
+		EXPECT_EQ(count, config.bufferCount)
+			<< "Allocated less buffers than expected for stream " << i;
+
+		++i;
+	}
 
 	camera_->requestCompleted.connect(this, &Capture::requestComplete);
 
@@ -64,9 +84,12 @@ void Capture::stop()
 
 	camera_->requestCompleted.disconnect(this);
 
-	Stream *stream = config_->at(0).stream();
 	requests_.clear();
-	allocator_->free(stream);
+
+	for (auto const &config : *config_) {
+		Stream *stream = config.stream();
+		allocator_->free(stream);
+	}
 }
 
 /* CaptureBalanced */
@@ -80,14 +103,12 @@ void CaptureBalanced::capture(unsigned int numRequests)
 {
 	start();
 
-	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) {
-		std::cout << "Camera needs " + std::to_string(buffers.size())
-			+ " requests, can't test only "
-			+ std::to_string(numRequests) << std::endl;
+	const unsigned int bufferCount = config_->at(0).bufferCount;
+	if (bufferCount > numRequests) {
+		std::cout << "Camera needs " << bufferCount
+			  << " requests, can't test only " << numRequests
+			  << std::endl;
 		GTEST_SKIP();
 	}
 
@@ -96,11 +117,21 @@ void CaptureBalanced::capture(unsigned int numRequests)
 	captureLimit_ = numRequests;
 
 	/* Queue the recommended number of requests. */
-	for (const std::unique_ptr<FrameBuffer> &buffer : buffers) {
+	for (unsigned int bufferIdx = 0; bufferIdx < bufferCount; ++bufferIdx) {
 		std::unique_ptr<Request> request = camera_->createRequest();
 		ASSERT_TRUE(request) << "Can't create request";
 
-		ASSERT_EQ(request->addBuffer(stream, buffer.get()), 0) << "Can't set buffer for request";
+		/* Add buffers for each stream. */
+		unsigned int i = 0;
+		for (const auto &config : *config_) {
+			Stream *stream = config.stream();
+			const auto &buffers = allocator_->buffers(stream);
+
+			ASSERT_EQ(request->addBuffer(stream, buffers[bufferIdx].get()), 0)
+				<< "Can't add buffers for stream " << i;
+
+			++i;
+		}
 
 		ASSERT_EQ(queueRequest(request.get()), 0) << "Failed to queue request";
 
@@ -152,18 +183,26 @@ void CaptureUnbalanced::capture(unsigned int numRequests)
 {
 	start();
 
-	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 requests. */
-	for (const std::unique_ptr<FrameBuffer> &buffer : buffers) {
+	const unsigned int bufferCount = config_->at(0).bufferCount;
+	for (unsigned int bufferIdx = 0; bufferIdx < bufferCount; ++bufferIdx) {
 		std::unique_ptr<Request> request = camera_->createRequest();
 		ASSERT_TRUE(request) << "Can't create request";
 
-		ASSERT_EQ(request->addBuffer(stream, buffer.get()), 0) << "Can't set buffer for request";
+		/* Add buffers for each stream. */
+		unsigned int i = 0;
+		for (const auto &config : *config_) {
+			Stream *stream = config.stream();
+			const auto &buffers = allocator_->buffers(stream);
+
+			ASSERT_EQ(request->addBuffer(stream, buffers[bufferIdx].get()), 0)
+				<< "Can't add buffers for stream " << i;
+
+			++i;
+		}
 
 		ASSERT_EQ(camera_->queueRequest(request.get()), 0) << "Failed to queue request";