Esempio n. 1
0
def _convert_texture(
        tex_info: Tuple[int, str], data_folder_path: Path
) -> Tuple[Material, Texture, Image, FileResource]:
    tex_id, texture_name = tex_info
    texture_path = PureWindowsPath(texture_name)
    full_texture_path = data_folder_path / "texture" / texture_path

    texture_data: Union[io.BytesIO, BinaryIO]
    alpha_mode = "MASK"
    if texture_path.suffix.lower() == ".bmp":
        texture_data = convert_bmp_to_png(full_texture_path)
        dest_path = texture_path.with_suffix(".png")
    elif texture_path.suffix.lower() == ".tga":
        alpha_mode = "BLEND"
        texture_data = convert_tga_to_png(full_texture_path)
        dest_path = texture_path.with_suffix(".png")
    else:
        texture_data = full_texture_path.open("rb")
        dest_path = texture_path

    return (Material(pbrMetallicRoughness=PBRMetallicRoughness(
        baseColorTexture=TextureInfo(index=tex_id),
        metallicFactor=0.0,
        roughnessFactor=1.0),
                     doubleSided=True,
                     alphaMode=alpha_mode), Texture(sampler=0, source=tex_id),
            Image(uri=dest_path.name, mimeType="image/png"),
            FileResource(dest_path.name, data=texture_data.read()))
Esempio n. 2
0
def create_gltf_file(faces, new_bin_file, new_gltf_file, existing_texture_file,
                     imageDimension):
    print("Putting faces in buffer...")
    mins, maxs = getMinMaxOfFaces(faces)

    bytearr = get_bytearray_from_faces(faces, imageDimension)

    print("Creating BIN file... (" + str(len(bytearr)) + " vertices)")
    binFile = FileResource(new_bin_file, data=bytearr)
    binFile.export()

    print("Creating GLTF file...")
    data = get_gltf_file_data(
        len(bytearr) / 20, new_bin_file, existing_texture_file, mins, maxs)
    with open(new_gltf_file, 'w') as gltfFile:
        json.dump(data, gltfFile)
Esempio n. 3
0
def pack_values(gltf: GLTF, uri: str, x: List[Tuple[float, ...]], btype) -> PV:
    md52PV = gltf.md52PV
    vertex_bytearray = bytearray()
    for vertex in x:
        for value in vertex:
            vertex_bytearray.extend(struct.pack("f", value))
    width = len(x[0])
    nelements = len(x)
    bytelen = len(vertex_bytearray)
    mins = [
        min([operator.itemgetter(i)(vertex) for vertex in x])
        for i in range(width)
    ]
    maxs = [
        max([operator.itemgetter(i)(vertex) for vertex in x])
        for i in range(width)
    ]

    md5 = get_md5(bytes(vertex_bytearray))
    if md5 not in md52PV:
        # noinspection PyTypeChecker

        fr = FileResource(uri, data=vertex_bytearray)
        gltf.resources.append(fr)
        buffer = Buffer(byteLength=bytelen, uri=uri)

        bi = add_buffer(gltf, buffer)
        bv = BufferView(buffer=bi,
                        byteOffset=0,
                        byteLength=bytelen,
                        target=BufferTarget.ARRAY_BUFFER.value)
        buffer_view_index = add_buffer_view(gltf, bv)
        # noinspection PyTypeChecker
        accessor = Accessor(
            bufferView=buffer_view_index,
            byteOffset=0,
            componentType=ComponentType.FLOAT.value,
            count=nelements,
            type=btype,
            min=mins,
            max=maxs,
        )

        accessor_index = add_accessor(gltf, accessor)
        md52PV[md5] = PV(
            buffer_index=bi,
            accessor_index=accessor_index,
            buffer_view_index=buffer_view_index,
            bytelen=bytelen,
        )
    return md52PV[md5]
Esempio n. 4
0
def make_texture(gltf: GLTF, fn: str) -> int:
    data = read_bytes_from_file(fn)
    md5 = get_md5(data)
    bn = os.path.basename(fn)
    noext, ext = os.path.splitext(bn)
    if fn not in gltf.fn2texture_index:
        uri = f"textures/{noext}-{md5}{ext}"
        rf = FileResource(uri, data=data)
        gltf.resources.append(rf)
        image = Image(uri=uri)
        image_index = add_image(gltf, image)
        texture = Texture(sampler=0, source=image_index)
        texture_index = add_texture(gltf, texture)
        gltf.fn2texture_index[fn] = texture_index

    return gltf.fn2texture_index[fn]
