-- mrg ui for image viewer mode of gegl ui

-- todo : implement thumb-bar of previous/next image

local path = GObject.Object(STATE).path
local source = GObject.Object(STATE).source

mrg:set_title(ffi.string(o.path))

cr:new_path()
cr:rectangle(0,0,mrg:width(),mrg:height())

function controls_keepalive()
  o.show_controls = 1
  hide_controls_id = mrg:add_timeout(2000, function()
    hide_controls_id = nil
    o.show_controls = 0
    mrg:queue_draw(nil)
    return 0
  end)
end

mrg:listen(Mrg.MOTION, function(event)
  if hide_controls_id ~= nil then
     mrg:remove_idle (hide_controls_id)
  end
  controls_keepalive()
end)

if not path:find(".lui$") then

mrg:listen(Mrg.DRAG, function(event)

  if zoom_pinch_coord == nil then
    zoom_pinch_coord = {{},{},{},{}}
  end
  controls_keepalive()

  if event.type == Mrg.DRAG_PRESS then

    if event.device_no == 5 then -- 5 = second finger
      zoom_pinch_coord[2].x = event.x;
      zoom_pinch_coord[2].y = event.y;

      zoom_pinch_coord[3].x = zoom_pinch_coord[1].x;
      zoom_pinch_coord[3].y = zoom_pinch_coord[1].y;
      zoom_pinch_coord[4].x = zoom_pinch_coord[2].x;
      zoom_pinch_coord[4].y = zoom_pinch_coord[2].y;

      zoom_pinch = 1
      orig_zoom = o.scale
    elseif event.device_no == 1 or event.device_no == 4 then -- mouse or first finger
      zoom_pinch_coord[1].x = event.x;
      zoom_pinch_coord[1].y = event.y;

    end
  elseif event.type == Mrg.DRAG_MOTION then
    if event.device_no == 1 or event.device_no == 4 then
      zoom_pinch_coord[1].x = event.x;
      zoom_pinch_coord[1].y = event.y;
    end
    if event.device_no == 5 then
      zoom_pinch_coord[2].x = event.x;
      zoom_pinch_coord[2].y = event.y;
    end

    if zoom_pinch ~= nil then
      local orig_dist = ffi.C.hypotf (zoom_pinch_coord[3].x - zoom_pinch_coord[2].x ,
                                      zoom_pinch_coord[3].y - zoom_pinch_coord[2].y );
      local dist = ffi.C.hypotf (zoom_pinch_coord[1].x - zoom_pinch_coord[2].x ,
                                 zoom_pinch_coord[1].y - zoom_pinch_coord[2].y);
      local x, y;
      local screen_cx = (zoom_pinch_coord[1].x + zoom_pinch_coord[2].x)/2;
      local screen_cy = (zoom_pinch_coord[1].y + zoom_pinch_coord[2].y)/2;

      local x = (o.u + screen_cx) / o.scale
      local y = (o.v + screen_cy) / o.scale

      o.scale = orig_zoom * dist / orig_dist;
      o.u = x * o.scale - screen_cx;
      o.v = y * o.scale - screen_cy;

      o.u = o.u - (event.delta_x )/2
      o.v = o.v - (event.delta_y )/2
      o.is_fit = 0;
    else
      o.u = o.u - event.delta_x
      o.v = o.v - event.delta_y
    end
  elseif event.type == Mrg.DRAG_RELEASE then
    zoom_pinch = nil
  end
  event:stop_propagate()
end)
end

mrg:listen(Mrg.SCROLL, function(event)
  local factor = 1.05
  local x = (o.u + event.device_x) / o.scale
  local y = (o.v + event.device_y) / o.scale

  if event.scroll_direction == Mrg.SCROLL_DIRECTION_DOWN then
    o.scale = o.scale * factor
  elseif event.scroll_direction == Mrg.SCROLL_DIRECTION_UP then
    o.scale = o.scale / factor
  end

  o.u = x * o.scale - event.device_x
  o.v = y * o.scale - event.device_y
  o.is_fit = 0
  mrg:queue_draw(nil)
end
)

