cache2.h 33 KB
Newer Older
Damien Leroux's avatar
Damien Leroux committed
1
2
3
4
5
6
7
8
9
#ifndef _SPEL_CACHE_2_H_
#define _SPEL_CACHE_2_H_

#include <future>
#include <tuple>
#include <unordered_map>
#include <iostream>
#include <string>
#include <sstream>
10
11
#include "function_wrapper.h"
#include "chrono.h"
Damien Leroux's avatar
Damien Leroux committed
12

13

14
15
16
17
18
19
20
21


struct chrono_trace {
    const std::string& name;
    chrono_trace(const std::string& n) : name(n) { chrono::increment(name); chrono::start(name); }
    ~chrono_trace() { chrono::stop(name); }
};

22
23
24
25
enum CachingPolicy : int { Oneshot = 0, Mem = 1, Disk = 2, Sync = 4 };
constexpr CachingPolicy operator | (CachingPolicy c1, CachingPolicy c2) { return CachingPolicy(int(c1) | int(c2)); }
constexpr CachingPolicy operator & (CachingPolicy c1, CachingPolicy c2) { return CachingPolicy(int(c1) & int(c2)); }

26
27
28
29
30
template <typename X> struct clean_type { typedef typename std::remove_reference<X>::type type; };
template <typename X> struct clean_type<const X&> { typedef X type; };
template <typename X> struct clean_type<X&> { typedef X type; };
template <typename X> struct clean_type<const X> { typedef X type; };

Damien Leroux's avatar
Damien Leroux committed
31
32
33
34
#include "error.h"
#include "input.h"
#include "cache/md5.h"
#include "cache/file.h"
35
#include "cache/registry.h"
Damien Leroux's avatar
Damien Leroux committed
36

37
extern "C" {
38
/*#include <dlfcn.h>*/
39
#include <malloc.h>
40
/*#include <string.h>*/
41
}
42
/*#include <cxxabi.h>*/
43
44
45
46
47
48
49
50

/*
char*
__cxa_demangle(const char* __mangled_name, char* __output_buffer, size_t* __length, int* __status);
*/

/*using demangle = abi::__cxa_demangle;*/

51
#if 0
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
static inline
std::unordered_map<void*, std::string>&
demangled_names_registry()
{
	static std::unordered_map<void*, std::string> _;
	return _;
}


template <typename Ret, typename... Args>
std::string&
get_func_name(Ret (*f) (Args...))
{
	union {
		Ret (*fptr) (Args...);
		void* vptr;
	} tmp = {f};
	std::string& ret = demangled_names_registry()[tmp.vptr];
	if (!ret.size()) {
		Dl_info info;
		dladdr(tmp.vptr, &info);
		int status = 0;
		char* buf = abi::__cxa_demangle(info.dli_sname, NULL, 0, &status);
		ret.assign(buf, strchr(buf, '('));
		free(buf);
		/*std::cout << tmp.vptr << " => " << info.dli_sname << std::endl;*/
	}
	return ret;
}


template <typename Ret, typename... Args>
std::string&
get_func_name(Ret (&f) (Args...))
{
	return get_func_name(&f);
}
89
#endif
90

Damien Leroux's avatar
Damien Leroux committed
91
92
93
/* forward */ struct md5_digest;


94
static inline std::string& cache_directory() { return active_settings->work_directory; }
Damien Leroux's avatar
Damien Leroux committed
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129

template <typename T> struct generic_value_interface;
template <typename T> struct value;
template <typename T> struct range;
template <typename T> struct collection;

template <typename T> struct fail : std::integral_constant<bool, false> {};

template <typename T> struct value<collection<T>> { static_assert(fail<T>::value, "Can't instantiate this"); };


template <typename T> struct immediate_value;

template <typename FuncType> struct async_computation;
template <typename FuncType> struct cached_computation;
template <typename FuncType> struct computed_value_factory;
template <typename FuncType> struct computed_value;
template <typename FuncType> struct cached_computed_value;

/* a value<T> behaves as a pointer to T */

template <typename T>
    struct generic_value_interface {
        typedef T value_type;

        virtual ~generic_value_interface() {}

        virtual value_type& operator * () = 0;
        virtual value_type* operator -> () = 0;

        virtual const value_type& operator * () const = 0;
        virtual const value_type* operator -> () const = 0;

        virtual size_t hash() const = 0;
        virtual md5_digest& md5(md5_digest&) const = 0;
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146

        bool operator == (const generic_value_interface<T>& gvi) const
        {
            return **this == *gvi;
        }
#if 0
        /*virtual bool equal(const generic_value_interface<T>&) const { return false; }*/
        virtual
            bool equal(const generic_value_interface<T>& gvi) const
            /*override*/
            {
                return **this == *gvi;
                /*return gvi.__equal__(m_storage);*/
            }

        /*virtual bool __equal__(const T&) const = 0;  // { return false; }*/
#endif
Damien Leroux's avatar
Damien Leroux committed
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
    };

