Files
Megapixels/src/dcp.c
2023-12-02 12:30:05 +01:00

86 lines
2.9 KiB
C

#include "dcp.h"
#include <libdng.h>
#include <stdbool.h>
#include <stdio.h>
#include <string.h>
#include <unistd.h>
bool
find_calibration_by_model(char *conffile, char *model, const char *sensor)
{
// Check config/%model,%sensor.dcp in the current working directory
sprintf(conffile, "config/%s,%s.dcp", model, sensor);
if (access(conffile, F_OK) != -1) {
printf("Found calibration file at %s\n", conffile);
return true;
}
// Check user overridden /etc/megapixels/config/%model,%sensor.dcp
sprintf(conffile, "%s/megapixels/config/%s,%s.dcp", "/etc", model, sensor);
if (access(conffile, F_OK) != -1) {
printf("Found calibration file at %s\n", conffile);
return true;
}
// Check packaged /usr/share/megapixels/config/%model,%sensor.ini
sprintf(conffile,
"%s/megapixels/config/%s,%s.dcp",
"/usr/share",
model,
sensor);
if (access(conffile, F_OK) != -1) {
printf("Found calibration file at %s\n", conffile);
return true;
}
printf("No calibration found for %s,%s\n", model, sensor);
return false;
}
bool
find_calibration(char *conffile, const char *sensor)
{
char model[512];
FILE *fp;
if (access("/proc/device-tree/compatible", F_OK) == -1) {
return false;
}
fp = fopen("/proc/device-tree/compatible", "r");
char *modelptr = model;
while (1) {
int c = fgetc(fp);
if (c == EOF) {
*(modelptr) = '\0';
return find_calibration_by_model(conffile, model, sensor);
}
*(modelptr++) = (char)c;
if (c == 0) {
bool res =
find_calibration_by_model(conffile, model, sensor);
if (res) {
return true;
}
modelptr = model;
}
}
}
struct MPCameraCalibration
parse_calibration_file(const char *path)
{
struct MPCameraCalibration result;
libdng_info temp = { 0 };
libdng_new(&temp);
libdng_load_calibration_file(&temp, path);
memcpy(result.color_matrix_1, temp.color_matrix_1, 9 * sizeof(float));
memcpy(result.color_matrix_2, temp.color_matrix_2, 9 * sizeof(float));
memcpy(result.forward_matrix_1, temp.forward_matrix_1, 9 * sizeof(float));
memcpy(result.forward_matrix_2, temp.forward_matrix_2, 9 * sizeof(float));
result.tone_curve_length = temp.tone_curve_length;
memcpy(result.tone_curve,
temp.tone_curve,
temp.tone_curve_length * sizeof(float));
return result;
}