RAPPEL : Opération de maintenance > ForgeMIA indisponible le 20 Janvier entre 7h et 12h

core.hpp 190 KB
Newer Older
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1
2
3
4
5
6
7
8
9
// Copyright (c) 2020 INRA Distributed under the Boost Software License,
// Version 1.0. (See accompanying file LICENSE_1_0.txt or copy at
// http://www.boost.org/LICENSE_1_0.txt)

#ifndef ORG_VLEPROJECT_IRRITATOR_2020
#define ORG_VLEPROJECT_IRRITATOR_2020

#include <algorithm>
#include <limits>
Gauthier Quesnel's avatar
Gauthier Quesnel committed
10
11
12

#ifdef __has_include
#if __has_include(<numbers>)
Gauthier Quesnel's avatar
Gauthier Quesnel committed
13
#include <numbers>
Gauthier Quesnel's avatar
Gauthier Quesnel committed
14
15
16
17
18
19
#define irt_have_numbers 1
#else
#define irt_have_numbers 0
#endif
#endif

Gauthier Quesnel's avatar
Gauthier Quesnel committed
20
#include <string_view>
21
#include <type_traits>
Gauthier Quesnel's avatar
Gauthier Quesnel committed
22
23
24
25

#include <cmath>
#include <cstdint>
#include <cstring>
K-H-Ismail's avatar
K-H-Ismail committed
26

27
28
29
30
31
32
/*****************************************************************************
 *
 * Helper macros
 *
 ****************************************************************************/

33
34
35
36
37
#ifndef irt_assert
#include <cassert>
#define irt_assert(_expr) assert(_expr)
#endif

38
39
40
41
42
43
namespace irt {

static inline bool is_fatal_breakpoint = true;

}

44
45
46
47
48
#ifndef NDEBUG
#if (defined(__i386__) || defined(__x86_64__)) && defined(__GNUC__) &&         \
  __GNUC__ >= 2
#define irt_breakpoint()                                                       \
    do {                                                                       \
49
50
        if (::irt::is_fatal_breakpoint)                                        \
            __asm__ __volatile__("int $03");                                   \
51
52
53
54
    } while (0)
#elif (defined(_MSC_VER) || defined(__DMC__)) && defined(_M_IX86)
#define irt_breakpoint()                                                       \
    do {                                                                       \
55
56
        if (::irt::is_fatal_breakpoint)                                        \
            __asm int 3h                                                       \
57
58
59
60
    } while (0)
#elif defined(_MSC_VER)
#define irt_breakpoint()                                                       \
    do {                                                                       \
61
62
        if (::irt::is_fatal_breakpoint)                                        \
            __debugbreak();                                                    \
63
64
65
66
67
    } while (0)
#elif defined(__alpha__) && !defined(__osf__) && defined(__GNUC__) &&          \
  __GNUC__ >= 2
#define irt_breakpoint()                                                       \
    do {                                                                       \
68
69
        if (::irt::is_fatal_breakpoint)                                        \
            __asm__ __volatile__("bpt");                                       \
70
71
72
73
    } while (0)
#elif defined(__APPLE__)
#define irt_breakpoint()                                                       \
    do {                                                                       \
74
75
        if (::irt::is_fatal_breakpoint)                                        \
            __builtin_trap();                                                  \
76
77
78
79
    } while (0)
#else /* !__i386__ && !__alpha__ */
#define irt_breakpoint()                                                       \
    do {                                                                       \
80
81
        if (::irt::is_fatal_breakpoint)                                        \
            raise(SIGTRAP);                                                    \
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
    } while (0)
#endif /* __i386__ */
#else
#define irt_breakpoint()                                                       \
    do {                                                                       \
    } while (0)
#endif

#define irt_bad_return(status__)                                               \
    do {                                                                       \
        irt_breakpoint();                                                      \
        return status__;                                                       \
    } while (0)

#define irt_return_if_bad(expr__)                                              \
    do {                                                                       \
        auto status__ = (expr__);                                              \
        if (status__ != status::success) {                                     \
            irt_breakpoint();                                                  \
            return status__;                                                   \
        }                                                                      \
    } while (0)

#define irt_return_if_fail(expr__, status__)                                   \
    do {                                                                       \
        if (!(expr__)) {                                                       \
            irt_breakpoint();                                                  \
            return status__;                                                   \
        }                                                                      \
    } while (0)

113
114
115
116
117
118
119
120
#if defined(__GNUC__)
#define irt_unreachable() __builtin_unreachable();
#elif defined(_MSC_VER)
#define irt_unreachable() __assume(0)
#else
#define irt_unreachable()
#endif

