Skip to main content

rucc_codegen/
split.rs

1//! Splitting critical edges, so that every edge that carries values has somewhere to put them.
2//!
3//! Design: `spec/10-backend.md` section 10.4.
4//!
5//! An edge carries values when the block it goes to takes parameters, and giving a parameter its
6//! value is a move. The move has to happen on the edge and not before it or after it, because
7//! before it is a block that goes somewhere else too and after it is a block that is arrived at
8//! from somewhere else too, and in either case the move would run on a path it was not written
9//! for. An edge out of a block with one successor can put its moves at the end of that block,
10//! since every path through it takes the edge. An edge into a block with one predecessor can put
11//! them at the start of that block, for the same reason the other way round. An edge that is
12//! neither, which is what a critical edge is, has neither place, and the allocator says so:
13//! `rucc_regalloc` asserts that it never sees one.
14//!
15//! So one is turned into two. A block with nothing in it goes on the edge, the arguments move on
16//! to the second half, and both halves are now uncritical: the first goes to a block with one
17//! predecessor and the second leaves a block with one successor. Which of the two the moves end
18//! up in is the allocator's answer and not this one's, and either is correct.
19//!
20//! # What it leaves behind
21//!
22//! An empty block, which is a jump to the next thing unless the layout puts it where it falls
23//! through. That is a cost, and it is why an edge with nothing to carry is left alone: there are
24//! no moves to find a place for, so splitting it would buy a jump and nothing else.
25//!
26//! # The other edge with nowhere to put a move
27//!
28//! A computed `goto` leaves its block through a register, and the moves an edge out of it carries
29//! would have to be written somewhere the jump has already gone past. So there is a second pass
30//! here, [`indirect`], which takes the values off those edges and puts them in a block of their
31//! own in front of each label. It runs first, and what it leaves behind is edges the splitting
32//! below then has nothing to do about.
33//!
34//! [`pads`] is here for the same reason and not for a reason of its own: the blocks those labels
35//! begin at are addresses an indirect branch arrives at, and a machine that checks the forward edge
36//! wants a landing pad at every one of them. Which block an address names is settled by the pass
37//! above, so the pad is written after it and not where the prologue's own pad is written.
38
39use std::collections::HashMap;
40
41use rucc_base::Interner;
42use rucc_mir as mir;
43use rucc_target::{BranchInsts, FrameInsts, RegClass};
44
45/// Splits every critical edge that carries values, and gives back how many it split.
46///
47/// Run after lowering and before allocation. Running it twice is running it once, because the
48/// blocks it adds have one successor each and are never the source of a critical edge.
49pub fn critical(func: &mut mir::Func) -> usize {
50    let preds = preds(func);
51    let blocks: Vec<mir::Block> = func.blocks().collect();
52    let mut split = 0;
53    for block in blocks {
54        if func[block].succs.len() < 2 {
55            continue;
56        }
57        for index in 0..func[block].succs.len() {
58            let call = func[block].succs[index].clone();
59            if call.args.is_empty() || preds[call.block.index()] < 2 {
60                continue;
61            }
62            // The new block is at the end of the layout, which is where a block that is a jump
63            // and nothing else does the least harm before the layout pass has an opinion.
64            //
65            // It runs exactly as often as the edge it sits on is taken, and both halves of that
66            // edge are now that edge, which is why the weight is copied onto all three rather
67            // than left at what a block nobody told anything runs. A block on a cold edge that
68            // claimed to run once per call would be one the layout put in the middle of the hot
69            // path.
70            let weight = call.weight;
71            let half = func.create_block();
72            func.set_weight(half, weight);
73            *func.succs_mut(half) = vec![call];
74            func.succs_mut(block)[index] = mir::BlockCall::to(half).taken(weight);
75            split += 1;
76        }
77    }
78    split
79}
80
81/// Takes the values off every edge out of a computed `goto`, and gives back how many blocks it
82/// made to hold them.
83///
84/// Run after lowering and before [`critical`], which then sees edges with nothing on them and
85/// leaves them alone. Running it twice is running it once, for the reason the splitting above is:
86/// the blocks it adds end in a jump rather than in a branch through a register.
87///
88/// # What is wrong with the edge it takes the values off
89///
90/// Every other edge in the function is out of a block whose last instruction the layout writes, so
91/// an edge that is the only way out of its block can put its moves at the end of that block and
92/// they land in front of the jump. A block that leaves through a register already ends in the jump
93/// when the allocator runs, because where it goes is a value and a value is something selection
94/// reads rather than something the layout knows. Moves at the end of that block would be written
95/// after the jump, where nothing runs them, and moves in front of it would be written across the
96/// register the jump reads, which the allocator believes is dead from the jump onwards and is free
97/// to hand to one of the moves.
98///
99/// So the moves go somewhere else. Each label an indirect branch reaches gets a block in front of
100/// it that carries the values, the branch goes to that block with nothing on the edge, and the
101/// address the `&&label` produces is the address of that block rather than of the label's own. The
102/// new block is arrived at one way and leaves one way, so its own edge has both of the places the
103/// splitting above talks about and the allocator is content.
104///
105/// # One label, one address, and two branches that disagree
106///
107/// A label has one address, so two computed `goto`s that reach it both arrive at whatever block
108/// that address names, and the values they carry are not the same values. One block in front of
109/// the label cannot move two different sets of registers.
110///
111/// So they are made to agree first. Each parameter of the label gets a register of its own, every
112/// branch writes that register in front of its jump, and the block in front of the label carries
113/// those registers and nothing else. That is what gcc does about the same problem, which it calls
114/// coalescing across an abnormal edge, done here rather than while the values are still the
115/// optimizer's.
116///
117/// Writing them in front of the jump is safe, which is not obvious, since a branch that goes five
118/// ways writes the registers of one of those ways on the path to all five. What makes it safe is
119/// that nothing reads those registers except the block in front of the label, and the only way to
120/// reach that block is an edge out of a branch, which writes them on the way. So a value written
121/// here and not used is a value overwritten before anything looks, whichever way the jump went.
122///
123/// # One register for one value, and not one for every place it is given to
124///
125/// That safety is also what makes the cost of it worth watching. A branch writes the registers of
126/// every label it can reach, so a register for every parameter of every label is a whole table's
127/// worth of moves in front of every jump in the function, and a dispatch table is a branch that
128/// reaches hundreds of labels. An interpreter hands each of them whatever its loop had in hand at
129/// the jump, which is the same few values over and over, so a register for each place one of them
130/// lands means those values written a hundred times over before every instruction the interpreter
131/// runs. That is not a small constant. It is what makes an interpreter built this way ten times
132/// slower than the same interpreter built with a `switch` instead of the computed `goto`.
133///
134/// So the register belongs to the value rather than to the place. Two parameters are given one
135/// register when they are drawn from the same class and every branch in the function gives them
136/// the same register, which is exactly when one register can stand for both, and a branch writes
137/// each register it has to write once however many labels asked for it. A dispatch table where
138/// every label wants the instruction pointer writes the instruction pointer once. A label
139/// something else reaches, or a label given something no other label is given, keeps a register of
140/// its own, and a branch that gives one label nothing shares nothing with it, since a register
141/// that branch never wrote is not one the label can be given.
142///
143/// # Panics
144///
145/// Panics on a class of register the machine named no move for, which is a function carrying a
146/// value of a kind the target never said how to copy, and on a branch that has lost the terminator
147/// it was found by, which nothing between the finding and the use of it can do. Both are a target
148/// description or a function that was built wrongly, and both are worth finding here rather than as
149/// a value that arrives somewhere it was never written.
150pub fn indirect(
151    func: &mut mir::Func,
152    branch: &BranchInsts,
153    frame: &FrameInsts,
154    names: &mut Interner,
155) -> usize {
156    let jump = mir::Opcode::new(names.intern(&format!("{}{}", branch.prefix, branch.indirect)));
157    let branches: Vec<mir::Block> = func
158        .blocks()
159        .filter(|&block| func.terminator(block).is_some_and(|last| func[last].opcode == jump))
160        .collect();
161    // Nothing at all in almost every function, and the walk at the bottom is over every instruction
162    // in it, so the answer is arrived at here rather than paid for everywhere.
163    if branches.is_empty() {
164        return 0;
165    }
166    // In the order the branches name them rather than in whatever order a hash gives, so that two
167    // runs of the compiler over one program write the same blocks.
168    let mut targets: Vec<mir::Block> = Vec::new();
169    for &block in &branches {
170        for call in &func[block].succs {
171            if !call.args.is_empty() && !targets.contains(&call.block) {
172                targets.push(call.block);
173            }
174        }
175    }
176
177    // One register per thing a branch has to give, rather than one per place it is given to. The
178    // key is what every branch gives that parameter, so two parameters given the same register by
179    // the same branches are given it in one register and a branch writes that register once.
180    let mut homes: HashMap<Given, mir::Reg> = HashMap::new();
181    // What each branch writes in front of its jump, in the order it was first asked for, and never
182    // the same register twice. Two parameters that share a register are given it by the one move.
183    let mut writes: Vec<Vec<(mir::Reg, mir::Reg, RegClass)>> = vec![Vec::new(); branches.len()];
184    let mut entries: HashMap<mir::Block, mir::Block> = HashMap::new();
185
186    for target in targets {
187        let params = func[target].params.clone();
188        let given = given(func, &branches, target);
189        let mut carried: Vec<mir::Reg> = Vec::new();
190        for (index, param) in params.iter().enumerate() {
191            let key: Given = (
192                param.class,
193                given.iter().map(|edges| edges.iter().map(|args| args[index]).collect()).collect(),
194            );
195            let home = *homes.entry(key).or_insert_with(|| func.new_vreg(param.class));
196            carried.push(home);
197            for (branch, edges) in given.iter().enumerate() {
198                for args in edges {
199                    if !writes[branch].iter().any(|&(written, _, _)| written == home) {
200                        writes[branch].push((home, args[index], param.class));
201                    }
202                }
203            }
204        }
205        let entry = func.create_block();
206        let mut total = mir::Weight::NEVER;
207        for &block in &branches {
208            for index in 0..func[block].succs.len() {
209                if func[block].succs[index].block != target {
210                    continue;
211                }
212                // The block in front of the label runs as often as every branch that reaches it,
213                // which is the same sum the weight of a block with that many edges into it would
214                // be.
215                let weight = func[block].succs[index].weight;
216                total = mir::Weight::parts(total.raw().saturating_add(weight.raw()));
217                func.succs_mut(block)[index] = mir::BlockCall::to(entry).taken(weight);
218            }
219        }
220        func.set_weight(entry, total);
221        *func.succs_mut(entry) = vec![mir::BlockCall::with(target, carried).taken(total)];
222        entries.insert(target, entry);
223    }
224
225    // And the moves themselves, once every label has asked for what it wants, since what one label
226    // asks for is what another may already have asked the same branch for.
227    for (branch, moves) in branches.iter().zip(&writes) {
228        let last = func.terminator(*branch).expect("a block that ends in a jump");
229        for &(home, arg, class) in moves {
230            let name = frame.moves(class).expect("a class this machine can move").mov;
231            let opcode = mir::Opcode::new(names.intern(&format!("{}{name}", frame.prefix)));
232            let inst = func.build_loose(opcode).def(home, class).uses(arg, class).finish();
233            func.insert_before(last, inst);
234        }
235    }
236
237    // And the addresses, which is the half of this that is not about edges. Every `&&label` in the
238    // function names a block, and a label with a block in front of it now begins at that block, so
239    // an address left pointing at the label's own block would be a jump past the moves.
240    let mut addresses: Vec<mir::MemRef> = Vec::new();
241    for block in func.blocks() {
242        for inst in func.insts(block) {
243            if let Some(mem) = func[inst].mem {
244                addresses.push(mem);
245            }
246        }
247    }
248    for mem in addresses {
249        if let Some(named) = func[mem].block {
250            if let Some(&entry) = entries.get(&named) {
251                func[mem].block = Some(entry);
252            }
253        }
254    }
255    // And the names, for the same reason. A block an image points at is one a `goto *p` arrives at,
256    // so a name left on the label's own block would be an address in a table that skips the moves,
257    // which is the one way into the block that would not have made them.
258    for (block, _) in &mut func.labels {
259        if let Some(&entry) = entries.get(block) {
260            *block = entry;
261        }
262    }
263    entries.len()
264}
265
266/// Puts a landing pad at the front of every block whose address is taken, and gives back how many
267/// it wrote.
268///
269/// Run after [`indirect`], because the block an address names is not settled until that has moved
270/// the addresses on to the blocks it made, and only when the command line asked for the forward
271/// edge to be checked. Nothing is written otherwise, which is why the name comes in as an option
272/// and why a target with no such instruction is a target this does nothing on.
273///
274/// The pad a prologue opens with is written elsewhere, in `crate::finish`, because the address it
275/// makes reachable is the address of the function rather than a place inside it. These are the
276/// other addresses an indirect branch may arrive at, and a machine that checks the forward edge
277/// faults on one that has no pad, so a computed `goto` compiled without this would be a program
278/// that ran everywhere except on the hardware the flag was turned on for.
279pub fn pads(
280    func: &mut mir::Func,
281    frame: &FrameInsts,
282    landing: Option<&'static str>,
283    names: &mut Interner,
284) -> usize {
285    let Some(name) = landing else { return 0 };
286    let opcode = mir::Opcode::new(names.intern(&format!("{}{name}", frame.prefix)));
287    let mut addressed: Vec<mir::Block> = Vec::new();
288    for block in func.blocks() {
289        for inst in func.insts(block) {
290            if let Some(mem) = func[inst].mem {
291                if let Some(named) = func[mem].block {
292                    if !addressed.contains(&named) {
293                        addressed.push(named);
294                    }
295                }
296            }
297        }
298    }
299    // And every arm of a jump table, which an indirect jump arrives at the same way.
300    for table in &func.tables {
301        let Some(jump) = func.block_of(table.jump) else { continue };
302        for &cell in &table.cells {
303            let named = func[jump].succs[cell as usize].block;
304            if !addressed.contains(&named) {
305                addressed.push(named);
306            }
307        }
308    }
309    for &block in &addressed {
310        let inst = func.build_loose(opcode).finish();
311        func.prepend_inst(block, inst);
312    }
313    addressed.len()
314}
315
316/// What decides whether two parameters can be given their value in one register: the class the
317/// parameter is drawn from, and the register every branch in the function gives it, in the order
318/// the branches are in and with one entry per edge inside that. A branch that does not reach the
319/// label gives nothing, which is a length of zero and is as much a part of the answer as a
320/// register is, since sharing with a parameter a branch never gives anything to would be reading a
321/// register that branch never wrote.
322type Given = (RegClass, Vec<Vec<mir::Reg>>);
323
324/// What each branch gives that label, edge by edge.
325///
326/// One entry per branch and in the branches' own order, since a label two branches reach and a
327/// label one branch reaches twice are not given the same thing. A branch is allowed to reach one
328/// label twice, which a table with the same label in two of its cells is, so what a branch gives
329/// is a list of what it gives rather than one set of registers.
330fn given(func: &mir::Func, branches: &[mir::Block], target: mir::Block) -> Vec<Vec<Vec<mir::Reg>>> {
331    branches
332        .iter()
333        .map(|&block| {
334            func[block]
335                .succs
336                .iter()
337                .filter(|call| call.block == target)
338                .map(|call| call.args.clone())
339                .collect()
340        })
341        .collect()
342}
343
344/// How many edges arrive at each block, counted by index rather than in layout order so that a
345/// block added while splitting can be looked up in the same table.
346fn preds(func: &mir::Func) -> Vec<usize> {
347    let mut counts = vec![0; func.block_count()];
348    for block in func.blocks() {
349        for call in &func[block].succs {
350            counts[call.block.index()] += 1;
351        }
352    }
353    counts
354}
355
356#[cfg(test)]
357mod tests {
358    use rucc_base::Interner;
359    use rucc_target::x86_64::{BRANCH, FRAME, GPR, REGS};
360
361    use super::*;
362
363    /// A diamond: one block that goes two ways and one block both ways arrive at, with as many
364    /// parameters on the block they arrive at as the test asks for.
365    fn diamond(params: usize) -> (Interner, mir::Func, [mir::Block; 4]) {
366        let mut names = Interner::new();
367        let mut func = mir::Func::new(names.intern("f"));
368        let head = func.create_block();
369        let left = func.create_block();
370        let right = func.create_block();
371        let join = func.create_block();
372        // The values arrive in the head, so that they have somewhere to be defined and the
373        // printer has a name for them. Nothing here runs an allocator, which is the one thing
374        // that would object to a first block with parameters.
375        let args: Vec<mir::Reg> = (0..params).map(|_| func.append_param(head, GPR)).collect();
376        for _ in 0..params {
377            func.append_param(join, GPR);
378        }
379        *func.succs_mut(head) = vec![mir::BlockCall::to(left), mir::BlockCall::to(right)];
380        *func.succs_mut(left) = vec![mir::BlockCall::with(join, args.clone())];
381        *func.succs_mut(right) = vec![mir::BlockCall::with(join, args)];
382        (names, func, [head, left, right, join])
383    }
384
385    /// Where each block goes, which is the whole of what this changes.
386    fn edges(func: &mir::Func) -> Vec<Vec<usize>> {
387        func.blocks()
388            .map(|block| func[block].succs.iter().map(|call| call.block.index()).collect())
389            .collect()
390    }
391
392    #[test]
393    fn an_edge_that_is_the_only_way_out_is_left_alone() {
394        let (_, mut func, _) = diamond(1);
395        // The two edges into the join carry a value each and neither is critical, because the
396        // block each leaves goes nowhere else.
397        assert_eq!(critical(&mut func), 0);
398        assert_eq!(edges(&func), vec![vec![1, 2], vec![3], vec![3], vec![]]);
399    }
400
401    #[test]
402    fn a_critical_edge_carrying_a_value_is_split_in_two() {
403        let (_, mut func, [head, _, _, join]) = diamond(1);
404        // Now the head goes straight to the join as well, so both of its arms are critical: it
405        // has two ways out and the join has three ways in.
406        let arg = func.append_param(head, GPR);
407        func.succs_mut(head).push(mir::BlockCall::with(join, vec![arg]));
408        func.succs_mut(head).swap(1, 2);
409
410        assert_eq!(critical(&mut func), 1);
411        assert_eq!(
412            edges(&func),
413            // The head's second arm is the new block and the new block goes to the join. The
414            // other two arms are untouched, because each goes to a block with one way in.
415            vec![vec![1, 4, 2], vec![3], vec![3], vec![], vec![3]]
416        );
417    }
418
419    #[test]
420    fn a_critical_edge_carrying_nothing_is_left_alone() {
421        let (_, mut func, [head, _, _, join]) = diamond(0);
422        func.succs_mut(head).push(mir::BlockCall::to(join));
423
424        // Critical and not split, because there is no move to find a place for and a block that
425        // is a jump and nothing else is worth more than nothing.
426        assert_eq!(critical(&mut func), 0);
427    }
428
429    #[test]
430    fn the_arguments_move_on_to_the_half_that_arrives() {
431        let (names, mut func, [head, _, _, join]) = diamond(1);
432        let arg = func.append_param(head, GPR);
433        func.succs_mut(head).push(mir::BlockCall::with(join, vec![arg]));
434
435        assert_eq!(critical(&mut func), 1);
436        // What the first half carries is nothing, since the block it goes to asks for nothing,
437        // and what the second half carries is what the whole edge used to.
438        let half = func.blocks().last().expect("the block the split added");
439        assert_eq!(func[head].succs[2].args, Vec::new());
440        assert_eq!(func[half].succs[0].args, vec![arg]);
441        assert_eq!(
442            mir::print_func(&func, &names, &REGS),
443            "mfunc @f {\nblock0(%0:gpr, %1:gpr):\n    block1, block2, block4\n\n\
444             block1:\n    block3(%0)\n\nblock2:\n    block3(%0)\n\n\
445             block3(%2:gpr):\n\nblock4:\n    block3(%1)\n}\n"
446        );
447    }
448
449    #[test]
450    fn splitting_twice_is_splitting_once() {
451        let (_, mut func, [head, _, _, join]) = diamond(1);
452        let arg = func.append_param(head, GPR);
453        func.succs_mut(head).push(mir::BlockCall::with(join, vec![arg]));
454
455        assert_eq!(critical(&mut func), 1);
456        assert_eq!(critical(&mut func), 0);
457    }
458
459    /// A function with one label whose address is taken and as many blocks leaving through that
460    /// address as the test asks for, each carrying as many values to the label as it asks for.
461    fn computed(branches: usize, params: usize) -> (Interner, mir::Func) {
462        let mut names = Interner::new();
463        let mut func = mir::Func::new(names.intern("f"));
464        let head = func.create_block();
465        let label = func.create_block();
466        for _ in 0..params {
467            func.append_param(label, GPR);
468        }
469        let lea = mir::Opcode::new(names.intern("x64.lea_64"));
470        let jump = mir::Opcode::new(names.intern("x64.jmp_reg"));
471        for _ in 0..branches {
472            // Every branch works the address out for itself, which is what a program that takes
473            // the address of a label twice looks like once the values are in registers.
474            let args: Vec<mir::Reg> = (0..params).map(|_| func.append_param(head, GPR)).collect();
475            let address = func.new_vreg(GPR);
476            let at = if branches == 1 { head } else { func.create_block() };
477            func.build(at, lea).def(address, GPR).mem(mir::Mem::block(label)).finish();
478            func.build(at, jump).operand(mir::Operand::read(address, GPR)).finish();
479            *func.succs_mut(at) = vec![mir::BlockCall::with(label, args)];
480        }
481        (names, func)
482    }
483
484    /// Which block each address in the function names, in the order the instructions are in.
485    fn addressed(func: &mir::Func) -> Vec<usize> {
486        func.blocks()
487            .flat_map(|block| func.insts(block).collect::<Vec<_>>())
488            .filter_map(|inst| func[inst].mem)
489            .filter_map(|mem| func[mem].block)
490            .map(mir::Block::index)
491            .collect()
492    }
493
494    #[test]
495    fn the_values_a_computed_goto_carries_move_into_a_block_in_front_of_the_label() {
496        let (mut names, mut func) = computed(1, 1);
497        assert_eq!(indirect(&mut func, &BRANCH, &FRAME, &mut names), 1);
498
499        // The branch goes to the new block carrying nothing, and the new block carries the value
500        // the branch used to. The address the `lea` works out is the new block's as well, since
501        // arriving at the label without going through the new block is arriving without the value.
502        assert_eq!(edges(&func), vec![vec![2], vec![], vec![1]]);
503        assert_eq!(func[mir::Block::new(0)].succs[0].args, Vec::new());
504        assert_eq!(addressed(&func), vec![2]);
505    }
506
507    #[test]
508    fn two_computed_gotos_that_reach_one_label_are_made_to_agree() {
509        let (mut names, mut func) = computed(2, 1);
510        assert_eq!(indirect(&mut func, &BRANCH, &FRAME, &mut names), 1);
511
512        // One block in front of the label and not two, because the label has one address and both
513        // branches arrive at it. What makes that sound is the move each branch writes in front of
514        // its own jump, which puts its value in the register that block carries.
515        assert_eq!(edges(&func), vec![vec![], vec![], vec![4], vec![4], vec![1]]);
516        let text = mir::print_func(&func, &names, &REGS);
517        assert_eq!(text.matches("x64.mov_rr_64").count(), 2, "{text}");
518        // In front of the jump rather than behind it, since nothing behind a jump runs.
519        for line in text.lines().collect::<Vec<_>>().windows(2) {
520            if line[1].contains("x64.jmp_reg") {
521                assert!(line[0].contains("x64.mov_rr_64"), "{text}");
522            }
523        }
524        assert_eq!(addressed(&func), vec![4, 4]);
525    }
526
527    /// A function with one computed `goto` that reaches as many labels as the test asks for, each
528    /// given the one value the branch has in hand, which is the shape of a dispatch table.
529    fn table(labels: usize) -> (Interner, mir::Func, Vec<mir::Block>) {
530        let mut names = Interner::new();
531        let mut func = mir::Func::new(names.intern("f"));
532        let head = func.create_block();
533        let arg = func.append_param(head, GPR);
534        let lea = mir::Opcode::new(names.intern("x64.lea_64"));
535        let jump = mir::Opcode::new(names.intern("x64.jmp_reg"));
536        let mut targets = Vec::new();
537        for _ in 0..labels {
538            let label = func.create_block();
539            func.append_param(label, GPR);
540            targets.push(label);
541            func.succs_mut(head).push(mir::BlockCall::with(label, vec![arg]));
542        }
543        let address = func.new_vreg(GPR);
544        func.build(head, lea).def(address, GPR).mem(mir::Mem::block(targets[0])).finish();
545        func.build(head, jump).operand(mir::Operand::read(address, GPR)).finish();
546        (names, func, targets)
547    }
548
549    #[test]
550    fn labels_a_branch_gives_the_same_value_are_given_it_in_one_register() {
551        let (mut names, mut func, _) = table(8);
552        assert_eq!(indirect(&mut func, &BRANCH, &FRAME, &mut names), 8);
553
554        // One move in front of the jump and not eight, because the eight labels are given the one
555        // value and it is now in the one register. Eight blocks were still made, since each label
556        // needs the block that moves that register on to its own parameter.
557        let text = mir::print_func(&func, &names, &REGS);
558        assert_eq!(text.matches("x64.mov_rr_64").count(), 1, "{text}");
559    }
560
561    #[test]
562    fn a_label_given_something_else_keeps_a_register_of_its_own() {
563        let (mut names, mut func, targets) = table(8);
564        let head = mir::Block::new(0);
565        let other = func.append_param(head, GPR);
566        let last = func[head].succs.len() - 1;
567        func.succs_mut(head)[last] = mir::BlockCall::with(targets[7], vec![other]);
568
569        assert_eq!(indirect(&mut func, &BRANCH, &FRAME, &mut names), 8);
570        // Two moves: one register for the seven labels given the same value, and one for the label
571        // given the other. Sharing is about what a label is given and not about how many there are.
572        let text = mir::print_func(&func, &names, &REGS);
573        assert_eq!(text.matches("x64.mov_rr_64").count(), 2, "{text}");
574    }
575
576    #[test]
577    fn labels_that_take_different_numbers_of_values_still_share_the_ones_they_agree_on() {
578        let (mut names, mut func, targets) = table(8);
579        let head = mir::Block::new(0);
580        let arg = func[head].params[0].reg;
581        let other = func.append_param(head, GPR);
582        // The last label takes a second value, which is what an interpreter looks like: each of
583        // its labels uses what it needs and no two of them need quite the same list.
584        func.append_param(targets[7], GPR);
585        let last = func[head].succs.len() - 1;
586        func.succs_mut(head)[last] = mir::BlockCall::with(targets[7], vec![arg, other]);
587
588        assert_eq!(indirect(&mut func, &BRANCH, &FRAME, &mut names), 8);
589        // Two moves, not nine. The first parameter of the long label is given what the other seven
590        // are given, so it takes the same register, and only the value nothing else is given needs
591        // one of its own.
592        let text = mir::print_func(&func, &names, &REGS);
593        assert_eq!(text.matches("x64.mov_rr_64").count(), 2, "{text}");
594    }
595
596    #[test]
597    fn an_edge_out_of_a_computed_goto_that_carries_nothing_is_left_alone() {
598        let (mut names, mut func) = computed(1, 0);
599
600        // No values to carry, so no block to carry them, and the address stays the label's own.
601        assert_eq!(indirect(&mut func, &BRANCH, &FRAME, &mut names), 0);
602        assert_eq!(addressed(&func), vec![1]);
603    }
604
605    #[test]
606    fn a_function_with_no_computed_goto_in_it_is_left_alone() {
607        let (mut names, mut func, _) = diamond(1);
608        assert_eq!(indirect(&mut func, &BRANCH, &FRAME, &mut names), 0);
609        assert_eq!(edges(&func), vec![vec![1, 2], vec![3], vec![3], vec![]]);
610    }
611
612    #[test]
613    fn what_it_leaves_is_nothing_for_the_splitting_below_to_do() {
614        let (mut names, mut func) = computed(2, 1);
615        indirect(&mut func, &BRANCH, &FRAME, &mut names);
616        // The edges out of the branches carry nothing now, and the edges out of the blocks it
617        // added are the only way out of those blocks, so neither kind is critical.
618        assert_eq!(critical(&mut func), 0);
619    }
620
621    /// The first instruction of each block, by opcode, and an empty string for a block with
622    /// nothing in it.
623    fn opens(func: &mir::Func, names: &Interner) -> Vec<String> {
624        func.blocks()
625            .map(|block| match func.insts(block).next() {
626                Some(inst) => names.resolve(func[inst].opcode.name()).to_owned(),
627                None => String::new(),
628            })
629            .collect()
630    }
631
632    #[test]
633    fn the_block_a_label_begins_at_gets_a_landing_pad_when_the_forward_edge_is_checked() {
634        let (mut names, mut func) = computed(2, 1);
635        indirect(&mut func, &BRANCH, &FRAME, &mut names);
636
637        // One pad, at the block in front of the label, because that is the block both addresses
638        // name once the values have been moved on to it. The label's own block is arrived at by an
639        // ordinary edge from there and wants nothing.
640        assert_eq!(pads(&mut func, &FRAME, FRAME.landing, &mut names), 1);
641        assert_eq!(opens(&func, &names), ["", "", "x64.lea_64", "x64.lea_64", "x64.endbr64"]);
642    }
643
644    #[test]
645    fn a_label_with_no_block_in_front_of_it_gets_the_pad_itself() {
646        let (mut names, mut func) = computed(1, 0);
647        indirect(&mut func, &BRANCH, &FRAME, &mut names);
648
649        // Nothing was moved on to anything, so the address still names the label and the pad goes
650        // where the address goes.
651        assert_eq!(pads(&mut func, &FRAME, FRAME.landing, &mut names), 1);
652        assert_eq!(opens(&func, &names), ["x64.lea_64", "x64.endbr64"]);
653    }
654
655    #[test]
656    fn nothing_is_written_when_the_forward_edge_is_not_checked() {
657        let (mut names, mut func) = computed(1, 0);
658        assert_eq!(pads(&mut func, &FRAME, None, &mut names), 0);
659        assert_eq!(opens(&func, &names), ["x64.lea_64", ""]);
660    }
661}