#import useful libraries, if you don't have them install them with pip

import sys

import bvh

sys.path.append(G.app.mhapi.locations.getInstallationPath()+'/plugins/4_rendering_opengl')

import mh2opengl

import gui3d

import numpy as np

import random 

global bvh

global mh2opengl

global gui3d

global posedir

global human

global bodyZones

global MHScript

#set the camera in a frontal position

G.app.modelCamera.setHorizontalRotation(0)

G.app.modelCamera.setVerticalInclination(5.0)

G.app.modelCamera.setPosition([0.15, 0.05, 0.0])

G.app.modelCamera.setZoomFactor(1.1)

#make sure this path corresponds with the folder with the poses in makehuman installation #folder

posedir = "/Applications/MakeHuman.app/Contents/Resources/data/poses/"

human = G.app.selectedHuman

global settings

settings = dict()

settings['scene'] = G.app.scene

settings['AA'] = True

settings['dimensions'] = (800, 800)

settings['lightmapSSS'] = False

#create vectors for modifiers with the values you want them to assume

random.seed(0)  # set random seed to 0 [for reproducibility]

rand_wheel_value = [ 0.1,0.2, 0.3, 0.4, 0.5, 0.6, 0.7, 0.8,0.9] #generates a vector with distinct decimal values

rand_wheel_muscle = [ 0.1,0.2, 0.3, 0.4, 0.5, 0.6]

rand_wheel_weight = [ 0.3, 0.4, 0.5, 0.6, 0.7, 0.8 ]

rand_wheel_breast = [0.6, 0.7, 0.8, 0.9, 1.0]

#get the modifier for waist to hip ratio #vary from -1 to 1

WHR_mod = human.getModifier('measure/measure-waisttohip-dist-decr|incr')

#get the modifier for hips circumference #vary from -1 to 1

hips_mod = human.getModifier('measure/measure-hips-circ-decr|incr')

rand_wheel_WHR = [-0.6,-0.5, -0.4, - 0.3, -0.2, -0.1, 0]

rand_wheel_hips = [0, 0.1,0.2,0.3,0.4,0.5]

#image = "image"

pose = "standing02"

#open and create a .txt file where you will store the parameters

f= open("/Users/ciao/Documents/MakeHuman/v1/render/female_render/female_models_parameters.txt","a+")

#insert your path above

bvh_file=bvh.load(posedir + pose + ".bvh", "auto")

#loop 130 times 4 intervals of femininity from 0 very feminine to 0.3 (1 is very masculine, #0.5 is androgynous)

sex = 0

i = 0

for i in range(1,131):

  for sex in np.arange(0,0.4,0.1):

  anim=bvh_file.createAnimationTrack(human.getBaseSkeleton())

  human.addAnimation(anim)

  human.setActiveAnimation(anim.name)

  human.setToFrame(0,update=False)

  human.setPosed(True)

  human.setGender(sex)

  #select a random number from vector above, to insert into the for loop

  human.setMuscle(random.choice(rand_wheel_muscle))  

  human.setWeight(random.choice(rand_wheel_weight))

  human.setBodyProportions(random.choice(rand_wheel_value))

  human.setBreastSize(random.choice(rand_wheel_breast))

  human.setBreastFirmness(random.choice(rand_wheel_value))

  human.setAge(random.uniform(0.6,0.8))

  human.setHeight(0.6)

  WHR_mod.setValue(random.choice(rand_wheel_WHR))

  hips_mod.setValue(random.choice(rand_wheel_hips))

  human.applyAllTargets()

  # Render.

  mh2opengl.Render(settings)

  # Save.

  filename = mh.getPath('render/female_render') + '/' + 'F' + str(sex) + '_' + str(i) +  '.png'

  gui3d.app.getCategory('Rendering').getTaskByName('Viewer').image.save(filename)

#write on the .txt file the parameters   

  f.write("muscle %.3f\r\n" % human.getMuscle() + "sex %.3f\r\n" % human.getGender() + "bodyprop %.3f\r\n" % human.getBodyProportions() + "weight %.3f\r\n" % human.getWeight() + "age %.3f\r\n" % human.getAge()   + "bodyno%d\r\n" % (i))  

  i = i + 1

f.close()