mirror of
https://github.com/IgorTimofeev/MineOS.git
synced 2025-12-22 03:59:21 +01:00
378 lines
11 KiB
Lua
378 lines
11 KiB
Lua
local fs = require("filesystem")
|
||
local unicode = require("unicode")
|
||
local gpu = require("component").gpu
|
||
|
||
local image = {}
|
||
|
||
local transparentSymbol = "#"
|
||
|
||
--------------------Все, что касается сжатого формата изображений (у нас он назван "JPG")----------------------------------------------------------------------------------
|
||
|
||
-- OC image format .ocif by Pirnogion
|
||
-- Спасибо, Пир
|
||
-- Охуенный форматик
|
||
local ocif_signature1 = 0x896F6369
|
||
local ocif_signature2 = 0x00661A0A --7 bytes: 89 6F 63 69 66 1A 0A
|
||
local ocif_signature_expand = { string.char(0x89), string.char(0x6F), string.char(0x63), string.char(0x69), string.char(0x66), string.char(0x1A), string.char(0x0A) }
|
||
|
||
local BYTE = 8
|
||
local NULL_CHAR = 0
|
||
|
||
local imageAPI = {}
|
||
|
||
local function readBytes(file, bytes)
|
||
local readedByte = 0
|
||
local readedNumber = 0
|
||
for i = bytes, 1, -1 do
|
||
readedByte = string.byte( file:read(1) or NULL_CHAR )
|
||
readedNumber = readedNumber + bit32.lshift(readedByte, i*8-8)
|
||
end
|
||
|
||
return readedNumber
|
||
end
|
||
|
||
local function HEXtoRGB(color)
|
||
local rr = bit32.rshift( color, 16 )
|
||
local gg = bit32.rshift( bit32.band(color, 0x00ff00), 8 )
|
||
local bb = bit32.band(color, 0x0000ff)
|
||
|
||
return rr, gg, bb
|
||
end
|
||
|
||
local function encodePixel(hexcolor_fg, hexcolor_bg, char)
|
||
local rr_fg, gg_fg, bb_fg = HEXtoRGB( hexcolor_fg )
|
||
local rr_bg, gg_bg, bb_bg = HEXtoRGB( hexcolor_bg )
|
||
local ascii_char1, ascii_char2 = string.byte( char, 1, 2 )
|
||
|
||
ascii_char1 = ascii_char1 or NULL_CHAR
|
||
ascii_char2 = ascii_char2 or NULL_CHAR
|
||
|
||
return rr_fg, gg_fg, bb_fg, rr_bg, gg_bg, bb_bg, ascii_char1, ascii_char2
|
||
end
|
||
|
||
local function decodeChar(char1, char2)
|
||
if ( char1 ~= 0 and char2 ~= 0 ) then
|
||
return string.char( char1, char2 )
|
||
elseif ( char1 ~= 0) then
|
||
return string.char( char1 )
|
||
elseif ( char2 ~= 0 ) then
|
||
return string.char( char2 )
|
||
end
|
||
end
|
||
|
||
--Конвертируем массив классического "сырого" формата в сжатый и оптимизированный
|
||
function image.convertImagetoGroupedImage(PNGMassiv)
|
||
local newPNGMassiv = { ["backgrounds"] = {} }
|
||
|
||
--Перебираем весь массив стандартного PNG-вида по высоте
|
||
for j = 1, #PNGMassiv do
|
||
for i = 1, #PNGMassiv[j] do
|
||
local back = PNGMassiv[j][i][1]
|
||
local fore = PNGMassiv[j][i][2]
|
||
local symbol = PNGMassiv[j][i][3]
|
||
|
||
newPNGMassiv["backgrounds"][back] = newPNGMassiv["backgrounds"][back] or {}
|
||
newPNGMassiv["backgrounds"][back][fore] = newPNGMassiv["backgrounds"][back][fore] or {}
|
||
|
||
table.insert(newPNGMassiv["backgrounds"][back][fore], {i, j, symbol} )
|
||
|
||
back, fore, symbol = nil, nil, nil
|
||
end
|
||
end
|
||
|
||
return newPNGMassiv
|
||
end
|
||
|
||
--Чтение сжатого формата
|
||
local function loadJPG(path)
|
||
local image = {}
|
||
local file = io.open(path, "rb")
|
||
|
||
local signature1, signature2 = readBytes(file, 4), readBytes(file, 3)
|
||
if ( signature1 ~= ocif_signature1 or signature2 ~= ocif_signature2 ) then
|
||
file:close()
|
||
return nil
|
||
end
|
||
|
||
image.width = readBytes(file, 1)
|
||
image.height = readBytes(file, 1)
|
||
image.depth = readBytes(file, 1)
|
||
|
||
for y = 1, image.height, 1 do
|
||
table.insert( image, {} )
|
||
for x = 1, image.width, 1 do
|
||
table.insert( image[y], {} )
|
||
image[y][x][2] = readBytes(file, 3)
|
||
image[y][x][1] = readBytes(file, 3)
|
||
image[y][x][3] = decodeChar(readBytes(file, 1), readBytes(file, 1))
|
||
end
|
||
end
|
||
|
||
file:close()
|
||
|
||
return image
|
||
end
|
||
|
||
--Рисование сжатого формата
|
||
function image.drawJPG(x, y, image)
|
||
x = x - 1
|
||
y = y - 1
|
||
|
||
local image2 = convertImagetoGroupedImage(image)
|
||
|
||
--Перебираем массив с фонами
|
||
for back, backValue in pairs(image2["backgrounds"]) do
|
||
gpu.setBackground(back)
|
||
for fore, foreValue in pairs(image2["backgrounds"][back]) do
|
||
gpu.setForeground(fore)
|
||
for pixel = 1, #image2["backgrounds"][back][fore] do
|
||
if image2["backgrounds"][back][fore][pixel][3] ~= transparentSymbol then
|
||
gpu.set(x + image2["backgrounds"][back][fore][pixel][1], y + image2["backgrounds"][back][fore][pixel][2], image2["backgrounds"][back][fore][pixel][3])
|
||
end
|
||
end
|
||
end
|
||
end
|
||
end
|
||
|
||
--Сохранение JPG в файл из существующего массива
|
||
function image.saveJPG(path, image)
|
||
|
||
-- Удаляем файл, если есть
|
||
-- И делаем папку к нему
|
||
fs.remove(path)
|
||
fs.makeDirectory(fs.path(path))
|
||
|
||
local file = io.open(path, "w")
|
||
|
||
--print("width = ", image.width)
|
||
|
||
file:write( table.unpack(ocif_signature_expand) )
|
||
file:write( string.char( image.width ) )
|
||
file:write( string.char( image.height ) )
|
||
file:write( string.char( image.depth ) )
|
||
|
||
for y = 1, image.height, 1 do
|
||
for x = 1, image.width, 1 do
|
||
local encodedPixel = { encodePixel( image[y][x][2], image[y][x][1], image[y][x][3] ) }
|
||
for i = 1, #encodedPixel do
|
||
file:write( string.char( encodedPixel[i] ) )
|
||
end
|
||
end
|
||
end
|
||
|
||
file:close()
|
||
end
|
||
|
||
---------------------------Все, что касается несжатого формата (у нас он назван "PNG")-------------------------------------------------------
|
||
|
||
-- Перевод HEX-цвета из файла (из 00ff00 делает 0x00ff00)
|
||
local function HEXtoSTRING(color,withNull)
|
||
local stro4ka = string.format("%x",color)
|
||
local sStro4ka = unicode.len(stro4ka)
|
||
|
||
if sStro4ka < 6 then
|
||
stro4ka = string.rep("0", 6 - sStro4ka) .. stro4ka
|
||
end
|
||
|
||
if withNull then return "0x"..stro4ka else return stro4ka end
|
||
end
|
||
|
||
--Загрузка ПНГ
|
||
local function loadPNG(path)
|
||
local file = io.open(path, "r")
|
||
local newPNGMassiv = { ["backgrounds"] = {} }
|
||
|
||
local pixelCounter, lineCounter = 1, 1
|
||
for line in file:lines() do
|
||
local dlinaStroki = unicode.len(line)
|
||
pixelCounter = 1
|
||
|
||
for i = 1, dlinaStroki, 16 do
|
||
local back = tonumber("0x"..unicode.sub(line, i, i + 5))
|
||
local fore = tonumber("0x"..unicode.sub(line, i + 7, i + 12))
|
||
local symbol = unicode.sub(line, i + 14, i + 14)
|
||
|
||
newPNGMassiv["backgrounds"][back] = newPNGMassiv["backgrounds"][back] or {}
|
||
newPNGMassiv["backgrounds"][back][fore] = newPNGMassiv["backgrounds"][back][fore] or {}
|
||
|
||
table.insert(newPNGMassiv["backgrounds"][back][fore], {pixelCounter, lineCounter, symbol} )
|
||
|
||
pixelCounter = pixelCounter + 1
|
||
back, fore, symbol = nil, nil, nil
|
||
end
|
||
|
||
lineCounter = lineCounter + 1
|
||
end
|
||
|
||
file:close()
|
||
pixelCounter, lineCounter = nil, nil
|
||
|
||
return newPNGMassiv
|
||
end
|
||
|
||
-- Сохранение существующего массива ПНГ в файл
|
||
function image.savePNG(path, MasterPixels)
|
||
-- Удаляем файл, если есть
|
||
-- И делаем папку к нему
|
||
fs.remove(path)
|
||
fs.makeDirectory(fs.path(path))
|
||
local f = io.open(path, "w")
|
||
|
||
for j=1, #MasterPixels do
|
||
for i=1,#MasterPixels[j] do
|
||
f:write(HEXtoSTRING(MasterPixels[j][i][1])," ",HEXtoSTRING(MasterPixels[j][i][2])," ",MasterPixels[j][i][3]," ")
|
||
end
|
||
f:write("\n")
|
||
end
|
||
|
||
f:close()
|
||
end
|
||
|
||
--Отрисовка ПНГ
|
||
function image.drawPNG(x, y, massivSudaPihay)
|
||
--Уменьшаем значения кординат на 1, т.к. циклы начинаются с единицы
|
||
x = x - 1
|
||
y = y - 1
|
||
|
||
--Конвертируем "сырой" формат PNG в оптимизированный и сгруппированный по цветам
|
||
--local massivSudaPihay = convertImagetoGroupedImage(massivSudaPihay2)
|
||
--Более не требуется, встроил конвертер в загрузчик файлов, по сути
|
||
--конвертации теперь вообще нет.
|
||
|
||
--Перебираем массив с фонами
|
||
for back, backValue in pairs(massivSudaPihay["backgrounds"]) do
|
||
gpu.setBackground(back)
|
||
for fore, foreValue in pairs(massivSudaPihay["backgrounds"][back]) do
|
||
gpu.setForeground(fore)
|
||
for pixel = 1, #massivSudaPihay["backgrounds"][back][fore] do
|
||
if massivSudaPihay["backgrounds"][back][fore][pixel][3] ~= transparentSymbol then
|
||
gpu.set(x + massivSudaPihay["backgrounds"][back][fore][pixel][1], y + massivSudaPihay["backgrounds"][back][fore][pixel][2], massivSudaPihay["backgrounds"][back][fore][pixel][3])
|
||
end
|
||
end
|
||
end
|
||
end
|
||
end
|
||
|
||
---------------------Глобальные функции данного API, с ними мы и работаем---------------------------------------------------------
|
||
|
||
--Конвертер из PNG в JPG
|
||
function image.PNGtoJPG(PNGMassiv)
|
||
local JPGMassiv = {}
|
||
local width, height = 0, 0
|
||
|
||
--Сохраняем пиксели
|
||
for j = 1, #PNGMassiv do
|
||
JPGMassiv[j] = {}
|
||
width = 0
|
||
for i = 1, #PNGMassiv[j] do
|
||
JPGMassiv[j][i] = { table.unpack(PNGMassiv[j][i]) }
|
||
width = width + 1
|
||
end
|
||
height = height + 1
|
||
end
|
||
|
||
JPGMassiv["width"] = width
|
||
JPGMassiv["height"] = height
|
||
JPGMassiv["depth"] = 8
|
||
|
||
return JPGMassiv
|
||
end
|
||
|
||
--Конвертер из JPG в PNG
|
||
function image.JPGtoPNG(JPGMassiv)
|
||
local PNGMassiv = {}
|
||
local width, height = 0, 0
|
||
|
||
--Сохраняем пиксели
|
||
for j = 1, #JPGMassiv do
|
||
PNGMassiv[j] = {}
|
||
width = 0
|
||
for i = 1, #JPGMassiv[j] do
|
||
PNGMassiv[j][i] = { table.unpack(JPGMassiv[j][i]) }
|
||
width = width + 1
|
||
end
|
||
height = height + 1
|
||
end
|
||
|
||
return PNGMassiv
|
||
end
|
||
|
||
-- Просканировать файловую систему на наличие .PNG
|
||
-- И сохранить рядом с ними аналогичную копию в формате .JPG
|
||
-- Осторожно, функция для дебага и знающих людей
|
||
-- С кривыми ручками сюда не лезь
|
||
function image.convertAllPNGtoJPG(path)
|
||
local list = ecs.getFileList(path)
|
||
for key, file in pairs(list) do
|
||
if fs.isDirectory(path.."/"..file) then
|
||
image.convertAllPNGtoJPG(path.."/"..file)
|
||
else
|
||
if ecs.getFileFormat(file) == ".png" or ecs.getFileFormat(file) == ".PNG" then
|
||
print("Найден .PNG в директории \""..path.."/"..file.."\"")
|
||
print("Загружаю этот файл...")
|
||
PNGFile = loadPNG(path.."/"..file)
|
||
print("Загрузка завершена!")
|
||
print("Конвертация в JPG начата...")
|
||
JPGFile = image.PNGtoJPG(PNGFile)
|
||
print("Ковертация завершена!")
|
||
print("Сохраняю .JPG в той же папке...")
|
||
image.saveJPG(path.."/"..ecs.hideFileFormat(file)..".jpg", JPGFile)
|
||
print("Сохранение завершено!")
|
||
print(" ")
|
||
end
|
||
end
|
||
end
|
||
end
|
||
|
||
--Загрузка любого изображения из доступных типов
|
||
function image.load(path)
|
||
|
||
local kartinka = {}
|
||
local fileFormat = ecs.getFileFormat(path)
|
||
|
||
if string.lower(fileFormat) == ".jpg" then
|
||
kartinka["format"] = ".jpg"
|
||
kartinka["image"] = loadJPG(path)
|
||
elseif string.lower(fileFormat) == ".png" then
|
||
kartinka["format"] = ".png"
|
||
kartinka["image"] = loadPNG(path)
|
||
else
|
||
ecs.error("Wrong file format! (not .png or .jpg)")
|
||
end
|
||
|
||
return kartinka
|
||
end
|
||
|
||
--Отрисовка этого изображения
|
||
function image.draw(x, y, kartinka)
|
||
if kartinka.format == ".jpg" then
|
||
image.drawJPG(x, y, kartinka["image"])
|
||
elseif kartinka.format == ".png" then
|
||
image.drawPNG(x, y, kartinka["image"])
|
||
end
|
||
end
|
||
|
||
---------------------------------------------------------------------------------------------------------------------
|
||
|
||
-- ecs.prepareToExit()
|
||
-- local cyka1 = image.load("System/OS/Icons/Love.png")
|
||
-- local cyka2 = image.load("Pastebin.app/Resources/Icon.png")
|
||
-- image.draw(2, 2, cyka1)
|
||
-- image.draw(40, 2, cyka2)
|
||
|
||
|
||
return image
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|
||
|