/* Lightweight */
template <typename T>
    struct value {
        typedef T value_type;

        value() : m_impl() {}
        value(generic_value_interface<T>* v) : m_impl(v) {}
        value(const T& immed) : m_impl(new immediate_value<T>(immed)) {}
        value(T&& immed) : m_impl(new immediate_value<T>(std::forward<T>(immed))) {}

        value_type& operator * () { return m_impl->operator * (); }
        value_type* operator -> () { return m_impl->operator -> (); }
        const value_type& operator * () const { return m_impl->operator * (); }
        const value_type* operator -> () const { return m_impl->operator -> (); }
        size_t hash() const { return m_impl->hash(); }
        md5_digest& md5(md5_digest& md) const { return m_impl->md5(md); }

166
        value<T>& operator = (const T& immed)
Damien Leroux's avatar
Damien Leroux committed
167
168
169
170
171
172
        {
            /*m_impl = new immediate_value<T>(immed);*/
            m_impl = std::make_shared<immediate_value<T>>(immed);
            return *this;
        }

173
        value<T>& operator = (std::shared_ptr<generic_value_interface<T>>& new_impl)
Damien Leroux's avatar
Damien Leroux committed
174
175
176
177
178
        {
            m_impl = new_impl;
            return *this;
        }

179
        value<T>& operator = (const value<T>& new_impl)
Damien Leroux's avatar
Damien Leroux committed
180
181
182
183
184
185
186
        {
            m_impl = new_impl.m_impl;
            return *this;
        }

        bool valid() const { return (bool) m_impl; }

187
188
		operator bool () const { return valid(); }

189
190
        /*bool equal(const value<T>& v) const { return m_impl == v.m_impl || m_impl->equal(*v.m_impl); }*/
        bool equal(const value<T>& v) const { return m_impl == v.m_impl || *m_impl == *v.m_impl; }
Damien Leroux's avatar
Damien Leroux committed
191
192
193
194
195
196
197
198

    protected:
        std::shared_ptr<generic_value_interface<T>> m_impl;
    };

template <typename T>
bool operator == (const value<T>& v1, const value<T>& v2) { return v1.equal(v2); }

199
200
201
template <typename T>
bool operator != (const value<T>& v1, const value<T>& v2) { return !v1.equal(v2); }

202
203
204
205
206

template <typename VT>
using clean_value_type = value<typename clean_type<VT>::type>;


Damien Leroux's avatar
Damien Leroux committed
207
208
209
210
211
212
213
214
namespace std {
    template <typename T>
        struct hash<value<T>> {
            size_t operator () (const value<T>& v) const
            {
                return v.hash();
            }
        };
215
216
217
218
219
220
221
222
223
224

    template <typename T>
        struct hash<collection<T>> {
            size_t operator () (const collection<T>& c) const
            {
                md5_digest h;
                h.update(c.begin(), c.end());
                return h.context;
            }
        };
Damien Leroux's avatar
Damien Leroux committed
225
226
227
228
229
}

template <typename VT>
std::ostream& operator << (std::ostream& os, const value<VT>& v)
{
230
231
232
233
234
    if (v.valid()) {
        return os << (*v);
    } else {
        return os << "<nil>";
    }
Damien Leroux's avatar
Damien Leroux committed
235
236
237
238
239
240
241
242
243
244
}

template <typename VT>
md5_digest& operator << (md5_digest& md5, const value<VT>& v)
{
    return v.md5(md5);
}


struct md5_hash_type {
245
    md5_digest md5;
Damien Leroux's avatar
Damien Leroux committed
246
247
248
249
250
251
    std::string accum;
    std::string append;
};



252
#define do_with_arg_pack(_expr) do { using _ = int[]; (void)_{0, ((_expr), void(), 0)...}; } while(0)
Damien Leroux's avatar
Damien Leroux committed
253

254
255
namespace new_redux {
    /* pattern found on http://stackoverflow.com/a/19098481 */
Damien Leroux's avatar
Damien Leroux committed
256

257
258
    template <typename... Args>
        size_t hash(Args... args)
259
        {
260
261
262
263
            size_t accum = 0;
            do_with_arg_pack(accum ^= std::hash<Args>()(args));
            /*using apply_to_pack = int[];*/
            /*(void)apply_to_pack{0, (accum ^= std::hash<Args>()(args), void(), 0)...};*/
264
265
266
            return accum;
        }

267
268
    template <typename... Args>
        md5_digest feed_md5(Args... args)
269
        {
270
271
272
            md5_digest m;
            do_with_arg_pack(m << args);
            return m;
273
274
        }

275
276
    template <typename... Args>
        std::string md5(Args... args)
Damien Leroux's avatar
Damien Leroux committed
277
        {
278
279
            /*do_with_arg_pack(std::cout << args << std::endl);*/
            return feed_md5(args...);
Damien Leroux's avatar
Damien Leroux committed
280
281
        }

282
283
    template <typename... Args>
        std::string md5_append(Args... args)
Damien Leroux's avatar
Damien Leroux committed
284
        {
285
286
287
288
            std::stringstream s;
            do_with_arg_pack(s << md5(args));
            /*std::cout << "* Long MD5 " << s.str() << std::endl;*/
            return s.str();
Damien Leroux's avatar
Damien Leroux committed
289
290
        }

291
}
Damien Leroux's avatar
Damien Leroux committed
292
293

