upscayl/electron/utils/get-arguments.ts

import { getPlatform } from "./get-device-specs";
import { ImageFormat } from "./types";
const slash: string = getPlatform() === "win" ? "\\" : "/";

export const getSingleImageArguments = ({
  inputDir,
  fullfileName,
  outFile,
  modelsPath,
  model,
  scale,
  gpuId,
  saveImageAs,
  customWidth,
}: {
  inputDir: string;
  fullfileName: string;
  outFile: string;
  modelsPath: string;
  model: string;
  scale: any;
  gpuId: string;
  saveImageAs: ImageFormat;
  customWidth: string;
}) => {
  return [
    "-i",
    inputDir + slash + fullfileName,
    "-o",
    outFile,
    customWidth ? "" : `-s ${scale}`,
    "-m",
    modelsPath,
    "-n",
    model,
    gpuId ? `-g ${gpuId}` : "",
    "-f",
    saveImageAs,
    customWidth ? `-w ${customWidth}` : "",
  ];
};

export const getDoubleUpscaleArguments = ({
  inputDir,
  fullfileName,
  outFile,
  modelsPath,
  model,
  gpuId,
  saveImageAs,
  scale,
  customWidth,
}: {
  inputDir: string;
  fullfileName: string;
  outFile: string;
  modelsPath: string;
  model: string;
  gpuId: string;
  saveImageAs: ImageFormat;
  scale: string;
  customWidth: string;
}) => {
  return [
    "-i",
    inputDir + slash + fullfileName,
    "-o",
    outFile,
    customWidth ? "" : `-s ${scale}`,
    "-m",
    modelsPath,
    "-n",
    model,
    gpuId ? `-g ${gpuId}` : "",
    "-f",
    saveImageAs,
    customWidth ? `-w ${customWidth}` : "",
  ];
};

export const getDoubleUpscaleSecondPassArguments = ({
  outFile,
  modelsPath,
  model,
  gpuId,
  saveImageAs,
  scale,
  customWidth,
}: {
  outFile: string;
  modelsPath: string;
  model: string;
  gpuId: string;
  saveImageAs: ImageFormat;
  scale: string;
  customWidth: string;
}) => {
  return [
    "-i",
    outFile,
    "-o",
    outFile,
    "-s",
    customWidth ? "" : `-s ${scale}`,
    "-m",
    modelsPath,
    "-n",
    model,
    gpuId ? `-g ${gpuId}` : "",
    "-f",
    saveImageAs,
    customWidth ? `-w ${customWidth}` : "",
  ];
};

export const getBatchArguments = ({
  inputDir,
  outputDir,
  modelsPath,
  model,
  gpuId,
  saveImageAs,
  scale,
  customWidth,
}: {
  inputDir: string;
  outputDir: string;
  modelsPath: string;
  model: string;
  gpuId: string;
  saveImageAs: ImageFormat;
  scale: string;
  customWidth: string;
}) => {
  return [
    "-i",
    inputDir,
    "-o",
    outputDir,
    customWidth ? "" : `-s ${scale}`,
    "-m",
    modelsPath,
    "-n",
    model,
    gpuId ? `-g ${gpuId}` : "",
    "-f",
    saveImageAs,
    customWidth ? `-w ${customWidth}` : "",
  ];
};