weights_sum.py

#!/usr/bin/python3
# -*- coding: utf-8 -*-

"""

**Project Name:**      MakeHuman-Helpers

**Authors:**           Black Punkduck

**Code Home Page:**    https://github.com/black-punkduck/MakeHuman-Helpers

**Copyright(c):**      Black Punkduck

**Licensing:**         AGPL3

    This file is part of the MakeHuman Support Tools

    This program is free software: you can redistribute it and/or modify
    it under the terms of the GNU Affero General Public License as
    published by the Free Software Foundation, either version 3 of the
    License, or (at your option) any later version.

    This program is distributed in the hope that it will be useful,
    but WITHOUT ANY WARRANTY; without even the implied warranty of
    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    GNU Affero General Public License for more details.

    You should have received a copy of the GNU Affero General Public License
    along with this program.  If not, see <http://www.gnu.org/licenses/>.

Abstract
--------
Skript to check if weights of a weight-file will sum up to 1.
"""

import re
import sys
import json
import argparse

configfilename = "data/mh_helper.json"

parser = argparse.ArgumentParser(description='Check if weights of a weight-file will sum up to 1.')
parser.add_argument('-f', default="", metavar='default_weights', help='use different default weight file (default: see configuration)')
parser.add_argument('-L', default=0.999, metavar='LOWEST', type=float,  help='lowest weight to be accepted (default: 0.999)')
parser.add_argument('-H', default=1.001, metavar='HIGHEST', type=float,  help='highest weight to be accepted (default: 1.001)')
parser.add_argument('-s', default=0, metavar='MINWEIGHTNUM', type=int,  help='lowest vertex number to check (default: 0)')
parser.add_argument('-m', default=0, metavar='MAXWEIGHTNUM', type=int,  help='highest vertex to check (default: see configuration ranges/_end_)')
args = parser.parse_args()

#
# read configuration
#
cfile = open (configfilename, "r")
config = json.load (cfile)
cfile.close()

minweights = args.s
if args.m == 0:
    maxweights = config['ranges']['_end_']
else:
    maxweights = args.m

#
# read weight-file
#
if len(args.f):
    weightfilename = args.f
else:
    weightfilename = config["default_weights"]

cfile = open (weightfilename, "r")
weights = json.load (cfile)
cfile.close()

verts = {}
for key in weights["weights"].keys():
    group = weights["weights"][key]

    for comb in range (0, len(group)):
        vnum = int(group[comb][0])
        val  = group[comb][1]
        if vnum not in verts:
            verts[vnum] = { '_sum' : val, key: val}
        else:
            verts[vnum][key] = val
            verts[vnum]['_sum'] += val


for vnum in range (minweights, maxweights):
    if vnum not in verts:
        print( str(vnum) + ": undefined")
    else:
        s = verts[vnum]['_sum']
        if s < args.L or s > args.H:
            t = str(vnum) + ": " + str(round(s, 4)) + ":="
            for group in verts[vnum].keys():
                if group != '_sum':
                    t += "  " + group + ":" + str(verts[vnum][group])
            print (t)