195 lines
7.9 KiB
Python
195 lines
7.9 KiB
Python
|
|
# bitmap image conversion predates the import/export system so it's a bit weird.
|
|
# conversion happens over time, so it merely kicks off the process.
|
|
|
|
import os
|
|
from PIL import Image
|
|
|
|
from ui_file_chooser_dialog import ImageFileChooserDialog
|
|
from ui_dialog import UIDialog, Field
|
|
from ui_art_dialog import ImportOptionsDialog
|
|
from image_convert import ImageConverter
|
|
from art_import import ArtImporter
|
|
from palette import PaletteFromFile
|
|
from art import DEFAULT_CHARSET, DEFAULT_PALETTE, DEFAULT_WIDTH, DEFAULT_HEIGHT
|
|
|
|
# custom chooser showing image previews, shares parent w/ "palette from image"
|
|
|
|
class ConvertImageChooserDialog(ImageFileChooserDialog):
|
|
|
|
title = 'Convert image'
|
|
confirm_caption = 'Choose'
|
|
|
|
def confirm_pressed(self):
|
|
filename = self.field_texts[0]
|
|
if not os.path.exists(filename) or not os.path.isfile(filename):
|
|
return
|
|
self.ui.app.last_import_dir = self.current_dir
|
|
self.dismiss()
|
|
# get dialog box class and invoke it
|
|
dialog_class = self.ui.app.importer.options_dialog_class
|
|
# tell the dialog which image we chose, store its size
|
|
w, h = Image.open(filename).size
|
|
options = {
|
|
'filename': filename,
|
|
'image_width': w,
|
|
'image_height': h
|
|
}
|
|
self.ui.open_dialog(dialog_class, options)
|
|
|
|
|
|
# custom dialog box providing convert options
|
|
|
|
class ConvertImageOptionsDialog(ImportOptionsDialog):
|
|
|
|
title = 'Convert bitmap image options'
|
|
field0_label = 'Color palette:'
|
|
field1_label = 'Current palette (%s)'
|
|
field2_label = 'From source image; # of colors:'
|
|
field3_label = ' '
|
|
field5_label = 'Converted art size:'
|
|
field6_label = 'Best fit to current size (%s)'
|
|
field7_label = '%% of source image: (%s)'
|
|
field8_label = ' '
|
|
field10_label = 'Smooth (bicubic) scale source image'
|
|
radio_groups = [(1, 2), (6, 7)]
|
|
field_width = UIDialog.default_short_field_width
|
|
# to get the layout we want, we must specify 0 padding lines and
|
|
# add some blank ones :/
|
|
y_spacing = 0
|
|
fields = [
|
|
Field(label=field0_label, type=None, width=0, oneline=True),
|
|
Field(label=field1_label, type=bool, width=0, oneline=True),
|
|
Field(label=field2_label, type=bool, width=0, oneline=True),
|
|
Field(label=field3_label, type=int, width=field_width, oneline=True),
|
|
Field(label='', type=None, width=0, oneline=True),
|
|
Field(label=field5_label, type=None, width=0, oneline=True),
|
|
Field(label=field6_label, type=bool, width=0, oneline=True),
|
|
Field(label=field7_label, type=bool, width=0, oneline=True),
|
|
Field(label=field8_label, type=float, width=field_width, oneline=True),
|
|
Field(label='', type=None, width=0, oneline=True),
|
|
Field(label=field10_label, type=bool, width=0, oneline=True),
|
|
Field(label='', type=None, width=0, oneline=True)
|
|
]
|
|
invalid_color_error = 'Palettes must be between 2 and 256 colors.'
|
|
invalid_scale_error = 'Scale must be greater than 0.0'
|
|
# redraw dynamic labels
|
|
always_redraw_labels = True
|
|
|
|
def get_initial_field_text(self, field_number):
|
|
if field_number == 1:
|
|
return UIDialog.true_field_text
|
|
elif field_number == 3:
|
|
# # of colors from source image
|
|
return '64'
|
|
elif field_number == 6:
|
|
return UIDialog.true_field_text
|
|
elif field_number == 8:
|
|
# % of source image size
|
|
return '50.0'
|
|
elif field_number == 10:
|
|
return ' '
|
|
return ''
|
|
|
|
def get_field_label(self, field_index):
|
|
label = self.fields[field_index].label
|
|
# custom label replacements to show palette, possible convert sizes
|
|
if field_index == 1:
|
|
label %= self.ui.active_art.palette.name if self.ui.active_art else DEFAULT_PALETTE
|
|
elif field_index == 6:
|
|
# can't assume any art is open, use defaults if needed
|
|
w = self.ui.active_art.width if self.ui.active_art else DEFAULT_WIDTH
|
|
h = self.ui.active_art.height if self.ui.active_art else DEFAULT_HEIGHT
|
|
label %= '%s x %s' % (w, h)
|
|
elif field_index == 7:
|
|
# scale # might not be valid
|
|
valid,_ = self.is_input_valid()
|
|
if not valid:
|
|
return label % '???'
|
|
label %= '%s x %s' % self.get_tile_scale()
|
|
return label
|
|
|
|
def get_tile_scale(self):
|
|
"returns scale in tiles of image dimensions"
|
|
# filename won't be set just after dialog is created
|
|
if not hasattr(self, 'filename'):
|
|
return 0, 0
|
|
scale = float(self.field_texts[8]) / 100
|
|
# can't assume any art is open, use defaults if needed
|
|
if self.ui.active_art:
|
|
cw = self.ui.active_art.charset.char_width
|
|
ch = self.ui.active_art.charset.char_height
|
|
else:
|
|
charset = self.ui.app.load_charset(DEFAULT_CHARSET)
|
|
cw, ch = charset.char_width, charset.char_height
|
|
width = self.image_width / cw
|
|
height = self.image_height / ch
|
|
width *= scale
|
|
height *= scale
|
|
return int(width), int(height)
|
|
|
|
def is_input_valid(self):
|
|
# colors: int between 2 and 256
|
|
try: int(self.field_texts[3])
|
|
except: return False, self.invalid_color_error
|
|
colors = int(self.field_texts[3])
|
|
if colors < 2 or colors > 256:
|
|
return False, self.invalid_color_error
|
|
# % scale: >0 float
|
|
try: float(self.field_texts[8])
|
|
except: return False, self.invalid_scale_error
|
|
if float(self.field_texts[8]) <= 0:
|
|
return False, self.invalid_scale_error
|
|
return True, None
|
|
|
|
def confirm_pressed(self):
|
|
valid, reason = self.is_input_valid()
|
|
if not valid: return
|
|
self.dismiss()
|
|
# compile options for importer
|
|
options = {}
|
|
# create new palette from image?
|
|
if self.field_texts[1].strip():
|
|
options['palette'] = self.ui.active_art.palette.name if self.ui.active_art else DEFAULT_PALETTE
|
|
else:
|
|
# create new palette
|
|
palette_filename = os.path.basename(self.filename)
|
|
colors = int(self.field_texts[3])
|
|
new_pal = PaletteFromFile(self.ui.app, self.filename,
|
|
palette_filename, colors)
|
|
# palette now loaded and saved to disk
|
|
options['palette'] = new_pal.name
|
|
# rescale art?
|
|
if self.field_texts[6].strip():
|
|
options['art_width'] = self.ui.active_art.width if self.ui.active_art else DEFAULT_WIDTH
|
|
options['art_height'] = self.ui.active_art.height if self.ui.active_art else DEFAULT_HEIGHT
|
|
else:
|
|
# art dimensions = scale% of image dimensions, in tiles
|
|
options['art_width'], options['art_height'] = self.get_tile_scale()
|
|
options['bicubic_scale'] = bool(self.field_texts[10].strip())
|
|
ImportOptionsDialog.do_import(self.ui.app, self.filename, options)
|
|
|
|
|
|
class BitmapImageImporter(ArtImporter):
|
|
format_name = 'Bitmap image'
|
|
format_description = """
|
|
Bitmap image in PNG, JPEG, or BMP format.
|
|
"""
|
|
file_chooser_dialog_class = ConvertImageChooserDialog
|
|
options_dialog_class = ConvertImageOptionsDialog
|
|
completes_instantly = False
|
|
|
|
def run_import(self, in_filename, options={}):
|
|
# modify self.app.ui.active_art based on options
|
|
palette = self.app.load_palette(options['palette'])
|
|
self.art.set_palette(palette)
|
|
width, height = options['art_width'], options['art_height']
|
|
self.art.resize(width, height) # Importer.init will adjust UI
|
|
bicubic_scale = options['bicubic_scale']
|
|
# let ImageConverter do the actual heavy lifting
|
|
ic = ImageConverter(self.app, in_filename, self.art, bicubic_scale)
|
|
# early failures: file no longer exists, PIL fails to load and convert image
|
|
if not ic.init_success:
|
|
return False
|
|
self.app.update_window_title()
|
|
return True
|