git.iko.soy/rowtype-yoga/purescript-native
on 22aec46 help sign in
rowtype-yoga/purescript-native/pcc/runtime/purescript.hh
555 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 char32_t c;
162 mutable bool b;
163 mutable cstring r;
164 mutable fn f;
165 mutable eff_fn e;
166 mutable void * v;
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 }
181 any(const double val) noexcept : tag(Tag::Double), d(val) {}
182 any(const char32_t val) noexcept : tag(Tag::Character), c(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 any(const char * val) : tag(Tag::String), s(make_managed<string>(val)) {}
189 any(const string& val) : tag(Tag::String), s(make_managed<string>(val)) {}
190 any(string&& val) noexcept : tag(Tag::String), s(make_managed<string>(std::move(val))) {}
192 any(const managed<string>& val) noexcept : tag(Tag::String), s(val) {}
193 any(managed<string>&& val) noexcept : tag(Tag::String), s(std::move(val)) {}
195 template <size_t N>
196 any(map<N>&& val) noexcept : tag(Tag::Map), p(make_managed<map<N>>(std::move(val))) {}
198 template <size_t N>
199 any(data<N>&& val) noexcept : tag(Tag::Data), p(make_managed<data<N>>(std::move(val))) {}
201 any(const array& val) : tag(Tag::Array), a(make_managed<array>(val)) {}
202 any(array&& val) noexcept : tag(Tag::Array), a(make_managed<array>(std::move(val))) {}
204 template <typename T>
205 any(const T& val, typename std::enable_if<std::is_convertible<T,fn>::value>::type* = 0) noexcept
206 : tag(Tag::Function), f(val) {}
208 template <typename T, typename = typename std::enable_if<!std::is_same<any,T>::value &&
209 !std::is_convertible<T,fn>::value>::type>
210 any(const T& val, typename std::enable_if<std::is_assignable<std::function<any(const any&)>,T>::value>::type* = 0)
211 : tag(Tag::Closure), l(make_managed<Closure_<T>>(val)) {}
213 template <typename T>
214 any(const T& val, typename std::enable_if<std::is_convertible<T,eff_fn>::value>::type* = 0) noexcept
215 : tag(Tag::EffFunction), e(val) {}
217 template <typename T,
218 typename = typename std::enable_if<!std::is_same<any,T>::value &&
219 !std::is_convertible<T,eff_fn>::value>::type>
220 any(const T& val, typename std::enable_if<std::is_assignable<std::function<any()>,T>::value>::type* = 0)
221 : tag(Tag::EffClosure), k(make_managed<EffClosure_<T>>(val)) {}
223 template <typename T>
224 any(const T& val, typename std::enable_if<std::is_convertible<T,thunk>::value>::type* = 0) noexcept
225 : tag(Tag::Thunk), t(val) {}
227 template <typename T,
228 typename = typename std::enable_if<std::is_class<typename std::remove_pointer<T>::type>::value>::type>
229 any(const T& val, typename std::enable_if<IS_POINTER_TYPE(T)::value>::type* = 0) noexcept
230 : tag(Tag::Pointer), p(val) {}
232 template <typename T>
233 any(T&& val, typename std::enable_if<std::is_assignable<managed<void>,T>::value>::type* = 0) noexcept
234 : tag(Tag::Pointer), p(std::move(val)) {}
236 // Explicit void* to raw pointer value
237 template <typename T>
238 any(const T& val, typename std::enable_if<std::is_same<T,void*>::value>::type* = 0) noexcept
239 : tag(Tag::RawPointer), v(val) {}
241 any(nullptr_t) noexcept : tag(Tag::RawPointer), v(nullptr) {}
244#if !defined(USE_GC)
245 private:
246 template <typename T,
247 typename U,
248 typename = typename std::enable_if<!std::is_reference<T>::value>::type>
249 static constexpr auto move_if_rvalue(U&& u) noexcept -> U&& {
250 return static_cast<U&&>(u);
251 }
253 template <typename T,
254 typename U,
255 typename = typename std::enable_if<std::is_reference<T>::value>::type>
256 static constexpr auto move_if_rvalue(U& u) noexcept -> U& {
257 return static_cast<U&>(u);
258 }
260 template <typename T>
261 auto assign(T&& other) const noexcept -> void {
262 switch (other.tag) {
263 case Tag::Thunk: t = other.t; break;
264 case Tag::Integer: i = other.i; break;
265 case Tag::Double: d = other.d; break;
266 case Tag::Character: c = other.c; break;
267 case Tag::Boolean: b = other.b; break;
268 case Tag::StringLiteral: r = other.r; break;
269 case Tag::Function: f = other.f; break;
270 case Tag::EffFunction: e = other.e; break;
271 case Tag::RawPointer: v = other.v; break;
273 case Tag::String: new (&s) managed<string>(move_if_rvalue<T>(other.s)); break;
274 case Tag::Array: new (&a) managed<array>(move_if_rvalue<T>(other.a)); break;
275 case Tag::Closure: new (&l) managed<Closure>(move_if_rvalue<T>(other.l)); break;
276 case Tag::EffClosure: new (&k) managed<EffClosure>(move_if_rvalue<T>(other.k)); break;
277 case Tag::Map:
278 case Tag::Data:
279 case Tag::Pointer: new (&p) managed<void>(move_if_rvalue<T>(other.p)); break;
281 default: assert(false && "Bad 'any' tag"); break;
282 }
283 }
285 auto destruct() -> void {
286 switch (tag) {
287 case Tag::String: s.~managed<string>(); break;
288 case Tag::Array: a.~managed<array>(); break;
289 case Tag::Closure: l.~managed<Closure>(); break;
290 case Tag::EffClosure: k.~managed<EffClosure>(); break;
291 case Tag::Map:
292 case Tag::Data:
293 case Tag::Pointer: p.~managed<void>(); break;
295 default: break;
296 }
297 }
299 public:
300 any(const any& other) noexcept : tag(other.tag) {
301 assign(other);
302 }
304 any(any&& other) noexcept : tag(other.tag) {
305 assign(std::move(other));
306 }
308 auto operator=(const any& rhs) noexcept -> any& {
309 destruct();
310 tag = rhs.tag;
311 assign(rhs);
312 return *this;
313 }
315 auto operator=(any&& rhs) noexcept -> any& {
316 destruct();
317 tag = rhs.tag;
318 assign(std::move(rhs));
319 return *this;
320 }
322 ~any() {
323 destruct();
324 }
325#endif // !defined(USE_GC)
327 public:
328 any() = delete;
330 auto operator()(const any&) const -> any;
331 auto operator()(const as_thunk) const -> const any&;
332 auto operator()() const -> any;
334 operator int() const;
335 operator double() const;
336 operator bool() const;
337 operator char32_t() const;
338 operator char() const;
339 operator size_t() const;
340 operator const char *() const;
341 operator const array&() const;
343 auto operator[](const size_t) const -> const any&;
345 auto size() const -> size_t;
346 auto empty() const -> bool;
347 auto contains(const Private::Symbol) const -> bool;
349 auto extractPointer(IF_DEBUG(const Tag)) const -> void*;
351 auto rawPointer() const -> void* {
352 const any& variant = unthunkVariant(*this);
353 assert(tag == Tag::RawPointer);
354 return variant.v;
355 }
357 static auto unthunkVariant(const any&) -> const any&;
359 #define DEFINE_OPERATOR(op) \
360 template <typename T> \
361 inline friend auto operator op (const any& lhs, T rhs) -> T { \
362 return static_cast<T>(lhs) op rhs; \
363 } \
364 template <typename T> \
365 inline friend auto operator op (T lhs, const any& rhs) -> T { \
366 return lhs op static_cast<T>(rhs); \
367 } \
368 friend auto operator op (const any&, const any&) -> any;
370 #define DEFINE_COMPARISON_OPERATOR(op) \
371 template <typename T> \
372 inline friend auto operator op (const any& lhs, T rhs) -> bool { \
373 return static_cast<T>(lhs) op rhs; \
374 } \
375 template <typename T> \
376 inline friend auto operator op (T lhs, const any& rhs) -> bool { \
377 return lhs op static_cast<T>(rhs); \
378 } \
379 friend auto operator op (const any&, const any&) -> bool; \
380 friend auto operator op (const any&, const char * const) -> bool; \
381 friend auto operator op (const char * const, const any&) -> bool;
383 DEFINE_COMPARISON_OPERATOR(==)
384 DEFINE_COMPARISON_OPERATOR(!=)
385 DEFINE_COMPARISON_OPERATOR(<)
386 DEFINE_COMPARISON_OPERATOR(<=)
387 DEFINE_COMPARISON_OPERATOR(>)
388 DEFINE_COMPARISON_OPERATOR(>=)
390 DEFINE_OPERATOR(+)
391 DEFINE_OPERATOR(-)
392 DEFINE_OPERATOR(*)
393 DEFINE_OPERATOR(/)
394 DEFINE_OPERATOR(%)
396 friend auto operator+(const any& lhs, const char * const rhs) -> string;
397 friend auto operator+(const char * const lhs, const any& rhs) -> string;
399 friend auto operator-(const any&) -> any; // unary negate
401}; // class any
403namespace Private {
404 template <typename T, typename U=void>
405 struct TagHelper {};
407 template <size_t N>
408 struct TagHelper<any::map<N>> {
409 static constexpr any::Tag tag = any::Tag::Map;
410 };
412 template <size_t N>
413 struct TagHelper<any::data<N>> {
414 static constexpr any::Tag tag = any::Tag::Data;
415 };
417 template <typename T>
418 struct TagHelper<T> {
419 static constexpr any::Tag tag = any::Tag::Pointer;
420 };
423// Pass-through
424template <typename T>
425constexpr auto cast(const T& a) -> const T& {
426 return a;
429template <typename T>
430inline auto cast(const any& a) ->
431 typename std::enable_if<std::is_arithmetic<T>::value ||
432 std::is_same<T,const char *>::value, T>::type {
433 return a;
436template <typename T>
437inline auto cast(const any& a) ->
438 typename std::enable_if<std::is_same<T, any::array>::value, const T&>::type {
439 return a;
442template <typename T, typename = typename std::enable_if<std::is_class<T>::value>::type>
443inline auto cast(const any& a) ->
444 typename std::enable_if<!std::is_same<T, any::array>::value, T&>::type {
445 return *static_cast<T*>(a.extractPointer(IF_DEBUG(Private::TagHelper<T>::tag)));
448template <typename T,
449 typename = typename std::enable_if<std::is_pointer<T>::value &&
450 !std::is_same<T,void*>::value &&
451 !std::is_same<T,const char *>::value>::type>
452inline auto cast(const any& a) -> T {
453 return static_cast<T>(a.extractPointer(IF_DEBUG(Private::TagHelper<T>::tag)));
456template <typename T>
457inline auto cast(const any& a) ->
458 typename std::enable_if<std::is_same<void*, T>::value, T>::type {
459 return a.rawPointer();
462namespace map {
463 template <size_t N, typename T>
464 inline auto get(const T& a) ->
465 typename std::enable_if<!std::is_same<any, T>::value, const any&>::type {
466 return a[N].second;
467 }
469 template <size_t N>
470 inline auto get(const any& a) -> const any& {
471 return cast<any::map<N+1>>(a)[N].second;
472 }
474 template <size_t N>
475 inline auto get(const Private::Symbol key, const any::map<N>& a) -> const any& {
476 static_assert(N > 0, "map size must be greater than zero");
477 typename std::remove_reference<decltype(a)>::type::size_type i = 0;
478 do {
479 if (a[i].first == key) {
480 return a[i].second;
481 }
482 } while (a[++i].first != nullptr);
483 assert(false && "map key not found");
484 static const any invalid_key(nullptr);
485 return invalid_key;
486 }
488 inline auto get(const Private::Symbol key, const any& a) -> const any& {
489 return get(key, cast<any::map<unknown_size>>(a));
490 }
492 #define P(N) any::map_pair&& arg ## N
493 #define T(N) any::map<N+1>
494 #define E(N) std::forward<decltype(arg ## N)>(arg ## N)
495 #define M(...) {{ __VA_ARGS__, { nullptr, nullptr } }}
497 inline auto make(P(1)) -> T(1) {
498 return T(1) M(E(1));
499 }
500 inline auto make(P(1), P(2)) -> T(2) {
501 return T(2) M(E(1), E(2));
502 }
503 inline auto make(P(1), P(2), P(3)) -> T(3) {
504 return T(3) M(E(1), E(2), E(3));
505 }
506 inline auto make(P(1), P(2), P(3), P(4)) -> T(4) {
507 return T(4) M(E(1), E(2), E(3), E(4));
508 }
509 inline auto make(P(1), P(2), P(3), P(4), P(5)) -> T(5) {
510 return T(5) M(E(1), E(2), E(3), E(4), E(5));
511 }
512 inline auto make(P(1), P(2), P(3), P(4), P(5), P(6)) -> T(6) {
513 return T(6) M(E(1), E(2), E(3), E(4), E(5), E(6));
514 }
515 inline auto make(P(1), P(2), P(3), P(4), P(5), P(6), P(7)) -> T(7) {
516 return T(7) M(E(1), E(2), E(3), E(4), E(5), E(6), E(7));
517 }
518 inline auto make(P(1), P(2), P(3), P(4), P(5), P(6), P(7), P(8)) -> T(8) {
519 return T(8) M(E(1), E(2), E(3), E(4), E(5), E(6), E(7), E(8));
520 }
522 #undef P
523 #undef T
524 #undef E
525 #undef M
528namespace record = map;
530namespace data {
531 template <size_t N, typename T>
532 inline auto get(const T& a) ->
533 typename std::enable_if<!std::is_same<any, T>::value, const any&>::type {
534 return a[N];
535 }
537 template <size_t N>
538 inline auto get(const any& a) -> const any& {
539 return cast<any::data<N+1>>(a)[N];
540 }
542 template <typename T>
543 inline auto ctor(const T& a) -> int {
544 return get<0>(a);
545 }
548} // namespace PureScript
550#undef WITH_ALLOCATOR
551#undef IS_POINTER_TYPE
552#undef DEFINE_OPERATOR
553#undef DEFINE_COMPARISON_OPERATOR
555#endif // PureScript_HH