enlightenment/src/modules/comp/e_mod_comp_wl.c

#include "e.h"
#include "e_mod_main.h"
#ifdef HAVE_WAYLAND_CLIENTS
# include <xcb/xcb_image.h>
# include "e_mod_comp_wl.h"
# include "e_mod_comp_wl_comp.h"
# include "e_mod_comp_wl_output.h"
# include "e_mod_comp_wl_input.h"
# include "e_mod_comp_wl_shell.h"
#endif

/* local function prototypes */
static Eina_Bool _e_mod_comp_wl_fd_handle(void *data, Ecore_Fd_Handler *hdl __UNUSED__);

/* private variables */
static Ecore_Fd_Handler *_wl_fd_handler = NULL;

/* extern variables */
struct wl_display *_wl_disp;

Eina_Bool 
e_mod_comp_wl_init(void)
{
   struct wl_event_loop *loop;
   int fd = 0;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* init wayland display */
   if (!(_wl_disp = wl_display_create()))
     {
        EINA_LOG_ERR("Failed to create wayland display\n");
        return EINA_FALSE;
     }

   if (wl_display_add_socket(_wl_disp, NULL))
     {
        wl_display_terminate(_wl_disp);
        EINA_LOG_ERR("Failed to add socket to wayland display\n");
        return EINA_FALSE;
     }

   /* init a wayland compositor ?? */
   if (!e_mod_comp_wl_comp_init())
     {
        wl_display_terminate(_wl_disp);
        EINA_LOG_ERR("Failed to create wayland compositor\n");
        return EINA_FALSE;
     }

   /* init output */
   if (!e_mod_comp_wl_output_init())
     {
        e_mod_comp_wl_comp_shutdown();
        wl_display_terminate(_wl_disp);
        EINA_LOG_ERR("Failed to create wayland output\n");
        return EINA_FALSE;
     }

   /* init input */
   if (!e_mod_comp_wl_input_init())
     {
        e_mod_comp_wl_output_shutdown();
        e_mod_comp_wl_comp_shutdown();
        wl_display_terminate(_wl_disp);
        EINA_LOG_ERR("Failed to create wayland input\n");
        return EINA_FALSE;
     }

   /* init a wayland shell */
   if (!e_mod_comp_wl_shell_init())
     {
        e_mod_comp_wl_input_shutdown();
        e_mod_comp_wl_output_shutdown();
        e_mod_comp_wl_comp_shutdown();
        wl_display_terminate(_wl_disp);
        EINA_LOG_ERR("Failed to create wayland shell\n");
        return EINA_FALSE;
     }

   loop = wl_display_get_event_loop(_wl_disp);
   fd = wl_event_loop_get_fd(loop);

   _wl_fd_handler = 
     ecore_main_fd_handler_add(fd, ECORE_FD_READ, 
                               _e_mod_comp_wl_fd_handle, NULL, NULL, NULL);

   wl_event_loop_dispatch(loop, 0);

   return EINA_TRUE;
}

void 
e_mod_comp_wl_shutdown(void)
{
   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   if (_wl_fd_handler)
     ecore_main_fd_handler_del(_wl_fd_handler);
   _wl_fd_handler = NULL;

   e_mod_comp_wl_shell_shutdown();
   e_mod_comp_wl_input_shutdown();
   e_mod_comp_wl_output_shutdown();
   e_mod_comp_wl_comp_shutdown();

   if (_wl_disp) wl_display_terminate(_wl_disp);
   _wl_disp = NULL;
}

uint32_t 
e_mod_comp_wl_time_get(void)
{
   struct timeval tv;

   gettimeofday(&tv, NULL);
   return tv.tv_sec * 1000 + tv.tv_usec / 1000;
}

Ecore_X_Pixmap 
e_mod_comp_wl_pixmap_get(Ecore_X_Window win)
{
   Wayland_Compositor *comp;
   Wayland_Surface *ws;
//   struct wl_list *list;
   Ecore_X_Pixmap pmap = 0;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   comp = e_mod_comp_wl_comp_get();
   if (wl_list_empty(&comp->surfaces)) return 0;

//   list = &comp->surfaces;
   wl_list_for_each(ws, &comp->surfaces, link)
     {
        if (!ws->buffer) continue;
        if (((ws->win) && (ws->win->border)) 
            && (ws->win->border->win == win))
          {
             Ecore_X_Connection *conn;
             Ecore_X_GC gc;
             uint8_t *pix = 0;
             int depth;

             if (ws->buffer)
               {
                  if (wl_buffer_is_shm(ws->buffer))
                    pix = (uint8_t *)wl_shm_buffer_get_data(ws->buffer);
                  else
                    {
                       if (ws->texture) pix = (uint8_t *)ws->texture;
                       else if (ws->saved_texture) 
                         pix = (uint8_t *)ws->saved_texture;
                    }
               }
             else if (ws->image)
               {
                  if (ws->texture) pix = (uint8_t *)ws->texture;
                  else if (ws->saved_texture) 
                    pix = (uint8_t *)ws->saved_texture;
               }

             if (!pix) return 0;

             depth = ecore_x_window_depth_get(win);
             conn = ecore_x_connection_get();

             pmap = xcb_generate_id(conn);
             xcb_create_pixmap(conn, depth, pmap, win, ws->w, ws->h);

             gc = ecore_x_gc_new(pmap, 0, NULL);
             xcb_put_image(conn, 2, pmap, gc, ws->w, ws->h, 
                           0, 0, 0, depth, 
                           (ws->w * ws->h * sizeof(int)), pix);
             ecore_x_gc_free(gc);
          }
     }

   return pmap;
}

/* local functions */
static Eina_Bool 
_e_mod_comp_wl_fd_handle(void *data __UNUSED__, Ecore_Fd_Handler *hdl __UNUSED__)
{
   struct wl_event_loop *loop;

   loop = wl_display_get_event_loop(_wl_disp);
   wl_event_loop_dispatch(loop, 0);
   return ECORE_CALLBACK_RENEW;
}