179 lines
7.7 KiB
C++
179 lines
7.7 KiB
C++
/*
|
|
* This file is part of RawTherapee.
|
|
*
|
|
* Copyright (c) 2004-2010 Gabor Horvath <hgabor@rawtherapee.com>
|
|
*
|
|
* RawTherapee 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 3 of the License, or
|
|
* (at your option) any later version.
|
|
*
|
|
* RawTherapee 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 RawTherapee. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
#ifndef _RAWIMAGESOURCE_
|
|
#define _RAWIMAGESOURCE_
|
|
|
|
#include <imagesource.h>
|
|
#include <lcms.h>
|
|
#define HR_SCALE 2
|
|
|
|
namespace rtengine {
|
|
|
|
template<class T> void freeArray (T** a, int H) {
|
|
for (int i=0; i<H; i++)
|
|
delete [] a[i];
|
|
delete [] a;
|
|
}
|
|
|
|
template<class T> T** allocArray (int W, int H) {
|
|
|
|
T** t = new T*[H];
|
|
for (int i=0; i<H; i++)
|
|
t[i] = new T[W];
|
|
return t;
|
|
}
|
|
|
|
|
|
template<class T> void freeArray2 (T** a, int H) {
|
|
for (int i=0; i<H; i++)
|
|
delete [] a[i];
|
|
}
|
|
|
|
class RawImageSource : public ImageSource {
|
|
|
|
protected:
|
|
Glib::Mutex isrcMutex;
|
|
|
|
int W, H;
|
|
ColorTemp wb;
|
|
ProgressListener* plistener;
|
|
double coeff[3][3];
|
|
double icoeff[3][3];
|
|
double cam[3][3];
|
|
double icam[3][3];
|
|
bool fuji;
|
|
bool d1x;
|
|
int border;
|
|
char** hpmap;
|
|
float** hrmap[3]; // for color propagation
|
|
char** needhr; // for color propagation
|
|
int max[3];
|
|
double defGain;
|
|
bool full;
|
|
Glib::ustring oldmethod;
|
|
cmsHPROFILE camProfile;
|
|
cmsHPROFILE embProfile;
|
|
|
|
RawImage* ri;
|
|
|
|
// to accelerate CIELAB conversion:
|
|
double lc00, lc01, lc02, lc10, lc11, lc12, lc20, lc21, lc22;
|
|
double* cache;
|
|
int threshold;
|
|
|
|
// the interpolated green plane:
|
|
unsigned short** green;
|
|
// the interpolated red plane:
|
|
unsigned short** red;
|
|
// the interpolated blue plane:
|
|
unsigned short** blue;
|
|
|
|
void hphd_vertical (float** hpmap, int col_from, int col_to);
|
|
void hphd_horizontal (float** hpmap, int row_from, int row_to);
|
|
void hphd_green ();
|
|
void correction_YIQ_LQ_ (Image16* im, int row_from, int row_to);
|
|
void hlRecovery (std::string method, unsigned short* red, unsigned short* green, unsigned short* blue, int i, int sx1, int width, int skip);
|
|
int defTransform (int tran);
|
|
void rotateLine (unsigned short* line, unsigned short** channel, int tran, int i, int w, int h);
|
|
void transformRect (PreviewProps pp, int tran, int &sx1, int &sy1, int &width, int &height, int &fw);
|
|
void transformPosition (int x, int y, int tran, int& tx, int& ty);
|
|
|
|
void updateHLRecoveryMap (std::string method, double rm, double gm, double bm);
|
|
void updateHLRecoveryMap_ColorPropagation ();
|
|
void HLRecovery_ColorPropagation (unsigned short* red, unsigned short* green, unsigned short* blue, int i, int sx1, int width, int skip);
|
|
|
|
public:
|
|
RawImageSource ();
|
|
~RawImageSource ();
|
|
|
|
int load (Glib::ustring fname);
|
|
void getImage (ColorTemp ctemp, int tran, Image16* image, PreviewProps pp, HRecParams hrp, ColorManagementParams cmp);
|
|
ColorTemp getWB () { return wb; }
|
|
ColorTemp getAutoWB ();
|
|
ColorTemp getSpotWB (std::vector<Coord2D> red, std::vector<Coord2D> green, std::vector<Coord2D>& blue, int tran);
|
|
|
|
double getDefGain () { return defGain; }
|
|
double getGamma () { return 2.2; }
|
|
|
|
void getFullSize (int& w, int& h, int tr = TR_NONE);
|
|
void getSize (int tran, PreviewProps pp, int& w, int& h);
|
|
|
|
ImageData* getImageData () { return idata; }
|
|
void setProgressListener (ProgressListener* pl) { plistener = pl; }
|
|
int getAEHistogram (unsigned int* histogram, int& histcompr);
|
|
|
|
static void colorSpaceConversion (Image16* im, ColorManagementParams cmp, cmsHPROFILE embedded, cmsHPROFILE camprofile, double cam[3][3], double& defgain);
|
|
static void inverse33 (double (*coeff)[3], double (*icoeff)[3]);
|
|
|
|
static void HLRecovery_Luminance (unsigned short* rin, unsigned short* gin, unsigned short* bin, unsigned short* rout, unsigned short* gout, unsigned short* bout, int width, int maxval);
|
|
static void HLRecovery_CIELab (unsigned short* rin, unsigned short* gin, unsigned short* bin, unsigned short* rout, unsigned short* gout, unsigned short* bout, int width, int maxval, double cam[3][3], double icam[3][3]);
|
|
|
|
protected:
|
|
typedef unsigned short ushort;
|
|
void correction_YIQ_LQ (Image16* i, int times);
|
|
inline void convert_row_to_YIQ (unsigned short* r, unsigned short* g, unsigned short* b, int* Y, int* I, int* Q, int W);
|
|
inline void convert_row_to_RGB (unsigned short* r, unsigned short* g, unsigned short* b, int* Y, int* I, int* Q, int W);
|
|
|
|
inline void convert_to_cielab_row (unsigned short* ar, unsigned short* ag, unsigned short* ab, short* oL, short* oa, short* ob);
|
|
inline void interpolate_row_g (unsigned short* agh, unsigned short* agv, int i);
|
|
inline void interpolate_row_rb (unsigned short* ar, unsigned short* ab, unsigned short* pg, unsigned short* cg, unsigned short* ng, int i);
|
|
inline void interpolate_row_rb_mul_pp (unsigned short* ar, unsigned short* ab, unsigned short* pg, unsigned short* cg, unsigned short* ng, int i, double r_mul, double g_mul, double b_mul, int x1, int width, int skip);
|
|
|
|
int LinEqSolve (int c, int dir, int nDim, float* pfMatr, float* pfVect, float* pfSolution);//Emil's CA auto correction
|
|
void CA_correct_RT ();
|
|
|
|
void cfa_clean (float thresh);//Emil's hot/dead pixel filter
|
|
void ddct8x8s(int isgn, float **a);
|
|
|
|
void cfa_linedn (float linenoiselevel);//Emil's line denoise
|
|
|
|
void green_equilibrate (float greenthresh);//Emil's green equilibration
|
|
|
|
|
|
void eahd_demosaic ();
|
|
void hphd_demosaic ();
|
|
void vng4_demosaic ();
|
|
void ppg_demosaic();
|
|
void amaze_demosaic_RT ();//Emil's code for AMaZE
|
|
void dcb_demosaic(int iterations, int dcb_enhance);
|
|
void ahd_demosaic();
|
|
void border_interpolate(int border, ushort (*image)[4]);
|
|
void dcb_initTileLimits(int &colMin, int &rowMin, int &colMax, int &rowMax, int x0, int y0, int border);
|
|
void fill_raw( ushort (*cache )[4], int x0, int y0, ushort** rawData);
|
|
void fill_border( ushort (*cache )[4], int border, int x0, int y0);
|
|
void copy_to_buffer(ushort (*image2)[3], ushort (*image)[4]);
|
|
void dcb_hid(ushort (*image)[4], ushort (*bufferH)[3], ushort (*bufferV)[3], int x0, int y0);
|
|
void dcb_color(ushort (*image)[4], int x0, int y0);
|
|
void dcb_hid2(ushort (*image)[4], int x0, int y0);
|
|
void dcb_map(ushort (*image)[4], int x0, int y0);
|
|
void dcb_correction(ushort (*image)[4], int x0, int y0);
|
|
void dcb_pp(ushort (*image)[4], int x0, int y0);
|
|
void dcb_correction2(ushort (*image)[4], int x0, int y0);
|
|
void restore_from_buffer(ushort (*image)[4], ushort (*image2)[3]);
|
|
void dcb_refinement(ushort (*image)[4], int x0, int y0);
|
|
void dcb_color_full(ushort (*image)[4], int x0, int y0, float (*chroma)[2]);
|
|
|
|
void transLine (unsigned short* red, unsigned short* green, unsigned short* blue, int i, Image16* image, int tran, int imw, int imh, int fw);
|
|
void hflip (Image16* im);
|
|
void vflip (Image16* im);
|
|
|
|
};
|
|
};
|
|
#endif
|