Gauthier Quesnel's avatar
Gauthier Quesnel committed
121
122
123
124
125
126
127
128
129
130
131
132
namespace irt {

using i8 = int8_t;
using i16 = int16_t;
using i32 = int32_t;
using i64 = int64_t;
using u8 = uint8_t;
using u16 = uint16_t;
using u32 = uint32_t;
using u64 = uint64_t;
using sz = size_t;

133
134
135
136
137
138
139
140
template<typename Integer>
constexpr typename std::make_unsigned<Integer>::type
to_unsigned(Integer value)
{
    irt_assert(value >= 0);
    return static_cast<typename std::make_unsigned<Integer>::type>(value);
}

Gauthier Quesnel's avatar
Gauthier Quesnel committed
141
142
143
144
145
146
147
148
149
/*****************************************************************************
 *
 * Return status of many function
 *
 ****************************************************************************/

enum class status
{
    success,
Gauthier Quesnel's avatar
Gauthier Quesnel committed
150
    unknown_dynamics,
Gauthier Quesnel's avatar
Gauthier Quesnel committed
151
152
153
154
155
156
157
158
159
160
    block_allocator_bad_capacity,
    block_allocator_not_enough_memory,
    head_allocator_bad_capacity,
    head_allocator_not_enough_memory,
    simulation_not_enough_model,
    simulation_not_enough_memory_message_list_allocator,
    simulation_not_enough_memory_input_port_list_allocator,
    simulation_not_enough_memory_output_port_list_allocator,
    data_array_init_capacity_error,
    data_array_not_enough_memory,
161
162
    data_array_archive_init_capacity_error,
    data_array_archive_not_enough_memory,
163
164
165
    array_init_capacity_zero,
    array_init_capacity_too_big,
    array_init_not_enough_memory,
166
167
168
    vector_init_capacity_zero,
    vector_init_capacity_too_big,
    vector_init_not_enough_memory,
Gauthier Quesnel's avatar
Gauthier Quesnel committed
169
170
171
    dynamics_unknown_id,
    dynamics_unknown_port_id,
    dynamics_not_enough_memory,
Gauthier Quesnel's avatar
Gauthier Quesnel committed
172
173
174
    model_connect_output_port_unknown,
    model_connect_input_port_unknown,
    model_connect_already_exist,
175
    model_connect_bad_dynamics,
176
177
178
179

    model_buffer_null_ta_source,
    model_buffer_empty_ta_source,

Gauthier Quesnel's avatar
Gauthier Quesnel committed
180
181
    model_integrator_dq_error,
    model_integrator_X_error,
Gauthier Quesnel's avatar
Gauthier Quesnel committed
182
183
184
185
    model_integrator_internal_error,
    model_integrator_output_error,
    model_integrator_running_without_x_dot,
    model_integrator_ta_with_bad_x_dot,
186
187
188
189
190
191

    model_generator_null_ta_source,
    model_generator_empty_ta_source,
    model_generator_null_value_source,
    model_generator_empty_value_source,

Gauthier Quesnel's avatar
Gauthier Quesnel committed
192
193
194
195
    model_quantifier_bad_quantum_parameter,
    model_quantifier_bad_archive_length_parameter,
    model_quantifier_shifting_value_neg,
    model_quantifier_shifting_value_less_1,
K-H-Ismail's avatar
K-H-Ismail committed
196
    model_time_func_bad_init_message,
197
198
    model_flow_bad_samplerate,
    model_flow_bad_data,
199
    gui_not_enough_memory,
200
    io_not_enough_memory,
Gauthier Quesnel's avatar
Gauthier Quesnel committed
201
202
203
204
205
206
207
    io_file_format_error,
    io_file_format_model_error,
    io_file_format_model_number_error,
    io_file_format_model_unknown,
    io_file_format_dynamics_unknown,
    io_file_format_dynamics_limit_reach,
    io_file_format_dynamics_init_error
Gauthier Quesnel's avatar
Gauthier Quesnel committed
208
209
};

210
211
212
213
214
215
216
constexpr i8
status_last() noexcept
{
    return static_cast<i8>(status::io_file_format_dynamics_init_error);
}

constexpr sz
217
218
status_size() noexcept
{
219
    return static_cast<sz>(status_last() + static_cast<i8>(1));
220
221
}

Gauthier Quesnel's avatar
Gauthier Quesnel committed
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
constexpr bool
is_success(status s) noexcept
{
    return s == status::success;
}

constexpr bool
is_bad(status s) noexcept
{
    return s != status::success;
}

template<typename... Args>
constexpr bool
is_status_equal(status s, Args... args) noexcept
{
    return ((s == args) || ... || false);
}

241
242
243
244
245
246
247
248
249
inline status
check_return(status s) noexcept
{
    if (s != status::success)
        irt_breakpoint();

    return s;
}

250
251
template<typename T, typename... Args>
constexpr bool
252
match(const T& s, Args... args) noexcept
253
254
255
256
{
    return ((s == args) || ... || false);
}

257
258
259
template<class T, class... Rest>
constexpr bool
are_all_same() noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
260
{
261
    return (std::is_same_v<T, Rest> && ...);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
262
263
}

Gauthier Quesnel's avatar
Gauthier Quesnel committed
264
265
266
267
268
269
270
271
272
template<class T>
typename std::enable_if<!std::numeric_limits<T>::is_integer, bool>::type
almost_equal(T x, T y, int ulp)
{
    return std::fabs(x - y) <=
             std::numeric_limits<T>::epsilon() * std::fabs(x + y) * ulp ||
           std::fabs(x - y) < std::numeric_limits<T>::min();
}

Gauthier Quesnel's avatar
Gauthier Quesnel committed
273
274
275
276
277
278
279
280
281
282
283
284
285
/*****************************************************************************
 *
 * Definition of a lightweight std::function
 *
 ****************************************************************************/

template<class F>
class function_ref;

template<class R, class... Args>
class function_ref<R(Args...)>
{
public:
286
    constexpr function_ref() noexcept = default;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342

