Code/Sandbox/Plugins/EditorCommon/NodeGraph/NodeGraphViewStyle.cpp

   1 // Copyright 2001-2018 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "stdafx.h"
   4 #include "NodeGraphViewStyle.h"
   5
   6 #include "NodeWidgetStyle.h"
   7 #include "GroupWidgetStyle.h"
   8 #include "CommentWidgetStyle.h"
   9 #include "ConnectionWidgetStyle.h"
  10 #include "NodePinWidgetStyle.h"
  11
  12 #include "NodeGraphView.h"
  13
  14 namespace CryGraphEditor {
  15
  16 CNodeGraphViewStyle::CNodeGraphViewStyle(const char* szStyleId)
  17         : QWidget()
  18 {
  19         setObjectName(szStyleId);
  20
  21         m_selectionColor = QColor(97, 172, 237);
  22         m_highlightColor = QColor(140, 140, 140);
  23
  24         m_gridBackgroundColor = QColor(0x5F, 0x5F, 0x5F);
  25         m_gridSegmentLineColor = QColor(0x46, 0x46, 0x46);
  26         m_gridSegmentLineWidth = 1.f;
  27         m_gridSegmentSize = 140.f;
  28
  29         m_gridSubSegmentLineColor = QColor(0x70, 0x70, 0x70);
  30         m_gridSubSegmentCount = 8;
  31         m_gridSubSegmentLineWidth = 1.f;
  32
  33         CNodeWidgetStyle* pNodeStyle = new CNodeWidgetStyle("Node", *this);
  34         CConnectionWidgetStyle* pConnectionStyle = new CConnectionWidgetStyle("Connection", *this);
  35         CNodePinWidgetStyle* pPinStyle = new CNodePinWidgetStyle("Pin", *this);
  36 }
  37
  38 void CNodeGraphViewStyle::RegisterNodeWidgetStyle(CNodeWidgetStyle* pStyle)
  39 {
  40         CRY_ASSERT_MESSAGE(pStyle->GetId(), "StyleId must be non-zero!");
  41         if (pStyle->GetId())
  42         {
  43                 const uint32 styleIdHash = pStyle->GetIdHash();
  44                 const bool iconExists = (m_nodeWidgetStylesById.find(styleIdHash) != m_nodeWidgetStylesById.end());
  45
  46                 if (iconExists == false)
  47                 {
  48                         pStyle->SetParent(this);
  49                         m_nodeWidgetStylesById[styleIdHash] = pStyle;
  50                 }
  51                 else
  52                 {
  53                         auto result = m_nodeWidgetStylesById.find(styleIdHash);
  54                         const stack_string resultStyleId = result->second->GetId();
  55                         if (pStyle->GetId() == resultStyleId)
  56                         {
  57                                 //CRY_ASSERT_MESSAGE(false, "Style id already exists.");
  58                         }
  59                         else
  60                         {
  61                                 CRY_ASSERT_MESSAGE(false, "Hash collison of style id '%s' and '%s'", pStyle->GetId(), resultStyleId.c_str());
  62                         }
  63                 }
  64         }
  65 }
  66
  67 const CNodeWidgetStyle* CNodeGraphViewStyle::GetNodeWidgetStyle(const char* styleId) const
  68 {
  69         CRY_ASSERT(styleId);
  70         const StyleIdHash styleIdHash = CCrc32::Compute(styleId);
  71         auto result = m_nodeWidgetStylesById.find(styleIdHash);
  72         if (result != m_nodeWidgetStylesById.end())
  73         {
  74                 return result->second;
  75         }
  76         return nullptr;
  77 }
  78
  79 void CNodeGraphViewStyle::RegisterGroupWidgetStyle(CGroupWidgetStyle* pStyle)
  80 {
  81         CRY_ASSERT_MESSAGE(pStyle->GetId(), "StyleId must be non-zero!");
  82         if (pStyle->GetId())
  83         {
  84                 const uint32 styleIdHash = pStyle->GetIdHash();
  85                 const bool iconExists = (m_groupWidgetStylesById.find(styleIdHash) != m_groupWidgetStylesById.end());
  86
  87                 if (iconExists == false)
  88                 {
  89                         pStyle->SetParent(this);
  90                         m_groupWidgetStylesById[styleIdHash] = pStyle;
  91                 }
  92                 else
  93                 {
  94                         auto result = m_groupWidgetStylesById.find(styleIdHash);
  95                         const stack_string resultStyleId = result->second->GetId();
  96                         if (pStyle->GetId() == resultStyleId)
  97                         {
  98                                 CRY_ASSERT_MESSAGE(false, "Style id already exists.");
  99                         }
 100                         else
 101                         {
 102                                 CRY_ASSERT_MESSAGE(false, "Hash collison of style id '%s' and '%s'", pStyle->GetId(), resultStyleId.c_str());
 103                         }
 104                 }
 105         }
 106 }
 107
 108 const CGroupWidgetStyle* CNodeGraphViewStyle::GetGroupWidgetStyle(const char* styleId) const
 109 {
 110         CRY_ASSERT_MESSAGE(styleId, "StyleId must be non-zero!");
 111         const StyleIdHash styleIdHash = CCrc32::Compute(styleId);
 112         auto result = m_groupWidgetStylesById.find(styleIdHash);
 113         if (result != m_groupWidgetStylesById.end())
 114         {
 115                 return result->second;
 116         }
 117         return nullptr;
 118 }
 119
 120 void CNodeGraphViewStyle::RegisterCommentWidgetStyle(CCommentWidgetStyle* pStyle)
 121 {
 122         CRY_ASSERT_MESSAGE(pStyle->GetId(), "StyleId must be non-zero!");
 123         if (pStyle->GetId())
 124         {
 125                 const uint32 styleIdHash = pStyle->GetIdHash();
 126                 const bool iconExists = (m_commentWidgetStylesById.find(styleIdHash) != m_commentWidgetStylesById.end());
 127
 128                 if (iconExists == false)
 129                 {
 130                         pStyle->SetParent(this);
 131                         m_commentWidgetStylesById[styleIdHash] = pStyle;
 132                 }
 133                 else
 134                 {
 135                         auto result = m_commentWidgetStylesById.find(styleIdHash);
 136                         const stack_string resultStyleId = result->second->GetId();
 137                         if (pStyle->GetId() == resultStyleId)
 138                         {
 139                                 //CRY_ASSERT_MESSAGE(false, "Style id already exists.");
 140                         }
 141                         else
 142                         {
 143                                 CRY_ASSERT_MESSAGE(false, "Hash collison of style id '%s' and '%s'", pStyle->GetId(), resultStyleId.c_str());
 144                         }
 145                 }
 146         }
 147 }
 148
 149 const CCommentWidgetStyle* CNodeGraphViewStyle::GetCommentWidgetStyle(const char* styleId) const
 150 {
 151         CRY_ASSERT_MESSAGE(styleId, "StyleId must be non-zero!");
 152         const StyleIdHash styleIdHash = CCrc32::Compute(styleId);
 153         auto result = m_commentWidgetStylesById.find(styleIdHash);
 154         if (result != m_commentWidgetStylesById.end())
 155         {
 156                 return result->second;
 157         }
 158         return nullptr;
 159 }
 160
 161 void CNodeGraphViewStyle::RegisterConnectionWidgetStyle(CConnectionWidgetStyle* pStyle)
 162 {
 163         CRY_ASSERT_MESSAGE(pStyle->GetId(), "StyleId must be non-zero!");
 164         if (pStyle->GetId())
 165         {
 166                 const StyleIdHash styleIdHash = pStyle->GetIdHash();
 167                 const bool iconExists = (m_connectionWidgetStylesById.find(styleIdHash) != m_connectionWidgetStylesById.end());
 168
 169                 if (iconExists == false)
 170                 {
 171                         pStyle->SetParent(this);
 172                         m_connectionWidgetStylesById[styleIdHash] = pStyle;
 173                 }
 174                 else
 175                 {
 176                         auto result = m_connectionWidgetStylesById.find(styleIdHash);
 177                         const stack_string resultStyleId = result->second->GetId();
 178                         if (pStyle->GetId() == resultStyleId)
 179                         {
 180                                 CRY_ASSERT_MESSAGE(false, "Style id already exists.");
 181                         }
 182                         else
 183                         {
 184                                 CRY_ASSERT_MESSAGE(false, "Hash collison of style id '%s' and '%s'", pStyle->GetId(), resultStyleId.c_str());
 185                         }
 186                 }
 187         }
 188 }
 189
 190 const CConnectionWidgetStyle* CNodeGraphViewStyle::GetConnectionWidgetStyle(const char* styleId) const
 191 {
 192         CRY_ASSERT(styleId);
 193         const StyleIdHash styleIdHash = CCrc32::Compute(styleId);
 194         auto result = m_connectionWidgetStylesById.find(styleIdHash);
 195         if (result != m_connectionWidgetStylesById.end())
 196         {
 197                 return result->second;
 198         }
 199         return nullptr;
 200 }
 201
 202 void CNodeGraphViewStyle::RegisterPinWidgetStyle(CNodePinWidgetStyle* pStyle)
 203 {
 204         CRY_ASSERT_MESSAGE(pStyle->GetId(), "StyleId must be non-zero!");
 205         if (pStyle->GetId())
 206         {
 207                 const StyleIdHash styleIdHash = pStyle->GetIdHash();
 208                 const bool iconExists = (m_pinWidgetStylesById.find(styleIdHash) != m_pinWidgetStylesById.end());
 209
 210                 if (iconExists == false)
 211                 {
 212                         pStyle->SetParent(this);
 213                         m_pinWidgetStylesById[styleIdHash] = pStyle;
 214                 }
 215                 else
 216                 {
 217                         auto result = m_pinWidgetStylesById.find(styleIdHash);
 218                         const stack_string resultStyleId = result->second->GetId();
 219                         if (pStyle->GetId() == resultStyleId)
 220                         {
 221                                 CRY_ASSERT_MESSAGE(false, "Style id already exists.");
 222                         }
 223                         else
 224                         {
 225                                 CRY_ASSERT_MESSAGE(false, "Hash collison of style id '%s' and '%s'", pStyle->GetId(), resultStyleId.c_str());
 226                         }
 227                 }
 228         }
 229 }
 230
 231 const CNodePinWidgetStyle* CNodeGraphViewStyle::GetPinWidgetStyle(const char* styleId) const
 232 {
 233         CRY_ASSERT(styleId);
 234         const StyleIdHash styleIdHash = CCrc32::Compute(styleId);
 235         auto result = m_pinWidgetStylesById.find(styleIdHash);
 236         if (result != m_pinWidgetStylesById.end())
 237         {
 238                 return result->second;
 239         }
 240         return nullptr;
 241 }
 242
 243 }