llvm-native-core 0.1.16

LLVM-native core semantic engine — IR, CodeGen, X86 MC, Clang frontend pipeline
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
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
89
90
91
92
93
94
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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
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
343
344
345
346
347
348
349
350
351
352
353
354
355
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
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
435
436
437
438
439
440
441
442
443
444
445
446
447
448
449
450
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
516
517
518
519
520
521
522
523
524
525
526
527
528
529
530
531
532
533
534
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
571
572
573
574
575
576
577
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
621
622
623
624
625
626
627
628
629
630
631
632
633
634
635
636
637
638
639
640
641
642
643
644
645
646
647
648
649
650
651
652
653
654
655
656
657
658
659
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
681
682
683
684
685
686
687
688
689
690
//! MIPS Register Information — complete register definitions for
//! MIPS32 and MIPS64 integer and floating-point register files per the
//! MIPS Architecture for Programmers and the MIPS ABI Supplement.
//!
//! Register categories:
//! - 32 General Purpose Registers (GPRs): $0/$zero through $31/$ra
//! - 32 Floating-Point Registers (FPRs): $f0 through $f31
//!   (32 single-precision or 16/32 double-precision depending on FR mode)
//! - Special registers: HI, LO (multiply/divide results), PC
//!
//! ABI names per the MIPS O32/N32/N64 conventions:
//!   zero | at | v0-v1 | a0-a3 | t0-t7 | t8-t9 | s0-s7 | k0-k1
//!   gp | sp | s8/fp | ra
//!
//! FP ABI names:
//!   f0-f1 (return) | f2-f3 | f4-f11 (temp) | f12-f13 (fp args)
//!   f14-f19 (temp) | f20-f31 (saved)
//!
//! Clean-room reconstruction from the MIPS Architecture for Programmers
//! and the MIPS ELF ABI supplements. Zero LLVM source code consultation.

// ============================================================================
// Register Identifiers — flat numbering scheme starting at 4000
// ============================================================================

/// All MIPS physical register IDs are `u16` constants.
/// Integer registers use IDs 4000–4031.
/// Special registers (HI, LO, PC) use IDs 4032–4034.
/// Floating-point registers use IDs 4050–4081.

// ============================================================================
// MIPS32/MIPS64 General Purpose Registers (4000–4031)
// ============================================================================

pub const ZERO: u16 = 4000;
pub const AT: u16 = 4001;
pub const V0: u16 = 4002;
pub const V1: u16 = 4003;
pub const A0: u16 = 4004;
pub const A1: u16 = 4005;
pub const A2: u16 = 4006;
pub const A3: u16 = 4007;
pub const T0: u16 = 4008;
pub const T1: u16 = 4009;
pub const T2: u16 = 4010;
pub const T3: u16 = 4011;
pub const T4: u16 = 4012;
pub const T5: u16 = 4013;
pub const T6: u16 = 4014;
pub const T7: u16 = 4015;
pub const S0: u16 = 4016;
pub const S1: u16 = 4017;
pub const S2: u16 = 4018;
pub const S3: u16 = 4019;
pub const S4: u16 = 4020;
pub const S5: u16 = 4021;
pub const S6: u16 = 4022;
pub const S7: u16 = 4023;
pub const T8: u16 = 4024;
pub const T9: u16 = 4025;
pub const K0: u16 = 4026;
pub const K1: u16 = 4027;
pub const GP: u16 = 4028;
pub const SP: u16 = 4029;
pub const FP: u16 = 4030;
pub const RA: u16 = 4031;

// ============================================================================
// MIPS Special Registers (4032–4034)
// ============================================================================

/// HI special register — holds upper 32/64 bits of multiply result
/// or remainder of divide.
pub const HI: u16 = 4032;
/// LO special register — holds lower 32/64 bits of multiply result
/// or quotient of divide.
pub const LO: u16 = 4033;
/// Program Counter (used as source/dest for JAL, JR, etc.).
pub const PC: u16 = 4034;

// ============================================================================
// MIPS Floating-Point Registers (4050–4081)
// ============================================================================

