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/declarations/local-variable.h>
9 #include <hilti/ast/expression.h>
10 #include <hilti/ast/types/list.h>
11 #include <hilti/ast/types/vector.h>
12 
13 namespace hilti::expression {
14 
17 public:
18  ListComprehension(Expression input, Expression output, const ID& id, std::optional<Expression> cond,
19  Meta m = Meta())
20  : NodeBase(nodes(std::move(input), std::move(output),
21  declaration::LocalVariable(id, type::auto_, true, id.meta()), std::move(cond),
22  type::List(type::auto_, m)),
23  std::move(m)) {}
24 
25  const auto& input() const { return child<Expression>(0); }
26  const auto& output() const { return child<Expression>(1); }
27  const auto& local() const { return child<declaration::LocalVariable>(2); }
28  auto localRef() const { return NodeRef(children()[2]); }
29  auto condition() const { return children()[3].tryAs<Expression>(); }
30 
35  IntrusivePtr<Scope> scope() const { return children()[1].scope(); }
36 
37  void setLocalType(const Type& t) { children()[2].as<declaration::LocalVariable>().setType(t); }
38  void setElementType(const Type& x) { children()[4] = type::List(x); }
39 
40  bool operator==(const ListComprehension& other) const {
41  return input() == other.input() && output() == other.output() && local() == other.local() &&
42  condition() == other.condition();
43  }
44 
46  bool isLhs() const { return false; }
48  bool isTemporary() const { return true; }
50  const Type& type() const { return children()[4].as<Type>(); }
51 
53  auto isConstant() const { return input().isConstant(); }
55  auto isEqual(const Expression& other) const { return node::isEqual(this, other); }
56 
58  auto properties() const { return node::Properties{}; }
59 };
60 
61 } // namespace hilti::expression
Definition: local-variable.h:18
IntrusivePtr< Scope > scope() const
Definition: list-comprehension.h:35
Definition: list-comprehension.h:16
auto properties() const
Definition: list-comprehension.h:58
auto isEqual(const Expression &other) const
Definition: list-comprehension.h:55
bool isTemporary() const
Definition: list-comprehension.h:48
const auto & children() const
Definition: node.h:471
Definition: expression.h:18
Definition: meta.h:19
Definition: list.h:54
Definition: type.h:160
auto isConstant() const
Definition: list-comprehension.h:53
Definition: intrusive-ptr.h:69
std::map< std::string, node::detail::PropertyValue > Properties
Definition: node.h:97
Definition: node-ref.h:45
bool isLhs() const
Definition: list-comprehension.h:46
auto & meta() const
Definition: node.h:475
const Type & type() const
Definition: list-comprehension.h:50
Definition: id.h:18
Definition: node.h:359
Definition: expression.h:21