gitweb.ps.run Git - toc/blob - src/find.h

   1 #pragma once\r
   2 \r
   3 #include "repr.h"\r
   4 #include "typeInfo.h"\r
   5 \r
   6 #include <functional>\r
   7 #include <optional>\r
   8 \r
   9 template<typename T>\r
  10 using opt = std::optional<T>;\r
  11 \r
  12 template<typename T>\r
  13 opt<T> find(const std::vector<T> & ts, std::function<bool(T)> f)\r
  14 {\r
  15   for (auto t : ts)\r
  16     if (f(t))\r
  17       return t;\r
  18   return nullopt;\r
  19 }\r
  20 \r
  21 opt<Function> findFunction(\r
  22   const Program & p,\r
  23   const std::string & name,\r
  24   const std::vector<std::string> & namespacePrefixes)\r
  25 {\r
  26   if (namespacePrefixes.empty())\r
  27   {\r
  28     return find<Function>(p.functions, [&](Function f) { return f.name == name; });\r
  29   }\r
  30   \r
  31   auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });\r
  32 \r
  33   if (!n.has_value())\r
  34     return nullopt;\r
  35 \r
  36   std::vector<Namespace> namespaces = { n.value() };\r
  37 \r
  38   for (int i = 1; i < namespacePrefixes.size(); i++)\r
  39   {\r
  40     n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });\r
  41     \r
  42     if (!n.has_value())\r
  43       return nullopt;\r
  44 \r
  45     namespaces.push_back(n.value());\r
  46   }\r
  47 \r
  48   for (int i = namespaces.size()-1; i >= 0; i--)\r
  49   {\r
  50     auto f = find<Function>(namespaces[i].functions, [&](Function f) { return f.name == name; });\r
  51     if (f.has_value())\r
  52       return f.value();\r
  53   }\r
  54 \r
  55   return find<Function>(p.functions, [&](Function f) { return f.name == name; });\r
  56 }\r
  57 \r
  58 opt<Struct> findStruct(\r
  59   const Program & p,\r
  60   const std::string & name,\r
  61   const std::vector<std::string> & namespacePrefixes)\r
  62 {\r
  63   if (namespacePrefixes.empty())\r
  64   {\r
  65     return find<Struct>(p.structs, [&](Struct s) { return s.name == name; });\r
  66   }\r
  67   \r
  68   auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });\r
  69   \r
  70   if (!n.has_value())\r
  71     return nullopt;\r
  72     \r
  73   for (int i = 1; i < namespacePrefixes.size(); i++)\r
  74   {\r
  75     n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });\r
  76 \r
  77     if (!n.has_value())\r
  78       return nullopt;\r
  79   }\r
  80   return find<Struct>(n.value().structs, [&](Struct s) { return s.name == name; });\r
  81 }\r
  82 \r
  83 opt<Variable> findVariable(\r
  84   const Program & p,\r
  85   const std::string & name,\r
  86   std::shared_ptr<Context> ctx)\r
  87 {\r
  88   auto it = ctx;\r
  89   while (it != nullptr)\r
  90   {\r
  91     auto v = find<Variable>(it->variables, [&](Variable v) { return v.name == name; });\r
  92     if (v.has_value())\r
  93       return v;\r
  94     it = it->parent;\r
  95   }\r
  96   return nullopt;\r
  97 }\r
  98 \r
  99 opt<Function> findStructMethod(\r
 100   const Program & p,\r
 101   const std::string & name,\r
 102   TypeInfo ti)\r
 103 {\r
 104   if (!ti.isStruct)\r
 105     return nullopt;\r
 106   auto s = findStruct(p, ti.type.name, ti.type.namespacePrefixes);\r
 107   if (!s.has_value())\r
 108     return nullopt;\r
 109   return find<StructMember<Function>>(s.value().methods, [&](Function f) { return f.name == name; });\r
 110 }\r
 111 \r
 112 opt<Variable> findStructMember(\r
 113   const Program & p,\r
 114   TypeInfo ti,\r
 115   const std::string & name)\r
 116 {\r
 117   auto s = findStruct(p, ti.type.name, ti.type.namespacePrefixes);\r
 118   if (!s.has_value())\r
 119     return nullopt;\r
 120   return find<StructMember<Variable>>(s.value().members, [&](Variable v) { return v.name == name; });\r
 121 }