mirror of
https://github.com/kunkundi/crossdesk.git
synced 2026-04-03 08:05:16 +08:00
100 lines
2.4 KiB
C++
100 lines
2.4 KiB
C++
/*
|
|
* @Author: DI JUNKUN
|
|
* @Date: 2026-03-22
|
|
* Copyright (c) 2026 by DI JUNKUN, All Rights Reserved.
|
|
*/
|
|
|
|
#ifndef _SCREEN_CAPTURER_WAYLAND_H_
|
|
#define _SCREEN_CAPTURER_WAYLAND_H_
|
|
|
|
struct DBusConnection;
|
|
struct pw_context;
|
|
struct pw_core;
|
|
struct pw_stream;
|
|
struct pw_thread_loop;
|
|
|
|
#include <atomic>
|
|
#include <cstdint>
|
|
#include <string>
|
|
#include <thread>
|
|
#include <vector>
|
|
|
|
#include "screen_capturer.h"
|
|
|
|
namespace crossdesk {
|
|
|
|
class ScreenCapturerWayland : public ScreenCapturer {
|
|
public:
|
|
ScreenCapturerWayland();
|
|
~ScreenCapturerWayland();
|
|
|
|
public:
|
|
int Init(const int fps, cb_desktop_data cb) override;
|
|
int Destroy() override;
|
|
int Start(bool show_cursor) override;
|
|
int Stop() override;
|
|
|
|
int Pause(int monitor_index) override;
|
|
int Resume(int monitor_index) override;
|
|
|
|
int SwitchTo(int monitor_index) override;
|
|
int ResetToInitialMonitor() override;
|
|
|
|
std::vector<DisplayInfo> GetDisplayInfoList() override;
|
|
|
|
private:
|
|
bool CheckPortalAvailability() const;
|
|
bool ConnectSessionBus();
|
|
bool CreatePortalSession();
|
|
bool SelectPortalSource();
|
|
bool StartPortalSession();
|
|
bool OpenPipeWireRemote();
|
|
bool SetupPipeWireStream();
|
|
|
|
void Run();
|
|
void CleanupPipeWire();
|
|
void CleanupDbus();
|
|
void ClosePortalSession();
|
|
void HandlePipeWireBuffer();
|
|
void UpdateDisplayGeometry(int width, int height);
|
|
|
|
private:
|
|
static constexpr int kFallbackWidth = 1920;
|
|
static constexpr int kFallbackHeight = 1080;
|
|
|
|
std::thread thread_;
|
|
std::atomic<bool> running_{false};
|
|
std::atomic<bool> paused_{false};
|
|
std::atomic<int> monitor_index_{0};
|
|
int initial_monitor_index_ = 0;
|
|
std::atomic<bool> show_cursor_{true};
|
|
int fps_ = 60;
|
|
cb_desktop_data callback_ = nullptr;
|
|
std::vector<DisplayInfo> display_info_list_;
|
|
|
|
DBusConnection* dbus_connection_ = nullptr;
|
|
std::string session_handle_;
|
|
std::string display_name_ = "WAYLAND0";
|
|
uint32_t pipewire_node_id_ = 0;
|
|
int pipewire_fd_ = -1;
|
|
|
|
pw_thread_loop* pw_thread_loop_ = nullptr;
|
|
pw_context* pw_context_ = nullptr;
|
|
pw_core* pw_core_ = nullptr;
|
|
pw_stream* pw_stream_ = nullptr;
|
|
void* stream_listener_ = nullptr;
|
|
bool pipewire_initialized_ = false;
|
|
bool pipewire_thread_loop_started_ = false;
|
|
uint32_t spa_video_format_ = 0;
|
|
int frame_width_ = 0;
|
|
int frame_height_ = 0;
|
|
int frame_stride_ = 0;
|
|
|
|
std::vector<uint8_t> y_plane_;
|
|
std::vector<uint8_t> uv_plane_;
|
|
};
|
|
|
|
} // namespace crossdesk
|
|
|
|
#endif
|