#ifndef _SCREEN_CAPTURER_WGC_H_
#define _SCREEN_CAPTURER_WGC_H_

#include <atomic>
#include <functional>
#include <mutex>
#include <string>
#include <thread>
#include <vector>

#include "screen_capturer.h"
#include "wgc_session.h"
#include "wgc_session_impl.h"

namespace crossdesk {

class ScreenCapturerWgc : public ScreenCapturer,
                          public WgcSession::wgc_session_observer {
 public:
  ScreenCapturerWgc();
  ~ScreenCapturerWgc();

 public:
  bool IsWgcSupported();

  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;

  std::vector<DisplayInfo> GetDisplayInfoList() { return display_info_list_; }

  int SwitchTo(int monitor_index);
  int ResetToInitialMonitor() override;

  void OnFrame(const WgcSession::wgc_session_frame& frame, int id);

 protected:
  void CleanUp();
  int RebuildSessions(int preferred_monitor_index);
  int StartSessions(bool show_cursor);

 private:
  HMONITOR monitor_;
  MONITORINFOEX monitor_info_;
  std::vector<DisplayInfo> display_info_list_;
  int monitor_index_ = 0;
  int initial_monitor_index_ = 0;

 private:
  class WgcSessionInfo {
   public:
    std::unique_ptr<WgcSession> session_;
    bool inited_ = false;
    bool running_ = false;
    bool paused_ = false;
  };

  std::vector<WgcSessionInfo> sessions_;

  std::atomic_bool running_{false};
  std::atomic_bool inited_{false};

  int fps_ = 60;

  cb_desktop_data on_data_ = nullptr;

  unsigned char* nv12_frame_ = nullptr;
  unsigned char* nv12_frame_scaled_ = nullptr;
  int nv12_width_ = 0;
  int nv12_height_ = 0;

  std::mutex frame_mutex_;
};
}  // namespace crossdesk
#endif
