program.hpp 1.46 KB
Newer Older
Paul's avatar
Paul committed
1
#ifndef RTG_GUARD_RTGLIB_PROGRAM_HPP
Paul's avatar
Paul committed
2
#define RTG_GUARD_RTGLIB_PROGRAM_HPP
Paul's avatar
Paul committed
3

Paul's avatar
Paul committed
4
#include <list>
Paul's avatar
Paul committed
5
#include <unordered_map>
Paul's avatar
Paul committed
6
#include <rtg/operation.hpp>
Paul's avatar
Paul committed
7
#include <rtg/literal.hpp>
Paul's avatar
Paul committed
8
#include <rtg/builtin.hpp>
Paul's avatar
Paul committed
9
#include <rtg/instruction_ref.hpp>
Paul's avatar
Paul committed
10
#include <algorithm>
Paul's avatar
Paul committed
11
12
13

namespace rtg {

Paul's avatar
Paul committed
14
15
struct program_impl;

Paul's avatar
Paul committed
16
17
struct program
{
Paul's avatar
Paul committed
18
19
    program();
    program(program&&) noexcept;
Paul's avatar
Paul committed
20
21
    program& operator=(program&&) noexcept;
    ~program() noexcept;
Paul's avatar
Paul committed
22

Paul's avatar
Paul committed
23
    template <class... Ts>
Paul's avatar
Paul committed
24
    instruction_ref add_instruction(operation op, Ts... args)
Paul's avatar
Paul committed
25
    {
Paul's avatar
Paul committed
26
        return add_instruction(op, {args...});
Paul's avatar
Paul committed
27
    }
Paul's avatar
Paul committed
28
    instruction_ref add_instruction(operation op, std::vector<instruction_ref> args);
Paul's avatar
Paul committed
29
30
31
32
33
34
35
36
    
    template <class... Ts>
    instruction_ref insert_instruction(instruction_ref ins, operation op, Ts... args)
    {
        return insert_instruction(ins, op, {args...});
    }
    instruction_ref insert_instruction(instruction_ref ins, operation op, std::vector<instruction_ref> args);

Paul's avatar
Paul committed
37
    template <class... Ts>
Paul's avatar
Paul committed
38
    instruction_ref add_literal(Ts&&... xs)
Paul's avatar
Paul committed
39
    {
Paul's avatar
Paul committed
40
        return add_literal(literal{std::forward<Ts>(xs)...});
Paul's avatar
Paul committed
41
42
    }

Paul's avatar
Paul committed
43
    instruction_ref add_literal(literal l);
Paul's avatar
Paul committed
44

Paul's avatar
Paul committed
45
    instruction_ref add_parameter(std::string name, shape s);
Paul's avatar
Paul committed
46
47

    literal eval(std::unordered_map<std::string, argument> params) const;
Paul's avatar
Paul committed
48

Paul's avatar
Paul committed
49
50
51
    // TODO: Change to stream operator
    void print() const;

Paul's avatar
Paul committed
52
    bool has_instruction(instruction_ref ins) const;
Paul's avatar
Paul committed
53

Paul's avatar
Paul committed
54
    private:
Paul's avatar
Paul committed
55
    std::unique_ptr<program_impl> impl;
Paul's avatar
Paul committed
56
57
};

Paul's avatar
Paul committed
58
} // namespace rtg
Paul's avatar
Paul committed
59
60

#endif