pub const F0: u16 = 4050;
pub const F1: u16 = 4051;
pub const F2: u16 = 4052;
pub const F3: u16 = 4053;
pub const F4: u16 = 4054;
pub const F5: u16 = 4055;
pub const F6: u16 = 4056;
pub const F7: u16 = 4057;
pub const F8: u16 = 4058;
pub const F9: u16 = 4059;
pub const F10: u16 = 4060;
pub const F11: u16 = 4061;
pub const F12: u16 = 4062;
pub const F13: u16 = 4063;
pub const F14: u16 = 4064;
pub const F15: u16 = 4065;
pub const F16: u16 = 4066;
pub const F17: u16 = 4067;
pub const F18: u16 = 4068;
pub const F19: u16 = 4069;
pub const F20: u16 = 4070;
pub const F21: u16 = 4071;
pub const F22: u16 = 4072;
pub const F23: u16 = 4073;
pub const F24: u16 = 4074;
pub const F25: u16 = 4075;
pub const F26: u16 = 4076;
pub const F27: u16 = 4077;
pub const F28: u16 = 4078;
pub const F29: u16 = 4079;
pub const F30: u16 = 4080;
pub const F31: u16 = 4081;

// ============================================================================
// Register counts
// ============================================================================

/// Number of integer (GPR) registers in the MIPS register file.
pub const MIPS_GPR_COUNT: usize = 32;

/// Number of floating-point registers in the MIPS register file.
pub const MIPS_FPR_COUNT: usize = 32;

/// Maximum register ID used in this backend.
pub const MIPS_MAX_REG_ID: u16 = 4081;

/// Base ID for integer registers.
pub const MIPS_GPR_BASE: u16 = 4000;

/// Base ID for special registers.
pub const MIPS_SPECIAL_BASE: u16 = 4032;

/// Base ID for floating-point registers.
pub const MIPS_FPR_BASE: u16 = 4050;

// ============================================================================
// Register Class Enum
// ============================================================================

/// MIPS register class.
#[derive(Debug, Clone, Copy, PartialEq, Eq, Hash)]
pub enum MipsRegClass {
    /// General Purpose Register (integer, 32-bit for MIPS32, 64-bit for MIPS64).
    GPR,
    /// Floating-Point Register (32-bit, single precision).
    FPR32,
    /// Floating-Point Register (64-bit, double precision).
    FPR64,
    /// Special register (HI, LO, PC).
    Special,
}

impl std::fmt::Display for MipsRegClass {
    fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
        match self {
            MipsRegClass::GPR => write!(f, "GPR"),
            MipsRegClass::FPR32 => write!(f, "FPR32"),
            MipsRegClass::FPR64 => write!(f, "FPR64"),
            MipsRegClass::Special => write!(f, "Special"),
        }
    }
}

// ============================================================================
// MipsRegisterInfo — register metadata queries
// ============================================================================

/// Provides register information queries for MIPS targets.
pub struct MipsRegisterInfo;

impl MipsRegisterInfo {
    /// Get the canonical assembly name for a register ID.
    pub fn get_asm_name(reg_id: u16) -> String {
        match reg_id {
            ZERO => "$zero".into(),
            AT => "$at".into(),
            V0 => "$v0".into(),
            V1 => "$v1".into(),
            A0 => "$a0".into(),
            A1 => "$a1".into(),
            A2 => "$a2".into(),
            A3 => "$a3".into(),
            T0 => "$t0".into(),
            T1 => "$t1".into(),
            T2 => "$t2".into(),
            T3 => "$t3".into(),
            T4 => "$t4".into(),
            T5 => "$t5".into(),
            T6 => "$t6".into(),
            T7 => "$t7".into(),
            S0 => "$s0".into(),
            S1 => "$s1".into(),
            S2 => "$s2".into(),
            S3 => "$s3".into(),
            S4 => "$s4".into(),
            S5 => "$s5".into(),
            S6 => "$s6".into(),
            S7 => "$s7".into(),
            T8 => "$t8".into(),
            T9 => "$t9".into(),
            K0 => "$k0".into(),
            K1 => "$k1".into(),
            GP => "$gp".into(),
            SP => "$sp".into(),
            FP => "$fp".into(),
            RA => "$ra".into(),
            HI => "$hi".into(),
            LO => "$lo".into(),
            PC => "$pc".into(),
            _ if reg_id >= F0 && reg_id <= F31 => {
                format!("$f{}", reg_id - F0)
            }
            _ => format!("${}", reg_id),
        }
    }

    /// Get the ABI name for a register ID.
    pub fn get_abi_name(reg_id: u16) -> String {
        Self::get_asm_name(reg_id)
    }

