def read_texture(self, filename):
"""
Reads a texture from the Pandora resources repository.
Returns a PNMImage object representing the image data.
Throws a PalettizerException if file could not be found.
:filename: Relative filename pointing to a texture file in the Pandora repository.
"""
full_filename = self.find_texture(filename)
if self.debug:
print(f'Reading {full_filename}...')
# We've found the source file! Let's load it using Panda3D.
img = PNMImage()
img.read(Filename.from_os_specific(full_filename))
needs_alpha_fill = img.num_channels not in (2, 4)
img.set_color_type(4)
if needs_alpha_fill:
# We need an alpha channel no matter what, so if the image does not have one,
# it needs to be filled immediately with opaque pixels as it starts out with transparent pixels
img.alpha_fill(1)
return img
def read_texture(self, filename, alpha=False):
"""
Reads a texture from the model path.
Throws a PalettizerException if file could not be found.
:filename: Relative filename pointing to a texture file in the model path.
:alpha: Do we need an alpha channel?
"""
img = PNMImage()
img.read(Filename.from_os_specific(filename))
if alpha:
needs_alpha_fill = img.num_channels not in (2, 4)
img.set_color_type(4)
if needs_alpha_fill:
# We need an alpha channel no matter what, so if the image does not have one,
# it needs to be filled immediately with opaque pixels as it starts out with transparent pixels
img.alpha_fill(1)
else:
img.set_color_type(3)
return img
def convert_texture(self, texture, model_path=None):
if not self.model_path:
self.print_exc('ERROR: No model path specified in ImageConverter.')
return
tex_path = texture[0]
tex_basename = os.path.splitext(os.path.basename(tex_path))[0]
if not os.path.isabs(tex_path):
if '../' in tex_path and model_path:
# This texture path is using relative paths.
# We assume that the working directory is the model's directory
tex_path = os.path.join(os.path.dirname(model_path), tex_path)
else:
tex_path = os.path.join(self.model_path, tex_path)
tex_path = tex_path.replace('\\', os.sep).replace('/', os.sep)
if not os.path.exists(tex_path):
self.print_exc('ERROR: Could not convert {}: Missing RGB texture!'.format(tex_path))
return
png_tex_path = os.path.join(os.path.dirname(tex_path), tex_basename + '.png')
png_tex_path = png_tex_path.replace('\\', os.sep).replace('/', os.sep)
print('Converting to PNG...', png_tex_path)
if len(texture) == 1:
# Only one texture, we can save this immediately
if tex_path.lower().endswith('.rgb'):
output_img = PNMImage()
output_img.read(Filename.from_os_specific(tex_path))
if output_img.num_channels in (1, 2) and 'golf_ball' not in tex_path and 'roll-o-dex' not in tex_path: # HACK: Toontown
output_img.set_color_type(4)
for i in range(output_img.get_x_size()):
for j in range(output_img.get_y_size()):
output_img.set_alpha(i, j, output_img.get_gray(i, j))
else:
output_img = self.read_texture(tex_path, alpha=False)
elif len(texture) == 2:
img = self.read_texture(tex_path, alpha=True)
# Two textures: the second one should be a RGB file
alpha_path = texture[1]
if not os.path.isabs(alpha_path):
if '../' in alpha_path and model_path:
# This texture path is using relative paths.
# We assume that the working directory is the model's directory
alpha_path = os.path.join(os.path.dirname(model_path), alpha_path)
else:
alpha_path = os.path.join(self.model_path, alpha_path)
alpha_path = alpha_path.replace('\\', os.sep).replace('/', os.sep)
if not os.path.exists(alpha_path):
self.print_exc('ERROR: Could not convert {} with alpha {}: Missing alpha texture!'.format(tex_path, alpha_path))
return
alpha_img = PNMImage()
alpha_img.read(Filename.from_os_specific(alpha_path))
alpha_img = self.resize_image(alpha_img, img.get_x_size(), img.get_y_size())
output_img = PNMImage(img.get_x_size(), img.get_y_size(), 4)
output_img.alpha_fill(1)
output_img.copy_sub_image(img, 0, 0, 0, 0, img.get_x_size(), img.get_y_size())
for i in range(img.get_x_size()):
for j in range(img.get_y_size()):
output_img.set_alpha(i, j, alpha_img.get_gray(i, j))
output_img.write(Filename.from_os_specific(png_tex_path))