    /// Creates a `function_ref` which refers to the same callable as `rhs`.
    constexpr function_ref(const function_ref<R(Args...)>& rhs) noexcept =
      default;

    /// Constructs a `function_ref` referring to `f`.
    ///
    /// \synopsis template <typename F> constexpr function_ref(F &&f) noexcept
    template<
      typename F,
      std::enable_if_t<!std::is_same<std::decay_t<F>, function_ref>::value &&
                       std::is_invocable_r<R, F&&, Args...>::value>* = nullptr>
    constexpr function_ref(F&& f) noexcept
      : obj(const_cast<void*>(reinterpret_cast<const void*>(std::addressof(f))))
    {
        cb = [](void* obj, Args... args) -> R {
            return std::invoke(
              *reinterpret_cast<typename std::add_pointer<F>::type>(obj),
              std::forward<Args>(args)...);
        };
    }

    /// Makes `*this` refer to the same callable as `rhs`.
    constexpr function_ref<R(Args...)>& operator=(
      const function_ref<R(Args...)>& rhs) noexcept = default;

    /// Makes `*this` refer to `f`.
    ///
    /// \synopsis template <typename F> constexpr function_ref &operator=(F &&f)
    /// noexcept;
    template<
      typename F,
      std::enable_if_t<std::is_invocable_r<R, F&&, Args...>::value>* = nullptr>
    constexpr function_ref<R(Args...)>& operator=(F&& f) noexcept
    {
        obj = reinterpret_cast<void*>(std::addressof(f));
        cb = [](void* obj, Args... args) {
            return std::invoke(
              *reinterpret_cast<typename std::add_pointer<F>::type>(obj),
              std::forward<Args>(args)...);
        };

        return *this;
    }

    constexpr void swap(function_ref<R(Args...)>& rhs) noexcept
    {
        std::swap(obj, rhs.obj);
        std::swap(cb, rhs.cb);
    }

    R operator()(Args... args) const
    {
        return cb(obj, std::forward<Args>(args)...);
    }

343
344
345
346
347
    constexpr bool empty() const noexcept
    {
        return obj == nullptr;
    }

348
349
350
351
352
353
    constexpr void reset() noexcept
    {
        obj = nullptr;
        cb = nullptr;
    }

Gauthier Quesnel's avatar
Gauthier Quesnel committed
354
355
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
private:
    void* obj = nullptr;
    R (*cb)(void*, Args...) = nullptr;
};

/// Swaps the referred callables of `lhs` and `rhs`.
template<typename R, typename... Args>
constexpr void
swap(function_ref<R(Args...)>& lhs, function_ref<R(Args...)>& rhs) noexcept
{
    lhs.swap(rhs);
}

template<typename R, typename... Args>
function_ref(R (*)(Args...)) -> function_ref<R(Args...)>;

template<typename R, typename... Args>
function_ref(R (*)(Args...) noexcept) -> function_ref<R(Args...) noexcept>;

373
374
375
376
377
378
379
/*****************************************************************************
 *
 * Allocator
 *
 ****************************************************************************/

using global_alloc_function_type = function_ref<void*(sz size)>;
380
using global_free_function_type = function_ref<void(void* ptr)>;
381

382
383
static inline void*
malloc_wrapper(sz size)
384
385
386
387
{
    return std::malloc(size);
}

388
389
static inline void
free_wrapper(void* ptr)
390
391
392
393
394
{
    if (ptr)
        std::free(ptr);
}

395
396
static inline global_alloc_function_type g_alloc_fn{ malloc_wrapper };
static inline global_free_function_type g_free_fn{ free_wrapper };
397

Gauthier Quesnel's avatar
Gauthier Quesnel committed
398
399
400
401
402
403
404
405
406
407
408
409
410
411
412
413
414
415
416
417
418
419
420
421
422
423
424
/*****************************************************************************
 *
 * Definition of Time
 *
 * TODO:
 * - enable template definition of float or [float|double,bool absolute]
 *   representation of time?
 *
 ****************************************************************************/

using time = double;

template<typename T>
struct time_domain
{};

template<>
struct time_domain<time>
{
    using time_type = time;