    /// Get the register class for a register ID.
    pub fn get_reg_class(reg_id: u16) -> MipsRegClass {
        if reg_id >= MIPS_GPR_BASE && reg_id < MIPS_GPR_BASE + 32 {
            MipsRegClass::GPR
        } else if reg_id >= HI && reg_id <= PC {
            MipsRegClass::Special
        } else if reg_id >= MIPS_FPR_BASE && reg_id < MIPS_FPR_BASE + 32 {
            MipsRegClass::FPR32
        } else {
            MipsRegClass::GPR
        }
    }

    /// Get the register width in bits.
    pub fn get_reg_width(reg_id: u16, is_64bit: bool) -> u32 {
        match Self::get_reg_class(reg_id) {
            MipsRegClass::GPR => {
                if is_64bit {
                    64
                } else {
                    32
                }
            }
            MipsRegClass::FPR32 => 32,
            MipsRegClass::FPR64 => 64,
            MipsRegClass::Special => {
                if is_64bit {
                    64
                } else {
                    32
                }
            }
        }
    }

    /// Get the DWARF register number for a register ID.
    pub fn get_dwarf_num(reg_id: u16) -> i32 {
        match reg_id {
            // GPRs map directly: $0 → dwarf 0, $1 → 1, ..., $31 → 31
            _ if reg_id >= ZERO && reg_id <= RA => (reg_id - ZERO) as i32,
            // FPRs map to 32..63
            _ if reg_id >= F0 && reg_id <= F31 => 32 + (reg_id - F0) as i32,
            // HI/LO are not standard DWARF, return -1
            HI => -1,
            LO => -1,
            PC => -1,
            _ => -1,
        }
    }

    /// Check if a register is callee-saved per the MIPS O32 ABI.
    pub fn is_callee_saved(reg_id: u16) -> bool {
        matches!(
            reg_id,
            S0 | S1 | S2 | S3 | S4 | S5 | S6 | S7 | FP | GP | SP | RA
        ) || (reg_id >= F20 && reg_id <= F31)
    }

    /// Check if a register is caller-saved per the MIPS O32 ABI.
    pub fn is_caller_saved(reg_id: u16) -> bool {
        matches!(
            reg_id,
            AT | V0
                | V1
                | A0
                | A1
                | A2
                | A3
                | T0
                | T1
                | T2
                | T3
                | T4
                | T5
                | T6
                | T7
                | T8
                | T9
                | K0
                | K1
        ) || (reg_id >= F0 && reg_id <= F19)
    }

    /// Check if a register is reserved (not available for allocation).
    pub fn is_reserved(reg_id: u16) -> bool {
        reg_id == ZERO || reg_id == AT || reg_id == K0 || reg_id == K1
    }

    /// Get the allocatable GPRs (excluding reserved registers).
    pub fn get_allocatable_gprs() -> Vec<u16> {
        let mut regs = Vec::new();
        for i in 1..32 {
            let r = MIPS_GPR_BASE + i as u16;
            if !Self::is_reserved(r) {
                regs.push(r);
            }
        }
        regs
    }

    /// Get the allocatable FPRs.
    pub fn get_allocatable_fprs() -> Vec<u16> {
        (F0..=F31).collect()
    }

    /// Get the argument registers (a0-a3).
    pub fn get_argument_regs() -> Vec<u16> {
        vec![A0, A1, A2, A3]
    }

    /// Get the FP argument registers (f12-f13 for O32, f12-f19 for N32/N64).
    pub fn get_fp_argument_regs(_is_64bit: bool) -> Vec<u16> {
        vec![F12, F13, F14, F15, F16, F17, F18, F19]
    }

    /// Get the return registers (v0-v1).
    pub fn get_return_regs() -> Vec<u16> {
        vec![V0, V1]
    }

    /// Get the FP return registers (f0-f1).
    pub fn get_fp_return_regs() -> Vec<u16> {
        vec![F0, F1]
    }

    /// Get the frame pointer register.
    pub fn get_frame_pointer_reg() -> u16 {
        FP
    }

    /// Get the return address register.
    pub fn get_return_address_reg() -> u16 {
        RA
    }

    /// Get the stack pointer register.
    pub fn get_stack_pointer_reg() -> u16 {
        SP
    }

    /// Get the global pointer register.
    pub fn get_global_pointer_reg() -> u16 {
        GP
    }

    /// Get the zero register.
    pub fn get_zero_reg() -> u16 {
        ZERO
    }

