diff options
| author | Patrick Schönberger | 2021-08-04 14:51:36 +0200 |
|---|---|---|
| committer | Patrick Schönberger | 2021-08-04 14:51:36 +0200 |
| commit | be56e29997e17685eebf8bd7cb745183c60de7db (patch) | |
| tree | fb41e72af9882ab2228a16f0fb2581cfaa5f50d0 /src/find.h | |
| parent | dbc4a22d3c8c4189459f0361cb9da06415ec2dc9 (diff) | |
| download | toc-be56e29997e17685eebf8bd7cb745183c60de7db.tar.gz toc-be56e29997e17685eebf8bd7cb745183c60de7db.zip | |
pre change
Diffstat (limited to 'src/find.h')
| -rw-r--r-- | src/find.h | 121 |
1 files changed, 121 insertions, 0 deletions
diff --git a/src/find.h b/src/find.h new file mode 100644 index 0000000..31400ab --- /dev/null +++ b/src/find.h @@ -0,0 +1,121 @@ +#pragma once
+
+#include "repr.h"
+#include "typeInfo.h"
+
+#include <functional>
+#include <optional>
+
+template<typename T>
+using opt = std::optional<T>;
+
+template<typename T>
+opt<T> find(const std::vector<T> & ts, std::function<bool(T)> f)
+{
+ for (auto t : ts)
+ if (f(t))
+ return t;
+ return nullopt;
+}
+
+opt<Function> findFunction(
+ const Program & p,
+ const std::string & name,
+ const std::vector<std::string> & namespacePrefixes)
+{
+ if (namespacePrefixes.empty())
+ {
+ return find<Function>(p.functions, [&](Function f) { return f.name == name; });
+ }
+
+ auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });
+
+ if (!n.has_value())
+ return nullopt;
+
+ for (int i = 1; i < namespacePrefixes.size(); i++)
+ {
+ n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });
+
+ if (!n.has_value())
+ return nullopt;
+ }
+
+ return find<Function>(n.value().functions, [&](Function f) { return f.name == name; });
+}
+
+opt<Struct> findStruct(
+ const Program & p,
+ const std::string & name,
+ const std::vector<std::string> & namespacePrefixes)
+{
+ if (namespacePrefixes.empty())
+ {
+ return find<Struct>(p.structs, [&](Struct s) { return s.name == name; });
+ }
+
+ auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });
+
+ if (!n.has_value())
+ return nullopt;
+
+ for (int i = 1; i < namespacePrefixes.size(); i++)
+ {
+ n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });
+
+ if (!n.has_value())
+ return nullopt;
+ }
+ return find<Struct>(n.value().structs, [&](Struct s) { return s.name == name; });
+}
+
+opt<Variable> findVariable(
+ const Program & p,
+ const std::string & name,
+ const std::vector<std::string> & namespacePrefixes)
+{
+ for (auto n : namespacePrefixes)
+ std::cout << n << std::endl;
+ if (namespacePrefixes.empty())
+ {
+ return find<Variable>(p.variables, [&](Variable v) { return v.name == name; });
+ }
+
+ auto n = find<Namespace>(p.namespaces, [&](Namespace n) { return n.name == namespacePrefixes[0]; });
+
+ if (!n.has_value())
+ return nullopt;
+
+ for (int i = 1; i < namespacePrefixes.size(); i++)
+ {
+ n = find<Namespace>(n.value().namespaces, [&](Namespace n) { return n.name == namespacePrefixes[i]; });
+
+ if (!n.has_value())
+ return nullopt;
+ }
+ return find<Variable>(n.value().variables, [&](Variable v) { return v.name == name; });
+}
+
+opt<Function> findStructMethod(
+ const Program & p,
+ const std::string & name,
+ TypeInfo ti)
+{
+ if (!ti.isStruct)
+ return nullopt;
+ auto s = findStruct(p, ti.type.name, ti.type.namespacePrefixes);
+ if (!s.has_value())
+ return nullopt;
+ return find<StructMember<Function>>(s.value().methods, [&](Function f) { return f.name == name; });
+}
+
+opt<Variable> findStructMember(
+ const Program & p,
+ TypeInfo ti,
+ const std::string & name)
+{
+ auto s = findStruct(p, ti.type.name, ti.type.namespacePrefixes);
+ if (!s.has_value())
+ return nullopt;
+ return find<StructMember<Variable>>(s.value().members, [&](Variable v) { return v.name == name; });
+}
\ No newline at end of file |