if o.is_video ~= 0 then
  mrg:listen(Mrg.PRESS, function(event)
    if o.playing ~= 0 then
      o.playing = 0
    else
      o.playing = 1
    end
  end)
end


function draw_grid(x,y,w,h)
  cr:new_path()
  cr:rectangle (0.00 *w + x, 0.00 * h + y, 0.33 * w, 0.33 * h)
  cr:rectangle (0.66 *w + x, 0.00 * h + y, 0.33 * w, 0.33 * h)
  cr:rectangle (0.00 *w + x, 0.66 * h + y, 0.33 * w, 0.33 * h)
  cr:rectangle (0.66 *w + x, 0.66 * h + y, 0.33 * w, 0.33 * h)
end

function draw_back(x,y,w,h)
  cr:new_path ();
  cr:new_path ();
  cr:move_to (x+0.9*w, y+0.1*h);
  cr:line_to (x+0.9*w, y+0.9*h);
  cr:line_to (x+0.1*w, y+0.5*h);
end

function draw_up (x,y,w,h)
  cr:new_path ();
  cr:move_to (x+0.1*w, y+0.9*h);
  cr:line_to (x+0.5*w, y+0.1*h);
  cr:line_to (x+0.9*w, y+0.9*h);
end

function draw_down (x,y,w,h)
  cr:new_path ();
  cr:move_to (x+0.1*w, y+0.1*h);
  cr:line_to (x+0.5*w, y+0.9*h);
  cr:line_to (x+0.9*w, y+0.1*h);
end

function draw_forward(x,y,w,h)
  cr:new_path ();
  cr:move_to (x+0.1*w, y+0.1*h);
  cr:line_to (x+0.1*w, y+0.9*h);
  cr:line_to (x+0.9*w, y+0.5*h);
end

function draw_pause (x,y,w,h)
  local margin = w * .1
  local bar = w * .3
  local gap = w * .16
  cr:new_path ();
  cr:rectangle (x + margin, y, bar, h)
  cr:rectangle (x + margin + bar + gap, y, bar, h)
end


function draw_edit(x,y,w,h)
  cr:new_path ();
  cr:arc (x+0.5*w, y+0.5*h, h * .4, 0.0, 3.1415 * 2);
end


-- ui_viewer

cr:save()
cr:rectangle(0,0,mrg:width(),mrg:height())

local width = mrg:width()
local height = mrg:height()

draw_grid(height* 0.1/4, height * 0.1/4, height * 0.1, height *0.1)

if o.show_controls ~= 0 then
  contrasty_stroke()
else
  cr:new_path()
end

cr:rectangle (0, 0, height * 0.15, height * 0.15);

if o.show_controls ~= 0 then
  cr:set_source_rgba (1,1,1,.1);
  cr:fill_preserve ();
end

mrg:listen(Mrg.PRESS,
  function(event)
    ffi.C.argvs_eval('parent')
    event:stop_propagate()
  end)

draw_back (height * .1 / 4, height * .5, height * .1, height *.1);
cr:close_path ();

if o.show_controls ~= 0 then
  contrasty_stroke ();
else
  cr:new_path ();
end

cr:rectangle (0, height * .3, height * .15, height *.7);

if o.show_controls ~= 0 then
  cr:set_source_rgba (1,1,1,.1);
  cr:fill_preserve ();
end

mrg:listen(Mrg.PRESS,
  function(event)
    ffi.C.argvs_eval('prev')
    event:stop_propagate()
  end)

cr:new_path ();

draw_forward (width - height * .12, height * .5, height * .1, height *.1);
cr:close_path ();

if o.show_controls ~= 0 then
  contrasty_stroke ();
else
  cr:new_path ();
end

cr:rectangle (width - height * .15, height * .3, height * .15, height *.7);

if o.show_controls ~= 0 then
  cr:set_source_rgba (1,1,1,.1);
  cr:fill_preserve ();
end

mrg:listen(Mrg.PRESS,
  function(event)
    ffi.C.argvs_eval('next')
    event:stop_propagate()
  end)