template <typename... Elems>
294
295
296
297
298
size_t compute_hash(const Elems&... e)
{
    /*redux::hash h; return redux::reduce()(0, h, e...);*/
    return new_redux::hash(e...);
}
Damien Leroux's avatar
Damien Leroux committed
299
300
301
302

template <typename... Elems>
std::string compute_md5(const Elems&... e)
{
303
304
305
306
307
    /*md5_digest md;*/
    /*redux::md5 m;*/
    /*redux::reduce()(md, m, e...);*/
    /*return md;*/
    return new_redux::md5(e...);
Damien Leroux's avatar
Damien Leroux committed
308
309
310
}

template <typename... Elems>
311
312
313
314
315
std::string append_md5(const Elems&... e)
{
    /*std::stringstream ss; redux::md5_append ma; return redux::reduce()(ss, ma, e...).str();*/
    return new_redux::md5_append(e...);
}
Damien Leroux's avatar
Damien Leroux committed
316
317
318
319
320
321
322
323
324
325


template <typename ValueType>
    struct immediate_value : generic_value_interface<ValueType> {
        typedef ValueType value_type;

        value_type m_storage;

        immediate_value(const ValueType& v) : m_storage(v) {}
        immediate_value(ValueType&& v) : m_storage(std::forward<ValueType>(v)) {}
326
327
        template <typename... Args>
            immediate_value(Args... x) : m_storage(x...) {}
Damien Leroux's avatar
Damien Leroux committed
328
329
330
331
332
333
334
335
336
337
338
339
340
341
342
343
344
345
346
347

        virtual
            value_type& operator * ()
            override { return m_storage; }
        virtual
            value_type* operator -> ()
            override { return &m_storage; }
        virtual
            const value_type& operator * () const
            override { return m_storage; }
        virtual
            const value_type* operator -> () const
            override { return &m_storage; }

        virtual
            size_t hash() const
            override { return std::hash<ValueType>()(m_storage); }
        virtual
            md5_digest& md5(md5_digest& md) const
            override { return md << m_storage; }
348
#if 0
Damien Leroux's avatar
Damien Leroux committed
349
350
351
352
353
354
        virtual
            bool __equal__(const ValueType& v) const
            override
            {
                return m_storage == v;
            }
355
#endif
Damien Leroux's avatar
Damien Leroux committed
356
357
358
359
360
361
362
363
364
365
366
367
368
369
370
371
372
373
374
375
376
377
378
379
380
381
382
383
384
385
386
387
388
389
390
391
392
393
394
    };


template <typename ValueType>
    struct unique_value : generic_value_interface<ValueType> {
        typedef ValueType value_type;

        value_type m_storage;

        unique_value(const ValueType& v) : m_storage(v) {}
        unique_value(ValueType&& v) : m_storage(std::forward<ValueType>(v)) {}
        unique_value() : m_storage() {}

        virtual
            value_type& operator * ()
            override { return m_storage; }
        virtual
            value_type* operator -> ()
            override { return &m_storage; }
        virtual
            const value_type& operator * () const
            override { return m_storage; }
        virtual
            const value_type* operator -> () const
            override { return &m_storage; }

        virtual
            size_t hash() const
            override { return std::hash<const void*>()(this); }
        virtual
            md5_digest& md5(md5_digest& md) const
            override { return md << m_storage; }
    };

/*template <typename M, typename R, typename C>*/
    /*struct immediate_value<labelled_matrix<M, R, C>>*/
        /*: unique_value<labelled_matrix<M, R, C>> {};*/


395
396
397
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
425
426
427
428
429
430
431
432
433
434
template <typename Ret, typename... Args>
computation_registry<value<Ret>, Ret (*) (Args...), value<typename clean_type<Args>::type>...>&
__get_registry()
{
    static computation_registry<value<Ret>, Ret (*) (Args...), value<typename clean_type<Args>::type>...> _reg_;
    /*MSG_DEBUG("Registry at " << (&_reg_));*/
    return _reg_;
}


template <typename Ret, typename... Args>
computation_registry<value<Ret>, Ret (*) (Args...), value<typename clean_type<Args>::type>...>&
__get_in_progress_registry()
{
    static computation_registry<value<Ret>, Ret (*) (Args...), value<typename clean_type<Args>::type>...> _reg_;
    /*MSG_DEBUG("Registry at " << (&_reg_));*/
    return _reg_;
}


template <typename Ret, typename... Args>
std::mutex& __get_in_progress_mutex() { static std::mutex _; return _; }

template <typename Ret, typename... Args>
void unregister_task_in_progress(Ret (*f) (Args...), const clean_value_type<Args>&... x)
{
    __get_in_progress_mutex<Ret, Args...>().lock();
    __get_registry<Ret, Args...>().remove(f, x...);
    __get_in_progress_mutex<Ret, Args...>().unlock();
}


template <typename Ret, typename... Args>
value<Ret> register_task_in_progress(Ret (&f) (Args...), const clean_value_type<Args>&... x, value<Ret>& v)
{
    __get_in_progress_mutex<Ret, Args...>().lock();
    __get_registry<Ret, Args...>().get(&f, x...) = v;
    __get_in_progress_mutex<Ret, Args...>().unlock();
}

