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 {
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),
21  declaration::LocalVariable(std::move(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(childs()[2]); }
29  auto condition() const { return childs()[3].tryAs<Expression>(); }
30 
35  IntrusivePtr<Scope> scope() const { return childs()[1].scope(); }
36 
37  void setLocalType(Type t) { childs()[2].as<declaration::LocalVariable>().setType(std::move(t)); }
38  void setElementType(const Type x) { childs()[4] = type::List(std::move(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 childs()[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 expression
62 } // namespace hilti
Definition: local-variable.h:19
IntrusivePtr< Scope > scope() const
Definition: list-comprehension.h:35
Definition: list-comprehension.h:17
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
Definition: expression.h:17
Definition: meta.h:18
Definition: list.h:55
Definition: type.h:159
auto isConstant() const
Definition: list-comprehension.h:53
Definition: intrusive-ptr.h:69
std::map< std::string, node::detail::PropertyValue > Properties
Definition: node.h:99
Definition: node-ref.h:44
bool isLhs() const
Definition: list-comprehension.h:46
const auto & childs() const
Definition: node.h:470
auto & meta() const
Definition: node.h:474
const Type & type() const
Definition: list-comprehension.h:50
Definition: id.h:18
Definition: node.h:358