    /// Check if a register ID is a GPR.
    pub fn is_gpr(reg_id: u16) -> bool {
        reg_id >= MIPS_GPR_BASE && reg_id < MIPS_GPR_BASE + 32
    }

    /// Check if a register ID is an FPR.
    pub fn is_fpr(reg_id: u16) -> bool {
        reg_id >= MIPS_FPR_BASE && reg_id < MIPS_FPR_BASE + 32
    }

    /// Get the index of a register within its class (0-31 for GPRs/FPRs).
    pub fn get_reg_index(reg_id: u16) -> Option<u8> {
        if Self::is_gpr(reg_id) {
            Some((reg_id - MIPS_GPR_BASE) as u8)
        } else if Self::is_fpr(reg_id) {
            Some((reg_id - MIPS_FPR_BASE) as u8)
        } else {
            None
        }
    }

    /// Check if a register can be used as a base register for addressing.
    pub fn can_be_base_reg(reg_id: u16) -> bool {
        Self::is_gpr(reg_id) && reg_id != ZERO
    }

    /// Get caller-saved GPRs.
    pub fn get_caller_saved_gprs() -> Vec<u16> {
        let mut regs = Vec::new();
        for i in 0..32 {
            let r = MIPS_GPR_BASE + i as u16;
            if i != 0 && Self::is_caller_saved(r) && !Self::is_reserved(r) {
                regs.push(r);
            }
        }
        regs
    }

    /// Get callee-saved GPRs.
    pub fn get_callee_saved_gprs() -> Vec<u16> {
        let mut regs = Vec::new();
        for i in 0..32 {
            let r = MIPS_GPR_BASE + i as u16;
            if Self::is_callee_saved(r) && !Self::is_reserved(r) {
                regs.push(r);
            }
        }
        regs
    }

    /// Get caller-saved FPRs.
    pub fn get_caller_saved_fprs() -> Vec<u16> {
        (F0..=F19).collect()
    }

    /// Get callee-saved FPRs.
    pub fn get_callee_saved_fprs() -> Vec<u16> {
        (F20..=F31).collect()
    }
}

// ============================================================================
// Tests
// ============================================================================

#[cfg(test)]
mod tests {
    use super::*;

    #[test]
    fn test_register_count_constants() {
        assert_eq!(MIPS_GPR_COUNT, 32);
        assert_eq!(MIPS_FPR_COUNT, 32);
        assert_eq!(MIPS_MAX_REG_ID, 4081);
        assert_eq!(MIPS_GPR_BASE, 4000);
        assert_eq!(MIPS_FPR_BASE, 4050);
    }

    #[test]
    fn test_register_ids_unique() {
        let all_regs = [
            ZERO, AT, V0, V1, A0, A1, A2, A3, T0, T1, T2, T3, T4, T5, T6, T7, S0, S1, S2, S3, S4,
            S5, S6, S7, T8, T9, K0, K1, GP, SP, FP, RA, HI, LO, PC,
        ];
        let mut seen = std::collections::HashSet::new();
        for &r in &all_regs {
            assert!(seen.insert(r), "Duplicate register ID: {}", r);
        }
    }

    #[test]
    fn test_fpr_ids_unique() {
        let fprs: Vec<u16> = (F0..=F31).collect();
        let mut seen = std::collections::HashSet::new();
        for &r in &fprs {
            assert!(seen.insert(r), "Duplicate FPR ID: {}", r);
        }
        assert_eq!(fprs.len(), 32);
    }

    #[test]
    fn test_abi_names() {
        assert_eq!(MipsRegisterInfo::get_asm_name(ZERO), "$zero");
        assert_eq!(MipsRegisterInfo::get_asm_name(SP), "$sp");
        assert_eq!(MipsRegisterInfo::get_asm_name(RA), "$ra");
        assert_eq!(MipsRegisterInfo::get_asm_name(FP), "$fp");
        assert_eq!(MipsRegisterInfo::get_asm_name(GP), "$gp");
        assert_eq!(MipsRegisterInfo::get_asm_name(V0), "$v0");
        assert_eq!(MipsRegisterInfo::get_asm_name(A0), "$a0");
        assert_eq!(MipsRegisterInfo::get_asm_name(T0), "$t0");
        assert_eq!(MipsRegisterInfo::get_asm_name(S0), "$s0");
    }

