require 'lfs'
local ut = require 'utils'
local www_utils = {}
-- isNorm is deprecated
function www_utils.saveIms(ims, imPath, isNorm, count)
count = count or 0
isNorm = isNorm or false
local numIms = ut.getLength(ims)
local filenames = {}
for i = 1, numIms do
filenames[i] = string.format("%08d.jpg", count + i)
-- Scale the image and save
local im = ims[i]
if isNorm then
im = ims[i]:clone()
local maxVal = im:max()
local minVal = im:min()
im:add(-minVal):mul(1.0 / (maxVal - minVal))
end
image.save(paths.concat(imPath, filenames[i]), im)
end
return filenames, count + numIms
end
-- ims and captions are both tables
function www_utils.renderTables(ims, captions, ncol, width, imPath)
assert(#captions == 0 or #captions == #ims, "#captions should be either 0 or equal to #ims")
local nrow = math.ceil(#ims / ncol)
local htmlStr = "\n"
local k = 0
for i = 1, nrow do
local captionRow = "| "..i.." | "
local imRow = ""..i.." | "
for j = 1, ncol do
k = k + 1
if #captions > 0 then
captionRow = captionRow .. string.format("%s | ", captions[k])
end
imRow = imRow .. string.format(" | ", imPath .. '/' .. ims[k])
if k >= #ims then break end
end
if #captions > 0 then
htmlStr = htmlStr .. "" .. captionRow .. "
\n"
end
htmlStr = htmlStr .. "" .. imRow .. "
\n"
if k >= #ims then break end
end
return htmlStr .. "
\n"
end
function www_utils.renderKeyValues(t)
-- Write all key_value pairs
local htmlStr = "\n| Key | Value |
\n"
for k, v in pairs(t) do
htmlStr = htmlStr .. "| " .. k .. " | " .. v .. " |
\n"
end
return htmlStr .. "
\n"
end
function www_utils.renderHeader()
return [[
]]
end
function www_utils.renderHtml(rootDir, ims, captions, ncol, isNorm, width, htmlFile)
captions = captions or {}
ncol = ncol or 8
isNorm = isNorm or false
width = width or 256
htmlFile = htmlFile or 'index.html'
local imRelativeDir = './im'
local imDir = paths.concat(rootDir, imRelativeDir)
lfs.mkdir(rootDir)
lfs.mkdir(imDir)
local count = 0
local f = io.open(paths.concat(rootDir, htmlFile), "w")
f:write('\n'..www_utils.renderHeader()..'\n')
local imNames = www_utils.saveIms(ims, imDir, isNorm, count)
local htmlStr = www_utils.renderTables(imNames, captions, ncol, width, imRelativeDir)
f:write(htmlStr .. '\n')
f:close()
return count
end
return www_utils