summaryrefslogtreecommitdiff
path: root/src/apps/cam/file_sink.h
blob: 71b7fe0feab5396d301ec3a65f8aa2d567f69f9e (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
/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright (C) 2019, Google Inc.
 *
 * File Sink
 */

#pragma once

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

#include <libcamera/stream.h>

#include "frame_sink.h"

class Image;

class FileSink : public FrameSink
{
public:
	FileSink(const libcamera::Camera *camera,
		 const std::map<const libcamera::Stream *, std::string> &streamNames);
	~FileSink();

	int setFilePattern(const std::string &pattern);

	int configure(const libcamera::CameraConfiguration &config) override;

	void mapBuffer(libcamera::FrameBuffer *buffer) override;

	bool processRequest(libcamera::Request *request) override;

private:
	static constexpr const char *kDefaultFilePattern = "frame-#.bin";

	enum class FileType {
		Binary,
		Dng,
		Ppm,
	};

	void writeBuffer(const libcamera::Stream *stream,
			 libcamera::FrameBuffer *buffer,
			 const libcamera::ControlList &metadata);

#ifdef HAVE_TIFF
	const libcamera::Camera *camera_;
#endif

	std::string pattern_;
	FileType fileType_;

	std::map<const libcamera::Stream *, std::string> streamNames_;
	std::map<libcamera::FrameBuffer *, std::unique_ptr<Image>> mappedBuffers_;
};