Source code

Revision control

Copy as Markdown

Other Tools

extern crate khronos_egl as egl;
use egl::API as egl;
use gl::types::{GLboolean, GLchar, GLenum, GLint, GLuint, GLvoid};
use std::ffi::CStr;
use std::ptr;
use std::sync::atomic::{AtomicBool, Ordering};
use std::sync::Arc;
use wayland_client::{
protocol::{wl_compositor::WlCompositor, wl_surface::WlSurface},
DispatchData, Display, EventQueue, Main,
};
use wayland_protocols::xdg_shell::client::{
xdg_surface::{self, XdgSurface},
xdg_wm_base::{self, XdgWmBase},
};
fn process_xdg_event(xdg: Main<XdgWmBase>, event: xdg_wm_base::Event, _dd: DispatchData) {
use xdg_wm_base::Event::*;
match event {
Ping { serial } => xdg.pong(serial),
_ => (),
}
}
struct DisplayConnection {
display: Display,
event_queue: EventQueue,
compositor: Main<WlCompositor>,
xdg: Main<XdgWmBase>,
}
fn setup_wayland() -> DisplayConnection {
let display =
wayland_client::Display::connect_to_env().expect("unable to connect to the wayland server");
let mut event_queue = display.create_event_queue();
let attached_display = display.clone().attach(event_queue.token());
let globals = wayland_client::GlobalManager::new(&attached_display);
// Roundtrip to retrieve the globals list
event_queue
.sync_roundtrip(&mut (), |_, _, _| unreachable!())
.unwrap();
// Get the compositor.
let compositor: Main<WlCompositor> = globals.instantiate_exact(1).unwrap();
// Xdg protocol.
let xdg: Main<XdgWmBase> = globals.instantiate_exact(1).unwrap();
xdg.quick_assign(process_xdg_event);
DisplayConnection {
display,
event_queue,
compositor,
xdg,
}
}
fn setup_egl(display: &Display) -> egl::Display {
let egl_display = unsafe {
egl.get_display(display.get_display_ptr() as *mut std::ffi::c_void)
.unwrap()
};
egl.initialize(egl_display).unwrap();
egl_display
}
fn create_context(display: egl::Display) -> (egl::Context, egl::Config) {
let attributes = [
egl::RED_SIZE,
8,
egl::GREEN_SIZE,
8,
egl::BLUE_SIZE,
8,
egl::NONE,
];
let config = egl
.choose_first_config(display, &attributes)
.expect("unable to choose an EGL configuration")
.expect("no EGL configuration found");
let context_attributes = [
egl::CONTEXT_MAJOR_VERSION,
4,
egl::CONTEXT_MINOR_VERSION,
0,
egl::CONTEXT_OPENGL_PROFILE_MASK,
egl::CONTEXT_OPENGL_CORE_PROFILE_BIT,
egl::NONE,
];
let context = egl
.create_context(display, config, None, &context_attributes)
.expect("unable to create an EGL context");
(context, config)
}
struct Surface {
handle: Main<WlSurface>,
initialized: AtomicBool,
}
fn create_surface(
ctx: &DisplayConnection,
egl_display: egl::Display,
egl_context: egl::Context,
egl_config: egl::Config,
width: i32,
height: i32,
) -> Arc<Surface> {
let wl_surface = ctx.compositor.create_surface();
let xdg_surface = ctx.xdg.get_xdg_surface(&wl_surface);
let xdg_toplevel = xdg_surface.get_toplevel();
xdg_toplevel.set_app_id("khronos-egl-test".to_string());
xdg_toplevel.set_title("Test".to_string());
wl_surface.commit();
ctx.display.flush().unwrap();
let surface = Arc::new(Surface {
handle: wl_surface,
initialized: AtomicBool::new(false),
});
let weak_surface = Arc::downgrade(&surface);
xdg_surface.quick_assign(
move |xdg_surface: Main<XdgSurface>, event: xdg_surface::Event, _dd: DispatchData| {
use xdg_surface::Event::*;
match event {
Configure { serial } => {
if let Some(surface) = weak_surface.upgrade() {
if !surface.initialized.swap(true, Ordering::Relaxed) {
let wl_egl_surface =
wayland_egl::WlEglSurface::new(&surface.handle, width, height);
let egl_surface = unsafe {
egl.create_window_surface(
egl_display,
egl_config,
wl_egl_surface.ptr() as egl::NativeWindowType,
None,
)
.expect("unable to create an EGL surface")
};
egl.make_current(
egl_display,
Some(egl_surface),
Some(egl_surface),
Some(egl_context),
)
.expect("unable to bind the context");
render();
egl.swap_buffers(egl_display, egl_surface)
.expect("unable to post the surface content");
xdg_surface.ack_configure(serial);
}
}
}
_ => (),
}
},
);
surface
}
fn main() {
// Setup Open GL.
egl.bind_api(egl::OPENGL_API)
.expect("unable to select OpenGL API");
gl::load_with(|name| egl.get_proc_address(name).unwrap() as *const std::ffi::c_void);
// Setup the Wayland client.
let mut ctx = setup_wayland();
// Setup EGL.
let egl_display = setup_egl(&ctx.display);
let (egl_context, egl_config) = create_context(egl_display);
// Create a surface.
// Note that it must be kept alive to the end of execution.
let _surface = create_surface(&ctx, egl_display, egl_context, egl_config, 800, 600);
loop {
ctx.event_queue
.dispatch(&mut (), |_, _, _| { /* we ignore unfiltered messages */ })
.unwrap();
}
}
const VERTEX: &'static [GLint; 8] = &[-1, -1, 1, -1, 1, 1, -1, 1];
const INDEXES: &'static [GLuint; 4] = &[0, 1, 2, 3];
const VERTEX_SHADER: &[u8] = b"#version 400
in vec2 position;
void main() {
gl_Position = vec4(position, 0.0f, 1.0f);
}
\0";
const FRAGMENT_SHADER: &[u8] = b"#version 400
out vec4 color;
void main() {
color = vec4(1.0f, 0.0f, 0.0f, 1.0f);
}
\0";
fn render() {
unsafe {
let vertex_shader = gl::CreateShader(gl::VERTEX_SHADER);
check_gl_errors();
let src = CStr::from_bytes_with_nul_unchecked(VERTEX_SHADER).as_ptr();
gl::ShaderSource(vertex_shader, 1, (&[src]).as_ptr(), ptr::null());
check_gl_errors();
gl::CompileShader(vertex_shader);
check_shader_status(vertex_shader);
let fragment_shader = gl::CreateShader(gl::FRAGMENT_SHADER);
check_gl_errors();
let src = CStr::from_bytes_with_nul_unchecked(FRAGMENT_SHADER).as_ptr();
gl::ShaderSource(fragment_shader, 1, (&[src]).as_ptr(), ptr::null());
check_gl_errors();
gl::CompileShader(fragment_shader);
check_shader_status(fragment_shader);
let program = gl::CreateProgram();
check_gl_errors();
gl::AttachShader(program, vertex_shader);
check_gl_errors();
gl::AttachShader(program, fragment_shader);
check_gl_errors();
gl::LinkProgram(program);
check_gl_errors();
gl::UseProgram(program);
check_gl_errors();
let mut buffer = 0;
gl::GenBuffers(1, &mut buffer);
check_gl_errors();
gl::BindBuffer(gl::ARRAY_BUFFER, buffer);
check_gl_errors();
gl::BufferData(
gl::ARRAY_BUFFER,
8 * 4,
VERTEX.as_ptr() as *const std::ffi::c_void,
gl::STATIC_DRAW,
);
check_gl_errors();
let mut vertex_input = 0;
gl::GenVertexArrays(1, &mut vertex_input);
check_gl_errors();
gl::BindVertexArray(vertex_input);
check_gl_errors();
gl::EnableVertexAttribArray(0);
check_gl_errors();
gl::VertexAttribPointer(0, 2, gl::INT, gl::FALSE as GLboolean, 0, 0 as *const GLvoid);
check_gl_errors();
let mut indexes = 0;
gl::GenBuffers(1, &mut indexes);
check_gl_errors();
gl::BindBuffer(gl::ELEMENT_ARRAY_BUFFER, indexes);
check_gl_errors();
gl::BufferData(
gl::ELEMENT_ARRAY_BUFFER,
4 * 4,
INDEXES.as_ptr() as *const std::ffi::c_void,
gl::STATIC_DRAW,
);
check_gl_errors();
gl::DrawElements(gl::TRIANGLE_FAN, 4, gl::UNSIGNED_INT, std::ptr::null());
check_gl_errors();
}
}
fn format_error(e: GLenum) -> &'static str {
match e {
gl::NO_ERROR => "No error",
gl::INVALID_ENUM => "Invalid enum",
gl::INVALID_VALUE => "Invalid value",
gl::INVALID_OPERATION => "Invalid operation",
gl::INVALID_FRAMEBUFFER_OPERATION => "Invalid framebuffer operation",
gl::OUT_OF_MEMORY => "Out of memory",
gl::STACK_UNDERFLOW => "Stack underflow",
gl::STACK_OVERFLOW => "Stack overflow",
_ => "Unknown error",
}
}
pub fn check_gl_errors() {
unsafe {
match gl::GetError() {
gl::NO_ERROR => (),
e => {
panic!("OpenGL error: {}", format_error(e))
}
}
}
}
unsafe fn check_shader_status(shader: GLuint) {
let mut status = gl::FALSE as GLint;
gl::GetShaderiv(shader, gl::COMPILE_STATUS, &mut status);
if status != (gl::TRUE as GLint) {
let mut len = 0;
gl::GetProgramiv(shader, gl::INFO_LOG_LENGTH, &mut len);
if len > 0 {
let mut buf = Vec::with_capacity(len as usize);
buf.set_len((len as usize) - 1); // subtract 1 to skip the trailing null character
gl::GetProgramInfoLog(
shader,
len,
ptr::null_mut(),
buf.as_mut_ptr() as *mut GLchar,
);
let log = String::from_utf8(buf).unwrap();
eprintln!("shader compilation log:\n{}", log);
}
panic!("shader compilation failed");
}
}