Damien Leroux's avatar
Damien Leroux committed
435
436
437
438
439
440
441
442

template <typename Ret, typename... Args>
    struct async_computation<Ret(Args...)> {
        typedef async_computation<Ret(Args...)> this_type;
        typedef Ret value_type;
        typedef Ret (*computation_function_pointer_type) (Args...);
        typedef std::packaged_task<Ret(Args...)> task_type;

443
        async_computation(CachingPolicy _Sync, computation_function_pointer_type func,
Damien Leroux's avatar
Damien Leroux committed
444
445
446
                          const value<typename clean_type<Args>::type>&... args)
            : dependencies(args...)
            , m_storage()
447
            , m_future(active_settings->enqueue(_Sync, [=] (Args... args) { chrono_trace _(get_func_name(func)); Ret ret = func(args...); unregister_task_in_progress(func, args...); return ret; }, *args...))
448
            , mutex()
Damien Leroux's avatar
Damien Leroux committed
449
450
        {}

451
        async_computation(CachingPolicy _Sync, std::function<Ret(Args...)>& func,
Damien Leroux's avatar
Damien Leroux committed
452
453
454
                          const value<typename clean_type<Args>::type>&... args)
            : dependencies(args...)
            , m_storage()
455
            , m_future(active_settings->enqueue(_Sync, func, *args...))
456
            /*, m_future(std::async(func, *args...))*/
457
            , mutex()
Damien Leroux's avatar
Damien Leroux committed
458
459
460
461
        {}

        value_type& __get_noconst()
        {
462
            std::unique_lock<std::mutex> read_guard(mutex);
Damien Leroux's avatar
Damien Leroux committed
463
464
465
466
467
468
469
470
471
472
473
474
475
476
477
            if (m_future.valid()) {
                return m_storage = m_future.get();
            }
            return m_storage;
        }

        const value_type& __get_const() const
        {
            return const_cast<this_type*>(this)->__get_noconst();
        }

    protected:
        std::tuple<value<typename clean_type<Args>::type>...> dependencies;
        value_type m_storage;
        std::future<Ret> m_future;
478
        std::mutex mutex;
Damien Leroux's avatar
Damien Leroux committed
479
480
481
482
483
484
485
    };


template <typename Ret, typename... Args>
    struct cached_computation<Ret(Args...)> {
        typedef Ret value_type;

486
        cached_computation(const std::string& name, Ret (*f) (Args...), const value<typename clean_type<Args>::type>&... args)
Damien Leroux's avatar
Damien Leroux committed
487
488
            : m_name(name)
            , m_func(f)
489
490
491
492
493
494
495
496
497
498
499
500
            /*, m_md5_hash{.accum = compute_md5(args...), .append = append_md5(args...)}*/
        {
            m_md5_hash.md5 = new_redux::feed_md5(args...);
            m_md5_hash.accum = m_md5_hash.md5;
            m_md5_hash.append = new_redux::md5_append(args...);
            /*std::cout << "NEW CACHED_COMPUTATION WITH ARGS" << std::endl;*/
            /*do_with_arg_pack(std::cout << args << std::endl);*/
            /*std::cout << "================================" << std::endl;*/
            /*std::cout << m_md5_hash.accum << std::endl;*/
            /*std::cout << m_md5_hash.append << std::endl;*/
            /*std::cout << "NEW CACHED_COMPUTATION " << m_md5_hash.accum << ' ' << m_md5_hash.append << std::endl;*/
        }
Damien Leroux's avatar
Damien Leroux committed
501
502
503
504

        std::string get_path() const
        {
            std::stringstream ss;
505
506
            ss << cache_directory() << '/' << m_name;
            ensure_directory_exists(ss.str());
507
            /*ss << cache_directory() << '/' << m_name << '_' << m_md5_hash.accum << '_' << typeid(Ret).name();*/
508
            ss << '/' << m_md5_hash.accum;
Damien Leroux's avatar
Damien Leroux committed
509
510
511
            return ss.str();
        }

512
513
514
515
516
517
518
519
520
521
522
523
524
525
526
        bool __load(const std::string& path, Ret& data)
        {
            std::ifstream ifs(path);
            cache_input ci(ifs);
            std::string check;
            /*   read string */
            ci & check;
            /*   compare to m_md5_hash.append */
            if (check == m_md5_hash.append) {
            /*   if same */
            /*     CACHE FOUND */
                    /*MSG_INFO('(' << m_name << ") Found data in cache");*/
                ci & data;
            /*     return value read from file */
                return true;
527
528
            } else {
                MSG_WARNING("Collision in cache! " << path);
529
530
531
532
            }
            return false;
        }

Damien Leroux's avatar
Damien Leroux committed
533
534
535
536
537
        Ret operator () (Args... x)
        {
            Ret data;
            std::string path = get_path();
            /* if cache file found */
538
            if (check_file(path, false, true, false)) {
Damien Leroux's avatar
Damien Leroux committed
539
540
541
542
543
544
545
546
547
                std::ifstream ifs(path);
                cache_input ci(ifs);
                std::string check;
            /*   read string */
                ci & check;
            /*   compare to m_md5_hash.append */
                if (check == m_md5_hash.append) {
            /*   if same */
            /*     CACHE FOUND */
548
                    /*MSG_INFO('(' << m_name << ") Found data in cache");*/
Damien Leroux's avatar
Damien Leroux committed
549
550
551
552
553
                    ci & data;
            /*     return value read from file */
                    return data;
                }
            /*   CACHE INVALID */
554
                /*MSG_INFO('(' << m_name << ") Cache is invalid. Computing data.");*/
Damien Leroux's avatar
Damien Leroux committed
555
556
557
            } else {
            /* else */
            /*   COMPUTE AND SAVE */
558
                /*MSG_INFO('(' << m_name << ") Computing data.");*/
Damien Leroux's avatar
Damien Leroux committed
559
560
561
562
563
            }
            /* compute value = m_func(x...) */
            data = m_func(x...);
            /* write m_md5_hash.append to file */
            std::ofstream ofs(path);
564
565
566
567
568
569
570
571
572
573
574
575
576
577
            {
                cache_output co(ofs);
                co & m_md5_hash.append;
                /* write value to file */
                co & data;
            }
            if (0) { /* check stored data */
                Ret check;
                if (!__load(path, check)) {
                    MSG_ERROR("Failed to load the data we just saved :(", "Call 911");
                } else if (!(check == data)) {
                    MSG_ERROR("The data we just saved is bogus :(", "Call 911");
                }
            }
Damien Leroux's avatar
Damien Leroux committed
578
579
580
581
            /* return value */
            return data;
        }

582
    /*protected:*/
Damien Leroux's avatar
Damien Leroux committed
583
584
585
586
587
588
589
590
591
592
        std::string m_name;
        Ret (*m_func) (Args...);
        md5_hash_type m_md5_hash;
    };

