117 lines
3.6 KiB
C++
117 lines
3.6 KiB
C++
/*
|
|
* This file is part of RawTherapee.
|
|
*
|
|
* Copyright (c) 2012 Oliver Duis <www.oliverduis.de>
|
|
*
|
|
* 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 _LCP_
|
|
#define _LCP_
|
|
|
|
#include "imagefloat.h"
|
|
#include <glibmm.h>
|
|
#include <map>
|
|
#include <list>
|
|
#include <string>
|
|
#include <expat.h>
|
|
|
|
namespace rtengine {
|
|
// Perspective model common data, also used for Vignette and Fisheye
|
|
class LCPModelCommon {
|
|
public:
|
|
float focLenX, focLenY, imgXCenter, imgYCenter;
|
|
float param[5]; // k1..k5
|
|
float scaleFac;
|
|
|
|
LCPModelCommon();
|
|
bool empty() const; // is it empty?
|
|
void print() const; // printf all values
|
|
};
|
|
|
|
class LCPPersModel {
|
|
public:
|
|
float focLen, focDist, aperture; // this is what it refers to
|
|
|
|
LCPModelCommon base; // base perspective correction
|
|
LCPModelCommon chromRG, chromG, chromBG; // red/green, green, blue/green (may be empty)
|
|
LCPModelCommon vignette; // vignette (may be empty)
|
|
|
|
LCPPersModel();
|
|
void print() const;
|
|
};
|
|
|
|
|
|
class LCPProfile {
|
|
// Temporary data for parsing
|
|
bool inCamProfiles,firstLIDone,inPerspect;
|
|
char lastTag[256];
|
|
LCPPersModel* pCurPersModel;
|
|
LCPModelCommon* pCurCommon;
|
|
|
|
static void XMLCALL XmlStartHandler(void *pLCPProfile, const char *el, const char **attr);
|
|
static void XMLCALL XmlTextHandler (void *pLCPProfile, const XML_Char *s, int len);
|
|
static void XMLCALL XmlEndHandler (void *pLCPProfile, const char *el);
|
|
|
|
public:
|
|
// Common data
|
|
Glib::ustring profileName, lensPrettyName, cameraPrettyName, lens, camera; // lens/camera(=model) can be auto-matched with DNG
|
|
bool isRaw,isFisheye;
|
|
|
|
// The correction frames
|
|
std::list<LCPPersModel*> persModels;
|
|
|
|
LCPProfile(Glib::ustring fname);
|
|
|
|
void calcBasePerspectiveParams(float focalLength, bool vignette, LCPModelCommon& corr); // Interpolates between the persModels frames
|
|
|
|
void print() const;
|
|
};
|
|
|
|
class LCPStore {
|
|
Glib::Mutex mtx;
|
|
|
|
|
|
// Maps file name to profile as cache
|
|
std::map<Glib::ustring, LCPProfile*> profileCache;
|
|
|
|
public:
|
|
Glib::ustring getDefaultCommonDirectory() const;
|
|
bool isValidLCPFileName(Glib::ustring filename) const;
|
|
LCPProfile* getProfile(Glib::ustring filename);
|
|
|
|
static LCPStore* getInstance();
|
|
};
|
|
|
|
#define lcpStore LCPStore::getInstance()
|
|
|
|
|
|
// Once precalculated class to correct a point
|
|
class LCPMapper {
|
|
double x0,y0,fx,fy;
|
|
LCPModelCommon mc;
|
|
bool swapXY;
|
|
bool mirrorX, mirrorY;
|
|
|
|
public:
|
|
// precalculates the mapper.
|
|
LCPMapper(LCPProfile* pProf, float focalLength, bool vignette, int fullWidth, int fullHeight,
|
|
const CoarseTransformParams& coarse, int rawRotationDeg);
|
|
|
|
void correctDistortion(double& x, double& y) const; // MUST be the first stage
|
|
float correctVignette (double x, double y) const; // MUST be in RAW
|
|
};
|
|
}
|
|
#endif
|