Harry Chen
6 years ago
parent
0acb65ff74
commit
8bc00324c8
@ -1,232 +0,0 @@
|
|||||||
//! Framebuffer
|
|
||||||
|
|
||||||
use super::mailbox;
|
|
||||||
use alloc::string::String;
|
|
||||||
use core::fmt;
|
|
||||||
use lazy_static::lazy_static;
|
|
||||||
use log::*;
|
|
||||||
use once::*;
|
|
||||||
use spin::Mutex;
|
|
||||||
|
|
||||||
/// Framebuffer information
|
|
||||||
#[repr(C)]
|
|
||||||
#[derive(Debug, Clone, Copy)]
|
|
||||||
pub struct FramebufferInfo {
|
|
||||||
/// visible width
|
|
||||||
pub xres: u32,
|
|
||||||
/// visible height
|
|
||||||
pub yres: u32,
|
|
||||||
/// virtual width
|
|
||||||
pub xres_virtual: u32,
|
|
||||||
/// virtual height
|
|
||||||
pub yres_virtual: u32,
|
|
||||||
/// virtual offset x
|
|
||||||
pub xoffset: u32,
|
|
||||||
/// virtual offset y
|
|
||||||
pub yoffset: u32,
|
|
||||||
|
|
||||||
/// bits per pixel
|
|
||||||
pub depth: u32,
|
|
||||||
/// bytes per line
|
|
||||||
pub pitch: u32,
|
|
||||||
|
|
||||||
/// bus address, starts from 0xC0000000/0x40000000
|
|
||||||
/// (see https://github.com/raspberrypi/firmware/wiki/Accessing-mailboxes)
|
|
||||||
pub bus_addr: u32,
|
|
||||||
/// screen buffer size
|
|
||||||
pub screen_size: u32,
|
|
||||||
}
|
|
||||||
|
|
||||||
#[repr(u32)]
|
|
||||||
#[derive(Debug, Clone, Copy)]
|
|
||||||
pub enum ColorDepth {
|
|
||||||
ColorDepth16 = 16,
|
|
||||||
ColorDepth32 = 32,
|
|
||||||
}
|
|
||||||
use self::ColorDepth::*;
|
|
||||||
|
|
||||||
#[repr(C)]
|
|
||||||
union ColorBuffer {
|
|
||||||
base_addr: usize,
|
|
||||||
buf16: &'static mut [u16],
|
|
||||||
buf32: &'static mut [u32],
|
|
||||||
}
|
|
||||||
|
|
||||||
impl ColorBuffer {
|
|
||||||
fn new(color_depth: ColorDepth, base_addr: usize, size: usize) -> ColorBuffer {
|
|
||||||
unsafe {
|
|
||||||
match color_depth {
|
|
||||||
ColorDepth16 => ColorBuffer {
|
|
||||||
buf16: core::slice::from_raw_parts_mut(base_addr as *mut u16, size / 2),
|
|
||||||
},
|
|
||||||
ColorDepth32 => ColorBuffer {
|
|
||||||
buf32: core::slice::from_raw_parts_mut(base_addr as *mut u32, size / 4),
|
|
||||||
},
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
#[inline]
|
|
||||||
fn read16(&self, index: u32) -> u16 {
|
|
||||||
unsafe { self.buf16[index as usize] }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[inline]
|
|
||||||
fn read32(&self, index: u32) -> u32 {
|
|
||||||
unsafe { self.buf32[index as usize] }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[inline]
|
|
||||||
fn write16(&mut self, index: u32, pixel: u16) {
|
|
||||||
unsafe { self.buf16[index as usize] = pixel }
|
|
||||||
}
|
|
||||||
|
|
||||||
#[inline]
|
|
||||||
fn write32(&mut self, index: u32, pixel: u32) {
|
|
||||||
unsafe { self.buf32[index as usize] = pixel }
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Frambuffer structure
|
|
||||||
pub struct Framebuffer {
|
|
||||||
pub fb_info: FramebufferInfo,
|
|
||||||
pub color_depth: ColorDepth,
|
|
||||||
buf: ColorBuffer,
|
|
||||||
}
|
|
||||||
|
|
||||||
impl fmt::Debug for Framebuffer {
|
|
||||||
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
|
|
||||||
let mut f = f.debug_struct("Framebuffer");
|
|
||||||
f.field("fb_info", &self.fb_info);
|
|
||||||
f.field("color_depth", &self.color_depth);
|
|
||||||
f.field("base_addr", unsafe { &self.buf.base_addr });
|
|
||||||
f.finish()
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
impl Framebuffer {
|
|
||||||
fn new(width: u32, height: u32, depth: u32) -> Result<Framebuffer, String> {
|
|
||||||
assert_has_not_been_called!("Framebuffer::new must be called only once");
|
|
||||||
|
|
||||||
let (width, height) = if width == 0 || height == 0 {
|
|
||||||
mailbox::framebuffer_get_physical_size()?
|
|
||||||
} else {
|
|
||||||
(width, height)
|
|
||||||
};
|
|
||||||
let depth = if depth == 0 {
|
|
||||||
mailbox::framebuffer_get_depth()?
|
|
||||||
} else {
|
|
||||||
depth
|
|
||||||
};
|
|
||||||
|
|
||||||
let info = mailbox::framebuffer_alloc(width, height, depth)?;
|
|
||||||
let color_depth = match info.depth {
|
|
||||||
16 => ColorDepth16,
|
|
||||||
32 => ColorDepth32,
|
|
||||||
_ => Err(format!("unsupported color depth {}", info.depth))?,
|
|
||||||
};
|
|
||||||
|
|
||||||
if info.bus_addr == 0 || info.screen_size == 0 {
|
|
||||||
Err(format!("mailbox call returned an invalid address/size"))?;
|
|
||||||
}
|
|
||||||
if info.pitch == 0 || info.pitch != info.xres * info.depth / 8 {
|
|
||||||
Err(format!(
|
|
||||||
"mailbox call returned an invalid pitch value {}",
|
|
||||||
info.pitch
|
|
||||||
))?;
|
|
||||||
}
|
|
||||||
|
|
||||||
use crate::arch::memory;
|
|
||||||
let paddr = info.bus_addr & !0xC0000000;
|
|
||||||
let vaddr = memory::ioremap(paddr as usize, info.screen_size as usize, "fb");
|
|
||||||
if vaddr == 0 {
|
|
||||||
Err(format!(
|
|
||||||
"cannot remap memory range [{:#x?}..{:#x?}]",
|
|
||||||
paddr,
|
|
||||||
paddr + info.screen_size
|
|
||||||
))?;
|
|
||||||
}
|
|
||||||
Ok(Framebuffer {
|
|
||||||
buf: ColorBuffer::new(color_depth, vaddr, info.screen_size as usize),
|
|
||||||
color_depth,
|
|
||||||
fb_info: info,
|
|
||||||
})
|
|
||||||
}
|
|
||||||
|
|
||||||
#[inline]
|
|
||||||
pub fn base_addr(&self) -> usize {
|
|
||||||
unsafe { self.buf.base_addr }
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Read pixel at `(x, y)`.
|
|
||||||
#[inline]
|
|
||||||
pub fn read(&self, x: u32, y: u32) -> u32 {
|
|
||||||
match self.color_depth {
|
|
||||||
ColorDepth16 => self.buf.read16(y * self.fb_info.xres + x) as u32,
|
|
||||||
ColorDepth32 => self.buf.read32(y * self.fb_info.xres + x),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Write pixel at `(x, y)`.
|
|
||||||
#[inline]
|
|
||||||
pub fn write(&mut self, x: u32, y: u32, pixel: u32) {
|
|
||||||
match self.color_depth {
|
|
||||||
ColorDepth16 => self.buf.write16(y * self.fb_info.xres + x, pixel as u16),
|
|
||||||
ColorDepth32 => self.buf.write32(y * self.fb_info.xres + x, pixel),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Copy buffer `[src_off .. src_off + size]` to `[dst_off .. dst_off + size]`.
|
|
||||||
/// `dst_off`, `src_off` and `size` must be aligned with `usize`.
|
|
||||||
pub fn copy(&mut self, dst_off: usize, src_off: usize, size: usize) {
|
|
||||||
const USIZE: usize = core::mem::size_of::<usize>();
|
|
||||||
let mut dst = self.base_addr() + dst_off;
|
|
||||||
let mut src = self.base_addr() + src_off;
|
|
||||||
let src_end = src + size;
|
|
||||||
while src < src_end {
|
|
||||||
unsafe { *(dst as *mut usize) = *(src as *mut usize) }
|
|
||||||
dst += USIZE;
|
|
||||||
src += USIZE;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Fill buffer `[offset .. offset + size]` with `pixel`.
|
|
||||||
/// `offset` and `size` must be aligned with `usize`.
|
|
||||||
pub fn fill(&mut self, offset: usize, size: usize, pixel: u32) {
|
|
||||||
const USIZE: usize = core::mem::size_of::<usize>();
|
|
||||||
let mut value: usize = 0;
|
|
||||||
let repeat = USIZE * 8 / self.fb_info.depth as usize;
|
|
||||||
let mask = ((1u64 << self.fb_info.depth) - 1) as usize;
|
|
||||||
for _i in 0..repeat {
|
|
||||||
value <<= self.fb_info.depth;
|
|
||||||
value += pixel as usize & mask;
|
|
||||||
}
|
|
||||||
|
|
||||||
let mut start = self.base_addr() + offset;
|
|
||||||
let end = start + size;
|
|
||||||
while start < end {
|
|
||||||
unsafe { *(start as *mut usize) = value }
|
|
||||||
start += USIZE;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Fill the entire buffer with `0`.
|
|
||||||
pub fn clear(&mut self) {
|
|
||||||
self.fill(0, self.fb_info.screen_size as usize, 0);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
lazy_static! {
|
|
||||||
pub static ref FRAME_BUFFER: Mutex<Option<Framebuffer>> = Mutex::new(None);
|
|
||||||
}
|
|
||||||
|
|
||||||
/// Initialize framebuffer
|
|
||||||
pub fn init() {
|
|
||||||
match Framebuffer::new(0, 0, 0) {
|
|
||||||
Ok(fb) => {
|
|
||||||
info!("framebuffer: init end\n{:#x?}", fb);
|
|
||||||
*FRAME_BUFFER.lock() = Some(fb);
|
|
||||||
}
|
|
||||||
Err(err) => warn!("framebuffer init failed: {}", err),
|
|
||||||
}
|
|
||||||
}
|
|
||||||
Loading…
Reference in new issue