feat(server): benchmark the remotefx tile encoder

Signed-off-by: Marc-André Lureau <marcandre.lureau@redhat.com>

Author: elmarco

Cargo.lock

@@ -135,15 +135,12 @@ pub fn to_64x64_ycbcr_tile(
     height: usize,
     stride: usize,
     format: PixelFormat,
-    y: &mut [i16],
-    cb: &mut [i16],
-    cr: &mut [i16],
+    y: &mut [i16; 64 * 64],
+    cb: &mut [i16; 64 * 64],
+    cr: &mut [i16; 64 * 64],
 ) {
     assert!(width <= 64);
     assert!(height <= 64);
-    assert_eq!(y.len(), 64 * 64);
-    assert_eq!(cb.len(), 64 * 64);
-    assert_eq!(cr.len(), 64 * 64);
 
     let to_rgb = pixel_format_to_rgb_fn(format);
     let bpp = format.bytes_per_pixel() as usize;

crates/ironrdp-graphics/src/color_conversion.rs

@@ -17,6 +17,7 @@ test = false
 
 [features]
 helper = ["dep:x509-cert", "dep:rustls-pemfile"]
+bench = []
 
 [dependencies]
 anyhow = "1.0"
@@ -41,6 +42,13 @@ rustls-pemfile = { version = "2.2.0", optional = true }
 
 [dev-dependencies]
 tokio = { version = "1", features = ["sync"] }
+criterion = "0.5"
+
+[[bench]]
+name = "bench"
+path = "benches/bench.rs"
+required-features = ["bench"]
+harness = false
 
 [lints]
 workspace = true

crates/ironrdp-server/Cargo.toml

@@ -0,0 +1,30 @@
+use std::num::NonZero;
+
+use criterion::{criterion_group, criterion_main, Criterion};
+use ironrdp_pdu::codecs::rfx::{EntropyAlgorithm, Quant};
+use ironrdp_server::{bench::encoder::*, BitmapUpdate};
+
+pub fn criterion_benchmark(c: &mut Criterion) {
+    let quant = Quant::default();
+    let algo = EntropyAlgorithm::Rlgr3;
+    let bitmap = BitmapUpdate {
+        top: 0,
+        left: 0,
+        width: NonZero::new(64).unwrap(),
+        height: NonZero::new(64).unwrap(),
+        format: ironrdp_server::PixelFormat::ARgb32,
+        data: vec![0; 64 * 64 * 4],
+        order: ironrdp_server::PixelOrder::BottomToTop,
+        stride: 64 * 4,
+    };
+    let enc = rfx::UpdateEncoder::new(&bitmap, quant, algo);
+    let mut buf = vec![0; 64 * 64 * 3];
+    c.bench_function("rfxenc", |b| {
+        b.iter(|| {
+            let _ = enc.encode(0, 0, &mut buf);
+        })
+    });
+}
+
+criterion_group!(benches, criterion_benchmark);
+criterion_main!(benches);

crates/ironrdp-server/benches/bench.rs

@@ -1,5 +1,7 @@
-pub(crate) mod bitmap;
-pub(crate) mod rfx;
+#![allow(unreachable_pub)]
+
+pub mod bitmap;
+pub mod rfx;
 
 use std::{cmp, mem};