#include "e.h"
#include "e_mod_main.h"
#include "e_smart_randr.h"
#include "e_smart_monitor.h"

/*
 * TODO:
 * 
 * Add Poller for Output Change events to listen for hotplug (4 seconds)
 * 
 */

/* local structures */
typedef struct _E_Smart_Data E_Smart_Data;
struct _E_Smart_Data
{
   /* base object */
   Evas_Object *o_base;

   /* grid object */
   Evas_Object *o_grid;

   /* virtual size */
   Evas_Coord vw, vh;

   /* visible flag */
   Eina_Bool visible : 1;

   /* list of monitors */
   Eina_List *monitors;
};

/* local function prototypes */
static void _e_smart_add(Evas_Object *obj);
static void _e_smart_del(Evas_Object *obj);
static void _e_smart_move(Evas_Object *obj, Evas_Coord x, Evas_Coord y);
static void _e_smart_resize(Evas_Object *obj, Evas_Coord w, Evas_Coord h);
static void _e_smart_show(Evas_Object *obj);
static void _e_smart_hide(Evas_Object *obj);
static void _e_smart_clip_set(Evas_Object *obj, Evas_Object *clip);
static void _e_smart_clip_unset(Evas_Object *obj);

static void _e_smart_randr_grid_cb_move(void *data, Evas *evas EINA_UNUSED, Evas_Object *obj EINA_UNUSED, void *event EINA_UNUSED);
static void _e_smart_randr_grid_cb_resize(void *data, Evas *evas EINA_UNUSED, Evas_Object *obj EINA_UNUSED, void *event EINA_UNUSED);

static Evas_Object *_e_smart_randr_monitor_crtc_find(E_Smart_Data *sd, Ecore_X_Randr_Crtc crtc);
static void _e_smart_randr_monitor_cb_changed(void *data, Evas_Object *obj EINA_UNUSED, void *event EINA_UNUSED);
static void _e_smart_randr_monitor_cb_moved(void *data, Evas_Object *obj, void *event EINA_UNUSED);

/* external functions exposed by this widget */
Evas_Object *
e_smart_randr_add(Evas *evas)
{
   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   static Evas_Smart *smart = NULL;
   static const Evas_Smart_Class sc = 
     {
        "smart_randr", EVAS_SMART_CLASS_VERSION, 
        _e_smart_add, _e_smart_del, _e_smart_move, _e_smart_resize, 
        _e_smart_show, _e_smart_hide, NULL, 
        _e_smart_clip_set, _e_smart_clip_unset, 
        NULL, NULL, NULL, NULL, NULL, NULL, NULL
     };

   /* if we have never created the smart class, do it now */
   if (!smart)
     if (!(smart = evas_smart_class_new(&sc)))
       return NULL;

   /* return a newly created smart randr widget */
   return evas_object_smart_add(evas, smart);
}

void 
e_smart_randr_virtual_size_calc(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Ecore_X_Window root = 0;
   Ecore_X_Randr_Crtc *crtcs;
   Evas_Coord vw = 0, vh = 0;
   int ncrtcs = 0;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* NB: The old code here used to get the modes from the e_randr_cfg.
    * I changed it to get directly from Xrandr because of attempts to 
    * run this in Xephyr. Getting the information from e_randr_cfg was not 
    * practical in those cases */

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* grab the root window */
   root = ecore_x_window_root_first_get();

   /* try to get the list of crtcs */
   if ((crtcs = ecore_x_randr_crtcs_get(root, &ncrtcs)))
     {
        int i = 0;
        Ecore_X_Randr_Output *routputs;
        Eina_List *outputs = NULL;

        /* loop the list of crtcs and try to get the outputs on each */
        for (i = 0; i < ncrtcs; i++)
          {
             int noutput = 0, j = 0;
             intptr_t *o;

             routputs = 
               ecore_x_randr_crtc_outputs_get(root, crtcs[i], &noutput);
             if ((noutput == 0) || (!routputs))
               {
                  /* find Possible outputs */
                  routputs = 
                    ecore_x_randr_crtc_possible_outputs_get(root, crtcs[i], 
                                                            &noutput);
                  if ((!noutput) || (!routputs)) continue;
               }

             for (j = 0; j < noutput; j++)
               {
                  Ecore_X_Randr_Crtc rcrtc;

                  rcrtc = ecore_x_randr_output_crtc_get(root, routputs[j]);
                  if ((rcrtc != 0) && (rcrtc != crtcs[i]))
                    continue;

                  outputs = 
                    eina_list_append(outputs, (intptr_t *)(long)routputs[j]);
               }

             /* loop the outputs and get the largest mode */
             EINA_LIST_FREE(outputs, o)
               {
                  Ecore_X_Randr_Output output;
                  Ecore_X_Randr_Mode *modes;
                  Evas_Coord mw = 0, mh = 0;
                  int nmode = 0;

                  output = (int)(long)o;

                  /* try to get the list of modes for this output */
                  modes = 
                    ecore_x_randr_output_modes_get(root, output, 
                                                   &nmode, NULL);
                  if (!modes) continue;

                  /* get the size of the largest mode */
                  ecore_x_randr_mode_size_get(root, modes[0], &mw, &mh);

                  vw += MAX(mw, mh);
                  vh += MAX(mw, mh);

                  /* free any allocated memory from ecore_x_randr */
                  free(modes);
               }

             /* free any allocated memory from ecore_x_randr */
             free(routputs);
          }

        /* free any allocated memory from ecore_x_randr */
        free(crtcs);
     }

   if ((vw == 0) && (vh == 0))
     {
        /* by default, set virtual size to the current screen size */
        ecore_x_randr_screen_current_size_get(root, &vw, &vh, NULL, NULL);
     }

   sd->vw = vw;
   sd->vh = vh;

   /* set the grid size */
   evas_object_grid_size_set(sd->o_grid, vw, vh);
}