    static constexpr const double infinity =
      std::numeric_limits<double>::infinity();
    static constexpr const double negative_infinity =
      -std::numeric_limits<double>::infinity();
    static constexpr const double zero = 0;

425
    static constexpr bool is_infinity(time t) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
426
427
428
429
    {
        return t == infinity || t == negative_infinity;
    }

430
    static constexpr bool is_zero(time t) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
431
432
433
434
435
436
437
438
439
440
441
442
443
444
445
446
447
448
    {
        return t == zero;
    }
};

/*****************************************************************************
 *
 * Small string
 *
 ****************************************************************************/

template<size_t length = 8>
class small_string
{
    char buffer_[length];
    u8 size_;

public:
Gauthier Quesnel's avatar
Gauthier Quesnel committed
449
450
    using iterator = char*;
    using const_iterator = const char*;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
451
452
453
454
455
456
457
458
459
460
461
462
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
478
479
480
481
482
483
484
485
486
487
488
489
490
491
492
493
494
495
496
497
498
499
500
501
502
503
504
505
506
507
508
509
510
511
512
513
514
515
    using size_type = u8;

    static_assert(length > size_t{ 1 } && length < size_t{ 254 });

    constexpr small_string() noexcept
    {
        clear();
    }

    constexpr small_string(const small_string& str) noexcept
    {
        std::copy_n(str.buffer_, str.size_, buffer_);
        buffer_[str.size_] = '\0';
        size_ = str.size_;
    }

    constexpr small_string(small_string&& str) noexcept
    {
        std::copy_n(str.buffer_, str.size_, buffer_);
        buffer_[str.size_] = '\0';
        size_ = str.size_;
        str.clear();
    }

    constexpr small_string& operator=(const small_string& str) noexcept
    {
        if (&str != this) {
            std::copy_n(str.buffer_, str.size_, buffer_);
            buffer_[str.size_] = '\0';
            size_ = str.size_;
        }

        return *this;
    }

    constexpr small_string& operator=(small_string&& str) noexcept
    {
        if (&str != this) {
            std::copy_n(str.buffer_, str.size_, buffer_);
            buffer_[str.size_] = '\0';
            size_ = str.size_;
        }

        return *this;
    }

    constexpr small_string(const char* str) noexcept
    {
        std::strncpy(buffer_, str, length - 1);
        buffer_[length - 1] = '\0';
        size_ = static_cast<u8>(std::strlen(buffer_));
    }

    constexpr small_string(const std::string_view str) noexcept
    {
        assign(str);
    }

    constexpr bool empty() const noexcept
    {
        constexpr unsigned char zero{ 0 };

        return zero == size_;
    }

Gauthier Quesnel's avatar
Gauthier Quesnel committed
516
517
    constexpr void size(std::size_t sz) noexcept
    {
518
519
        size_ = static_cast<u8>(std::min(sz, length - 1));
        buffer_[size_] = '\0';
Gauthier Quesnel's avatar
Gauthier Quesnel committed
520
521
    }

Gauthier Quesnel's avatar
Gauthier Quesnel committed
522
523
524
525
526
527
528
    constexpr std::size_t size() const noexcept
    {
        return size_;
    }

    constexpr std::size_t capacity() const noexcept
    {
529
        return length;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
530
531
532
533
    }

    constexpr void assign(const std::string_view str) noexcept
    {
534
        const size_t copy_length = std::min(str.size(), length - 1);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
535
536

        std::memcpy(buffer_, str.data(), copy_length);
537
538
        buffer_[copy_length] = '\0';

539
        size_ = static_cast<u8>(copy_length);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
    }

    constexpr std::string_view sv() const noexcept
    {
        return { buffer_, size_ };
    }

    constexpr void clear() noexcept
    {
        std::fill_n(buffer_, length, '\0');
        size_ = 0;
    }

    constexpr const char* c_str() const noexcept
    {
        return buffer_;
    }

Gauthier Quesnel's avatar
Gauthier Quesnel committed
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
    constexpr iterator begin() noexcept
    {
        return buffer_;
    }

    constexpr iterator end() noexcept
    {
        return buffer_ + size_;
    }

    constexpr const_iterator begin() const noexcept
    {
        return buffer_;
    }

    constexpr const_iterator end() const noexcept
    {
        return buffer_ + size_;
    }

Gauthier Quesnel's avatar
Gauthier Quesnel committed
578
579
580
581
582
583
584
585
586
587
588
589
590
591
592
593
594
595
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
618
619
620
    constexpr bool operator==(const small_string& rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs.buffer_, length) == 0;
    }

