BambuStudio/slic3r/GUI/AuxiliaryDataViewModel.hpp

#ifndef slic3r_GUI_AuxiliaryDataViewModel_hpp_
#define slic3r_GUI_AuxiliaryDataViewModel_hpp_

#include "wx/wxprec.h"
#include "wx/dataview.h"
#include "wx/hashmap.h"
#include "wx/vector.h"

#include "I18N.hpp"

#include <boost/filesystem.hpp>

class AuxiliaryModelNode;
WX_DEFINE_ARRAY_PTR(AuxiliaryModelNode*, AuxiliaryModelNodePtrArray);

namespace fs = boost::filesystem;

class AuxiliaryModelNode
{
public:
    AuxiliaryModelNode()
    {
        m_parent = NULL;
        name = "";
        m_container = true;
        m_root = true;
    }

    AuxiliaryModelNode(AuxiliaryModelNode* parent, const wxString& abs_path, bool is_container)
    {
        m_parent = parent;
        m_container = is_container;
        m_root = false;
        path = abs_path;
        fs::path path_obj(path.c_str());
        name = path_obj.filename().generic_wstring();

        parent->Append(this);
    }

    ~AuxiliaryModelNode()
    {
        // free all our children nodes
        size_t count = m_children.GetCount();
        for (size_t i = 0; i < count; i++)
        {
            AuxiliaryModelNode* child = m_children[i];
            delete child;
        }
    }

    bool IsContainer() const
    {
        return m_container;
    }

    AuxiliaryModelNode* GetParent()
    {
        return m_parent;
    }

    void SetParent(AuxiliaryModelNode* parent)
    {
        m_parent = parent;
    }

    AuxiliaryModelNodePtrArray& GetChildren()
    {
        return m_children;
    }
    AuxiliaryModelNode* GetNthChild(unsigned int n)
    {
        return m_children.Item(n);
    }
    void Insert(AuxiliaryModelNode* child, unsigned int n)
    {
        m_children.Insert(child, n);
    }
    void Append(AuxiliaryModelNode* child)
    {
        m_children.Add(child);
    }
    unsigned int GetChildCount() const
    {
        return m_children.GetCount();
    }

public:
    wxString    name;
    wxString    path;

private:
    AuxiliaryModelNode* m_parent;
    AuxiliaryModelNodePtrArray   m_children;
    bool m_container;
    bool m_root;

};

class AuxiliaryModel : public wxDataViewModel
{
public:
    AuxiliaryModel();
    ~AuxiliaryModel();

    // helper methods to change the model
    wxDataViewItem CreateFolder(wxString name = wxEmptyString);
    wxDataViewItemArray ImportFile(AuxiliaryModelNode* sel, wxArrayString file_paths);
    void Delete(const wxDataViewItem& item);
    void MoveItem(const wxDataViewItem& dropped_item, const wxDataViewItem& dragged_item);
    bool IsOrphan(const wxDataViewItem& item);
    bool Rename(const wxDataViewItem& item, const wxString& name);
    AuxiliaryModelNode* GetParent(AuxiliaryModelNode* node) const;
    void Reparent(AuxiliaryModelNode* node, AuxiliaryModelNode* new_parent);

    void Init(wxString aux_path);
    void Reload(wxString aux_path);

    // override sorting to always sort branches ascendingly

    int Compare(const wxDataViewItem& item1, const wxDataViewItem& item2,
        unsigned int column, bool ascending) const wxOVERRIDE;

    // implementation of base class virtuals to define model

    virtual unsigned int GetColumnCount() const wxOVERRIDE
    {
        return 1;
    }

    virtual wxString GetColumnType(unsigned int col) const wxOVERRIDE
    {
        return "string";
    }

    virtual void GetValue(wxVariant& variant,
        const wxDataViewItem& item, unsigned int col) const wxOVERRIDE;
    virtual bool SetValue(const wxVariant& variant,
        const wxDataViewItem& item, unsigned int col) wxOVERRIDE;

    virtual bool IsEnabled(const wxDataViewItem& item,
        unsigned int col) const wxOVERRIDE;

    virtual wxDataViewItem GetParent(const wxDataViewItem& item) const wxOVERRIDE;
    virtual bool IsContainer(const wxDataViewItem& item) const wxOVERRIDE;
    virtual unsigned int GetChildren(const wxDataViewItem& parent,
        wxDataViewItemArray& array) const wxOVERRIDE;

private:
    AuxiliaryModelNode* m_root;
    wxString m_root_dir;
};

#endif // slic3r_GUI_AuxiliaryDataViewModel_hpp_