template <typename Ret, typename... Args>
    struct computed_value<Ret(Args...)> : generic_value_interface<Ret> {
        typedef Ret value_type;
        typedef std::function<Ret(Args...)> computation_type;

593
        computed_value(CachingPolicy _Sync, Ret (*func) (Args...), const value<typename clean_type<Args>::type>&... args)
Damien Leroux's avatar
Damien Leroux committed
594
            : m_hash(compute_hash(args...))
595
            , m_task(_Sync, func, args...)
Damien Leroux's avatar
Damien Leroux committed
596
597
598
599
600
601
602
603
604
605
606
607
608
609
610
611
612
613
614
615
616
617
        {}

        virtual
            value_type& operator * ()
            override { return m_task.__get_noconst(); }
        virtual
            value_type* operator -> ()
            override { return &m_task.__get_noconst(); }
        virtual
            const value_type& operator * () const
            override { return m_task.__get_const(); }
        virtual
            const value_type* operator -> () const
            override { return &m_task.__get_const(); }

        virtual
            size_t hash() const
            override { return m_hash; }
        virtual
            md5_digest& md5(md5_digest& md) const
            override { return md; }

618
    /*protected:*/
Damien Leroux's avatar
Damien Leroux committed
619
620
621
622
623
624
625
626
627
        size_t m_hash;
        async_computation<Ret(Args...)> m_task;
    };

template <typename Ret, typename... Args>
    struct cached_computed_value<Ret(Args...)> : generic_value_interface<Ret> {
        typedef Ret value_type;
        typedef std::function<Ret(Args...)> computation_type;

628
        cached_computed_value(CachingPolicy _Sync, Ret (*func) (Args...), const value<typename clean_type<Args>::type>&... args)
629
            : m_comp(get_func_name(func), func, args...)
Damien Leroux's avatar
Damien Leroux committed
630
            , m_comp_proxy([this](Args... x) { return m_comp(x...); })
631
            , m_task(_Sync, m_comp_proxy, args...)
Damien Leroux's avatar
Damien Leroux committed
632
        {}
633
634
            /*m_hash = m_comp.m_md5_hash.md5.context;*/
        /*}*/
Damien Leroux's avatar
Damien Leroux committed
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649

        virtual
            value_type& operator * ()
            override { return m_task.__get_noconst(); }
        virtual
            value_type* operator -> ()
            override { return &m_task.__get_noconst(); }
        virtual
            const value_type& operator * () const
            override { return m_task.__get_const(); }
        virtual
            const value_type* operator -> () const
            override { return &m_task.__get_const(); }

        virtual
650
651
652
653
654
655
            size_t hash() const override
            /*override { return m_hash; }*/
            {
                (void)m_task.__get_const();
                return m_comp.m_md5_hash.md5.context;
            }
Damien Leroux's avatar
Damien Leroux committed
656
        virtual
657
658
659
660
661
662
            md5_digest& md5(md5_digest& md) const override
            {
                (void)m_task.__get_const();
                return md.blend(m_comp.m_md5_hash.md5.context);
            }
            /*override { return md; }*/
Damien Leroux's avatar
Damien Leroux committed
663

664
    /*protected:*/
Damien Leroux's avatar
Damien Leroux committed
665
666
667
        cached_computation<Ret(Args...)> m_comp;
        std::function<Ret(Args...)> m_comp_proxy;
        async_computation<Ret(Args...)> m_task;
668
        /*size_t m_hash;*/
Damien Leroux's avatar
Damien Leroux committed
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
710
    };


