LensCalImageCtrl.h

Go to the documentation of this file.

// -*- c-basic-offset: 4 -*-
/*  This 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
 *  Lesser General Public License for more details.
 *
 *  You should have received a copy of the GNU General Public
 *  License along with this software. If not, see
 *  <http://www.gnu.org/licenses/>.
 *
 */

#ifndef LensCalImageCtrl_H
#define LensCalImageCtrl_H

#include <base_wx/wxImageCache.h>
#include <lcms2.h>
#include "lines/FindLines.h"
#include "LensCalTypes.h"

class LensCalImageCtrl : public wxPanel
{
public:
    enum LensCalPreviewMode
    {
        mode_original=0,
        mode_edge=1,
        mode_corrected=2
    };
    LensCalImageCtrl();
    void SetImage(ImageLineList* newList, unsigned int newIndex);
    void SetEmptyImage();
    void SetShowLines(bool showLines);
    void SetMode(const LensCalPreviewMode newMode);
    const LensCalPreviewMode GetMode();
    void SetLens(const HuginBase::SrcPanoImage::Projection newProjection,const double newFocallength, const double newCropfactor);
    void SetLensDistortions(const double newA, const double newB, const double newC, const double newD, const double newE);

protected:
    void DrawView();
    void OnMouseEvent(wxMouseEvent &e);

private:
    void Resize ( wxSizeEvent & e );
    void OnPaint(wxPaintEvent & dc);
    void SetEdgeImage();
    void GenerateRemappedImage(const unsigned int newWidth,const unsigned int newHeight);

    ImageLineList* m_imageLines;
    unsigned int m_imageIndex;
    wxImage m_img;
    vigra::BRGBImage m_edgeImage;
    vigra::BRGBImage m_remappedImage;
    wxImage m_edge;
    wxImage m_remapped_img;
    wxBitmap m_scaled_img;
    wxBitmap m_display_img;
    float m_scale;
    bool m_showLines;
    LensCalPreviewMode m_previewMode;
    cmsHPROFILE m_monitorProfile;
    bool m_hasMonitorProfile;

    // some actual lens parameters
    HuginBase::SrcPanoImage::Projection m_projection;
    double m_focallength;
    double m_cropfactor;
    double m_a;
    double m_b;
    double m_c;
    double m_d;
    double m_e;
    // SrcPanoImage and PanoramaOptions to calculate remapped image and transform lines
    HuginBase::SrcPanoImage m_panoimage;
    HuginBase::PanoramaOptions m_opts;

    DECLARE_EVENT_TABLE()
    DECLARE_DYNAMIC_CLASS(CenterCanvas)
};

class LensCalImageCtrlXmlHandler : public wxXmlResourceHandler
{
    DECLARE_DYNAMIC_CLASS(LensCalImageCtrlXmlHandler)

public:
    LensCalImageCtrlXmlHandler();
    virtual wxObject *DoCreateResource();
    virtual bool CanHandle(wxXmlNode *node);
};

#endif // LensCalImageCtrl_H