void 
e_smart_randr_monitors_create(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Evas *evas;
   Ecore_X_Window root = 0;
   Ecore_X_Randr_Crtc *crtcs;
   int ncrtcs = 0;
   Evas_Coord gx = 0, gy = 0, gw = 0, gh = 0;
   Evas_Coord nx = 0;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* NB: The old code here used to get the outputs from the e_randr_cfg.
    * I changed it to get directly from Xrandr because of attempts to 
    * run this in Xephyr. Getting the information from e_randr_cfg was not 
    * practical in those cases */

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* grab the canvas of the grid object */
   evas = evas_object_evas_get(sd->o_grid);

   /* get the geometry of the grid */
   evas_object_geometry_get(sd->o_grid, &gx, &gy, &gw, &gh);

   /* grab the root window */
   root = ecore_x_window_root_first_get();

   /* try to get the list of crtcs */
   if ((crtcs = ecore_x_randr_crtcs_get(root, &ncrtcs)))
     {
        int i = 0;
        Ecore_X_Randr_Output *routputs;
        Eina_List *outputs = NULL;
        Evas_Coord cx = 0, cy = 0, cw = 0, ch = 0;

        /* loop the list of crtcs and try to get the outputs on each */
        for (i = 0; i < ncrtcs; i++)
          {
             int noutput = 0, j = 0;
             intptr_t *o;

             printf("Checking Crtc: %d\n", crtcs[i]);

             /* TODO: Add EFL 1.8 check here and optimize this code */

             /* get the geometry for this crtc */
             ecore_x_randr_crtc_geometry_get(root, crtcs[i], 
                                             &cx, &cy, &cw, &ch);
             printf("\tGeometry: %d %d %d %d\n", cx, cy, cw, ch);

             routputs = 
               ecore_x_randr_crtc_outputs_get(root, crtcs[i], &noutput);
             printf("\t\tNum Of Outputs: %d\n", noutput);

             if ((noutput == 0) || (!routputs))
               {
                  /* find Possible outputs */
                  routputs = 
                    ecore_x_randr_crtc_possible_outputs_get(root, crtcs[i], 
                                                            &noutput);
                  /* printf("\t\tNum Of Possible: %d\n", noutput); */
                  if ((!noutput) || (!routputs)) continue;
               }

             /* loop possible outputs */
             for (j = 0; j < noutput; j++)
               {
                  Ecore_X_Randr_Crtc rcrtc;

                  /* get this outputs crtc */
                  rcrtc = ecore_x_randr_output_crtc_get(root, routputs[j]);

                  /* if crtc is not 0 (disabled), then check if it matches 
                   * the one we are currently working with. If it does not 
                   * match, then this is not a possible output for us */
                  if ((rcrtc != 0) && (rcrtc != crtcs[i]))
                    continue;

                  outputs = 
                    eina_list_append(outputs, (intptr_t *)(long)routputs[j]);
               }

             /* loop the outputs and create monitors for each */
             EINA_LIST_FREE(outputs, o)
               {
                  Evas_Object *mon;
                  Ecore_X_Randr_Output output;

                  /* for each output, try to create a monitor */
                  if (!(mon = e_smart_monitor_add(evas)))
                    continue;

                  /* hook into monitor changed callback */
                  evas_object_smart_callback_add(mon, "monitor_changed", 
                                                 _e_smart_randr_monitor_cb_changed, obj);
                  evas_object_smart_callback_add(mon, "monitor_moved", 
                                                 _e_smart_randr_monitor_cb_moved, obj);

                  /* add this monitor to our list */
                  sd->monitors = eina_list_append(sd->monitors, mon);

                  /* tell monitor what the grid's virtual size is */
                  e_smart_monitor_grid_virtual_size_set(mon, sd->vw, sd->vh);

                  /* tell monitor what the grid is and it's geometry */
                  e_smart_monitor_grid_set(mon, sd->o_grid, gx, gy, gw, gh);

                  output = (int)(long)o;

                  printf("\t\t\tOutput %d Crtc Is: %d\n", output, 
                         ecore_x_randr_output_crtc_get(root, output));

                  if ((cw == 0) && (ch == 0))
                    {
                       Ecore_X_Randr_Mode *modes;
                       Evas_Coord mw = 0, mh = 0;
                       int nmode = 0;

                       /* try to get the list of modes for this output */
                       modes = 
                         ecore_x_randr_output_modes_get(root, output, 
                                                        &nmode, NULL);
                       if (!modes) continue;

                       /* get the size of the largest mode */
                       ecore_x_randr_mode_size_get(root, modes[0], &mw, &mh);

                       /* free any allocated memory from ecore_x_randr */
                       free(modes);

                       /* tell monitor what it's current position is
                        * NB: This also packs into the grid */
                       e_smart_monitor_current_geometry_set(mon, nx, 0, mw, mh);

                       nx += mw;
                    }
                  else
                    {
                       /* tell monitor what it's current position is
                        * NB: This also packs into the grid */
                       e_smart_monitor_current_geometry_set(mon, cx, cy, cw, ch);

                       nx += cw;
                    }

                  /* tell monitor what crtc it uses and current position */
                  e_smart_monitor_crtc_set(mon, crtcs[i], cx, cy, cw, ch);

                  /* tell monitor what output it uses */
                  e_smart_monitor_output_set(mon, output);

                  /* tell monitor to set the background preview */
                  e_smart_monitor_background_set(mon, cx, cy);

                  /* if we have a previous crtc */
                  if (i > 0)
                    {
                       /* check if this crtc is a clone of that one */
                       Ecore_X_Randr_Crtc pcrtc = 0;
                       Evas_Coord px = 0, py = 0;

                       pcrtc = crtcs[i - 1];

                       /* get the geometry for this previous crtc */
                       ecore_x_randr_crtc_geometry_get(root, pcrtc, 
                                                       &px, &py, NULL, NULL);

                       /* check if the geometry's match */
                       if ((px == cx) && (py == cy))
                         {
                            Ecore_X_Randr_Orientation orient, porient;

                            orient = 
                              ecore_x_randr_crtc_orientation_get(root, crtcs[i]);

                            porient = 
                              ecore_x_randr_crtc_orientation_get(root, pcrtc);

                            /* check orientation */
                            if ((porient == orient))
                              {
                                 Ecore_X_Randr_Mode mode = 0, pmode = 0;

                                 mode = 
                                   ecore_x_randr_crtc_mode_get(root, crtcs[i]);
                                 pmode = 
                                   ecore_x_randr_crtc_mode_get(root, pcrtc);

                                 /* check mode */
                                 if ((pmode == mode))
                                   {
                                      Evas_Object *pmon;

                                      /* find monitor for this prev crtc */
                                      if ((pmon = _e_smart_randr_monitor_crtc_find(sd, pcrtc)))
                                        {
                                           /* tell This monitor it is cloned into previous */
                                           e_smart_monitor_clone_set(mon, pmon);
                                        }
                                   }
                              }
                         }
                    }
               }

             /* free any allocated memory from ecore_x_randr */
             free(routputs);
          }

        /* free any allocated memory from ecore_x_randr */
        free(crtcs);
     }
}

