/*
 *  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/>.
 */
#include <string>
#include <glibmm/ustring.h>
#include <map>
#include <cmath>
#include "rawimage.h"

namespace rtengine{

class ffInfo
{
public:

	Glib::ustring pathname; // filename of flat field
	std::list<Glib::ustring> pathNames; // other similar dark frames, used for average
	std::string maker;  ///< manufacturer
	std::string model;  ///< model
	std::string lens;   ///< lens
	int iso;            ///< ISO (gain)
	double shutter;     ///< shutter or exposure time in sec
	double aperture;	///< aperture in stops
	double focallength; ///< focal length in mm
	time_t timestamp;   ///< seconds since 1 Jan 1970


	ffInfo(const Glib::ustring &name, const std::string &mak, const std::string &mod,const std::string & len,double focal,double apert,time_t t)
	:pathname(name),maker(mak),model(mod),lens(len),aperture(apert),focallength(focal),timestamp(t),ri(NULL){}

	ffInfo( const ffInfo &o)
	:pathname(o.pathname),maker(o.maker),model(o.model),lens(o.lens),aperture(o.aperture),focallength(o.focallength),timestamp(o.timestamp),ri(NULL){}
	~ffInfo() { if( ri ) delete ri; }

	
	ffInfo &operator =(const ffInfo &o);
	bool operator <(const ffInfo &e2) const;

	// Calculate virtual distance between two shots; different model return infinite
	double distance(const std::string &mak, const std::string &mod, const std::string &lens, double focallength, double aperture) const;

	static std::string key(const std::string &mak, const std::string &mod, const std::string &len, double focal, double apert );
	std::string key(){ return key( maker,model,lens,focallength,aperture); }

	RawImage *getRawImage();

protected:
	RawImage *ri; ///< Flat Field raw data

	void updateRawImage();
};

class FFManager
{
public:
	void init( Glib::ustring pathname );
	Glib::ustring getPathname(){ return currentPath; };
	void getStat( int &totFiles, int &totTemplate);
	RawImage *searchFlatField( const std::string &mak, const std::string &mod, const std::string &len, double focallength, double apert, time_t t );
	RawImage *searchFlatField( const Glib::ustring filename );

protected:
	typedef std::multimap<std::string,ffInfo> ffList_t;
	typedef std::map<std::string, std::list<badPix> > bpList_t;
	ffList_t ffList;
	bool initialized;
	Glib::ustring currentPath;
	ffInfo *addFileInfo(const Glib::ustring &filename, bool pool=true );
	ffInfo *find( const std::string &mak, const std::string &mod, const std::string &len, double focal, double apert, time_t t );
};

extern FFManager ffm;

}