import sys
import struct
from struct import pack
import os
from util import *


nud = open(sys.argv[1], "rb+")

texID = int(sys.argv[2])

#setup are arrays
boneCount = []
vertexStart_array=[]
vertexAmount_array=[]
vertexSize_array=[]
polyStart_array=[]
polyAmount_array=[]
polySize_array=[]
vertexAddStart_array=[]
UVSize_array=[]
polyName_array=[]
singleBind_array=[]
texturePropertiesL1Start_array=[]
texturePropertiesL2Start_array=[]
texturePropertiesL3Start_array=[]
texturePropertiesL4Start_array=[]
textureNuml1_array=[]
textureNuml2_array=[]
textureNuml3_array=[]
textureNuml4_array=[]

nud.seek(0x4)

fileSize = readu32be(nud)
#print("File Size is %i bytes" % (fileSize))
nud.seek(2,1)
polysets = readu16be(nud)
nud.seek(4,1)
polyClumpStart = (readu32be(nud) + 0x30)
polyClumpSize = readu32be(nud)
vertexClumpStart = (polyClumpStart + polyClumpSize)
vertexClumpSize = readu32be(nud)
vertexAddStart = (vertexClumpSize + vertexClumpStart)
vertexAddClumpSize = readu32be(nud)
nameClumpStart = (vertexAddClumpSize + vertexAddStart)
nud.seek(0x10,1)
objCount = 0

for z in range(polysets):
        nud.seek(0x20,1)
        #print(hex(nud.tell()))
        polynamestart = readu32be(nud)
        indentifiera = readu32be(nud)
        singleBind = readu16be(nud)
        polyamount = readu16be(nud)
        positionb = readu32be(nud)
        objCount = (objCount + polyamount)
        for s in range(polyamount):
                polyName_array.append(polynamestart)
                singleBind_array.append(singleBind)
for p in range(objCount):
        polyStart = (readu32be(nud) + polyClumpStart)
        vertexStart = (readu32be(nud) + vertexAddStart)
        vertexAddStart = (readu32be(nud) + vertexAddClumpSize)
        vertexAmount = readu16be(nud)
        #print("There is %i Vertexs" % int(vertexAmount))
        vertexSize = readByte(nud)
        uvSize = readByte(nud)
        textureLayer1Properties = readu32be(nud)
        textureLayer2Properties = readu32be(nud)
        textureLayer3Properties = readu32be(nud)
        textureLayer4Properties = readu32be(nud)
        polyAmount = readu16be(nud)
        #print("There is %i Faces" % int(polyAmount))
        polySize = readByte(nud)
        polyFlag = readByte(nud)
        nud.seek(0xC,1)
        vertexStart_array.append(vertexStart)
        polyStart_array.append(polyStart)
        vertexAddStart_array.append(vertexAddStart)
        vertexAmount_array.append(vertexAmount)
        polyAmount_array.append(polyAmount)
        vertexSize_array.append(vertexSize)
        UVSize_array.append(uvSize)
        polySize_array.append(polySize)
        texturePropertiesL1Start_array.append(textureLayer1Properties)
        texturePropertiesL2Start_array.append(textureLayer2Properties)
        texturePropertiesL3Start_array.append(textureLayer3Properties)
        texturePropertiesL4Start_array.append(textureLayer4Properties)

for z in range(objCount):
        nud.seek(texturePropertiesL1Start_array[z])
        nud.seek(8,1)
        texSomethingCount = readu16be(nud)
        texPropCount = readu16be(nud)
        nud.seek(20,1)
        for x in range(texPropCount):
                nud.seek(2,1)
                texNum = readu16be(nud)
                if texNum != 0:
                        nud.seek(-2,1)
                        store = texID * 4
                        nud.write(pack("B", store))
                        nud.seek(-1,1)
                texNum = readu16be(nud)
                print(hex(texNum))
                if x == 1:
                        textureNuml1_array.append(texNum)
                if x == 2:
                        textureNuml2_array.append(texNum)
                if x == 3:
                        textureNuml3_array.append(texNum)
                if x == 4:
                        textureNuml4_array.append(texNum)
                nud.seek(20,1)
        headerRead = 0x20
        while headerRead == 0x20:
                headerRead = readu32be(nud)
                texPropNameStart = readu32be(nud)
                nud.seek(24,1)
                backUp = nud.tell()
                nud.seek(nameClumpStart + texPropNameStart)
                texPropName = getString(nud)
                nud.seek(backUp)
        if texturePropertiesL2Start_array[z] != 0:
                nud.seek(texturePropertiesL2Start_array[z])
                nud.seek(8,1)
                texSomethingCount = readu16be(nud)
                texPropCount = readu16be(nud)
                nud.seek(20,1)
                for x in range(texPropCount):
                        nud.seek(2,1)
                        texNum = readu16be(nud)
                        nud.seek(20,1)
                headerRead = 0x20
                while headerRead == 0x20:
                        headerRead = readu32be(nud)
                        texPropNameStart = readu32be(nud)
                        nud.seek(24,1)
                        backUp = nud.tell()
                        nud.seek(nameClumpStart + texPropNameStart)
                        texPropName = getString(nud)
                        nud.seek(backUp)
        if texturePropertiesL3Start_array[z] != 0:
                        nud.seek(texturePropertiesL2Start_array[z])
                        nud.seek(8,1)
                        texSomethingCount = readu16be(nud)
                        texPropCount = readu16be(nud)
                        nud.seek(20,1)
                        for x in range(texPropCount):
                                nud.seek(2,1)
                                texNum = readu16be(nud)
                                nud.seek(20,1)
                        headerRead = 0x20
                        while headerRead == 0x20:
                                headerRead = readu32be(nud)
                                texPropNameStart = readu32be(nud)
                                nud.seek(24,1)
                                backUp = nud.tell()
                                nud.seek(nameClumpStart + texPropNameStart)
                                texPropName = getString(nud)
                                nud.seek(backUp)
        if texturePropertiesL4Start_array[z] != 0:
                        nud.seek(texturePropertiesL2Start_array[z])
                        nud.seek(8,1)
                        texSomethingCount = readu16be(nud)
                        texPropCount = readu16be(nud)
                        nud.seek(20,1)
                        for x in range(texPropCount):
                                nud.seek(2,1)
                                texNum = readu16be(nud)
                                nud.seek(20,1)
                        headerRead = 0x20
                        while headerRead == 0x20:
                                headerRead = readu32be(nud)
                                texPropNameStart = readu32be(nud)
                                nud.seek(24,1)
                                backUp = nud.tell()
                                nud.seek(nameClumpStart + texPropNameStart)
                                texPropName = getString(nud)
                                nud.seek(backUp)
nud.close()