Code/Sandbox/Plugins/EditorSchematyc2/Editor/AddContainerDlg.cpp

   1 // Copyright 2001-2016 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "AddContainerDlg.h"
   5
   6 #include <CrySchematyc2/IEnvTypeDesc.h>
   7
   8 #include "Resource.h"
   9 #include "PluginUtils.h"
  10
  11 namespace Schematyc2
  12 {
  13         //////////////////////////////////////////////////////////////////////////
  14         CAddContainerDlg::CAddContainerDlg(CWnd* pParent, CPoint pos, TScriptFile& scriptFile, const SGUID& scopeGUID)
  15                 : CDialog(IDD_SCHEMATYC_ADD_CONTAINER, pParent)
  16                 , m_pos(pos)
  17                 , m_scriptFile(scriptFile)
  18                 , m_scopeGUID(scopeGUID)
  19         {}
  20
  21         //////////////////////////////////////////////////////////////////////////
  22         SGUID CAddContainerDlg::GetContainerGUID() const
  23         {
  24                 return m_containerGUID;
  25         }
  26
  27         //////////////////////////////////////////////////////////////////////////
  28         BOOL CAddContainerDlg::OnInitDialog()
  29         {
  30                 SetWindowPos(NULL, m_pos.x, m_pos.y, 0, 0, SWP_NOSIZE | SWP_NOZORDER);
  31                 CDialog::OnInitDialog();
  32
  33                 GetSchematyc()->GetEnvRegistry().VisitTypeDescs(EnvTypeDescVisitor::FromMemberFunction<CAddContainerDlg, &CAddContainerDlg::VisitTypeDesc>(*this));
  34                 ScriptIncludeRecursionUtils::VisitEnumerations(m_scriptFile, ScriptIncludeRecursionUtils::EnumerationVisitor::FromMemberFunction<CAddContainerDlg, &CAddContainerDlg::VisitScriptEnumeration>(*this), SGUID(), true);
  35                 ScriptIncludeRecursionUtils::VisitStructures(m_scriptFile, ScriptIncludeRecursionUtils::StructureVisitor::FromMemberFunction<CAddContainerDlg, &CAddContainerDlg::VisitScriptStructure>(*this), SGUID(), true);
  36                 for(TGUIDStringPairVector::const_iterator iType = m_types.begin(), iEndType = m_types.end(); iType != iEndType; ++ iType)
  37                 {
  38                         m_typesCtrl.AddString(iType->second.c_str());
  39                 }
  40                 m_typesCtrl.SetCurSel(0);
  41
  42                 return true;
  43         }
  44
  45         //////////////////////////////////////////////////////////////////////////
  46         void CAddContainerDlg::DoDataExchange(CDataExchange* pDX)
  47         {
  48                 DDX_Control(pDX, IDC_SCHEMATYC_CONTAINER_TYPES, m_typesCtrl);
  49                 DDX_Control(pDX, IDC_SCHEMATYC_CONTAINER_NAME, m_nameCtrl);
  50                 CDialog::DoDataExchange(pDX);
  51         }
  52
  53         //////////////////////////////////////////////////////////////////////////
  54         void CAddContainerDlg::OnOK()
  55         {
  56                 CString name;
  57                 m_nameCtrl.GetWindowText(name);
  58                 if(PluginUtils::ValidateScriptElementName(CDialog::GetSafeHwnd(), m_scriptFile, m_scopeGUID, name.GetString(), true))
  59                 {
  60                         const IScriptContainer* pContainer = m_scriptFile.AddContainer(m_scopeGUID, name.GetString(), GetSelection());
  61                         if(pContainer)
  62                         {
  63                                 m_containerGUID = pContainer->GetGUID();
  64                         }
  65                         CDialog::OnOK();
  66                 }
  67         }
  68
  69         //////////////////////////////////////////////////////////////////////////
  70         EVisitStatus CAddContainerDlg::VisitTypeDesc(const IEnvTypeDesc& typeDesc)
  71         {
  72                 m_types.push_back(TGUIDStringPair(typeDesc.GetGUID(), typeDesc.GetName()));
  73                 return EVisitStatus::Continue;
  74         }
  75
  76         //////////////////////////////////////////////////////////////////////////
  77         void CAddContainerDlg::VisitScriptEnumeration(const TScriptFile& enumerationFile, const IScriptEnumeration& enumeration)
  78         {
  79                 stack_string    fullName;
  80                 DocUtils::GetFullElementName(enumerationFile, enumeration.GetGUID(), fullName);
  81                 m_types.push_back(TGUIDStringPair(enumeration.GetGUID(), fullName.c_str()));
  82         }
  83
  84         //////////////////////////////////////////////////////////////////////////
  85         void CAddContainerDlg::VisitScriptStructure(const TScriptFile& structureFile, const IScriptStructure& structure)
  86         {
  87                 stack_string    fullName;
  88                 DocUtils::GetFullElementName(structureFile, structure.GetGUID(), fullName);
  89                 m_types.push_back(TGUIDStringPair(structure.GetGUID(), fullName.c_str()));
  90         }
  91
  92         //////////////////////////////////////////////////////////////////////////
  93         SGUID CAddContainerDlg::GetSelection() const
  94         {
  95                 const int       curSel = m_typesCtrl.GetCurSel();
  96                 if((curSel != LB_ERR) && (curSel >= 0))
  97                 {
  98                         return m_types[curSel].first;
  99                 }
 100                 return SGUID();
 101         }
 102 }