/* ranges and collections */

/* T must have operators + and < */
template <typename T>
    struct range {
        T m_min, m_max, m_step;
        range(T min, T max, T step)
            : m_min(min), m_max(max), m_step(step)
        {}
        struct iterator {
            value<T> m_data;
            T m_step;
            T m_max;
            iterator(T value, T max, T step)
                : m_data(new immediate_value<T>(value)), m_step(step), m_max(max)
            {}
            iterator& operator ++ ()
            {
                *m_data += m_step;
                if (*m_data > m_max) {
                    *m_data = m_max;
                }
                return *this;
            }
            bool operator == (const iterator& i) const
            {
                return *m_data == *i.m_data;
            }
            bool operator != (const iterator& i) const
            {
                return *m_data != *i.m_data;
            }
            const value<T>& operator * () const { return m_data; }
        };
        iterator begin() const { return {m_min, m_max, m_step}; }
        iterator end() const { return {m_max, m_max, 0}; }
    };


711
712
713
714
715
716
717
718
719
720
721
722
723
724
725
726
727
728
729
730
731
732
733
734
735
736
737
738
739
740
741
742
743
744
745
746
747
748
749
template <typename _Coll>
    struct as_collection {
        typedef typename _Coll::value_type T;
        typedef typename _Coll::const_iterator ci_type;

        ci_type m_begin, m_end;

        as_collection(const _Coll& c)
            : m_begin(c.begin()), m_end(c.end())
        {}

        struct iterator {
            ci_type ci;
            iterator(const ci_type& it)
                : ci(it)
            {}
            iterator& operator ++ ()
            {
                ++ci;
                return *this;
            }
            bool operator == (const iterator& i) const
            {
                return ci == i.ci;
            }
            bool operator != (const iterator& i) const
            {
                return ci != i.ci;
            }
            value<T> operator * () const { return value<T>{*ci}; }
        };

        iterator begin() const { return {m_begin}; }
        iterator end() const { return {m_end}; }
    };

template <typename _Coll>
as_collection<_Coll> values_of(const _Coll& c) { return {c}; }

Damien Leroux's avatar
Damien Leroux committed
750
751
752
753
754
755
756
757
758
759
760
#if 0
template <typename ValueType, typename... AllArgs> struct registry_impl_type;
template <typename ValueType>
    struct registry_impl_type<ValueType> { typedef ValueType type; };
template <typename ValueType, typename Arg0, typename... OtherArgs>
    struct registry_impl_type<ValueType, Arg0, OtherArgs...> {
        typedef std::unordered_map<Arg0, registry_impl_type<ValueType, OtherArgs...>::type> type;
    };
#endif


761
762
763
764
765
766
767
768
769
770
771
772
773
774
775
#if 0
static inline
std::mutex& __get_lock(void* fptr)
{
    struct m_wrap { std::mutex mutex; m_wrap() : mutex() {} m_wrap(const m_wrap&) : mutex() {} };
    static std::unordered_map<void*, m_wrap> _;
    return _[fptr].mutex;
    /*auto it = _.find(fptr);*/
    /*if (it == _.end()) {*/
        /*bool discard;*/
        /*std::tie(it, discard) = _.insert({fptr, {}});*/
    /*}*/
    /*return it->second;*/
}
#endif
Damien Leroux's avatar
Damien Leroux committed
776
777
778


template <typename Ret, typename... Args>
779
780
781
struct with_disk_cache_traits {
	typedef cached_computed_value<Ret(Args...)> type;
};
Damien Leroux's avatar
Damien Leroux committed
782
783

template <typename Ret, typename... Args>
784
785
786
struct without_disk_cache_traits {
	typedef computed_value<Ret(Args...)> type;
};
Damien Leroux's avatar
Damien Leroux committed
787

788
789
template <int _Policy, typename Ret, typename... Args>
struct disk_cache_traits;
Damien Leroux's avatar
Damien Leroux committed
790

791
template <typename Ret, typename... Args>
792
struct disk_cache_traits<Oneshot, Ret, Args...> : without_disk_cache_traits<Ret, Args...> {};
793
794

template <typename Ret, typename... Args>
795
struct disk_cache_traits<Disk, Ret, Args...> : with_disk_cache_traits<Ret, Args...> {};
Damien Leroux's avatar
Damien Leroux committed
796

797
798
799
800
801
802
803
template <typename Ret, typename... Args>
struct with_mem_cache_traits {
	typedef value<Ret>& return_type;

	template <typename _Maker>
		static
		return_type
804
		create(CachingPolicy _Sync, Ret (&f) (Args...), const clean_value_type<Args>&... x)
805
		{
806
807
            static std::mutex _;
            std::unique_lock<std::mutex> lock_guard(_);
808
809
810
811
812
813
814

            value<Ret>* ret_in_progress = __get_in_progress_registry<Ret, Args...>().find(&f, x...);

            if (ret_in_progress) {
                return *ret_in_progress;
            }

815
816
817
			/*MSG_DEBUG("new value with mem cache");*/
			return_type ret = __get_registry<Ret, Args...>().get(&f, x...);
			if (!ret.valid()) {
818
				ret = new _Maker(_Sync, f, x...);
819
820
821
822
823
824
825
826
827
828
829
830
			}
			return ret;
		}
};

