fmt_sho_ps2_map_bsp.py

"""Noesis Python Plugin

      File: fmt_sho_ps2_map_bsp.py
   Authors: Rodolfo Nuñez (roocker666)
            Laurynas Zubavičius (Sparagas)
   Purpose: Silent Hill Origins / Silent Hill Zero (Sony - PlayStation 2)
  Category: 3D static map models
 File Mask: *.bsp
  ID Bytes: 
"""

from inc_noesis import *


HAS_PADDING = b'\x6C'
HAS_NO_PADDING = b'\x68'


def registerNoesisTypes():
    handle = noesis.register("SHO Map PS2", ".bsp")
    noesis.setHandlerTypeCheck(handle, CheckType)
    noesis.setHandlerLoadModel(handle, LoadModel)
    return 1


def CheckType(data):
    return 1


def LoadModel(data, mdlList):
    bs = NoeBitStream(data)
    ctx = rapi.rpgCreateContext()


    search_pos = 0
    i = 0
    while True:
        # Find the next occurrence of the pattern
        search_pos = data.find(b'\x05\x04\x01\x00\x01\x00', search_pos)
        if search_pos == -1:
            break  # Exit the loop if no more occurrences are found

        search_pos += 7  # Skip searched bytes
        
        # try to skip bad SHSM data
        original_pos = search_pos
        next_pos = data.find(b'\x05\x04\x01\x00\x01\x00', search_pos)
        if next_pos > 0 and next_pos - original_pos < 50:
            search_pos = next_pos + 7
        
        # sorry was lazy to do it in a loop
        original_pos = search_pos
        next_pos = data.find(b'\x05\x04\x01\x00\x01\x00', search_pos)
        if next_pos > 0 and next_pos - original_pos < 50:
            search_pos = next_pos + 7
        
        bs.seek(search_pos)

        vnum = bs.readByte()
        check_pad = bs.read(1)

        if check_pad == HAS_PADDING:
            vbuf = bs.read(vnum * 16)
        elif check_pad == HAS_NO_PADDING:
            vbuf = bs.read(vnum * 12)

        bs.seek(data.find(b'\x05\x04\x01\x00\x01\x01', bs.tell()) + 9)
        uvbuf = bs.read(vnum * 8)

        bs.seek(data.find(b'\x05\x04\x01\x00\x01\x02', bs.tell()) + 9)
        vcolbuf = bs.read(vnum * 4)

        bs.seek(data.find(b'\x05\x04\x01\x00\x01\x03', bs.tell()) + 9)
        print('found ', i)

        if check_pad == HAS_PADDING:
            rapi.rpgBindPositionBuffer(vbuf, noesis.RPGEODATA_FLOAT, 16)
        elif check_pad == HAS_NO_PADDING:
            rapi.rpgBindPositionBuffer(vbuf, noesis.RPGEODATA_FLOAT, 12)

        rapi.rpgBindUV1Buffer(uvbuf, noesis.RPGEODATA_FLOAT, 8)
        rapi.rpgBindColorBuffer(vcolbuf, noesis.RPGEODATA_UBYTE, 4, 4)

        rapi.rpgCommitTriangles(None, noesis.RPGEO_NONE, vnum, noesis.RPGEO_TRIANGLE_STRIP)

        rapi.rpgSetName('mesh{}'.format(i))
        i += 1

    mdl = rapi.rpgConstructModel()
    mdl.setModelMaterials(NoeModelMaterials([], [NoeMaterial('mat0', '')]))
    mdlList.append(mdl)
    return 1