107 lines
3.6 KiB
C++
107 lines
3.6 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 <https://www.gnu.org/licenses/>.
|
|
*/
|
|
#pragma once
|
|
|
|
#include <cmath>
|
|
#include <list>
|
|
#include <map>
|
|
#include <string>
|
|
|
|
#include <glibmm/ustring.h>
|
|
|
|
#include "pixelsmap.h"
|
|
|
|
namespace rtengine
|
|
{
|
|
|
|
class RawImage;
|
|
class dfInfo final
|
|
{
|
|
public:
|
|
|
|
Glib::ustring pathname; // filename of dark frame
|
|
std::list<Glib::ustring> pathNames; // other similar dark frames, used for average
|
|
std::string maker; ///< manufacturer
|
|
std::string model; ///< model
|
|
int iso; ///< ISO (gain)
|
|
double shutter; ///< shutter or exposure time in sec
|
|
time_t timestamp; ///< seconds since 1 Jan 1970
|
|
|
|
|
|
dfInfo(const Glib::ustring &name, const std::string &mak, const std::string &mod, int iso, double shut, time_t t)
|
|
: pathname(name), maker(mak), model(mod), iso(iso), shutter(shut), timestamp(t), ri(nullptr) {}
|
|
|
|
dfInfo( const dfInfo &o)
|
|
: pathname(o.pathname), maker(o.maker), model(o.model), iso(o.iso), shutter(o.shutter), timestamp(o.timestamp), ri(nullptr) {}
|
|
~dfInfo();
|
|
|
|
dfInfo &operator =(const dfInfo &o);
|
|
bool operator <(const dfInfo &e2) const;
|
|
|
|
// Calculate virtual distance between two shots; different model return infinite
|
|
double distance(const std::string &mak, const std::string &mod, int iso, double shutter) const;
|
|
|
|
static std::string key(const std::string &mak, const std::string &mod, int iso, double shut );
|
|
std::string key()
|
|
{
|
|
return key( maker, model, iso, shutter);
|
|
}
|
|
|
|
RawImage *getRawImage();
|
|
std::vector<badPix> &getHotPixels();
|
|
|
|
protected:
|
|
RawImage *ri; ///< Dark Frame raw data
|
|
std::vector<badPix> badPixels; ///< Extracted hot pixels
|
|
|
|
void updateBadPixelList( RawImage *df );
|
|
void updateRawImage();
|
|
};
|
|
|
|
class DFManager final
|
|
{
|
|
public:
|
|
void init(const Glib::ustring &pathname);
|
|
Glib::ustring getPathname()
|
|
{
|
|
return currentPath;
|
|
};
|
|
void getStat( int &totFiles, int &totTemplate);
|
|
RawImage *searchDarkFrame( const std::string &mak, const std::string &mod, int iso, double shut, time_t t );
|
|
RawImage *searchDarkFrame( const Glib::ustring filename );
|
|
std::vector<badPix> *getHotPixels ( const std::string &mak, const std::string &mod, int iso, double shut, time_t t );
|
|
std::vector<badPix> *getHotPixels ( const Glib::ustring filename );
|
|
std::vector<badPix> *getBadPixels ( const std::string &mak, const std::string &mod, const std::string &serial);
|
|
|
|
protected:
|
|
typedef std::multimap<std::string, dfInfo> dfList_t;
|
|
typedef std::map<std::string, std::vector<badPix> > bpList_t;
|
|
dfList_t dfList;
|
|
bpList_t bpList;
|
|
bool initialized;
|
|
Glib::ustring currentPath;
|
|
dfInfo *addFileInfo(const Glib::ustring &filename, bool pool = true );
|
|
dfInfo *find( const std::string &mak, const std::string &mod, int isospeed, double shut, time_t t );
|
|
int scanBadPixelsFile( Glib::ustring filename );
|
|
};
|
|
|
|
extern DFManager dfm;
|
|
|
|
}
|