MaterialTreeView.h

Go to the documentation of this file.

/*
===========================================================================

Doom 3 GPL Source Code
Copyright (C) 1999-2011 id Software LLC, a ZeniMax Media company. 

This file is part of the Doom 3 GPL Source Code (?Doom 3 Source Code?).  

Doom 3 Source Code 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 3 of the License, or
(at your option) any later version.

Doom 3 Source Code 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 General Public License for more details.

You should have received a copy of the GNU General Public License
along with Doom 3 Source Code.  If not, see <http://www.gnu.org/licenses/>.

In addition, the Doom 3 Source Code is also subject to certain additional terms. You should have received a copy of these additional terms immediately following the terms and conditions of the GNU General Public License which accompanied the Doom 3 Source Code.  If not, please request a copy in writing from id Software at the address below.

If you have questions concerning this license or the applicable additional terms, you may contact in writing id Software LLC, c/o ZeniMax Media Inc., Suite 120, Rockville, Maryland 20850 USA.

===========================================================================
*/
#pragma once

#include <afxcview.h>
#include "../common/PropTree/PropTreeView.h"
#include "MaterialEditor.h"
#include "MaterialView.h"

typedef struct {
        idStr           materialName;
        HTREEITEM       treeItem;
} MaterialTreeItem_t;

class MaterialTreeView : public CTreeView, public MaterialView {

public:
        virtual                 ~MaterialTreeView();

        void                    InitializeMaterialList(bool includeFile = true, const char* filename = NULL);
        void                    BuildMaterialList(bool includeFile = true, const char* filename = NULL);
        
        //Material Interface
        virtual void    MV_OnMaterialChange(MaterialDoc* pMaterial);
        virtual void    MV_OnMaterialApply(MaterialDoc* pMaterial);
        virtual void    MV_OnMaterialSaved(MaterialDoc* pMaterial);
        virtual void    MV_OnMaterialAdd(MaterialDoc* pMaterial);
        virtual void    MV_OnMaterialDelete(MaterialDoc* pMaterial);
        virtual void    MV_OnMaterialNameChanged(MaterialDoc* pMaterial, const char* oldName);
        virtual void    MV_OnFileReload(const char* filename);
        
        bool                    CanCopy();
        bool                    CanPaste();
        bool                    CanCut();
        bool                    CanDelete();
        bool                    CanRename();
        bool                    CanSaveFile();
        idStr                   GetSaveFilename();

        bool                    FindNextMaterial(MaterialSearchData_t* searchData);
        HTREEITEM               FindNextMaterial(HTREEITEM item, MaterialSearchData_t* searchData);
        HTREEITEM               GetNextSeachItem(HTREEITEM item, bool stayInFile);

        void                    DeleteFolder(HTREEITEM item, bool addUndo = true);
        HTREEITEM               AddFolder(const char* name, HTREEITEM parent);
        void                    RenameFolder(HTREEITEM item, const char* name);
        

protected:
        MaterialTreeView();
        DECLARE_DYNCREATE(MaterialTreeView)

        
        enum {
                TYPE_ROOT = 0,
                TYPE_FOLDER,
                TYPE_FILE,
                TYPE_MATERIAL_FOLDER,
                TYPE_MATERIAL
        };

        //Overrides
        virtual BOOL    PreTranslateMessage(MSG* pMsg);

        //Window Messages
        afx_msg int             OnCreate(LPCREATESTRUCT lpCreateStruct);
        afx_msg void    OnTvnSelchanged(NMHDR *pNMHDR, LRESULT *pResult);
        afx_msg void    OnTvnBeginlabeledit(NMHDR *pNMHDR, LRESULT *pResult);
        afx_msg void    OnTvnEndlabeledit(NMHDR *pNMHDR, LRESULT *pResult);
        afx_msg void    OnContextMenu(CWnd* pWnd, CPoint point);
        afx_msg void    OnNMRclick(NMHDR *pNMHDR, LRESULT *pResult);
        afx_msg void    OnChar(UINT nChar, UINT nRepCnt, UINT nFlags);
        afx_msg void    OnTvnBegindrag(NMHDR *pNMHDR, LRESULT *pResult);
        afx_msg void    OnMouseMove(UINT nFlags, CPoint point);
        afx_msg void    OnLButtonUp(UINT nFlags, CPoint point);

        //Menu Commands
        afx_msg void    OnApplyMaterial();
        afx_msg void    OnApplyFile();
        afx_msg void    OnApplyAll();
        afx_msg void    OnSaveMaterial();
        afx_msg void    OnSaveFile();
        afx_msg void    OnSaveAll();
        afx_msg void    OnRenameMaterial();
        afx_msg void    OnAddMaterial();
        afx_msg void    OnAddFolder();
        afx_msg void    OnDeleteMaterial();
        afx_msg void    OnReloadFile();
        afx_msg void    OnCut();
        afx_msg void    OnCopy();
        afx_msg void    OnPaste();

        //Internal Messages
        afx_msg LRESULT OnRenameFolderComplete(WPARAM wParam, LPARAM lParam);
        afx_msg LRESULT OnRenameMaterialComplete(WPARAM wParam, LPARAM lParam);

        DECLARE_MESSAGE_MAP()
        

        //Utility methods
        void                    RenameMaterial(HTREEITEM item, const char* originalName);
        bool                    GetFileName(HTREEITEM item, idStr& out);
        idStr                   GetMediaPath(HTREEITEM item, DWORD type);
        void                    GetMaterialPaths(HTREEITEM item, idList<MaterialTreeItem_t>* list);
        void                    AddStrList(const char *root, idStrList *list, bool includeFile);
        void                    PopupMenu(CPoint* pt);
        void                    SetItemImage(HTREEITEM item, bool mod, bool apply, bool children);
        

        //Methods for working with the quicktree
        void                    CleanLookupTrees(HTREEITEM item);
        void                    BuildLookupTrees(HTREEITEM item);
        idStr                   GetQuicktreePath(HTREEITEM item);


protected:
        CImageList                              m_image;
        bool                                    treeWithFile;

        //Hashtables for quick lookups
        idHashTable<HTREEITEM>  quickTree;
        idHashTable<HTREEITEM>  materialToTree;
        idHashTable<HTREEITEM>  fileToTree;


        //Member variables for renaming folders
        HTREEITEM                               renamedFolder;
        idList<MaterialTreeItem_t> affectedMaterials;

        CImageList*                             dragImage;
        bool                                    bDragging;
        CPoint                                  dropPoint;
        HTREEITEM                               dragItem;

        //Hover Expand
        HTREEITEM                               hoverItem;
        DWORD                                   hoverStartTime;

        bool                                    internalChange;
};