    #[test]
    fn test_fpr_names() {
        assert_eq!(MipsRegisterInfo::get_asm_name(F0), "$f0");
        assert_eq!(MipsRegisterInfo::get_asm_name(F12), "$f12");
        assert_eq!(MipsRegisterInfo::get_asm_name(F31), "$f31");
    }

    #[test]
    fn test_special_reg_names() {
        assert_eq!(MipsRegisterInfo::get_asm_name(HI), "$hi");
        assert_eq!(MipsRegisterInfo::get_asm_name(LO), "$lo");
        assert_eq!(MipsRegisterInfo::get_asm_name(PC), "$pc");
    }

    #[test]
    fn test_get_reg_class() {
        assert_eq!(MipsRegisterInfo::get_reg_class(ZERO), MipsRegClass::GPR);
        assert_eq!(MipsRegisterInfo::get_reg_class(RA), MipsRegClass::GPR);
        assert_eq!(MipsRegisterInfo::get_reg_class(HI), MipsRegClass::Special);
        assert_eq!(MipsRegisterInfo::get_reg_class(LO), MipsRegClass::Special);
        assert_eq!(MipsRegisterInfo::get_reg_class(F0), MipsRegClass::FPR32);
        assert_eq!(MipsRegisterInfo::get_reg_class(F31), MipsRegClass::FPR32);
    }

    #[test]
    fn test_get_reg_width() {
        assert_eq!(MipsRegisterInfo::get_reg_width(V0, false), 32);
        assert_eq!(MipsRegisterInfo::get_reg_width(V0, true), 64);
        assert_eq!(MipsRegisterInfo::get_reg_width(F0, false), 32);
        assert_eq!(MipsRegisterInfo::get_reg_width(HI, true), 64);
    }

    #[test]
    fn test_get_dwarf_num() {
        assert_eq!(MipsRegisterInfo::get_dwarf_num(ZERO), 0);
        assert_eq!(MipsRegisterInfo::get_dwarf_num(AT), 1);
        assert_eq!(MipsRegisterInfo::get_dwarf_num(RA), 31);
        assert_eq!(MipsRegisterInfo::get_dwarf_num(F0), 32);
        assert_eq!(MipsRegisterInfo::get_dwarf_num(F31), 63);
        assert_eq!(MipsRegisterInfo::get_dwarf_num(HI), -1);
    }

    #[test]
    fn test_is_callee_saved() {
        assert!(MipsRegisterInfo::is_callee_saved(S0));
        assert!(MipsRegisterInfo::is_callee_saved(S7));
        assert!(MipsRegisterInfo::is_callee_saved(FP));
        assert!(MipsRegisterInfo::is_callee_saved(RA));
        assert!(MipsRegisterInfo::is_callee_saved(F20));
        assert!(!MipsRegisterInfo::is_callee_saved(T0));
        assert!(!MipsRegisterInfo::is_callee_saved(A0));
    }

    #[test]
    fn test_is_caller_saved() {
        assert!(MipsRegisterInfo::is_caller_saved(T0));
        assert!(MipsRegisterInfo::is_caller_saved(A0));
        assert!(MipsRegisterInfo::is_caller_saved(V0));
        assert!(MipsRegisterInfo::is_caller_saved(F0));
        assert!(!MipsRegisterInfo::is_caller_saved(S0));
    }

    #[test]
    fn test_is_reserved() {
        assert!(MipsRegisterInfo::is_reserved(ZERO));
        assert!(MipsRegisterInfo::is_reserved(K0));
        assert!(MipsRegisterInfo::is_reserved(K1));
        assert!(!MipsRegisterInfo::is_reserved(V0));
        assert!(!MipsRegisterInfo::is_reserved(SP));
    }

    #[test]
    fn test_get_allocatable_gprs() {
        let regs = MipsRegisterInfo::get_allocatable_gprs();
        assert!(regs.len() >= 25);
        // Should not contain zero, at, k0, k1
        assert!(!regs.contains(&ZERO));
        assert!(!regs.contains(&AT));
        assert!(!regs.contains(&K0));
        assert!(!regs.contains(&K1));
    }

    #[test]
    fn test_get_allocatable_fprs() {
        let regs = MipsRegisterInfo::get_allocatable_fprs();
        assert_eq!(regs.len(), 32);
    }

    #[test]
    fn test_get_argument_regs() {
        let regs = MipsRegisterInfo::get_argument_regs();
        assert_eq!(regs, vec![A0, A1, A2, A3]);
    }

