Skip to main content

asmkit/riscv/
operands.rs

1//! RISC-V Operands definition.
2use crate::{
3    core::{
4        arch_traits::{Arch, ArchTraits},
5        operand::*,
6        types::TypeId,
7    },
8    define_abstract_reg, define_final_reg, define_operand_cast, define_reg_traits,
9};
10
11macro_rules! impl_deref_for_wrapper {
12    ($wrapper:ty, $target:ty) => {
13        impl core::ops::Deref for $wrapper {
14            type Target = $target;
15
16            fn deref(&self) -> &Self::Target {
17                &self.0
18            }
19        }
20
21        impl core::ops::DerefMut for $wrapper {
22            fn deref_mut(&mut self) -> &mut Self::Target {
23                &mut self.0
24            }
25        }
26    };
27}
28
29#[derive(Clone, Copy, PartialEq, PartialOrd, Eq, Ord, Hash)]
30pub struct Reg(pub BaseReg);
31
32impl_deref_for_wrapper!(Reg, BaseReg);
33
34define_abstract_reg!(Reg, BaseReg);
35
36define_reg_traits!(RISCV32Gp, RegGroup::Gp, 32, TypeId::Int32);
37define_reg_traits!(RISCV64Gp, RegGroup::Gp, 64, TypeId::Int64);
38define_reg_traits!(RISCVPC, RegGroup::PC, 64, TypeId::Int64);
39define_reg_traits!(RISCVVec, RegGroup::Vec, 128, TypeId::Int32x4);
40define_reg_traits!(RISCVFp, RegGroup::Vec, 64, TypeId::Float64);
41
42impl Reg {
43    pub const fn signature_of(typ: RegType) -> OperandSignature {
44        ArchTraits::by_arch(Arch::RISCV64).reg_type_to_signature(typ)
45    }
46
47    pub fn is_gp(&self) -> bool {
48        self.is_reg_group_of(RegGroup::Gp)
49    }
50
51    pub fn is_fp(&self) -> bool {
52        self.has_base_signature(RISCVFp::SIGNATURE)
53    }
54
55    pub fn is_vec(&self) -> bool {
56        self.has_base_signature(RISCVFp::SIGNATURE)
57    }
58
59    pub fn set_reg_t<T: RegTraits>(&mut self, rid: u32) {
60        self.set_signature(T::SIGNATURE.into());
61        self.set_id(rid);
62    }
63
64    pub fn set_type_and_id(&mut self, typ: RegType, id: u32) {
65        self.set_signature(Self::signature_of(typ));
66        self.set_id(id);
67    }
68    pub const SIGNATURE: u32 = BaseReg::SIGNATURE;
69}
70
71#[derive(Clone, Copy, PartialEq, PartialOrd, Eq, Ord, Hash)]
72pub struct Gp(pub Reg);
73
74impl_deref_for_wrapper!(Gp, Reg);
75
76impl Gp {
77    pub const fn signature_of(typ: RegType) -> OperandSignature {
78        ArchTraits::by_arch(Arch::RISCV64).reg_type_to_signature(typ)
79    }
80
81    pub const X0: u32 = 0;
82    pub const X1: u32 = 1;
83    pub const X2: u32 = 2;
84    pub const X3: u32 = 3;
85    pub const X4: u32 = 4;
86    pub const X5: u32 = 5;
87    pub const X6: u32 = 6;
88    pub const X7: u32 = 7;
89    pub const X8: u32 = 8;
90    pub const X9: u32 = 9;
91    pub const X10: u32 = 10;
92    pub const X11: u32 = 11;
93    pub const X12: u32 = 12;
94    pub const X13: u32 = 13;
95    pub const X14: u32 = 14;
96    pub const X15: u32 = 15;
97    pub const X16: u32 = 16;
98    pub const X17: u32 = 17;
99    pub const X18: u32 = 18;
100    pub const X19: u32 = 19;
101    pub const X20: u32 = 20;
102    pub const X21: u32 = 21;
103    pub const X22: u32 = 22;
104    pub const X23: u32 = 23;
105    pub const X24: u32 = 24;
106    pub const X25: u32 = 25;
107    pub const X26: u32 = 26;
108    pub const X27: u32 = 27;
109    pub const X28: u32 = 28;
110    pub const X29: u32 = 29;
111    pub const X30: u32 = 30;
112    pub const X31: u32 = 31;
113}
114
115define_final_reg!(Gp, Reg, RISCV64Gp);
116
117#[derive(Clone, Copy, PartialEq, PartialOrd, Eq, Ord, Hash)]
118pub struct Fp(pub Reg);
119
120impl_deref_for_wrapper!(Fp, Reg);
121
122define_final_reg!(Fp, Reg, RISCVFp);
123
124impl Fp {
125    pub const fn signature_of(typ: RegType) -> OperandSignature {
126        ArchTraits::by_arch(Arch::RISCV64).reg_type_to_signature(typ)
127    }
128
129    pub const F0: u32 = 0;
130    pub const F1: u32 = 1;
131    pub const F2: u32 = 2;
132    pub const F3: u32 = 3;
133    pub const F4: u32 = 4;
134    pub const F5: u32 = 5;
135    pub const F6: u32 = 6;
136    pub const F7: u32 = 7;
137    pub const F8: u32 = 8;
138    pub const F9: u32 = 9;
139    pub const F10: u32 = 10;
140    pub const F11: u32 = 11;
141    pub const F12: u32 = 12;
142    pub const F13: u32 = 13;
143    pub const F14: u32 = 14;
144    pub const F15: u32 = 15;
145    pub const F16: u32 = 16;
146    pub const F17: u32 = 17;
147    pub const F18: u32 = 18;
148    pub const F19: u32 = 19;
149    pub const F20: u32 = 20;
150    pub const F21: u32 = 21;
151    pub const F22: u32 = 22;
152    pub const F23: u32 = 23;
153    pub const F24: u32 = 24;
154    pub const F25: u32 = 25;
155    pub const F26: u32 = 26;
156    pub const F27: u32 = 27;
157    pub const F28: u32 = 28;
158    pub const F29: u32 = 29;
159    pub const F30: u32 = 30;
160    pub const F31: u32 = 31;
161}
162
163#[derive(Clone, Copy, PartialEq, PartialOrd, Eq, Ord, Hash)]
164pub struct Vp(pub Reg);
165
166impl_deref_for_wrapper!(Vp, Reg);
167
168define_final_reg!(Vp, Reg, RISCVVec);
169
170impl Vp {
171    pub const fn signature_of(typ: RegType) -> OperandSignature {
172        ArchTraits::by_arch(Arch::RISCV64).reg_type_to_signature(typ)
173    }
174
175    pub const V0: u32 = 0;
176    pub const V1: u32 = 1;
177    pub const V2: u32 = 2;
178    pub const V3: u32 = 3;
179    pub const V4: u32 = 4;
180    pub const V5: u32 = 5;
181    pub const V6: u32 = 6;
182    pub const V7: u32 = 7;
183    pub const V8: u32 = 8;
184    pub const V9: u32 = 9;
185    pub const V10: u32 = 10;
186    pub const V11: u32 = 11;
187    pub const V12: u32 = 12;
188    pub const V13: u32 = 13;
189    pub const V14: u32 = 14;
190    pub const V15: u32 = 15;
191    pub const V16: u32 = 16;
192    pub const V17: u32 = 17;
193    pub const V18: u32 = 18;
194    pub const V19: u32 = 19;
195    pub const V20: u32 = 20;
196    pub const V21: u32 = 21;
197    pub const V22: u32 = 22;
198    pub const V23: u32 = 23;
199    pub const V24: u32 = 24;
200    pub const V25: u32 = 25;
201    pub const V26: u32 = 26;
202    pub const V27: u32 = 27;
203    pub const V28: u32 = 28;
204    pub const V29: u32 = 29;
205    pub const V30: u32 = 30;
206    pub const V31: u32 = 31;
207}
208
209pub mod regs {
210    pub use super::*;
211
212    pub const fn x(id: u32) -> Gp {
213        Gp::from_id(id)
214    }
215
216    pub const fn f(id: u32) -> Fp {
217        Fp::from_id(id)
218    }
219
220    pub const fn v(id: u32) -> Vp {
221        Vp::from_id(id)
222    }
223
224    pub const X0: Gp = x(0);
225    pub const X1: Gp = x(1);
226    pub const X2: Gp = x(2);
227    pub const X3: Gp = x(3);
228    pub const X4: Gp = x(4);
229    pub const X5: Gp = x(5);
230    pub const X6: Gp = x(6);
231    pub const X7: Gp = x(7);
232    pub const X8: Gp = x(8);
233    pub const X9: Gp = x(9);
234    pub const X10: Gp = x(10);
235    pub const X11: Gp = x(11);
236    pub const X12: Gp = x(12);
237    pub const X13: Gp = x(13);
238    pub const X14: Gp = x(14);
239    pub const X15: Gp = x(15);
240    pub const X16: Gp = x(16);
241    pub const X17: Gp = x(17);
242    pub const X18: Gp = x(18);
243    pub const X19: Gp = x(19);
244    pub const X20: Gp = x(20);
245    pub const X21: Gp = x(21);
246    pub const X22: Gp = x(22);
247    pub const X23: Gp = x(23);
248    pub const X24: Gp = x(24);
249    pub const X25: Gp = x(25);
250    pub const X26: Gp = x(26);
251    pub const X27: Gp = x(27);
252    pub const X28: Gp = x(28);
253    pub const X29: Gp = x(29);
254    pub const X30: Gp = x(30);
255    pub const X31: Gp = x(31);
256
257    pub const ZERO: Gp = X0;
258    pub const RA: Gp = X1;
259    pub const SP: Gp = X2;
260    pub const GP: Gp = X3;
261    pub const TP: Gp = X4;
262    pub const T0: Gp = X5;
263    pub const T1: Gp = X6;
264    pub const T2: Gp = X7;
265    pub const S0: Gp = X8;
266    pub const S1: Gp = X9;
267    pub const A0: Gp = X10;
268    pub const A1: Gp = X11;
269    pub const A2: Gp = X12;
270    pub const A3: Gp = X13;
271    pub const A4: Gp = X14;
272    pub const A5: Gp = X15;
273    pub const A6: Gp = X16;
274    pub const A7: Gp = X17;
275    pub const S2: Gp = X18;
276    pub const S3: Gp = X19;
277    pub const S4: Gp = X20;
278    pub const S5: Gp = X21;
279    pub const S6: Gp = X22;
280    pub const S7: Gp = X23;
281    pub const S8: Gp = X24;
282    pub const S9: Gp = X25;
283    pub const S10: Gp = X26;
284    pub const S11: Gp = X27;
285    pub const T3: Gp = X28;
286    pub const T4: Gp = X29;
287    pub const T5: Gp = X30;
288    pub const T6: Gp = X31;
289
290    pub const F0: Fp = f(0);
291    pub const F1: Fp = f(1);
292    pub const F2: Fp = f(2);
293    pub const F3: Fp = f(3);
294    pub const F4: Fp = f(4);
295    pub const F5: Fp = f(5);
296    pub const F6: Fp = f(6);
297    pub const F7: Fp = f(7);
298    pub const F8: Fp = f(8);
299    pub const F9: Fp = f(9);
300    pub const F10: Fp = f(10);
301    pub const F11: Fp = f(11);
302    pub const F12: Fp = f(12);
303    pub const F13: Fp = f(13);
304    pub const F14: Fp = f(14);
305    pub const F15: Fp = f(15);
306    pub const F16: Fp = f(16);
307    pub const F17: Fp = f(17);
308    pub const F18: Fp = f(18);
309    pub const F19: Fp = f(19);
310    pub const F20: Fp = f(20);
311    pub const F21: Fp = f(21);
312    pub const F22: Fp = f(22);
313    pub const F23: Fp = f(23);
314    pub const F24: Fp = f(24);
315    pub const F25: Fp = f(25);
316    pub const F26: Fp = f(26);
317    pub const F27: Fp = f(27);
318    pub const F28: Fp = f(28);
319    pub const F29: Fp = f(29);
320    pub const F30: Fp = f(30);
321    pub const F31: Fp = f(31);
322
323    pub const V0: Vp = v(0);
324    pub const V1: Vp = v(1);
325    pub const V2: Vp = v(2);
326    pub const V3: Vp = v(3);
327    pub const V4: Vp = v(4);
328    pub const V5: Vp = v(5);
329    pub const V6: Vp = v(6);
330    pub const V7: Vp = v(7);
331    pub const V8: Vp = v(8);
332    pub const V9: Vp = v(9);
333    pub const V10: Vp = v(10);
334    pub const V11: Vp = v(11);
335    pub const V12: Vp = v(12);
336    pub const V13: Vp = v(13);
337    pub const V14: Vp = v(14);
338    pub const V15: Vp = v(15);
339    pub const V16: Vp = v(16);
340    pub const V17: Vp = v(17);
341    pub const V18: Vp = v(18);
342    pub const V19: Vp = v(19);
343    pub const V20: Vp = v(20);
344    pub const V21: Vp = v(21);
345    pub const V22: Vp = v(22);
346    pub const V23: Vp = v(23);
347    pub const V24: Vp = v(24);
348    pub const V25: Vp = v(25);
349    pub const V26: Vp = v(26);
350    pub const V27: Vp = v(27);
351    pub const V28: Vp = v(28);
352    pub const V29: Vp = v(29);
353    pub const V30: Vp = v(30);
354    pub const V31: Vp = v(31);
355}