compile_meshes.py

#!/usr/bin/python3
import os
import json
import argparse

os.environ['MAKEHUMAN2TOOL'] = str(True)

from core.environ import UserEnvironment
from core.attached_asset import attachedAsset
from obj3d.object3d import object3d

#
# we need a dummy class for global containing the environment 
# and a logLine function provided in inside environment
#

class globalObjects():
    def __init__(self, env):
        self.env = env

def logLine(level, line):
    if level & 8:
        print (line)

def getNumVerts(basename, systemspace, userspace):
    """
    get numverts from base.json
    """

    bfile = os.path.join(userspace, "base", basename, "base.json")
    if not os.path.isfile(bfile):
        bfile = os.path.join(systemspace, "base", basename, "base.json")

    if os.path.isfile(bfile):
        with open(bfile, 'r') as f:
            baselines = json.load(f)
            if "numverts" in baselines:
                return baselines["numverts"]
    return -1

def compressFile(glob, eqtype, path, source):
    filename = os.path.join(path, source)
    asset =  attachedAsset(glob, eqtype, glob.env.numverts)
    asset.mhcloToMHBin(filename)

def compressSingleFile(glob, name):
    if name.endswith(".obj"):
        basemesh = object3d(glob, None, "base")
        (res, err) = basemesh.load(name, True)
        if res == 0:
            print (err)
            exit(10)
        basemesh.exportBinary()
    elif name.endswith(".mhclo") or name.endswith(".proxy"):
        p, f = os.path.split(name)
        p, sd = os.path.split(p)
        p, m = os.path.split(p)
        p, eqtype = os.path.split(p)
        asset =  attachedAsset(glob, eqtype, glob.env.numverts)
        asset.mhcloToMHBin(name)

if __name__ == '__main__':
    # get predefined environment parameters (standardmesh)
    #
    dirname = os.path.abspath(os.path.dirname(__file__))
    os.chdir(dirname)
    systemspace = os.path.join(dirname,"data")

    release_info = os.path.join("data", "makehuman2_version.json")
    if os.path.isfile(release_info):
        with open(release_info, 'r') as f:
            release = json.load(f)


    uenv = UserEnvironment()
    uenv.getPlatform()
    #
    # now add a few additional environment variables
    #
    uenv.logLine = logLine                          # the function we supply directly
    uenv.verbose = 0                                # do not print comments from makehuman2
    uenv.basename = release["standardmesh"]         # the meshname


    conffile = uenv.getUserConfigFilenames()[0]
    userspace = None
    if os.path.isfile(conffile):
        with open(conffile, 'r') as f:
            conf = json.load(f)
            userspace = os.path.join(conf["path_home"], "data")

    parser = argparse.ArgumentParser(description="Compile objects (mhclo + obj) to binary form (mhbin) (usually works interactive).")
    parser.add_argument("-s", action="store_true", help="compile system space objects")
    parser.add_argument("-b", "--base", type=str, default=uenv.basename, help="preselect base mesh")
    parser.add_argument("-f", "--file", type=str, help="compile only this file")

    if userspace is not None:
        parser.add_argument("-u", action="store_true", help="compile user space instead of system space")

    parser.add_argument("-n", action="store_true", help="compile non interactive")
    parser.add_argument("filename", nargs="?", type=str, help="compile only assets which are similar to this filename")

    args = parser.parse_args()

    # without knowing the base it self we have to know the number of vertices
    # for creating the delete-bool-array for clothes, so it is parameter of base.json
    #

    uenv.numverts = getNumVerts(args.base, systemspace, userspace)
    if uenv.numverts < 0:
        print ("Cannot evaluate number of vertices from 'base.json'")
        exit (2)

    # first handle the case where only one file should be compiled
    #
    if args.file:
        glob = globalObjects(uenv)
        compressSingleFile(glob, args.file)
        exit(0)


    space = None
    if args.u:
        if userspace is None:
            print ("No user space found")
            exit (2)
        space = userspace
    if args.s:
        space = systemspace


    # no decision, ask user
    #
    if space is None:
        print("[1] User   space: " + userspace)
        print("[2] System space: " + systemspace)

        okay = False
        while not okay:
            line = input('Enter 1, 2 or a to abort: ')
            if line == "a":
                exit (0)
            if line == "1":
                space = userspace
                okay = True
            if line == "2":
                space = systemspace
                okay = True

    print ("Compile objects in: " + space + "\n")
    if args.n is False:
        okay = False
        while not okay:
            line = input('Enter a to abort, c to compress: ')
            if line == "a":
                exit (0)
            if line == "c":
                okay = True

    glob = globalObjects(uenv)

    # first compile base if added to user space or system space
    #
    num = 0
    base =  os.path.join(space, "base", args.base, "base.obj")
    if os.path.isfile (base):
        if args.filename is None or "base" in args.filename:
            print ("Found: " + base)
            basemesh = object3d(glob, None, "base")
            (res, err) = basemesh.load(base, True)
            if res == 0:
                print (err)
                exit(10)
            basemesh.exportBinary()
            num += 1

    for folder in ["clothes", "eyebrows", "eyelashes", "eyes", "hair", "proxy", "teeth", "tongue"]:
        absfolder = os.path.join(space, folder, args.base)
        if os.path.isdir(absfolder):
            for root, dirs, files in os.walk(absfolder, topdown=True):
                for name in files:
                    if name.endswith(".mhclo") or name.endswith(".proxy"):
                        if args.filename:
                            if args.filename in name:
                                compressFile (glob, folder, root, name)
                                num += 1
                        else:
                            compressFile (glob, folder, root, name)
                            num += 1

    print (str(num) + " mesh(es) compiled.")
    exit(0)