Spicy
list-comprehension.h
1 // Copyright (c) 2020-2021 by the Zeek Project. See LICENSE for details.
2 
3 #pragma once
4 
5 #include <memory>
6 #include <utility>
7 
8 #include <hilti/ast/expression.h>
9 #include <hilti/ast/types/computed.h>
10 #include <hilti/ast/types/list.h>
11 #include <hilti/ast/types/vector.h>
12 
13 namespace hilti {
14 namespace expression {
15 
18 public:
19  ListComprehension(Expression input, Expression output, ID id, std::optional<Expression> cond, Meta m = Meta())
20  : NodeBase(nodes(std::move(input), std::move(output), std::move(id), std::move(cond), type::unknown),
21  std::move(m)) {
22  _computeTypes();
23  }
24 
25  const auto& input() const { return child<Expression>(0); }
26  const auto& output() const { return child<Expression>(1); }
27  const auto& id() const { return child<ID>(2); }
28  auto condition() const { return childs()[3].tryReferenceAs<Expression>(); }
29 
34  IntrusivePtr<Scope> scope() const { return childs()[1].scope(); }
35 
36  bool operator==(const ListComprehension& other) const {
37  return input() == other.input() && output() == other.output() && id() == other.id() &&
38  condition() == other.condition();
39  }
40 
42  bool isLhs() const { return false; }
44  bool isTemporary() const { return true; }
46  Type type() const { return childs()[4].as<Type>(); }
47  Node& typeNode() { return childs()[4]; }
48 
50  auto isConstant() const { return input().isConstant(); }
52  auto isEqual(const Expression& other) const { return node::isEqual(this, other); }
53 
55  auto properties() const { return node::Properties{}; }
56 
57 private:
58  void _computeTypes() {}
59 };
60 
61 } // namespace expression
62 } // namespace hilti
auto & childs() const
Definition: node.h:445
IntrusivePtr< Scope > scope() const
Definition: list-comprehension.h:34
Definition: list-comprehension.h:17
auto properties() const
Definition: list-comprehension.h:55
auto isEqual(const Expression &other) const
Definition: list-comprehension.h:52
bool isTemporary() const
Definition: list-comprehension.h:44
Definition: expression.h:16
Definition: meta.h:18
auto isConstant() const
Definition: list-comprehension.h:50
Type type() const
Definition: list-comprehension.h:46
Definition: intrusive-ptr.h:69
std::map< std::string, node::detail::PropertyValue > Properties
Definition: node.h:83
Definition: node.h:97
bool isLhs() const
Definition: list-comprehension.h:42
Definition: id.h:18
Definition: node.h:318