devs/foxbox
1
0
Fork 0
forked from len0rd/rockbox
Code Pull requests Activity

lua add rlimg.lua example script split large includes to separate files

Browse source 
Change-Id: I67cac5bc4ce5525ab30abf9443f6cc1a33190512
This commit is contained in:
William Wilgus 2019-08-04 09:48:09 -05:00
parent 4209c09770
commit f85df30e3e
8 changed files with 1464 additions and 393 deletions
Download patch file Download diff file Expand all files Collapse all files

216
apps/plugins/lua/include_lua/draw.lua
View file

@ -29,7 +29,6 @@
_draw.ellipse_filled _draw.ellipse_filled
_draw.ellipse_rect_filled _draw.ellipse_rect_filled
_draw.ellipse_rect _draw.ellipse_rect
_draw.flood_fill
_draw.hline _draw.hline
_draw.image _draw.image
_draw.line _draw.line
@ -39,7 +38,6 @@
_draw.rect_filled _draw.rect_filled
_draw.rounded_rect _draw.rounded_rect
_draw.rounded_rect_filled _draw.rounded_rect_filled
_draw.text
_draw.vline _draw.vline
]] ]]
@ -56,21 +54,17 @@ local _draw = {} do
setmetatable(_draw, rocklib_image) setmetatable(_draw, rocklib_image)
-- Internal Constants -- Internal Constants
local _LCD = rb.lcd_framebuffer()
local LCD_W, LCD_H = rb.LCD_WIDTH, rb.LCD_HEIGHT
local BSAND = 8 -- blits color to dst if src <> 0 local BSAND = 8 -- blits color to dst if src <> 0
local _NIL = nil -- nil placeholder local _NIL = nil -- nil placeholder
local _abs = math.abs
local _clear = rocklib_image.clear local _clear = rocklib_image.clear
local _copy = rocklib_image.copy local _copy = rocklib_image.copy
local _ellipse = rocklib_image.ellipse local _ellipse = rocklib_image.ellipse
local _get = rocklib_image.get local _get = rocklib_image.get
local _line = rocklib_image.line local _line = rocklib_image.line
local _marshal = rocklib_image.marshal
local _min = math.min local _min = math.min
local _newimg = rb.new_image local _newimg = rb.new_image
local _points = rocklib_image.points
-- line -- line
_draw.line = function(img, x1, y1, x2, y2, color, bClip) _draw.line = function(img, x1, y1, x2, y2, color, bClip)
@ -87,33 +81,22 @@ local _draw = {} do
_line(img, x, y, _NIL, y + length, color, bClip) _line(img, x, y, _NIL, y + length, color, bClip)
end end
-- draws a non-filled figure based on points in t-points -- draws a non-filled rect based on points in t-points
local function polyline(img, x, y, t_points, color, bClosed, bClip) local function polyrect(img, x, y, t_points, color, bClip)
if #t_points < 2 then error("not enough points", 3) end local pt_first_last = t_points[1]
local pt1, pt2
local pt_first_last for i = 1, 4, 1 do
pt1 = t_points[i]
if bClosed then pt2 = t_points[i + 1] or pt_first_last-- first and last point
pt_first_last = t_points[1]
else
pt_first_last = t_points[#t_points]
end
for i = 1, #t_points, 1 do
local pt1 = t_points[i]
local pt2 = t_points[i + 1] or pt_first_last-- first and last point
_line(img, pt1[1] + x, pt1[2] + y, pt2[1] + x, pt2[2] + y, color, bClip) _line(img, pt1[1] + x, pt1[2] + y, pt2[1] + x, pt2[2] + y, color, bClip)
end end
end end
-- rectangle -- rectangle
local function rect(img, x, y, width, height, color, bClip) local function rect(img, x, y, width, height, color, bClip)
if width == 0 or height == 0 then return end if width == 0 or height == 0 then return end
polyline(img, x, y, {{0, 0}, {width, 0}, {width, height}, {0, height}}, color, true, bClip) polyrect(img, x, y, {{0, 0}, {width, 0}, {width, height}, {0, height}}, color, bClip)
end end
@ -240,190 +223,9 @@ local _draw = {} do
_copy(dst, src, x, y, 1, 1, _NIL, _NIL, bClip) _copy(dst, src, x, y, 1, 1, _NIL, _NIL, bClip)
end end
-- floods an area of targetclr with fillclr x, y specifies the start seed
_draw.flood_fill = function(img, x, y, targetclr, fillclr)
-- scanline 4-way flood algorithm
-- ^
-- <--------x--->
-- v
-- check that target color doesn't = fill and the first point is target color
if targetclr == fillclr or targetclr ~= _get(img, x, y, true) then return end
local max_w = img:width()
local max_h = img:height()
local qpt = {} -- FIFO queue
-- rather than moving elements around in our FIFO queue
-- for each read; increment 'qhead' by 2
-- set both elements to nil and let the
-- garbage collector worry about it
-- for each write; increment 'qtail' by 2
-- x coordinates are in odd indices while
-- y coordinates are in even indices
local qtail = 0
local function check_ns(val, x, y)
if targetclr == val then
y = y - 1
if targetclr == _get(img, x, y, true) then -- north
qtail = qtail + 2
qpt[qtail - 1] = x
qpt[qtail] = y
end
y = y + 2
if targetclr == _get(img, x, y, true) then -- south
qtail = qtail + 2
qpt[qtail - 1] = x
qpt[qtail] = y
end
return fillclr
end
return _NIL -- signal marshal to stop
end
local function seed_pt(x, y)
-- should never hit max but make sure not to end early
for qhead = 2, 0x40000000, 2 do
if targetclr == _get(img, x, y, true) then
_marshal(img, x, y, 1, y, _NIL, _NIL, true, check_ns) -- west
_marshal(img, x + 1, y, max_w, y, _NIL, _NIL, true, check_ns) -- east
end
x = qpt[qhead - 1]
qpt[qhead - 1] = _NIL
if not x then break end
y = qpt[qhead]
qpt[qhead] = _NIL
end
end
seed_pt(x, y) -- Begin
end -- flood_fill
-- draws a closed figure based on points in t_points
_draw.polygon = function(img, x, y, t_points, color, fillcolor, bClip)
if #t_points < 2 then error("not enough points", 3) end
if fillcolor then
local x_min, x_max = 0, 0
local y_min, y_max = 0, 0
local w, h = 0, 0
-- find boundries of polygon
for i = 1, #t_points, 1 do
local pt = t_points[i]
if pt[1] < x_min then x_min = pt[1] end
if pt[1] > x_max then x_max = pt[1] end
if pt[2] < y_min then y_min = pt[2] end
if pt[2] > y_max then y_max = pt[2] end
end
w = _abs(x_max) + _abs(x_min)
h = _abs(y_max) + _abs(y_min)
x_min = x_min - 2 -- leave a border to use flood_fill
y_min = y_min - 2
local fill_img = _newimg(w + 3, h + 3)
_clear(fill_img, 0x1)
for i = 1, #t_points, 1 do
local pt1 = t_points[i]
local pt2 = t_points[i + 1] or t_points[1]-- first and last point
_line(fill_img, pt1[1] - x_min, pt1[2] - y_min,
pt2[1]- x_min, pt2[2] - y_min, 0)
end
_draw.flood_fill(fill_img, fill_img:width(), fill_img:height() , 0x1, 0x0)
_copy(img, fill_img, x - 1, y - 1, _NIL, _NIL, _NIL, _NIL, bClip, BSAND, fillcolor)
end
polyline(img, x, y, t_points, color, true, bClip)
end
-- draw text onto image if width/height are supplied text is centered
_draw.text = function(img, x, y, width, height, font, color, text)
font = font or rb.FONT_UI
local opts = {x = 0, y = 0, width = LCD_W - 1, height = LCD_H - 1,
font = font, drawmode = 3, fg_pattern = 0x1, bg_pattern = 0}
if rb.LCD_DEPTH == 2 then -- invert 2-bit screens
--vp.drawmode = bit.bxor(vp.drawmode, 4)
opts.fg_pattern = 3 - opts.fg_pattern
opts.bg_pattern = 3 - opts.bg_pattern
end
rb.set_viewport(opts)
local res, w, h = rb.font_getstringsize(text, font)
if not width then
width = 0
else
width = (width - w) / 2
end
if not height then
height = 0
else
height = (height - h) / 2
end
-- make a copy of the current screen for later
local screen_img = _newimg(LCD_W, LCD_H)
_copy(screen_img, _LCD)
-- check if the screen buffer is supplied image if so set img to the copy
if img == _LCD then
img = screen_img
end
-- we will be printing the text to the screen then blitting into img
rb.lcd_clear_display()
if w > LCD_W then -- text is too long for the screen do it in chunks
local l = 1
local resp, wp, hp
local lenr = text:len()
while lenr > 1 do
l = lenr
resp, wp, hp = rb.font_getstringsize(text:sub(1, l), font)
while wp >= LCD_W and l > 1 do
l = l - 1
resp, wp, hp = rb.font_getstringsize(text:sub( 1, l), font)
end
rb.lcd_putsxy(0, 0, text:sub(1, l))
text = text:sub(l)
if x + width > img:width() or y + height > img:height() then
break
end
-- using the mask we made blit color into img
_copy(img, _LCD, x + width, y + height, _NIL, _NIL, _NIL, _NIL, false, BSAND, color)
x = x + wp
rb.lcd_clear_display()
lenr = text:len()
end
else --w <= LCD_W
rb.lcd_putsxy(0, 0, text)
-- using the mask we made blit color into img
_copy(img, _LCD, x + width, y + height, _NIL, _NIL, _NIL, _NIL, false, BSAND, color)
end
_copy(_LCD, screen_img) -- restore screen
rb.set_viewport() -- set viewport default
return res, w, h
end
-- expose internal functions to the outside through _draw table -- expose internal functions to the outside through _draw table
_draw.hline = hline _draw.hline = hline
_draw.vline = vline _draw.vline = vline
_draw.polyline = polyline
_draw.rect = rect _draw.rect = rect
_draw.rounded_rect = rounded_rect _draw.rounded_rect = rounded_rect
end -- _draw functions end -- _draw functions

95
apps/plugins/lua/include_lua/draw_floodfill.lua Normal file
View file

@ -0,0 +1,95 @@
--[[ Lua Floodfill function
/***************************************************************************
* __________ __ ___.
* Open \______ \ ____ ____ | | _\_ |__ _______ ___
* Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
* Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
* Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
* \/ \/ \/ \/ \/
* $Id$
*
* Copyright (C) 2017 William Wilgus
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
* KIND, either express or implied.
*
****************************************************************************/
]]
-- floods an area of targetclr with fillclr x, y specifies the start seed
-- flood_fill(img, x, y, targetclr, fillclr)
if not rb.lcd_framebuffer then rb.splash(rb.HZ, "No Support!") return nil end
do
local rocklib_image = getmetatable(rb.lcd_framebuffer())
local _NIL = nil -- nil placeholder
local _get = rocklib_image.get
local _line = rocklib_image.line
local _marshal = rocklib_image.marshal
return function(img, x, y, targetclr, fillclr)
-- scanline 4-way flood algorithm
-- ^
-- <--------x--->
-- v
-- check that target color doesn't = fill and the first point is target color
if targetclr == fillclr or targetclr ~= _get(img, x, y, true) then return end
local max_w = img:width()
local max_h = img:height()
local qpt = {} -- FIFO queue
-- rather than moving elements around in our FIFO queue
-- for each read; increment 'qhead' by 2
-- set both elements to nil and let the
-- garbage collector worry about it
-- for each write; increment 'qtail' by 2
-- x coordinates are in odd indices while
-- y coordinates are in even indices
local qtail = 0
local function check_ns(val, x, y)
if targetclr == val then
y = y - 1
if targetclr == _get(img, x, y, true) then -- north
qtail = qtail + 2
qpt[qtail - 1] = x
qpt[qtail] = y
end
y = y + 2
if targetclr == _get(img, x, y, true) then -- south
qtail = qtail + 2
qpt[qtail - 1] = x
qpt[qtail] = y
end
return fillclr
end
return _NIL -- signal marshal to stop
end
local function seed_pt(x, y)
-- should never hit max but make sure not to end early
for qhead = 2, 0x40000000, 2 do
if targetclr == _get(img, x, y, true) then
_marshal(img, x, y, 1, y, _NIL, _NIL, true, check_ns) -- west
_marshal(img, x + 1, y, max_w, y, _NIL, _NIL, true, check_ns) -- east
end
x = qpt[qhead - 1]
qpt[qhead - 1] = _NIL
if not x then break end
y = qpt[qhead]
qpt[qhead] = _NIL
end
end
seed_pt(x, y) -- Begin
end -- flood_fill
end

103
apps/plugins/lua/include_lua/draw_poly.lua Normal file
View file

@ -0,0 +1,103 @@
--[[ Lua Poly Drawing functions
/***************************************************************************
* __________ __ ___.
* Open \______ \ ____ ____ | | _\_ |__ _______ ___
* Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
* Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
* Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
* \/ \/ \/ \/ \/
* $Id$
*
* Copyright (C) 2017 William Wilgus
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
* KIND, either express or implied.
*
****************************************************************************/
]]
--[[ Exposed Functions
_poly.polygon
_poly.polyline
]]
if not rb.lcd_framebuffer then rb.splash(rb.HZ, "No Support!") return nil end
local _poly = {} do
local BSAND = 8 -- blits color to dst if src <> 0
local _NIL = nil -- nil placeholder
local _abs = math.abs
local _copy = rocklib_image.copy
local _line = rocklib_image.line
local flood_fill = require("draw_floodfill")
-- draws a non-filled figure based on points in t-points
local function polyline(img, x, y, t_points, color, bClosed, bClip)
if #t_points < 2 then error("not enough points", 3) end
local pt_first_last
if bClosed then
pt_first_last = t_points[1]
else
pt_first_last = t_points[#t_points]
end
for i = 1, #t_points, 1 do
local pt1 = t_points[i]
local pt2 = t_points[i + 1] or pt_first_last-- first and last point
_line(img, pt1[1] + x, pt1[2] + y, pt2[1] + x, pt2[2] + y, color, bClip)
end
end
-- draws a closed figure based on points in t_points
_poly.polygon = function(img, x, y, t_points, color, fillcolor, bClip)
if #t_points < 2 then error("not enough points", 3) end
if fillcolor then
local x_min, x_max = 0, 0
local y_min, y_max = 0, 0
local w, h = 0, 0
-- find boundries of polygon
for i = 1, #t_points, 1 do
local pt = t_points[i]
if pt[1] < x_min then x_min = pt[1] end
if pt[1] > x_max then x_max = pt[1] end
if pt[2] < y_min then y_min = pt[2] end
if pt[2] > y_max then y_max = pt[2] end
end
w = _abs(x_max) + _abs(x_min)
h = _abs(y_max) + _abs(y_min)
x_min = x_min - 2 -- leave a border to use flood_fill
y_min = y_min - 2
local fill_img = _newimg(w + 3, h + 3)
_clear(fill_img, 0x1)
for i = 1, #t_points, 1 do
local pt1 = t_points[i]
local pt2 = t_points[i + 1] or t_points[1]-- first and last point
_line(fill_img, pt1[1] - x_min, pt1[2] - y_min,
pt2[1]- x_min, pt2[2] - y_min, 0)
end
flood_fill(fill_img, fill_img:width(), fill_img:height() , 0x1, 0x0)
_copy(img, fill_img, x - 1, y - 1, _NIL, _NIL, _NIL, _NIL, bClip, BSAND, fillcolor)
end
polyline(img, x, y, t_points, color, true, bClip)
end
-- expose internal functions to the outside through _poly table
_poly.polyline = polyline
end
return _poly

121
apps/plugins/lua/include_lua/draw_text.lua Normal file
View file

@ -0,0 +1,121 @@
--[[ Lua Draw Text function
/***************************************************************************
* __________ __ ___.
* Open \______ \ ____ ____ | | _\_ |__ _______ ___
* Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
* Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
* Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
* \/ \/ \/ \/ \/
* $Id$
*
* Copyright (C) 2017 William Wilgus
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
* KIND, either express or implied.
*
****************************************************************************/
]]
-- draw text onto image if width/height are supplied text is centered
if not rb.lcd_framebuffer then rb.splash(rb.HZ, "No Support!") return nil end
do
-- Internal Constants
local rocklib_image = getmetatable(rb.lcd_framebuffer())
local _LCD = rb.lcd_framebuffer()
local LCD_W, LCD_H = rb.LCD_WIDTH, rb.LCD_HEIGHT
local BSAND = 8 -- blits color to dst if src <> 0
local _NIL = nil -- nil placeholder
local _clear = rocklib_image.clear
local _copy = rocklib_image.copy
local _newimg = rb.new_image
return function(img, x, y, width, height, font, color, text)
font = font or rb.FONT_UI
local opts = {x = 0, y = 0, width = LCD_W - 1, height = LCD_H - 1,
font = font, drawmode = 3, fg_pattern = 0x1, bg_pattern = 0}
if rb.LCD_DEPTH == 2 then -- invert 2-bit screens
--vp.drawmode = bit.bxor(vp.drawmode, 4)
opts.fg_pattern = 3 - opts.fg_pattern
opts.bg_pattern = 3 - opts.bg_pattern
end
rb.set_viewport(opts)
local res, w, h = rb.font_getstringsize(text, font)
if not width then
width = 0
else
width = (width - w) / 2
end
if not height then
height = 0
else
height = (height - h) / 2
end
-- make a copy of the current screen for later
--local screen_img = _newimg(LCD_W, LCD_H)
local screen_img = _newimg(LCD_W, h * 2)
_copy(screen_img, _LCD)
-- check if the screen buffer is supplied image if so set img to the copy
if img == _LCD then
img = screen_img
end
-- we will be printing the text to the screen then blitting into img
--rb.lcd_clear_display()
_clear(_LCD, opts.bg_pattern or 0, 1, 1, LCD_W, h * 2)
if w > LCD_W then -- text is too long for the screen do it in chunks
local l = 1
local resp, wp, hp
local lenr = text:len()
while lenr > 1 do
l = lenr
resp, wp, hp = rb.font_getstringsize(text:sub(1, l), font)
while wp >= LCD_W and l > 1 do
l = l - 1
resp, wp, hp = rb.font_getstringsize(text:sub( 1, l), font)
end
rb.lcd_putsxy(0, 0, text:sub(1, l))
text = text:sub(l)
if x + width > img:width() or y + height > img:height() then
break
end
-- using the mask we made blit color into img
_copy(img, _LCD, x + width, y + height, _NIL, _NIL, _NIL, _NIL, false, BSAND, color)
x = x + wp
--rb.lcd_clear_display()
_clear(_LCD, opts.bg_pattern or 0, 1, 1, LCD_W, h * 2)
lenr = text:len()
end
else --w <= LCD_W
rb.lcd_putsxy(0, 0, text)
-- using the mask we made blit color into img
_copy(img, _LCD, x + width, y + height, _NIL, _NIL, _NIL, _NIL, false, BSAND, color)
end
_copy(_LCD, screen_img) -- restore screen
rb.set_viewport() -- set viewport default
return res, w, h
end
end

184
apps/plugins/lua/include_lua/image.lua
View file

@ -23,7 +23,6 @@
--[[ Exposed Functions --[[ Exposed Functions
_img.save
_img.search _img.search
_img.rotate _img.rotate
_img.resize _img.resize
@ -156,189 +155,6 @@ local _img = {} do
return r_img return r_img
end end
-- saves img to file: name
_img.save = function(img, name)
-- bmp saving derived from rockbox - screendump.c
-- bitdepth is limited by the device
-- eg. device displays greyscale, rgb images are saved greyscale
local file
local bbuffer = {} -- concat buffer for s_bytes
local fbuffer = {} -- concat buffer for file writes, reused
local function s_bytesLE(bits, value)
-- bits must be multiples of 8 (sizeof byte)
local byte
local nbytes = bit.rshift(bits, 3)
for b = 1, nbytes do
if value > 0 then
byte = value % 256
value = (value - byte) / 256
else
byte = 0
end
bbuffer[b] = string.char(byte)
end
return table.concat(bbuffer, _NIL, 1, nbytes)
end
local function s_bytesBE(bits, value)
-- bits must be multiples of 8 (sizeof byte)
local byte
local nbytes = bit.rshift(bits, 3)
for b = nbytes, 1, -1 do
if value > 0 then
byte = value % 256
value = (value - byte) / 256
else
byte = 0
end
bbuffer[b] = string.char(byte)
end
return table.concat(bbuffer, _NIL, 1, nbytes)
end
local cmp = {["r"] = function(c) return bit.band(bit.rshift(c, 16), 0xFF) end,
["g"] = function(c) return bit.band(bit.rshift(c, 08), 0xFF) end,
["b"] = function(c) return bit.band(c, 0xFF) end}
local function bmp_color(color)
return s_bytesLE(8, cmp.b(color))..
s_bytesLE(8, cmp.g(color))..
s_bytesLE(8, cmp.r(color))..
s_bytesLE(8, 0) .. ""
end -- c_cmp(color, c.r))
local function bmp_color_mix(c1, c2, num, den)
-- mixes c1 and c2 as ratio of numerator / denominator
-- used 2x each save results
local bc1, gc1, rc1 = cmp.b(c1), cmp.g(c1), cmp.r(c1)
return s_bytesLE(8, cmp.b(c2) - bc1 * num / den + bc1)..
s_bytesLE(8, cmp.g(c2) - gc1 * num / den + gc1)..
s_bytesLE(8, cmp.r(c2) - rc1 * num / den + rc1)..
s_bytesLE(8, 0) .. ""
end
local w, h = img:width(), img:height()
local depth = tonumber(img:__tostring(6)) -- RLI_INFO_DEPTH = 0x6
local format = tonumber(img:__tostring(7)) -- RLI_INFO_FORMAT = 0x7
local bpp, bypl -- bits per pixel, bytes per line
-- bypl, pad rows to a multiple of 4 bytes
if depth <= 4 then
bpp = 8 -- 256 color image
bypl = (w + 3)
elseif depth <= 16 then
bpp = 16
bypl = (w * 2 + 3)
elseif depth <= 24 then
bpp = 24
bypl = (w * 3 + 3)
else
bpp = 32
bypl = (w * 4 + 3)
end
local linebytes = bit.band(bypl, bit.bnot(3))
local bytesperpixel = bit.rshift(bpp, 3)
local headersz = 54
local imgszpad = h * linebytes
local compression, n_colors = 0, 0
local h_ppm, v_ppm = 0x00000EC4, 0x00000EC4 --Pixels Per Meter ~ 96 dpi
if depth == 16 then
compression = 3 -- BITFIELDS
n_colors = 3
elseif depth <= 8 then
n_colors = bit.lshift(1, depth)
end
headersz = headersz + (4 * n_colors)
file = io.open('/' .. name, "w+") -- overwrite, rb ignores the 'b' flag
if not file then
rb.splash(rb.HZ, "Error opening /" .. name)
return
end
-- create a bitmap header 'rope' with image details -- concatenated at end
local bmpheader = fbuffer
bmpheader[01] = "BM"
bmpheader[02] = s_bytesLE(32, headersz + imgszpad)
bmpheader[03] = "\0\0\0\0" -- WORD reserved 1 & 2
bmpheader[04] = s_bytesLE(32, headersz) -- BITMAPCOREHEADER size
bmpheader[05] = s_bytesLE(32, 40) -- BITMAPINFOHEADER size
bmpheader[06] = s_bytesLE(32, w)
bmpheader[07] = s_bytesLE(32, h)
bmpheader[08] = "\1\0" -- WORD color planes ALWAYS 1
bmpheader[09] = s_bytesLE(16, bpp) -- bits/pixel
bmpheader[10] = s_bytesLE(32, compression)
bmpheader[11] = s_bytesLE(32, imgszpad)
bmpheader[12] = s_bytesLE(32, h_ppm) -- biXPelsPerMeter
bmpheader[13] = s_bytesLE(32, v_ppm) -- biYPelsPerMeter
bmpheader[14] = s_bytesLE(32, n_colors)
bmpheader[15] = s_bytesLE(32, n_colors)
-- Color Table (#n_colors entries)
if depth == 1 then -- assuming positive display
bmpheader[#bmpheader + 1] = bmp_color(0xFFFFFF)
bmpheader[#bmpheader + 1] = bmp_color(0x0)
elseif depth == 2 then
bmpheader[#bmpheader + 1] = bmp_color(0xFFFFFF)
bmpheader[#bmpheader + 1] = bmp_color_mix(0xFFFFFF, 0, 1, 3)
bmpheader[#bmpheader + 1] = bmp_color_mix(0xFFFFFF, 0, 2, 3)
bmpheader[#bmpheader + 1] = bmp_color(0x0)
elseif depth == 16 then
if format == 555 then
-- red bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x00007C00)
-- green bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x000003E0)
-- blue bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x0000001F)
else --565
-- red bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x0000F800)
-- green bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x000007E0)
-- blue bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x0000001F)
end
end
file:write(table.concat(fbuffer))-- write the header to the file now
for i=1, #fbuffer do fbuffer[i] = _NIL end -- reuse table
local imgdata = fbuffer
-- pad rows to a multiple of 4 bytes
local bytesleft = linebytes - (bytesperpixel * w)
local t_data = {}
local fs_bytes_E = s_bytesLE -- default save in Little Endian
if format == 3553 then -- RGB565SWAPPED
fs_bytes_E = s_bytesBE -- Saves in Big Endian
end
-- Bitmap lines start at bottom unless biHeight is negative
for point in _points(img, 1, h, w + bytesleft, 1) do
imgdata[#imgdata + 1] = fs_bytes_E(bpp, point or 0)
if #fbuffer >= 31 then -- buffered write, increase # for performance
file:write(table.concat(fbuffer))
for i=1, #fbuffer do fbuffer[i] = _NIL end -- reuse table
end
end
file:write(table.concat(fbuffer)) --write leftovers to file
fbuffer = _NIL
file:close()
end -- save(img, name)
--searches an image for target color --searches an image for target color
_img.search = function(img, x1, y1, x2, y2, targetclr, variation, stepx, stepy) _img.search = function(img, x1, y1, x2, y2, targetclr, variation, stepx, stepy)

215
apps/plugins/lua/include_lua/image_save.lua Normal file
View file

@ -0,0 +1,215 @@
--[[ Lua Image save
/***************************************************************************
* __________ __ ___.
* Open \______ \ ____ ____ | | _\_ |__ _______ ___
* Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
* Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
* Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
* \/ \/ \/ \/ \/
* $Id$
*
* Copyright (C) 2017 William Wilgus
*
* This program is free software; you can redistribute it and/or
* modify it under the terms of the GNU General Public License
* as published by the Free Software Foundation; either version 2
* of the License, or (at your option) any later version.
*
* This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
* KIND, either express or implied.
*
****************************************************************************/
]]
-- save(img, path/name)
-- bmp saving derived from rockbox - screendump.c
-- bitdepth is limited by the device
-- eg. device displays greyscale, rgb images are saved greyscale
if not rb.lcd_framebuffer then rb.splash(rb.HZ, "No Support!") return nil end
do
local rocklib_image = getmetatable(rb.lcd_framebuffer())
-- internal constants
local _NIL = nil -- _NIL placeholder
local _points = rocklib_image.points
-- saves img to file: name
return function(img, name)
local file
local bbuffer = {} -- concat buffer for s_bytes
local fbuffer = {} -- concat buffer for file writes, reused
local function s_bytesLE(bits, value)
-- bits must be multiples of 8 (sizeof byte)
local byte
local nbytes = bit.rshift(bits, 3)
for b = 1, nbytes do
if value > 0 then
byte = value % 256
value = (value - byte) / 256
else
byte = 0
end
bbuffer[b] = string.char(byte)
end
return table.concat(bbuffer, _NIL, 1, nbytes)
end
local function s_bytesBE(bits, value)
-- bits must be multiples of 8 (sizeof byte)
local byte
local nbytes = bit.rshift(bits, 3)
for b = nbytes, 1, -1 do
if value > 0 then
byte = value % 256
value = (value - byte) / 256
else
byte = 0
end
bbuffer[b] = string.char(byte)
end
return table.concat(bbuffer, _NIL, 1, nbytes)
end
local cmp = {["r"] = function(c) return bit.band(bit.rshift(c, 16), 0xFF) end,
["g"] = function(c) return bit.band(bit.rshift(c, 08), 0xFF) end,
["b"] = function(c) return bit.band(c, 0xFF) end}
local function bmp_color(color)
return s_bytesLE(8, cmp.b(color))..
s_bytesLE(8, cmp.g(color))..
s_bytesLE(8, cmp.r(color))..
s_bytesLE(8, 0) .. ""
end -- c_cmp(color, c.r))
local function bmp_color_mix(c1, c2, num, den)
-- mixes c1 and c2 as ratio of numerator / denominator
-- used 2x each save results
local bc1, gc1, rc1 = cmp.b(c1), cmp.g(c1), cmp.r(c1)
return s_bytesLE(8, cmp.b(c2) - bc1 * num / den + bc1)..
s_bytesLE(8, cmp.g(c2) - gc1 * num / den + gc1)..
s_bytesLE(8, cmp.r(c2) - rc1 * num / den + rc1)..
s_bytesLE(8, 0) .. ""
end
local w, h = img:width(), img:height()
local depth = tonumber(img:__tostring(6)) -- RLI_INFO_DEPTH = 0x6
local format = tonumber(img:__tostring(7)) -- RLI_INFO_FORMAT = 0x7
local bpp, bypl -- bits per pixel, bytes per line
-- bypl, pad rows to a multiple of 4 bytes
if depth <= 4 then
bpp = 8 -- 256 color image
bypl = (w + 3)
elseif depth <= 16 then
bpp = 16
bypl = (w * 2 + 3)
elseif depth <= 24 then
bpp = 24
bypl = (w * 3 + 3)
else
bpp = 32
bypl = (w * 4 + 3)
end
local linebytes = bit.band(bypl, bit.bnot(3))
local bytesperpixel = bit.rshift(bpp, 3)
local headersz = 54
local imgszpad = h * linebytes
local compression, n_colors = 0, 0
local h_ppm, v_ppm = 0x00000EC4, 0x00000EC4 --Pixels Per Meter ~ 96 dpi
if depth == 16 then
compression = 3 -- BITFIELDS
n_colors = 3
elseif depth <= 8 then
n_colors = bit.lshift(1, depth)
end
headersz = headersz + (4 * n_colors)
file = io.open('/' .. name, "w+") -- overwrite, rb ignores the 'b' flag
if not file then
rb.splash(rb.HZ, "Error opening /" .. name)
return
end
-- create a bitmap header 'rope' with image details -- concatenated at end
local bmpheader = fbuffer
bmpheader[01] = "BM"
bmpheader[02] = s_bytesLE(32, headersz + imgszpad)
bmpheader[03] = "\0\0\0\0" -- WORD reserved 1 & 2
bmpheader[04] = s_bytesLE(32, headersz) -- BITMAPCOREHEADER size
bmpheader[05] = s_bytesLE(32, 40) -- BITMAPINFOHEADER size
bmpheader[06] = s_bytesLE(32, w)
bmpheader[07] = s_bytesLE(32, h)
bmpheader[08] = "\1\0" -- WORD color planes ALWAYS 1
bmpheader[09] = s_bytesLE(16, bpp) -- bits/pixel
bmpheader[10] = s_bytesLE(32, compression)
bmpheader[11] = s_bytesLE(32, imgszpad)
bmpheader[12] = s_bytesLE(32, h_ppm) -- biXPelsPerMeter
bmpheader[13] = s_bytesLE(32, v_ppm) -- biYPelsPerMeter
bmpheader[14] = s_bytesLE(32, n_colors)
bmpheader[15] = s_bytesLE(32, n_colors)
-- Color Table (#n_colors entries)
if depth == 1 then -- assuming positive display
bmpheader[#bmpheader + 1] = bmp_color(0xFFFFFF)
bmpheader[#bmpheader + 1] = bmp_color(0x0)
elseif depth == 2 then
bmpheader[#bmpheader + 1] = bmp_color(0xFFFFFF)
bmpheader[#bmpheader + 1] = bmp_color_mix(0xFFFFFF, 0, 1, 3)
bmpheader[#bmpheader + 1] = bmp_color_mix(0xFFFFFF, 0, 2, 3)
bmpheader[#bmpheader + 1] = bmp_color(0x0)
elseif depth == 16 then
if format == 555 then
-- red bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x00007C00)
-- green bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x000003E0)
-- blue bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x0000001F)
else --565
-- red bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x0000F800)
-- green bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x000007E0)
-- blue bitfield mask
bmpheader[#bmpheader + 1] = s_bytesLE(32, 0x0000001F)
end
end
file:write(table.concat(fbuffer))-- write the header to the file now
for i=1, #fbuffer do fbuffer[i] = _NIL end -- reuse table
local imgdata = fbuffer
-- pad rows to a multiple of 4 bytes
local bytesleft = linebytes - (bytesperpixel * w)
local t_data = {}
local fs_bytes_E = s_bytesLE -- default save in Little Endian
if format == 3553 then -- RGB565SWAPPED
fs_bytes_E = s_bytesBE -- Saves in Big Endian
end
-- Bitmap lines start at bottom unless biHeight is negative
for point in _points(img, 1, h, w + bytesleft, 1) do
imgdata[#imgdata + 1] = fs_bytes_E(bpp, point or 0)
if #fbuffer >= 31 then -- buffered write, increase # for performance
file:write(table.concat(fbuffer))
for i=1, #fbuffer do fbuffer[i] = _NIL end -- reuse table
end
end
file:write(table.concat(fbuffer)) --write leftovers to file
fbuffer = _NIL
file:close()
end -- save(img, name)
end

4
apps/plugins/lua/lua.make
View file

@ -16,8 +16,8 @@ LUA_OBJ := $(call c2obj, $(LUA_SRC))
OTHER_SRC += $(LUA_SRC) OTHER_SRC += $(LUA_SRC)
LUA_INCLUDEDIR := $(LUA_SRCDIR)/include_lua LUA_INCLUDEDIR := $(LUA_SRCDIR)/include_lua
LUA_INCLUDELIST := $(addprefix $(LUA_BUILDDIR)/,audio.lua blit.lua color.lua draw.lua \ LUA_INCLUDELIST := $(addprefix $(LUA_BUILDDIR)/,audio.lua blit.lua color.lua draw.lua draw_floodfill.lua draw_poly.lua \
image.lua lcd.lua math_ex.lua print.lua \ draw_text.lua image.lua image_save.lua lcd.lua math_ex.lua print.lua \
timer.lua playlist.lua pcm.lua sound.lua \ timer.lua playlist.lua pcm.lua sound.lua \
rbcompat.lua printtable.lua) rbcompat.lua printtable.lua)

919
apps/plugins/lua_scripts/rlimg.lua Executable file
View file

@ -0,0 +1,919 @@
--[[
__________ __ ___.
Open \______ \ ____ ____ | | _\_ |__ _______ ___
Source | _// _ \_/ ___\| |/ /| __ \ / _ \ \/ /
Jukebox | | ( <_> ) \___| < | \_\ ( <_> > < <
Firmware |____|_ /\____/ \___ >__|_ \|___ /\____/__/\_ \
\/ \/ \/ \/ \/
$Id$
Example Lua RBIMAGE script
Copyright (C) 2009 by Maurus Cuelenaere -- some prior work
Copyright (C) 2017 William Wilgus
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This software is distributed on an "AS IS" basis, WITHOUT WARRANTY OF ANY
KIND, either express or implied.
]]--
require("actions") -- Contains rb.actions & rb.contexts
-- require("buttons") -- Contains rb.buttons -- not needed for this example
local _math = require("math_ex") -- missing math sine cosine, sqrt, clamp functions
local _timer = require("timer")
local _clr = require("color") -- clrset, clrinc provides device independent colors
local _lcd = require("lcd") -- lcd helper functions
local _print = require("print") -- advanced text printing
local _img = require("image") -- image manipulation save, rotate, resize, tile, new, load
local _img_save = require("image_save")
local _blit = require("blit") -- handy list of blit operations
local _draw = require("draw") -- draw all the things (primitives)
local _draw_floodfill = require("draw_floodfill")
local _draw_text = require("draw_text")
--local scrpath = rb.current_path()"
--package.path = scrpath .. "/?.lua;" .. package.path --add lua_scripts directory to path
require("printmenu") --menu
--[[ RBIMAGE library functions
NOTE!! on x, y coordinates + width & height
----------------------------------------------
When making a new image you specify the width and height say (10, 20).
Intutively (at least to me) (0,0) (offset addressing) would reference the first
pixel (top left) and the last pixel (bottom, right) would be (w-1, h-1) or (9, 19)
but the original rbimage library (like lua) uses 1-based arrays (ordinal addressing)
for the image data so the first pixel is (1,1) and the last pixel is (w, h)
this presents a bit of a problem when interfacing with the internal rockbox
functions as you must remeber to convert all lua coordinates to 0 based coordinates.
I have opted to not change this in the name of compatibility with old scripts
NOTE2!! on width & height versus offset_x & offset_y :copy()
------------------------------------------------------------------------
The only place where RBIMAGE deviates from (ordinal addressing) is in the blit
/ copy function sx, sy and dx, dy still refer to the same pixel as all the
other functions but offset_x, and offset_y are zero based
Example (assuming same sized images)
dst:copy(src, 1,1, 1,1, 0, 0)
would copy the first pixel from src to the first pixel of dst
dst:copy(src, 1,1, 1,1, dst:width()-1, dst:height()-1) would copy all of src to dst
this might be a bit confusing but makes reversing images (negative offsets) easier.
Since offsets are auto calculated if empty or out of range you could call it as
dst:copy(src, 1,1, 1,1, dst:width(), dst:height()) or even
dst:copy(src, 1,1, 1,1) if you prefered and get the same result
]]
--'CONSTANTS' (in lua there really is no such thing as all vars are mutable)
--------------------------------------------------------
--colors for fg/bg ------------------------
local WHITE = _clr.set(-1, 255, 255, 255)
local BLACK = _clr.set(0, 0, 0, 0)
local RED = _clr.set(WHITE, 255)
local GREEN = _clr.set(WHITE, 0, 255)
local BLUE = _clr.set(WHITE, 0, 0, 255)
-------------------------------------------
local clrs
local CANCEL_BUTTON = rb.actions.PLA_CANCEL
-- EXAMPLES ---------------------------------------------------------------------- EXAMPLES---------------------------------------------------------------------
function my_blit(dst_val, dx, dy, src_val, sx, sy)
-- user defined blit operation
--this function gets called for every pixel you specify
--you may change pixels in both the source and dest image
--return nil to stop early
if _lcd.DEPTH < 2 then
return src_val
end
if dst_val == WHITE and bit.band(dx, 1) == 1 and bit.band(dy, 1) == 1 then
return BLACK;
elseif dst_val == WHITE then
return src_val
end
return dst_val
end
function create_logo()
--[[creates a small logo from data array]]
-- moves scope of white and black from global to local
local WHITE = WHITE
local BLACK = BLACK
--[[small array with 16 bits of data per element (16-bits in y direction)
while the number of elements (32) is the x direction.
in other words 16 rows X 32 columns, totally abritrary actually
you could easily rotate or flip it by changing the for loops below ]]
local logo = {0xFFFF, 0xFFFF, 0x8001, 0x8001, 0x9E7F, 0x9E7F, 0x9E7F, 0x9E7F,
0x9E3F, 0x9E3F, 0x804F, 0x804F, 0xC0E1, 0xC0F1, 0xFFFD, 0xFFFF,
0xFFFF, 0xFFFF, 0x8001, 0x8001, 0xFF01, 0xFF01, 0xFEFB, 0xFEFB,
0xFDFD, 0xFDFD, 0xFDFD, 0xFCF9, 0xFE03, 0xFE03, 0xFFFF, 0xFFFF}
local img, img1, img2, img3
img = _img.new(_lcd.W, _lcd.H)
img:clear(BLACK) --[[always clear an image after you create it if you
intend to do any thing besides copy something
entirely to it as there is no guarantee what
state the data within is in, it could be all
0's or it could be every digit of PI ]]
-- check for an error condition bail if error
if(not img or not logo) then
return nil
end
local logosz = table.getn(logo)
local bits = 16 -- each element contains 16 pixels
local data = 0
for i=1, math.min(logosz, _lcd.W) do
for j=0, math.min(bits, _lcd.H) do
if bit.band(bit.lshift(1, bits - j), logo[i]) > 0 then
data = WHITE
else
data = BLACK
end
-- Set the pixel at position i, j+1 to the copied data
img:set(i, j + 1, data)
end
end
-- make a new image the size of our generated logo
img1 = _img.new(logosz + 1, bits + 1)
-- img.copy(dest, source, dx, dy, sx, sy, [w, h])
img1:copy(img, 1, 1, 1, 1)
--[[lua does auto garbage collection, but it is still
a good idea to set large items to nil when done anyways]]
img = nil
local sl -- new image size
if _lcd.W < _lcd.H then
sl = _lcd.W / 3
else
sl = _lcd.H / 3
end
-- make sl always even by subtracting 1 if needed
sl = bit.band(sl, bit.bnot(1))
if sl < 16 then
sl = 16
end
img2 = _img.new(sl, sl)
--img2:clear(BLACK) -- doesn't need cleared since we copy to it entirely
--[[we are going to resize the image since the data supplied is 32 x 16
pixels its really tiny on most screens]]
_img.resize(img2, img1)
img1 = nil
img3 = _img.new(sl, sl)
img3:clear(BLACK)
if IS_COLOR_TARGET == true then
local c_yellow = _clr.set(WHITE, 0xFC, 0xC0, 0x00)
local c_grey = _clr.set(WHITE, 0xD4, 0xE3, 0xF3)
local c_dkgrey = _clr.set(WHITE, 0xB4, 0xC3, 0xD3)
-- if dest pixel == source pixel make dest pixel yellow
img3:copy(img2, 1, 1, 1, 1, nil, nil, false, _blit.BDEQS, c_yellow)
-- xor src pixel to dest pixel if both are 0 or both are 1 dest = 0
img2:copy(img3, 1, 1, 2, 1, nil, nil, false, _blit.BXOR)
-- if dest pixel color > src pixel color copy grey to dest
img2:copy(img3, 1, 1, 1, 1, nil, nil, false, _blit.BDGTS, c_grey)
-- set img3 to grey
img3:clear(c_dkgrey)
end
-- make a WHITE square in the middle
img3:clear(WHITE, 4,4, img3:width() - 3, img3:height() - 3)
img3:copy(img2, 1, 1, 1, 1, nil, nil, false, _blit.CUSTOM, my_blit)
img2 = nil
_img_save(img3, "pix.bmp")
return img3
end -- copy_logo
-- draws an olive erm ball and returns it
function create_ball()
local sl -- image size
if _lcd.W < _lcd.H then
sl = _lcd.W / 5
else
sl = _lcd.H / 5
end
-- make sl always even by subtracting 1 if needed
sl = bit.band(sl, bit.bnot(1))
if sl < 16 then
sl = 16
end
local img = _img.new(sl, sl)
img:clear(0)
_draw.circle_filled(img, sl/2, sl/2, sl/2 - 1, _clr.set(-1, 255), _clr.set(0, 100))
_draw.circle_filled(img, sl/3, sl/3, sl/10, _clr.set(-1, 255, 0, 0))
return img
end
-- bounces img around on screen
function bounce_image(img)
local timer = _timer() -- creates a new timer -- saves index
local wait
-- make a copy of the current screen for later
local screen_img = _lcd:duplicate()
local img_sqy = _img.new(img:width() + img:width() / 8, img:height())
local img_sqx = _img.new(img:width(), img:height() + img:height() / 8)
_img.resize(img_sqy, img)
_img.resize(img_sqx, img)
-- moves definition of CANCEL_BUTTON from global to local
local CANCEL_BUTTON = CANCEL_BUTTON
--------------------------------------------------------
local imgn = img
local hold = 0
local sx = 1 -- starting x
local sy = 1 -- starting y
local ex = _lcd.W - img:width() - 2
local ey = _lcd.H - img:width() - 2
-- threshold resets speed, inverts image
local tx = ex / 5
local ty = ey / 5
local last_x = sx
local last_y = sy
local x = sx
local y = sy
local dx = 1
local dy = 1
-- negative width\height cause the image to be drawn from the opposite end
local fx = _lcd.W
local fy = _lcd.H
local function invert_images()
img:invert();
img_sqx:invert()
img_sqy:invert()
end
local loops = (_lcd.W * _lcd.H) / 2
while (loops > 0) do
if IS_COLOR_TARGET then
if bit.band(loops, 128) == 128 then
_lcd:copy(imgn, x, y, 1, 1, fx, fy, false, _blit.BOR)
_lcd:copy(screen_img, x, y, x, y, imgn:width(), imgn:height(),
false, _blit.BDEQC, imgn:get(1,1))
else
_lcd:copy(imgn, x, y, 1, 1, fx, fy, false, _blit.BSNEC, imgn:get(1,1))
end
else
local blitop
if imgn:get(1,1) ~= 0 then
blitop = _blit.BSNOT
else
blitop = _blit.BXOR
end
_lcd:copy(imgn, x, y, 1, 1, fx, fy, false, blitop, WHITE)
end
if hold < 1 then
imgn = img
else
hold = hold - 1
end
_lcd:update()
x = x + dx
y = y + dy
if y >= ey or y <= sy then
dy = (-dy)
fy = (-fy)
imgn = img_sqy
hold = 3
if dx < 0 and y < 10 then
dx = dx - 5
end
if dx > tx then
dx = 5;
dy = 5;
invert_images()
end
end
if x >= ex or x <= sx then
dx = (-dx)
fx = (-fx)
imgn = img_sqx
hold = 3
if dy < 0 and x < 10 then
dy = dy - 5
end
if dy > ty then
dx = 5;
dy = 5;
invert_images()
end
end
x = _math.clamp(x, sx, ex)
y = _math.clamp(y, sy, ey)
-- copy original image back to screen
_lcd:copy(screen_img)
loops = loops -1
wait = timer:check(true) --checks timer and updates last time
if wait >= 5 then
wait = 0
elseif wait < 5 then
wait = 5 - wait
end
-- 0 = timeout immediately
-- ( -1 would be never timeout, and >0 is amount of 'ticks' before timeout)
if rb.get_plugin_action(wait) == CANCEL_BUTTON then
break;
end
end
timer:stop() -- destroys timer, also returns time since last time
-- leave the screen how we found it
_lcd:copy(screen_img)
end -- image_bounce
-- draws a gradient using available colors
function draw_gradient(img, x, y, w, h, direction, clrs)
x = x or 1
y = y or 1
w = w or img:width() - x
h = h or img:height() - y
local zstep = 0
local step = 1
if IS_COLOR_TARGET == true then -- Only do this when we're on a color target
local z = 1
local c = 1
clrs = clrs or {255,255,255}
local function gradient_rgb(p, c1, c2)
-- tried squares of difference but blends were very abrupt
local r = c1.r + (p * (c2.r - c1.r) / 10500)
local g = c1.g + (p * (c2.g - c1.g) / 10000)
local b = c1.b + (p * (c2.b - c1.b) / 09999)
return _clr.set(nil, r, g, b)
end
local function check_z()
if z > 10000 and c < #clrs - 1 then
z = 1
c = c + 1
elseif z > 10000 then
z = 10000
end
end
if direction == "V" then
zstep = math.max(1, (10000 / ((w - 1) / (#clrs - 1)) + bit.band(#clrs, 1)))
for i=x, w + x do
check_z()
_draw.vline(img, i, y, h, gradient_rgb(z, clrs[c], clrs[c + 1]))
z = z + zstep
end
else
zstep = math.max(1, (10000 / ((h - 1) / (#clrs - 1)) + bit.band(#clrs, 1)))
for j=y, h + y do
check_z()
_draw.hline(img, x, j, w, gradient_rgb(z, clrs[c], clrs[c + 1]))
z = z + zstep
end
end
else -- not a color target but might be greyscale
local clr = _clr.set(-1)
for i=x, w + x do
for j=y, h + y do
-- Set the pixel at position i, j to the specified color
img:set(i, j, clr)
end
clr = _clr.inc(clr, 1)
end
end
--rb.sleep(1000)
end -- draw_gradient
function twist(img)
--[[ local fg
if rb.lcd_get_foreground then
fg = rb.lcd_get_foreground()
end]]
local ims = {}
ims.strip = _img.tile(img, img:width(), _lcd.H + img:height())
ims.y_init = {img:height(), 1}
ims.y_finl = {1, img:height()}
ims.y_inc = {-2, 4}
ims.y_pos = nil
-- together define the width of the overall object
local x_off=(_lcd.W/2)
local m = _lcd.W
local z = -m
local zi = 1
-- angles of rotation for each leaf
local ANGLES = {0, 45, 90, 135, 180, 225, 270, 315}
local elems = #ANGLES
local _XCOORD = {}
-- color alternates each leaf
local colors = { WHITE, _clr.set(0, 0, 0, 0) }
local c = 0
-- calculated position of each point in the sine wave(s)
local xs, xe
--[[--Profiling code
local timer = _timer.start()]]
for rot = 0, 6000, 4 do
_lcd:clear(BLACK)
for y=1, _lcd.H do
local function get_sines()
local sc = m + z
if sc == 0 then
sc = 1 -- prevent divide by 0
elseif sc + z > _lcd.W then
zi = -1
colors[2] = _clr.inc(colors[2], 1, 0, 50, 0)
elseif sc + z < -(_lcd.W) then
zi = 1
colors[1] = _clr.inc(colors[1], -1, 0, 10, 0)
end
if colors[2] == colors[1] then
colors[2] = _clr.inc(colors[2], 1)
end
for j = 1, elems do
_XCOORD[j] = _math.d_sin(y + ANGLES[j] + rot) / sc + x_off
end
_XCOORD[0] = _XCOORD[elems] -- loop table for efficient wrap
end
get_sines()
for k = 1, elems do
xs = _XCOORD[k]
xe = _XCOORD[(k+1) % elems]
if xs < 1 or xe > _lcd.W then
while xs < 1 or xe > _lcd.W do
m = m + 1 -- shift m in order to scale object min/max
get_sines()
xs = _XCOORD[k]
xe = _XCOORD[(k+1) % elems]
end
end
c = (c % 2) + 1
if xs < xe then
-- defines the direction of leaves & fills them
_lcd:set(xs, y, colors[c])
_lcd:set(xe, y, colors[c])
_lcd:line(xs + 1, y, xe - 1, y, colors[3 - c], true)
end
end
end
do -- stripes and shifts image strips; blits it into the colors(1) leaves
local y
local y_col = 1
local w = ims.strip:width() - 1
local h = ims.strip:height() - 1
if ims.y_pos ~= nil then
for i = 1, #ims.y_pos do
ims.y_pos[i] = ims.y_pos[i] + ims.y_inc[i]
if (ims.y_inc[i] > 0 and ims.y_pos[i] > ims.y_finl[i])
or (ims.y_inc[i] < 0 and ims.y_pos[i] < ims.y_finl[i]) then
ims.y_pos[i] = ims.y_init[i]
end
end
else
ims.y_pos = {ims.y_init[1], ims.y_init[2]}
end
for ix = 1, _lcd.W, w do
y_col = y_col + 1
y = ims.y_pos[(y_col % 2) + 1]
if _lcd.DEPTH > 1 then
_lcd:copy(ims.strip, ix, 1, 1, y, w, h, false, _blit.BDEQC, colors[1])
else
_lcd:copy(ims.strip, ix, 1, 1, y, w, h, false, _blit.BSAND)
end
end
end
_lcd:update()
z = z + zi
if rb.get_plugin_action(0) == CANCEL_BUTTON then
break
end
collectgarbage("step")
end
--[[--Profiling code
_print.f("%d", _timer.stop(timer))
rb.sleep(rb.HZ * 10)]]
end -- twist
function draw_target(img)
local clr = _clr.set(0, 0, 0, 0)
-- make a copy of original screen for restoration
local screen_img = _lcd:duplicate()
rad_m = math.min(_lcd.W, _lcd.H)
for s = -_lcd.W /4, 16 do
img:copy(screen_img)
s = math.max(1, math.abs(s))
for r = 1, rad_m /2 - 10, s do
clr = _clr.inc(clr, 1, r * 5, r * 10, r * 20)
_draw.circle(img, _lcd.CX, _lcd.CY, r, clr)
end
_lcd:update()
if rb.get_plugin_action( 20) == CANCEL_BUTTON then
z = 16;
break;
end
end
end -- draw_target
function draw_sweep(img, cx, cy, radius, color)
local timer = _timer() --creates a new timer saves index
local wait
local x
local y
--make a copy of original screen for restoration
local screen_img = _lcd:duplicate()
_draw.circle(img, cx, cy, radius, color)
for d = 630, 270, - 5 do
if d % 45 == 0 then
img:copy(screen_img)
_draw.circle(img, cx, cy, radius, color)
l = 0
end
x = cx + radius * _math.d_cos(d) / 10000
y = cy + radius * _math.d_sin(d) / 10000
_draw.line(img, cx, cy, x, y, color)
l = l + 1
if l > 1 then
x1 = cx + (radius - 1) * _math.d_cos(d + 1) / 10000
y1 = cy + (radius - 1) * _math.d_sin(d + 1) / 10000
_draw_floodfill(img, x1, y1, BLACK, color)
end
_lcd:update()
wait = timer:check(true) --checks timer and updates last time
if wait >= 50 then
wait = 0
elseif wait < 50 then
wait = 50 - wait
end
if rb.get_plugin_action( wait) == CANCEL_BUTTON then
break
end
end
timer:stop() --destroys timer, also returns time since last time
screen_img = nil
end -- draw_sweep
function rotate_image(img)
local blitop = _blit.BOR
local i = 1
local d = 0
local ximg
local x, y, w, h, xr, yr
ximg = _img.rotate(img, 45) -- image will be largest at this point
w = ximg:width()
h = ximg:height()
xr = (_lcd.W - w) / 2
yr = (_lcd.H - h) / 2
--make a copy of original screen for restoration
local screen_img -- = _lcd:duplicate()
screen_img =_img.new(w, h)
screen_img :copy(_LCD, 1, 1, xr, yr, w, h)
--_print.f("CW")
--[[--Profiling code
local timer = _timer.start()]]
while d >= 0 do
ximg = _img.rotate(img, d)
w = ximg:width()
h = ximg:height()
x = (_lcd.W - w) / 2
y = (_lcd.H - h) / 2
-- copy our rotated image onto the background
_lcd:copy(ximg, x, y, 1, 1, w, h, false, blitop)
_lcd:update()
--restore the portion of the background we destroyed
_lcd:copy(screen_img, xr, yr, 1, 1)
if d > 0 and d % 360 == 0 then
_lcd:copy(ximg, x, y, 1, 1, w, h)
_lcd:update()
if i == 1 then i = 0 end
if d == 1440 or i < 0 then
if i < 0 then
i = i - 5
else
i = - 5
end
blitop = _blit.BXOR
--_print.f("CCW")
--rb.sleep(rb.HZ)
else
i = i + 5
--_print.f("CW")
--rb.sleep(rb.HZ)
end
end
d = d + i
if rb.get_plugin_action(0) == CANCEL_BUTTON then
break;
end
end
_lcd:copy(ximg, x, y, 1, 1, w, h)
--[[-- Profiling code
_print.f("%d", _timer.stop(timer))
rb.sleep(rb.HZ * 10)]]
end -- rotate_image
-- shows blitting with a mask
function blit_mask(dst)
local timer = _timer()
local r = math.min(_lcd.CX, _lcd.CY) / 5
local bmask = _img.new(_lcd.W, _lcd.H)
bmask:clear(0)
_draw.circle_filled(bmask, _lcd.CX, _lcd.CY, r, WHITE)
local color = _clr.set(0, 0, 0 ,0)
for z = 0, 100 do
z = z + timer:check(true)
color = _clr.inc(color, 1, z * 5, z * 10, z * 20)
dst:copy(bmask, 1, 1, 1, 1, nil, nil, false, _blit.BSAND, color)
_lcd:update()
if rb.get_plugin_action(0) == CANCEL_BUTTON then
break
end
end
end -- blit_mask
-- draws an X on the screen
function draw_x()
_draw.line(_LCD, 1, 1, _lcd.W, _lcd.H, WHITE)
_draw.line(_LCD, _lcd.W, 1, 1, _lcd.H, WHITE)
_draw.hline(_LCD, 10, _lcd.CY , _lcd.W - 21, WHITE)
_draw.vline(_LCD, _lcd.CX, 20 , _lcd.H - 41, WHITE)
_draw.rect(_LCD, _lcd.CX - 17, _lcd.CY - 17, 34, 34, WHITE)
_lcd:update()
rb.sleep(100)
end -- draw_x
--fills an image with random colors
function random_img(img)
local min = _clr.set(0, 0, 0, 0)
local max = _clr.set(-1, 255, 255, 255)
math.randomseed(rb.current_tick())
for x = 1, img:width() do
for y = 1, img:height() do
img:set(x, y, math.random(min, max))
end
end
end -- random_img
function rainbow_img(img)
--draw a gradient using available colors
if IS_COLOR_TARGET == true then
--R O Y G B I V
clrs = {
{r = 255, g = 255, b = 255}, -- white
{r = 000, g = 000, b = 000}, -- black
{r = 200, g = 000, b = 000}, -- red
{r = 255, g = 000, b = 000}, -- red
{r = 255, g = 100, b = 033}, -- orange
{r = 255, g = 127, b = 000}, -- orange
{r = 255, g = 200, b = 033}, -- yellow
{r = 255, g = 255, b = 000}, -- yellow
{r = 050, g = 255, b = 000}, -- green
{r = 000, g = 125, b = 125}, -- green
{r = 000, g = 000, b = 255}, -- blue
{r = 033, g = 025, b = 200}, -- indigo
{r = 075, g = 000, b = 150}, -- indigo
{r = 127, g = 000, b = 150}, -- violet
{r = 150, g = 000, b = 255}, -- violet
{r = 255, g = 255, b = 255}, -- white
{r = 000, g = 000, b = 000}, -- black
}
else
end
draw_gradient(img, 1, 1, nil, nil, "V", clrs)
end -- rainbow_img
-- draws a rounded rectangle with text
function rock_lua()
local res, w, h = _lcd:text_extent("W", rb.FONT_UI)
w = _lcd.W - 10
h = h + 4
_draw.rounded_rect_filled(_LCD, 5, 5, w, h, 15, WHITE)
_draw_text(_LCD, 5, 5, w, h, nil, BLACK, "ROCKlua!")
_lcd:update()
rb.sleep(100)
end -- rock_lua
-- draws a rounded rectangle with long text
function long_text()
local txt = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ1234567890"
local res, w, h = _lcd:text_extent(txt, rb.FONT_UI)
local resp, wp, hp = _lcd:text_extent(" ", rb.FONT_UI)
local wait = 0
w = w + wp * 3
h = h + 4
local img = _img.new(w + 1, h)
img:clear(BLACK)
_draw.rounded_rect_filled(img, 1, 1, w, h, 15, WHITE)
_draw_text(img, 1, 2, nil, nil, nil, BLACK, txt)
for p = -w + 1, w - 1 do
wait = 0
p = math.abs(p)
_lcd:copy(img, 1, _lcd.H - h, w - p, 1)
_lcd:update()
if p == 0 or w - p == 1 then wait = 100; rb.sleep(50) end
if rb.get_plugin_action(wait) == CANCEL_BUTTON then
break
end
end
end -- long_text
-- creates or loads an image to use as logo
function get_logo()
local logo_img = _img.load("/pix.bmp") --loads the previously saved image (if we saved it before)
--create a logo if an image wasn't saved previously
if(not logo_img) then
logo_img = create_logo()
end
--fallback
if(not logo_img) then
-- Load a BMP file in the variable backdrop
local base = "/.rockbox/icons/"
local backdrop = _img.load("/image.bmp") --user supplied image
or _img.load(base .. "tango_small_viewers.bmp")
or _img.load(base .. "tango_small_viewers_mono.bmp")
or _img.load(base .. "tango_small_mono.bmp")
or _img.load(base .. "tango_icons.16x16.bmp")
or _img.load(base .. "tango_icons_viewers.16x16.bmp")
or _img.load(base .. "viewers.bmp")
or _img.load(base .. "viewers.6x8x1.bmp")
or _img.load(base .. "viewers.6x8x2.bmp")
or _img.load(base .. "viewers.6x8x10.bmp")
or _img.load(base .. "viewers.6x8x16.bmp")
logo_img = backdrop
end
return logo_img
end -- get_logo
-- uses print_table to display a menu
function main_menu()
local mt = {
[1] = "Rocklua RLI Example",
[2] = "The X",
[3] = "Blit Mask",
[4] = "Target",
[5] = "Sweep",
[6] = "Bouncing Ball (olive)",
[7] = "The Twist",
[8] = "Image Rotation",
[9] = "Long Text",
[10] = "Rainbow Image",
[11] = "Random Image",
[12] = "Clear Screen",
[13] = "Save Screen",
[14] = "Exit"
}
local ft = {
[0] = exit_now, --if user cancels do this function
[1] = function(TITLE) return true end, -- shouldn't happen title occupies this slot
[2] = function(THE_X) draw_x() end,
[3] = function(BLITM) blit_mask(_lcd()) end,
[4] = function(TARGT) draw_target(_lcd()) end,
[5] = function(SWEEP)
local r = math.min(_lcd.CX, _lcd.CY) - 20
draw_sweep(_lcd(), _lcd.CX, _lcd.CY, r, _clr.set(-1, 0, 255, 0))
end,
[6] = function(BOUNC) bounce_image(create_ball()) end,
[7] = function(TWIST) twist(get_logo()) end,
[8] = function(ROTAT) rotate_image(get_logo()) end,
[9] = long_text,
[10] = function(RAINB)
rainbow_img(_lcd()); _lcd:update(); rb.sleep(rb.HZ)
end,
[11] = function(RANDM)
random_img(_lcd()); _lcd:update(); rb.sleep(rb.HZ)
end,
[12] = function(CLEAR) _lcd:clear(BLACK); rock_lua() end,
[13] = function(SAVEI) _LCD:invert(); _img_save(_LCD, "/rocklua.bmp") end,
[14] = function(EXIT_) return true end
}
if _lcd.DEPTH < 2 then
table.remove(mt, 10)
table.remove(ft, 10)
end
print_menu(mt, ft)
end
--------------------------------------------------------------------------------
function exit_now()
_lcd:update()
_lcd:splashf(rb.HZ * 5, "ran for %d seconds", _timer.stop("main") / rb.HZ)
os.exit()
end -- exit_now
--------------------------------------------------------------------------------
--MAIN PROGRAM------------------------------------------------------------------
_timer("main") -- keep track of how long the program ran
-- Clear the screen
_lcd:clear(BLACK)
if _lcd.DEPTH > 1 then
--draw a gradient using available colors
if IS_COLOR_TARGET == true then
clrs = {
{r = 255, g = 000, b = 000}, -- red
{r = 000, g = 255, b = 000}, -- green
{r = 000, g = 000, b = 255}, -- blue
}
else
end
local w = bit.rshift(_lcd.W, 2)
local h = bit.rshift(_lcd.H, 2)
draw_gradient(_lcd(), (_lcd.W - w)/2 - 1, (_lcd.H - h)/3 - 1, w, h, "H", clrs)
_lcd:update()
rb.sleep(rb.HZ)
end
do
local img = _img.load("/rocklua.bmp")
if not img then
rock_lua()
else
_lcd:image(img)
end
end
_lcd:update()
rb.sleep(rb.HZ / 2)
if rb.cpu_boost then rb.cpu_boost(true) end
main_menu()
if rb.cpu_boost then rb.cpu_boost(false) end
exit_now()
-- For a reference list of all functions available, see apps/plugins/lua/rocklib.c (and apps/plugin.h)