gui.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
import argparse
import io
import shutil
import sys
import traceback
from pathlib import Path
import winreg

import tkinter as tk
import tkinter.ttk as ttk
from tkinter.filedialog import askopenfilename

from PIL import Image, ImageTk
from scframe import VerticalScrolledFrame

from character import KoikatuCharacter
from save_data import KoikatuSaveData
from resource import Resource as RM
from status import StatusPanel

class PropertyPanel(ttk.Frame):

    def __init__(self, parent, character, *args, **kwargs):
        super().__init__(parent, *args, **kwargs)
        self.character = character

        row = 0
        label1 = ttk.Label(self, text=RM.res('lastname'))
        self._lastname = tk.StringVar(value=character.lastname)
        entry1 = ttk.Entry(self, textvariable=self._lastname)
        label1.grid(row=row, column=0, sticky='E', columnspan=1)
        entry1.grid(row=row, column=1, sticky='W', columnspan=1)

        label2 = ttk.Label(self, text=RM.res('firstname'))
        self._firstname = tk.StringVar(value=character.firstname)
        entry2 = ttk.Entry(self, textvariable=self._firstname)
        label2.grid(row=row, column=2, sticky='E', columnspan=1)
        entry2.grid(row=row, column=3, sticky='W', columnspan=1)

        row = row + 1
        label3 = ttk.Label(self, text=RM.res('nickname'))
        self._nickname = tk.StringVar(value=character.nickname)
        entry3 = ttk.Entry(self, textvariable=self._nickname)
        label3.grid(row=row, column=0, sticky='E', columnspan=1)
        entry3.grid(row=row, column=1, sticky='W', columnspan=1)

        if character.sex == 0:
            label4 = ttk.Label(self, text=RM.res('intellect'))
            self._intellect = tk.StringVar(value=character.intelligence)
            entry4 = ttk.Entry(self, textvariable=self._intellect)
            label4.grid(row=row, column=2, sticky='E', columnspan=1)
            entry4.grid(row=row, column=3, sticky='W', columnspan=1)

            row = row + 1
            label5 = ttk.Label(self, text=RM.res('physical'))
            self._physical = tk.StringVar(value=character.strength)
            entry5 = ttk.Entry(self, textvariable=self._physical)
            label5.grid(row=row, column=0, sticky='E', columnspan=1)
            entry5.grid(row=row, column=1, sticky='W', columnspan=1)

            label6 = ttk.Label(self, text=RM.res('hentai'))
            self._hentai = tk.StringVar(value=character.ero)
            entry6 = ttk.Entry(self, textvariable=self._hentai)
            label6.grid(row=row, column=2, sticky='E', columnspan=1)
            entry6.grid(row=row, column=3, sticky='W', columnspan=1)

        elif character.sex == 1:
            values = (RM.res('male'), RM.res('female'))
            label4 = ttk.Label(self, text=RM.res('sex'))
            self._sex = tk.StringVar(value=values[character.sex])
            entry4 = ttk.Combobox(self, values=values,
                                textvariable=self._sex, state='readonly')
            label4.grid(row=row, column=2, sticky="E", columnspan=1)
            entry4.grid(row=row, column=3, sticky="W", columnspan=1)

            row = row + 1
            values = RM.res('personalities')
            label5 = ttk.Label(self, text=RM.res('personality'))
            self._personality = tk.StringVar(value=values[character.personality])
            entry5 = ttk.Combobox(self, values=values,
                                textvariable=self._personality, state='readonly')
            label5.grid(row=row, column=0, sticky="E", columnspan=1)
            entry5.grid(row=row, column=1, sticky="W", columnspan=1)

            values = RM.res('weak_points')
            label6 = ttk.Label(self, text=RM.res('weak_point'))
            self._weak_point = tk.StringVar(value=values[character.weak_point])
            entry6 = ttk.Combobox(self, values=values,
                                textvariable=self._weak_point, state='readonly')
            label6.grid(row=row, column=2, sticky="E", columnspan=1)
            entry6.grid(row=row, column=3, sticky="W", columnspan=1)

            # answer
            row = row + 1
            self._answers = {}
            frame = ttk.LabelFrame(self, text=RM.res('answer'))
            for i, name in enumerate(character.answers.keys()):
                self._answers[name] = self._make_boolean_prop(frame,
                                                            RM.res(name),
                                                            character.answers[name], i, 5)
            frame.grid(row=row, column=0, columnspan=4, sticky='W')

            # denail
            row = row + 1
            self._denials = {}
            frame = ttk.LabelFrame(self, text=RM.res('denial'))
            for i, name in enumerate(character.denials.keys()):
                self._denials[name] = self._make_boolean_prop(frame,
                                                            RM.res(name),
                                                            character.denials[name], i, 5)
            frame.grid(row=row, column=0, columnspan=4, sticky='W')

            # attribute
            row = row + 1
            self._attributes = {}
            frame = ttk.LabelFrame(self, text=RM.res('attribute'))
            for i, name in enumerate(character.attributes.keys()):
                self._attributes[name] = self._make_boolean_prop(frame,
                                                                RM.res(name),
                                                                character.attributes[name], i, 5)
            frame.grid(row=row, column=0, columnspan=4, sticky='W')

    @property
    def firstname(self):
        return self._firstname.get()

    @property
    def lastname(self):
        return self._lastname.get()
        
    @property
    def nickname(self):
        return self._nickname.get()

    @property
    def intellect(self):
        return self._intellect.get()
    @property
    def physical(self):
        return self._physical.get()
    @property
    def hentai(self):
        return self._hentai.get()


    @property
    def sex(self):
        return [RM.res('male'), RM.res('female')].index(self._sex.get())

    @property
    def personality(self):
        values = RM.res('personalities')
        return values.index(self._personality.get())

    @personality.setter
    def personality(self, value):
        values = RM.res('personalities')
        return self._personality.set(values[value])

    @property
    def weak_point(self):
        values = RM.res('weak_points')
        return values.index(self._weak_point.get())

    @weak_point.setter
    def weak_point(self, value):
        values = RM.res('weak_points')
        self._weak_point.set(values[value])

    @property
    def answers(self):
        return {key:self._answers[key].get() for key in self._answers}

    @property
    def denials(self):
        return {key:self._denials[key].get() for key in self._denials}

    @property
    def attributes(self):
        return {key:self._attributes[key].get() for key in self._attributes}

    def update_character(self, character):
        self._firstname.set(character.firstname)
        self._lastname.set(character.lastname)
        self._nickname.set(character.nickname)

        if character.sex == 0:
            self._intellect.set(character.intelligence)
            self._physical.set(character.strength)
            self._hentai.set(character.ero)
        

        self.personality = character.personality
        self.weak_point = character.weak_point

        for key in self._answers:
            self._answers[key].set(character.answers[key])

        for key in self._denials:
            self._denials[key].set(character.denials[key])

        for key in self._attributes:
                self._attributes[key].set(character.attributes.setdefault(key,False))


    def _make_boolean_prop(self, frame, name, value, i, cols):
        var = tk.BooleanVar()
        checkbox = ttk.Checkbutton(frame, text=name, variable=var)
        var.set(value)
        row = i // cols
        col = i % cols
        checkbox.grid(row=row, column=col, sticky='W')
        return var



