git.iko.soy/rowtype-yoga/purescript-native
on 4e33a7e help sign in
rowtype-yoga/purescript-native/pcc/runtime/purescript.hh
572 lines 17 kB history raw
1///////////////////////////////////////////////////////////////////////////////
2//
3// Module : purescript.hh
4// Copyright : (c) Andy Arvanitis 2015
5// License : MIT
6//
7// Maintainer : Andy Arvanitis <andy.arvanitis@gmail.com>
8// Stability : experimental
9// Portability :
10//
11// Basic types and functions to support purescript-to-C++1x rendering
12//
13///////////////////////////////////////////////////////////////////////////////
14//
15#ifndef PureScript_HH
16#define PureScript_HH
18#if !defined(__cplusplus)
19#error This is a C++ file!
20#endif
22// Standard includes
24#if defined(DEBUG)
25 #include <cassert>
26 #include <limits>
27 #define IF_DEBUG(x) x
28#else
29 #define NDEBUG
30 #undef assert
31 #define assert(x)
32 #define IF_DEBUG(_)
33#endif
35#include <functional>
36#include <string>
37#include <array>
38#include <deque>
39#include <utility>
40#include <stdexcept>
41#include <iso646.h> // mostly for MS Visual Studio compiler
42#include "purescript_memory.hh"
44namespace PureScript {
46using cstring = const char *;
47using std::string;
48using std::nullptr_t;
49using std::runtime_error;
51namespace Private {
52 struct SymbolGeneratorAnchor {};
54 template <typename T>
55 struct SymbolGenerator {
56 constexpr static SymbolGeneratorAnchor anchor = SymbolGeneratorAnchor{};
57 };
58 template <typename T>
59 constexpr SymbolGeneratorAnchor SymbolGenerator<T>::anchor;
61 using Symbol = const SymbolGeneratorAnchor *;
64#define define_symbol(S) namespace PureScript { \
65 namespace Private { \
66 namespace Symbols { \
67 struct S_ ## S {}; \
68 } \
69 } \
70 }
71#define symbol(S) (&Private::SymbolGenerator<::PureScript::Private::Symbols::S_ ## S>::anchor)
73constexpr bool undefined = false;
75// TODO: Not a real limit, just used for simpler accessors
76static constexpr size_t unknown_size = 64;
78// A variant data class designed to provide some features of dynamic typing.
79//
80class any {
82 public:
83 enum class Tag {
84 Thunk = 0x10,
85 Integer,
86 Double,
87 Character,
88 Boolean,
89 StringLiteral,
90 Function,
91 EffFunction,
92 RawPointer,
93 String,
94 Map,
95 Data,
96 Array,
97 Closure,
98 EffClosure,
99 Pointer
100 };
102 private:
103 mutable Tag tag;
105 public:
106 struct as_thunk {
107 };
108 static constexpr as_thunk unthunk = as_thunk{};
110 using map_pair = std::pair<const Private::Symbol, const any>;
112 template <size_t N>
113 using map = std::array<const map_pair, N>;
115 template <size_t N>
116 using data = std::array<const any, N>;
118 using array = std::deque<any WITH_ALLOCATOR(any)>;
119 using fn = auto (*)(const any&) -> any;
120 using eff_fn = auto (*)() -> any;
121 using thunk = auto (*)(const as_thunk) -> const any&;
123 private:
124 class Closure {
125 public:
126 virtual auto operator()(const any&) const -> any = 0;
127 virtual ~Closure();
128 };
130 template <typename T>
131 class Closure_ : public Closure {
132 const T lambda;
133 public:
134 Closure_(const T& l) noexcept : lambda(l) {}
135 auto operator()(const any& arg) const -> any override {
136 return lambda(arg);
137 }
138 };
140 class EffClosure {
141 public:
142 virtual auto operator()() const -> any = 0;
143 virtual ~EffClosure();
144 };
146 template <typename T>
147 class EffClosure_ : public EffClosure {
148 const T lambda;
149 public:
150 EffClosure_(const T& l) noexcept : lambda(l) {}
151 auto operator()() const -> any override {
152 return lambda();
153 }
154 };
156 private:
157 union {
158 mutable thunk t;
159 mutable int i;
160 mutable double d;
161 mutable char c;
162 mutable bool b;
163 mutable cstring r;
164 mutable fn f;
165 mutable eff_fn e;
166 mutable void * u;
167 mutable managed<string> s;
168 mutable managed<array> a;
169 mutable managed<Closure> l;
170 mutable managed<EffClosure> k;
171 mutable managed<void> p;
172 };
174 public:
176 any(const int val) noexcept : tag(Tag::Integer), i(val) {}
177 any(const long val) noexcept : tag(Tag::Integer), i(static_cast<decltype(i)>(val)) {
178 assert(val >= std::numeric_limits<decltype(i)>::min() &&
179 val <= std::numeric_limits<decltype(i)>::max());
180 }
182 any(const double val) noexcept : tag(Tag::Double), d(val) {}
183 any(const char val) noexcept : tag(Tag::Character), c(val) {}
185 template <typename T, typename = typename std::enable_if<std::is_same<bool,T>::value>::type>
186 any(const T val) noexcept : tag(Tag::Boolean), b(val) {}
188 template <size_t N>
189 any(const char (&val)[N]) noexcept : tag(Tag::StringLiteral), r(val) {}
190 any(char * val) : tag(Tag::String), s(make_managed<string>(val)) {}
192 any(const string& val) : tag(Tag::String), s(make_managed<string>(val)) {}
193 any(string&& val) noexcept : tag(Tag::String), s(make_managed<string>(std::move(val))) {}
195 any(const managed<string>& val) noexcept : tag(Tag::String), s(val) {}
196 any(managed<string>&& val) noexcept : tag(Tag::String), s(std::move(val)) {}
198 template <size_t N>
199 any(map<N>&& val) noexcept : tag(Tag::Map), p(make_managed<map<N>>(std::move(val))) {}
201 template <size_t N>
202 any(data<N>&& val) noexcept : tag(Tag::Data), p(make_managed<data<N>>(std::move(val))) {}
204 any(const array& val) : tag(Tag::Array), a(make_managed<array>(val)) {}
205 any(array&& val) noexcept : tag(Tag::Array), a(make_managed<array>(std::move(val))) {}
207 template <typename T>
208 any(const T& val, typename std::enable_if<std::is_convertible<T,fn>::value>::type* = 0) noexcept
209 : tag(Tag::Function), f(val) {}
211 template <typename T, typename = typename std::enable_if<!std::is_same<any,T>::value &&
212 !std::is_convertible<T,fn>::value>::type>
213 any(const T& val, typename std::enable_if<std::is_assignable<std::function<any(const any&)>,T>::value>::type* = 0)
214 : tag(Tag::Closure), l(make_managed<Closure_<T>>(val)) {}
216 template <typename T>
217 any(const T& val, typename std::enable_if<std::is_convertible<T,eff_fn>::value>::type* = 0) noexcept
218 : tag(Tag::EffFunction), e(val) {}
220 template <typename T,
221 typename = typename std::enable_if<!std::is_same<any,T>::value &&
222 !std::is_convertible<T,eff_fn>::value>::type>
223 any(const T& val, typename std::enable_if<std::is_assignable<std::function<any()>,T>::value>::type* = 0)
224 : tag(Tag::EffClosure), k(make_managed<EffClosure_<T>>(val)) {}
226 template <typename T>
227 any(const T& val, typename std::enable_if<std::is_convertible<T,thunk>::value>::type* = 0) noexcept
228 : tag(Tag::Thunk), t(val) {}
230 template <typename T,
231 typename = typename std::enable_if<std::is_class<typename std::remove_pointer<T>::type>::value>::type>
232 any(const T& val, typename std::enable_if<IS_POINTER_TYPE(T)::value>::type* = 0) noexcept
233 : tag(Tag::Pointer), p(val) {}
235 template <typename T>
236 any(T&& val, typename std::enable_if<std::is_assignable<managed<void>,T>::value>::type* = 0) noexcept
237 : tag(Tag::Pointer), p(std::move(val)) {}
239 // Explicit void* to raw pointer value
240 template <typename T>
241 any(const T& val, typename std::enable_if<std::is_same<T,void*>::value>::type* = 0) noexcept
242 : tag(Tag::RawPointer), u(val) {}
244 any(nullptr_t) noexcept : tag(Tag::RawPointer), u(nullptr) {}
247#if !defined(USE_GC)
248 private:
249 template <typename T,
250 typename U,
251 typename = typename std::enable_if<!std::is_reference<T>::value>::type>
252 static constexpr auto move_if_rvalue(U&& u) noexcept -> U&& {
253 return static_cast<U&&>(u);
254 }
256 template <typename T,
257 typename U,
258 typename = typename std::enable_if<std::is_reference<T>::value>::type>
259 static constexpr auto move_if_rvalue(U& u) noexcept -> U& {
260 return static_cast<U&>(u);
261 }
263 template <typename T>
264 auto assign(T&& other) const noexcept -> void {
265 switch (other.tag) {
266 case Tag::Thunk: t = other.t; break;
267 case Tag::Integer: i = other.i; break;
268 case Tag::Double: d = other.d; break;
269 case Tag::Character: c = other.c; break;
270 case Tag::Boolean: b = other.b; break;
271 case Tag::StringLiteral: r = other.r; break;
272 case Tag::Function: f = other.f; break;
273 case Tag::EffFunction: e = other.e; break;
274 case Tag::RawPointer: u = other.u; break;
276 case Tag::String: new (&s) managed<string>(move_if_rvalue<T>(other.s)); break;
277 case Tag::Array: new (&a) managed<array>(move_if_rvalue<T>(other.a)); break;
278 case Tag::Closure: new (&l) managed<Closure>(move_if_rvalue<T>(other.l)); break;
279 case Tag::EffClosure: new (&k) managed<EffClosure>(move_if_rvalue<T>(other.k)); break;
280 case Tag::Map:
281 case Tag::Data:
282 case Tag::Pointer: new (&p) managed<void>(move_if_rvalue<T>(other.p)); break;
284 default: assert(false && "Bad 'any' tag"); break;
285 }
286 }
288 auto destruct() -> void {
289 switch (tag) {
290 case Tag::String: s.~managed<string>(); break;
291 case Tag::Array: a.~managed<array>(); break;
292 case Tag::Closure: l.~managed<Closure>(); break;
293 case Tag::EffClosure: k.~managed<EffClosure>(); break;
294 case Tag::Map:
295 case Tag::Data:
296 case Tag::Pointer: p.~managed<void>(); break;
298 default: break;
299 }
300 }
302 public:
303 any(const any& other) noexcept : tag(other.tag) {
304 assign(other);
305 }
307 any(any&& other) noexcept : tag(other.tag) {
308 assign(std::move(other));
309 }
311 auto operator=(const any& rhs) noexcept -> any& {
312 destruct();
313 tag = rhs.tag;
314 assign(rhs);
315 return *this;
316 }
318 auto operator=(any&& rhs) noexcept -> any& {
319 destruct();
320 tag = rhs.tag;
321 assign(std::move(rhs));
322 return *this;
323 }
325 ~any() {
326 destruct();
327 }
328#endif // !defined(USE_GC)
330 public:
331 any() = delete;
333 auto operator()(const any&) const -> any;
334 auto operator()(const as_thunk) const -> const any&;
335 auto operator()() const -> any;
337 operator int() const;
338 operator double() const;
339 operator bool() const;
340 operator char() const;
341 operator size_t() const;
342 operator cstring() const;
343 operator const array&() const;
345 auto operator[](const size_t) const -> const any&;
347 auto size() const -> size_t;
348 auto empty() const -> bool;
349 auto contains(const Private::Symbol) const -> bool;
351 auto extractPointer(IF_DEBUG(const Tag)) const -> void*;
353 auto rawPointer() const -> void* {
354 const any& variant = unthunkVariant(*this);
355 assert(tag == Tag::RawPointer);
356 return variant.u;
357 }
359 static auto unthunkVariant(const any&) -> const any&;
361 #define DEFINE_OPERATOR(op, ty, rty) \
362 inline friend auto operator op (const any& lhs, ty rhs) -> rty { \
363 return (ty)lhs op rhs; \
364 } \
365 inline friend auto operator op (ty lhs, const any& rhs) -> rty { \
366 return lhs op (ty)rhs; \
367 } \
369 #define DECLARE_COMPARISON_OPERATOR(op) \
370 friend auto operator op (const any&, const any&) -> bool; \
371 DEFINE_OPERATOR(op, int, bool) \
372 DEFINE_OPERATOR(op, double, bool) \
373 DEFINE_OPERATOR(op, char, bool) \
374 DEFINE_OPERATOR(op, bool, bool) \
375 friend auto operator op (const any&, const char * const) -> bool; \
376 friend auto operator op (const char * const, const any&) -> bool;
378 DECLARE_COMPARISON_OPERATOR(==)
379 DECLARE_COMPARISON_OPERATOR(!=)
380 DECLARE_COMPARISON_OPERATOR(<)
381 DECLARE_COMPARISON_OPERATOR(<=)
382 DECLARE_COMPARISON_OPERATOR(>)
383 DECLARE_COMPARISON_OPERATOR(>=)
385 friend auto operator+(const any&, const any&) -> any;
387 DEFINE_OPERATOR(+, int, int)
388 DEFINE_OPERATOR(+, double, double)
389 DEFINE_OPERATOR(+, char, char)
390 friend auto operator+(const any& lhs, const char * const rhs) -> string;
391 friend auto operator+(const char * const lhs, const any& rhs) -> string;
393 friend auto operator-(const any&, const any&) -> any;
395 DEFINE_OPERATOR(-, int, int)
396 DEFINE_OPERATOR(-, double, double)
397 DEFINE_OPERATOR(-, char, char)
399 friend auto operator*(const any&, const any&) -> any;
401 DEFINE_OPERATOR(*, int, int)
402 DEFINE_OPERATOR(*, double, double)
403 DEFINE_OPERATOR(*, char, char)
405 friend auto operator/(const any&, const any&) -> any;
407 DEFINE_OPERATOR(/, int, int)
408 DEFINE_OPERATOR(/, double, double)
409 DEFINE_OPERATOR(/, char, char)
411 friend auto operator%(const any&, const any&) -> any;
413 DEFINE_OPERATOR(%, int, int)
414 DEFINE_OPERATOR(%, char, char)
416 friend auto operator-(const any&) -> any; // unary negate
418}; // class any
420namespace Private {
421 template <typename T, typename U=void>
422 struct TagHelper {};
424 template <size_t N>
425 struct TagHelper<any::map<N>> {
426 static constexpr any::Tag tag = any::Tag::Map;
427 };
429 template <size_t N>
430 struct TagHelper<any::data<N>> {
431 static constexpr any::Tag tag = any::Tag::Data;
432 };
434 template <typename T>
435 struct TagHelper<T> {
436 static constexpr any::Tag tag = any::Tag::Pointer;
437 };
440// Pass-through
441template <typename T>
442constexpr auto cast(const T& a) -> const T& {
443 return a;
446template <typename T>
447inline auto cast(const any& a) ->
448 typename std::enable_if<std::is_arithmetic<T>::value ||
449 std::is_same<T,cstring>::value, T>::type {
450 return a;
453template <typename T>
454inline auto cast(const any& a) ->
455 typename std::enable_if<std::is_same<T, any::array>::value, const T&>::type {
456 return a;
459template <typename T, typename = typename std::enable_if<std::is_class<T>::value>::type>
460inline auto cast(const any& a) ->
461 typename std::enable_if<!std::is_same<T, any::array>::value, T&>::type {
462 return *static_cast<T*>(a.extractPointer(IF_DEBUG(Private::TagHelper<T>::tag)));
465template <typename T,
466 typename = typename std::enable_if<std::is_pointer<T>::value &&
467 !std::is_same<T,void*>::value &&
468 !std::is_same<T,cstring>::value>::type>
469inline auto cast(const any& a) -> T {
470 return static_cast<T>(a.extractPointer(IF_DEBUG(Private::TagHelper<T>::tag)));
473template <typename T>
474inline auto cast(const any& a) ->
475 typename std::enable_if<std::is_same<void*, T>::value, T>::type {
476 return a.rawPointer();
479namespace map {
480 template <size_t N, typename T>
481 inline auto get(const T& a) ->
482 typename std::enable_if<!std::is_same<any, T>::value, const any&>::type {
483 return a[N].second;
484 }
486 template <size_t N>
487 inline auto get(const any& a) -> const any& {
488 return cast<any::map<N+1>>(a)[N].second;
489 }
491 template <size_t N>
492 inline auto get(const Private::Symbol key, const any::map<N>& a) -> const any& {
493 static_assert(N > 0, "map size must be greater than zero");
494 typename std::remove_reference<decltype(a)>::type::size_type i = 0;
495 do {
496 if (a[i].first == key) {
497 return a[i].second;
498 }
499 } while (a[++i].first != nullptr);
500 assert(false && "map key not found");
501 static const any invalid_key(nullptr);
502 return invalid_key;
503 }
505 inline auto get(const Private::Symbol key, const any& a) -> const any& {
506 return get(key, cast<any::map<unknown_size>>(a));
507 }
509 #define P(N) any::map_pair&& arg ## N
510 #define T(N) any::map<N+1>
511 #define E(N) std::forward<decltype(arg ## N)>(arg ## N)
512 #define M(...) {{ __VA_ARGS__, { nullptr, nullptr } }}
514 inline auto make(P(1)) -> T(1) {
515 return T(1) M(E(1));
516 }
517 inline auto make(P(1), P(2)) -> T(2) {
518 return T(2) M(E(1), E(2));
519 }
520 inline auto make(P(1), P(2), P(3)) -> T(3) {
521 return T(3) M(E(1), E(2), E(3));
522 }
523 inline auto make(P(1), P(2), P(3), P(4)) -> T(4) {
524 return T(4) M(E(1), E(2), E(3), E(4));
525 }
526 inline auto make(P(1), P(2), P(3), P(4), P(5)) -> T(5) {
527 return T(5) M(E(1), E(2), E(3), E(4), E(5));
528 }
529 inline auto make(P(1), P(2), P(3), P(4), P(5), P(6)) -> T(6) {
530 return T(6) M(E(1), E(2), E(3), E(4), E(5), E(6));
531 }
532 inline auto make(P(1), P(2), P(3), P(4), P(5), P(6), P(7)) -> T(7) {
533 return T(7) M(E(1), E(2), E(3), E(4), E(5), E(6), E(7));
534 }
535 inline auto make(P(1), P(2), P(3), P(4), P(5), P(6), P(7), P(8)) -> T(8) {
536 return T(8) M(E(1), E(2), E(3), E(4), E(5), E(6), E(7), E(8));
537 }
539 #undef P
540 #undef T
541 #undef E
542 #undef M
545namespace record = map;
547namespace data {
548 template <size_t N, typename T>
549 inline auto get(const T& a) ->
550 typename std::enable_if<!std::is_same<any, T>::value, const any&>::type {
551 return a[N];
552 }
554 template <size_t N>
555 inline auto get(const any& a) -> const any& {
556 return cast<any::data<N+1>>(a)[N];
557 }
559 template <typename T>
560 inline auto ctor(const T& a) -> int {
561 return get<0>(a);
562 }
565} // namespace PureScript
567#undef WITH_ALLOCATOR
568#undef IS_POINTER_TYPE
569#undef DEFINE_OPERATOR
570#undef DECLARE_COMPARISON_OPERATOR
572#endif // PureScript_HH