From 4c82053d4a1aa0ae5437349663ccfc5c66d387ab Mon Sep 17 00:00:00 2001
From: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
Date: Sat, 14 Mar 2020 16:14:27 +0200
Subject: libcamera: v4l2_videodevice: Refactor allocateBuffers()
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

Move the buffer creation code out of allocateBuffers() to a
createBuffers() function. This prepare for the rework of buffer export
and will avoid code duplication.

Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com>
Reviewed-by: Niklas Söderlund <niklas.soderlund@ragnatech.se>
---
 src/libcamera/v4l2_videodevice.cpp | 68 ++++++++++++++++++++------------------
 1 file changed, 36 insertions(+), 32 deletions(-)

(limited to 'src/libcamera/v4l2_videodevice.cpp')

diff --git a/src/libcamera/v4l2_videodevice.cpp b/src/libcamera/v4l2_videodevice.cpp
index 100a92b5..e5911b49 100644
--- a/src/libcamera/v4l2_videodevice.cpp
+++ b/src/libcamera/v4l2_videodevice.cpp
@@ -1036,61 +1036,65 @@ int V4L2VideoDevice::requestBuffers(unsigned int count,
  */
 int V4L2VideoDevice::allocateBuffers(unsigned int count,
 				     std::vector<std::unique_ptr<FrameBuffer>> *buffers)
+{
+	int ret = createBuffers(count, buffers);
+	if (ret < 0)
+		return ret;
+
+	cache_ = new V4L2BufferCache(*buffers);
+	memoryType_ = V4L2_MEMORY_MMAP;
+
+	return ret;
+}
+
+int V4L2VideoDevice::createBuffers(unsigned int count,
+				   std::vector<std::unique_ptr<FrameBuffer>> *buffers)
 {
 	if (cache_) {
 		LOG(V4L2, Error) << "Buffers already allocated";
 		return -EINVAL;
 	}
 
-	memoryType_ = V4L2_MEMORY_MMAP;
-
 	int ret = requestBuffers(count, V4L2_MEMORY_MMAP);
 	if (ret < 0)
 		return ret;
 
 	for (unsigned i = 0; i < count; ++i) {
-		struct v4l2_buffer buf = {};
-		struct v4l2_plane planes[VIDEO_MAX_PLANES] = {};
-
-		buf.index = i;
-		buf.type = bufferType_;
-		buf.memory = memoryType_;
-		buf.length = ARRAY_SIZE(planes);
-		buf.m.planes = planes;
-
-		ret = ioctl(VIDIOC_QUERYBUF, &buf);
-		if (ret < 0) {
-			LOG(V4L2, Error)
-				<< "Unable to query buffer " << i << ": "
-				<< strerror(-ret);
-			goto err_buf;
-		}
-
-		std::unique_ptr<FrameBuffer> buffer = createBuffer(buf);
+		std::unique_ptr<FrameBuffer> buffer = createBuffer(i);
 		if (!buffer) {
 			LOG(V4L2, Error) << "Unable to create buffer";
-			ret = -EINVAL;
-			goto err_buf;
+
+			requestBuffers(0, V4L2_MEMORY_MMAP);
+			buffers->clear();
+
+			return -EINVAL;
 		}
 
 		buffers->push_back(std::move(buffer));
 	}
 
-	cache_ = new V4L2BufferCache(*buffers);
-
 	return count;
+}
 
-err_buf:
-	requestBuffers(0, V4L2_MEMORY_MMAP);
+std::unique_ptr<FrameBuffer> V4L2VideoDevice::createBuffer(unsigned int index)
+{
+	struct v4l2_plane v4l2Planes[VIDEO_MAX_PLANES] = {};
+	struct v4l2_buffer buf = {};
 
-	buffers->clear();
+	buf.index = index;
+	buf.type = bufferType_;
+	buf.memory = V4L2_MEMORY_MMAP;
+	buf.length = ARRAY_SIZE(v4l2Planes);
+	buf.m.planes = v4l2Planes;
 
-	return ret;
-}
+	int ret = ioctl(VIDIOC_QUERYBUF, &buf);
+	if (ret < 0) {
+		LOG(V4L2, Error)
+			<< "Unable to query buffer " << index << ": "
+			<< strerror(-ret);
+		return nullptr;
+	}
 
-std::unique_ptr<FrameBuffer>
-V4L2VideoDevice::createBuffer(const struct v4l2_buffer &buf)
-{
 	const bool multiPlanar = V4L2_TYPE_IS_MULTIPLANAR(buf.type);
 	const unsigned int numPlanes = multiPlanar ? buf.length : 1;
 
-- 
cgit v1.2.1