nordic-dev.net/rust-gpu
2
0
Fork 0
mirror of https://github.com/EmbarkStudios/rust-gpu.git synced 2025-02-20 10:56:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Implement switch

Browse Source
This commit is contained in:
khyperia 2020-08-14 14:53:47 +02:00
parent 56e354f0d5
commit 541e5701e6
1 changed files with 28 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
rustc_codegen_spirv/src/trans.rs
View File

@ -251,6 +251,34 @@ fn trans_terminator<'tcx>(ctx: &mut FnCtx<'_, 'tcx>, term: &Terminator<'tcx>) {
let destination_bb = ctx.get_basic_block(destination.1);
ctx.spirv().branch(destination_bb).unwrap();
}
TerminatorKind::SwitchInt {
ref discr,
switch_ty: _,
ref values,
ref targets,
} => {
// the last target is the default block
assert_eq!(values.len() + 1, targets.len());
let (selector, _selector_type) = trans_operand(ctx, discr);
let target = values
.iter()
.zip(targets.iter())
.map(|(&value, &target)| {
if value > u32::max_value() as _ {
// TODO: Fall back to if/else chain for these
panic!(
"Unimplemented: switch value too high for spir-v: {:?}",
value
);
}
let value_spirv = value as u32;
let target_spirv = ctx.get_basic_block(target);
(value_spirv, target_spirv)
})
.collect::<Vec<_>>();
let default = ctx.get_basic_block(*targets.last().unwrap());
ctx.spirv().switch(selector, default, target).unwrap();
}
ref thing => panic!("Unknown terminator: {:?}", thing),
}
}