draw_edit (width - height * .15, height * .0, height * .15, height *.15);
if o.show_controls ~= 0 then
  contrasty_stroke ();
else
  cr:new_path ();
end

cr:rectangle (width - height * .15, height * .0, height * .15, height *.15);
if o.show_controls ~= 0 then
 cr:set_source_rgba (1,1,1,.1);
 cr:fill_preserve ();
end

mrg:listen(Mrg.PRESS,
  function(event)
    ffi.C.argvs_eval('toggle editing')
    event:stop_propagate()
  end)

cr:new_path ();

if o.is_video ~= 0 then

  local source = GObject.Object(STATE).source
  local frame = source:get_property('frame').value
  local frames = source:get_property('frames').value
  if o.show_controls ~= 0 then
    cr:set_line_width(2)
    cr:new_path()
    cr:move_to(height * .15, height * .95);
    cr:line_to(width - height * .15, height * .95);
    cr:set_source_rgba(1,1,1,1)

    cr:stroke()
    cr:move_to(height * .15 + frame * (width - height * .3)  / frames, height * 0.975)
    cr:rel_line_to(0, -0.05 * height)
    cr:stroke()

    if o.playing ~= 0 then
      draw_pause(0.17 * height, height * .86, height/20, height/20)
    else
      draw_forward(0.17 *height, height * .86, height/20, height/20)
    end

    cr:set_source_rgba(1,1,1,.5)
    cr:fill()



  end

  cr:rectangle(height * .1 , height * .9, width - height * .2, height *.1)
  mrg:listen(Mrg.DRAG, function(event)
    source:set_property('frame',
       GObject.Value(GObject.Type.INT,
        (event.x - height *.15) / (width - height * .3) * frames
    ))
    event:stop_propagate()
  end)
  cr:new_path()
end


if 1 ~= 0 then

  local frame = o.pos;
  local frames = o.duration;

  local source = GObject.Object(STATE).sink

  if o.show_controls ~= 0 then
    cr:set_line_width(2)
    cr:new_path()
    cr:move_to(height * .15, height * .95);
    cr:line_to(width - height * .15, height * .95);
    cr:set_source_rgba(1,1,1,1)

    cr:stroke()
    cr:move_to(height * .15 + frame * (width - height * .3)  / frames, height * 0.975)
    cr:rel_line_to(0, -0.05 * height)
    cr:stroke()

    cr:set_source_rgba(1,1,1,.5)
    cr:fill()


  end

  cr:rectangle(height * .1 , height * .9, width - height * .2, height *.1)
  mrg:listen(Mrg.DRAG, function(event)
      o:set_clip_position((event.x - height *.15) / (width - height * .3) * frames)
    -- the following commented out lua is deprecated by above, the below
    -- lua does not do the correct time to fps quantization that set_clip_position does

    --o.pos = (event.x - height *.15) / (width - height * .3) * frames

    --sink:set_time(o.pos + o.start)
    --if o.is_video ~= 0 then
    --  source = GObject.Object(STATE).source
    --   local frames = source:get_property('frames').value
    --   local pos = (event.x - height *.15) / (width - height * .3)
    --source:set_property('frame',
    --   GObject.Value(GObject.Type.INT,
    --        math.floor(pos * frames)
    --))
    --end
      event:stop_propagate()
  end)
  cr:new_path()
end


function pdf_next_page()
  local pages = source:get_property('pages').value
  local page = source:get_property('page').value
  source:set_property('page', GObject.Value(GObject.Type.INT, page + 1))
end

function pdf_prev_page()
  local pages = source:get_property('pages').value
  local page = source:get_property('page').value
  source:set_property('page', GObject.Value(GObject.Type.INT, page - 1))
end

function draw_thumb_bar()
  local cr=mrg:cr()
  cr:set_source_rgba(1,1,1,.1)
  cr:rectangle(0, mrg:height()*0.8, mrg:width(), mrg:height()*0.2)
  mrg:listen(Mrg.MOTION, function(e)
   -- print('a')
  end)
  cr:fill()
  -- mrg:print("thumbbar" .. o:item_no() .. '' .. o:item_path())