    constexpr bool operator!=(const small_string& rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs.buffer_, length) != 0;
    }

    constexpr bool operator>(const small_string& rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs.buffer_, length) > 0;
    }

    constexpr bool operator<(const small_string& rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs.buffer_, length) < 0;
    }

    constexpr bool operator==(const char* rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs, length) == 0;
    }

    constexpr bool operator!=(const char* rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs, length) != 0;
    }

    constexpr bool operator>(const char* rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs, length) > 0;
    }

    constexpr bool operator<(const char* rhs) const noexcept
    {
        return std::strncmp(buffer_, rhs, length) < 0;
    }
};

/*****************************************************************************
 *
621
 * value
Gauthier Quesnel's avatar
Gauthier Quesnel committed
622
623
624
 *
 ****************************************************************************/

625
struct message
Gauthier Quesnel's avatar
Gauthier Quesnel committed
626
{
627
628
    using size_type = std::size_t;
    using difference_type = std::ptrdiff_t;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
629

630
    double real[4];
Gauthier Quesnel's avatar
Gauthier Quesnel committed
631

632
633
634
635
636
637
    constexpr size_type size() const noexcept
    {
        return real[3] ? 4u : real[2] ? 3u : real[1] ? 2u : real[0] ? 1u : 0u;
    }

    constexpr difference_type ssize() const noexcept
638
    {
639
        return real[3] ? 4 : real[2] ? 3 : real[1] ? 2 : real[0] ? 1 : 0;
640
    }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
641

642
    constexpr message() noexcept
643
      : real{ 0., 0., 0., 0. }
644
    {}
Gauthier Quesnel's avatar
Gauthier Quesnel committed
645

646
    constexpr message(const double v) noexcept
647
      : real{ v, 0., 0., 0. }
648
649
650
    {}

    constexpr message(const double v1, const double v2) noexcept
651
      : real{ v1, v2, 0., 0. }
652
653
654
655
656
    {}

    constexpr message(const double v1,
                      const double v2,
                      const double v3) noexcept
657
658
659
660
661
662
663
664
      : real{ v1, v2, v3, 0. }
    {}

    constexpr message(const double v1,
                      const double v2,
                      const double v3,
                      const double v4) noexcept
      : real{ v1, v2, v3, v4 }
665
    {}
Gauthier Quesnel's avatar
Gauthier Quesnel committed
666

667
    constexpr double operator[](const difference_type i) const noexcept
668
    {
669
        return real[i];
Gauthier Quesnel's avatar
Gauthier Quesnel committed
670
    }
671

672
    constexpr double& operator[](const difference_type i) noexcept
673
    {
674
        return real[i];
675
    }
676
677
678
679
680
681
682
683

    constexpr void reset() noexcept
    {
        real[0] = 0.;
        real[1] = 0.;
        real[2] = 0.;
        real[3] = 0.;
    }
684
};
Gauthier Quesnel's avatar
Gauthier Quesnel committed
685

686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
/*****************************************************************************
 *
 * Flat list
 *
 ****************************************************************************/

template<typename T>
class block_allocator
{
public:
    static_assert(std::is_trivially_destructible_v<T>,
                  "block_allocator is only for POD object");

    using value_type = T;

