Code/Sandbox/Plugins/SchematycEditor/GenericWidgetModel.cpp

   1 // Copyright 2001-2016 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "GenericWidgetModel.h"
   5
   6 #include <Controls/DictionaryWidget.h>
   7
   8 #include <ProxyModels/MergingProxyModel.h>
   9 #include <ProxyModels/ItemModelAttribute.h>
  10
  11 #include <CrySchematyc2/IFramework.h>
  12 #include <CrySchematyc2/Script/IScriptGraph.h>
  13 #include <CrySchematyc2/Script/IScriptRegistry.h>
  14
  15 #include "Util.h"
  16
  17 namespace Cry {
  18 namespace SchematycEd {
  19
  20 //////////////////////////////////////////////////////////////////////////
  21 CGenericWidgetDictionaryEntry::CGenericWidgetDictionaryEntry(CGenericWidgetDictionaryModel& dictionaryModel)
  22         : m_node(Type_Undefined)
  23         , m_dictionaryModel(dictionaryModel)
  24 {
  25 }
  26
  27 //////////////////////////////////////////////////////////////////////////
  28 uint32 CGenericWidgetDictionaryEntry::GetType() const
  29 {
  30         return m_node;
  31 }
  32
  33 //////////////////////////////////////////////////////////////////////////
  34 int32 CGenericWidgetDictionaryEntry::GetNumChildEntries() const
  35 {
  36         return m_childs.size();
  37 }
  38
  39 //////////////////////////////////////////////////////////////////////////
  40 const CAbstractDictionaryEntry* CGenericWidgetDictionaryEntry::GetChildEntry(int32 index) const
  41 {
  42         if ((index >= 0) && (index < m_childs.size()))
  43                 return m_dictionaryModel.GetEntryByGUID(m_childs[index]);
  44
  45         return nullptr;
  46 }
  47
  48 //////////////////////////////////////////////////////////////////////////
  49 const CAbstractDictionaryEntry* CGenericWidgetDictionaryEntry::GetParentEntry() const
  50 {
  51         return m_dictionaryModel.GetEntryByGUID(m_parent);
  52 }
  53
  54 //////////////////////////////////////////////////////////////////////////
  55 QString  CGenericWidgetDictionaryEntry::GetToolTip() const
  56 {
  57         return m_description;
  58 }
  59
  60 //////////////////////////////////////////////////////////////////////////
  61 QString CGenericWidgetDictionaryEntry::GetName() const
  62 {
  63         return m_name;
  64 }
  65
  66 //////////////////////////////////////////////////////////////////////////
  67 QString CGenericWidgetDictionaryEntry::GetFullName() const
  68 {
  69         return m_fullName;
  70 }
  71
  72 //////////////////////////////////////////////////////////////////////////
  73 SGUID CGenericWidgetDictionaryEntry::GetTypeGUID()
  74 {
  75         return m_guid;
  76 }
  77
  78 //////////////////////////////////////////////////////////////////////////
  79 QVariant CGenericWidgetDictionaryEntry::GetColumnValue(int32 columnIndex) const
  80 {
  81         switch (columnIndex)
  82         {
  83                 case CGenericWidgetDictionaryModel::Column_Name:
  84                 {
  85                         return QVariant::fromValue(m_name);
  86                 }
  87                 case CGenericWidgetDictionaryModel::Column_Type:
  88                 {
  89                         return QVariant::fromValue(m_type);
  90                 }
  91                 default:
  92                 {
  93                         break;
  94                 }
  95         }
  96
  97         return QVariant();
  98 }
  99
 100 //////////////////////////////////////////////////////////////////////////
 101 void CGenericWidgetDictionaryModel::ClearScriptElementModel()
 102 {
 103         m_entryGuid.clear();
 104         m_entryPool.clear();
 105         m_entryDict.clear();
 106
 107         m_entryPool.reserve(128);
 108 }
 109
 110 //////////////////////////////////////////////////////////////////////////
 111 CGenericWidgetDictionaryEntry* CGenericWidgetDictionaryModel::BuildScriptElementEntry(const IScriptElement& element, const char* typeName)
 112 {
 113         m_entryPool.emplace_back(*this);
 114         CGenericWidgetDictionaryEntry& entry = m_entryPool.back();
 115
 116         entry.m_name = element.GetName();
 117         entry.m_guid = element.GetGUID();
 118         entry.m_node = CAbstractDictionaryEntry::Type_Entry;
 119         entry.m_type = typeName;
 120
 121         m_entryDict.insert(std::make_pair(entry.m_guid, &entry));
 122         return &entry;
 123 }
 124
 125 void CGenericWidgetDictionaryModel::BuildScriptElementParent(const IScriptElement& element, CGenericWidgetDictionaryEntry* childEntry, const std::vector<EScriptElementType>& parentType)
 126 {
 127         const IScriptElement* parent = element.GetParent();
 128         for (auto& type : parentType)
 129         {
 130                 if (parent && (parent->GetElementType() == type))
 131                 {
 132                         CGenericWidgetDictionaryEntry* parentEntry = GetEntryByGUID(parent->GetGUID());
 133                         if (!parentEntry)
 134                                 break;
 135
 136                         parentEntry->m_childs.emplace_back(childEntry->m_guid);
 137                         parentEntry->m_node = CAbstractDictionaryEntry::Type_Folder;
 138
 139                         childEntry->m_parent = parent->GetGUID();
 140                         return;
 141                 }
 142         }
 143
 144         m_entryGuid.push_back(childEntry->m_guid);
 145         return;
 146 }
 147
 148 //////////////////////////////////////////////////////////////////////////
 149 int32 CGenericWidgetDictionaryModel::GetNumEntries() const
 150 {
 151         return m_entryGuid.size();
 152 }
 153
 154 //////////////////////////////////////////////////////////////////////////
 155 const CAbstractDictionaryEntry* CGenericWidgetDictionaryModel::GetEntry(int32 index) const
 156 {
 157         if (index < m_entryGuid.size())
 158         {
 159                 return GetEntryByGUID(m_entryGuid[index]);
 160         }
 161
 162         return nullptr;
 163 }
 164
 165 //////////////////////////////////////////////////////////////////////////
 166 int32 CGenericWidgetDictionaryModel::GetNumColumns() const
 167 {
 168         return Column_Count;
 169 };
 170
 171 //////////////////////////////////////////////////////////////////////////
 172 QString CGenericWidgetDictionaryModel::GetColumnName(int32 index) const
 173 {
 174         auto pAttribute = CGenericWidgetDictionaryModel::GetColumnAttributeInternal(index);
 175         if (pAttribute)
 176         {
 177                 return QString(pAttribute->GetName());
 178         }
 179
 180         return QString();
 181 }
 182
 183 //////////////////////////////////////////////////////////////////////////
 184 int32 CGenericWidgetDictionaryModel::GetDefaultFilterColumn() const
 185 {
 186         return Column_Type;
 187 }
 188
 189 //////////////////////////////////////////////////////////////////////////
 190 int32 CGenericWidgetDictionaryModel::GetDefaultSortColumn() const
 191 {
 192         return Column_Name;
 193 }
 194
 195 //////////////////////////////////////////////////////////////////////////
 196 const CItemModelAttribute* CGenericWidgetDictionaryModel::GetColumnAttribute(int32 index) const
 197 {
 198         return GetColumnAttributeInternal(index);
 199 }
 200
 201 //////////////////////////////////////////////////////////////////////////
 202 CGenericWidgetDictionaryEntry* CGenericWidgetDictionaryModel::GetEntryByGUID(const Schematyc2::SGUID& guid) const
 203 {
 204         auto it = m_entryDict.find(guid);
 205         if (it != m_entryDict.end())
 206         {
 207                 return it->second;
 208         }
 209
 210         return nullptr;
 211 }
 212
 213 const CItemModelAttribute* CGenericWidgetDictionaryModel::GetColumnAttributeInternal(int column)
 214 {
 215         static CItemModelAttribute Name("Name", eAttributeType_String);
 216         static CItemModelAttribute Type("Type", eAttributeType_String);
 217         static CItemModelAttribute None("    ", eAttributeType_String);
 218
 219         switch (column)
 220         {
 221                 case Column_Name: return &Name;
 222                 case Column_Type: return &Type;
 223         }
 224
 225         return &None;
 226 }
 227
 228 } //namespace SchematycEd
 229 } //namespace Cry