end

mrg:add_binding("page-up", NULL, "previous image",
  function() ffi.C.argvs_eval ("prev") end)

mrg:add_binding("page-down", NULL, "next image",
  function() ffi.C.argvs_eval ("next") end)

if path:find(".pdf$") or path:find('.PDF$') then
  local pages = source:get_property('pages').value
  local page = source:get_property('page').value

  mrg:set_style("background:transparent;color:white")
  mrg:set_xy(width - height * .12, height - height * .15)
  if o.show_controls ~= 0 then
    mrg:print(' ' .. page .. '/' .. pages)
  end

  draw_up (width - height * .12, height * .71, height * .1, height *.1);
  cr:close_path ();
  mrg:listen(Mrg.PRESS, function(event)
    pdf_prev_page()
    event:stop_propagate()
  end)
  if o.show_controls ~= 0 then
    contrasty_stroke ()
  else
    cr:new_path()
  end

  draw_down (width - height * .12, height * .87, height * .1, height *.1);
  cr:close_path ();
  mrg:listen(Mrg.PRESS, function(event)
    pdf_next_page()
    event:stop_propagate()
  end)
  if o.show_controls ~= 0 then
    contrasty_stroke ()
  else
    cr:new_path()
  end

  -- override default page-up / page-down bindings
  mrg:add_binding("page-up", NULL, "previous page",
    function(e) pdf_prev_page() e:stop_propagate() end)

  mrg:add_binding("page-down", NULL, "next page",
    function(e) pdf_next_page() e:stop_propagate() end)

end

--draw_thumb_bar()

mrg:add_binding("alt-right", NULL, "next image",
  function() ffi.C.argvs_eval ("next") end)

mrg:add_binding("alt-left", NULL, "rev image",
  function() ffi.C.argvs_eval ("prev") end)


mrg:add_binding("control-m", NULL, "toggle mipmap",
  function() ffi.C.argvs_eval ("toggle mipmap") end)

mrg:add_binding("control-y", NULL, "toggle display profile",
  function() ffi.C.argvs_eval ("toggle colormanaged-display") end)

mrg:add_binding("control-delete", NULL, "discard",
  function() ffi.C.argvs_eval ("discard") end)

if o.commandline[0] == 0 then
  mrg:add_binding("space", NULL, "toggle playing",
    function() ffi.C.argvs_eval ("toggle playing") end)

  mrg:add_binding("1", NULL, "give one star",
    function() ffi.C.argvs_eval ("star 1") end)
  mrg:add_binding("2", NULL, "give two stars",
    function() ffi.C.argvs_eval ("star 2") end)
  mrg:add_binding("3", NULL, "give three stars",
    function() ffi.C.argvs_eval ("star 3") end)
  mrg:add_binding("4", NULL, "give four stars",
    function() ffi.C.argvs_eval ("star 4") end)
  mrg:add_binding("5", NULL, "give five stars",
    function() ffi.C.argvs_eval ("star 5") end)
  mrg:add_binding("0", NULL, "give no stars",
    function() ffi.C.argvs_eval ("star 0") end)
end

if o.show_controls ~= 0 then

  local stars = o:get_key_int(o.path, "stars")

  if stars < 0 then stars = 0 end

  --if stars >= 0 then
  if true then
    mrg:start('div.viewerstars')
    mrg:set_xy(mrg:height()*.15 + 2,mrg:em())
    for i=1,stars do
      mrg:text_listen(Mrg.PRESS, function(e)
          ffi.C.argvs_eval('star ' .. i)
          e:stop_propagate()
      end)
      mrg:print('★')
      mrg:text_listen_done()
    end
    mrg:set_style('color:gray;')
    for i=stars+1,5 do
      mrg:text_listen(Mrg.PRESS, function(e)
          ffi.C.argvs_eval('star ' .. i)
          e:stop_propagate()
      end)
      mrg:print('★')
      mrg:text_listen_done()
    end
    mrg:close()
  end
end

cr:restore()

--if o.playing ~= 0 then
 --  print 'o'
--end