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

#ifndef _IMAGESTREE_H
#define _IMAGESTREE_H

#include <map>
#include "PT/Panorama.h"
#include <panodata/StandardImageVariableGroups.h>
#include "treelistctrl.h"
#include "GuiLevel.h"
#include "hugin/PanoOperation.h"

using namespace PT;

class ImagesTreeCtrl: public wxTreeListCtrl, public PT::PanoramaObserver
{
public:
    enum GroupMode
    {
        GROUP_NONE=0,
        GROUP_LENS=1,
        GROUP_STACK=2,
        GROUP_OUTPUTLAYERS=3,
        GROUP_OUTPUTSTACK=4,
    };
    enum DisplayMode
    {
        DISPLAY_GENERAL=0,
        DISPLAY_EXIF=1,
        DISPLAY_POSITION=2,
        DISPLAY_LENS=3,
        DISPLAY_PHOTOMETRICS=4,
        DISPLAY_PHOTOMETRICS_IMAGES=32,
        DISPLAY_PHOTOMETRICS_LENSES=33,
    };
    ImagesTreeCtrl();

    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(PT::Panorama * pano);
   
    virtual ~ImagesTreeCtrl(void) ;

    void SetGroupMode(GroupMode newMode);
    void SetDisplayMode(DisplayMode newMode);

    void SetGuiLevel(GuiLevel newSetting);
    void SetOptimizerMode();
    virtual void panoramaChanged(PT::Panorama & pano);
    void panoramaImagesChanged(PT::Panorama &pano, const PT::UIntSet & imgNr);
    HuginBase::UIntSet GetSelectedImages();

protected:
    void UpdateImageText(wxTreeItemId item);
    void UpdateGroupText(wxTreeItemId item);
    void UpdateGroup(wxTreeItemId parent, const UIntSet imgs, UIntSet& changed);
    void OnColumnWidthChange(wxListEvent & e );
    void OnLinkImageVariables(wxCommandEvent &e);
    void OnUnlinkImageVariables(wxCommandEvent &e);
    void OnEditImageVariables(wxCommandEvent &e);
    void OnBeginDrag(wxTreeEvent &e);
    void OnEndDrag(wxMouseEvent &e);
    void OnLeftDown(wxMouseEvent &e);
    void OnLeftDblClick(wxMouseEvent &e);
    void OnSelectAll(wxCommandEvent &e);
    void OnUnselectAll(wxCommandEvent &e);
    void OnSelectLensStack(wxCommandEvent &e);
    void OnUnselectLensStack(wxCommandEvent &e);
    void OnChar(wxTreeEvent &e);
    void OnBeginEdit(wxTreeEvent &e);
    void OnEndEdit(wxTreeEvent &e);
    void OnExecuteOperation(wxCommandEvent & e);
    void OnContextMenu(wxTreeEvent & e);
    void OnHeaderContextMenu(wxListEvent & e);

private:
    void CreateColumns();
    void UnLinkImageVariables(bool linked);
    void SelectAllParameters(bool select, bool allImages);
    void UpdateOptimizerVariables();
    void GenerateSubMenu(wxMenu* menu, PanoOperation::PanoOperationVector* operations, int& id);

    // the model
    Panorama * m_pano;
    GroupMode m_groupMode;
    
    // image variable group information
    HuginBase::StandardImageVariableGroups * m_variable_groups;

    // number of digits for display
    int m_degDigits;
    int m_distDigits;
    int m_pixelDigits;
    GuiLevel m_guiLevel;
    bool m_optimizerMode;
    DisplayMode m_displayMode;
    std::map<std::string,size_t> m_columnMap;
    std::vector<std::string> m_columnVector;
    HuginBase::UIntSet m_editableColumns;
    std::vector<HuginBase::ImageVariableGroup::ImageVariableEnum> m_variableVector;
    std::map<int,PanoOperation::PanoOperation*> m_menuOperation;
    size_t m_selectedColumn;
    HuginBase::UIntSet m_draggingImages;
    bool m_dragging;
    double m_editVal;
    wxString m_editOldString;
    bool m_needsUpdate;

    wxTreeItemId m_root;

    //for saving column width
    wxString m_configClassName;
    DECLARE_EVENT_TABLE()
    DECLARE_DYNAMIC_CLASS(ImagesTreeCtrl)
};


class ImagesTreeCtrlXmlHandler : public wxTreeListCtrlXmlHandler
{
    DECLARE_DYNAMIC_CLASS(ImagesTreeCtrlXmlHandler)

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

#endif // _IMAGESTREE_H

---