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 = "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();
    // switch between single or multiselectio mode for images list box
    void SetSingleSelection(bool singleSelMode);
 
    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