Esempio n. 5
0
def _convert_animations(rsm_version: int,
                        frame_rate_per_second: Optional[float],
                        nodes: List[AbstractNode],
                        gltf_model: GLTFModel) -> List[FileResource]:
    gltf_resources = []

    if frame_rate_per_second:
        delay_between_frames = 1.0 / frame_rate_per_second
    else:
        delay_between_frames = 1.0 / 1000.0

    model_anim = Animation(name="animation", samplers=[], channels=[])
    input_buffer_id = None
    input_stream = io.BytesIO()
    rot_buffer_id = None
    rot_output_stream = io.BytesIO()
    scale_buffer_id = None
    scale_output_stream = io.BytesIO()
    pos_buffer_id = None
    pos_output_stream = io.BytesIO()

    for node_id, node in enumerate(nodes):
        rsm_node = node.impl
        node_name = decode_string(rsm_node.name)

        # Rotation
        rotation_frame_count = len(rsm_node.rot_key_frames)
        if rotation_frame_count > 0:
            if input_buffer_id is None:
                input_buffer_id = len(gltf_model.buffers)
                gltf_model.buffers.append(Buffer(byteLength=0))
            if rot_buffer_id is None:
                rot_buffer_id = len(gltf_model.buffers)
                gltf_model.buffers.append(Buffer(byteLength=0))

            input_values = [
                delay_between_frames * rot_frame.frame_id
                for rot_frame in rsm_node.rot_key_frames
            ]

            input_view_offset = input_stream.tell()
            input_written = serialize_floats(input_values, input_stream)
            output_view_offset = rot_output_stream.tell()
            output_written = 0
            for frame in rsm_node.rot_key_frames:
                if rsm_version < 0x200:
                    gltf_quat = [
                        frame.quaternion[0],
                        frame.quaternion[2],
                        frame.quaternion[1],
                        frame.quaternion[3],
                    ]
                else:
                    gltf_quat = frame.quaternion
                for value in gltf_quat:
                    output_written += rot_output_stream.write(
                        struct.pack('f', value))

            curr_buffer_view_id = len(gltf_model.bufferViews)
            gltf_model.bufferViews += [
                BufferView(buffer=input_buffer_id,
                           byteOffset=input_view_offset,
                           byteLength=input_written),
                BufferView(buffer=rot_buffer_id,
                           byteOffset=output_view_offset,
                           byteLength=output_written)
            ]
            curr_accessor_id = len(gltf_model.accessors)
            gltf_model.accessors += [
                Accessor(bufferView=curr_buffer_view_id,
                         byteOffset=0,
                         componentType=ComponentType.FLOAT.value,
                         count=rotation_frame_count,
                         type=AccessorType.SCALAR.value,
                         min=[min(input_values)],
                         max=[max(input_values)]),
                Accessor(bufferView=curr_buffer_view_id + 1,
                         byteOffset=0,
                         componentType=ComponentType.FLOAT.value,
                         count=rotation_frame_count,
                         type=AccessorType.VEC4.value)
            ]

            rot_sampler = AnimationSampler(input=curr_accessor_id,
                                           output=curr_accessor_id + 1)
            sampler_id = len(model_anim.samplers)
            rot_channel = Channel(sampler=sampler_id,
                                  target=Target(path="rotation", node=node_id))

            model_anim.samplers.append(rot_sampler)
            model_anim.channels.append(rot_channel)

        # Scale
        if rsm_version >= 0x106:
            scale_frame_count = len(rsm_node.scale_key_frames)
            if scale_frame_count > 0:
                if input_buffer_id is None:
                    input_buffer_id = len(gltf_model.buffers)
                    gltf_model.buffers.append(Buffer(byteLength=0))
                if scale_buffer_id is None:
                    scale_buffer_id = len(gltf_model.buffers)
                    gltf_model.buffers.append(Buffer(byteLength=0))

                input_values = [
                    delay_between_frames * scale_frame.frame_id
                    for scale_frame in rsm_node.scale_key_frames
                ]

                input_view_offset = input_stream.tell()
                input_written = serialize_floats(input_values, input_stream)
                output_view_offset = scale_output_stream.tell()
                output_written = 0
                for frame in rsm_node.scale_key_frames:
                    for value in frame.scale:
                        output_written += scale_output_stream.write(
                            struct.pack('f', value))

                curr_buffer_view_id = len(gltf_model.bufferViews)
                gltf_model.bufferViews += [
                    BufferView(buffer=input_buffer_id,
                               byteOffset=input_view_offset,
                               byteLength=input_written),
                    BufferView(buffer=scale_buffer_id,
                               byteOffset=output_view_offset,
                               byteLength=output_written)
                ]
                curr_accessor_id = len(gltf_model.accessors)
                gltf_model.accessors += [
                    Accessor(bufferView=curr_buffer_view_id,
                             byteOffset=0,
                             componentType=ComponentType.FLOAT.value,
                             count=scale_frame_count,
                             type=AccessorType.SCALAR.value,
                             min=[min(input_values)],
                             max=[max(input_values)]),
                    Accessor(bufferView=curr_buffer_view_id + 1,
                             byteOffset=0,
                             componentType=ComponentType.FLOAT.value,
                             count=scale_frame_count,
                             type=AccessorType.VEC3.value)
                ]

                scale_sampler = AnimationSampler(input=curr_accessor_id,
                                                 output=curr_accessor_id + 1)
                sampler_id = len(model_anim.samplers)
                scale_channel = Channel(sampler=sampler_id,
                                        target=Target(path="scale",
                                                      node=node_id))

                model_anim.samplers.append(scale_sampler)
                model_anim.channels.append(scale_channel)

        # Translation
        if rsm_version >= 0x203:
            translation_frame_count = len(rsm_node.pos_key_frames)
            if translation_frame_count > 0:
                if input_buffer_id is None:
                    input_buffer_id = len(gltf_model.buffers)
                    gltf_model.buffers.append(Buffer(byteLength=0))
                if pos_buffer_id is None:
                    pos_buffer_id = len(gltf_model.buffers)
                    gltf_model.buffers.append(Buffer(byteLength=0))

                input_values = [
                    delay_between_frames * pos_frame.frame_id
                    for pos_frame in rsm_node.pos_key_frames
                ]

                input_view_offset = input_stream.tell()
                input_written = serialize_floats(input_values, input_stream)
                output_view_offset = pos_output_stream.tell()
                output_written = 0
                for frame in rsm_node.pos_key_frames:
                    for value in frame.position:
                        output_written += pos_output_stream.write(
                            struct.pack('f', value))

                curr_buffer_view_id = len(gltf_model.bufferViews)
                gltf_model.bufferViews += [
                    BufferView(buffer=input_buffer_id,
                               byteOffset=input_view_offset,
                               byteLength=input_written),
                    BufferView(buffer=pos_buffer_id,
                               byteOffset=output_view_offset,
                               byteLength=output_written)
                ]
                curr_accessor_id = len(gltf_model.accessors)
                gltf_model.accessors += [
                    Accessor(bufferView=curr_buffer_view_id,
                             byteOffset=0,
                             componentType=ComponentType.FLOAT.value,
                             count=translation_frame_count,
                             type=AccessorType.SCALAR.value,
                             min=[min(input_values)],
                             max=[max(input_values)]),
                    Accessor(bufferView=curr_buffer_view_id + 1,
                             byteOffset=0,
                             componentType=ComponentType.FLOAT.value,
                             count=translation_frame_count,
                             type=AccessorType.VEC3.value)
                ]

                pos_sampler = AnimationSampler(input=curr_accessor_id,
                                               output=curr_accessor_id + 1)
                sampler_id = len(model_anim.samplers)
                pos_channel = Channel(sampler=sampler_id,
                                      target=Target(path="translation",
                                                    node=node_id))

                model_anim.samplers.append(pos_sampler)
                model_anim.channels.append(pos_channel)

    if input_buffer_id:
        # Add input data
        input_stream.seek(0)
        input_data = input_stream.read()
        input_file_name = "anim_in.bin"
        gltf_resources.append(FileResource(input_file_name, data=input_data))
        gltf_model.buffers[input_buffer_id].uri = input_file_name
        gltf_model.buffers[input_buffer_id].byteLength = len(input_data)

        # Add rotation data
        if rot_buffer_id:
            rot_output_stream.seek(0)
            rot_data = rot_output_stream.read()
            rot_file_name = 'anim_rot.bin'
            gltf_resources.append(FileResource(rot_file_name, data=rot_data))
            gltf_model.buffers[rot_buffer_id].uri = rot_file_name
            gltf_model.buffers[rot_buffer_id].byteLength = len(rot_data)

        # Add scale data
        if scale_buffer_id:
            scale_output_stream.seek(0)
            scale_data = scale_output_stream.read()
            scale_file_name = 'anim_scale.bin'
            gltf_resources.append(
                FileResource(scale_file_name, data=scale_data))
            gltf_model.buffers[scale_buffer_id].uri = scale_file_name
            gltf_model.buffers[scale_buffer_id].byteLength = len(scale_data)

        # Add tanslation data
        if pos_buffer_id:
            pos_output_stream.seek(0)
            pos_data = pos_output_stream.read()
            pos_file_name = 'anim_pos.bin'
            gltf_resources.append(FileResource(pos_file_name, data=pos_data))
            gltf_model.buffers[pos_buffer_id].uri = pos_file_name
            gltf_model.buffers[pos_buffer_id].byteLength = len(pos_data)

        gltf_model.animations = [model_anim]

    return gltf_resources
