MacaqueBlender/AddTargetObjects_Color.py at master · Phenomenal-Cat/MacaqueBlender · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167

#=================== EditTargetObjects.py =======================
# These functions are used to add or remove target objects to or
# from the scene.
#

import bpy
import math
import numpy as np
import mathutils as mu

def AddTargetObjects(TargetObjFile = []):

    #================= Set target appearance
    SphereRad       = 0.01                                              # Set sphere radius (meters)
    SphereDepth     = -0.2
    SphereEcc       = 0.12
    SphereNumber    = 8
    SpherePolAng    = np.linspace(0,2*math.pi, SphereNumber+1)
    SphereLocs      = np.zeros((SphereNumber,3))
    SphereMultiCol  = 0

    if SphereMultiCol == 0:
        SphereColor    = (0,0,1)                    # Set target colors (RGB)

    elif SphereMultiCol == 1:
        np.random.seed(0)                                               # Seed numpy's random number generator
        SphereColor     = ((1,0,0),(1,0.5,0),(1,1,0),(0.5,1,0),(0,1,0),(0,1,0.5),(0,1,1),(0,0.5,1),(0,0,1),(1,0,1),(1,1,1))
        ColorOrder      = np.random.permutation(len(SphereColor))       # Randomize color locations
        SphereColor     = SphereColor[ColorOrder]

    for sph in range(0,SphereNumber):
        SphereLocs[sph][0] = SphereEcc*math.sin(SpherePolAng[sph])
        SphereLocs[sph][2] = SphereEcc*math.cos(SpherePolAng[sph])
        SphereLocs[sph][1] = SphereDepth


    #================= Create material
    AllMats = [mat for mat in bpy.data.materials if mat.name.startswith("Target")]
    if all(AllMats):
        for m in (0,SphereNumber):
            mat                         = bpy.data.materials.new("TargetMat_%s" % m)    # Create new material
            if SphereMultiCol == 0:
                mat.diffuse_color       = SphereColor                                   # Set material color
            elif SphereMultiCol == 1:
                mat.diffuse_color       = SphereColor[m]                                # Set material color
            mat.specular_intensity  = 0                                                 # Set intensity of specular reflection (0-1)
    elif not(all(AllMats)):
        mat = AllMats[0]

    #============ Add texture to target surfaces
    #heightTex           = bpy.data.textures.new('TargetTex', type = 'CLOUDS')
    #heightTex.noise_scale = 0.002


    #================= Create targets
    #bpy.ops.group.create(name="Targets")
    ##bpy.ops.object.group_link(group="Targets")

    group                   = bpy.data.groups['Targets']
    instance                = bpy.data.objects.new('dupli_group', None)
    instance.dupli_type     = 'GROUP'
    instance.dupli_group    = group
    bpy.context.scene.objects.link(instance)

    if not TargetObjFile:
        for sph in range(0, len(SphereLocs)):
            if sph == 0:
                bpy.ops.mesh.primitive_ico_sphere_add(
                    subdivisions    = 100,
                    size            = SphereRad,
                    calc_uvs        = False,
                    view_align      = False,
                    enter_editmode  = False,
                    location        = SphereLocs[sph],
                    rotation        = (0,0,0))
                bpy.data.objects['Icosphere'].active_material   = mat                           # Set target color
                bpy.data.objects['Icosphere'].name              = "Target %d" % sph             # Rename object as target ID
                dispMod             = bpy.context.scene.objects["Target 0"].modifiers.new("Displace", type='DISPLACE')
                dispMod.texture     = heightTex
                dispMod.strength    = 0.001
            elif sph  > 0:
                d = bpy.data.objects['Target 0'].copy()
                bpy.context.scene.objects.link(d)
                d.location      = SphereLocs[sph]
                d.name          = "Target %d" % sph
    else:
        for sph in range(0, len(SphereLocs)):
            if sph == 0:
                Import  = bpy.ops.import_scene.obj(filepath=TargetObjFile)          # Import target geometry from '.obj' file
                Target  = bpy.context.selected_objects[0]                           # Get target object handle
                Target.scale            = ((SphereRad, SphereRad, SphereRad))       # Set target size
                Target.location         = SphereLocs[sph]                           # Set target location
                Target.active_material  = mat                                       # Set target material
                Target.name             = "Target %d" % sph                         # Rename object as target ID
            elif shp > 0:
                d = bpy.data.objects['Target 0'].copy()                             # Copy first target object
                bpy.context.scene.objects.link(d)
                d.location      = SphereLocs[sph]                                   # Update location of new object
                d.name          = "Target %d" % sph

    bpy.context.scene.objects["Target %d" % sph].select = True
    bpy.ops.object.group_link(group="Targets")
    bpy.context.scene.objects["Target %d"  % sph].select = False
    #bpy.context.scene.objects.active = d
    #bpy.ops.group.objects_add_active(group = "Targets")
    return SphereLocs


def RemoveTargetObjects():
    scene        = bpy.context.scene
    AllTargets   = [obj for obj in scene.objects if obj.name.startswith("Target")]  # Find all objects in scene with 'Target' in name
    for t in range(0,len(AllTargets)):
        AllTargets[t].select = True         # Select next target object
        bpy.ops.object.delete()             # Delete target object


def EditTargetObjects(Add):

    if Add == 1:
        Locs = AddTargetObjects()
    elif Add == 0:
        RemoveTargetObjects()

def MonkeyLookAt(HeadLoc, GazeLoc):
    bpy.data.objects["HeaDRig"].pose.bones['EyesTracker'].location = mu.Vector((GazeLoc)) - bpy.data.objects["HeaDRig"].location
    bpy.data.objects["HeaDRig"].pose.bones['HeadTracker'].location = mu.Vector((HeadLoc[0],HeadLoc[2],HeadLoc[1])) #- bpy.data.objects["HeaDRig"].location


def RenderAllViews(Locs):
    Congruent = 0

    #RenderDir = 'P:\murphya\Blender\Renders\CuedAttention'
    RenderDir = '/Volumes/Seagate Backup 1/NIH_PhD_nonthesis/7. 3DMacaqueFaces/Renders/CuedLocations'
    for l in range(0, len(Locs)):
        if Congruent == 1:
            GazeLoc = (0, 0.03, 0.2)
            MonkeyLookAt(Locs[l], GazeLoc)
        elif Congruent == 0:
            MonkeyLookAt((0,-0.2,0), (Locs[l][0], Locs[l][1], 0.1) )

        bpy.ops.wm.redraw_timer(type='DRAW_WIN_SWAP', iterations=1)
        Filename = "CuedLocation_V3_headLoc%s.png" % (l)
        print("Now rendering: " + Filename + " . . .\n")
        bpy.context.scene.render.filepath = RenderDir + "/" + Filename
        bpy.ops.render.render(write_still=True, use_viewport=True)


Prefix = '/Volumes/projects'
TargetObjFile = Prefix + '/murphya/MacaqueFace3D/GameRenders/Golf_Ball.obj'
#RemoveTargetObjects()
Locs = AddTargetObjects(TargetObjFile)

#bpy.data.objects["HeaDRig"].pose.bones['blink'].location = mu.Vector((0,0,0.007))   # Close eye lids (blink)

#RenderAllViews(Locs)

#Loc = [0.15,0.2,0]
#MonkeyLookAt(Locs[3], Locs[3])