template <typename Ret, typename... Args>
struct without_mem_cache_traits {
	typedef value<Ret> return_type;

	template <typename _Maker>
		static
		return_type
831
		create(CachingPolicy _Sync, Ret (&f) (Args...), const clean_value_type<Args>&... x)
832
		{
833
834
835
836
837
838
            value<Ret>* ret_in_progress = __get_in_progress_registry<Ret, Args...>().find(&f, x...);

            if (ret_in_progress) {
                return *ret_in_progress;
            }

839
			/*MSG_DEBUG("new value without mem cache");*/
840
			return_type ret = new _Maker(_Sync, f, x...);
841
842
843
844
845
846
847
848
			return ret;
		}
};

template <int _Policy, typename Ret, typename... Args>
struct mem_cache_traits;

template <typename Ret, typename... Args>
849
struct mem_cache_traits<Oneshot, Ret, Args...> : without_mem_cache_traits<Ret, Args...> {};
850
851

template <typename Ret, typename... Args>
852
struct mem_cache_traits<Mem, Ret, Args...> : with_mem_cache_traits<Ret, Args...> {};
853

854
855
template <CachingPolicy _Policy = Oneshot, typename Ret, typename... Args>
typename mem_cache_traits<_Policy & Mem, Ret, Args...>::return_type
856
857
make_value(Ret (&f) (Args...), const clean_value_type<Args>&... x)
{
858
859
860
	typedef mem_cache_traits<_Policy & Mem, Ret, Args...> mem_policy;
	typedef disk_cache_traits<_Policy & Disk, Ret, Args...> disk_policy;
	return mem_policy::template create<typename disk_policy::type>(_Policy & Sync, f, x...);
861
}
Damien Leroux's avatar
Damien Leroux committed
862
863
864

template <typename T> struct collection : std::vector<value<T>> {
    using std::vector<value<T>>::vector;
865
866
867
868
869
870
871
872
873
    using std::vector<value<T>>::operator [];
    template <typename INTEGRAL_TYPE>
        value<T>&
        operator [] (const value<INTEGRAL_TYPE>& i)
        { return (*this)[*i]; }
    template <typename INTEGRAL_TYPE>
        const value<T>&
        operator [] (const value<INTEGRAL_TYPE>& i) const
        { return (*this)[*i]; }
Damien Leroux's avatar
Damien Leroux committed
874
875
};

876
877
template <typename... X> struct tuple;

878
template <CachingPolicy C, typename F, typename R, typename P> struct make_coll_impl;
Damien Leroux's avatar
Damien Leroux committed
879
880
881
882
883
884
885
886
887
888

template <typename A, typename B> struct types_must_be_identical : std::integral_constant<bool, false> {};
template <> struct types_must_be_identical<int, double> : std::integral_constant<bool, false> {};
template <> struct types_must_be_identical<double, int> : std::integral_constant<bool, false> {};
template <typename A> struct types_must_be_identical<A, A> : std::integral_constant<bool, true> {};
template <typename A> struct types_must_be_identical<const A&, A> : std::integral_constant<bool, true> {};
template <typename A> struct types_must_be_identical<A&, A> : std::integral_constant<bool, true> {};
template <typename A> struct types_must_be_identical<const A, A> : std::integral_constant<bool, true> {};

template <typename LA, typename LB> struct type_lists_must_be_identical;
889
template <> struct type_lists_must_be_identical<tuple<>, tuple<>> {};
Damien Leroux's avatar
Damien Leroux committed
890
891

template <typename A0, typename... A, typename B0, typename... B>
892
893
    struct type_lists_must_be_identical<tuple<A0, A...>, tuple<B0, B...>>
            : type_lists_must_be_identical<tuple<A...>, tuple<B...>> {
Damien Leroux's avatar
Damien Leroux committed
894
895
896
        static_assert(types_must_be_identical<A0, B0>::value, "INVALID PARAMETER TYPE.");
    };

897
template <CachingPolicy _Policy, typename Ret, typename... FuncArgs, typename... ErrArgs>
898
struct make_coll_impl<_Policy, Ret(FuncArgs...), tuple<>, tuple<ErrArgs...>> {
Damien Leroux's avatar
Damien Leroux committed
899
900
901
902
    void operator () (collection<Ret>& coll,
                      Ret (&f) (FuncArgs...),
                      ErrArgs... errargs)
    {
903
        type_lists_must_be_identical<tuple<FuncArgs...>, tuple<typename ErrArgs::value_type...>>();
904
        coll.push_back(make_value<_Policy>(f, errargs...));
Damien Leroux's avatar
Damien Leroux committed
905
906
907
    }
};