    #[test]
    fn test_get_fp_argument_regs() {
        let regs = MipsRegisterInfo::get_fp_argument_regs(false);
        assert_eq!(regs.len(), 8);
        assert!(regs.contains(&F12));
        assert!(regs.contains(&F19));
    }

    #[test]
    fn test_get_return_regs() {
        let regs = MipsRegisterInfo::get_return_regs();
        assert_eq!(regs, vec![V0, V1]);
    }

    #[test]
    fn test_get_fp_return_regs() {
        let regs = MipsRegisterInfo::get_fp_return_regs();
        assert_eq!(regs, vec![F0, F1]);
    }

    #[test]
    fn test_special_regs() {
        assert_eq!(MipsRegisterInfo::get_frame_pointer_reg(), FP);
        assert_eq!(MipsRegisterInfo::get_return_address_reg(), RA);
        assert_eq!(MipsRegisterInfo::get_stack_pointer_reg(), SP);
        assert_eq!(MipsRegisterInfo::get_global_pointer_reg(), GP);
        assert_eq!(MipsRegisterInfo::get_zero_reg(), ZERO);
    }

    #[test]
    fn test_is_gpr_and_fpr() {
        assert!(MipsRegisterInfo::is_gpr(ZERO));
        assert!(MipsRegisterInfo::is_gpr(RA));
        assert!(!MipsRegisterInfo::is_gpr(F0));
        assert!(!MipsRegisterInfo::is_gpr(HI));
        assert!(MipsRegisterInfo::is_fpr(F0));
        assert!(MipsRegisterInfo::is_fpr(F31));
        assert!(!MipsRegisterInfo::is_fpr(ZERO));
    }

    #[test]
    fn test_get_reg_index() {
        assert_eq!(MipsRegisterInfo::get_reg_index(ZERO), Some(0));
        assert_eq!(MipsRegisterInfo::get_reg_index(AT), Some(1));
        assert_eq!(MipsRegisterInfo::get_reg_index(RA), Some(31));
        assert_eq!(MipsRegisterInfo::get_reg_index(F0), Some(0));
        assert_eq!(MipsRegisterInfo::get_reg_index(F31), Some(31));
        assert_eq!(MipsRegisterInfo::get_reg_index(HI), None);
    }

    #[test]
    fn test_can_be_base_reg() {
        assert!(!MipsRegisterInfo::can_be_base_reg(ZERO));
        assert!(MipsRegisterInfo::can_be_base_reg(SP));
        assert!(MipsRegisterInfo::can_be_base_reg(T0));
        assert!(!MipsRegisterInfo::can_be_base_reg(F0));
    }

    #[test]
    fn test_reg_class_display() {
        assert_eq!(MipsRegClass::GPR.to_string(), "GPR");
        assert_eq!(MipsRegClass::FPR32.to_string(), "FPR32");
        assert_eq!(MipsRegClass::FPR64.to_string(), "FPR64");
        assert_eq!(MipsRegClass::Special.to_string(), "Special");
    }

    #[test]
    fn test_caller_saved_gprs_count() {
        let regs = MipsRegisterInfo::get_caller_saved_gprs();
        assert!(regs.len() >= 13); // t0-t9, v0-v1, a0-a3
    }

    #[test]
    fn test_callee_saved_gprs_count() {
        let regs = MipsRegisterInfo::get_callee_saved_gprs();
        assert!(regs.len() >= 8); // s0-s7
        assert!(regs.contains(&S0));
        assert!(regs.contains(&S7));
    }

    #[test]
    fn test_caller_saved_fprs_count() {
        let regs = MipsRegisterInfo::get_caller_saved_fprs();
        assert_eq!(regs.len(), 20); // f0-f19
    }

    #[test]
    fn test_callee_saved_fprs_count() {
        let regs = MipsRegisterInfo::get_callee_saved_fprs();
        assert_eq!(regs.len(), 12); // f20-f31
    }

    #[test]
    fn test_all_gpr_ids_in_range() {
        for i in 0..32 {
            let r = MIPS_GPR_BASE + i as u16;
            assert!(r >= MIPS_GPR_BASE && r < MIPS_GPR_BASE + 32);
        }
    }

    #[test]
    fn test_all_fpr_ids_in_range() {
        for i in 0..32 {
            let r = MIPS_FPR_BASE + i as u16;
            assert!(r >= MIPS_FPR_BASE && r < MIPS_FPR_BASE + 32);
        }
    }
}