1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34

use std::result;

use windows::Win32::Graphics::Dxgi::{DXGI_OUTDUPL_FRAME_INFO, DXGI_OUTPUT_DESC};

pub trait OutputDescExt {
  fn width(&self) -> u32;
  fn height(&self) -> u32;
}

impl OutputDescExt for DXGI_OUTPUT_DESC {
  fn width(&self) -> u32 {
    (self.DesktopCoordinates.right - self.DesktopCoordinates.left) as u32
  }
  fn height(&self) -> u32 {
    (self.DesktopCoordinates.bottom - self.DesktopCoordinates.top) as u32
  }
}

pub type Result<T> = result::Result<T, &'static str>;

/// Return needed buffer size, in bytes.
pub fn calc_buffer_size(desc: DXGI_OUTPUT_DESC) -> usize {
  (desc.width() * desc.height() * 4) as usize // 4 for BGRA32
}

pub trait FrameInfoExt {
  fn is_new_frame(&self) -> bool;
}

impl FrameInfoExt for DXGI_OUTDUPL_FRAME_INFO {
  fn is_new_frame(&self) -> bool {
    self.LastPresentTime > 0
  }
}