void 
e_smart_randr_min_size_get(Evas_Object *obj, Evas_Coord *mw, Evas_Coord *mh)
{
   E_Smart_Data *sd;

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   if (mw) *mw = (sd->vw / 10);
   if (mh) *mh = (sd->vh / 10);
}

Eina_Bool 
e_smart_randr_changed_get(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Eina_Bool changed = EINA_FALSE;
   Eina_List *l = NULL;
   Evas_Object *mon;

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return EINA_FALSE;

   EINA_LIST_FOREACH(sd->monitors, l, mon)
     {
        E_Smart_Monitor_Changes changes = E_SMART_MONITOR_CHANGED_NONE;

        changes = e_smart_monitor_changes_get(mon);
        if (changes > E_SMART_MONITOR_CHANGED_NONE)
          {
             changed = EINA_TRUE;
             break;
          }
     }

   return changed;
}

void 
e_smart_randr_changes_apply(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Eina_List *l = NULL;
   Evas_Object *mon;
   Eina_Bool need_reset = EINA_FALSE;

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* tell each monitor to apply it's changes */
   EINA_LIST_FOREACH(sd->monitors, l, mon)
     if (e_smart_monitor_changes_apply(mon))
       need_reset = EINA_TRUE;

   if (need_reset) 
     ecore_x_randr_screen_reset(ecore_x_window_root_first_get());
}

