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/unknown.h>
14 
15 namespace hilti {
16 namespace declaration {
17 
20 public:
21  LocalVariable(ID id, ::hilti::Type type, std::optional<hilti::Expression> init = {}, bool const_ = false,
22  Meta m = Meta())
23  : NodeBase(nodes(std::move(id), std::move(type), std::move(init)), std::move(m)), _const(const_) {}
24 
25  LocalVariable(ID id, ::hilti::Type type, std::vector<hilti::Expression> args,
26  std::optional<hilti::Expression> init = {}, bool const_ = false, Meta m = Meta())
27  : NodeBase(nodes(std::move(id), std::move(type), std::move(init), std::move(args)), std::move(m)),
28  _const(const_) {}
29 
30  LocalVariable(ID id, hilti::Expression init, bool const_ = false, Meta m = Meta())
31  : NodeBase(nodes(std::move(id), node::none, std::move(init)), std::move(m)), _const(const_) {}
32 
33  auto init() const { return childs()[2].tryReferenceAs<hilti::Expression>(); }
34  auto typeArguments() const { return childs<hilti::Expression>(3, -1); }
35 
36  ::hilti::Type type() const {
37  if ( auto t = childs()[1].tryAs<::hilti::Type>(); t && *t != type::unknown )
38  return type::effectiveType(std::move(*t));
39 
40  if ( auto i = init() )
41  return i->type();
42 
43  return type::unknown;
44  }
45 
50  auto hasAutomaticType() const { return ! childs()[1].isA<::hilti::Type>(); }
51 
52  bool operator==(const LocalVariable& other) const {
53  return id() == other.id() && type() == other.type() && init() == other.init();
54  }
55 
57  bool isConstant() const { return _const; }
59  const ID& id() const { return child<ID>(0); }
61  Linkage linkage() const { return Linkage::Private; }
63  std::string displayName() const { return "local variable"; };
65  auto isEqual(const Declaration& other) const { return node::isEqual(this, other); }
66 
68  auto properties() const { return node::Properties{{"const", _const}}; }
69 
77  static Declaration setType(const LocalVariable& d, std::optional<hilti::Type> t) {
78  auto x = Declaration(d)._clone().as<LocalVariable>();
79  if ( t )
80  x.childs()[1] = *t;
81  else
82  x.childs()[1] = node::none;
83 
84  return std::move(x);
85  }
86 
94  static Declaration setInit(const LocalVariable& d, std::optional<hilti::Expression> i) {
95  auto x = Declaration(d)._clone().as<LocalVariable>();
96  if ( i )
97  x.childs()[2] = *i;
98  else
99  x.childs()[2] = node::none;
100 
101  return std::move(x);
102  }
103 
111  static Declaration setTypeArguments(const LocalVariable& d, std::vector<hilti::Expression> args) {
112  auto x = Declaration(d)._clone().as<LocalVariable>();
113  // We keep the first 3 children, which are ID, type and init expression.
114  x.childs() = x.childs<Node>(0, 3);
115  for ( auto&& a : args )
116  x.childs().emplace_back(std::move(a));
117 
118  return std::move(x);
119  }
120 
121 private:
122  bool _const;
123 };
124 
125 } // namespace declaration
126 } // namespace hilti
Definition: local-variable.h:19
auto & childs() const
Definition: node.h:445
std::vector< T > childs(int begin, int end) const
Definition: node.h:373
static Declaration setType(const LocalVariable &d, std::optional< hilti::Type > t)
Definition: local-variable.h:77
const ID & id() const
Definition: local-variable.h:59
const Node none
Definition: node.cc:12
auto isEqual(const Declaration &other) const
Definition: local-variable.h:65
Definition: meta.h:18
static Declaration setInit(const LocalVariable &d, std::optional< hilti::Expression > i)
Definition: local-variable.h:94
static Declaration setTypeArguments(const LocalVariable &d, std::vector< hilti::Expression > args)
Definition: local-variable.h:111
std::map< std::string, node::detail::PropertyValue > Properties
Definition: node.h:83
Linkage linkage() const
Definition: local-variable.h:61
bool isConstant() const
Definition: local-variable.h:57
Definition: node.h:97
Definition: declaration.h:15
auto hasAutomaticType() const
Definition: local-variable.h:50
auto properties() const
Definition: local-variable.h:68
std::string displayName() const
Definition: local-variable.h:63
Definition: id.h:18
Definition: node.h:318