test/color-space.cpp


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

#include <libcamera/camera_manager.h>

class CameraTest
{
public:
	CameraTest(const char *name, bool isolate = false);
	~CameraTest();

protected:
	libcamera::CameraManager *cm_;
	std::shared_ptr<libcamera::Camera> camera_;
	int status_;
};
/* SPDX-License-Identifier: GPL-2.0-or-later */
/*
 * Copyright (C) 2022, Laurent Pinchart <laurent.pinchart@ideasonboard.com>
 *
 * libcamera ColorSpace test
 */

#include <array>
#include <iostream>

#include <libcamera/color_space.h>

#include "test.h"

using namespace libcamera;
using namespace std;

class ColorSpaceTest : public Test
{
protected:
	int run()
	{
		if (ColorSpace::toString(std::nullopt) != "Unset") {
			std::cerr << "Conversion from nullopt to string failed" << std::endl;
			return TestFail;
		}

		const std::array<std::pair<ColorSpace, std::string>, 10> colorSpaces = { {
			{ ColorSpace::Raw, "RAW" },
			{ ColorSpace::Srgb, "sRGB" },
			{ ColorSpace::Sycc, "sYCC" },
			{ ColorSpace::Smpte170m, "SMPTE170M" },
			{ ColorSpace::Rec709, "Rec709" },
			{ ColorSpace::Rec2020, "Rec2020" },
			{
				ColorSpace{
					ColorSpace::Primaries::Raw,
					ColorSpace::TransferFunction::Linear,
					ColorSpace::YcbcrEncoding::None,
					ColorSpace::Range::Limited
				},
				"RAW/Linear/None/Limited"
			}, {
				ColorSpace{
					ColorSpace::Primaries::Smpte170m,
					ColorSpace::TransferFunction::Srgb,
					ColorSpace::YcbcrEncoding::Rec601,
					ColorSpace::Range::Full
				},
				"SMPTE170M/sRGB/Rec601/Full"
			}, {
				ColorSpace{
					ColorSpace::Primaries::Rec709,
					ColorSpace::TransferFunction::Rec709,
					ColorSpace::YcbcrEncoding::Rec709,
					ColorSpace::Range::Full
				},
				"Rec709/Rec709/Rec709/Full"
			}, {
				ColorSpace{
					ColorSpace::Primaries::Rec2020,
					ColorSpace::TransferFunction::Linear,
					ColorSpace::YcbcrEncoding::Rec2020,
					ColorSpace::Range::Limited
				},
				"Rec2020/Linear/Rec2020/Limited"
			},
		} };

		for (const auto &[colorSpace, name] : colorSpaces) {
			if (colorSpace.toString() != name) {
				std::cerr
					<< "Conversion from ColorSpace to string failed: "
					<< "expected " << name
					<< ", got " << colorSpace.toString()
					<< std::endl;
				return TestFail;
			}

			if (ColorSpace::fromString(name) != colorSpace) {
				std::cerr
					<< "Conversion from string "
					<< name << " to ColorSpace failed"
					<< std::endl;
				return TestFail;
			}
		}

		if (ColorSpace::fromString("Invalid")) {
			std::cerr << "Conversion from invalid name string to color space succeeded"
				  << std::endl;
			return TestFail;
		}

		if (ColorSpace::fromString("Rec709/Rec709/Rec710/Limited")) {
			std::cerr << "Conversion from invalid component string to color space succeeded"
				  << std::endl;
			return TestFail;
		}

		return TestPass;
	}
};

TEST_REGISTER(ColorSpaceTest)