mirror of
https://github.com/upscayl/upscayl.git
synced 2024-11-28 01:10:52 +01:00
512 lines
20 KiB
JavaScript
512 lines
20 KiB
JavaScript
"use strict";
|
|
var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
|
|
function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
|
|
return new (P || (P = Promise))(function (resolve, reject) {
|
|
function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
|
|
function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
|
|
function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
|
|
step((generator = generator.apply(thisArg, _arguments || [])).next());
|
|
});
|
|
};
|
|
var __importDefault = (this && this.__importDefault) || function (mod) {
|
|
return (mod && mod.__esModule) ? mod : { "default": mod };
|
|
};
|
|
Object.defineProperty(exports, "__esModule", { value: true });
|
|
// Native
|
|
const path_1 = require("path");
|
|
const url_1 = require("url");
|
|
const child_process_1 = require("child_process");
|
|
const fs_1 = __importDefault(require("fs"));
|
|
const electron_updater_1 = require("electron-updater");
|
|
const getPlatform_1 = __importDefault(require("./getPlatform"));
|
|
const upscayl_ffmpeg_1 = __importDefault(require("upscayl-ffmpeg"));
|
|
const binaries_1 = require("./binaries");
|
|
// Packages
|
|
const electron_1 = require("electron");
|
|
const electron_is_dev_1 = __importDefault(require("electron-is-dev"));
|
|
const electron_next_1 = __importDefault(require("electron-next"));
|
|
const commands_1 = __importDefault(require("./commands"));
|
|
// Prepare the renderer once the app is ready
|
|
let mainWindow;
|
|
electron_1.app.on("ready", () => __awaiter(void 0, void 0, void 0, function* () {
|
|
yield (0, electron_next_1.default)("./renderer");
|
|
console.log("🚀 ICON PATH: ", (0, path_1.join)(__dirname, "build", "icon.png"));
|
|
console.log("🚀 UPSCAYL EXEC PATH: ", (0, binaries_1.execPath)(""));
|
|
console.log("🚀 MODELS PATH: ", binaries_1.modelsPath);
|
|
console.log("🚀 FFMPEG PATH: ", upscayl_ffmpeg_1.default.path);
|
|
mainWindow = new electron_1.BrowserWindow({
|
|
icon: (0, path_1.join)(__dirname, "build", "icon.png"),
|
|
width: 1300,
|
|
height: 940,
|
|
minHeight: 500,
|
|
minWidth: 500,
|
|
show: false,
|
|
backgroundColor: "#171717",
|
|
webPreferences: {
|
|
nodeIntegration: true,
|
|
webSecurity: false,
|
|
preload: (0, path_1.join)(__dirname, "preload.js"),
|
|
},
|
|
});
|
|
const url = electron_is_dev_1.default
|
|
? "http://localhost:8000"
|
|
: (0, url_1.format)({
|
|
pathname: (0, path_1.join)(__dirname, "../renderer/out/index.html"),
|
|
protocol: "file:",
|
|
slashes: true,
|
|
});
|
|
mainWindow.setMenuBarVisibility(false);
|
|
mainWindow.loadURL(url);
|
|
mainWindow.webContents.setWindowOpenHandler(({ url }) => {
|
|
electron_1.shell.openExternal(url);
|
|
return { action: "deny" };
|
|
});
|
|
mainWindow.once("ready-to-show", () => {
|
|
mainWindow.show();
|
|
mainWindow.webContents.setZoomFactor(1);
|
|
});
|
|
if (!electron_is_dev_1.default) {
|
|
electron_updater_1.autoUpdater.checkForUpdates();
|
|
}
|
|
}));
|
|
// Quit the app once all windows are closed
|
|
electron_1.app.on("window-all-closed", electron_1.app.quit);
|
|
console.log(electron_1.app.getAppPath());
|
|
//------------------------Select File-----------------------------//
|
|
// ! DONT FORGET TO RESTART THE APP WHEN YOU CHANGE CODE HERE
|
|
electron_1.ipcMain.handle(commands_1.default.SELECT_FILE, () => __awaiter(void 0, void 0, void 0, function* () {
|
|
const { canceled, filePaths } = yield electron_1.dialog.showOpenDialog({
|
|
properties: ["openFile", "multiSelections"],
|
|
});
|
|
if (canceled) {
|
|
console.log("operation cancelled");
|
|
return "cancelled";
|
|
}
|
|
else {
|
|
console.log(filePaths[0]);
|
|
// CREATE input AND upscaled FOLDER
|
|
return filePaths[0];
|
|
}
|
|
}));
|
|
//------------------------Select Folder-----------------------------//
|
|
electron_1.ipcMain.handle(commands_1.default.SELECT_FOLDER, (event, message) => __awaiter(void 0, void 0, void 0, function* () {
|
|
const { canceled, filePaths } = yield electron_1.dialog.showOpenDialog({
|
|
properties: ["openDirectory"],
|
|
});
|
|
if (canceled) {
|
|
console.log("operation cancelled");
|
|
return "cancelled";
|
|
}
|
|
else {
|
|
console.log(filePaths[0]);
|
|
return filePaths[0];
|
|
}
|
|
}));
|
|
//------------------------Open Folder-----------------------------//
|
|
electron_1.ipcMain.on(commands_1.default.OPEN_FOLDER, (event, payload) => __awaiter(void 0, void 0, void 0, function* () {
|
|
console.log(payload);
|
|
electron_1.shell.openPath(payload);
|
|
}));
|
|
//------------------------Double Upscayl-----------------------------//
|
|
electron_1.ipcMain.on(commands_1.default.DOUBLE_UPSCAYL, (event, payload) => __awaiter(void 0, void 0, void 0, function* () {
|
|
const model = payload.model;
|
|
let inputDir = payload.imagePath.match(/(.*)[\/\\]/)[1] || "";
|
|
let outputDir = payload.outputPath;
|
|
const gpuId = payload.gpuId;
|
|
const saveImageAs = payload.saveImageAs;
|
|
// COPY IMAGE TO TMP FOLDER
|
|
const platform = (0, getPlatform_1.default)();
|
|
const fullfileName = platform === "win"
|
|
? payload.imagePath.split("\\").slice(-1)[0]
|
|
: payload.imagePath.split("/").slice(-1)[0];
|
|
const fileName = (0, path_1.parse)(fullfileName).name;
|
|
const fileExt = (0, path_1.parse)(fullfileName).ext;
|
|
const outFile = outputDir + "/" + fileName + "_upscayl_8x_" + model + "." + saveImageAs;
|
|
// UPSCALE
|
|
let upscayl = (0, child_process_1.spawn)((0, binaries_1.execPath)("realesrgan"), [
|
|
"-i",
|
|
inputDir + "/" + fullfileName,
|
|
"-o",
|
|
outFile,
|
|
"-s",
|
|
4,
|
|
"-m",
|
|
binaries_1.modelsPath,
|
|
"-n",
|
|
model,
|
|
gpuId ? `-g ${gpuId}` : "",
|
|
"-f",
|
|
saveImageAs,
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
console.log("🆙 COMMAND:", "-i", inputDir + "/" + fullfileName, "-o", outFile, "-s", 4, "-m", binaries_1.modelsPath, "-n", model, gpuId ? `-g ${gpuId}` : "", "-f", saveImageAs);
|
|
let failed = false;
|
|
// TAKE UPSCAYL OUTPUT
|
|
upscayl.stderr.on("data", (data) => {
|
|
// CONVERT DATA TO STRING
|
|
data = data.toString();
|
|
// PRINT TO CONSOLE
|
|
console.log(data);
|
|
// SEND UPSCAYL PROGRESS TO RENDERER
|
|
mainWindow.webContents.send(commands_1.default.DOUBLE_UPSCAYL_PROGRESS, data);
|
|
// IF PROGRESS HAS ERROR, UPSCAYL FAILED
|
|
if (data.includes("invalid gpu") || data.includes("failed")) {
|
|
failed = true;
|
|
}
|
|
});
|
|
// IF ERROR
|
|
upscayl.on("error", (data) => {
|
|
data.toString();
|
|
// SEND UPSCAYL PROGRESS TO RENDERER
|
|
mainWindow.webContents.send(commands_1.default.DOUBLE_UPSCAYL_PROGRESS, data);
|
|
// SET FAILED TO TRUE
|
|
failed = true;
|
|
return;
|
|
});
|
|
// ON UPSCAYL DONE
|
|
upscayl.on("close", (code) => {
|
|
// IF NOT FAILED
|
|
if (!failed) {
|
|
// UPSCALE
|
|
let upscayl2 = (0, child_process_1.spawn)((0, binaries_1.execPath)("realesrgan"), [
|
|
"-i",
|
|
outFile,
|
|
"-o",
|
|
outFile,
|
|
"-s",
|
|
4,
|
|
"-m",
|
|
binaries_1.modelsPath,
|
|
"-n",
|
|
model,
|
|
gpuId ? `-g ${gpuId}` : "",
|
|
"-f",
|
|
saveImageAs,
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
let failed2 = false;
|
|
// TAKE UPSCAYL OUTPUT
|
|
upscayl2.stderr.on("data", (data) => {
|
|
// CONVERT DATA TO STRING
|
|
data = data.toString();
|
|
// PRINT TO CONSOLE
|
|
console.log(data);
|
|
// SEND UPSCAYL PROGRESS TO RENDERER
|
|
mainWindow.webContents.send(commands_1.default.DOUBLE_UPSCAYL_PROGRESS, data);
|
|
// IF PROGRESS HAS ERROR, UPSCAYL FAILED
|
|
if (data.includes("invalid gpu") || data.includes("failed")) {
|
|
failed2 = true;
|
|
}
|
|
});
|
|
// IF ERROR
|
|
upscayl2.on("error", (data) => {
|
|
data.toString();
|
|
// SEND UPSCAYL PROGRESS TO RENDERER
|
|
mainWindow.webContents.send(commands_1.default.DOUBLE_UPSCAYL_PROGRESS, data);
|
|
// SET FAILED TO TRUE
|
|
failed2 = true;
|
|
return;
|
|
});
|
|
upscayl2.on("close", (code) => {
|
|
if (!failed2) {
|
|
console.log("Done upscaling");
|
|
mainWindow.webContents.send(commands_1.default.DOUBLE_UPSCAYL_DONE, outFile);
|
|
}
|
|
});
|
|
}
|
|
});
|
|
}));
|
|
//------------------------Image Upscayl-----------------------------//
|
|
electron_1.ipcMain.on(commands_1.default.UPSCAYL, (event, payload) => __awaiter(void 0, void 0, void 0, function* () {
|
|
const model = payload.model;
|
|
const scale = payload.scaleFactor;
|
|
const gpuId = payload.gpuId;
|
|
const saveImageAs = payload.saveImageAs;
|
|
let inputDir = payload.imagePath.match(/(.*)[\/\\]/)[1] || "";
|
|
let outputDir = payload.outputPath;
|
|
// COPY IMAGE TO TMP FOLDER
|
|
const fullfileName = payload.imagePath.replace(/^.*[\\\/]/, "");
|
|
console.log(fullfileName);
|
|
const fileName = (0, path_1.parse)(fullfileName).name;
|
|
const fileExt = (0, path_1.parse)(fullfileName).ext;
|
|
const outFile = model.includes("models-DF2K")
|
|
? outputDir +
|
|
"/" +
|
|
fileName +
|
|
"_upscayl_sharpened_" +
|
|
scale +
|
|
"x_" +
|
|
model +
|
|
"." +
|
|
saveImageAs
|
|
: outputDir +
|
|
"/" +
|
|
fileName +
|
|
"_upscayl_" +
|
|
scale +
|
|
"x_" +
|
|
model +
|
|
"." +
|
|
saveImageAs;
|
|
// UPSCALE
|
|
if (fs_1.default.existsSync(outFile)) {
|
|
// If already upscayled, just output that file
|
|
mainWindow.webContents.send(commands_1.default.UPSCAYL_DONE, outFile);
|
|
}
|
|
else {
|
|
let upscayl = null;
|
|
switch (model) {
|
|
default:
|
|
upscayl = (0, child_process_1.spawn)((0, binaries_1.execPath)("realesrgan"), [
|
|
"-i",
|
|
inputDir + "/" + fullfileName,
|
|
"-o",
|
|
outFile,
|
|
"-s",
|
|
scale === 2 ? 4 : scale,
|
|
"-m",
|
|
binaries_1.modelsPath,
|
|
"-n",
|
|
model,
|
|
gpuId ? `-g ${gpuId}` : "",
|
|
"-f",
|
|
saveImageAs,
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
console.log("🆙 COMMAND: ", "-i", inputDir + "/" + fullfileName, "-o", outFile, "-s", scale === 2 ? 4 : scale, "-m", binaries_1.modelsPath, "-n", model, gpuId ? `-g ${gpuId}` : "", "-f", saveImageAs);
|
|
break;
|
|
case "models-DF2K":
|
|
upscayl = (0, child_process_1.spawn)((0, binaries_1.execPath)("realsr"), [
|
|
"-i",
|
|
inputDir + "/" + fullfileName,
|
|
"-o",
|
|
outFile,
|
|
"-s",
|
|
scale,
|
|
"-x",
|
|
"-m",
|
|
binaries_1.modelsPath + "/" + model,
|
|
gpuId ? `-g ${gpuId}` : "",
|
|
"-f",
|
|
saveImageAs,
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
console.log("🆙 COMMAND: ", "-i", inputDir + "/" + fullfileName, "-o", outFile, "-s", scale, "-x", "-m", binaries_1.modelsPath + "/" + model, gpuId ? `-g ${gpuId}` : "", "-f", saveImageAs);
|
|
break;
|
|
}
|
|
let failed = false;
|
|
upscayl === null || upscayl === void 0 ? void 0 : upscayl.stderr.on("data", (data) => {
|
|
console.log("🚀 => upscayl.stderr.on => stderr.toString()", data.toString());
|
|
data = data.toString();
|
|
mainWindow.webContents.send(commands_1.default.UPSCAYL_PROGRESS, data.toString());
|
|
if (data.includes("invalid gpu") || data.includes("failed")) {
|
|
failed = true;
|
|
}
|
|
});
|
|
upscayl === null || upscayl === void 0 ? void 0 : upscayl.on("error", (data) => {
|
|
mainWindow.webContents.send(commands_1.default.UPSCAYL_PROGRESS, data.toString());
|
|
failed = true;
|
|
return;
|
|
});
|
|
// Send done comamnd when
|
|
upscayl === null || upscayl === void 0 ? void 0 : upscayl.on("close", (code) => {
|
|
if (failed !== true) {
|
|
console.log("Done upscaling");
|
|
mainWindow.webContents.send(commands_1.default.UPSCAYL_DONE, outFile);
|
|
}
|
|
});
|
|
}
|
|
}));
|
|
//------------------------Upscayl Folder-----------------------------//
|
|
electron_1.ipcMain.on(commands_1.default.FOLDER_UPSCAYL, (event, payload) => __awaiter(void 0, void 0, void 0, function* () {
|
|
// GET THE MODEL
|
|
const model = payload.model;
|
|
const gpuId = payload.gpuId;
|
|
const saveImageAs = payload.saveImageAs;
|
|
// GET THE IMAGE DIRECTORY
|
|
let inputDir = payload.batchFolderPath;
|
|
console.log("🚀 => file: index.ts => line 471 => inputDir", inputDir);
|
|
// GET THE OUTPUT DIRECTORY
|
|
let outputDir = model.includes("models-DF2K")
|
|
? payload.outputPath + "_sharpened"
|
|
: payload.outputPath;
|
|
console.log("🚀 => file: index.ts => line 474 => outputDir", outputDir);
|
|
if (!fs_1.default.existsSync(outputDir)) {
|
|
fs_1.default.mkdirSync(outputDir, { recursive: true });
|
|
}
|
|
// UPSCALE
|
|
let upscayl = null;
|
|
switch (model) {
|
|
default:
|
|
upscayl = (0, child_process_1.spawn)((0, binaries_1.execPath)("realesrgan"), [
|
|
"-i",
|
|
inputDir,
|
|
"-o",
|
|
outputDir,
|
|
"-s",
|
|
4,
|
|
"-m",
|
|
binaries_1.modelsPath,
|
|
"-n",
|
|
model,
|
|
gpuId ? `-g ${gpuId}` : "",
|
|
"-f",
|
|
saveImageAs,
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
console.log("🆙 COMMAND:", "-i", inputDir, "-o", outputDir, "-s", 4, "-m", binaries_1.modelsPath, "-n", model, gpuId ? `-g ${gpuId}` : "", "-f", saveImageAs);
|
|
break;
|
|
case "models-DF2K":
|
|
upscayl = (0, child_process_1.spawn)((0, binaries_1.execPath)("realsr"), [
|
|
"-i",
|
|
inputDir,
|
|
"-o",
|
|
outputDir,
|
|
"-s",
|
|
4,
|
|
"-x",
|
|
"-m",
|
|
binaries_1.modelsPath + "/" + model,
|
|
gpuId ? `-g ${gpuId}` : "",
|
|
"-f",
|
|
saveImageAs,
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
console.log("🆙 COMMAND:", "-i", inputDir, "-o", outputDir, "-s", 4, "-x", "-m", binaries_1.modelsPath + "/" + model, gpuId ? `-g ${gpuId}` : "", "-f", saveImageAs);
|
|
break;
|
|
}
|
|
let failed = false;
|
|
upscayl === null || upscayl === void 0 ? void 0 : upscayl.stderr.on("data", (data) => {
|
|
console.log("🚀 => upscayl.stderr.on => stderr.toString()", data.toString());
|
|
data = data.toString();
|
|
mainWindow.webContents.send(commands_1.default.FOLDER_UPSCAYL_PROGRESS, data.toString());
|
|
if (data.includes("invalid gpu") || data.includes("failed")) {
|
|
failed = true;
|
|
}
|
|
});
|
|
upscayl === null || upscayl === void 0 ? void 0 : upscayl.on("error", (data) => {
|
|
mainWindow.webContents.send(commands_1.default.FOLDER_UPSCAYL_PROGRESS, data.toString());
|
|
failed = true;
|
|
return;
|
|
});
|
|
// Send done comamnd when
|
|
upscayl === null || upscayl === void 0 ? void 0 : upscayl.on("close", (code) => {
|
|
if (failed !== true) {
|
|
console.log("Done upscaling");
|
|
mainWindow.webContents.send(commands_1.default.FOLDER_UPSCAYL_DONE, outputDir);
|
|
}
|
|
});
|
|
}));
|
|
//------------------------Video Upscayl-----------------------------//
|
|
electron_1.ipcMain.on(commands_1.default.UPSCAYL_VIDEO, (event, payload) => __awaiter(void 0, void 0, void 0, function* () {
|
|
// Extract the model
|
|
const model = payload.model;
|
|
// Extract the Video Directory
|
|
let videoFileName = payload.videoPath.replace(/^.*[\\\/]/, "");
|
|
const justFileName = (0, path_1.parse)(videoFileName).name;
|
|
let inputDir = payload.videoPath.match(/(.*)[\/\\]/)[1] || "";
|
|
console.log("🚀 => file: index.ts => line 337 => inputDir", inputDir);
|
|
// Set the output directory
|
|
let outputDir = payload.outputPath + "_frames";
|
|
console.log("🚀 => file: index.ts => line 340 => outputDir", outputDir);
|
|
let frameExtractionPath = (0, path_1.join)(inputDir, justFileName + "_f");
|
|
let frameUpscalePath = (0, path_1.join)(inputDir, justFileName + "_u");
|
|
console.log("🚀 => file: index.ts => line 342 => frameExtractionPath", frameExtractionPath, frameUpscalePath);
|
|
if (!fs_1.default.existsSync(frameExtractionPath)) {
|
|
fs_1.default.mkdirSync(frameExtractionPath, { recursive: true });
|
|
}
|
|
if (!fs_1.default.existsSync(frameUpscalePath)) {
|
|
fs_1.default.mkdirSync(frameUpscalePath, { recursive: true });
|
|
}
|
|
let ffmpegProcess = null;
|
|
ffmpegProcess = (0, child_process_1.spawn)(upscayl_ffmpeg_1.default.path, [
|
|
"-i",
|
|
inputDir + "/" + videoFileName,
|
|
frameExtractionPath + "/" + "out%d.png",
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
let failed = false;
|
|
ffmpegProcess === null || ffmpegProcess === void 0 ? void 0 : ffmpegProcess.stderr.on("data", (data) => {
|
|
console.log("🚀 => file: index.ts:420 => data", data.toString());
|
|
data = data.toString();
|
|
mainWindow.webContents.send(commands_1.default.FFMPEG_VIDEO_PROGRESS, data.toString());
|
|
});
|
|
ffmpegProcess === null || ffmpegProcess === void 0 ? void 0 : ffmpegProcess.on("error", (data) => {
|
|
mainWindow.webContents.send(commands_1.default.FFMPEG_VIDEO_PROGRESS, data.toString());
|
|
failed = true;
|
|
return;
|
|
});
|
|
// Send done comamnd when
|
|
ffmpegProcess === null || ffmpegProcess === void 0 ? void 0 : ffmpegProcess.on("close", (code) => {
|
|
if (failed !== true) {
|
|
console.log("Frame extraction successful!");
|
|
mainWindow.webContents.send(commands_1.default.FFMPEG_VIDEO_DONE, outputDir);
|
|
// UPSCALE
|
|
let upscayl = null;
|
|
upscayl = (0, child_process_1.spawn)((0, binaries_1.execPath)("realesrgan"), [
|
|
"-i",
|
|
frameExtractionPath,
|
|
"-o",
|
|
frameUpscalePath,
|
|
"-s",
|
|
4,
|
|
"-m",
|
|
binaries_1.modelsPath,
|
|
"-n",
|
|
model,
|
|
], {
|
|
cwd: undefined,
|
|
detached: false,
|
|
});
|
|
upscayl === null || upscayl === void 0 ? void 0 : upscayl.stderr.on("data", (data) => {
|
|
console.log("🚀 => upscayl.stderr.on => stderr.toString()", data.toString());
|
|
data = data.toString();
|
|
mainWindow.webContents.send(commands_1.default.FFMPEG_VIDEO_PROGRESS, data.toString());
|
|
});
|
|
}
|
|
});
|
|
}));
|
|
//------------------------Auto-Update Code-----------------------------//
|
|
// ! AUTO UPDATE STUFF
|
|
electron_updater_1.autoUpdater.on("update-available", ({ releaseNotes, releaseName }) => {
|
|
const dialogOpts = {
|
|
type: "info",
|
|
buttons: ["Ok"],
|
|
title: "Application Update",
|
|
message: process.platform === "win32"
|
|
? releaseNotes
|
|
: releaseName,
|
|
detail: "A new version is being downloaded.",
|
|
};
|
|
electron_1.dialog.showMessageBox(dialogOpts).then((returnValue) => { });
|
|
});
|
|
electron_updater_1.autoUpdater.on("update-downloaded", (event) => {
|
|
const dialogOpts = {
|
|
type: "info",
|
|
buttons: ["Restart", "Later"],
|
|
title: "Application Update",
|
|
message: process.platform === "win32"
|
|
? event.releaseNotes
|
|
: event.releaseName,
|
|
detail: "A new version has been downloaded. Restart the application to apply the updates.",
|
|
};
|
|
electron_1.dialog.showMessageBox(dialogOpts).then((returnValue) => {
|
|
if (returnValue.response === 0)
|
|
electron_updater_1.autoUpdater.quitAndInstall();
|
|
});
|
|
});
|