MaterialPropTreeView.cpp

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.

===========================================================================
*/
#include "../../idlib/precompiled.h"
#pragma hdrstop

#include "MaterialPropTreeView.h"

#define PROP_TREE_VIEW "PropTreeView"


IMPLEMENT_DYNCREATE(MaterialPropTreeView, CPropTreeView)

BEGIN_MESSAGE_MAP(MaterialPropTreeView, CPropTreeView)
        ON_NOTIFY( PTN_ITEMCHANGED, IDC_PROPERTYTREE, OnPropertyChangeNotification )
        ON_NOTIFY( PTN_ITEMEXPANDING, IDC_PROPERTYTREE, OnPropertyItemExpanding )
END_MESSAGE_MAP()


MaterialPropTreeView::MaterialPropTreeView() {
        registry.Init("Software\\id Software\\DOOM3\\Tools\\MaterialEditor\\PropertySettings");
        internalChange = false;
}

MaterialPropTreeView::~MaterialPropTreeView() {
}

void MaterialPropTreeView::SetPropertyListType(int listType, int stageNum) {

        currentListType = listType;
        currentStage = stageNum;

        m_Tree.DeleteAllItems();

        //idList<MaterialProp_t*>* propList = NULL;
        MaterialDefList* propList = MaterialDefManager::GetMaterialDefs(currentListType);
        currentPropDefs = propList;

        if(!propList)
                return;

        CPropTreeItem* pCurrentGroup = NULL;
        CPropTreeItem* pCurrentNode = NULL;

        for(int i = 0; i < propList->Num(); i++) {
                switch((*propList)[i]->type) {
                        case MaterialDef::MATERIAL_DEF_TYPE_GROUP:
                                {
                                        pCurrentGroup = m_Tree.InsertItem(new CPropTreeItem());
                                        pCurrentNode = pCurrentGroup;

                                        if(!registry.GetBool(va("Expand%d%s", currentListType, (*propList)[i]->displayName.c_str())))
                                                pCurrentGroup->Expand();
                                }
                                break;
                        case MaterialDef::MATERIAL_DEF_TYPE_BOOL:
                                {
                                        CPropTreeItemCheck* pCheck;
                                        pCheck = (CPropTreeItemCheck*)m_Tree.InsertItem(new CPropTreeItemCheck(), pCurrentGroup);
                                        pCheck->CreateCheckBox();
                                        pCurrentNode = pCheck;
                                }
                                break;
                        case MaterialDef::MATERIAL_DEF_TYPE_STRING:
                                {
                                        //pCurrentNode = m_Tree.InsertItem(new CPropTreeItemEdit(), pCurrentGroup);     
                                        pCurrentNode = m_Tree.InsertItem(new CPropTreeItemFileEdit(), pCurrentGroup);

                                }
                                break;
                }

                if(pCurrentNode) {
                        (*propList)[i]->SetViewData(PROP_TREE_VIEW, pCurrentNode->GetCtrlID());
                        pCurrentNode->SetLabelText((*propList)[i]->displayName);
                        pCurrentNode->SetInfoText((*propList)[i]->displayInfo);
                }
        }

        RefreshProperties();
}

void MaterialPropTreeView::LoadSettings() {
        registry.Load();
}

void MaterialPropTreeView::SaveSettings() {
        registry.Save();
}

void MaterialPropTreeView::MV_OnMaterialChange(MaterialDoc* pMaterial) {
        
        if(materialDocManager->GetCurrentMaterialDoc()) {
                idStr currentName = materialDocManager->GetCurrentMaterialDoc()->name;
                if(!internalChange && !pMaterial->name.Icmp(currentName)) {
                        RefreshProperties();
                }
        }
}

void MaterialPropTreeView::OnPropertyChangeNotification( NMHDR *nmhdr, LRESULT *lresult ) {

        NMPROPTREE      *nmProp = (NMPROPTREE *)nmhdr;
        CPropTreeItem   *item = nmProp->pItem;

        internalChange = true;

        MaterialDef* propItem = FindDefForTreeID(item->GetCtrlID());
        if(propItem) {
                MaterialDoc* materialDoc = materialDocManager->GetCurrentMaterialDoc();

                switch(propItem->type) {
                        case MaterialDef::MATERIAL_DEF_TYPE_BOOL:
                                {
                                        BOOL val = item->GetItemValue();
                                        materialDoc->SetAttributeBool(currentStage, propItem->dictName, val ? true : false);
                                }
                                break;
                        case MaterialDef::MATERIAL_DEF_TYPE_STRING:
                                {
                                        idStr val = (LPCTSTR)item->GetItemValue();                                                              
                                        materialDoc->SetAttribute(currentStage, propItem->dictName, val);
                                }
                                break;
                }
        }

        internalChange = false;

        *lresult = 0;
}

void MaterialPropTreeView::OnPropertyItemExpanding( NMHDR *nmhdr, LRESULT *lresult ) {

        NMPROPTREE      *nmProp = (NMPROPTREE *)nmhdr;
        CPropTreeItem   *item = nmProp->pItem;

        //The item isn't toggled till after this returns so use the opposite of the current state.
        registry.SetBool(va("Expand%d%s", currentListType, item->GetLabelText()), item->IsExpanded() ? true : false);
        registry.Save();

        *lresult = 0;
}

MaterialDef* MaterialPropTreeView::FindDefForTreeID(UINT treeID) {

        int c = currentPropDefs->Num();
        for(int i = 0; i < c; i++) {
                if((*currentPropDefs)[i]->GetViewData(PROP_TREE_VIEW) == treeID)
                        return (*currentPropDefs)[i];
        }

        return NULL;
}

void MaterialPropTreeView::RefreshProperties() {
        
        MaterialDefList* propList = MaterialDefManager::GetMaterialDefs(currentListType);

        if(!propList)
                return;

        MaterialDoc* materialDoc = materialDocManager->GetCurrentMaterialDoc();

        for(int i = 0; i < propList->Num(); i++) {
                switch((*propList)[i]->type) {
                        case MaterialDef::MATERIAL_DEF_TYPE_BOOL:
                                {
                                        bool val = materialDoc->GetAttributeBool(currentStage, (*propList)[i]->dictName);
                                        CPropTreeItemCheck* item = (CPropTreeItemCheck*)m_Tree.FindItem((*propList)[i]->GetViewData(PROP_TREE_VIEW));
                                        item->SetCheckState(val ? TRUE:FALSE);
                                }
                                break;
                        case MaterialDef::MATERIAL_DEF_TYPE_STRING:
                                {
                                        idStr val = materialDoc->GetAttribute(currentStage, (*propList)[i]->dictName);
                                        CPropTreeItemEdit* item = (CPropTreeItemEdit*)m_Tree.FindItem((*propList)[i]->GetViewData(PROP_TREE_VIEW));
                                        item->SetItemValue((LPARAM)val.c_str());
                                }
                                break;
                }
        }

        Invalidate();
}