    union block
Gauthier Quesnel's avatar
Gauthier Quesnel committed
702
    {
703
704
705
        block* next;
        typename std::aligned_storage<sizeof(T), alignof(T)>::type storage;
    };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
706

707
private:
Gauthier Quesnel's avatar
Gauthier Quesnel committed
708
    block* blocks{ nullptr };    // contains all preallocated blocks
709
710
711
712
    block* free_head{ nullptr }; // a free list
    sz size{ 0 };                // number of active elements allocated
    sz max_size{ 0 }; // number of elements allocated (with free_head)
    sz capacity{ 0 }; // capacity of the allocator
Gauthier Quesnel's avatar
Gauthier Quesnel committed
713

714
715
public:
    block_allocator() = default;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
716

717
718
    block_allocator(const block_allocator&) = delete;
    block_allocator& operator=(const block_allocator&) = delete;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
719

720
721
722
    ~block_allocator() noexcept
    {
        if (blocks)
723
            g_free_fn(blocks);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
724
725
    }

726
    status init(std::size_t new_capacity) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
727
    {
728
729
        if (new_capacity == 0)
            return status::block_allocator_bad_capacity;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
730

731
732
        if (new_capacity != capacity) {
            if (blocks)
733
734
                g_free_fn(blocks);

735
            blocks =
736
              static_cast<block*>(g_alloc_fn(new_capacity * sizeof(block)));
737
738
739
            if (blocks == nullptr)
                return status::block_allocator_not_enough_memory;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
740

741
742
743
744
        size = 0;
        max_size = 0;
        capacity = new_capacity;
        free_head = nullptr;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
745

746
        return status::success;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
747
748
    }

749
    void reset() noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
750
    {
751
752
753
754
755
        if (capacity > 0) {
            size = 0;
            max_size = 0;
            free_head = nullptr;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
756
757
    }

758
    T* alloc() noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
759
    {
760
761
762
763
764
765
        block* new_block = nullptr;

        if (free_head != nullptr) {
            new_block = free_head;
            free_head = free_head->next;
        } else {
766
            irt_assert(max_size < capacity);
767
            new_block = reinterpret_cast<block*>(&blocks[max_size++]);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
768
        }
769
        ++size;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
770

771
        return reinterpret_cast<T*>(new_block);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
772
773
    }

774
775
776
777
778
    bool can_alloc() noexcept
    {
        return free_head != nullptr || max_size < capacity;
    }

779
    void free(T* n) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
780
    {
781
        irt_assert(n);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
782

783
        block* ptr = reinterpret_cast<block*>(n);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
784

785
786
        ptr->next = free_head;
        free_head = ptr;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
787

788
        --size;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
789

790
791
792
793
        if (size == 0) {         // A special part: if it no longer exists
            max_size = 0;        // we reset the free list and the number
            free_head = nullptr; // of elements allocated.
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
794
    }
795
796
797
798
799

    bool can_alloc(size_t number) const noexcept
    {
        return number + size < capacity;
    }
800
};
Gauthier Quesnel's avatar
Gauthier Quesnel committed
801

802
803
804
805
806
template<typename T>
class flat_list
{
public:
    struct node_type
Gauthier Quesnel's avatar
Gauthier Quesnel committed
807
    {
808
809
810
        T value;
        node_type* next = nullptr;
    };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
811

812
813
814
815
public:
    using allocator_type = block_allocator<node_type>;
    using value_type = T;
    using reference = T&;
816
    using const_reference = const T&;
817
    using pointer = T*;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
818

819
    class iterator
Gauthier Quesnel's avatar
Gauthier Quesnel committed
820
    {
821
822
    private:
        node_type* node{ nullptr };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
823

824
825
826
827
828
829
    public:
        using iterator_category = std::forward_iterator_tag;
        using value_type = T;
        using pointer = T*;
        using reference = T&;
        using difference_type = std::ptrdiff_t;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
830

831
        iterator() noexcept = default;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
832

833
834
835
        iterator(node_type* n) noexcept
          : node(n)
        {}
Gauthier Quesnel's avatar
Gauthier Quesnel committed
836

837
838
839
        iterator(const iterator& other) noexcept
          : node(other.node)
        {}
Gauthier Quesnel's avatar
Gauthier Quesnel committed
840

841
842
843
844
845
        iterator& operator=(const iterator& other) noexcept
        {
            node = other.node;
            return *this;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
846

847
848
849
850
        T& operator*() noexcept
        {
            return node->value;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
851

852
853
854
855
        T* operator->() noexcept
        {
            return &node->value;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
856

857
858
859
860
        iterator operator++() noexcept
        {
            if (node != nullptr)
                node = node->next;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
861

862
863
            return *this;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
864

865
866
867
        iterator operator++(int) noexcept
        {
            iterator tmp(*this);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
868

869
870
            if (node != nullptr)
                node = node->next;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
871

872
873
            return tmp;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
874

875
876
877
878
        bool operator==(const iterator& other) const noexcept
        {
            return node == other.node;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
879

880
881
882
883
        bool operator!=(const iterator& other) const noexcept
        {
            return node != other.node;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
884

885
886
887
888
        void swap(iterator& other) noexcept
        {
            std::swap(node, other.node);
        }
889

Gauthier Quesnel's avatar
Gauthier Quesnel committed
890
        friend class flat_list<T>;
891
    };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
892

893
    class const_iterator
Gauthier Quesnel's avatar
Gauthier Quesnel committed
894
    {
895
896
    private:
        const node_type* node{ nullptr };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
897

898
899
900
901
902
903
    public:
        using iterator_category = std::forward_iterator_tag;
        using value_type = T;
        using pointer = T*;
        using reference = T&;
        using difference_type = std::ptrdiff_t;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
904

905
        const_iterator() noexcept = default;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
906

907
908
909
        const_iterator(node_type* n) noexcept
          : node(n)
        {}
Gauthier Quesnel's avatar
Gauthier Quesnel committed
910

911
912
913
914
915
916
917
918
919
920
921
922
923
924
925
926
927
928
929
930
931
932
933
934
935
936
937
938
939
940
941
942
943
944
945
946
947
948
949
950
951
952
953
954
955
956
957
958
959
960
961
962
        const_iterator(const const_iterator& other) noexcept
          : node(other.node)
        {}

        const_iterator& operator=(const const_iterator& other) noexcept
        {
            node = other.node;
            return *this;
        }

        const T& operator*() noexcept
        {
            return node->value;
        }

        const T* operator->() noexcept
        {
            return &node->value;
        }

        const_iterator operator++() noexcept
        {
            if (node != nullptr)
                node = node->next;

            return *this;
        }

        const_iterator operator++(int) noexcept
        {
            const_iterator tmp(*this);

            if (node != nullptr)
                node = node->next;

            return tmp;
        }

        bool operator==(const const_iterator& other) const noexcept
        {
            return node == other.node;
        }

        bool operator!=(const const_iterator& other) const noexcept
        {
            return node != other.node;
        }

        void swap(const_iterator& other) noexcept
        {
            std::swap(node, other.node);
        }
963

Gauthier Quesnel's avatar
Gauthier Quesnel committed
964
        friend class flat_list<T>;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
965
966
    };

967
968
969
private:
    allocator_type* allocator{ nullptr };
    node_type* node{ nullptr };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
970

971
972
public:
    flat_list() = default;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
973

974
975
    flat_list(allocator_type* allocator_new) noexcept
      : allocator(allocator_new)
Gauthier Quesnel's avatar
Gauthier Quesnel committed
976
977
    {}

978
979
980
981
982
983
    flat_list(const flat_list& other) = delete;
    flat_list& operator=(const flat_list& other) = delete;

    flat_list(flat_list&& other) noexcept
      : allocator(other.allocator)
      , node(other.node)
Gauthier Quesnel's avatar
Gauthier Quesnel committed
984
    {
985
986
        other.allocator = nullptr;
        other.node = nullptr;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
987
988
    }

989
    void set_allocator(allocator_type* allocator_new) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
990
    {
991
992
993
        clear();
        allocator = allocator_new;
        node = nullptr;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
994
995
    }

996
    flat_list& operator=(flat_list&& other) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
997
    {
998
999
1000
1001
1002
1003
1004
1005
1006
        if (this != &other) {
            clear();
            allocator = other.allocator;
            other.allocator = nullptr;
            node = other.node;
            other.node = nullptr;
        }

        return *this;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1007
1008
    }

1009
    ~flat_list() noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1010
    {
1011
        clear();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1012
1013
    }

1014
    void clear() noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1015
    {
1016
1017
1018
1019
1020
1021
1022
        node_type* prev = node;

        while (node != nullptr) {
            node = node->next;
            allocator->free(prev);
            prev = node;
        }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1023
1024
    }

1025
    bool empty() const noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1026
    {
1027
        return node == nullptr;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1028
1029
    }

1030
    iterator begin() noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1031
    {
1032
1033
        return iterator(node);
    }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1034

1035
1036
1037
1038
    iterator end() noexcept
    {
        return iterator(nullptr);
    }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1039

1040
1041
1042
1043
    const_iterator begin() const noexcept
    {
        return const_iterator(node);
    }
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1044

1045
1046
1047
    const_iterator end() const noexcept
    {
        return const_iterator(nullptr);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1048
1049
    }

1050
1051
    reference front() noexcept
    {
1052
        irt_assert(!empty());
1053
1054
1055
1056
1057
1058

        return node->value;
    }

    const_reference front() const noexcept
    {
1059
        irt_assert(!empty());
1060
1061
1062
1063

        return node->value;
    }

1064
1065
    template<typename... Args>
    iterator emplace_front(Args&&... args) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1066
    {
1067
        node_type* new_node = allocator->alloc();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1068

1069
        new (&new_node->value) T(std::forward<Args>(args)...);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1070

1071
1072
        new_node->next = node;
        node = new_node;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1073

1074
        return begin();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1075
1076
    }

1077
1078
    template<typename... Args>
    iterator emplace_after(iterator it, Args&&... args) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1079
    {
1080
1081
        node_type* new_node = allocator->alloc();
        new (&new_node->value) T(std::forward<Args>(args)...);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1082

1083
1084
        if (it->node == nullptr)
            return emplace_front(std::forward<Args>(args)...);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1085

1086
1087
        new_node->next = it->node->next;
        it->node->next = new_node;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1088

1089
        return iterator(new_node);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1090
1091
    }

1092
1093
    template<typename... Args>
    iterator try_emplace_front(Args&&... args) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1094
    {
1095
        auto [success, new_node] = allocator->try_alloc();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1096

1097
1098
        if (!success)
            return end();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1099

1100
        new (&new_node->value) T(std::forward<Args>(args)...);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1101

1102
1103
1104
1105
        new_node->next = node;
        node = new_node;

        return begin();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1106
1107
    }

1108
1109
    template<typename... Args>
    iterator try_emplace_after(iterator it, Args&&... args) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1110
    {
1111
        auto [success, new_node] = allocator->try_alloc();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1112

1113
1114
        if (!success)
            return end();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1115

1116
        new (&new_node->value) T(std::forward<Args>(args)...);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1117

1118
1119
        if (it->node == nullptr)
            return emplace_front(std::forward<Args>(args)...);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1120

1121
1122
        new_node->next = it->node->next;
        it->node->next = new_node;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1123

1124
        return iterator(new_node);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1125
1126
    }

1127
    void pop_front() noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1128
    {
1129
1130
        if (node == nullptr)
            return;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1131

1132
1133
        node_type* to_delete = node;
        node = node->next;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1134

1135
        if constexpr (!std::is_trivial_v<T>)
1136
            to_delete->value.~T();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1137

1138
        allocator->free(to_delete);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1139
1140
    }

1141
    iterator erase_after(iterator it) noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1142
    {
1143
        irt_assert(allocator);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1144

1145
1146
        if (it.node == nullptr)
            return end();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1147

1148
1149
1150
        node_type* to_delete = it.node->next;
        if (to_delete == nullptr)
            return end();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1151

1152
1153
        node_type* next = to_delete->next;
        it.node->next = next;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1154

1155
        if constexpr (!std::is_trivial_v<T>)
1156
            to_delete->value.~T();
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1157

1158
        allocator->free(to_delete);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1159

1160
        return iterator(next);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1161
1162
1163
1164
    }
};

template<typename T>
1165
class flat_double_list
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1166
{
1167
private:
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1168
1169
1170
1171
    struct node_type
    {
        T value;
        node_type* next = nullptr;
1172
        node_type* prev = nullptr;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1173
1174
1175
1176
1177
1178
1179
1180
1181
1182
1183
    };

public:
    using allocator_type = block_allocator<node_type>;
    using value_type = T;
    using reference = T&;
    using pointer = T*;

    class iterator
    {
    private:
1184
        flat_double_list* list{ nullptr };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1185
1186
1187
        node_type* node{ nullptr };

    public:
1188
        using iterator_category = std::bidirectional_iterator_tag;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1189
1190
1191
1192
1193
1194
        using value_type = T;
        using pointer = T*;
        using reference = T&;

        iterator() noexcept = default;

1195
1196
1197
        iterator(flat_double_list* lst, node_type* n) noexcept
          : list(lst)
          , node(n)
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1198
1199
1200
        {}

        iterator(const iterator& other) noexcept
1201
1202
          : list(other.list)
          , node(other.node)
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1203
1204
1205
1206
        {}

        iterator& operator=(const iterator& other) noexcept
        {
1207
            list = other.list;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1208
1209
1210
1211
1212
1213
1214
1215
1216
            node = other.node;
            return *this;
        }

        T& operator*() noexcept
        {
            return node->value;
        }

1217
1218
1219
1220
1221
1222
1223
1224
1225
1226
1227
        T* operator*() const noexcept
        {
            return node->value;
        }

        pointer operator->() noexcept
        {
            return &(node->value);
        }

        pointer operator->() const noexcept
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1228
1229
1230
1231
1232
1233
        {
            return &node->value;
        }

        iterator operator++() noexcept
        {
1234
            node = (node == nullptr) ? list->m_front : node->next;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1235
1236
1237
1238
1239
1240
1241
1242

            return *this;
        }

        iterator operator++(int) noexcept
        {
            iterator tmp(*this);

1243
1244
1245
1246
1247
1248
1249
1250
1251
1252
1253
1254
1255
1256
1257
1258
1259
            node = (node == nullptr) ? list->m_front : node->next;

            return tmp;
        }

        iterator operator--() noexcept
        {
            node = (node == nullptr) ? list->m_back : node->prev;

            return *this;
        }

        iterator operator--(int) noexcept
        {
            iterator tmp(*this);

            node = (node == nullptr) ? list->m_back : node->prev;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1260
1261
1262
1263
1264
1265
1266
1267
1268
1269
1270
1271
1272
1273
1274
1275

            return tmp;
        }

        bool operator==(const iterator& other) const noexcept
        {
            return node == other.node;
        }

        bool operator!=(const iterator& other) const noexcept
        {
            return node != other.node;
        }

        void swap(iterator& other) noexcept
        {
1276
            std::swap(list, other.list);
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1277
1278
1279
1280
1281
1282
1283
            std::swap(node, other.node);
        }
    };

    class const_iterator
    {
    private:
1284
1285
        const flat_double_list* list{ nullptr };
        node_type* node{ nullptr };
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1286
1287

    public:
1288
        using const_iterator_category = std::bidirectional_iterator_tag;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1289
1290
1291
1292
1293
1294
        using value_type = T;
        using pointer = T*;
        using reference = T&;

        const_iterator() noexcept = default;

1295
1296
1297
        const_iterator(const flat_double_list* lst, node_type* n) noexcept
          : list(lst)
          , node(n)
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1298
1299
1300
        {}

        const_iterator(const const_iterator& other) noexcept
1301
1302
          : list(other.list)
          , node(other.node)
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1303
1304
1305
1306
        {}

        const_iterator& operator=(const const_iterator& other) noexcept
        {
1307
            list = other.list;
Gauthier Quesnel's avatar
Gauthier Quesnel committed
1308
1309
1310
1311
1312
1313
1314
1315
1316
1317
1318
            node = other.node;