raw.c
/*
* Copyright (C) 2013 Magic Lantern Team
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This program is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
* GNU General Public License for more details.
*
* You should have received a copy of the GNU General Public License
* along with this program; if not, write to the
* Free Software Foundation, Inc.,
* 51 Franklin Street, Fifth Floor,
* Boston, MA 02110-1301, USA.
*/
#include "dryos.h"
#include "raw.h"
#include "property.h"
#include "math.h"
#include "bmp.h"
#include "lens.h"
#include "module.h"
#include "menu.h"
#include "edmac-memcpy.h"
#include "imgconv.h"
#undef RAW_DEBUG /* define it to help with porting */
#undef RAW_DEBUG_DUMP /* if you want to save the raw image buffer and the DNG from here */
#undef RAW_DEBUG_BLACK /* for checking black level calibration */
/* see also RAW_ZEBRA_TEST and RAW_SPOTMETER_TEST in zebra.c */
#ifdef RAW_DEBUG
#define dbg_printf(fmt,...) { console_printf(fmt, ## __VA_ARGS__); }
#else
#define dbg_printf(fmt,...) {}
#endif
static struct semaphore * raw_sem = 0;
static int dirty = 0;
/* dual ISO interface */
static int (*dual_iso_get_recovery_iso)() = MODULE_FUNCTION(dual_iso_get_recovery_iso);
static int (*dual_iso_get_dr_improvement)() = MODULE_FUNCTION(dual_iso_get_dr_improvement);
/*********************** Camera-specific constants ****************************/
/**
* LiveView raw buffer address
* To find it, call("lv_save_raw") and look for an EDMAC channel that becomes active (Debug menu)
**/
#if defined(CONFIG_5D2) || defined(CONFIG_50D)
#define RAW_LV_EDMAC 0xC0F04508
#endif
#if defined(CONFIG_500D) || defined(CONFIG_550D) || defined(CONFIG_7D)
#define RAW_LV_EDMAC 0xC0F26008
#endif
#if defined(CONFIG_DIGIC_V) || defined(CONFIG_600D) || defined(CONFIG_60D)
/* probably all new cameras use this address */
#define RAW_LV_EDMAC 0xC0F26208
#endif
#ifdef CONFIG_EDMAC_RAW_SLURP
/* undefine so we don't use it by mistake */
#undef RAW_LV_EDMAC
/* hardcode Canon's raw buffer directly */
/* you can find it from lv_raw_dump, arg1 passed to dump_file:
*
* raw_buffer = get_raw_buffer()
* sprintf_maybe(filename, '%08lx.mm1', raw_buffer)
* ...
* dump_file(filename, raw_buffer, 7*something...)
*/
#ifdef CONFIG_60D
#define DEFAULT_RAW_BUFFER MEM(MEM(0x5028))
#endif
#ifdef CONFIG_600D
#define DEFAULT_RAW_BUFFER MEM(MEM(0x51FC))
#endif
#ifdef CONFIG_5D3
#define DEFAULT_RAW_BUFFER MEM(0x2600C + 0x2c) /* 113 */
//~ #define DEFAULT_RAW_BUFFER MEM(0x25f1c + 0x34) /* 123 */
#endif
#else
/* with Canon lv_save_raw, just read it from EDMAC */
#define DEFAULT_RAW_BUFFER shamem_read(RAW_LV_EDMAC)
#endif
/**
* Photo-mode raw buffer address
* To find it, lookup CCDWriteEDmacCompleteCBR in ASM code, and find the corresponding EDMAC channnel.
*
* example for 5D2:
* ffa3763c: e24f1f6f sub r1, pc, #444 ; @str:CCDWriteEDmacCompleteCBR
* ffa37640: e3a00002 mov r0, #2 ; so, it uses EDMAC channel #2 => RAW_PHOTO_EDMAC 0xc0f04208
* ffa37644: ebfdb453 bl @EDMAC_RegisterCompleteCBR
*
* On old cameras, it can be intercepted from SDSf3 state object, right after sdsMem1toRAWcompress.
* On new cameras, use the SSS state, sssCompleteMem1ToRaw.
*
* See state-object.c for intercepting code,
* and http://a1ex.bitbucket.org/ML/states/ for state diagrams.
*/
#if defined(CONFIG_5D2) || defined(CONFIG_50D) || defined(CONFIG_60D) || defined(CONFIG_550D) || defined(CONFIG_500D) || defined(CONFIG_600D) || defined(CONFIG_1100D) || defined(CONFIG_7D)
#define RAW_PHOTO_EDMAC 0xc0f04208
#endif
#if defined(CONFIG_5D3) || defined(CONFIG_700D) || defined(CONFIG_6D) || defined(CONFIG_EOSM) || defined(CONFIG_650D)
#define RAW_PHOTO_EDMAC 0xc0f04008
#endif
/**
* Raw type (optional)
* decompile lv_af_raw
* => (5D3) lv_set_raw_type(arg0 ? 4 : 7)
* => MEM(0x2D168) = a bunch of values, default 34, 18 with lv_af on, 14 with lv_af off.
* see also http://www.magiclantern.fm/forum/index.php?topic=5614.msg39696#msg39696
*/
#ifdef CONFIG_5D3
/**
* Renato [http://www.magiclantern.fm/forum/index.php?topic=5614.msg41070#msg41070]:
* "Best images in: 17, 35, 37, 39, 81, 83, 99"
* note: values are off by 1
*/
#define PREFERRED_RAW_TYPE 16
#define RAW_TYPE_ADDRESS 0x2D168
#endif
/**
* RAW_TYPE 78 (and others) have a frame-wide green pattern on them
* ACR can clear them but also kills some details and does not do
* a good job in general. TL;DR Use pink dot remover.
* http://www.magiclantern.fm/forum/index.php?topic=6658.0
*/
/*
#ifdef CONFIG_700D
#define PREFERRED_RAW_TYPE 78
#define RAW_TYPE_ADDRESS 0x351B8
#endif
#ifdef CONFIG_650D
#define PREFERRED_RAW_TYPE 78
#define RAW_TYPE_ADDRESS 0x350B4
#endif
*/
/**
* White level
* one size fits all: should work on most cameras and can't be wrong by more than 0.1 EV
*/
#define WHITE_LEVEL 15000
/** there may be exceptions */
#ifdef CONFIG_6D
#undef WHITE_LEVEL
#define WHITE_LEVEL 13000
#endif
/**
* Color matrix should be copied from DCRAW.
* It will also work with the values from some other camera, but colors may be a little off.
**/
#ifdef CONFIG_5D2
//~ { "Canon EOS 5D Mark II", 0, 0x3cf0,
//~ { 4716,603,-830,-7798,15474,2480,-1496,1937,6651 } },
#define CAM_COLORMATRIX1 \
4716, 10000, 603, 10000, -830, 10000, \
-7798, 10000, 15474, 10000, 2480, 10000, \
-1496, 10000, 1937, 10000, 6651, 10000
#endif
#ifdef CONFIG_5D3
//~ { "Canon EOS 5D Mark III", 0, 0x3c80,
//~ { 6722,-635,-963,-4287,12460,2028,-908,2162,5668 } },
#define CAM_COLORMATRIX1 \
6722, 10000, -635, 10000, -963, 10000, \
-4287, 10000, 12460, 10000, 2028, 10000, \
-908, 10000, 2162, 10000, 5668, 10000
#endif
#ifdef CONFIG_550D
//~ { "Canon EOS 550D", 0, 0x3dd7,
//~ { 6941,-1164,-857,-3825,11597,2534,-416,1540,6039 } },
#define CAM_COLORMATRIX1 \
6461, 10000, -1164, 10000, -857, 10000,\
-3825, 10000, 11597, 10000, 2534, 10000,\
-416, 10000, 1540, 10000, 6039, 10000
#endif
#ifdef CONFIG_6D
//~ { "Canon EOS 6D", 0, 0,
//~ { 7034,-804,-1014,-4420,12564,2058,-851,1994,5758 } },
#define CAM_COLORMATRIX1 \
7034, 10000, -804, 10000, -1014, 10000,\
-4420, 10000, 12564, 10000, 2058, 10000, \
-851, 10000, 1994, 10000, 5758, 10000
#endif
#ifdef CONFIG_500D
//~ { "Canon EOS 500D", 0, 0x3479,
//~ { 4763,712,-646,-6821,14399,2640,-1921,3276,6561 } },
#define CAM_COLORMATRIX1 \
4763, 10000, 712, 10000, -646, 10000, \
-6821, 10000, 14399, 10000, 2640, 10000, \
-1921, 10000, 3276, 10000, 6561, 10000
#endif
#ifdef CONFIG_600D
//~ { "Canon EOS 600D", 0, 0x3510,
//~ { 6461,-907,-882,-4300,12184,2378,-819,1944,5931 } },
#define CAM_COLORMATRIX1 \
6461, 10000, -907, 10000, -882, 10000,\
-4300, 10000, 12184, 10000, 2378, 10000, \
-819, 10000, 1944, 10000, 5931, 10000
#endif
#ifdef CONFIG_1100D
//~ { "Canon EOS 1100D", 0, 0x3510,
//~ { 6444,-904,-893,-4563,12308,2535,-903,2016,6728 } },
#define CAM_COLORMATRIX1 \
6444, 10000, -904, 10000, -893, 10000,\
-4563, 10000, 12308, 10000, 2535, 10000, \
-903, 10000, 2016, 10000, 6728, 10000
#endif
#ifdef CONFIG_60D
//~ { "Canon EOS 60D", 0, 0x2ff7,
//~ { 6719,-994,-925,-4408,12426,2211,-887,2129,6051 } },
#define CAM_COLORMATRIX1 \
6719, 10000, -994, 10000, -925, 10000,\
-4408, 10000, 12426, 10000, 2211, 10000, \
-887, 10000, 2129, 10000, 6051, 10000
#endif
#ifdef CONFIG_50D // these values are in ufraw-0.19.2
//~{ "Canon EOS 50D", 0, 0x3d93,
//~{ 4920,616,-593,-6493,13964,2784,-1774,3178,7005 } },
#define CAM_COLORMATRIX1 \
4920, 10000, 616, 10000, -593, 10000, \
-6493, 10000, 12964, 10000, 2784, 10000, \
-1774, 10000, 3178, 10000, 7005, 10000
#endif
#if defined(CONFIG_650D) || defined(CONFIG_EOSM) || defined(CONFIG_700D) || defined(CONFIG_100D) //Same sensor??. TODO: Check 700D/100D
//~ { "Canon EOS 650D", 0, 0x354d,
//~ { "Canon EOS M", 0, 0,
//~ { 6602,-841,-939,-4472,12458,2247,-975,2039,6148 } },
#define CAM_COLORMATRIX1 \
6602, 10000, -841, 10000, -939, 10000,\
-4472, 10000, 12458, 10000, 2247, 10000, \
-975, 10000, 2039, 10000, 6148, 10000
#endif
#ifdef CONFIG_7D
//~ { "Canon EOS 7D", 0, 0x3510,
//~ { 6844,-996,-856,-3876,11761,2396,-593,1772,6198 } },
#define CAM_COLORMATRIX1 \
6844, 10000, -996, 10000, -856, 10000,\
-3876, 10000, 11761, 10000, 2396, 10000, \
-593, 10000, 1772, 10000, 6198, 10000
#endif
struct raw_info raw_info = {
.api_version = 1,
.bits_per_pixel = 14,
.black_level = 1024,
.white_level = 13000,
.cfa_pattern = 0x02010100, // Red Green Green Blue
.calibration_illuminant1 = 1, // Daylight
.color_matrix1 = {CAM_COLORMATRIX1},// camera-specific, from dcraw.c
.dynamic_range = 1100, // not correct; use numbers from DxO instead
};
/**
* Dynamic range, from DxO
* e.g. http://www.dxomark.com/index.php/Cameras/Camera-Sensor-Database/Canon/EOS-5D-Mark-III
* Measurements | Dynamic range | Screen
* You can hover over the points to list the measured EV (thanks Audionut).
*
* This is only used in photo LiveView, where we can't compute it
*/
#ifdef CONFIG_5D3
static int dynamic_ranges[] = {1097, 1087, 1069, 1041, 994, 923, 830, 748, 648, 552, 464};
#endif
#ifdef CONFIG_5D2
static int dynamic_ranges[] = {1116, 1112, 1092, 1066, 1005, 909, 813, 711, 567};
#endif
#ifdef CONFIG_6D
static int dynamic_ranges[] = {1143, 1139, 1122, 1087, 1044, 976, 894, 797, 683, 624, 505};
#endif
#ifdef CONFIG_500D
static int dynamic_ranges[] = {1104, 1094, 1066, 1007, 933, 848, 737, 625};
#endif
#ifdef CONFIG_550D
//static int dynamic_ranges[] = {1157, 1154, 1121, 1070, 979, 906, 805, 707}; I took the values Greg recommended
static int dynamic_ranges[] = {1095, 1092, 1059, 1008, 917, 844, 744, 645};
#endif
#ifdef CONFIG_600D
static int dynamic_ranges[] = {1146, 1139, 1116, 1061, 980, 898, 806, 728};
#endif
#ifdef CONFIG_1100D
static int dynamic_ranges[] = {1099, 1098, 1082, 1025, 965, 877, 784}; // No ISO 12800 available
#endif
#ifdef CONFIG_650D
static int dynamic_ranges[] = {1062, 1047, 1021, 963, 888, 804, 695, 623, 548};
#endif
#ifdef CONFIG_700D
static int dynamic_ranges[] = {1058, 1053, 1032, 967, 893, 807, 704, 618, 510};
#endif
#ifdef CONFIG_60D
static int dynamic_ranges[] = {1091, 1072, 1055, 999, 910, 824, 736, 662};
#endif
#ifdef CONFIG_50D
static int dynamic_ranges[] = {1100, 1094, 1060, 1005, 919, 826, 726, 633};
#endif
#ifdef CONFIG_EOSM
static int dynamic_ranges[] = {1060, 1063, 1037, 982, 901, 831, 718, 622, 536};
#endif
#ifdef CONFIG_7D
static int dynamic_ranges[] = {1112, 1108, 1076, 1010, 902, 826, 709, 622};
#endif
static int autodetect_black_level(int* black_mean, int* black_stdev);
static int compute_dynamic_range(int black_mean, int black_stdev, int white_level);
static int autodetect_white_level(int initial_guess);
#ifdef CONFIG_RAW_LIVEVIEW
/* returns 1 on success */
static int raw_lv_get_resolution(int* width, int* height)
{
#ifdef CONFIG_EDMAC_RAW_SLURP
/* params useful for hardcoding buffer sizes, according to video mode */
int mv = is_movie_mode();
int mv720 = mv && video_mode_resolution == 1;
int mv1080 = mv && video_mode_resolution == 0;
int mv640 = mv && video_mode_resolution == 2;
int mv1080crop = mv && video_mode_resolution == 0 && video_mode_crop;
int mv640crop = mv && video_mode_resolution == 2 && video_mode_crop;
/* note: 6D reports 129 = 0x81 for zoom x1, and it behaves just like plain (unzoomed) LiveView) */
int zoom = (lv_dispsize & 0xF) > 1;
/* silence warnings; not all cameras have all these modes */
(void)mv640; (void)mv720; (void)mv1080; (void)mv640; (void)mv1080crop; (void)mv640crop; (void)zoom;
#ifdef CONFIG_5D3
/* don't know how to get the resolution without relying on Canon's lv_save_raw */
*width = zoom ? 3744 : mv720 ? 2080 : 2080;
*height = zoom ? 1380 : mv720 ? 692 : 1318; /* height must be exact! copy it from Debug->EDMAC */
return 1;
#endif
#ifdef CONFIG_60D
*width = zoom ? 2520 : mv640crop ? 920 : mv720 || mv640 ? 1888 : 1888;
*height = zoom ? 1106 : mv640crop ? 624 : mv720 || mv640 ? 720 : 1182;
return 1;
#endif
#ifdef CONFIG_600D
*width = zoom ? 2520 : mv1080crop ? 1952 : mv720 ? 1888 : 1888;
*height = zoom ? 1106 : mv1080crop ? 1048 : mv720 ? 720 : 1182;
return 1;
#endif
/* unknown camera? */
return 0;
#else
/* autodetect raw size from EDMAC */
uint32_t lv_raw_height = shamem_read(RAW_LV_EDMAC+4);
uint32_t lv_raw_size = shamem_read(RAW_LV_EDMAC+8);
if (!lv_raw_size) return 0;
int pitch = lv_raw_size & 0xFFFF;
*width = pitch * 8 / 14;
/* 5D2 uses lv_raw_size >> 16, 5D3 uses lv_raw_height, so this hopefully covers both cases */
*height = MAX((lv_raw_height & 0xFFFF) + 1, ((lv_raw_size >> 16) & 0xFFFF) + 1);
return 1;
#endif
}
#endif
static int raw_update_params_work()
{
//~ static int k = 0;
//~ bmp_printf(FONT_MED, 250, 100, "raw update %d called from %s ", k++, get_task_name_from_id(get_current_task()));
#ifdef RAW_DEBUG
console_show();
#endif
int width = 0;
int height = 0;
int skip_left = 0;
int skip_right = 0;
int skip_top = 0;
int skip_bottom = 0;
/* params useful for hardcoding buffer sizes, according to video mode */
int mv = is_movie_mode();
int mv720 = mv && video_mode_resolution == 1;
int mv1080 = mv && video_mode_resolution == 0;
int mv640 = mv && video_mode_resolution == 2;
int mv1080crop = mv && video_mode_resolution == 0 && video_mode_crop;
int mv640crop = mv && video_mode_resolution == 2 && video_mode_crop;
int zoom = lv_dispsize > 1;
/* silence warnings; not all cameras have all these modes */
(void)mv640; (void)mv720; (void)mv1080; (void)mv640; (void)mv1080crop; (void)mv640crop; (void)zoom;
if (lv)
{
#ifdef CONFIG_RAW_LIVEVIEW
if (!raw_lv_is_enabled())
{
dbg_printf("LV raw disabled\n");
return 0;
}
#ifdef CONFIG_RAW_DISABLE_IN_10X_ZOOM
if (lv_dispsize == 10)
{
dbg_printf("LV raw has trouble with 10x zoom\n");
return 0;
}
#endif
raw_info.buffer = (void*) DEFAULT_RAW_BUFFER;
if (!raw_info.buffer)
{
dbg_printf("LV raw buffer null\n");
return 0;
}
if (!raw_lv_get_resolution(&width, &height))
{
dbg_printf("LV RAW size error\n");
return 0;
}
/* the raw edmac might be used by something else, and wrong numbers may be still there */
/* e.g. 5D2: 1244x1, obviously wrong */
if (width < 320 || height < 160)
{
dbg_printf("LV RAW size too small\n");
return 0;
}
/**
* The RAW file has unused areas, usually black; we need to skip them.
*
* To find the skip values, start with 0,
* load the RAW in your favorite photo editor (e.g. ufraw+gimp),
* then find the usable area, read the coords and plug the skip values here.
*
* Try to use even offsets only, otherwise the colors will be screwed up.
*/
#ifdef CONFIG_5D2
skip_top = zoom ? 52 : 18;
skip_left = 160;
#endif
#ifdef CONFIG_5D3
skip_top = zoom ? 60 : mv720 ? 20 : 28;
skip_left = 146;
skip_right = 2;
#endif
#ifdef CONFIG_6D
/* same skip offsets in 1080p and 720p; top/left bar is the same in x5 zoom as well */
skip_top = 28;
skip_left = 80;
skip_right = zoom ? 0 : 10;
#endif
#ifdef CONFIG_500D
#warning FIXME: are these values correct for 1080p or 720p? (which of them?)
skip_top = 24;
skip_left = zoom ? 64 : 74;
#endif
#if defined(CONFIG_550D) || defined(CONFIG_600D)
#warning FIXME: are these values correct for 720p and crop modes?
skip_top = 26;
skip_left = zoom ? 0 : 152;
skip_right = zoom ? 0 : 2;
#endif
#ifdef CONFIG_60D
#warning FIXME: are these values correct for 720p and crop modes?
skip_top = 26;
skip_left = zoom ? 0 : 152;
skip_right = zoom ? 0 : 2;
#endif
#ifdef CONFIG_50D
skip_top = 26;
skip_left = zoom ? 64: 74;
skip_right = 0;
skip_bottom = 0;
#endif
#if defined(CONFIG_650D) || defined(CONFIG_EOSM) || defined(CONFIG_100D)
#warning FIXME: are these values correct for 720p and crop modes?
skip_top = 28;
skip_left = 74;
skip_right = 0;
skip_bottom = 6;
#endif
#ifdef CONFIG_700D
skip_top = 28;
skip_left = 72;
skip_right = 0;
skip_bottom = zoom ? 0 : mv1080crop ? 0 : 4;
#endif
#ifdef CONFIG_7D
#warning FIXME: are these values correct for 720p and crop modes?
skip_top = 26;
skip_left = zoom ? 0 : 256;
#endif
dbg_printf("LV raw buffer: %x (%dx%d)\n", raw_info.buffer, width, height);
dbg_printf("Skip left:%d right:%d top:%d bottom:%d\n", skip_left, skip_right, skip_top, skip_bottom);
#else
#warning RAW FEATURES WILL NOT WORK IN LIVEVIEW ON THIS BUILD
#endif
}
else if (QR_MODE) // image review after taking pics
{
#ifdef CONFIG_RAW_PHOTO
if (!can_use_raw_overlays_photo())
{
return 0;
}
raw_info.buffer = (void*) shamem_read(RAW_PHOTO_EDMAC);
if (!raw_info.buffer)
{
dbg_printf("Photo raw buffer null\n");
return 0;
}
/* autodetect image size from EDMAC */
width = shamem_read(RAW_PHOTO_EDMAC + 8) * 8 / 14; /* size B */
height = shamem_read(RAW_PHOTO_EDMAC + 4) + 1; /* size N */
/* in photo mode, raw buffer size is from ~12 Mpix (1100D) to ~24 Mpix (5D3) */
/* (this EDMAC may be reused for something else, usually smaller, or with a different size encoding - refuse to run if this happens) */
if ((width & 0xFFFFE000) || (height & 0xFFFFE000) || (width*height < 10e6) || (width*height > 30e6))
{
dbg_printf("Photo raw size error\n");
return 0;
}
/**
* The RAW file has unused areas, called "optical black" (OB); we need to skip them.
*
* To check the skip offsets, load raw_diag.mo (from the CMOS/ADTG ISO research thread),
* select the "OB zones" option, and adjust the skip offsets until the picture looks like this:
* https://dl.dropboxusercontent.com/u/4124919/bleeding-edge/iso50/ob/ob-zones-5d3-6400.png
*
* Use even offsets only, otherwise the colors will be screwed up.
*/
#ifdef CONFIG_5D2
skip_left = 160;
skip_top = 52;
#endif
#ifdef CONFIG_5D3
skip_left = 138; /* this gives a tight fit */
skip_right = 2;
skip_top = 80; /* matches dcraw */
#endif
#ifdef CONFIG_500D
skip_left = 62;
skip_top = 24;
/* skip one line */
raw_info.buffer += width * 14/8;
height--;
#endif
#if defined(CONFIG_550D) || defined(CONFIG_60D) || defined(CONFIG_600D)
skip_left = 142;
skip_top = 52;
#endif
#ifdef CONFIG_1100D
skip_top = 16;
skip_left = 62;
raw_info.buffer += width * 14/8;
height--;
#endif
#ifdef CONFIG_6D
skip_left = 72;
skip_right = 0;
skip_top = 52;
#endif
#if defined(CONFIG_50D)
skip_left = 64;
skip_top = 54;
#endif
#if defined(CONFIG_650D) || defined(CONFIG_EOSM) || defined(CONFIG_700D) || defined(CONFIG_100D)
skip_left = 72;
skip_top = 52;
#endif
#ifdef CONFIG_7D /* very similar to 5D2 */
skip_left = 158;
skip_top = 50;
#endif
dbg_printf("Photo raw buffer: %x (%dx%d)\n", raw_info.buffer, width, height);
dbg_printf("Skip left:%d right:%d top:%d bottom:%d\n", skip_left, skip_right, skip_top, skip_bottom);
#endif
}
else
{
dbg_printf("Neither LV nor QR\n");
return 0;
}
/*********************** Portable code ****************************************/
if (width != raw_info.width || height != raw_info.height)
{
/* raw dimensions changed? force a full update, including preview window */
dirty = 1;
}
#ifdef CONFIG_RAW_LIVEVIEW
/* zoom mode changed? refresh params */
{
static int prev_zoom = 0;
int zoom = lv_dispsize;
if (zoom != prev_zoom)
dirty = 1;
prev_zoom = zoom;
}
/* in zoom mode: yuv position changed? force a refresh */
if (lv_dispsize > 1)
{
static int prev_delta_x = 0;
static int prev_delta_y = 0;
int delta_x, delta_y;
focus_box_get_raw_crop_offset(&delta_x, &delta_y);
if (delta_x != prev_delta_x || delta_y != prev_delta_y)
dirty = 1;
prev_delta_x = delta_x;
prev_delta_y = delta_y;
}
#endif
/* black and white autodetection are time-consuming */
/* only refresh once per second or if dirty, but never while recording */
static int bw_aux = INT_MIN;
int recompute_black_and_white = NOT_RECORDING && (dirty || should_run_polling_action(1000, &bw_aux));
if (dirty)
{
raw_set_geometry(width, height, skip_left, skip_right, skip_top, skip_bottom);
dirty = 0;
}
if (!recompute_black_and_white)
{
/* keep the old values */
return 1;
}
int black_mean, black_stdev_x100;
raw_info.white_level = WHITE_LEVEL;
raw_info.black_level = autodetect_black_level(&black_mean, &black_stdev_x100);
if (!lv)
{
/* at ISO 160, 320 etc, the white level is decreased by -1/3 EV */
/* in LiveView, it doesn't change */
int iso = 0;
if (!iso) iso = lens_info.raw_iso;
if (!iso) iso = lens_info.raw_iso_auto;
static int last_iso = 0;
if (!iso) iso = last_iso;
last_iso = iso;
if (!iso) return 0;
int iso_rounded = COERCE((iso + 3) / 8 * 8, 72, 200);
float iso_digital = (iso - iso_rounded) / 8.0f;
if (iso_digital <= 0)
{
raw_info.white_level -= raw_info.black_level;
raw_info.white_level *= powf(2, iso_digital);
raw_info.white_level += raw_info.black_level;
}
raw_info.white_level = autodetect_white_level(raw_info.white_level);
raw_info.dynamic_range = compute_dynamic_range(black_mean, black_stdev_x100, raw_info.white_level);
}
#ifdef CONFIG_RAW_LIVEVIEW
else if (!is_movie_mode())
{
/* in photo mode, LV iso is not equal to photo ISO because of ExpSim *
* the digital ISO will not change the raw histogram
* but we want the histogram to mimic the CR2 one as close as possible
* so we do this by compensating the white level manually
* warning: this may exceed 16383!
*/
int shad_gain = shamem_read(0xc0f08030);
raw_info.white_level -= raw_info.black_level;
raw_info.white_level = raw_info.white_level * 3444 / shad_gain; /* 0.25 EV correction, so LiveView matches CR2 exposure */
raw_info.white_level += raw_info.black_level;
/* in photo LiveView, ISO is not the one selected from Canon menu,
* so the computed dynamic range has nothing to do with the one from CR2 pics
* => we will use the DxO values
*/
int iso = 0;
if (!iso) iso = lens_info.raw_iso;
if (!iso) iso = lens_info.raw_iso_auto;
static int last_iso = 0;
if (!iso) iso = last_iso;
last_iso = iso;
if (!iso) return 0;
int iso2 = dual_iso_get_recovery_iso();
if (iso2) iso = MIN(iso, iso2);
int dr_boost = dual_iso_get_dr_improvement();
raw_info.dynamic_range = get_dxo_dynamic_range(iso) + dr_boost;
dbg_printf("dynamic range: %d.%02d EV (iso=%d)\n", raw_info.dynamic_range/100, raw_info.dynamic_range%100, raw2iso(iso));
}
else /* movie mode, no tricks here */
{
raw_info.dynamic_range = compute_dynamic_range(black_mean, black_stdev_x100, raw_info.white_level);
}
#endif
dbg_printf("black=%d white=%d\n", raw_info.black_level, raw_info.white_level);
#ifdef RAW_DEBUG_DUMP
dbg_printf("saving raw buffer...\n");
dump_seg(raw_info.buffer, MAX(raw_info.frame_size, 1000000), "raw.buf");
dbg_printf("saving DNG...\n");
save_dng("raw.dng", &raw_info);
reverse_bytes_order(raw_info.buffer, raw_info.frame_size);
dbg_printf("done\n");
#endif
return 1;
}
int raw_update_params()
{
get_yuv422_vram(); /* refresh VRAM parameters */
int ans = 0;
take_semaphore(raw_sem, 0);
ans = raw_update_params_work();
give_semaphore(raw_sem);
return ans;
}
static int preview_rect_x;
static int preview_rect_y;
static int preview_rect_w;
static int preview_rect_h;
void raw_set_preview_rect(int x, int y, int w, int h)
{
preview_rect_x = x;
preview_rect_y = y;
preview_rect_w = w;
preview_rect_h = h;
/* preview area (usually active area) should match the area from os.x0/y0 to os.x_max/y_max */
/* note: this will call BMP_LOCK */
/* not exactly a good idea when we have already acquired raw_sem */
//~ get_yuv422_vram(); // update vram parameters
/* scaling factor: raw width should match os.x_ex, same for raw height and os.y_ex */
lv2raw.sx = 1024 * w / BM2LV_DX(os.x_ex);
lv2raw.sy = 1024 * h / BM2LV_DY(os.y_ex);
/* translation: raw top-left corner (x,y) should match (os.x0,os.y0) */
int x0_lv = BM2LV_X(os.x0);
int y0_lv = BM2LV_Y(os.y0);
lv2raw.tx = x - LV2RAW_DX(x0_lv);
lv2raw.ty = y - LV2RAW_DY(y0_lv);
}
void raw_set_geometry(int width, int height, int skip_left, int skip_right, int skip_top, int skip_bottom)
{
raw_info.width = width;
raw_info.height = height;
raw_info.pitch = raw_info.width * 14 / 8;
raw_info.frame_size = raw_info.height * raw_info.pitch;
raw_info.active_area.x1 = skip_left;
raw_info.active_area.y1 = skip_top;
raw_info.active_area.x2 = raw_info.width - skip_right;
raw_info.active_area.y2 = raw_info.height - skip_bottom;
raw_info.jpeg.x = 0;
raw_info.jpeg.y = 0;
raw_info.jpeg.width = raw_info.width - skip_left - skip_right;
raw_info.jpeg.height = raw_info.height - skip_top - skip_bottom;
dbg_printf("active area: x=%d..%d, y=%d..%d\n", raw_info.active_area.x1, raw_info.active_area.x2, raw_info.active_area.y1, raw_info.active_area.y2);
/* Canon does not render the entire active area - these numbers give pixel-perfect alignment on 5D3 */
int preview_skip_left = skip_left + 14;
int preview_skip_top = skip_top + 8;
int preview_width = raw_info.jpeg.width - 28;
int preview_height = raw_info.jpeg.height - 16;
#ifdef CONFIG_RAW_LIVEVIEW
if (lv_dispsize > 1)
{
int delta_x, delta_y;
if (focus_box_get_raw_crop_offset(&delta_x, &delta_y))
{
/* in 10x, the yuv area is twice as small than in 5x */
int zoom_corr = lv_dispsize == 10 ? 2 : 1;
/**
* |<-----------------raw_info.width--------------------------->|
* | |
* | raw_info.jpeg.width |
* | |<---------------------------------------------------------|
* | |
*->|-|<--- skip_left |
* | | | skip_top
* +-------preview_height |
* .-----:------------------------------------------------------. -----------------v-----------------------------
* | |```:``````````````````````````````````````````````````````| `````````````````^`` ^ ^
* | | : |<-preview_width->| | delta_y | | preview_skip_top
* | | _V_ _________________ | | | _____________v___
* | | | | | | v |
* | | | | | C_raw | ------+-- raw_info.height
* | | | | C_yuv | | ------+-- |
* | | | | | | ^ |
* | | _|_ |_________________| | | |
* | | ^ | v
* '------------------------------------------------------------' ----------------------------
* | |
* | | |---delta_x--->|
* | |
*->|----------|<-- preview_skip_left
* | |
*
*/
/* if the yuv window is on the left side, delta_x is > 0 */
preview_skip_left += (raw_info.jpeg.width - vram_hd.width / zoom_corr) / 2 - delta_x;
preview_skip_top += (raw_info.jpeg.height - vram_hd.height / zoom_corr) / 2 - delta_y;
preview_width = vram_hd.width / zoom_corr;
preview_height = vram_hd.height / zoom_corr;
}
}
#endif
raw_set_preview_rect(preview_skip_left, preview_skip_top, preview_width, preview_height);
dbg_printf("lv2raw sx:%d sy:%d tx:%d ty:%d\n", lv2raw.sx, lv2raw.sy, lv2raw.tx, lv2raw.ty);
dbg_printf("raw2lv test: (%d,%d) - (%d,%d)\n", RAW2LV_X(raw_info.active_area.x1), RAW2LV_Y(raw_info.active_area.y1), RAW2LV_X(raw_info.active_area.x2), RAW2LV_Y(raw_info.active_area.y2));
dbg_printf(" should be: (%d,%d) - (%d,%d)\n", BM2LV_X(os.x0), BM2LV_Y(os.y0), BM2LV_X(os.x_max), BM2LV_Y(os.y_max));
dbg_printf("raw2bm test: (%d,%d) - (%d,%d)\n", RAW2BM_X(raw_info.active_area.x1), RAW2BM_Y(raw_info.active_area.y1), RAW2BM_X(raw_info.active_area.x2), RAW2BM_Y(raw_info.active_area.y2));
dbg_printf(" should be: (%d,%d) - (%d,%d)\n", os.x0, os.y0, os.x_max, os.y_max);
dbg_printf("bm2raw test: (%d,%d) - (%d,%d)\n", BM2RAW_X(os.x0), BM2RAW_Y(os.y0), BM2RAW_X(os.x_max), BM2RAW_Y(os.y_max));
dbg_printf(" should be: (%d,%d) - (%d,%d)\n", raw_info.active_area.x1, raw_info.active_area.y1, raw_info.active_area.x2, raw_info.active_area.y2);
}
int FAST raw_red_pixel(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2;
int i = ((y * raw_info.width + x) / 8);
return buf[i].a;
}
int FAST raw_green_pixel(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2;
int i = ((y * raw_info.width + x) / 8);
return buf[i].h;
}
int FAST raw_blue_pixel(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2 - 1;
int i = ((y * raw_info.width + x) / 8);
return buf[i].h;
}
int FAST raw_red_pixel_dark(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2;
int i = ((y * raw_info.width + x) / 8);
return MIN(buf[i].a, buf[i - raw_info.width*2/8].a);
}
int FAST raw_green_pixel_dark(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2;
int i = ((y * raw_info.width + x) / 8);
return MIN(buf[i].h, buf[i - raw_info.width*2/8].h);
}
int FAST raw_blue_pixel_dark(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2 - 1;
int i = ((y * raw_info.width + x) / 8);
return MIN(buf[i].h, buf[i - raw_info.width*2/8].h);
}
int FAST raw_red_pixel_bright(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2;
int i = ((y * raw_info.width + x) / 8);
return MAX(buf[i].a, buf[i - raw_info.width*2/8].a);
}
int FAST raw_green_pixel_bright(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2;
int i = ((y * raw_info.width + x) / 8);
return MAX(buf[i].h, buf[i - raw_info.width*2/8].h);
}
int FAST raw_blue_pixel_bright(int x, int y)
{
struct raw_pixblock * buf = (void*)raw_info.buffer;
y = (y/2) * 2 - 1;
int i = ((y * raw_info.width + x) / 8);
return MAX(buf[i].h, buf[i - raw_info.width*2/8].h);
}
int FAST raw_get_pixel(int x, int y) {
struct raw_pixblock * p = (void*)raw_info.buffer + y * raw_info.pitch + (x/8)*14;
switch (x%8) {
case 0: return p->a;
case 1: return p->b_lo | (p->b_hi << 12);
case 2: return p->c_lo | (p->c_hi << 10);
case 3: return p->d_lo | (p->d_hi << 8);
case 4: return p->e_lo | (p->e_hi << 6);
case 5: return p->f_lo | (p->f_hi << 4);
case 6: return p->g_lo | (p->g_hi << 2);
case 7: return p->h;
}
return p->a;
}
int FAST raw_get_pixel_ex(void* raw_buffer, int x, int y) {
struct raw_pixblock * p = (void*)raw_buffer + y * raw_info.pitch + (x/8)*14;
switch (x%8) {
case 0: return p->a;
case 1: return p->b_lo | (p->b_hi << 12);
case 2: return p->c_lo | (p->c_hi << 10);
case 3: return p->d_lo | (p->d_hi << 8);
case 4: return p->e_lo | (p->e_hi << 6);
case 5: return p->f_lo | (p->f_hi << 4);
case 6: return p->g_lo | (p->g_hi << 2);
case 7: return p->h;
}
return p->a;
}
void FAST raw_set_pixel(int x, int y, int value)
{
struct raw_pixblock * p = (void*)raw_info.buffer + y * raw_info.pitch + (x/8)*14;
switch (x%8) {
case 0: p->a = value; break;
case 1: p->b_lo = value; p->b_hi = value >> 12; break;
case 2: p->c_lo = value; p->c_hi = value >> 10; break;
case 3: p->d_lo = value; p->d_hi = value >> 8; break;
case 4: p->e_lo = value; p->e_hi = value >> 6; break;
case 5: p->f_lo = value; p->f_hi = value >> 4; break;
case 6: p->g_lo = value; p->g_hi = value >> 2; break;
case 7: p->h = value; break;
}
}
int FAST raw_get_gray_pixel(int x, int y, int gray_projection)
{
int (*red_pixel)(int x, int y) = raw_red_pixel;
int (*green_pixel)(int x, int y) = raw_green_pixel;
int (*blue_pixel)(int x, int y) = raw_blue_pixel;
switch (gray_projection & GRAY_PROJECTION_BRIGHT_DARK_MASK)
{
case GRAY_PROJECTION_DARK_ONLY:
red_pixel = raw_red_pixel_dark;
green_pixel = raw_green_pixel_dark;
blue_pixel = raw_blue_pixel_dark;
break;
case GRAY_PROJECTION_BRIGHT_ONLY:
red_pixel = raw_red_pixel_bright;
green_pixel = raw_green_pixel_bright;
blue_pixel = raw_blue_pixel_bright;
break;
default:
break;
}
switch (gray_projection & 0xFF)
{
case GRAY_PROJECTION_RED:
return red_pixel(x, y);
case GRAY_PROJECTION_GREEN:
return green_pixel(x, y);
case GRAY_PROJECTION_BLUE:
return blue_pixel(x, y);
case GRAY_PROJECTION_AVERAGE_RGB:
return (red_pixel(x, y) + green_pixel(x, y) + blue_pixel(x, y)) / 3;
case GRAY_PROJECTION_MAX_RGB:
return MAX(MAX(red_pixel(x, y), green_pixel(x, y)), blue_pixel(x, y));
case GRAY_PROJECTION_MAX_RB:
return MAX(red_pixel(x, y), blue_pixel(x, y));
case GRAY_PROJECTION_MEDIAN_RGB:
{
int r = red_pixel(x, y);
int g = green_pixel(x, y);
int b = blue_pixel(x, y);
int M = MAX(MAX(r,g),b);
int m = MIN(MIN(r,g),b);
if (r >= m && r <= M) return r;
if (g >= m && g <= M) return g;
return b;
}
default:
return -1;
}
}
/* input: 0 - 16384 (valid range: from black level to white level) */
/* output: -14 ... 0 */
float FAST raw_to_ev(int raw)
{
int raw_max = raw_info.white_level - raw_info.black_level;
if (unlikely(raw_info.white_level > 16383) && unlikely(raw > 10000))
{
/**
* Hack for photo mode LV raw overlays (histogram & friends)
* to show correct overexposure warnings when ExpSim is done with -1/3 EV digital ISO.
*
* Canon implements ExpSim by varying iso/shutter/aperture in full stops, and digital ISO for 1/3 stops.
* Digital ISO does not affect the raw histogram, so they add -1/3, 0 or +1/3 EV when developing the raw for LV display
* We did the same adjustment by adjusting the white level in raw_update_params.
* But when the correction is -1/3 EV, the white level is greater than 16383,
* so the overexposure indicators will read a negative EV instead of 0 (they will no longer indicate overexposure).
*
* With this hack, we are pushing raw values greater than 10000 towards 0 EV (overexposed) level,
* thus keeping the correct horizontal position of the histogram at midtones (raw - 1/3 EV)
* and getting correct overexposure indicators for highlights (0 EV).
*
* Math:
* at raw=10000 we keep the original white level,
* at raw=15000 or more, white level becomes 15000,
* with linear interpolation, thus stretching the histogram in the brightest half-stop.
*
* Feel free to optimize it with fixed point.
*
* This hack has no effect in movie mode or outside LV, because white level is normally under 16383.
*/
float k = COERCE((raw - 10000) / 5000.0, 0.0, 1.0);
int adjusted_white = raw_info.white_level * (1-k) + 15000 * k;
raw_max = adjusted_white - raw_info.black_level;
}
float raw_ev = -log2f(raw_max) + log2f(COERCE(raw - raw_info.black_level, 1, raw_max));
return raw_ev;
}
int FAST ev_to_raw(float ev)
{
int raw_max = MIN(raw_info.white_level, 16383) - raw_info.black_level;
return raw_info.black_level + powf(2, ev) * raw_max;
}
static void autodetect_black_level_calc(int x1, int x2, int y1, int y2, int dx, int dy, int* out_mean, int* out_stdev_x100)
{
int black = 0;
int num = 0;
/* compute average level */
for (int y = y1; y < y2; y += dy)
{
for (int x = x1; x < x2; x += dx)
{
int p = raw_get_pixel(x, y);
if (p == 0) continue; /* bad pixel */
black += p;
num++;
}
}
int mean = black / num;
/* compute standard deviation */
int stdev = 0;
for (int y = y1; y < y2; y += dy)
{
for (int x = x1; x < x2; x += dx)
{
int p = raw_get_pixel(x, y);
if (p == 0) continue;
int dif = p - mean;
stdev += dif * dif;
#ifdef RAW_DEBUG_BLACK
/* to check if we are reading the black level from the proper spot, enable RAW_DEBUG_BLACK here and in save_dng. */
raw_set_pixel(x, y, rand());
#endif
}
}
if (num)
{
stdev = sqrtf((float)stdev / num) * 100.0;
}
else
{
/* use some "sane" values instead of inf/nan */
stdev = 800;
mean = 2048;
}
*out_mean = mean;
*out_stdev_x100 = stdev;
}
static int autodetect_black_level(int* black_mean, int* black_stdev_x100)
{
//~ static int k = 0;
//~ bmp_printf(FONT_MED, 250, 50, "black refresh: %d ", k++);
/* also handle black level for dual ISO */
int mean1 = 0;
int stdev1 = 0;
int mean2 = 0;
int stdev2 = 0;
if (raw_info.active_area.x1 > 10) /* use the left black bar for black calibration */
{
autodetect_black_level_calc(
16, raw_info.active_area.x1 - 16,
raw_info.active_area.y1 + 20, raw_info.active_area.y2 - 20,
3, 16,
&mean1, &stdev1
);
autodetect_black_level_calc(
16, raw_info.active_area.x1 - 16,
raw_info.active_area.y1 + 22, raw_info.active_area.y2 - 20,
3, 16,
&mean2, &stdev2
);
}
else /* use the top black bar for black calibration */
{
autodetect_black_level_calc(
raw_info.active_area.x1 + 20, raw_info.active_area.x2 - 20,
4, raw_info.active_area.y1 - 4,
16, 4,
&mean1, &stdev1
);
autodetect_black_level_calc(
raw_info.active_area.x1 + 20, raw_info.active_area.x2 - 20,
6, raw_info.active_area.y1 - 4,
16, 4,
&mean2, &stdev2
);
}
/* does it look like dual ISO? take the DR from the cleanest half */
/* correct DR is high-iso DR + ABS(ISO difference) */
/* or low-iso DR + DR improvement */
*black_mean = (mean1 + mean2) / 2;
*black_stdev_x100 = MIN(stdev1, stdev2);
return *black_mean;
}
static int autodetect_white_level(int initial_guess)
{
int white = initial_guess - 3000;
int max = white + 500;
int confirms = 0;
//~ bmp_printf(FONT_MED, 50, 50, "White...");
int raw_height = raw_info.active_area.y2 - raw_info.active_area.y1;
for (int y = raw_info.active_area.y1 + raw_height/10; y < raw_info.active_area.y2 - raw_height/10; y += 5)
{
int pitch = raw_info.width/8*14;
int row = (intptr_t) raw_info.buffer + y * pitch;
int skip_5p = ((raw_info.active_area.x2 - raw_info.active_area.x1) * 6/128)/8*14; /* skip 5% */
int row_crop_start = row + raw_info.active_area.x1/8*14 + skip_5p;
int row_crop_end = row + raw_info.active_area.x2/8*14 - skip_5p;
for (struct raw_pixblock * p = (void*)row_crop_start; (void*)p < (void*)row_crop_end; p += 5)
{
if (p->a > max)
{
max = p->a;
confirms = 1;
}
else if (p->a == max)
{
confirms++;
if (confirms > 5)
{
white = max - 500;
}
}
}
}
//~ bmp_printf(FONT_MED, 50, 50, "White: %d ", white);
return white;
}
static int compute_dynamic_range(int black_mean, int black_stdev_x100, int white_level)
{
/**
* A = full well capacity / read-out noise
* DR in dB = 20 log10(A)
* DR in stops = dB / 6 = log2(A)
* I guess noise level is the RMS value, which is identical to stdev
*
* This is quite close to DxO measurements (within +/- 0.5 EV),
* except at very high ISOs where there seems to be noise reduction applied to raw data
*/
#ifdef RAW_DEBUG_DR
int mean = black_mean * 100;
int stdev = black_stdev_x100;
bmp_printf(FONT_MED, 50, 100, "mean=%d.%02d stdev=%d.%02d white=%d", mean/100, mean%100, stdev/100, stdev%100, white_level);
white_level = autodetect_white_level(15000);
#endif
int dr = (int)roundf((log2f(white_level - black_mean) - log2f(black_stdev_x100 / 100.0)) * 100);
#ifdef RAW_DEBUG_DR
bmp_printf(FONT_MED, 50, 120, "=> dr=%d.%02d", dr/100, dr%100);
#endif
/* dual ISO enabled? */
dr += dual_iso_get_dr_improvement();
return dr;
}
#ifdef CONFIG_RAW_LIVEVIEW
static int lv_raw_enabled = 0;
#ifdef CONFIG_EDMAC_RAW_SLURP
static void* redirected_raw_buffer = 0;
#endif
/* to be called from vsync hooks */
void FAST raw_lv_redirect_edmac(void* ptr)
{
#ifdef CONFIG_EDMAC_RAW_SLURP
redirected_raw_buffer = (void*) CACHEABLE(ptr);
#else
MEM(RAW_LV_EDMAC) = (intptr_t) CACHEABLE(ptr);
#endif
}
#ifdef CONFIG_EDMAC_RAW_SLURP
void FAST raw_lv_vsync()
{
/* where should we save the raw data? */
void* buf = redirected_raw_buffer ? redirected_raw_buffer : (void*) DEFAULT_RAW_BUFFER;
if (buf && lv_raw_enabled)
{
//~ bmp_printf(FONT_MED, 50, 50, "%x %dx%d ", buf, raw_info.pitch, raw_info.height);
/* pull the raw data into "buf" */
edmac_raw_slurp(CACHEABLE(buf), raw_info.pitch, raw_info.height);
}
else
{
//~ bmp_printf(FONT_MED, 50, 50, "%x %d... ", buf, lv_raw_enabled);
}
/* overriding the buffer is only valid for one frame */
redirected_raw_buffer = 0;
#ifdef PREFERRED_RAW_TYPE
/* this needs to be set for every single frame */
EngDrvOutLV(MEM(RAW_TYPE_ADDRESS-4), PREFERRED_RAW_TYPE);
#endif
}
#endif
int raw_lv_settings_still_valid()
{
/* should be fast enough for vsync calls */
if (!lv_raw_enabled) return 0;
int w, h;
if (!raw_lv_get_resolution(&w, &h)) return 0;
if (w != raw_info.width || h != raw_info.height) return 0;
return 1;
}
#endif
/* For accessing the pixels in a struct raw_pixblock, faster than via raw_get_pixel */
/* todo: move in raw.h? */
#define PA ((int)(p->a))
#define PB ((int)(p->b_lo | (p->b_hi << 12)))
#define PC ((int)(p->c_lo | (p->c_hi << 10)))
#define PD ((int)(p->d_lo | (p->d_hi << 8)))
#define PE ((int)(p->e_lo | (p->e_hi << 6)))
#define PF ((int)(p->f_lo | (p->f_hi << 4)))
#define PG ((int)(p->g_lo | (p->g_hi << 2)))
#define PH ((int)(p->h))
/* a second set of pixels */
#define QA ((int)(q->a))
#define QB ((int)(q->b_lo | (q->b_hi << 12)))
#define QC ((int)(q->c_lo | (q->c_hi << 10)))
#define QD ((int)(q->d_lo | (q->d_hi << 8)))
#define QE ((int)(q->e_lo | (q->e_hi << 6)))
#define QF ((int)(q->f_lo | (q->f_hi << 4)))
#define QG ((int)(q->g_lo | (q->g_hi << 2)))
#define QH ((int)(q->h))
static void FAST raw_preview_color_work(void* raw_buffer, void* lv_buffer, int y1, int y2)
{
uint16_t* lv16 = CACHEABLE(lv_buffer);
uint32_t* lv32 = (uint32_t*) lv16;
if (!lv16) return;
struct raw_pixblock * raw = CACHEABLE(raw_buffer);
if (!raw) return;
/* white balance 2,1,2 => use two gamma curves to simplify code */
uint8_t gamma_rb[1024];
uint8_t gamma_g[1024];
for (int i = 0; i < 1024; i++)
{
/* only show 10 bits */
int black = (raw_info.black_level>>4);
int g_rb = (i > black) ? (log2f(i - black) + 1) * 255 / 10 : 0;
int g_g = (i > black) ? (log2f(i - black)) * 255 / 10 : 0;
gamma_rb[i] = COERCE(g_rb * g_rb / 255, 0, 255); /* idk, looks better this way */
gamma_g[i] = COERCE(g_g * g_g / 255, 0, 255); /* (it's like a nonlinear curve applied on top of log) */
}
int x1 = BM2LV_X(os.x0);
int x2 = BM2LV_X(os.x_max);
x1 = MAX(x1, RAW2LV_X(MAX(raw_info.active_area.x1, preview_rect_x)));
x2 = MIN(x2, RAW2LV_X(MIN(raw_info.active_area.x2, preview_rect_x + preview_rect_w)));
if (x2 < x1) return;
/* cache the LV to RAW transformation for the inner loop to make it faster */
/* we will always choose a green pixel */
int* lv2rx = malloc(x2 * 4);
if (!lv2rx) return;
for (int x = x1; x < x2; x++)
lv2rx[x] = LV2RAW_X(x) & ~1;
/* full-res vertically */
for (int y = y1; y < y2; y++)
{
int yr = LV2RAW_Y(y) & ~1;
/* on HDMI screens, BM2LV_DX() may get negative */
if((yr <= preview_rect_y || yr >= preview_rect_y + preview_rect_h) && BM2LV_DX(x2-x1) > 0)
{
/* out of range, just fill with black */
memset(&lv32[LV(0,y)/4], 0, BM2LV_DX(x2-x1)*2);
continue;
}
struct raw_pixblock * row = (void*)raw + yr * raw_info.pitch;
/* half-res horizontally, to simplify YUV422 math */
for (int x = x1; x < x2; x += 2)
{
int xr = lv2rx[x];
struct raw_pixblock * p = row + (xr/8); /* RG (xr and yr are multiples of 2) */
struct raw_pixblock * q = (void*) p + raw_info.pitch; /* GB, next line */
int r,g,b;
/* RGGB cell */
/* note: at 1920 horizontal resolution in raw, downsampling by 8 would result in 240px horizontally => looks ugly */
switch (xr%8)
{
case 0:
r = PA >> 4;
g = (PB + QA) >> 5;
b = QB >> 4;
break;
case 2:
r = PC >> 4;
g = (PD + QC) >> 5;
b = QD >> 4;
break;
case 4:
r = PE >> 4;
g = (PF + QE) >> 5;
b = QF >> 4;
break;
case 6:
r = PG >> 4;
g = (PH + QG) >> 5;
b = QH >> 4;
break;
default:
r = g = b = 0;
}
r = gamma_rb[r];
g = gamma_g [g];
b = gamma_rb[b];
uint32_t yuv = rgb2yuv422(r,g,b);
lv32[LV(x,y)/4] = yuv;
}
}
free(lv2rx);
}
static void FAST raw_preview_fast_work(void* raw_buffer, void* lv_buffer, int y1, int y2)
{
uint16_t* lv16 = CACHEABLE(lv_buffer);
uint64_t* lv64 = (uint64_t*) lv16;
if (!lv16) return;
struct raw_pixblock * raw = CACHEABLE(raw_buffer);
if (!raw) return;
uint8_t gamma[1024];
for (int i = 0; i < 1024; i++)
{
int g = (i > (raw_info.black_level>>4)) ? log2f(i - (raw_info.black_level>>4)) * 255 / 10 : 0;
gamma[i] = g * g / 255; /* idk, looks better this way */
}
int x1 = BM2LV_X(os.x0);
int x2 = BM2LV_X(os.x_max);
x1 = MAX(x1, RAW2LV_X(MAX(raw_info.active_area.x1, preview_rect_x)));
x2 = MIN(x2, RAW2LV_X(MIN(raw_info.active_area.x2, preview_rect_x + preview_rect_w)));
if (x2 < x1) return;
/* cache the LV to RAW transformation for the inner loop to make it faster */
/* we will always choose a green pixel */
int* lv2rx = malloc(x2 * 4);
if (!lv2rx) return;
for (int x = x1; x < x2; x++)
lv2rx[x] = LV2RAW_X(x) & ~1;
for (int y = y1; y < y2; y++)
{
int yr = LV2RAW_Y(y) | 1;
/* on HDMI screens, BM2LV_DX() may get negative */
if((yr <= preview_rect_y || yr >= preview_rect_y + preview_rect_h) && BM2LV_DX(x2-x1) > 0)
{
/* out of range, just fill with black */
memset(&lv64[LV(0,y)/8], 0, BM2LV_DX(x2-x1)*2);
continue;
}
struct raw_pixblock * row = (void*)raw + yr * raw_info.pitch;
if (y%2) continue;
for (int x = x1; x < x2; x += 4)
{
int xr = lv2rx[x];
struct raw_pixblock * p = row + (xr/8);
int c = p->a;
uint64_t Y = gamma[c >> 4];
Y = (Y << 8) | (Y << 24) | (Y << 40) | (Y << 56);
int idx = LV(x,y)/8;
lv64[idx] = Y;
lv64[idx + vram_lv.pitch/8] = Y;
}
}
free(lv2rx);
}
void FAST raw_preview_fast_ex(void* raw_buffer, void* lv_buffer, int y1, int y2, int quality)
{
yuv422_buffer_check();
if (raw_buffer == (void*)-1)
raw_buffer = (void*)raw_info.buffer;
if (lv_buffer == (void*)-1)
lv_buffer = (void*)YUV422_LV_BUFFER_DISPLAY_ADDR;
if (y1 == -1)
y1 = BM2LV_Y(os.y0);
if (y2 == -1)
y2 = BM2LV_Y(os.y_max);
if (quality == -1)
quality = 0;
switch (quality)
{
case RAW_PREVIEW_GRAY_ULTRA_FAST:
raw_preview_fast_work(raw_buffer, lv_buffer, y1, y2);
break;
case RAW_PREVIEW_COLOR_HALFRES:
default:
raw_preview_color_work(raw_buffer, lv_buffer, y1, y2);
break;
}
}
void FAST raw_preview_fast()
{
raw_preview_fast_ex((void*)-1, (void*)-1, -1, -1, -1);
}
#ifdef CONFIG_RAW_LIVEVIEW
#ifndef CONFIG_EDMAC_RAW_SLURP
#ifdef PREFERRED_RAW_TYPE
static int old_raw_type = -1;
#endif
#endif
static void raw_lv_enable()
{
/* make sure LiveView is fully started before enabling the raw flag */
/* if enabled too early, right after the property is fired, the raw stream may not come up (race condition in Canon code?) */
wait_lv_frames(2);
lv_raw_enabled = 1;
#ifndef CONFIG_EDMAC_RAW_SLURP
call("lv_save_raw", 1);
#ifdef PREFERRED_RAW_TYPE
/* set new raw type; Canon's lv_save_raw will apply it */
old_raw_type = MEM(RAW_TYPE_ADDRESS);
MEM(RAW_TYPE_ADDRESS) = PREFERRED_RAW_TYPE;
#endif
#endif
}
static void raw_lv_disable()
{
lv_raw_enabled = 0;
#ifndef CONFIG_EDMAC_RAW_SLURP
call("lv_save_raw", 0);
#ifdef PREFERRED_RAW_TYPE
/* restore old raw type */
if (old_raw_type != -1)
{
MEM(RAW_TYPE_ADDRESS) = old_raw_type;
old_raw_type = -1;
}
#endif
#endif
}
int raw_lv_is_enabled()
{
return lv_raw_enabled;
}
static int raw_lv_request_count = 0;
static void raw_lv_update()
{
int new_state = raw_lv_request_count > 0;
if (new_state && !lv_raw_enabled)
{
raw_lv_enable();
for (int i = 0; i < 20; i++)
{
if (raw_update_params_work())
break;
msleep(50);
}
}
else if (!new_state && lv_raw_enabled)
{
/* in zoom mode, these cameras have pink preview in zoom mode (even after disabling raw flag, the pink preview remains) */
/* lookup 0xc0f08114 in raw_rec.c for more info */
//~ #define PINK_FIX_TEST
#ifdef PINK_FIX_TEST
/* with zoom on halfshutter, the image should get pink for 2 seconds, then it should be back to normal */
msleep(1000);
beep(); /* first beep: disabling raw mode, image remains pink */
#endif
/* disable the raw flag */
raw_lv_disable();
msleep(50);
#if defined(CONFIG_5D2) || defined(CONFIG_50D) || defined(CONFIG_500D)
#ifdef PINK_FIX_TEST
msleep(1000);
beep(); /* second beep: changing raw type to something that isn't pink (this will be reset as soon as you enable raw back) */
#endif
/* fix pink preview in zoom */
if (lv && lv_dispsize > 1 && DISPLAY_IS_ON)
{
/* todo: enqueue it in a vsync hook? */
EngDrvOutLV(0xc0f08114, 0);
}
#endif
}
}
void raw_lv_request()
{
/* refresh VRAM parameters */
/* the BMP_LOCK is just to make sure this will not conflict with other locks */
/* (get_yuv422_vram will only call BMP_LOCK if it has to refresh something, that is, once in a blue moon) */
BMP_LOCK( get_yuv422_vram(); )
/* this one should be called only in LiveView */
ASSERT(lv);
take_semaphore(raw_sem, 0);
raw_lv_request_count++;
raw_lv_update();
give_semaphore(raw_sem);
}
void raw_lv_release()
{
take_semaphore(raw_sem, 0);
raw_lv_request_count--;
ASSERT(raw_lv_request_count >= 0);
raw_lv_update();
give_semaphore(raw_sem);
}
#endif
/* may not be correct on 4:3 screens */
void raw_force_aspect_ratio_1to1()
{
if (lv2raw.sy < lv2raw.sx) /* image too tall */
{
lv2raw.sy = lv2raw.sx;
int height = RAW2LV_DY(preview_rect_h);
int offset = (BM2LV_DY(os.y_ex) - height) / 2;
int skip_top = preview_rect_y;
lv2raw.ty = skip_top - LV2RAW_DY(os.y0) - LV2RAW_DY(offset);
}
else if (lv2raw.sx < lv2raw.sy) /* image too wide */
{
lv2raw.sx = lv2raw.sy;
int width = RAW2LV_DX(preview_rect_w);
int offset = (vram_lv.width - width) / 2;
int skip_left = preview_rect_x;
lv2raw.tx = skip_left - LV2RAW_DX(os.x0) - LV2RAW_DX(offset);
}
}
void raw_set_dirty()
{
dirty = 1;
}
int get_dxo_dynamic_range(int raw_iso)
{
int iso = raw_iso;
int iso_rounded = COERCE((iso + 3) / 8 * 8, 72, 72 + (COUNT(dynamic_ranges)-1) * 8);
int dr_index = COERCE((iso_rounded - 72) / 8, 0, COUNT(dynamic_ranges)-1);
float iso_digital = (iso - iso_rounded) / 8.0f;
int dr = dynamic_ranges[dr_index];
if (iso_digital > 0)
{
/* at ISO 250, 500, 1000, dynamic range is lowered,
* because data is amplified but white level stays the same
*/
dr -= iso_digital * 100;
}
else if (iso_digital < 0)
{
/* at ISO 160, 320 and so on, the DR is:
* - pretty much the same on old cameras (a tiny bit lost because of quantization error)
* - 0.1 stops on new cameras (best guess: starting from 550D)
*
* important?
*/
}
return dr;
}
/* helpers for menu and other code that wants to use raw */
int can_use_raw_overlays_photo()
{
// MRAW/SRAW are causing trouble.
// Besides buffer address different on some cameras, these formats are lossy and break all my noise analysis tools
// RAW and RAW+JPEG are OK
if ((pic_quality & 0xFE00FF) == (PICQ_RAW & 0xFE00FF))
return 1;
return 0;
}
int can_use_raw_overlays()
{
if (QR_MODE && can_use_raw_overlays_photo())
return 1;
#ifdef CONFIG_RAW_LIVEVIEW
if (lv && raw_lv_is_enabled())
return 1;
#endif
return 0;
}
int can_use_raw_overlays_menu()
{
#ifdef CONFIG_RAW_LIVEVIEW
if (is_movie_mode())
{
/* in movie mode, raw overlays don't make much sense for H.264 video, so only show them for raw video */
if (lv && raw_lv_is_enabled())
return 1;
}
else
#endif
{
/* outside LiveView: only pure RAW is known to work */
if (can_use_raw_overlays_photo())
return 1;
#ifdef CONFIG_RAW_LIVEVIEW
/* in LiveView: we can display the raw overlays no matter what */
/* so use them also for sRAW and mRAW, even if this may not work in QR mode */
int raw = pic_quality & 0x60000;
if (lv && raw)
return 1;
#endif
}
return 0;
}
MENU_UPDATE_FUNC(menu_set_warning_raw)
{
MENU_SET_WARNING(MENU_WARN_NOT_WORKING,
is_movie_mode() ? "[MOVIE] This feature requires you shooting RAW." :
"[PHOTO] Set picture quality to RAW in Canon menu."
);
}
MENU_UPDATE_FUNC(menu_checkdep_raw)
{
if (!can_use_raw_overlays_menu())
{
menu_set_warning_raw(entry, info);
}
}
#ifdef CONFIG_RAW_DISABLE_IN_10X_ZOOM
/* workaround: disable raw flag in zoom mode (it crashes on some cameras) */
/* not exactly nice, but better than crashing */
PROP_HANDLER(PROP_LV_DISPSIZE)
{
if (lv_raw_enabled)
{
if (buf[0] == 10)
{
lv_raw_enabled = -1;
#ifndef CONFIG_RAW_DISABLE_IN_10X_ZOOM_WEAK
/* crash reported on 50D if this is called; however, other cameras (550D, 500D) require it to avoid crashes */
call("lv_save_raw", 0);
#endif
}
else
{
lv_raw_enabled = 1;
#ifndef CONFIG_RAW_DISABLE_IN_10X_ZOOM_WEAK
call("lv_save_raw", 1);
#endif
}
}
}
#endif
static void raw_init()
{
raw_sem = create_named_semaphore("raw_sem", 1);
}
INIT_FUNC("raw", raw_init);