class CharacterPanel(ttk.Frame):
    def __init__(self, app, parent, character, *args, **kwargs):
        super().__init__(parent,
                         relief='ridge',
                         *args, **kwargs)
        self.app = app
        self.parent = parent
        self._character = character
        self.dirty = False

        png = Image.open(io.BytesIO(character.png))
        self.image = ImageTk.PhotoImage(png)

        self.photo = tk.Label(self,
                              image=self.image,
                              width=self.image.width(),
                              height=self.image.height())
        self.photo.grid(row=0, column=0, rowspan=3, padx=2, pady=2)
        self.property_panel = PropertyPanel(self, character)
        if character.sex == 1:
            self.property_panel.grid(row=0, column=1, rowspan=1, padx=2, pady=2)
            self.status_panel = StatusPanel(self, character)
            self.status_panel.grid(row=1, column=1, rowspan=1, padx=4, pady=2, sticky='N')
        else:
            self.property_panel.grid(row=0, column=1, rowspan=1, columnspan=2, padx=2, pady=2, sticky='W')


        self._load_btn = ttk.Button(self,
                                    text='Load Character Card',
                                    command=self._open_dialog)
        self._load_btn.grid(row=2, column=1, sticky='W', pady=4)


    @property
    def character(self):
        chara = self._character
        panel = self.property_panel

        chara.firstname = panel.firstname
        chara.lastname = panel.lastname
        chara.nickname = panel.nickname
        if chara.sex == 0:
            chara.intelligence = panel.intellect
            chara.strength = panel.physical
            chara.ero = panel.hentai

        elif chara.sex == 1:
            chara.personality = panel.personality
            chara.weak_point = panel.weak_point
            chara.answers = self.property_panel.answers
            chara.denials = self.property_panel.denials
            chara.attributes = self.property_panel.attributes

            panel = self.status_panel
            chara.feeling = panel.feeling
            chara.m_love  = panel.m_love
            chara.h_count = panel.h_count
            chara.koikatu = panel.koikatu
            chara.lover = panel.relation
            chara.date = panel.date
            chara.intimacy = panel.intimacy

            ac = [
                'mune', 'kokan', 'anal', 'siri', 'tikubi',
                'kokan_piston', 'anal_piston', 'houshi'
            ]
            for name in ac:
                chara.set_ac(name, panel.ac(name))

        self.dirty = True
        return chara

    def _update_character(self, character):
        self._character.custom = character.custom
        self._character.coordinates = character.coordinates
        self._character.parameter = character.parameter
        self._character.status = character.status
        self._character.png_length = character.png_length
        self._character.png = character.png
        self.dirty = True

        png = Image.open(io.BytesIO(character.png))
        self.image = ImageTk.PhotoImage(png)
        self.photo.config(image=self.image)

        self.property_panel.update_character(self._character)

    def _open_dialog(self):
        name = askopenfilename(filetype=[("koikatu card", "*.png")],
                               initialdir=self.app.card_dir)
        if name is not None:
            with open(name, 'rb') as infile:
                chara = KoikatuCharacter(infile, True)
                self._update_character(chara)
        else:
            self.quit()


