Spaces:
Runtime error
Runtime error
File size: 3,635 Bytes
c19ca42 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 |
import os
import platform
import shutil
import subprocess
import sys
import uuid
from tempfile import mkdtemp
from typing import List
import numpy as np
from nodes.log import logger
from ...utils.utils import split_file_path
from ..image_utils import cv_save_image
from .format import SRGB_FORMATS, DxgiFormat
__TEXCONV_DIR = os.path.join(
os.path.dirname(sys.modules["__main__"].__file__), "texconv" # type: ignore
)
__TEXCONV_EXE = os.path.join(__TEXCONV_DIR, "texconv.exe")
def __decode(b: bytes) -> str:
try:
return b.decode(encoding="iso8859-1")
except:
try:
return b.decode(encoding="utf-8")
except:
return str(b)
def __run_texconv(args: List[str], error_message: str):
if platform.system() != "Windows":
# texconv is only supported on Windows.
raise ValueError(
"Texconv is only supported on Windows."
" Reading and writing DDS files is only partially supported on other systems."
)
result = subprocess.run(
[__TEXCONV_EXE, "-nologo", *args],
check=False,
capture_output=True,
)
if result.returncode != 0:
output = (__decode(result.stdout) + __decode(result.stderr)).replace("\r", "")
logger.error(f"chaiNNer: failed to run texconv: texconv: {__TEXCONV_EXE} args: {args} exit code: {result.returncode} output: {output}")
raise ValueError(f"{error_message}: Code {result.returncode}: {output}")
def dds_to_png_texconv(path: str) -> str:
"""
Converts the given DDS file to PNG by creating a temporary PNG file.
"""
prefix = uuid.uuid4().hex
_, basename, _ = split_file_path(path)
tempdir = mkdtemp(prefix="chaiNNer-")
__run_texconv(
[
"-f",
"rgba",
"-srgb",
"-ft",
"png",
"-px",
prefix,
"-o",
tempdir,
path,
],
"Unable to convert DDS",
)
return os.path.join(tempdir, prefix + basename + ".png")
def save_as_dds(
path: str,
image: np.ndarray,
dds_format: DxgiFormat,
mipmap_levels: int = 0,
uniform_weighting: bool = False,
dithering: bool = False,
minimal_compression: bool = False,
maximum_compression: bool = False,
dx9: bool = False,
separate_alpha: bool = False,
):
"""
Saves an image as DDS using texconv.
See the following page for more information on save options:
https://github.com/Microsoft/DirectXTex/wiki/Texconv
"""
target_dir, name, ext = split_file_path(path)
assert ext == ".dds", "The file to save must end with '.dds'"
tempDir = mkdtemp(prefix="chaiNNer-")
try:
tempPng = os.path.join(tempDir, f"{name}.png")
cv_save_image(tempPng, image, [])
args = [
"-y",
"-f",
dds_format,
"-dx9" if dx9 else "-dx10",
"-m",
str(mipmap_levels),
# use texconv to directly produce the target file
"-o",
target_dir,
]
bc = ""
bc += "u" if uniform_weighting else ""
bc += "d" if dithering else ""
bc += "q" if minimal_compression else ""
bc += "x" if maximum_compression else ""
if bc:
args.extend(["-bc", f"-{bc}"])
if dds_format in SRGB_FORMATS:
args.append("-srgbi")
if separate_alpha:
args.append("-sepalpha")
args.append(tempPng)
__run_texconv(args, "Unable to write DDS")
finally:
shutil.rmtree(tempDir)
|