Spicy
local-variable.h
1 // Copyright (c) 2020-2021 by the Zeek Project. See LICENSE for details.
2 
3 #pragma once
4 
5 #include <string>
6 #include <utility>
7 #include <vector>
8 
9 #include <hilti/ast/declaration.h>
10 #include <hilti/ast/expression.h>
11 #include <hilti/ast/id.h>
12 #include <hilti/ast/type.h>
13 #include <hilti/ast/types/auto.h>
14 
15 namespace hilti::declaration {
16 
19 public:
20  LocalVariable(ID id, ::hilti::Type type, std::optional<hilti::Expression> init, bool const_ = false,
21  Meta m = Meta())
22  : DeclarationBase(nodes(std::move(id), std::move(type), std::move(init)), std::move(m)), _const(const_) {}
23 
24  LocalVariable(ID id, ::hilti::Type type, bool const_ = false, Meta m = Meta())
25  : DeclarationBase(nodes(std::move(id), std::move(type), node::none), std::move(m)), _const(const_) {}
26 
27  LocalVariable(ID id, hilti::Expression init, bool const_ = false, Meta m = Meta())
28  : DeclarationBase(nodes(std::move(id), node::none, std::move(init)), std::move(m)), _const(const_) {}
29 
30  LocalVariable(ID id, ::hilti::Type type, std::vector<hilti::Expression> args,
31  std::optional<hilti::Expression> init = {}, bool const_ = false, Meta m = Meta())
32  : DeclarationBase(nodes(std::move(id), std::move(type), std::move(init), std::move(args)), std::move(m)),
33  _const(const_) {}
34 
35  LocalVariable(ID id, bool const_ = false, Meta m = Meta())
36  : DeclarationBase(nodes(std::move(id), type::auto_, node::none), std::move(m)), _const(const_) {}
37 
38  auto init() const { return children()[2].tryAs<hilti::Expression>(); }
39 
40  const auto& type() const {
41  if ( auto t = children()[1].tryAs<hilti::Type>() )
42  return *t;
43  else {
44  assert(init());
45  return init()->type();
46  }
47  }
48 
49  auto typeArguments() const { return children<hilti::Expression>(3, -1); }
50 
51  void setInit(const hilti::Expression& i) { children()[2] = i; }
52  void setType(const hilti::Type& t) { children()[1] = t; }
53  void setTypeArguments(std::vector<hilti::Expression> args) {
54  auto& c = children();
55  c.erase(c.begin() + 3, c.end());
56  for ( auto&& a : args )
57  c.emplace_back(std::move(a));
58  }
59 
60  bool operator==(const LocalVariable& other) const {
61  return id() == other.id() && type() == other.type() && init() == other.init();
62  }
63 
65  bool isConstant() const { return _const; }
67  const ID& id() const { return child<ID>(0); }
69  Linkage linkage() const { return Linkage::Private; }
71  std::string displayName() const { return "local variable"; };
73  auto isEqual(const Declaration& other) const { return node::isEqual(this, other); }
74 
76  auto properties() const { return node::Properties{{"const", _const}}; }
77 
78 private:
79  bool _const;
80 };
81 
82 } // namespace hilti::declaration
Definition: local-variable.h:18
Definition: declaration.h:54
const ID & id() const
Definition: local-variable.h:67
const Node none
Definition: node.cc:14
const auto & children() const
Definition: node.h:472
auto isEqual(const Declaration &other) const
Definition: local-variable.h:73
Definition: meta.h:19
Definition: type.h:160
Definition: declaration.h:88
std::map< std::string, node::detail::PropertyValue > Properties
Definition: node.h:98
Linkage linkage() const
Definition: local-variable.h:69
bool isConstant() const
Definition: local-variable.h:65
Definition: declaration.h:19
auto properties() const
Definition: local-variable.h:76
std::string displayName() const
Definition: local-variable.h:71
Definition: id.h:18