def load_single_model(data, mdlList):
    '''Loads a single model. For testing purposes'''
    
    if rapi.checkFileExt(rapi.getInputName(), ".mbn"):
        bs = NoeBitStream(data)
        idstring = bs.readUInt()
        numFiles = bs.readUInt()
        bs.read('2L')
        filesizes = []
        for i in range(numFiles):
            unk, size, crc, null = bs.read('4L')
            filesizes.append(size)
            
        for i in range(numFiles):
            size = filesizes[i]
            mdl = bs.readBytes(size)
            parser = mdl_parser(mdl)
            parser.parse_file()
            mdl = rapi.rpgConstructModel()
            mdl.setModelMaterials(NoeModelMaterials(parser.texList, parser.matList))
            mdlList.append(mdl)            
        
    elif rapi.checkFileExt(rapi.getInputName(), ".mdl"):
        parser = mdl_parser(data)
        parser.parse_file()
        mdl = rapi.rpgConstructModel()
        mdl.setModelMaterials(NoeModelMaterials(parser.texList, parser.matList))
        mdlList.append(mdl)

 def findMODL(self):
     fName=rapi.getInputName().split('\\')[-1]
     dirPath = rapi.getDirForFilePath(rapi.getInputName())
     f = open(dirPath + "/" + fName,'rb')
     file = f.read()
     f.close()
     offset = file.find(b"MODL")
     self.data.seek(offset)

    def __init__(self, data):

        self.inFile = NoeBitStream(data)
        self.animList = []
        self.texList = []
        self.matList = []
        self.boneList = []
        self.dirpath = rapi.getDirForFilePath(rapi.getInputName())
        self.basename = rapi.getExtensionlessName(rapi.getInputName())

def noepyCheckType(data):
    fName=rapi.getInputName().split('\\')[-1]
    dirPath = rapi.getDirForFilePath(rapi.getInputName())
    bs = open(dirPath + "/" + fName)
    idstring = line = bs.readline()
    idstring =''.join(idstring.split('\x00'))
    bs.close()
    if not "version" in idstring.lower() : return 0
    return 1

def noepyLoadModel(data, mdlList):
    
    ctx = rapi.rpgCreateContext()
    global dirPath
    global fileName
    dirPath     = rapi.getDirForFilePath(rapi.getInputName())
    fileName    = rapi.getLocalFileName(rapi.getInputName()) 
    file = GOBJ(data)
    mdlList.append(file.mdl)
    rapi.rpgClearBufferBinds()
    return 1

 def __init__(self):
     self.filename       = rapi.getInputName().split('\\')[-1]
     self.dirPath        = rapi.getDirForFilePath(rapi.getInputName())
     self.matList        = []
     self.texList        = []
     self.bones          = []
     self.anims          = []
     self.GetMaterials()
     self.LoadModel()
     try:self.LoadArmature()
     except:print('Armature failed to init');pass

 def __init__(self, data):    
     '''Initialize some data. Refer to Sanae.py to see what is already
     initialized'''
     
     self.inFile = NoeBitStream(data)
     self.meshes = []
     self.boneID = 0
     self.animList = []
     self.texList = []
     self.matList = []
     self.boneList = []
     self.abspath = rapi.getInputName()
     self.dirpath = rapi.getDirForFilePath(rapi.getInputName())

 def GetMaterial(self):
     fName=rapi.getInputName().split('\\')[-1][:-5]
     self.name = fName
     dirPath = rapi.getDirForFilePath(rapi.getInputName())
     self.path = dirPath
     try:
         matFile = open(dirPath+"/"+fName+".material","r")
         while True:
             line = matFile.readline()
             #print(line)
             if "texture " in line: line=line.split(' ')[-1][:-5];print(line);return line
     except:print("[*] Material not found.", );return 0
     print(fName)

    def parse_materials(self, filename):

        dirpath = rapi.getDirForFilePath(rapi.getInputName())
        for i in range(self.numMat):
            name = self.read_name(128)
            matName = self.read_name(128)
            if not matName:
                matName = "material[%d]" % i
            self.inFile.seek(44, 1)
            numTex = self.inFile.readUInt()
            for i in range(numTex):
                if i == 1:
                    self.inFile.read("2L")
                    texName = self.read_name(256)
                    texName = os.path.basename(texName)
                elif i == 2:
                    self.inFile.read("2L")
                    self.read_name(256)
                    os.path.basename(texName)
                else:
                    self.inFile.read("2L")
                    self.read_name(256)

            self.matNames.append(matName)
            self.texNames.append(dirpath + "tex\\" + texName)

 def parse_textures(self, filename):
     
     try:
         dirpath = rapi.getDirForFilePath(rapi.getInputName())
         f = open(dirpath + filename + ".tex", 'rb')
         data = f.read()
         bs = NoeBitStream(data)
         
         header = bs.readBytes(4)
         size, numTex, null = bs.read('3L')
         for i in range(numTex):
             texSize = bs.readUInt()
             texFmt = noeStrFromBytes(bs.readBytes(4))
             if texFmt == "tex":
                 texFmt = "dds"
             bs.read('2L')
             texName = noeStrFromBytes(bs.readBytes(32))
             texData = bs.readBytes(texSize)
             
             tex = rapi.loadTexByHandler(texData, "." + texFmt)
             if tex is not None:
                 tex.name = texName
                 self.texList.append(tex)
     except:
         pass

