#!/usr/bin/env python3
# -*- coding: utf-8 -*-

# ZetpeR xax007@yandex.ru
# The Simpsons Virtual Springfield

import os, sys, io, struct
from PIL import Image
import numpy as np
import zlib

NAME = "The Simpsons Virtual Springfield"
TYPES_FILES = [('vsa Images', ('*.vsa')),('vsb Images', ('*.vsb')),('vs3 Images', ('*.vs3'))]
FORMATS_FILES = ["vsa","vsb","vs3"]
GAMES = ["The Simpsons Virtual Springfield"]
AUTHOR = "ZetpeR xax007@yandex.ru"

class Game_Res:
    def __init__(self,app):
        self.file = None
        self.data = []
        self.app = app
        self.sup_formats = ["vsa",
                            "vsb",
                            "rut",]
        self.sup_types = {"vsa":2,
                          "vsb":1,
                          "vs3":2}
        self.images = []
        self.sound = None
        self.fwav = io.BytesIO() # Соединенные звуки

    def open_files(self,files):
        self.data = files
    
    def unpack(self,data_res):
        self.images = []
        self.sound = None
        
        name = data_res[0]
        size = data_res[2]
        format = data_res[3]
        f = open(name,"rb")
        f2 = io.BytesIO(f.read(size))
        f.close()

        if format == "vsa":
            self.Unpack_VSA(f2,name)
        elif format == "vsb":
            self.Unpack_VSB(f2)
        elif format == "vs3":
            self.Unpack_VS3(f2)

    def Unpack_VSA(self, f,name):
        dirname, filename = os.path.split(name)
        ss = 0 # Номера картинки Нужно чтоб исправить ошибку в файле 109_11F.VSA
        f.seek(0,2)
        end_f = f.tell()
        f.seek(0)
        type = f.read(4)
        if type != b'VSA_': # Проверка
            print("ЭТО НЕ КАРТИНКИ",type)
            return(0)

        f2 = io.BytesIO()
        for i in range(256):
            rgb = f.read(3)
            a = f.read(1)
            f2.write(rgb)
            if i == 15: # Номер цвета прозрачности
                f2.write(b'\x00')
            else:
                f2.write(b'\xFF') # Специально записано 255 чтоб было видно картинку
        f2.seek(0) 

        Pal = b''
        for i in range(256):
            Pal += f2.read(3)
            f2.read(1)
        f2.close()
                
        while True:
            if f.tell() == end_f: # Если достигли конца файла
                break
            tip = struct.unpack("<H",f.read(2))[0] # Тип файлов
            if tip == 64 or tip == 256: # Сжатая картинка
                ss += 1
                unclear = struct.unpack("<I",f.read(4))[0] # Непонятно может тут 2 числа по 2 байта
                w,h = struct.unpack("HH",f.read(4)) # Ширина высота
                size = struct.unpack("<I",f.read(4))[0] # Размер сжатых данных
                #print(w,h,"Шир Выс,Размер сжатых данных",size)
                #print("Позиция",f.tell()-2,"Тип",tip,"Непонятно",unclear)
                f.read(18) # Всегда нули

                if size == 0: # Если размер сжатых данных равен нулю то начать сначало
                    continue

                if filename == "109_11F.VSA": # Сделано чтоб исправить ошибку в файле 109_11F.VSA
                    if ss == 4: 
                        w = 320
                        h = 155
                    elif ss == 5:
                        w = 208
                        h = 126
                    elif ss == 6:
                        w = 88
                        h = 142
                    elif ss == 7:
                        w = 64
                        h = 112
                    elif ss == 8:
                        w = 72
                        h = 70
                    elif ss == 9:
                        w = 48
                        h = 45
                    elif ss == 10:
                        w = 48
                        h = 19
                    elif ss == 11:
                        w = 8
                        h = 1
                
                size_comp = f.tell()+size # Узнаем конец сжатых данных
                self.Unpacking(f,size_comp,Pal,w,h)

            elif tip == 128: # Звук
                f.read(8) # Всегда нули
                size = struct.unpack("<I",f.read(4))[0] # Размер файла
                f.read(18) # Всегда нули
                fd = f.read(size)
                self.fwav.write(fd)

            else:
                print("Не понятно какой тип ",tip,"позиция",f.tell()-2)
                break

        self.Unpack_NEW_WAV()
        f.close()

    def Unpacking(self, f,size_comp,Pal,w,h):
        f4 = io.BytesIO()
        while True:
            if f.tell() == size_comp: # Конец файла
                break 
            compr = struct.unpack("B",f.read(1))[0] # Байт управления
            if compr & 128 == 128: # Если левый бит 1
                #col = compr & 0x7f # ПЕРВОЕ ЗНАЧЕНИЕ ПРОСТО ЧИСЛО  НАПРИМЕР 81      0x7f
                col = 256 - (compr-1) # -1 надо для правельного расчёта
                fd = f.read(1) # Читаем байт который надо повторить
                for i in range(col):
                    f4.write(fd)
                    
            else: # Байт не сжат
                fd = f.read(compr+1) # Читаем байты
                f4.write(fd)

        f4.seek(0)
        f_image = Image.frombuffer('P', (w,h), f4.read(w*h), 'raw', 'P', 0, 1) 
        f_image.putpalette(Pal)
        self.images.append(f_image)
        f4.close()
        
    def Unpack_NEW_WAV(self):
        size = self.fwav.tell()
        if size > 0: # Если записан звук
            self.fwav.seek(0)
            fd = self.fwav.read()

            wav = b""
            wav += b"RIFF"
            wav += struct.pack("<I", size+44-8)# chunkSize размер файла-8
            wav += b"WAVE" # format WAVE
            wav += b"fmt " # subchunk1Id fmt 0x666d7420    
            subchunk1Size = 16 
            audioFormat = 1 
            numChannels = 1    # Количество каналов
            sampleRate = 22050 # Частота файла
            byteRate = 22050   # Частота выхода звука, для расчёта длины звучания
            blockAlign = 2
            bitsPerSample = 8 # Битность звука
            wav += struct.pack("<IHHIIHH", subchunk1Size, audioFormat, numChannels, sampleRate, byteRate, blockAlign, bitsPerSample)
            wav += b"data"
            wav += struct.pack("<I", size)
            wav += fd # Данные
            f3 = io.BytesIO(wav)
            self.sound = f3
            self.fwav = io.BytesIO() # Чистим память
            
    def pal_data(self): # Палитра
        pal_data = b'x\x9cc``\xd8\xcf\x00\xc2\xfb\xc1\xd4~0\xeb\x00\x10\xdc9\xb0\xec\xd4\x07\x06Tp\xee\xce\xe3\xd5\xabWs\xef2`\x90V^\xba\xf4\xff\x94\xaeg\xcd\xf9g\x8bB\xb6$Z\xcc\x0e\x90m\xb2g\xca\xfc\xdd\x7f88\xfa\xff\x8f\xf3\xcc\xcf\x9f\xff\xdf\x18\xe3\xf1\x1f\xac\x8b\x87\x81\x81\x9f\x8bUTTTXQHA\x86WNNNYH\xceL\\\xdcXI\xc6V\x8b\xd5\xcdZ\xdc\xc7X&\xcaP8DR\xc0_D\xc4G@>[F\xa6H\x92/\x8f\x83/]\x84\xa1S\x9d\xa1\xceE\xb2\xdeU\xad\xd3Rm\xb2\xa7\xe1R/\xe5\xb9\x91\n\x8bb\xf4\xb6\xa5*\xaf,f\xdb\xe5\xc7p\xc6\x95\xfd\xa45\xc3zk\xe3\x95bFG\x99\xd4\xef\xb1j>\xe3\xe0}\xc7\xc2\xb2\x99A\xa8\x9aA=\xcb\xd44\xdb\xc4\xaa\xd0C/\xd5E1\xdaK\xae*Cuz\x93Na\xbfC\xd0\x820\x9fR\x1b\xe5bC\xe9&U\x86`}\x86\x08WA\x17\x1b\x0e\x0bG>S\x1f\tCO\x0e}S\x06##\x06k#\x06\x139\x06\x19\x19fI\x19.e]\x06y\x0b&\xee\x08\x19\xceP\x13\x06+~\x069>\x06)\r\x06\x069K9ccc777C\x970M\xafd\x93\x80\x02\xdd\xf0"\xe5\xa4\x99\xea&\xab\x18\xccN\xb2f\xdfbH}\xc5\x19\xf7\x8d\xd5\xeb\xbfh\xcf\x19\x86\xfa\xb5\x12\x8b\x9b\x04\xebs\xa2&\xf7\x94L\x9c\xd9\xd9\xbf\xa2\xafj\xd5\x84\t\x13f\xcf\x9e\xbdh\xd1\xa2\x15+V\xac\x9d\xbe{{\xcf\xdee\x99\xfb**\x1eE\xc5^\xf6\xafX_\x94\xd6\\PP\x10\x1d\x1d]\xe0\xef_\xe3\xee\xde\xea\xef69)\xa9\xaf bW\x9e\xd7\x9e\xba\xa4\xed\xd3\xb2/\xcf*z\xd7^\xf9iF\xf7\xef\xe5\xb5\xff\x17\xc7\xfe\xdf\x9b\xf8\xe1\xa2\xff\xff\x87N\x7f\x1f\xe6\xfc\xbf\xdc\xf1\xff\xd0\xdc\xaf\xdb\x16\xfc_\xb5\xe3\xfd\x8c\xad\xff{\xd6\xfdoZ\xf0\xbf~\xfe\xff\x8a\xdeO\x99]\xff\xc3\xbb\xfeGU}\xf6\xa9\xf9bS\xf7\xdf \xe2\xbfP\xc5q\xb1\xac\xa5\xe6\xd3\xda]\xfb\x92\xddz\xca,\xd3\x02m\x93\x9c\xd43\xfc\xd4\x9d\xd3\xb9\xb3\x03\x18\x9a\x19D\xaa\nO\xac\x9drt\xe7\xdc3;\x96\x9c\xd8\xbdv\xff\x05`\x1a\xb9~\xfd\xfa\x91KON\x9f\xf9pz\xeb\x8b\xf5\x9b^\xf5\xce\xf8\xefw\xe6\xbf\xd6\xde\xff\xb5_\xfe\xbf|\xf9\xeb\xff\xff\xffO\xff\xdc|\xf0\xe7\xfc\xe7\xbbG>\xdd\xde\xf8\xff\xdb\x96\xff\xe7\xae\xff?\xf5\xe1E\xe7\xdd\x8b[\xfav\xcd\xefY~ \xb3\xe9t\xc6\x81\x03\xa6\x0f7I?\x9e\xca\xf1p:\xcb\xff^\x96\xffej?\x92\xc4\xff\x87\xb1\xfc\xf7\xe1z\x15\xac{7)\xe4F}\xd8\xff\xda\x88\x97\x8b\\\xce\xbdr^xO\xbat/\xc3\xff\x13L\xff\x9f1\x10\t\xfe\xff\xfe\xb0`\xc1\x92\x86\x86\x06Pj\xfc\xcf\xf0\x1fL\xfd\x07\xb3\x80\x00\x00P\xf8D\xaf'
        fd = zlib.decompress(pal_data)
        f4 = io.BytesIO(fd) 
        Pal = f4.read(768)
        f4.close()
        return(Pal)
            
    def Unpack_VSB(self, f):
        Pal = self.pal_data() # Если понадобится внешняя палитра
        f.seek(0,2)
        end_f = f.tell() # Узнаёт конец файла
        f.seek(0)
        type = f.read(4)
        if type != b'VSB_': # Проверка
            print("ЭТО НЕ ЗАДНИЙ ФОН",type)
            return(0)
        
        f.seek(16)
        w,h = struct.unpack("HH",f.read(4)) # Ширина  и Высота
        #print("Ширина и высота",w,h)
        f.seek(132) # Начало сжатых данных 132
        
        f2 = io.BytesIO()
        while True:
            if f2.tell() == w*h: # Остановить распаковку
                break
            compr = f.read(1)
            if compr[0] & 192 == 192: # Проверяет два бита в байте слева
                col = compr[0] - 192 # Количество повторений
                fd = f.read(1) # Повторить байт
                for i in range(col):
                    f2.write(fd)
     
            else: # Если байт не сжат прочто запись
                f2.write(compr)
        #print("Конец распаковки",f.tell())
        
        posf0 = f.tell()
        unclear = struct.unpack("B",f.read(1))[0] # Непонятный байт
        ostatok = end_f-f.tell()
        f.seek(posf0)
        check = f.read(4) # Байты проверки

        #print("Осталось в конце файла байт",ostatok,"Непонятный байт",unclear)
        if ostatok == 768: # Палитра
            f.seek(posf0+1)
            Pal = f.read(768)

        elif check == b'VSB_':
            #print("Тут VSB ещё сжатая картинка или палитра ?")
            # Используется внешняя палитра pal_data
            pass
        else:
            print("НЕПОНЯТНО №№№№№№№№№№№№№№№№№№№№№№№№№№№№№№№№№№№№")

        f2.seek(0)
        f_image = Image.frombuffer('P', (w,h), f2.read(w*h), 'raw', 'P', 0, 1) 
        f_image.putpalette(Pal)
        self.images.append(f_image)
        f2.close()
        f.close()

    def Unpack_VS3(self, f):
        Pal = self.pal_data() # Если понадобится внешняя палитра
    
        f.seek(0,2) # Перейти на конец файла
        end_f = f.tell() # Узнаёт длину файла
        f.seek(8)
        w,h = struct.unpack("HH",f.read(4)) # Ширина и высота
        #print("Ширина и высота картинки",w,h,"Непонятные байты",unclear_1,unclear_2)

        f.seek(144) # Начало сжатых данных
        sizef = struct.unpack("<I",f.read(4))[0]
        f.read(3)
        fd = f.read(sizef-7) # ВОЗМОЖНО НЕПРАВЕЛЬНЫЙ РАСЧЁТ
        f4 = io.BytesIO(fd) # Нужно чтоб правельно прочетать файл FOX.VS3
        f5 = io.BytesIO() # Картинка распакованная
        sizeline = w # Чтоб прочитать непонятный байт
        while True:
            if f5.tell() == w*h: # Остановка когда вся картинка распаковалась
                break 
            if f5.tell() == sizeline: # Размер строчки
                sizeline += w
                unclear = struct.unpack("B",f4.read(1))[0] # Непонятный байт
            compr = struct.unpack("B",f4.read(1))[0] # Управляющий байт
            if compr & 128 == 0: # Проверяет 1 бит в байте слева
                fd = f4.read(1) # Читаем байт который надо повторить
                for i in range(compr):
                    f5.write(fd)

            else: # Просто прочетать байты
                fd = f4.read(256 - compr)
                f5.write(fd)
        f4.close()
        
        f5.seek(0) # Картинка первый кадр
        f_image = Image.frombuffer('P', (w,h), f5.read(w*h), 'raw', 'P', 0, 1) 
        f_image.putpalette(Pal)
        self.images.append(f_image)
        #print("Позиция после распакованного файла",f.tell(),"\n")
        f5.close()
        
        while True:
            poz = f.tell()
            if poz == end_f: # Остановка в конце файла
                #print("Конец файла",poz)
                break

            unclear = struct.unpack("<I",f.read(4))[0] # Непонятно
            check = f.read(2) # b'\xFA\xF1' может быть Частота размер 22050
            bit = struct.unpack("<B",f.read(1))[0] # 0 или 1
            #print("Непонятно",unclear,"Позиция",poz)

            if check == b'\x22\x56': # Если ровно частоте 22050 Звук в файлах DELOGO.VS3, FOX.VS3, VORTEX.VS3
                #print("Звук")
                f.seek(poz+16)
                fd = f.read(22050)
                self.fwav.write(fd)

            elif check == b'\xFA\xF1' and bit == 1:
                #print("Данные FA F1 Позиция",poz)
                fd = f.read(9) # Нули
                sizecomp = struct.unpack("<I",f.read(4))[0] # Размер блока кадра
                f.seek(poz+sizecomp+16)
                #print("Позиция после чтения кадра",f.tell(),"\n")

            elif check == b'\xFA\xF1' and bit == 0:
                #print("Чтение блока данных без информации",poz,"\n")
                f.seek(poz)
                f.read(16)

            else:
                print("Непонятно",poz)
                return(0)

        self.Unpack_NEW_WAV()
        f.close()