class App:
    def __init__(self, root, filename, out_filename):
        self.root = root
        self.filename = filename
        self.out_filename = out_filename
        self.save_data = KoikatuSaveData(filename)
        try:
            self.card_dir=get_default_chara_folder()
        except:
            self.card_dir=Path.cwd()

        style = ttk.Style()
        style.configure('.', padding='2 4 2 4')
        frame = VerticalScrolledFrame(self.root)

        self.panels = []
        row_count = 0
        for chara in self.save_data.characters:
                panel = CharacterPanel(self, frame.interior, chara)
                panel.grid(row=int(row_count/2),column = int(row_count%2))
                self.panels.append(panel)
                row_count+=1

        btn_frame = ttk.Frame(self.root)
        save_btn = ttk.Button(btn_frame, text='Save & Quit', command=self.save_and_quit)
        quit_btn = ttk.Button(btn_frame, text='Quit', command=self.quit)
        quit_btn.pack(side='right', pady=2)
        save_btn.pack(side='right', pady=2)


        frame.grid(row=0, column=0)
        btn_frame.grid(row=1, column=0, pady=2, sticky='E')

        y_padding = 4
        width = 1440
        height = 355 * 3 + btn_frame.winfo_height() + y_padding * 2
        WindowsTaskbarHeight=48
        if self.root.winfo_screenheight() - WindowsTaskbarHeight < height:
            height = int(355 * 2) + btn_frame.winfo_height() + y_padding * 2 
        self.root.geometry(f'{width}x{height}')
        


        def _configure(event):
            fh = self.root.winfo_height() - btn_frame.winfo_height() - y_padding 
            frame.canvas.config(height=fh)
        self.root.bind('<Configure>', _configure)


    def save(self):
        for i, panel in enumerate(self.panels):
            chara = panel.character
            if i > 0 and panel.dirty:
                self.save_data.replace(i, chara)

        if self.filename == self.out_filename:
            # create backup
            path = Path(self.filename).resolve()
            backup = path.parent / (path.stem + '.old.dat')
            shutil.copy(self.filename, backup)

        self.save_data.save(self.out_filename)


    def save_and_quit(self, *args):
        self.save()
        self.root.destroy()

    def quit(self, *args):
        self.root.destroy()


    def run(self):
        self.root.mainloop()



def main():
    default_resource = Path(sys.argv[0]).parent / 'resources_ja.json'

    parser = argparse.ArgumentParser()
    parser.add_argument('save_data',
                        help='koikatu save data')
    parser.add_argument('-o', dest='output',
                        help='output file')
    parser.add_argument('-r',
                        dest='resources',
                        default=default_resource,
                        help='resource file name')

    root = tk.Tk()
    root.title('Koikatu Save data editor')

    if len(sys.argv) > 1:
        args = parser.parse_args()
        save_data = args.save_data

        if args.output is not None:
            out_filename = args.output
        else:
            out_filename = save_data
        resources = args.resources
    else:
        try:
            save_folder=get_default_save_folder()
        except:
            save_folder=Path.cwd()
        save_data = askopenfilename(filetype=[("koikatu save data", "*.dat")],
                                    initialdir=save_folder)
        print(save_data)
        if save_data is None or len(save_data) == 0:
            root.destroy()
            sys.exit(-1)

        out_filename = save_data
        resources = default_resource

    RM.load(resources)
    print('out:', out_filename)
    App(root, save_data, out_filename).run()

def get_default_save_folder():
    u"""get default save folder from windows registry """
    path = r'Software\illusion\Koikatu\koikatu'
    key = winreg.OpenKeyEx(winreg.HKEY_CURRENT_USER, path)
    data, regtype = winreg.QueryValueEx(key, 'INSTALLDIR')
    return data+'UserData\save\game'

def get_default_chara_folder():
    u"""get default save folder from windows registry """
    path = r'Software\illusion\Koikatu\koikatu'
    key = winreg.OpenKeyEx(winreg.HKEY_CURRENT_USER, path)
    data, regtype = winreg.QueryValueEx(key, 'INSTALLDIR')
    return data+'UserData\chara'

if __name__ == '__main__':
    try:
        main()
    except Exception as e:
        traceback.print_exc()
        print('push Enter key')
        input()