/* local functions */
static void 
_e_smart_add(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Evas *evas;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to allocate the smart data structure */
   if (!(sd = E_NEW(E_Smart_Data, 1))) return;

   /* grab the canvas */
   evas = evas_object_evas_get(obj);

   sd->o_base = edje_object_add(evas);
   e_theme_edje_object_set(sd->o_base, "base/theme/widgets", 
                           "e/conf/randr/main");
   evas_object_smart_member_add(sd->o_base, obj);

   /* create the virtual grid */
   sd->o_grid = evas_object_grid_add(evas);
   edje_object_part_swallow(sd->o_base, "e.swallow.content", sd->o_grid);

   /* setup grid move callback */
   evas_object_event_callback_add(sd->o_grid, EVAS_CALLBACK_MOVE, 
                                  _e_smart_randr_grid_cb_move, sd);
   evas_object_event_callback_add(sd->o_grid, EVAS_CALLBACK_RESIZE, 
                                  _e_smart_randr_grid_cb_resize, sd);

   /* set the object's smart data */
   evas_object_smart_data_set(obj, sd);
}

static void 
_e_smart_del(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Evas_Object *mon;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* free the monitors */
   EINA_LIST_FREE(sd->monitors, mon)
     {
        evas_object_smart_callback_del(mon, "monitor_changed", 
                                       _e_smart_randr_monitor_cb_changed);
        evas_object_smart_callback_del(mon, "monitor_moved", 
                                       _e_smart_randr_monitor_cb_moved);
        evas_object_del(mon);
     }

   /* remove grid move callback */
   evas_object_event_callback_del(sd->o_grid, EVAS_CALLBACK_MOVE, 
                                  _e_smart_randr_grid_cb_move);
   evas_object_event_callback_del(sd->o_grid, EVAS_CALLBACK_RESIZE, 
                                  _e_smart_randr_grid_cb_resize);

   /* delete the grid object */
   evas_object_del(sd->o_grid);

   /* delete the base object */
   evas_object_del(sd->o_base);

   /* try to free the allocated structure */
   E_FREE(sd);

   /* set the objects smart data to null */
   evas_object_smart_data_set(obj, NULL);
}

static void 
_e_smart_move(Evas_Object *obj, Evas_Coord x, Evas_Coord y)
{
   E_Smart_Data *sd;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* move the base object */
   evas_object_move(sd->o_base, x, y);
}

static void 
_e_smart_resize(Evas_Object *obj, Evas_Coord w, Evas_Coord h)
{
   E_Smart_Data *sd;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* resize the base object */
   evas_object_resize(sd->o_base, w, h);
}

static void 
_e_smart_show(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Eina_List *l = NULL;
   Evas_Object *mon;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* if it is already visible, get out */
   if (sd->visible) return;

   /* show the base object */
   evas_object_show(sd->o_base);

   /* show any monitors */
   EINA_LIST_FOREACH(sd->monitors, l, mon)
     evas_object_show(mon);

   /* set visibility flag */
   sd->visible = EINA_TRUE;
}

static void 
_e_smart_hide(Evas_Object *obj)
{
   E_Smart_Data *sd;
   Eina_List *l = NULL;
   Evas_Object *mon;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* if it is not visible, we have nothing to do */
   if (!sd->visible) return;

   /* hide any monitors */
   EINA_LIST_FOREACH(sd->monitors, l, mon)
     evas_object_hide(mon);

   /* hide the base object */
   evas_object_hide(sd->o_base);

   /* set visibility flag */
   sd->visible = EINA_FALSE;
}

