MaskEditorPanel.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 _MaskEditorPanel_H
#define _MaskEditorPanel_H


//-----------------------------------------------------------------------------
// Headers
//-----------------------------------------------------------------------------

#include <panodata/Panorama.h>
#include <wx/clrpicker.h>
#include "MaskImageCtrl.h"
#include <panodata/StandardImageVariableGroups.h>

class MaskEditorPanel : public wxPanel, public HuginBase::PanoramaObserver
{
public:
    MaskEditorPanel();

    bool Create(wxWindow* parent, wxWindowID id = wxID_ANY, const wxPoint& pos = wxDefaultPosition, const wxSize& size = wxDefaultSize, long style = wxTAB_TRAVERSAL, const wxString& name = wxT("panel"));

    void Init(HuginBase::Panorama * pano);

    virtual ~MaskEditorPanel();

    void SetPano(HuginBase::Panorama * panorama)
        { m_pano = panorama; };
    void setImage(unsigned int imgNr, bool updateListSelection=false);
    void setMask(unsigned int maskNr);
    void UpdateMask();
    void AddMask();
    void SelectMask(unsigned int newMaskNr);

    void UpdateCrop(bool updateFromImgCtrl=false);
    void UpdateCropFromImage();
    void SwitchToCropMode();

    void panoramaChanged(HuginBase::Panorama &pano);
    void panoramaImagesChanged(HuginBase::Panorama &pano, const HuginBase::UIntSet & imgNr);

    void OnImageListChar(wxKeyEvent& e);
    void OnImageSelect(wxListEvent &e);
    void OnMaskSelect(wxListEvent &e);
    void OnMaskTypeChange(wxCommandEvent &e);
    void OnMaskAdd(wxCommandEvent &e);
    void OnMaskSave(wxCommandEvent &e);
    void OnMaskLoad(wxCommandEvent &e);
    void OnMaskCopy(wxCommandEvent &e);
    void OnMaskPaste(wxCommandEvent &e);
    void OnMaskDelete(wxCommandEvent &e);
    void OnZoom(wxCommandEvent & e);
    void OnColourChanged(wxColourPickerEvent &e);
    void OnShowActiveMasks(wxCommandEvent &e);
    // reset crop area. 
    void OnResetButton(wxCommandEvent & e);
    void OnSetLeft(wxCommandEvent & e);
    void OnSetRight(wxCommandEvent & e);
    void OnSetTop(wxCommandEvent & e);
    void OnSetBottom(wxCommandEvent & e);
    void OnAutoCenter(wxCommandEvent & e);
    void OnModeChanged(wxNotebookEvent & e);

private:

    void UpdateMaskList(bool restoreSelection=false);
    unsigned int GetSelectedMask();
    const HuginBase::UIntSet GetSelectedImages();
    void OnImagesColumnWidthChange(wxListEvent& e);
    void OnMaskColumnWidthChange( wxListEvent & e );
    MaskImageCtrl::ImageRotation GetRot(const unsigned int imgNr);
    void DisplayCrop(int imgNr);
    void UpdateImage(size_t imgNr);

    void UpdateCropDisplay();
    // ensure that the crop roi is centered
    void CenterCrop();

    size_t GetImgNr();

    // GUI controls
    MaskImageCtrl *m_editImg;
    wxListCtrl *m_imagesListMask;
    wxListCtrl *m_maskList;
    wxChoice *m_maskType;
    wxNotebook *m_maskCropCtrl;

    // my data
    HuginBase::Panorama * m_pano;
    HuginBase::ConstStandardImageVariableGroups* m_imageGroups=nullptr;
    // current masks vector
    HuginBase::MaskPolygonVector m_currentMasks;
    HuginBase::MaskPolygon::MaskType m_defaultMaskType;
    // mask or crop mode
    bool m_maskMode;
    // the current mask
    unsigned int m_MaskNr;
    // the filename of the current image
    std::string m_File;

    // controls for crop editing
    wxTextCtrl * m_left_textctrl;
    wxTextCtrl * m_right_textctrl;
    wxTextCtrl * m_top_textctrl;
    wxTextCtrl * m_bottom_textctrl;
    wxCheckBox * m_autocenter_cb;
    wxCheckBox * m_cropLens;
    HuginBase::SrcPanoImage::CropMode m_cropMode;
    vigra::Rect2D m_cropRect;
    bool m_autoCenterCrop = true;
    vigra::Point2D m_cropCenter;

    DECLARE_DYNAMIC_CLASS(MaskEditorPanel)
};

class MaskEditorPanelXmlHandler : public wxXmlResourceHandler
{
    DECLARE_DYNAMIC_CLASS(MaskEditorPanelXmlHandler)

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


#endif // _MaskEditorPanel_H