Esempio n. 6
0
def _convert_nodes(
        rsm_version: int, nodes: List[AbstractNode],
        tex_id_by_node: List[List[int]],
        gltf_model: GLTFModel) -> Tuple[List[FileResource], List[int]]:
    root_nodes = []

    model_bbox = calculate_model_bounding_box(rsm_version, nodes)
    for node in nodes:
        if node.parent is None:
            _compute_transform_matrices(rsm_version, node,
                                        len(nodes) == 0, model_bbox)

    nodes_children: Dict[str, List[int]] = {}
    vertex_bytearray = bytearray()
    tex_vertex_bytearray = bytearray()
    byteoffset = 0
    tex_byteoffset = 0
    for node_id, node in enumerate(nodes):
        rsm_node = node.impl
        node_name = decode_string(rsm_node.name)
        if node.parent is not None:
            parent_name = decode_string(node.parent.impl.name)
            if parent_name in nodes_children:
                nodes_children[parent_name] += [node_id]
            else:
                nodes_children[parent_name] = [node_id]

        if rsm_version >= 0x203:
            node_tex_ids = tex_id_by_node[node_id]
        else:
            node_tex_ids = rsm_node.texture_ids
        vertices_by_texture = _sort_vertices_by_texture(rsm_node, node_tex_ids)

        gltf_primitives = []
        for tex_id, vertices in vertices_by_texture.items():
            # Model vertices
            bytelen = _serialize_vertices(vertices[0], vertex_bytearray)
            # Texture vertices
            tex_bytelen = _serialize_vertices(vertices[1],
                                              tex_vertex_bytearray)

            (mins, maxs) = _calculate_vertices_bounds(vertices[0])
            (tex_mins, tex_maxs) = _calculate_vertices_bounds(vertices[1])

            gltf_model.bufferViews.append(
                BufferView(
                    buffer=0,  # Vertices
                    byteOffset=byteoffset,
                    byteLength=bytelen,
                    target=BufferTarget.ARRAY_BUFFER.value))
            gltf_model.bufferViews.append(
                BufferView(
                    buffer=1,  # Texture vertices
                    byteOffset=tex_byteoffset,
                    byteLength=tex_bytelen,
                    target=BufferTarget.ARRAY_BUFFER.value))

            buffer_view_id = len(gltf_model.accessors)
            gltf_model.accessors.append(
                Accessor(bufferView=buffer_view_id,
                         byteOffset=0,
                         componentType=ComponentType.FLOAT.value,
                         count=len(vertices[0]),
                         type=AccessorType.VEC3.value,
                         min=mins,
                         max=maxs))
            gltf_model.accessors.append(
                Accessor(bufferView=buffer_view_id + 1,
                         byteOffset=0,
                         componentType=ComponentType.FLOAT.value,
                         count=len(vertices[1]),
                         type=AccessorType.VEC2.value,
                         min=tex_mins,
                         max=tex_maxs))
            gltf_primitives.append(
                Primitive(attributes=Attributes(POSITION=buffer_view_id + 0,
                                                TEXCOORD_0=buffer_view_id + 1),
                          material=tex_id))

            byteoffset += bytelen
            tex_byteoffset += tex_bytelen

        gltf_model.meshes.append(Mesh(primitives=gltf_primitives))

        # Decompose matrix to TRS
        translation, rotation, scale = decompose_matrix(
            node.gltf_transform_matrix)

        gltf_model.nodes.append(
            Node(name=node_name,
                 mesh=node_id,
                 translation=translation.to_list() if translation else None,
                 rotation=rotation.to_list() if rotation else None,
                 scale=scale.to_list() if scale else None))
        if node.parent is None:
            root_nodes.append(node_id)

    # Register vertex buffers
    vtx_file_name = 'vertices.bin'
    tex_vtx_file_name = 'tex_vertices.bin'
    gltf_resources = [
        FileResource(vtx_file_name, data=vertex_bytearray),
        FileResource(tex_vtx_file_name, data=tex_vertex_bytearray)
    ]
    gltf_model.buffers = [
        Buffer(byteLength=byteoffset, uri=vtx_file_name),
        Buffer(byteLength=tex_byteoffset, uri=tex_vtx_file_name)
    ]

    # Update nodes' children
    # Note(LinkZ): Consume children with `pop` to avoid issues with models
    # containing multiple nodes with the same name
    for gltf_node in gltf_model.nodes:
        gltf_node.children = nodes_children.pop(gltf_node.name, None)

    return gltf_resources, root_nodes
