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 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214
|
#include <stdbool.h>
#include <tiffio.h>
#include "greatest.h"
#include "libdng.h"
static enum greatest_test_res
check_str_tag(TIFF *im, uint32_t tag, const char *name, const char *expected)
{
char *temp;
if (TIFFGetField(im, tag, &temp) != 1) {
FAILm(name);
}
ASSERT_STR_EQm(name, expected, temp);
PASS();
}
static enum greatest_test_res
check_int_tag(TIFF *im, uint32_t tag, const char *name, int expected)
{
uint32_t temp = 0;
if (TIFFGetField(im, tag, &temp) != 1) {
FAILm(name);
}
ASSERT_EQ_FMTm(name, expected, temp, "%d");
PASS();
}
static enum greatest_test_res
check_float_tag(TIFF *im, uint32_t tag, const char *name, float expected)
{
float temp;
if (TIFFGetField(im, tag, &temp) != 1) {
FAILm(name);
}
ASSERT_IN_RANGEm(name, expected, temp, 0.0000001f);
PASS();
}
TEST generate_simple_dng(void)
{
libdng_info info = {0};
libdng_new(&info);
ASSERT_EQm("Set mode", 1, libdng_set_mode_from_name(&info, "RGGB"));
ASSERT_EQm("Set make", 1, libdng_set_make_model(&info, "Make", "Model"));
ASSERT_EQm("Set software", 1, libdng_set_software(&info, "Software"));
ASSERT_EQm("Set orientation", 1, libdng_set_orientation(&info, 4));
ASSERT_EQm("Set exposuretime", 1, libdng_set_exposure_time(&info, 0.04f));
ASSERT_EQm("Set fnumber", 1, libdng_set_fnumber(&info, 2.8f));
ASSERT_EQm("Set fnumber", 1, libdng_set_focal_length(&info, 50.0f, 1.5f));
ASSERT_EQm("Set framerate", 1, libdng_set_frame_rate(&info, 30.0f));
ASSERT_EQm("Set distortion", 1, libdng_set_distortion(&info, 1.0f, 2.0f, 3.0f));
ASSERT_EQm("Set vignette", 1, libdng_set_vignette(&info, 1.0f, -1.0f, 6.0f));
ASSERT_EQm("Set color matrix 1", 1, libdng_set_color_matrix_1(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
ASSERT_EQm("Set color matrix 2", 1, libdng_set_color_matrix_2(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
ASSERT_EQm("Set forward matrix 1", 1, libdng_set_forward_matrix_1(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
ASSERT_EQm("Set forward matrix 2", 1, libdng_set_forward_matrix_2(&info, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f, 1.0f));
uint8_t *data = malloc(1280 * 720);
ASSERT_EQm("Write DNG", 1, libdng_write(&info, "test.dng", 1280, 720, data, 1280 * 720));
free(data);
libdng_free(&info);
// Use LibTIFF directly to verify results
TIFF *im = TIFFOpen("test.dng", "r");
if (im == NULL) {
FAILm("Could not open result");
}
toff_t exif_offset;
if (TIFFGetField(im, TIFFTAG_EXIFIFD, &exif_offset) != 1) {
FAILm("Could not find EXIF data");
}
// Check IFD0 with most metadata and the thumbnail image
CHECK_CALL(check_int_tag(im, TIFFTAG_ORIENTATION, "ORIENTATION", 4));
CHECK_CALL(check_int_tag(im, TIFFTAG_BITSPERSAMPLE, "THUMB_BPS", 8));
CHECK_CALL(check_int_tag(im, TIFFTAG_SAMPLESPERPIXEL, "THUMB_CHANNELS", 3));
CHECK_CALL(check_int_tag(im, TIFFTAG_PHOTOMETRIC, "THUMB_PHOTOMETRIC", PHOTOMETRIC_RGB));
CHECK_CALL(check_str_tag(im, TIFFTAG_MAKE, "MAKE", "Make"));
CHECK_CALL(check_str_tag(im, TIFFTAG_MODEL, "MODEL", "Model"));
CHECK_CALL(check_str_tag(im, TIFFTAG_UNIQUECAMERAMODEL, "UCM", "Make Model"));
CHECK_CALL(check_str_tag(im, TIFFTAG_SOFTWARE, "SOFTWARE", "Software"));
CHECK_CALL(check_int_tag(im, TIFFTAG_ORIENTATION, "ORIENTATION", 4));
// Check the custom tags
#define MPTAG_VERSION 49982
#define MPTAG_DISTORTION 49983
#define MPTAG_VIGNETTE 49984
unsigned short count;
float *fvalues;
uint8_t *u8values;
if (TIFFGetField(im, MPTAG_VERSION, &count, &u8values) != 1) {
FAILm("Could not read MPTAG_VERSION");
}
ASSERT_EQ_FMTm("MPTAG_VERSION count", 4, count, "%d");
ASSERT_EQ_FMTm("MPTAG_VERSION[0]", 1, u8values[0], "%d");
ASSERT_EQ_FMTm("MPTAG_VERSION[1]", 0, u8values[1], "%d");
ASSERT_EQ_FMTm("MPTAG_VERSION[2]", 0, u8values[2], "%d");
ASSERT_EQ_FMTm("MPTAG_VERSION[3]", 0, u8values[3], "%d");
if (TIFFGetField(im, MPTAG_DISTORTION, &fvalues) != 1) {
FAILm("Could not read MPTAG_DISTORTION");
}
ASSERT_IN_RANGEm("MPTAG_DISTORTION[0]", 1.0f, fvalues[0], 0.0000001f);
ASSERT_IN_RANGEm("MPTAG_DISTORTION[1]", 2.0f, fvalues[1], 0.0000001f);
ASSERT_IN_RANGEm("MPTAG_DISTORTION[2]", 3.0f, fvalues[2], 0.0000001f);
if (TIFFGetField(im, MPTAG_VIGNETTE, &fvalues) != 1) {
FAILm("Could not read MPTAG_VIGNETTE");
}
ASSERT_IN_RANGEm("MPTAG_VIGNETTE[0]", 1.0f, fvalues[0], 0.0000001f);
ASSERT_IN_RANGEm("MPTAG_VIGNETTE[1]", -1.0f, fvalues[1], 0.0000001f);
ASSERT_IN_RANGEm("MPTAG_VIGNETTE[2]", 6.0f, fvalues[2], 0.0000001f);
// Switch to IFD1 which has the raw data
int subifd_count = 0;
void *ptr;
toff_t subifd_offsets[2];
TIFFGetField(im, TIFFTAG_SUBIFD, &subifd_count, &ptr);
memcpy(subifd_offsets, ptr, subifd_count * sizeof(subifd_offsets[0]));
TIFFSetSubDirectory(im, subifd_offsets[0]);
// Check IFD1 metadata
CHECK_CALL(check_int_tag(im, TIFFTAG_IMAGEWIDTH, "RAW_WIDTH", 1280));
CHECK_CALL(check_int_tag(im, TIFFTAG_IMAGELENGTH, "RAW_HEIGHT", 720));
CHECK_CALL(check_int_tag(im, TIFFTAG_BITSPERSAMPLE, "RAW_BPS", 8));
CHECK_CALL(check_int_tag(im, TIFFTAG_SAMPLESPERPIXEL, "RAW_CHANNELS", 1));
CHECK_CALL(check_int_tag(im, TIFFTAG_PHOTOMETRIC, "RAW_PHOTOMETRIC", PHOTOMETRIC_CFA));
// Switch to the EXIF block with the generic picture metadata
TIFFReadEXIFDirectory(im, exif_offset);
CHECK_CALL(check_float_tag(im, EXIFTAG_EXPOSURETIME, "EXPOSURETIME", 0.04f));
CHECK_CALL(check_float_tag(im, EXIFTAG_FNUMBER, "FNUMBER", 2.8f));
CHECK_CALL(check_float_tag(im, EXIFTAG_FOCALLENGTH, "FOCALLENGTH", 50.0f));
CHECK_CALL(check_int_tag(im, EXIFTAG_FOCALLENGTHIN35MMFILM, "FOCALLENGTHIN35MMFILM", 75));
PASS();
}
TEST read_dng(void)
{
// Generate test timestamp
struct tm testtime;
strptime("2024-08-27 01:02:03", "%Y-%m-%d %H:%M:%S", &testtime);
// Generate a file to read
libdng_info info = {0};
libdng_new(&info);
libdng_set_datetime(&info, testtime);
libdng_set_mode_from_name(&info, "RGGB");
libdng_set_make_model(&info, "Make", "Model");
libdng_set_software(&info, "Software");
libdng_set_orientation(&info, LIBDNG_ORIENTATION_RIGHTBOT);
libdng_set_exposure_time(&info, 1.4f);
libdng_set_exposure_program(&info, LIBDNG_EXPOSUREPROGRAM_PORTRAIT);
libdng_set_fnumber(&info, 1.8f);
libdng_set_focal_length(&info, 75.0f, 1.6f);
uint8_t *data = malloc(1280 * 720);
libdng_write(&info, "test.dng", 1280, 720, data, 1280 * 720);
free(data);
libdng_free(&info);
// Read the file again
libdng_info dng = {0};
libdng_read(&dng, "test.dng");
ASSERT_EQm("Bit depth", 8, dng.bit_depth);
ASSERT_EQm("CFFA[0]", 0, dng.cfapattern[0]);
ASSERT_EQm("CFFA[1]", 1, dng.cfapattern[1]);
ASSERT_EQm("CFFA[2]", 1, dng.cfapattern[2]);
ASSERT_EQm("CFFA[3]", 2, dng.cfapattern[3]);
ASSERT_STR_EQm("Make", "Make", dng.camera_make);
ASSERT_STR_EQm("Model", "Model", dng.camera_model);
ASSERT_STR_EQm("Software", "Software", dng.software);
ASSERT_EQm("Orientation", LIBDNG_ORIENTATION_RIGHTBOT, dng.orientation);
ASSERT_IN_RANGEm("Exposure time", 1.4f, dng.exposure_time, 0.0000001f);
ASSERT_EQm("Exposure program", LIBDNG_EXPOSUREPROGRAM_PORTRAIT, dng.exposure_program);
ASSERT_IN_RANGEm("FNumber", 1.8f, dng.fnumber, 0.0000001f);
ASSERT_IN_RANGEm("Focal length", 75.0f, dng.focal_length, 0.0000001f);
ASSERT_IN_RANGEm("Crop factor", 1.6f, dng.crop_factor, 0.0000001f);
char orig[sizeof "2011-10-08T07:07:09Z"];
char parsed[sizeof "2011-10-08T07:07:09Z"];
strftime(orig, sizeof orig, "%FT%TZ", &testtime);
strftime(parsed, sizeof parsed, "%FT%TZ", &dng.datetime);
ASSERT_STR_EQm("DateTime", orig, parsed);
ASSERT_EQ_FMTm("Year", testtime.tm_year, dng.datetime.tm_year, "%d");
ASSERT_EQ_FMTm("Month", testtime.tm_mon, dng.datetime.tm_mon, "%d");
ASSERT_EQ_FMTm("Day", testtime.tm_mday, dng.datetime.tm_mday, "%d");
ASSERT_EQ_FMTm("Hour", testtime.tm_hour, dng.datetime.tm_hour, "%d");
ASSERT_EQ_FMTm("Minute", testtime.tm_min, dng.datetime.tm_min, "%d");
ASSERT_EQ_FMTm("Second", testtime.tm_sec, dng.datetime.tm_sec, "%d");
PASS();
}
SUITE (test_suite)
{
RUN_TEST(generate_simple_dng);
RUN_TEST(read_dng);
}
GREATEST_MAIN_DEFS();
int
main(int argc, char **argv)
{
GREATEST_MAIN_BEGIN();
libdng_init();
RUN_SUITE(test_suite);
GREATEST_MAIN_END();
}
|