Code/Sandbox/Plugins/EditorSchematyc2/Bridge/Private/Bridge_LibUtils.cpp

   1 // Copyright 2001-2016 Crytek GmbH / Crytek Group. All rights reserved.
   2
   3 #include "StdAfx.h"
   4 #include "Bridge_LibUtils.h"
   5
   6 #include "CrySchematyc2/LibUtils.h"
   7
   8 namespace Schematyc2 {
   9
  10 namespace LibUtils {
  11
  12 void FindReferences(SGUID refGuid, SGUID refGoBack, const char* szItemName, const TScriptFile* pFile, SReferencesLogInfo& result)
  13 {
  14         result.header.Format("Searching references of %s in %s", szItemName, pFile->GetFileName());
  15
  16         auto visitAbstractInterfaces = [refGuid, refGoBack, szItemName, &result](const IScriptAbstractInterfaceImplementation& abstractInterface)->EVisitStatus
  17         {
  18                 std::stack<string> path;
  19                 if (abstractInterface.GetRefGUID() == refGuid)
  20                 {
  21                         stack_string graphPath;
  22                         graphPath.append("<font color = \"yellow\">");
  23                         const IScriptElement *pParent = abstractInterface.GetParent();
  24                         while (pParent)
  25                         {
  26                                 path.push(pParent->GetName());
  27                                 pParent = pParent->GetParent();
  28                         }
  29
  30                         while (!path.empty())
  31                         {
  32                                 graphPath.append(path.top().c_str());
  33                                 graphPath.append("<font color = \"black\">");
  34                                 graphPath.append("//");
  35                                 graphPath.append("<font color = \"yellow\">");
  36                                 path.pop();
  37                         }
  38
  39                         graphPath.append(abstractInterface.GetName());
  40
  41                         SLogMetaItemGUID gotoItem(abstractInterface.GetGUID());
  42                         SLogMetaChildGUID gobackItem(refGoBack);
  43
  44                         stack_string pathLink;
  45                         pathLink.Format("Reference of <font color = \"yellow\">%s <font color = \"lightgray\">found in %s", szItemName, graphPath.c_str());
  46
  47                         result.references.emplace_back(gotoItem, gobackItem, pathLink.c_str());
  48                 }
  49
  50                 return EVisitStatus::Continue;
  51         };
  52
  53         auto visitGraph = [refGuid, refGoBack, szItemName, &result](const IDocGraph& graph) -> EVisitStatus
  54         {
  55                 auto visitNode = [refGuid, refGoBack, &graph, szItemName, &result](const IScriptGraphNode& node) -> EVisitStatus
  56                 {
  57                         bool foundInOutput(false);
  58                         for (size_t i = 0, count = node.GetOutputCount(); i < count; ++i)
  59                         {
  60                                 CAggregateTypeId typeId = node.GetOutputTypeId(i);
  61                                 if (typeId.AsScriptTypeGUID() == refGuid)
  62                                 {
  63                                         foundInOutput = true;
  64                                         break;
  65                                 }
  66                         }
  67
  68                         std::stack<string> path;
  69                         if (node.GetRefGUID() == refGuid || graph.GetContextGUID() == refGuid || foundInOutput)
  70                         {
  71                                 stack_string graphPath;
  72                                 graphPath.append("<font color = \"yellow\">");
  73                                 const IScriptElement *parent = graph.GetParent();
  74                                 while (parent)
  75                                 {
  76                                         path.push(parent->GetName());
  77                                         parent = parent->GetParent();
  78                                 }
  79
  80                                 while (!path.empty())
  81                                 {
  82                                         graphPath.append(path.top().c_str());
  83                                         graphPath.append("<font color = \"black\">");
  84                                         graphPath.append("//");
  85                                         graphPath.append("<font color = \"yellow\">");
  86                                         path.pop();
  87                                 }
  88
  89                                 graphPath.append(graph.GetName());
  90
  91                                 SLogMetaItemGUID gotoItem(graph.GetType() == EScriptGraphType::Transition ? graph.GetScopeGUID() : graph.GetGUID());
  92                                 SLogMetaChildGUID gobackItem(refGoBack);
  93
  94                                 stack_string pathLink;
  95                                 pathLink.Format("Reference of <font color = \"yellow\">%s <font color = \"lightgray\">found in %s", szItemName, graphPath.c_str());
  96
  97                                 result.references.emplace_back(gotoItem, gobackItem, pathLink.c_str());
  98                         }
  99
 100                         return EVisitStatus::Continue;
 101                 };
 102
 103                 graph.VisitNodes(ScriptGraphNodeConstVisitor::FromLambdaFunction(visitNode));
 104                 return EVisitStatus::Continue;
 105         };
 106
 107         auto visitActions = [refGuid, refGoBack, szItemName, &result](const IScriptActionInstance& action)->EVisitStatus
 108         {
 109                 std::stack<string> path;
 110                 if (action.GetActionGUID() == refGuid)
 111                 {
 112                         stack_string graphPath;
 113                         graphPath.append("<font color = \"yellow\">");
 114                         const IScriptElement *parent = action.GetParent();
 115                         while (parent)
 116                         {
 117                                 path.push(parent->GetName());
 118                                 parent = parent->GetParent();
 119                         }
 120
 121                         while (!path.empty())
 122                         {
 123                                 graphPath.append(path.top().c_str());
 124                                 graphPath.append("<font color = \"black\">");
 125                                 graphPath.append("//");
 126                                 graphPath.append("<font color = \"yellow\">");
 127                                 path.pop();
 128                         }
 129
 130                         graphPath.append(action.GetName());
 131
 132                         SLogMetaItemGUID gotoItem(action.GetGUID());
 133                         SLogMetaChildGUID gobackItem(refGoBack);
 134
 135                         stack_string pathLink;
 136                         pathLink.Format("Reference of <font color = \"yellow\">%s <font color = \"lightgray\">found in %s", szItemName, graphPath.c_str());
 137
 138                         result.references.emplace_back(gotoItem, gobackItem, pathLink.c_str());
 139                 }
 140
 141                 return EVisitStatus::Continue;
 142         };
 143
 144         auto visitComponentInstances = [refGuid, refGoBack, szItemName, &result](const IScriptComponentInstance& componentInstance) -> EVisitStatus
 145         {
 146                 if (const IScriptGraphExtension* pGraph = static_cast<const IScriptGraphExtension*>(componentInstance.GetExtensions().QueryExtension(Graph)))
 147                 {
 148                         const IScriptElement& graph = pGraph->GetElement_New();
 149
 150                         auto visitNode = [refGuid, refGoBack, &graph, szItemName, &result, &componentInstance](const IScriptGraphNode& node) -> EVisitStatus
 151                         {
 152                                 std::stack<string> path;
 153                                 if (node.GetRefGUID() == refGuid)
 154                                 {
 155                                         stack_string graphPath;
 156                                         graphPath.append("<font color = \"yellow\">");
 157                                         const IScriptElement *parent = graph.GetParent();
 158                                         while (parent)
 159                                         {
 160                                                 path.push(parent->GetName());
 161                                                 parent = parent->GetParent();
 162                                         }
 163
 164                                         while (!path.empty())
 165                                         {
 166                                                 graphPath.append(path.top().c_str());
 167                                                 graphPath.append("<font color = \"black\">");
 168                                                 graphPath.append("//");
 169                                                 graphPath.append("<font color = \"yellow\">");
 170                                                 path.pop();
 171                                         }
 172
 173                                         graphPath.append(graph.GetName());
 174
 175                                         SLogMetaItemGUID gotoItem(componentInstance.GetGUID());
 176                                         SLogMetaChildGUID gobackItem(refGoBack);
 177
 178                                         stack_string pathLink;
 179                                         pathLink.Format("Reference of <font color = \"yellow\">%s <font color = \"lightgray\">found in %s", szItemName, graphPath.c_str());
 180
 181                                         result.references.emplace_back(gotoItem, gobackItem, pathLink.c_str());
 182                                 }
 183
 184                                 return EVisitStatus::Continue;
 185                         };
 186
 187                         pGraph->VisitNodes(ScriptGraphNodeConstVisitor::FromLambdaFunction(visitNode));
 188                 }
 189
 190                 return EVisitStatus::Continue;
 191         };
 192
 193         pFile->VisitGraphs(DocGraphConstVisitor::FromLambdaFunction(visitGraph), SGUID(), true);
 194         pFile->VisitComponentInstances(ScriptComponentInstanceConstVisitor::FromLambdaFunction(visitComponentInstances), SGUID(), true);
 195         pFile->VisitActionInstances(ScriptActionInstanceConstVisitor::FromLambdaFunction(visitActions), SGUID(), true);
 196         pFile->VisitAbstractInterfaceImplementations(ScriptAbstractInterfaceImplementationConstVisitor::FromLambdaFunction(visitAbstractInterfaces), SGUID(), true);
 197
 198         result.footer.Format("References found: %d", result.GetReferenceCount());
 199 }
 200
 201 void FindAndLogReferences(SGUID refGuid, SGUID refGoBack, const char* szItemName, const TScriptFile* pFile, bool searchInAllFiles)
 202 {
 203         // Check if the refGuid is actually a action and assign the proper search guid
 204         if (const IScriptActionInstance* pAction = pFile->GetActionInstance(refGuid))
 205         {
 206                 refGuid = pAction->GetActionGUID();
 207         }
 208
 209         if (searchInAllFiles)
 210         {
 211                 auto visitFiles = [refGuid, refGoBack, szItemName](const TScriptFile& file)->EVisitStatus
 212                 {
 213                         SReferencesLogInfo refInfo;
 214                         FindReferences(refGuid, refGoBack, szItemName, &file, refInfo);
 215                         if (!refInfo.IsEmpty())
 216                         {
 217                                 PrintReferencesToSchematycConsole(refInfo);
 218                         }
 219                         return EVisitStatus::Continue;
 220                 };
 221
 222                 GetSchematyc()->GetScriptRegistry().VisitFiles(TScriptFileConstVisitor::FromLambdaFunction(visitFiles));
 223         }
 224         else
 225         {
 226                 SReferencesLogInfo refInfo;
 227                 FindReferences(refGuid, refGoBack, szItemName, pFile, refInfo);
 228                 PrintReferencesToSchematycConsole(refInfo);
 229         }
 230 }
 231
 232 } // namespace LibUtils
 233
 234 } // namespace Schematyc