908
template <CachingPolicy _Policy, typename Ret, typename... FuncArgs>
909
struct make_coll_impl<_Policy, Ret(FuncArgs...), tuple<>, tuple<clean_value_type<FuncArgs>...>> {
Damien Leroux's avatar
Damien Leroux committed
910
911
912
913
    void operator () (collection<Ret>& coll,
                      Ret (&f) (FuncArgs...),
                      const value<typename clean_type<FuncArgs>::type>&... pargs)
    {
914
        coll.push_back(make_value<_Policy>(f, pargs...));
Damien Leroux's avatar
Damien Leroux committed
915
916
917
918
    }
};


919
template <CachingPolicy _Policy, typename Ret, typename... FuncArgs, typename VT, typename... ArgsRemaining, typename... PreviousArgs>
920
struct make_coll_impl<_Policy, Ret(FuncArgs...), tuple<value<VT>, ArgsRemaining...>, tuple<PreviousArgs...>> {
Damien Leroux's avatar
Damien Leroux committed
921
922
923
924
925
926
927
    void operator () (collection<Ret>& coll,
                      Ret (&f) (FuncArgs...),
                      const value<VT>& car,
                      const ArgsRemaining&... cdr,
                      const PreviousArgs&... pargs)
    {
        make_coll_impl<
928
			_Policy,
Damien Leroux's avatar
Damien Leroux committed
929
            Ret(FuncArgs...),
930
931
            tuple<ArgsRemaining...>,
            tuple<PreviousArgs..., value<VT>>>() (coll, f, cdr..., pargs..., car);
Damien Leroux's avatar
Damien Leroux committed
932
933
934
    }
};

935
template <CachingPolicy _Policy, typename Ret, typename... FuncArgs, typename T, typename... ArgsRemaining, typename... PreviousArgs>
936
struct make_coll_impl<_Policy, Ret(FuncArgs...), tuple<range<T>, ArgsRemaining...>, tuple<PreviousArgs...>> {
Damien Leroux's avatar
Damien Leroux committed
937
938
939
940
941
942
943
944
945
    void operator () (collection<Ret>& coll,
                      Ret (&f) (FuncArgs...),
                      const range<T>& car,
                      const ArgsRemaining&... cdr,
                      const PreviousArgs&... pargs)
    {
        typedef value<T> vtype;
        for (const vtype& v: car) {
            make_coll_impl<
946
				_Policy,
Damien Leroux's avatar
Damien Leroux committed
947
                Ret(FuncArgs...),
948
949
                tuple<ArgsRemaining...>,
                tuple<PreviousArgs..., vtype>>() (coll, f, cdr..., pargs..., v);
Damien Leroux's avatar
Damien Leroux committed
950
951
952
953
        }
    }
};

954
template <CachingPolicy _Policy, typename Ret, typename... FuncArgs, typename T, typename... ArgsRemaining, typename... PreviousArgs>
955
struct make_coll_impl<_Policy, Ret(FuncArgs...), tuple<collection<T>, ArgsRemaining...>, tuple<PreviousArgs...>> {
Damien Leroux's avatar
Damien Leroux committed
956
957
958
959
960
961
962
963
964
    void operator () (collection<Ret>& coll,
                      Ret (&f) (FuncArgs...),
                      const collection<T>& car,
                      const ArgsRemaining&... cdr,
                      const PreviousArgs&... pargs)
    {
        typedef value<T> vtype;
        for (const vtype& v: car) {
            make_coll_impl<
965
				_Policy,
Damien Leroux's avatar
Damien Leroux committed
966
                Ret(FuncArgs...),
967
968
                tuple<ArgsRemaining...>,
                tuple<PreviousArgs..., vtype>>() (coll, f, cdr..., pargs..., v);
Damien Leroux's avatar
Damien Leroux committed
969
970
971
972
        }
    }
};

973
974
975
976
977
978
979
980
981
982
983
984
985
986
987
988
989
990
991
template <CachingPolicy _Policy, typename Ret, typename... FuncArgs, typename T, typename... ArgsRemaining, typename... PreviousArgs>
struct make_coll_impl<_Policy, Ret(FuncArgs...), tuple<as_collection<T>, ArgsRemaining...>, tuple<PreviousArgs...>> {
    void operator () (collection<Ret>& coll,
                      Ret (&f) (FuncArgs...),
                      const as_collection<T>& car,
                      const ArgsRemaining&... cdr,
                      const PreviousArgs&... pargs)
    {
        typedef value<typename T::value_type> vtype;
        for (const vtype& v: car) {
            make_coll_impl<
				_Policy,
                Ret(FuncArgs...),
                tuple<ArgsRemaining...>,
                tuple<PreviousArgs..., vtype>>() (coll, f, cdr..., pargs..., v);
        }
    }
};

Damien Leroux's avatar
Damien Leroux committed
992
993


994
template <CachingPolicy _Policy = Oneshot, typename Ret, typename... Args, typename... CollArgs>
Damien Leroux's avatar
Damien Leroux committed
995
996
997
998
collection<Ret>
make_collection(Ret (&f) (Args...), CollArgs... args)
{
    collection<Ret> ret;
999
    make_coll_impl<_Policy, Ret(Args...), tuple<CollArgs...>, tuple<>>() (ret, f, args...);
Damien Leroux's avatar
Damien Leroux committed
1000
    return ret;