Esempio n. 7
0
 def transform(g: GLTF) -> GLTF:
     rf = FileResource(uri, data=data)
     g.resources.append(rf)
     g.model.images[0] = Image(name=tname, uri=uri)
     return g
Esempio n. 8
0
def resource_from_file(fn: FilePath) -> FileResource:
    data = read_bytes_from_file(fn)

    bn = os.path.basename(fn)
    return FileResource(bn, data=data)
Esempio n. 9
0
def export_gltf(icon, filename, metadata=None):
    basename = PurePath(filename).stem

    vertex_info_format = ("3f" * icon.animation_shapes) + "3f 2f 3f"
    float_size = struct.calcsize("f")
    animation_speed = 0.1

    animation_present = icon.animation_shapes > 1

    model_data = bytearray()

    mins = {}
    maxs = {}

    for i, vertex in enumerate(icon.vertices):
        for j, position in enumerate(vertex.positions):
            if j == 0:
                values_basis = [
                    position.x / 4096, -position.y / 4096, -position.z / 4096
                ]
                values = values_basis
            else:
                # Subtract basis position to compensate for shape keys being relative to basis
                values = [
                    position.x / 4096 - values_basis[0],
                    -position.y / 4096 - values_basis[1],
                    -position.z / 4096 - values_basis[2]
                ]

            if j not in mins:
                mins[j] = values.copy()
            else:
                if values[0] < mins[j][0]: mins[j][0] = values[0]
                if values[1] < mins[j][1]: mins[j][1] = values[1]
                if values[2] < mins[j][2]: mins[j][2] = values[2]

            if j not in maxs:
                maxs[j] = values.copy()
            else:
                if values[0] > maxs[j][0]: maxs[j][0] = values[0]
                if values[1] > maxs[j][1]: maxs[j][1] = values[1]
                if values[2] > maxs[j][2]: maxs[j][2] = values[2]

            model_data.extend(struct.pack("3f", *values))

        model_data.extend(
            struct.pack("3f 2f 3f", vertex.normal.x / 4096,
                        -vertex.normal.y / 4096, -vertex.normal.z / 4096,
                        1.0 - (vertex.tex_coord.u / 4096),
                        1.0 - (vertex.tex_coord.v / 4096),
                        vertex.color.r / 255, vertex.color.g / 255,
                        vertex.color.b / 255))

    # Generate animation data if multiple animation shapes are present

    if animation_present:
        animation_offset = len(model_data)

        for i in range(icon.frame_count + 1):
            model_data.extend(struct.pack("f", i * animation_speed))

        for i, frame in enumerate(icon.frames + [icon.frames[0]]):
            segment = [struct.pack("f", 0.0)] * (icon.animation_shapes - 1)

            if frame.shape_id != 0:
                segment[frame.shape_id - 1] = struct.pack("f", 1.0)

            for item in segment:
                model_data.extend(item)

        animation_length = len(model_data) - animation_offset

    # Generate texture

    if isinstance(icon.texture, Ps2ico.CompressedTexture):
        image_data = convert_compressed_texture_data(icon.texture.size,
                                                     icon.texture.data)
    elif isinstance(icon.texture, Ps2ico.UncompressedTexture):
        image_data = convert_uncompressed_texture_data(icon.texture.data)

    with BytesIO() as png:
        PILImage.frombytes("RGB", (128, 128), image_data).save(png, "png")
        texture_data = png.getvalue()

    # Basic glTF info

    model = GLTFModel()

    model.asset = Asset(version="2.0", generator=f"ico2gltf v{VERSION}")

    model.scenes = [Scene(nodes=[0])]

    model.scene = 0

    model.nodes = [Node(mesh=0)]

    # If present, embed metadata

    if metadata:
        # Normalize title: turn japanese full-width characters into normal ones and insert the line break
        title = unicodedata.normalize("NFKC", metadata.title).rstrip("\x00")
        title = title[:metadata.offset_2nd_line //
                      2] + "\n" + title[metadata.offset_2nd_line // 2:]

        model.extras = {
            "title":
            title,
            "background_opacity":
            metadata.bg_opacity / 0x80,
            "background_bottom_left_color": [
                metadata.bg_color_lowerleft.r / 0x80,
                metadata.bg_color_lowerleft.g / 0x80,
                metadata.bg_color_lowerleft.b / 0x80,
                metadata.bg_color_lowerleft.a / 0x80
            ],
            "background_bottom_right_color": [
                metadata.bg_color_lowerright.r / 0x80,
                metadata.bg_color_lowerright.g / 0x80,
                metadata.bg_color_lowerright.b / 0x80,
                metadata.bg_color_lowerright.a / 0x80
            ],
            "background_top_left_color": [
                metadata.bg_color_upperleft.r / 0x80,
                metadata.bg_color_upperleft.g / 0x80,
                metadata.bg_color_upperleft.b / 0x80,
                metadata.bg_color_upperleft.a / 0x80
            ],
            "background_top_right_color": [
                metadata.bg_color_upperright.r / 0x80,
                metadata.bg_color_upperright.g / 0x80,
                metadata.bg_color_upperright.b / 0x80,
                metadata.bg_color_lowerright.a / 0x80
            ],
            "ambient_color": [
                metadata.light_ambient_color.r, metadata.light_ambient_color.g,
                metadata.light_ambient_color.b
            ],
            "light1_direction": [
                metadata.light1_direction.x, metadata.light1_direction.y,
                metadata.light1_direction.z
            ],
            "light1_color": [
                metadata.light1_color.r, metadata.light1_color.g,
                metadata.light1_color.b, metadata.light1_color.a
            ],
            "light2_direction": [
                metadata.light2_direction.x, metadata.light2_direction.y,
                metadata.light2_direction.z
            ],
            "light2_color": [
                metadata.light2_color.r, metadata.light2_color.g,
                metadata.light2_color.b, metadata.light2_color.a
            ],
            "light3_direction": [
                metadata.light3_direction.x, metadata.light3_direction.y,
                metadata.light3_direction.z
            ],
            "light3_color": [
                metadata.light3_color.r, metadata.light3_color.g,
                metadata.light3_color.b, metadata.light3_color.a
            ],
        }

    # Meshes

    primitive = Primitive(attributes=Attributes(
        POSITION=0,
        NORMAL=icon.animation_shapes,
        TEXCOORD_0=icon.animation_shapes + 1,
        COLOR_0=icon.animation_shapes + 2),
                          material=0)

    if animation_present:
        primitive.targets = [{
            "POSITION": i + 1
        } for i in range(icon.animation_shapes - 1)]

    model.meshes = [Mesh(name="Icon", primitives=[primitive])]

    # Buffers

    model.buffers = [
        Buffer(uri=f"{basename}.bin", byteLength=len(model_data)),
        Buffer(uri=f"{basename}.png", byteLength=len(texture_data))
    ]

    # Materials

    model.images = [Image(bufferView=1, mimeType="image/png")]

    model.textures = [Texture(source=0)]

    model.materials = [
        Material(name="Material",
                 pbrMetallicRoughness=PBRMetallicRoughness(
                     baseColorTexture=TextureInfo(index=0),
                     roughnessFactor=1,
                     metallicFactor=0))
    ]

    # Animations

    if animation_present:
        model.animations = [
            Animation(name="Default",
                      samplers=[
                          AnimationSampler(
                              input=icon.animation_shapes + 3,
                              output=icon.animation_shapes + 4,
                              interpolation=Interpolation.LINEAR.value)
                      ],
                      channels=[
                          Channel(sampler=0,
                                  target=Target(node=0, path="weights"))
                      ]),
        ]

    # Buffer Views

    model.bufferViews = [
        BufferView(name="Data",
                   buffer=0,
                   byteStride=struct.calcsize(vertex_info_format),
                   byteLength=len(model_data)),
        BufferView(name="Texture", buffer=1, byteLength=len(texture_data)),
    ]

    if animation_present:
        model.bufferViews.append(
            BufferView(name="Animation",
                       buffer=0,
                       byteOffset=animation_offset,
                       byteLength=animation_length), )

    # Accessors

    model.accessors = [
        Accessor(name=f"Position {i}",
                 bufferView=0,
                 byteOffset=i * 3 * float_size,
                 min=mins[i],
                 max=maxs[i],
                 count=len(icon.vertices),
                 componentType=ComponentType.FLOAT.value,
                 type=AccessorType.VEC3.value)
        for i in range(icon.animation_shapes)
    ]

    model.accessors.extend([
        Accessor(name="Normal",
                 bufferView=0,
                 byteOffset=((icon.animation_shapes - 1) * 3 * float_size) +
                 3 * float_size,
                 count=len(icon.vertices),
                 componentType=ComponentType.FLOAT.value,
                 type=AccessorType.VEC3.value),
        Accessor(name="UV",
                 bufferView=0,
                 byteOffset=((icon.animation_shapes - 1) * 3 * float_size) +
                 6 * float_size,
                 count=len(icon.vertices),
                 componentType=ComponentType.FLOAT.value,
                 type=AccessorType.VEC2.value),
        Accessor(name="Color",
                 bufferView=0,
                 byteOffset=((icon.animation_shapes - 1) * 3 * float_size) +
                 8 * float_size,
                 count=len(icon.vertices),
                 componentType=ComponentType.FLOAT.value,
                 type=AccessorType.VEC3.value),
    ])

    if animation_present:
        model.accessors.extend([
            Accessor(name="Animation Time",
                     bufferView=2,
                     byteOffset=0,
                     min=[0.0],
                     max=[(icon.frame_count) * animation_speed],
                     count=(icon.frame_count + 1),
                     componentType=ComponentType.FLOAT.value,
                     type=AccessorType.SCALAR.value),
            Accessor(name="Animation Data",
                     bufferView=2,
                     byteOffset=(icon.frame_count + 1) * float_size,
                     min=[0.0],
                     max=[1.0],
                     count=(icon.frame_count + 1) *
                     (icon.animation_shapes - 1),
                     componentType=ComponentType.FLOAT.value,
                     type=AccessorType.SCALAR.value)
        ])

    resources = [
        FileResource(f"{basename}.bin", data=model_data),
        FileResource(f"{basename}.png", data=texture_data)
    ]

    gltf = GLTF(model=model, resources=resources)
    gltf.export(filename)
Esempio n. 10
0
def resource_from_file(fn: str) -> FileResource:
    with open(fn, "rb") as f:
        data = f.read()

    bn = os.path.basename(fn)
    return FileResource(bn, data=data)