def noepyLoadModel(data, mdlList):
    '''Build the model, set materials, bones, and animations. You do not
    need all of them as long as they are empty lists (they are by default)'''
    
    ctx = rapi.rpgCreateContext()
    filename = rapi.getLocalFileName(rapi.getInputName())
    fileID = ''.join(c for c in filename if c.isdigit())    
    bs = NoeBitStream(data)
    idstring = bs.readUInt()
    idstring2 = bs.readUInt()
    
    if idstring == 1213416781: #MESH
        if idstring2 == 1:
            parser = StaticParser1(data)
        elif idstring2 == 2:
            parser = StaticParser2(data)
    else:
        parser = SanaeParser(data)
    print(idstring)
    parser.parse_file()
    mdl = rapi.rpgConstructModel()
    mdl.setModelMaterials(NoeModelMaterials(parser.texList, parser.matList))
    mdl.setBones(parser.boneList)
    mdl.setAnims(parser.animList)
    mdlList.append(mdl)
    return 1

def load_all_models(mdlList):
    '''Load all models'''

    #carry over from previous models
    matList = []
    texList = []
    
    ##load face
    #facePath = "E:\\My Documents\\Workspace\\sample\\Age of Wushu\\g_face.xmod"
    #f = open(facePath, 'rb')
    #data2 = f.read()
    #parser = AgeOfWushu_XMOD(data2)
    #parser.parse_file()
    #material = NoeMaterial("g_face", "E:\\My Documents\\Workspace\\sample\\Age of Wushu\\g_face_1.dds")
    #matList.append(material)
    
    #load the outfit
    
    dirPath = rapi.getDirForFilePath(rapi.getInputName())
    fileList = [file for file in os.listdir(dirPath) if file.lower().endswith(".xmod")]    
    for file in fileList:
        f = open(dirPath + file, 'rb')
        data2 = f.read()
        parser = AgeOfWushu_XMOD(data2)
        parser.parse_file()
        matList.extend(parser.matList)
        texList.extend(parser.texList)
        mdl = rapi.rpgConstructModel()
    mdl.setModelMaterials(NoeModelMaterials(texList, matList))
    mdlList.append(mdl)    

    def parse_material(self):
        
        dirpath = rapi.getDirForFilePath(rapi.getInputName())
        
        matName = self.read_name()
        self.inFile.readByte() #?
        self.inFile.read('17f')
                
        hasDiff = self.inFile.readUInt()
        if hasDiff:
            texPath = self.read_name()

        hasSpec = self.inFile.readUInt()
        if hasSpec:
            specPath = self.read_name()
        
        hasNorm = self.inFile.readUInt()
        if hasNorm:
            normPath = self.read_name()
        self.inFile.seek(7, 1)
        blend = self.read_name()
        
        texName = os.path.basename(texPath)
        tempPath = dirpath + "texture\\" + texName 
        material = NoeMaterial(matName, tempPath)
        material.setDefaultBlend(0)
        self.matList.append(material)
        return matName

def LoadTexture(texName, file):
    
    if MODPATH != None:
        name = MODPATH +'\\'+ texName
    else:
        name = rapi.getDirForFilePath(rapi.getInputName()) + texName.split('\\')[-1]
        print(rapi.getDirForFilePath(rapi.getInputName()))
    if not rapi.checkFileExists(name):
        print("Texture not found: %s" %name)
        
    else:
        
        tex = open(name,'rb').read()
        noepyLoadRGBA(tex,file.texList)
        file.texList[-1].name = name
        
        return file.texList[-1]

 def __init__(self, data):    
     self.inFile = NoeBitStream(data)
     self.animList = []
     self.texList = []
     self.matList = []
     self.boneList = []
     self.dirpath = rapi.getDirForFilePath(rapi.getInputName())
     self.texpath = self.dirpath + "texture\\"

def load_single_model(data, mdlList):
    '''Loads a single model. For testing purposes'''
    
    filename = rapi.getExtensionlessName(rapi.getInputName())
    parser = SanaeParser(data, filename)
    parser.parse_file()
    mdl = rapi.rpgConstructModel()
    mdl.setModelMaterials(NoeModelMaterials(parser.texList, parser.matList))
    mdlList.append(mdl)       

 def __init__(self, data):    
     self.inFile = NoeBitStream(data)
     self.animList = []
     self.texList = []
     self.matList = []
     self.boneList = []
     self.meshes = []
     self.materials = []
     self.dirpath = rapi.getDirForFilePath(rapi.getInputName())

def load_single_model(data, mdlList):
    '''Loads a single model. For testing purposes'''
    
    filename, ext = os.path.splitext(rapi.getLocalFileName(rapi.getInputName()))
    parser = SanaeParser(data)
    parser.parse_file(filename)
    mdl = rapi.rpgConstructModel()
    mdl.setModelMaterials(NoeModelMaterials(parser.texList, parser.matList))
    mdlList.append(mdl)       

def get_type(data):
    
    filename = rapi.getLocalFileName(rapi.getInputName())
    if rapi.checkFileExt(filename, '.3DO'):
        return Shaiya3DO(data)
    elif rapi.checkFileExt(filename, '.3DC'):
        return Shaiya3DC(data)
    elif rapi.checkFileExt(filename, '.SMOD'):
        return ShaiyaSMOD(data)

def LoadTexture(texName):
    if ARTPATH == None:
        name = rapi.getDirForFilePath(rapi.getInputName())+texName
    else:
        name = texName.split('\\')[-1]
    tex = open(name,'rb').read()
    tex = rapi.loadTexByHandler(tex,'.dds')
    tex.name = texName
    return tex