apps: Share common source between applications
Multiple source files in the src/apps/cam/ directory are used by cam, qcam and lc-compliance. They are compiled separately for each application. Move them to a new src/apps/common/ directory and compile them in a static library to decrease the number of compilation operations. Signed-off-by: Laurent Pinchart <laurent.pinchart@ideasonboard.com> Reviewed-by: Umang Jain <umang.jain@ideasonboard.com> Reviewed-by: Kieran Bingham <kieran.bingham@ideasonboard.com>
This commit is contained in:
@@ -0,0 +1,653 @@
|
||||
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
||||
/*
|
||||
* Copyright (C) 2020, Raspberry Pi Ltd
|
||||
*
|
||||
* dng_writer.cpp - DNG writer
|
||||
*/
|
||||
|
||||
#include "dng_writer.h"
|
||||
|
||||
#include <algorithm>
|
||||
#include <iostream>
|
||||
#include <map>
|
||||
|
||||
#include <tiffio.h>
|
||||
|
||||
#include <libcamera/control_ids.h>
|
||||
#include <libcamera/formats.h>
|
||||
#include <libcamera/property_ids.h>
|
||||
|
||||
using namespace libcamera;
|
||||
|
||||
enum CFAPatternColour : uint8_t {
|
||||
CFAPatternRed = 0,
|
||||
CFAPatternGreen = 1,
|
||||
CFAPatternBlue = 2,
|
||||
};
|
||||
|
||||
struct FormatInfo {
|
||||
uint8_t bitsPerSample;
|
||||
CFAPatternColour pattern[4];
|
||||
void (*packScanline)(void *output, const void *input,
|
||||
unsigned int width);
|
||||
void (*thumbScanline)(const FormatInfo &info, void *output,
|
||||
const void *input, unsigned int width,
|
||||
unsigned int stride);
|
||||
};
|
||||
|
||||
struct Matrix3d {
|
||||
Matrix3d()
|
||||
{
|
||||
}
|
||||
|
||||
Matrix3d(float m0, float m1, float m2,
|
||||
float m3, float m4, float m5,
|
||||
float m6, float m7, float m8)
|
||||
{
|
||||
m[0] = m0, m[1] = m1, m[2] = m2;
|
||||
m[3] = m3, m[4] = m4, m[5] = m5;
|
||||
m[6] = m6, m[7] = m7, m[8] = m8;
|
||||
}
|
||||
|
||||
Matrix3d(const Span<const float> &span)
|
||||
: Matrix3d(span[0], span[1], span[2],
|
||||
span[3], span[4], span[5],
|
||||
span[6], span[7], span[8])
|
||||
{
|
||||
}
|
||||
|
||||
static Matrix3d diag(float diag0, float diag1, float diag2)
|
||||
{
|
||||
return Matrix3d(diag0, 0, 0, 0, diag1, 0, 0, 0, diag2);
|
||||
}
|
||||
|
||||
static Matrix3d identity()
|
||||
{
|
||||
return Matrix3d(1, 0, 0, 0, 1, 0, 0, 0, 1);
|
||||
}
|
||||
|
||||
Matrix3d transpose() const
|
||||
{
|
||||
return { m[0], m[3], m[6], m[1], m[4], m[7], m[2], m[5], m[8] };
|
||||
}
|
||||
|
||||
Matrix3d cofactors() const
|
||||
{
|
||||
return { m[4] * m[8] - m[5] * m[7],
|
||||
-(m[3] * m[8] - m[5] * m[6]),
|
||||
m[3] * m[7] - m[4] * m[6],
|
||||
-(m[1] * m[8] - m[2] * m[7]),
|
||||
m[0] * m[8] - m[2] * m[6],
|
||||
-(m[0] * m[7] - m[1] * m[6]),
|
||||
m[1] * m[5] - m[2] * m[4],
|
||||
-(m[0] * m[5] - m[2] * m[3]),
|
||||
m[0] * m[4] - m[1] * m[3] };
|
||||
}
|
||||
|
||||
Matrix3d adjugate() const
|
||||
{
|
||||
return cofactors().transpose();
|
||||
}
|
||||
|
||||
float determinant() const
|
||||
{
|
||||
return m[0] * (m[4] * m[8] - m[5] * m[7]) -
|
||||
m[1] * (m[3] * m[8] - m[5] * m[6]) +
|
||||
m[2] * (m[3] * m[7] - m[4] * m[6]);
|
||||
}
|
||||
|
||||
Matrix3d inverse() const
|
||||
{
|
||||
return adjugate() * (1.0 / determinant());
|
||||
}
|
||||
|
||||
Matrix3d operator*(const Matrix3d &other) const
|
||||
{
|
||||
Matrix3d result;
|
||||
for (unsigned int i = 0; i < 3; i++) {
|
||||
for (unsigned int j = 0; j < 3; j++) {
|
||||
result.m[i * 3 + j] =
|
||||
m[i * 3 + 0] * other.m[0 + j] +
|
||||
m[i * 3 + 1] * other.m[3 + j] +
|
||||
m[i * 3 + 2] * other.m[6 + j];
|
||||
}
|
||||
}
|
||||
return result;
|
||||
}
|
||||
|
||||
Matrix3d operator*(float f) const
|
||||
{
|
||||
Matrix3d result;
|
||||
for (unsigned int i = 0; i < 9; i++)
|
||||
result.m[i] = m[i] * f;
|
||||
return result;
|
||||
}
|
||||
|
||||
float m[9];
|
||||
};
|
||||
|
||||
void packScanlineSBGGR8(void *output, const void *input, unsigned int width)
|
||||
{
|
||||
const uint8_t *in = static_cast<const uint8_t *>(input);
|
||||
uint8_t *out = static_cast<uint8_t *>(output);
|
||||
|
||||
std::copy(in, in + width, out);
|
||||
}
|
||||
|
||||
void packScanlineSBGGR10P(void *output, const void *input, unsigned int width)
|
||||
{
|
||||
const uint8_t *in = static_cast<const uint8_t *>(input);
|
||||
uint8_t *out = static_cast<uint8_t *>(output);
|
||||
|
||||
/* \todo Can this be made more efficient? */
|
||||
for (unsigned int x = 0; x < width; x += 4) {
|
||||
*out++ = in[0];
|
||||
*out++ = (in[4] & 0x03) << 6 | in[1] >> 2;
|
||||
*out++ = (in[1] & 0x03) << 6 | (in[4] & 0x0c) << 2 | in[2] >> 4;
|
||||
*out++ = (in[2] & 0x0f) << 4 | (in[4] & 0x30) >> 2 | in[3] >> 6;
|
||||
*out++ = (in[3] & 0x3f) << 2 | (in[4] & 0xc0) >> 6;
|
||||
in += 5;
|
||||
}
|
||||
}
|
||||
|
||||
void packScanlineSBGGR12P(void *output, const void *input, unsigned int width)
|
||||
{
|
||||
const uint8_t *in = static_cast<const uint8_t *>(input);
|
||||
uint8_t *out = static_cast<uint8_t *>(output);
|
||||
|
||||
/* \todo Can this be made more efficient? */
|
||||
for (unsigned int i = 0; i < width; i += 2) {
|
||||
*out++ = in[0];
|
||||
*out++ = (in[2] & 0x0f) << 4 | in[1] >> 4;
|
||||
*out++ = (in[1] & 0x0f) << 4 | in[2] >> 4;
|
||||
in += 3;
|
||||
}
|
||||
}
|
||||
|
||||
void thumbScanlineSBGGRxxP(const FormatInfo &info, void *output,
|
||||
const void *input, unsigned int width,
|
||||
unsigned int stride)
|
||||
{
|
||||
const uint8_t *in = static_cast<const uint8_t *>(input);
|
||||
uint8_t *out = static_cast<uint8_t *>(output);
|
||||
|
||||
/* Number of bytes corresponding to 16 pixels. */
|
||||
unsigned int skip = info.bitsPerSample * 16 / 8;
|
||||
|
||||
for (unsigned int x = 0; x < width; x++) {
|
||||
uint8_t value = (in[0] + in[1] + in[stride] + in[stride + 1]) >> 2;
|
||||
*out++ = value;
|
||||
*out++ = value;
|
||||
*out++ = value;
|
||||
in += skip;
|
||||
}
|
||||
}
|
||||
|
||||
void packScanlineIPU3(void *output, const void *input, unsigned int width)
|
||||
{
|
||||
const uint8_t *in = static_cast<const uint8_t *>(input);
|
||||
uint16_t *out = static_cast<uint16_t *>(output);
|
||||
|
||||
/*
|
||||
* Upscale the 10-bit format to 16-bit as it's not trivial to pack it
|
||||
* as 10-bit without gaps.
|
||||
*
|
||||
* \todo Improve packing to keep the 10-bit sample size.
|
||||
*/
|
||||
unsigned int x = 0;
|
||||
while (true) {
|
||||
for (unsigned int i = 0; i < 6; i++) {
|
||||
*out++ = (in[1] & 0x03) << 14 | (in[0] & 0xff) << 6;
|
||||
if (++x >= width)
|
||||
return;
|
||||
|
||||
*out++ = (in[2] & 0x0f) << 12 | (in[1] & 0xfc) << 4;
|
||||
if (++x >= width)
|
||||
return;
|
||||
|
||||
*out++ = (in[3] & 0x3f) << 10 | (in[2] & 0xf0) << 2;
|
||||
if (++x >= width)
|
||||
return;
|
||||
|
||||
*out++ = (in[4] & 0xff) << 8 | (in[3] & 0xc0) << 0;
|
||||
if (++x >= width)
|
||||
return;
|
||||
|
||||
in += 5;
|
||||
}
|
||||
|
||||
*out++ = (in[1] & 0x03) << 14 | (in[0] & 0xff) << 6;
|
||||
if (++x >= width)
|
||||
return;
|
||||
|
||||
in += 2;
|
||||
}
|
||||
}
|
||||
|
||||
void thumbScanlineIPU3([[maybe_unused]] const FormatInfo &info, void *output,
|
||||
const void *input, unsigned int width,
|
||||
unsigned int stride)
|
||||
{
|
||||
uint8_t *out = static_cast<uint8_t *>(output);
|
||||
|
||||
for (unsigned int x = 0; x < width; x++) {
|
||||
unsigned int pixel = x * 16;
|
||||
unsigned int block = pixel / 25;
|
||||
unsigned int pixelInBlock = pixel - block * 25;
|
||||
|
||||
/*
|
||||
* If the pixel is the last in the block cheat a little and
|
||||
* move one pixel backward to avoid reading between two blocks
|
||||
* and having to deal with the padding bits.
|
||||
*/
|
||||
if (pixelInBlock == 24)
|
||||
pixelInBlock--;
|
||||
|
||||
const uint8_t *in = static_cast<const uint8_t *>(input)
|
||||
+ block * 32 + (pixelInBlock / 4) * 5;
|
||||
|
||||
uint16_t val1, val2, val3, val4;
|
||||
switch (pixelInBlock % 4) {
|
||||
case 0:
|
||||
val1 = (in[1] & 0x03) << 14 | (in[0] & 0xff) << 6;
|
||||
val2 = (in[2] & 0x0f) << 12 | (in[1] & 0xfc) << 4;
|
||||
val3 = (in[stride + 1] & 0x03) << 14 | (in[stride + 0] & 0xff) << 6;
|
||||
val4 = (in[stride + 2] & 0x0f) << 12 | (in[stride + 1] & 0xfc) << 4;
|
||||
break;
|
||||
case 1:
|
||||
val1 = (in[2] & 0x0f) << 12 | (in[1] & 0xfc) << 4;
|
||||
val2 = (in[3] & 0x3f) << 10 | (in[2] & 0xf0) << 2;
|
||||
val3 = (in[stride + 2] & 0x0f) << 12 | (in[stride + 1] & 0xfc) << 4;
|
||||
val4 = (in[stride + 3] & 0x3f) << 10 | (in[stride + 2] & 0xf0) << 2;
|
||||
break;
|
||||
case 2:
|
||||
val1 = (in[3] & 0x3f) << 10 | (in[2] & 0xf0) << 2;
|
||||
val2 = (in[4] & 0xff) << 8 | (in[3] & 0xc0) << 0;
|
||||
val3 = (in[stride + 3] & 0x3f) << 10 | (in[stride + 2] & 0xf0) << 2;
|
||||
val4 = (in[stride + 4] & 0xff) << 8 | (in[stride + 3] & 0xc0) << 0;
|
||||
break;
|
||||
case 3:
|
||||
val1 = (in[4] & 0xff) << 8 | (in[3] & 0xc0) << 0;
|
||||
val2 = (in[6] & 0x03) << 14 | (in[5] & 0xff) << 6;
|
||||
val3 = (in[stride + 4] & 0xff) << 8 | (in[stride + 3] & 0xc0) << 0;
|
||||
val4 = (in[stride + 6] & 0x03) << 14 | (in[stride + 5] & 0xff) << 6;
|
||||
break;
|
||||
}
|
||||
|
||||
uint8_t value = (val1 + val2 + val3 + val4) >> 10;
|
||||
*out++ = value;
|
||||
*out++ = value;
|
||||
*out++ = value;
|
||||
}
|
||||
}
|
||||
|
||||
static const std::map<PixelFormat, FormatInfo> formatInfo = {
|
||||
{ formats::SBGGR8, {
|
||||
.bitsPerSample = 8,
|
||||
.pattern = { CFAPatternBlue, CFAPatternGreen, CFAPatternGreen, CFAPatternRed },
|
||||
.packScanline = packScanlineSBGGR8,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SGBRG8, {
|
||||
.bitsPerSample = 8,
|
||||
.pattern = { CFAPatternGreen, CFAPatternBlue, CFAPatternRed, CFAPatternGreen },
|
||||
.packScanline = packScanlineSBGGR8,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SGRBG8, {
|
||||
.bitsPerSample = 8,
|
||||
.pattern = { CFAPatternGreen, CFAPatternRed, CFAPatternBlue, CFAPatternGreen },
|
||||
.packScanline = packScanlineSBGGR8,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SRGGB8, {
|
||||
.bitsPerSample = 8,
|
||||
.pattern = { CFAPatternRed, CFAPatternGreen, CFAPatternGreen, CFAPatternBlue },
|
||||
.packScanline = packScanlineSBGGR8,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SBGGR10_CSI2P, {
|
||||
.bitsPerSample = 10,
|
||||
.pattern = { CFAPatternBlue, CFAPatternGreen, CFAPatternGreen, CFAPatternRed },
|
||||
.packScanline = packScanlineSBGGR10P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SGBRG10_CSI2P, {
|
||||
.bitsPerSample = 10,
|
||||
.pattern = { CFAPatternGreen, CFAPatternBlue, CFAPatternRed, CFAPatternGreen },
|
||||
.packScanline = packScanlineSBGGR10P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SGRBG10_CSI2P, {
|
||||
.bitsPerSample = 10,
|
||||
.pattern = { CFAPatternGreen, CFAPatternRed, CFAPatternBlue, CFAPatternGreen },
|
||||
.packScanline = packScanlineSBGGR10P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SRGGB10_CSI2P, {
|
||||
.bitsPerSample = 10,
|
||||
.pattern = { CFAPatternRed, CFAPatternGreen, CFAPatternGreen, CFAPatternBlue },
|
||||
.packScanline = packScanlineSBGGR10P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SBGGR12_CSI2P, {
|
||||
.bitsPerSample = 12,
|
||||
.pattern = { CFAPatternBlue, CFAPatternGreen, CFAPatternGreen, CFAPatternRed },
|
||||
.packScanline = packScanlineSBGGR12P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SGBRG12_CSI2P, {
|
||||
.bitsPerSample = 12,
|
||||
.pattern = { CFAPatternGreen, CFAPatternBlue, CFAPatternRed, CFAPatternGreen },
|
||||
.packScanline = packScanlineSBGGR12P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SGRBG12_CSI2P, {
|
||||
.bitsPerSample = 12,
|
||||
.pattern = { CFAPatternGreen, CFAPatternRed, CFAPatternBlue, CFAPatternGreen },
|
||||
.packScanline = packScanlineSBGGR12P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SRGGB12_CSI2P, {
|
||||
.bitsPerSample = 12,
|
||||
.pattern = { CFAPatternRed, CFAPatternGreen, CFAPatternGreen, CFAPatternBlue },
|
||||
.packScanline = packScanlineSBGGR12P,
|
||||
.thumbScanline = thumbScanlineSBGGRxxP,
|
||||
} },
|
||||
{ formats::SBGGR10_IPU3, {
|
||||
.bitsPerSample = 16,
|
||||
.pattern = { CFAPatternBlue, CFAPatternGreen, CFAPatternGreen, CFAPatternRed },
|
||||
.packScanline = packScanlineIPU3,
|
||||
.thumbScanline = thumbScanlineIPU3,
|
||||
} },
|
||||
{ formats::SGBRG10_IPU3, {
|
||||
.bitsPerSample = 16,
|
||||
.pattern = { CFAPatternGreen, CFAPatternBlue, CFAPatternRed, CFAPatternGreen },
|
||||
.packScanline = packScanlineIPU3,
|
||||
.thumbScanline = thumbScanlineIPU3,
|
||||
} },
|
||||
{ formats::SGRBG10_IPU3, {
|
||||
.bitsPerSample = 16,
|
||||
.pattern = { CFAPatternGreen, CFAPatternRed, CFAPatternBlue, CFAPatternGreen },
|
||||
.packScanline = packScanlineIPU3,
|
||||
.thumbScanline = thumbScanlineIPU3,
|
||||
} },
|
||||
{ formats::SRGGB10_IPU3, {
|
||||
.bitsPerSample = 16,
|
||||
.pattern = { CFAPatternRed, CFAPatternGreen, CFAPatternGreen, CFAPatternBlue },
|
||||
.packScanline = packScanlineIPU3,
|
||||
.thumbScanline = thumbScanlineIPU3,
|
||||
} },
|
||||
};
|
||||
|
||||
int DNGWriter::write(const char *filename, const Camera *camera,
|
||||
const StreamConfiguration &config,
|
||||
const ControlList &metadata,
|
||||
[[maybe_unused]] const FrameBuffer *buffer,
|
||||
const void *data)
|
||||
{
|
||||
const ControlList &cameraProperties = camera->properties();
|
||||
|
||||
const auto it = formatInfo.find(config.pixelFormat);
|
||||
if (it == formatInfo.cend()) {
|
||||
std::cerr << "Unsupported pixel format" << std::endl;
|
||||
return -EINVAL;
|
||||
}
|
||||
const FormatInfo *info = &it->second;
|
||||
|
||||
TIFF *tif = TIFFOpen(filename, "w");
|
||||
if (!tif) {
|
||||
std::cerr << "Failed to open tiff file" << std::endl;
|
||||
return -EINVAL;
|
||||
}
|
||||
|
||||
/*
|
||||
* Scanline buffer, has to be large enough to store both a RAW scanline
|
||||
* or a thumbnail scanline. The latter will always be much smaller than
|
||||
* the former as we downscale by 16 in both directions.
|
||||
*/
|
||||
uint8_t scanline[(config.size.width * info->bitsPerSample + 7) / 8];
|
||||
|
||||
toff_t rawIFDOffset = 0;
|
||||
toff_t exifIFDOffset = 0;
|
||||
|
||||
/*
|
||||
* Start with a thumbnail in IFD 0 for compatibility with TIFF baseline
|
||||
* readers, as required by the TIFF/EP specification. Tags that apply to
|
||||
* the whole file are stored here.
|
||||
*/
|
||||
const uint8_t version[] = { 1, 2, 0, 0 };
|
||||
|
||||
TIFFSetField(tif, TIFFTAG_DNGVERSION, version);
|
||||
TIFFSetField(tif, TIFFTAG_DNGBACKWARDVERSION, version);
|
||||
TIFFSetField(tif, TIFFTAG_FILLORDER, FILLORDER_MSB2LSB);
|
||||
TIFFSetField(tif, TIFFTAG_MAKE, "libcamera");
|
||||
|
||||
const auto &model = cameraProperties.get(properties::Model);
|
||||
if (model) {
|
||||
TIFFSetField(tif, TIFFTAG_MODEL, model->c_str());
|
||||
/* \todo set TIFFTAG_UNIQUECAMERAMODEL. */
|
||||
}
|
||||
|
||||
TIFFSetField(tif, TIFFTAG_SOFTWARE, "qcam");
|
||||
TIFFSetField(tif, TIFFTAG_ORIENTATION, ORIENTATION_TOPLEFT);
|
||||
|
||||
/*
|
||||
* Thumbnail-specific tags. The thumbnail is stored as an RGB image
|
||||
* with 1/16 of the raw image resolution. Greyscale would save space,
|
||||
* but doesn't seem well supported by RawTherapee.
|
||||
*/
|
||||
TIFFSetField(tif, TIFFTAG_SUBFILETYPE, FILETYPE_REDUCEDIMAGE);
|
||||
TIFFSetField(tif, TIFFTAG_IMAGEWIDTH, config.size.width / 16);
|
||||
TIFFSetField(tif, TIFFTAG_IMAGELENGTH, config.size.height / 16);
|
||||
TIFFSetField(tif, TIFFTAG_BITSPERSAMPLE, 8);
|
||||
TIFFSetField(tif, TIFFTAG_COMPRESSION, COMPRESSION_NONE);
|
||||
TIFFSetField(tif, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_RGB);
|
||||
TIFFSetField(tif, TIFFTAG_SAMPLESPERPIXEL, 3);
|
||||
TIFFSetField(tif, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
|
||||
TIFFSetField(tif, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_UINT);
|
||||
|
||||
/*
|
||||
* Fill in some reasonable colour information in the DNG. We supply
|
||||
* the "neutral" colour values which determine the white balance, and the
|
||||
* "ColorMatrix1" which converts XYZ to (un-white-balanced) camera RGB.
|
||||
* Note that this is not a "proper" colour calibration for the DNG,
|
||||
* nonetheless, many tools should be able to render the colours better.
|
||||
*/
|
||||
float neutral[3] = { 1, 1, 1 };
|
||||
Matrix3d wbGain = Matrix3d::identity();
|
||||
/* From http://www.brucelindbloom.com/index.html?Eqn_RGB_XYZ_Matrix.html */
|
||||
const Matrix3d rgb2xyz(0.4124564, 0.3575761, 0.1804375,
|
||||
0.2126729, 0.7151522, 0.0721750,
|
||||
0.0193339, 0.1191920, 0.9503041);
|
||||
Matrix3d ccm = Matrix3d::identity();
|
||||
/*
|
||||
* Pick a reasonable number eps to protect against singularities. It
|
||||
* should be comfortably larger than the point at which we run into
|
||||
* numerical trouble, yet smaller than any plausible gain that we might
|
||||
* apply to a colour, either explicitly or as part of the colour matrix.
|
||||
*/
|
||||
const double eps = 1e-2;
|
||||
|
||||
const auto &colourGains = metadata.get(controls::ColourGains);
|
||||
if (colourGains) {
|
||||
if ((*colourGains)[0] > eps && (*colourGains)[1] > eps) {
|
||||
wbGain = Matrix3d::diag((*colourGains)[0], 1, (*colourGains)[1]);
|
||||
neutral[0] = 1.0 / (*colourGains)[0]; /* red */
|
||||
neutral[2] = 1.0 / (*colourGains)[1]; /* blue */
|
||||
}
|
||||
}
|
||||
|
||||
const auto &ccmControl = metadata.get(controls::ColourCorrectionMatrix);
|
||||
if (ccmControl) {
|
||||
Matrix3d ccmSupplied(*ccmControl);
|
||||
if (ccmSupplied.determinant() > eps)
|
||||
ccm = ccmSupplied;
|
||||
}
|
||||
|
||||
/*
|
||||
* rgb2xyz is known to be invertible, and we've ensured above that both
|
||||
* the ccm and wbGain matrices are non-singular, so the product of all
|
||||
* three is guaranteed to be invertible too.
|
||||
*/
|
||||
Matrix3d colorMatrix1 = (rgb2xyz * ccm * wbGain).inverse();
|
||||
|
||||
TIFFSetField(tif, TIFFTAG_COLORMATRIX1, 9, colorMatrix1.m);
|
||||
TIFFSetField(tif, TIFFTAG_ASSHOTNEUTRAL, 3, neutral);
|
||||
|
||||
/*
|
||||
* Reserve space for the SubIFD and ExifIFD tags, pointing to the IFD
|
||||
* for the raw image and EXIF data respectively. The real offsets will
|
||||
* be set later.
|
||||
*/
|
||||
TIFFSetField(tif, TIFFTAG_SUBIFD, 1, &rawIFDOffset);
|
||||
TIFFSetField(tif, TIFFTAG_EXIFIFD, exifIFDOffset);
|
||||
|
||||
/* Write the thumbnail. */
|
||||
const uint8_t *row = static_cast<const uint8_t *>(data);
|
||||
for (unsigned int y = 0; y < config.size.height / 16; y++) {
|
||||
info->thumbScanline(*info, &scanline, row,
|
||||
config.size.width / 16, config.stride);
|
||||
|
||||
if (TIFFWriteScanline(tif, &scanline, y, 0) != 1) {
|
||||
std::cerr << "Failed to write thumbnail scanline"
|
||||
<< std::endl;
|
||||
TIFFClose(tif);
|
||||
return -EINVAL;
|
||||
}
|
||||
|
||||
row += config.stride * 16;
|
||||
}
|
||||
|
||||
TIFFWriteDirectory(tif);
|
||||
|
||||
/* Create a new IFD for the RAW image. */
|
||||
const uint16_t cfaRepeatPatternDim[] = { 2, 2 };
|
||||
const uint8_t cfaPlaneColor[] = {
|
||||
CFAPatternRed,
|
||||
CFAPatternGreen,
|
||||
CFAPatternBlue
|
||||
};
|
||||
|
||||
TIFFSetField(tif, TIFFTAG_SUBFILETYPE, 0);
|
||||
TIFFSetField(tif, TIFFTAG_IMAGEWIDTH, config.size.width);
|
||||
TIFFSetField(tif, TIFFTAG_IMAGELENGTH, config.size.height);
|
||||
TIFFSetField(tif, TIFFTAG_BITSPERSAMPLE, info->bitsPerSample);
|
||||
TIFFSetField(tif, TIFFTAG_COMPRESSION, COMPRESSION_NONE);
|
||||
TIFFSetField(tif, TIFFTAG_PHOTOMETRIC, PHOTOMETRIC_CFA);
|
||||
TIFFSetField(tif, TIFFTAG_SAMPLESPERPIXEL, 1);
|
||||
TIFFSetField(tif, TIFFTAG_PLANARCONFIG, PLANARCONFIG_CONTIG);
|
||||
TIFFSetField(tif, TIFFTAG_SAMPLEFORMAT, SAMPLEFORMAT_UINT);
|
||||
TIFFSetField(tif, TIFFTAG_CFAREPEATPATTERNDIM, cfaRepeatPatternDim);
|
||||
if (TIFFLIB_VERSION < 20201219)
|
||||
TIFFSetField(tif, TIFFTAG_CFAPATTERN, info->pattern);
|
||||
else
|
||||
TIFFSetField(tif, TIFFTAG_CFAPATTERN, 4, info->pattern);
|
||||
TIFFSetField(tif, TIFFTAG_CFAPLANECOLOR, 3, cfaPlaneColor);
|
||||
TIFFSetField(tif, TIFFTAG_CFALAYOUT, 1);
|
||||
|
||||
const uint16_t blackLevelRepeatDim[] = { 2, 2 };
|
||||
float blackLevel[] = { 0.0f, 0.0f, 0.0f, 0.0f };
|
||||
uint32_t whiteLevel = (1 << info->bitsPerSample) - 1;
|
||||
|
||||
const auto &blackLevels = metadata.get(controls::SensorBlackLevels);
|
||||
if (blackLevels) {
|
||||
Span<const int32_t, 4> levels = *blackLevels;
|
||||
|
||||
/*
|
||||
* The black levels control is specified in R, Gr, Gb, B order.
|
||||
* Map it to the TIFF tag that is specified in CFA pattern
|
||||
* order.
|
||||
*/
|
||||
unsigned int green = (info->pattern[0] == CFAPatternRed ||
|
||||
info->pattern[1] == CFAPatternRed)
|
||||
? 0 : 1;
|
||||
|
||||
for (unsigned int i = 0; i < 4; ++i) {
|
||||
unsigned int level;
|
||||
|
||||
switch (info->pattern[i]) {
|
||||
case CFAPatternRed:
|
||||
level = levels[0];
|
||||
break;
|
||||
case CFAPatternGreen:
|
||||
level = levels[green + 1];
|
||||
green = (green + 1) % 2;
|
||||
break;
|
||||
case CFAPatternBlue:
|
||||
default:
|
||||
level = levels[3];
|
||||
break;
|
||||
}
|
||||
|
||||
/* Map the 16-bit value to the bits per sample range. */
|
||||
blackLevel[i] = level >> (16 - info->bitsPerSample);
|
||||
}
|
||||
}
|
||||
|
||||
TIFFSetField(tif, TIFFTAG_BLACKLEVELREPEATDIM, &blackLevelRepeatDim);
|
||||
TIFFSetField(tif, TIFFTAG_BLACKLEVEL, 4, &blackLevel);
|
||||
TIFFSetField(tif, TIFFTAG_WHITELEVEL, 1, &whiteLevel);
|
||||
|
||||
/* Write RAW content. */
|
||||
row = static_cast<const uint8_t *>(data);
|
||||
for (unsigned int y = 0; y < config.size.height; y++) {
|
||||
info->packScanline(&scanline, row, config.size.width);
|
||||
|
||||
if (TIFFWriteScanline(tif, &scanline, y, 0) != 1) {
|
||||
std::cerr << "Failed to write RAW scanline"
|
||||
<< std::endl;
|
||||
TIFFClose(tif);
|
||||
return -EINVAL;
|
||||
}
|
||||
|
||||
row += config.stride;
|
||||
}
|
||||
|
||||
/* Checkpoint the IFD to retrieve its offset, and write it out. */
|
||||
TIFFCheckpointDirectory(tif);
|
||||
rawIFDOffset = TIFFCurrentDirOffset(tif);
|
||||
TIFFWriteDirectory(tif);
|
||||
|
||||
/* Create a new IFD for the EXIF data and fill it. */
|
||||
TIFFCreateEXIFDirectory(tif);
|
||||
|
||||
/* Store creation time. */
|
||||
time_t rawtime;
|
||||
struct tm *timeinfo;
|
||||
char strTime[20];
|
||||
|
||||
time(&rawtime);
|
||||
timeinfo = localtime(&rawtime);
|
||||
strftime(strTime, 20, "%Y:%m:%d %H:%M:%S", timeinfo);
|
||||
|
||||
/*
|
||||
* \todo Handle timezone information by setting OffsetTimeOriginal and
|
||||
* OffsetTimeDigitized once libtiff catches up to the specification and
|
||||
* has EXIFTAG_ defines to handle them.
|
||||
*/
|
||||
TIFFSetField(tif, EXIFTAG_DATETIMEORIGINAL, strTime);
|
||||
TIFFSetField(tif, EXIFTAG_DATETIMEDIGITIZED, strTime);
|
||||
|
||||
const auto &analogGain = metadata.get(controls::AnalogueGain);
|
||||
if (analogGain) {
|
||||
uint16_t iso = std::min(std::max(*analogGain * 100, 0.0f), 65535.0f);
|
||||
TIFFSetField(tif, EXIFTAG_ISOSPEEDRATINGS, 1, &iso);
|
||||
}
|
||||
|
||||
const auto &exposureTime = metadata.get(controls::ExposureTime);
|
||||
if (exposureTime)
|
||||
TIFFSetField(tif, EXIFTAG_EXPOSURETIME, *exposureTime / 1e6);
|
||||
|
||||
TIFFWriteCustomDirectory(tif, &exifIFDOffset);
|
||||
|
||||
/* Update the IFD offsets and close the file. */
|
||||
TIFFSetDirectory(tif, 0);
|
||||
TIFFSetField(tif, TIFFTAG_SUBIFD, 1, &rawIFDOffset);
|
||||
TIFFSetField(tif, TIFFTAG_EXIFIFD, exifIFDOffset);
|
||||
TIFFWriteDirectory(tif);
|
||||
|
||||
TIFFClose(tif);
|
||||
|
||||
return 0;
|
||||
}
|
||||
@@ -0,0 +1,27 @@
|
||||
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
||||
/*
|
||||
* Copyright (C) 2020, Raspberry Pi Ltd
|
||||
*
|
||||
* dng_writer.h - DNG writer
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#ifdef HAVE_TIFF
|
||||
#define HAVE_DNG
|
||||
|
||||
#include <libcamera/camera.h>
|
||||
#include <libcamera/controls.h>
|
||||
#include <libcamera/framebuffer.h>
|
||||
#include <libcamera/stream.h>
|
||||
|
||||
class DNGWriter
|
||||
{
|
||||
public:
|
||||
static int write(const char *filename, const libcamera::Camera *camera,
|
||||
const libcamera::StreamConfiguration &config,
|
||||
const libcamera::ControlList &metadata,
|
||||
const libcamera::FrameBuffer *buffer, const void *data);
|
||||
};
|
||||
|
||||
#endif /* HAVE_TIFF */
|
||||
@@ -0,0 +1,150 @@
|
||||
/* SPDX-License-Identifier: GPL-2.0-or-later */
|
||||
/*
|
||||
* Copyright (C) 2019, Google Inc.
|
||||
*
|
||||
* event_loop.cpp - cam - Event loop
|
||||
*/
|
||||
|
||||
#include "event_loop.h"
|
||||
|
||||
#include <assert.h>
|
||||
#include <event2/event.h>
|
||||
#include <event2/thread.h>
|
||||
#include <iostream>
|
||||
|
||||
EventLoop *EventLoop::instance_ = nullptr;
|
||||
|
||||
EventLoop::EventLoop()
|
||||
{
|
||||
assert(!instance_);
|
||||
|
||||
evthread_use_pthreads();
|
||||
base_ = event_base_new();
|
||||
instance_ = this;
|
||||
}
|
||||
|
||||
EventLoop::~EventLoop()
|
||||
{
|
||||
instance_ = nullptr;
|
||||
|
||||
events_.clear();
|
||||
event_base_free(base_);
|
||||
libevent_global_shutdown();
|
||||
}
|
||||
|
||||
EventLoop *EventLoop::instance()
|
||||
{
|
||||
return instance_;
|
||||
}
|
||||
|
||||
int EventLoop::exec()
|
||||
{
|
||||
exitCode_ = -1;
|
||||
event_base_loop(base_, EVLOOP_NO_EXIT_ON_EMPTY);
|
||||
return exitCode_;
|
||||
}
|
||||
|
||||
void EventLoop::exit(int code)
|
||||
{
|
||||
exitCode_ = code;
|
||||
event_base_loopbreak(base_);
|
||||
}
|
||||
|
||||
void EventLoop::callLater(const std::function<void()> &func)
|
||||
{
|
||||
{
|
||||
std::unique_lock<std::mutex> locker(lock_);
|
||||
calls_.push_back(func);
|
||||
}
|
||||
|
||||
event_base_once(base_, -1, EV_TIMEOUT, dispatchCallback, this, nullptr);
|
||||
}
|
||||
|
||||
void EventLoop::addFdEvent(int fd, EventType type,
|
||||
const std::function<void()> &callback)
|
||||
{
|
||||
std::unique_ptr<Event> event = std::make_unique<Event>(callback);
|
||||
short events = (type & Read ? EV_READ : 0)
|
||||
| (type & Write ? EV_WRITE : 0)
|
||||
| EV_PERSIST;
|
||||
|
||||
event->event_ = event_new(base_, fd, events, &EventLoop::Event::dispatch,
|
||||
event.get());
|
||||
if (!event->event_) {
|
||||
std::cerr << "Failed to create event for fd " << fd << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
int ret = event_add(event->event_, nullptr);
|
||||
if (ret < 0) {
|
||||
std::cerr << "Failed to add event for fd " << fd << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
events_.push_back(std::move(event));
|
||||
}
|
||||
|
||||
void EventLoop::addTimerEvent(const std::chrono::microseconds period,
|
||||
const std::function<void()> &callback)
|
||||
{
|
||||
std::unique_ptr<Event> event = std::make_unique<Event>(callback);
|
||||
event->event_ = event_new(base_, -1, EV_PERSIST, &EventLoop::Event::dispatch,
|
||||
event.get());
|
||||
if (!event->event_) {
|
||||
std::cerr << "Failed to create timer event" << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
struct timeval tv;
|
||||
tv.tv_sec = period.count() / 1000000ULL;
|
||||
tv.tv_usec = period.count() % 1000000ULL;
|
||||
|
||||
int ret = event_add(event->event_, &tv);
|
||||
if (ret < 0) {
|
||||
std::cerr << "Failed to add timer event" << std::endl;
|
||||
return;
|
||||
}
|
||||
|
||||
events_.push_back(std::move(event));
|
||||
}
|
||||
|
||||
void EventLoop::dispatchCallback([[maybe_unused]] evutil_socket_t fd,
|
||||
[[maybe_unused]] short flags, void *param)
|
||||
{
|
||||
EventLoop *loop = static_cast<EventLoop *>(param);
|
||||
loop->dispatchCall();
|
||||
}
|
||||
|
||||
void EventLoop::dispatchCall()
|
||||
{
|
||||
std::function<void()> call;
|
||||
|
||||
{
|
||||
std::unique_lock<std::mutex> locker(lock_);
|
||||
if (calls_.empty())
|
||||
return;
|
||||
|
||||
call = calls_.front();
|
||||
calls_.pop_front();
|
||||
}
|
||||
|
||||
call();
|
||||
}
|
||||
|
||||
EventLoop::Event::Event(const std::function<void()> &callback)
|
||||
: callback_(callback), event_(nullptr)
|
||||
{
|
||||
}
|
||||
|
||||
EventLoop::Event::~Event()
|
||||
{
|
||||
event_del(event_);
|
||||
event_free(event_);
|
||||
}
|
||||
|
||||
void EventLoop::Event::dispatch([[maybe_unused]] int fd,
|
||||
[[maybe_unused]] short events, void *arg)
|
||||
{
|
||||
Event *event = static_cast<Event *>(arg);
|
||||
event->callback_();
|
||||
}
|
||||
@@ -0,0 +1,68 @@
|
||||
/* SPDX-License-Identifier: GPL-2.0-or-later */
|
||||
/*
|
||||
* Copyright (C) 2019, Google Inc.
|
||||
*
|
||||
* event_loop.h - cam - Event loop
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <chrono>
|
||||
#include <functional>
|
||||
#include <list>
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
|
||||
#include <event2/util.h>
|
||||
|
||||
struct event_base;
|
||||
|
||||
class EventLoop
|
||||
{
|
||||
public:
|
||||
enum EventType {
|
||||
Read = 1,
|
||||
Write = 2,
|
||||
};
|
||||
|
||||
EventLoop();
|
||||
~EventLoop();
|
||||
|
||||
static EventLoop *instance();
|
||||
|
||||
int exec();
|
||||
void exit(int code = 0);
|
||||
|
||||
void callLater(const std::function<void()> &func);
|
||||
|
||||
void addFdEvent(int fd, EventType type,
|
||||
const std::function<void()> &handler);
|
||||
|
||||
using duration = std::chrono::steady_clock::duration;
|
||||
void addTimerEvent(const std::chrono::microseconds period,
|
||||
const std::function<void()> &handler);
|
||||
|
||||
private:
|
||||
struct Event {
|
||||
Event(const std::function<void()> &callback);
|
||||
~Event();
|
||||
|
||||
static void dispatch(int fd, short events, void *arg);
|
||||
|
||||
std::function<void()> callback_;
|
||||
struct event *event_;
|
||||
};
|
||||
|
||||
static EventLoop *instance_;
|
||||
|
||||
struct event_base *base_;
|
||||
int exitCode_;
|
||||
|
||||
std::list<std::function<void()>> calls_;
|
||||
std::list<std::unique_ptr<Event>> events_;
|
||||
std::mutex lock_;
|
||||
|
||||
static void dispatchCallback(evutil_socket_t fd, short flags,
|
||||
void *param);
|
||||
void dispatchCall();
|
||||
};
|
||||
@@ -0,0 +1,109 @@
|
||||
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
||||
/*
|
||||
* Copyright (C) 2021, Ideas on Board Oy
|
||||
*
|
||||
* image.cpp - Multi-planar image with access to pixel data
|
||||
*/
|
||||
|
||||
#include "image.h"
|
||||
|
||||
#include <assert.h>
|
||||
#include <errno.h>
|
||||
#include <iostream>
|
||||
#include <map>
|
||||
#include <string.h>
|
||||
#include <sys/mman.h>
|
||||
#include <unistd.h>
|
||||
|
||||
using namespace libcamera;
|
||||
|
||||
std::unique_ptr<Image> Image::fromFrameBuffer(const FrameBuffer *buffer, MapMode mode)
|
||||
{
|
||||
std::unique_ptr<Image> image{ new Image() };
|
||||
|
||||
assert(!buffer->planes().empty());
|
||||
|
||||
int mmapFlags = 0;
|
||||
|
||||
if (mode & MapMode::ReadOnly)
|
||||
mmapFlags |= PROT_READ;
|
||||
|
||||
if (mode & MapMode::WriteOnly)
|
||||
mmapFlags |= PROT_WRITE;
|
||||
|
||||
struct MappedBufferInfo {
|
||||
uint8_t *address = nullptr;
|
||||
size_t mapLength = 0;
|
||||
size_t dmabufLength = 0;
|
||||
};
|
||||
std::map<int, MappedBufferInfo> mappedBuffers;
|
||||
|
||||
for (const FrameBuffer::Plane &plane : buffer->planes()) {
|
||||
const int fd = plane.fd.get();
|
||||
if (mappedBuffers.find(fd) == mappedBuffers.end()) {
|
||||
const size_t length = lseek(fd, 0, SEEK_END);
|
||||
mappedBuffers[fd] = MappedBufferInfo{ nullptr, 0, length };
|
||||
}
|
||||
|
||||
const size_t length = mappedBuffers[fd].dmabufLength;
|
||||
|
||||
if (plane.offset > length ||
|
||||
plane.offset + plane.length > length) {
|
||||
std::cerr << "plane is out of buffer: buffer length="
|
||||
<< length << ", plane offset=" << plane.offset
|
||||
<< ", plane length=" << plane.length
|
||||
<< std::endl;
|
||||
return nullptr;
|
||||
}
|
||||
size_t &mapLength = mappedBuffers[fd].mapLength;
|
||||
mapLength = std::max(mapLength,
|
||||
static_cast<size_t>(plane.offset + plane.length));
|
||||
}
|
||||
|
||||
for (const FrameBuffer::Plane &plane : buffer->planes()) {
|
||||
const int fd = plane.fd.get();
|
||||
auto &info = mappedBuffers[fd];
|
||||
if (!info.address) {
|
||||
void *address = mmap(nullptr, info.mapLength, mmapFlags,
|
||||
MAP_SHARED, fd, 0);
|
||||
if (address == MAP_FAILED) {
|
||||
int error = -errno;
|
||||
std::cerr << "Failed to mmap plane: "
|
||||
<< strerror(-error) << std::endl;
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
info.address = static_cast<uint8_t *>(address);
|
||||
image->maps_.emplace_back(info.address, info.mapLength);
|
||||
}
|
||||
|
||||
image->planes_.emplace_back(info.address + plane.offset, plane.length);
|
||||
}
|
||||
|
||||
return image;
|
||||
}
|
||||
|
||||
Image::Image() = default;
|
||||
|
||||
Image::~Image()
|
||||
{
|
||||
for (Span<uint8_t> &map : maps_)
|
||||
munmap(map.data(), map.size());
|
||||
}
|
||||
|
||||
unsigned int Image::numPlanes() const
|
||||
{
|
||||
return planes_.size();
|
||||
}
|
||||
|
||||
Span<uint8_t> Image::data(unsigned int plane)
|
||||
{
|
||||
assert(plane <= planes_.size());
|
||||
return planes_[plane];
|
||||
}
|
||||
|
||||
Span<const uint8_t> Image::data(unsigned int plane) const
|
||||
{
|
||||
assert(plane <= planes_.size());
|
||||
return planes_[plane];
|
||||
}
|
||||
@@ -0,0 +1,50 @@
|
||||
/* SPDX-License-Identifier: LGPL-2.1-or-later */
|
||||
/*
|
||||
* Copyright (C) 2021, Ideas on Board Oy
|
||||
*
|
||||
* image.h - Multi-planar image with access to pixel data
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <memory>
|
||||
#include <stdint.h>
|
||||
#include <vector>
|
||||
|
||||
#include <libcamera/base/class.h>
|
||||
#include <libcamera/base/flags.h>
|
||||
#include <libcamera/base/span.h>
|
||||
|
||||
#include <libcamera/framebuffer.h>
|
||||
|
||||
class Image
|
||||
{
|
||||
public:
|
||||
enum class MapMode {
|
||||
ReadOnly = 1 << 0,
|
||||
WriteOnly = 1 << 1,
|
||||
ReadWrite = ReadOnly | WriteOnly,
|
||||
};
|
||||
|
||||
static std::unique_ptr<Image> fromFrameBuffer(const libcamera::FrameBuffer *buffer,
|
||||
MapMode mode);
|
||||
|
||||
~Image();
|
||||
|
||||
unsigned int numPlanes() const;
|
||||
|
||||
libcamera::Span<uint8_t> data(unsigned int plane);
|
||||
libcamera::Span<const uint8_t> data(unsigned int plane) const;
|
||||
|
||||
private:
|
||||
LIBCAMERA_DISABLE_COPY(Image)
|
||||
|
||||
Image();
|
||||
|
||||
std::vector<libcamera::Span<uint8_t>> maps_;
|
||||
std::vector<libcamera::Span<uint8_t>> planes_;
|
||||
};
|
||||
|
||||
namespace libcamera {
|
||||
LIBCAMERA_FLAGS_ENABLE_OPERATORS(Image::MapMode)
|
||||
}
|
||||
@@ -0,0 +1,26 @@
|
||||
# SPDX-License-Identifier: CC0-1.0
|
||||
|
||||
apps_sources = files([
|
||||
'image.cpp',
|
||||
'options.cpp',
|
||||
'stream_options.cpp',
|
||||
])
|
||||
|
||||
apps_cpp_args = []
|
||||
|
||||
if libevent.found()
|
||||
apps_sources += files([
|
||||
'event_loop.cpp',
|
||||
])
|
||||
endif
|
||||
|
||||
if libtiff.found()
|
||||
apps_cpp_args += ['-DHAVE_TIFF']
|
||||
apps_sources += files([
|
||||
'dng_writer.cpp',
|
||||
])
|
||||
endif
|
||||
|
||||
apps_lib = static_library('apps', apps_sources,
|
||||
cpp_args : apps_cpp_args,
|
||||
dependencies : [libcamera_public])
|
||||
File diff suppressed because it is too large
Load Diff
@@ -0,0 +1,157 @@
|
||||
/* SPDX-License-Identifier: GPL-2.0-or-later */
|
||||
/*
|
||||
* Copyright (C) 2019, Google Inc.
|
||||
*
|
||||
* options.h - cam - Options parsing
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <ctype.h>
|
||||
#include <list>
|
||||
#include <map>
|
||||
#include <tuple>
|
||||
#include <vector>
|
||||
|
||||
class KeyValueParser;
|
||||
class OptionValue;
|
||||
struct Option;
|
||||
|
||||
enum OptionArgument {
|
||||
ArgumentNone,
|
||||
ArgumentRequired,
|
||||
ArgumentOptional,
|
||||
};
|
||||
|
||||
enum OptionType {
|
||||
OptionNone,
|
||||
OptionInteger,
|
||||
OptionString,
|
||||
OptionKeyValue,
|
||||
};
|
||||
|
||||
template<typename T>
|
||||
class OptionsBase
|
||||
{
|
||||
public:
|
||||
OptionsBase() : valid_(false) {}
|
||||
|
||||
bool empty() const;
|
||||
bool valid() const;
|
||||
bool isSet(const T &opt) const;
|
||||
const OptionValue &operator[](const T &opt) const;
|
||||
|
||||
void invalidate();
|
||||
|
||||
private:
|
||||
friend class KeyValueParser;
|
||||
friend class OptionsParser;
|
||||
|
||||
bool parseValue(const T &opt, const Option &option, const char *value);
|
||||
|
||||
std::map<T, OptionValue> values_;
|
||||
bool valid_;
|
||||
};
|
||||
|
||||
class KeyValueParser
|
||||
{
|
||||
public:
|
||||
class Options : public OptionsBase<std::string>
|
||||
{
|
||||
};
|
||||
|
||||
KeyValueParser();
|
||||
virtual ~KeyValueParser();
|
||||
|
||||
bool addOption(const char *name, OptionType type, const char *help,
|
||||
OptionArgument argument = ArgumentNone);
|
||||
|
||||
virtual Options parse(const char *arguments);
|
||||
|
||||
private:
|
||||
KeyValueParser(const KeyValueParser &) = delete;
|
||||
KeyValueParser &operator=(const KeyValueParser &) = delete;
|
||||
|
||||
friend class OptionsParser;
|
||||
unsigned int maxOptionLength() const;
|
||||
void usage(int indent);
|
||||
|
||||
std::map<std::string, Option> optionsMap_;
|
||||
};
|
||||
|
||||
class OptionsParser
|
||||
{
|
||||
public:
|
||||
class Options : public OptionsBase<int>
|
||||
{
|
||||
};
|
||||
|
||||
OptionsParser();
|
||||
~OptionsParser();
|
||||
|
||||
bool addOption(int opt, OptionType type, const char *help,
|
||||
const char *name = nullptr,
|
||||
OptionArgument argument = ArgumentNone,
|
||||
const char *argumentName = nullptr, bool array = false,
|
||||
int parent = 0);
|
||||
bool addOption(int opt, KeyValueParser *parser, const char *help,
|
||||
const char *name = nullptr, bool array = false,
|
||||
int parent = 0);
|
||||
|
||||
Options parse(int argc, char *argv[]);
|
||||
void usage();
|
||||
|
||||
private:
|
||||
OptionsParser(const OptionsParser &) = delete;
|
||||
OptionsParser &operator=(const OptionsParser &) = delete;
|
||||
|
||||
void usageOptions(const std::list<Option> &options, unsigned int indent);
|
||||
|
||||
std::tuple<OptionsParser::Options *, const Option *>
|
||||
childOption(const Option *parent, Options *options);
|
||||
bool parseValue(const Option &option, const char *arg, Options *options);
|
||||
|
||||
std::list<Option> options_;
|
||||
std::map<unsigned int, Option *> optionsMap_;
|
||||
};
|
||||
|
||||
class OptionValue
|
||||
{
|
||||
public:
|
||||
enum ValueType {
|
||||
ValueNone,
|
||||
ValueInteger,
|
||||
ValueString,
|
||||
ValueKeyValue,
|
||||
ValueArray,
|
||||
};
|
||||
|
||||
OptionValue();
|
||||
OptionValue(int value);
|
||||
OptionValue(const char *value);
|
||||
OptionValue(const std::string &value);
|
||||
OptionValue(const KeyValueParser::Options &value);
|
||||
|
||||
void addValue(const OptionValue &value);
|
||||
|
||||
ValueType type() const { return type_; }
|
||||
bool empty() const { return type_ == ValueType::ValueNone; }
|
||||
|
||||
operator int() const;
|
||||
operator std::string() const;
|
||||
|
||||
int toInteger() const;
|
||||
std::string toString() const;
|
||||
const KeyValueParser::Options &toKeyValues() const;
|
||||
const std::vector<OptionValue> &toArray() const;
|
||||
|
||||
const OptionsParser::Options &children() const;
|
||||
|
||||
private:
|
||||
ValueType type_;
|
||||
int integer_;
|
||||
std::string string_;
|
||||
KeyValueParser::Options keyValues_;
|
||||
std::vector<OptionValue> array_;
|
||||
OptionsParser::Options children_;
|
||||
};
|
||||
@@ -0,0 +1,134 @@
|
||||
/* SPDX-License-Identifier: GPL-2.0-or-later */
|
||||
/*
|
||||
* Copyright (C) 2020, Raspberry Pi Ltd
|
||||
*
|
||||
* stream_options.cpp - Helper to parse options for streams
|
||||
*/
|
||||
#include "stream_options.h"
|
||||
|
||||
#include <iostream>
|
||||
|
||||
#include <libcamera/color_space.h>
|
||||
|
||||
using namespace libcamera;
|
||||
|
||||
StreamKeyValueParser::StreamKeyValueParser()
|
||||
{
|
||||
addOption("role", OptionString,
|
||||
"Role for the stream (viewfinder, video, still, raw)",
|
||||
ArgumentRequired);
|
||||
addOption("width", OptionInteger, "Width in pixels",
|
||||
ArgumentRequired);
|
||||
addOption("height", OptionInteger, "Height in pixels",
|
||||
ArgumentRequired);
|
||||
addOption("pixelformat", OptionString, "Pixel format name",
|
||||
ArgumentRequired);
|
||||
addOption("colorspace", OptionString, "Color space",
|
||||
ArgumentRequired);
|
||||
}
|
||||
|
||||
KeyValueParser::Options StreamKeyValueParser::parse(const char *arguments)
|
||||
{
|
||||
KeyValueParser::Options options = KeyValueParser::parse(arguments);
|
||||
StreamRole role;
|
||||
|
||||
if (options.valid() && options.isSet("role") &&
|
||||
!parseRole(&role, options)) {
|
||||
std::cerr << "Unknown stream role "
|
||||
<< options["role"].toString() << std::endl;
|
||||
options.invalidate();
|
||||
}
|
||||
|
||||
return options;
|
||||
}
|
||||
|
||||
StreamRoles StreamKeyValueParser::roles(const OptionValue &values)
|
||||
{
|
||||
/* If no configuration values to examine default to viewfinder. */
|
||||
if (values.empty())
|
||||
return { StreamRole::Viewfinder };
|
||||
|
||||
const std::vector<OptionValue> &streamParameters = values.toArray();
|
||||
|
||||
StreamRoles roles;
|
||||
for (auto const &value : streamParameters) {
|
||||
StreamRole role;
|
||||
|
||||
/* If role is invalid or not set default to viewfinder. */
|
||||
if (!parseRole(&role, value.toKeyValues()))
|
||||
role = StreamRole::Viewfinder;
|
||||
|
||||
roles.push_back(role);
|
||||
}
|
||||
|
||||
return roles;
|
||||
}
|
||||
|
||||
int StreamKeyValueParser::updateConfiguration(CameraConfiguration *config,
|
||||
const OptionValue &values)
|
||||
{
|
||||
if (!config) {
|
||||
std::cerr << "No configuration provided" << std::endl;
|
||||
return -EINVAL;
|
||||
}
|
||||
|
||||
/* If no configuration values nothing to do. */
|
||||
if (values.empty())
|
||||
return 0;
|
||||
|
||||
const std::vector<OptionValue> &streamParameters = values.toArray();
|
||||
|
||||
if (config->size() != streamParameters.size()) {
|
||||
std::cerr
|
||||
<< "Number of streams in configuration "
|
||||
<< config->size()
|
||||
<< " does not match number of streams parsed "
|
||||
<< streamParameters.size()
|
||||
<< std::endl;
|
||||
return -EINVAL;
|
||||
}
|
||||
|
||||
unsigned int i = 0;
|
||||
for (auto const &value : streamParameters) {
|
||||
KeyValueParser::Options opts = value.toKeyValues();
|
||||
StreamConfiguration &cfg = config->at(i++);
|
||||
|
||||
if (opts.isSet("width") && opts.isSet("height")) {
|
||||
cfg.size.width = opts["width"];
|
||||
cfg.size.height = opts["height"];
|
||||
}
|
||||
|
||||
if (opts.isSet("pixelformat"))
|
||||
cfg.pixelFormat = PixelFormat::fromString(opts["pixelformat"].toString());
|
||||
|
||||
if (opts.isSet("colorspace"))
|
||||
cfg.colorSpace = ColorSpace::fromString(opts["colorspace"].toString());
|
||||
}
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
bool StreamKeyValueParser::parseRole(StreamRole *role,
|
||||
const KeyValueParser::Options &options)
|
||||
{
|
||||
if (!options.isSet("role"))
|
||||
return false;
|
||||
|
||||
std::string name = options["role"].toString();
|
||||
|
||||
if (name == "viewfinder") {
|
||||
*role = StreamRole::Viewfinder;
|
||||
return true;
|
||||
} else if (name == "video") {
|
||||
*role = StreamRole::VideoRecording;
|
||||
return true;
|
||||
} else if (name == "still") {
|
||||
*role = StreamRole::StillCapture;
|
||||
return true;
|
||||
} else if (name == "raw") {
|
||||
*role = StreamRole::Raw;
|
||||
return true;
|
||||
}
|
||||
|
||||
return false;
|
||||
}
|
||||
@@ -0,0 +1,28 @@
|
||||
/* SPDX-License-Identifier: GPL-2.0-or-later */
|
||||
/*
|
||||
* Copyright (C) 2020, Raspberry Pi Ltd
|
||||
*
|
||||
* stream_options.h - Helper to parse options for streams
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include <libcamera/camera.h>
|
||||
|
||||
#include "options.h"
|
||||
|
||||
class StreamKeyValueParser : public KeyValueParser
|
||||
{
|
||||
public:
|
||||
StreamKeyValueParser();
|
||||
|
||||
KeyValueParser::Options parse(const char *arguments) override;
|
||||
|
||||
static libcamera::StreamRoles roles(const OptionValue &values);
|
||||
static int updateConfiguration(libcamera::CameraConfiguration *config,
|
||||
const OptionValue &values);
|
||||
|
||||
private:
|
||||
static bool parseRole(libcamera::StreamRole *role,
|
||||
const KeyValueParser::Options &options);
|
||||
};
|
||||
Reference in New Issue
Block a user