1/*
2 * Copyright © 2022 Imagination Technologies Ltd.
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a copy
5 * of this software and associated documentation files (the "Software"), to deal
6 * in the Software without restriction, including without limitation the rights
7 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
8 * copies of the Software, and to permit persons to whom the Software is
9 * furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice (including the next
12 * paragraph) shall be included in all copies or substantial portions of the
13 * Software.
14 *
15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
18 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
19 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
20 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
21 * SOFTWARE.
22 */
23
24#ifndef ROGUE_REGALLOC_H
25#define ROGUE_REGALLOC_H
26
27#include <stdbool.h>
28#include <stddef.h>
29
30#include "util/list.h"
31
32/**
33 * \brief Register classes used for allocation.
34 */
35enum rogue_reg_class {
36   ROGUE_REG_CLASS_TEMP,
37   ROGUE_REG_CLASS_VEC4,
38
39   ROGUE_REG_CLASS_COUNT,
40};
41
42/**
43 * \brief Register data for each class.
44 */
45struct rogue_reg_data {
46   enum rogue_operand_type type;
47   size_t count;
48   size_t stride;
49
50   size_t offset;
51   struct ra_class *class;
52   size_t num_used;
53};
54
55/**
56 * \brief Register allocation context.
57 */
58struct rogue_ra {
59   struct ra_regs *regs;
60
61   struct rogue_reg_data reg_data[ROGUE_REG_CLASS_COUNT];
62};
63
64struct rogue_ra *rogue_ra_init(void *mem_ctx);
65bool rogue_ra_alloc(struct list_head *instr_list,
66                    struct rogue_ra *ra,
67                    size_t *temps_used,
68                    size_t *internals_used);
69
70#endif /* ROGUE_REGALLOC_H */
71