blob: 7097779777582eb512ab2bdd55a59ce7af51ff3d (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
|
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
#pragma once
#include "value_type.h"
#include <map>
namespace vespalib::eval {
namespace nodes { struct Node; }
class Function;
/**
* Class keeping track of the output type of all intermediate
* calculations for a single function. The constructor performs type
* resolution for each node in the AST based on the type of all
* function parameters. The default constructor creates an empty type
* repo where all lookups will result in error types.
**/
class NodeTypes
{
private:
ValueType _not_found;
std::map<const nodes::Node*,ValueType> _type_map;
std::vector<vespalib::string> _errors;
public:
NodeTypes();
NodeTypes(NodeTypes &&rhs) = default;
NodeTypes &operator=(NodeTypes &&rhs) = default;
NodeTypes(const nodes::Node &const_node);
NodeTypes(const Function &function, const std::vector<ValueType> &input_types);
~NodeTypes();
const std::vector<vespalib::string> &errors() const { return _errors; }
NodeTypes export_types(const nodes::Node &root) const;
const ValueType &get_type(const nodes::Node &node) const;
template <typename F>
void each(F &&f) const {
for (const auto &entry: _type_map) {
f(*entry.first, entry.second);
}
}
bool all_types_are_double() const {
bool all_double = true;
each([&all_double](const nodes::Node &, const ValueType &type)
{
all_double &= type.is_double();
});
return (all_double && (_type_map.size() > 0));
}
};
}
|
