LensDB.h

Go to the documentation of this file.

// -*- c-basic-offset: 4 -*-

/*  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 software 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 software; if not, write to the Free Software
 *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
 *
 */

#ifndef _LENS_DB_H
#define _LENS_DB_H

#include <hugin_shared.h>
#include <string>
#ifndef Hugin_shared
#define CONF_LENSFUN_STATIC
#endif
#include <lensfun.h>
#include <panodata/SrcPanoImage.h>
#include <hugin_math/hugin_math.h>

namespace HuginBase
{

namespace LensDB
{

class IMPEX LensDBList
{
public:
    LensDBList();
    ~LensDBList();
    const size_t GetLensCount() const;
    const lfLens* GetLens(size_t index) const;
    void SetLenses(const lfLens** lenses);
    std::string GetLensName(size_t index) const;
    void SetCameraModelMaker(const std::string camMaker, const std::string camModel);
private:
    const lfLens** m_lenses;
    size_t m_lensCount;
    std::string m_camMaker;
    std::string m_camModel;
};

class IMPEX LensDB
{
public:
    LensDB();
    virtual ~LensDB();

    void SetMainDBPath(std::string mainPath);
    std::string GetMainDBPath();
    std::string GetUserDBPath();
    void ReloadUserPart();

    // routines to read from database
    bool GetCropFactor(std::string maker, std::string model, double &cropFactor);
    bool GetCameraMount(std::string maker, std::string model, std::string &mount);
    bool FindLens(std::string camMaker, std::string camModel, std::string lens);
    void SetActiveLens(const lfLens* activeLens);
    bool CheckLensFocal(double focal);
    bool CheckLensAperture(double aperture);
    bool FindLenses(std::string camMaker, std::string camModel, std::string lensname, LensDBList &foundLenses, bool fuzzy=false);
    bool GetMounts(std::vector<std::string> &foundMounts);
    bool GetProjection(BaseSrcPanoImage::Projection & projection);
    bool GetCrop(double focal, BaseSrcPanoImage::CropMode &cropMode, hugin_utils::FDiff2D &cropLeftTop, hugin_utils::FDiff2D &cropRightBottom);
    bool GetFov(double focal,double &fov);
    bool GetDistortion(double focal, std::vector<double> &distortion);
    bool GetVignetting(double focal, double aperture, double distance, std::vector<double> &vignetting);

    //routines to save to database
    bool SaveCameraCrop(std::string filename, std::string maker, std::string model, std::string mount, double cropfactor);

    int BeginSaveLens(std::string filename, std::string maker, std::string lensname, std::string mount, BaseSrcPanoImage::Projection projection, double cropfactor);
    void SaveHFOV(double focal, double hfov);
    void SaveCrop(double focal, BaseSrcPanoImage::CropMode cropMode, hugin_utils::FDiff2D cropLeftTop, hugin_utils::FDiff2D cropRightBottom);
    void SaveDistortion(double focal, std::vector<double> distortion);
    void SaveVignetting(double focal, double aperture, double distance, std::vector<double> vignetting);
    bool EndSaveLens();

    // access to single database class
    static LensDB& GetSingleton();
    static void Clean();
private:
    bool InitDB();
    bool LoadFilesInDir(std::string path);
    void CleanSaveInformation();
    bool IsNewMount(std::string newMount);
    void FreeLensList();
    void CleanUpdatedMounts();
    struct lfDatabase *m_db;
    struct lfDatabase *m_newDB;
    const struct lfLens **m_lenses;
    bool m_needLensCleanup;
    struct lfLens **m_updatedLenses;
    struct lfMount **m_updatedMounts;
    struct lfLens *m_currentLens;
    std::string m_lensFilename;
    bool m_initialized;
    // stores the pathes
    std::string m_main_db_path;
    std::string m_user_db_path;
    static LensDB* m_instance;
};

}; //namespace LensDB
}; //namespace HuginBase

#endif //_LENS_DB_H

---