static void 
_e_smart_clip_set(Evas_Object *obj, Evas_Object *clip)
{
   E_Smart_Data *sd;

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* set the clip */
   evas_object_clip_set(sd->o_base, clip);
}

static void 
_e_smart_clip_unset(Evas_Object *obj)
{
   E_Smart_Data *sd;

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(obj))) return;

   /* unset the clip */
   evas_object_clip_unset(sd->o_base);
}

static void 
_e_smart_randr_grid_cb_move(void *data, Evas *evas EINA_UNUSED, Evas_Object *obj EINA_UNUSED, void *event EINA_UNUSED)
{
   E_Smart_Data *sd;
   Evas_Coord gx = 0, gy = 0, gw = 0, gh = 0;
   Eina_List *l = NULL;
   Evas_Object *mon;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to get the smart data */
   if (!(sd = data)) return;

   /* get the grid geometry */
   evas_object_geometry_get(sd->o_grid, &gx, &gy, &gw, &gh);

   /* loop the monitors and update grid geometry */
   EINA_LIST_FOREACH(sd->monitors, l, mon)
     e_smart_monitor_grid_set(mon, sd->o_grid, gx, gy, gw, gh);
}

static void 
_e_smart_randr_grid_cb_resize(void *data, Evas *evas EINA_UNUSED, Evas_Object *obj EINA_UNUSED, void *event EINA_UNUSED)
{
   E_Smart_Data *sd;
   Evas_Coord gx = 0, gy = 0, gw = 0, gh = 0;
   Eina_List *l = NULL;
   Evas_Object *mon;

   LOGFN(__FILE__, __LINE__, __FUNCTION__);

   /* try to get the smart data */
   if (!(sd = data)) return;

   /* get the grid geometry */
   evas_object_geometry_get(sd->o_grid, &gx, &gy, &gw, &gh);

   /* loop the monitors and update grid geometry */
   EINA_LIST_FOREACH(sd->monitors, l, mon)
     e_smart_monitor_grid_set(mon, sd->o_grid, gx, gy, gw, gh);
}

static Evas_Object *
_e_smart_randr_monitor_crtc_find(E_Smart_Data *sd, Ecore_X_Randr_Crtc crtc)
{
   Eina_List *l = NULL;
   Evas_Object *mon;

   if ((!sd) || (!crtc)) return NULL;

   EINA_LIST_FOREACH(sd->monitors, l, mon)
     {
        Ecore_X_Randr_Crtc mcrtc;

        if ((mcrtc = e_smart_monitor_crtc_get(mon)))
          if ((mcrtc == crtc)) return mon;
     }

   return NULL;
}

static void 
_e_smart_randr_monitor_cb_changed(void *data, Evas_Object *obj EINA_UNUSED, void *event EINA_UNUSED)
{
   Evas_Object *randr;

   if (!(randr = data)) return;

   /* tell main dialog that something changed and to enable apply button */
   evas_object_smart_callback_call(randr, "randr_changed", NULL);
}

static void 
_e_smart_randr_monitor_cb_moved(void *data, Evas_Object *obj, void *event EINA_UNUSED)
{
   E_Smart_Data *sd;
   Evas_Object *randr, *mon;
   Eina_List *l = NULL;
   Evas_Coord minx = 0, miny = 0;

   if (!(randr = data)) return;

   /* try to get the objects smart data */
   if (!(sd = evas_object_smart_data_get(randr))) return;

   /* ONE MONITOR MUST BE AT 0 X and ONE MUST BE AT 0 Y */

   /* normalize output positions so that upper left corner of all 
    * outputs is at 0,0 */

   minx = 32768;
   miny = 32768;

   EINA_LIST_FOREACH(sd->monitors, l, mon)
     {
        Evas_Coord mx = 0, my = 0;

        /* get the geometry for this monitor */
        e_smart_monitor_current_geometry_get(mon, &mx, &my, NULL, NULL);
        if (mx < minx) minx = mx;
        if (my < miny) miny = my;
     }

   if ((minx) || (miny))
     {
        EINA_LIST_FOREACH(sd->monitors, l, mon)
          {
             Evas_Coord mx = 0, my = 0, mw = 0, mh = 0;

             /* get the geometry for this monitor */
             e_smart_monitor_current_geometry_get(mon, &mx, &my, &mw, &mh);

             mx -= minx;
             my -= miny;

             /* move monitor to new position */
             e_smart_monitor_current_geometry_set(mon, mx, my, mw, mh);
          }
     }

   /* tell main dialog that something changed and to enable apply button */
   evas_object_smart_callback_call(randr, "randr_changed", NULL);
}