如何用重复的 u16 值填充 [u8] 数组？

Question

我希望构建一个数组，该数组将被复制到 RAM 中并发送到 LCD 驱动程序。我希望能够传入一个颜色值 (color: u16) 并使用它来填充数组。请注意，我使用的是 #![no_std]，因为它适用于嵌入式应用程序。

显而易见的简单方法如下：

let mut block: [u8;256] = [0; 256];
for i in (0..block.len()).step_by(2) {
    block[i] = ((color && 0xFF00) >> 8) as u8;
    block[i+1] = (color && 0xFF) as u8;
}

由于我的个人经验更多来自 C，我想知道是否有使用类型转换和转换的更好解决方案。例如，使用 Rust 初始化一个 u16 数组并转换为 u8 数组：

let block_u16: [u16; 128] = [color; 128]
let block_u8 = block_u16 as [u8; 256];

供参考，目标函数原型为：

spi::setup_block(block: &[u8]);

Answer 1

您可以通过 transmute 走不安全的路线，但我不推荐这样做。我不保证 C 风格的版本在小端和大端机器上做同样的事情。

就我个人而言，我会在切片上使用块迭代器。

let color_bytes = color.to_le_bytes();
for word in block.chunks_exact_mut(2) {
    word[0] = color_bytes[0];
    word[1] = color_bytes[1];
}

Answer 2

您可以使用第二种方法（将 u16 数组转换为 u8 指针）。

let color: u16 = 0x0102;
let mut block: [ u16; 128 ] = [ color; 128 ];
let mut block = unsafe { 
    core::slice::from_raw_parts_mut( block.as_mut_ptr( ) as *mut u8, block.len( ) * 2 ) 
};

注意 endianness 使用此解决方案，它可能无法在您的目标架构上执行您想要的操作。

Answer 3

for chunk in block.chunks_exact_mut(2) {
    chunk.copy_from_slice(&color.to_be_bytes());
}

如何用重复的 u16 值填充 [u8] 数组？

How to fill a [u8] array with a repeated u16 value?

type-conversion

type-punning

rust

rust-no-std