for p in range(o['nPoses']): O = bvp.Object(obID=o['name'],Lib=Lib,pos3D=pos,size3D=s,rot3D=(0,0,rotZ),pose=p) ObToAdd.append(O) if scaleObj: ScObSz = 10.*scaleObj.size3D/O.size3D ScObToAdd.append else: O = bvp.Object(obID=o['name'],Lib=Lib,pos3D=pos,size3D=s,rot3D=(0,0,rotZ)) ObToAdd.append(O) # Add scale object in here somehwhere... Scale for each object! if scaleObj: ScObSz = 10.*scaleObj.size3D/O.size3D ScObToAdd.append # Lights (Sky), Background Sky = bvp.Sky() BG = bvp.Background() # Objects for Obj in ObToAdd: # Camera for c in SceneCam: #Cam = bvp.Camera() # w/ some parameters... # Create Scene ScCt+=1 fpath = 'Sc%04d_##'%(ScCt) ScnL.append(bvp.Scene(Num=ScCt,Obj=(Obj,),BG=BG,Sky=Sky, Shadow=None,Cam=c,FrameRange=(1,nFrames), fpath=fpath,FrameRate=15)) # Convert list of scenes to SceneList SL = bvp.SceneList(ScnList=ScnL,RenderOptions=RO) # Save SL.Save(os.path.join(LibDir,'SilhouetteSceneList.pik'))
# Frame count
frames = (1,1)
# set standard lights (Sky)
Sky = bvp.Sky()
# Get dummy objects to put in scenes:
ObL = []
for o in DummyObjects:
ObL.append(bvp.Object(obID=o,Lib=Lib,size3D=None)
# Misc Setup
BGCt = 0;
ScnL = []
for bg in ToRender:
BGCt+=1
# Create Scene
BG = bvp.Background(bgID=bg['name'],Lib=Lib)
for p in range(nCamLoc):
cNum = p+1
Cam = bvp.Camera(location=BG.CamConstraint.sampleCamPos(frames),fixPos=BG.CamConstraint.sampleFixPos(frames),frames=frames)
S = bvp.Scene(Num=BGCt,BG=BG,Sky=Sky,Obj=None,
Shadow=None,Cam=Cam,FrameRange=(1,1),
fpath='%s_%s_cp%d_fr##'%(BG.semantic_category[0],BG.name,cNum),
FrameRate=15)
try:
# Allow re-set of camera position with each attempt to populate scene
S.populate_scene(ObL,ResetCam=True)
except:
print('Unable to populate scene %s!'%S.fpath)
ScnL.append(S)
# Convert list of scenes to SceneList
SL = bvp.SceneList(ScnList=ScnL,RenderOptions=RO)
def RenderSkies(self, subCat=None, Is_Overwrite=False):
"""
Render (all) skies in bvpLibrary to folder <LibDir>/LibBackgrounds/<category>_<name>.png
subCat = None # lambda x: 'dome' in x['semantic_category']
"""
raise Exception('Not done yet!')
RO = bvp.RenderOptions()
RO.BVPopts['BasePath'] = os.path.join(self.LibDir, 'Images', 'Skies',
'%s')
RO.resolution_x = RO.resolution_y = 256 # smaller images
if subCat:
ToRender = self.getSCL(subCat, 'backgrounds')
else:
ToRender = self.backgrounds # all backgrounds
# Frame count
frames = (1, 1)
# set standard lights (Sky)
Sky = bvp.Sky()
# Get dummy objects to put in scenes:
ObL = []
for o in dummyObjects:
ObL.append(bvp.Object(obID=o, Lib=self, size3D=None))
# Misc Setup
BGCt = 0
ScnL = []
for bg in ToRender:
BGCt += 1
# Create Scene
BG = bvp.Background(bgID=bg['name'], Lib=self)
for p in range(nCamLoc):
cNum = p + 1
fpath = '%s_%s_cp%d_fr##' % (BG.semantic_category[0], BG.name,
cNum)
fChk = RO.BVPopts['BasePath'] % fpath.replace(
'##', '01.' + RO.file_format.lower())
print('Checking for file: %s' % (fChk))
if os.path.exists(fChk) and not Is_Overwrite:
print('Found it!')
# Only append scenes to render that DO NOT have previews already rendered!
continue
Cam = bvp.Camera(
location=BG.CamConstraint.sampleCamPos(frames),
fixPos=BG.CamConstraint.sampleFixPos(frames),
frames=frames)
S = bvp.Scene(Num=BGCt,
BG=BG,
Sky=Sky,
Obj=None,
Shadow=None,
Cam=Cam,
FrameRange=(1, 1),
fpath=fpath,
FrameRate=15)
#try:
# Allow re-set of camera position with each attempt to populate scene
S.populate_scene(ObL, ResetCam=True)
#except:
# print('Unable to populate scene %s!'%S.fpath)
ScnL.append(S)
# Convert list of scenes to SceneList
SL = bvp.SceneList(ScnList=ScnL, RenderOptions=RO)
SL.RenderSlurm(RenderGroupSize=nCamLoc)
def RenderBGs(self,
subCat=None,
dummyObjects=(),
nCamLoc=5,
Is_Overwrite=False):
"""
Render (all) backgrounds in bvpLibrary to folder <LibDir>/Images/Backgrounds/<category>_<name>.png
subCat = None #lambda x: x['name']=='BG_201_mHouse_1fl_1' #None #'outdoor'
dummyObjects = ['*human', '*artifact', '*vehicle']
"""
RO = bvp.RenderOptions()
RO.BVPopts['BasePath'] = os.path.join(self.LibDir, 'Images',
'Backgrounds', '%s')
RO.resolution_x = RO.resolution_y = 256 # smaller images
if subCat:
ToRender = self.getSCL(subCat, 'backgrounds')
else:
ToRender = self.backgrounds # all backgrounds
# Frame count
frames = (1, 1)
# Get dummy objects to put in scenes:
# Misc Setup
BGCt = 0
ScnL = []
for bg in ToRender:
BGCt += 1
# Create Scene
BG = bvp.Background(bgID=bg['name'], Lib=self)
ObL = []
for o in dummyObjects:
ObL.append(bvp.Object(obID=o, Lib=self, size3D=None))
for p in range(nCamLoc):
cNum = p + 1
fpath = '%s_%s_cp%02d_fr##' % (BG.semantic_category[0],
BG.name, cNum)
fChk = RO.BVPopts['BasePath'] % fpath.replace(
'##', '01.' + RO.image_settings['file_format'].lower())
print('Checking for file: %s' % (fChk))
if os.path.exists(fChk) and not Is_Overwrite:
print('Found it!')
# Only append scenes to render that DO NOT have previews already rendered!
continue
Cam = bvp.Camera(
location=BG.CamConstraint.sampleCamPos(frames),
fixPos=BG.CamConstraint.sampleFixPos(frames),
frames=frames)
Sky = bvp.Sky('*' + BG.sky_semantic_category[0], Lib=self)
if Sky.semantic_category:
if 'dome' in Sky.semantic_category:
if len(Sky.lightLoc) > 1:
Shad = None
elif len(Sky.lightLoc) == 1:
if 'sunset' in Sky.semantic_category:
Shad = bvp.Shadow('*west', self)
else:
fn = lambda x: 'clouds' in x[
'semantic_category'] and not 'west' in x[
'semantic_category']
Shad = bvp.Shadow(fn, self)
else:
Shad = None
else:
Shad = None
S = bvp.Scene(Num=BGCt,
BG=BG,
Sky=Sky,
Obj=None,
Shadow=Shad,
Cam=Cam,
FrameRange=frames,
fpath=fpath,
FrameRate=15)
try:
# Allow re-set of camera position with each attempt to populate scene
S.populate_scene(ObL, ResetCam=True)
except:
print('Unable to populate scene %s!' % S.fpath)
ScnL.append(S)
# Convert list of scenes to SceneList
SL = bvp.SceneList(ScnList=ScnL, RenderOptions=RO)
SL.RenderSlurm(RenderGroupSize=nCamLoc)
def RenderObjectVox(self,
nGrid=10,
xL=(-5, 5),
yL=(-5, 5),
zL=(0, 10),
maxFilesPerDir=5000,
subCat=None,
render_Pose=True,
Is_Overwrite=False):
"""
Render (all) objects in bvpLibrary in voxelized 3D form
TODO: longer help!
ScaleObj = optional scale object to render along with this object (NOT FINISHED!)
"""
if bvp.Is_Blender:
print('Sorry, this won'
't run inside Blender; it requires access to slurm!')
return
RO = bvp.RenderOptions()
RO.BVPopts['Voxels'] = True # This will over-ride all other options...
RO.BVPopts['BasePath'] = os.path.join(self.LibDir, 'Images', 'Objects',
'Voxels', '%s')
RO.BVPopts['Type'] = 'all'
RO.resolution_x = RO.resolution_y = 5 # itty-bitty images for inside/outside test
if subCat:
ToRender = self.getSCL(subCat, 'objects')
else:
ToRender = self.objects # all objects
ObCt = 0
for o in ToRender:
ScnL = []
# Get all object variations to add as separate scenes
ObToAdd = []
if o['nPoses'] and render_Pose:
for p in range(o['nPoses']):
O = bvp.Object(obID=o['name'],
Lib=self,
pos3D=(0, 0, 0),
size3D=10,
pose=p)
ObToAdd.append(O)
else:
O = bvp.Object(obID=o['name'],
Lib=self,
pos3D=(0, 0, 0),
size3D=10)
ObToAdd.append(O)
# Lights (Sky) & Background
Sky = bvp.Sky()
Sky.WorldParams['horizon_color'] = (0, 0, 0)
BG = bvp.Background()
# Get all (nGrid**3) camera positions
cPos = bvp.utils.basics.gridPos(nGrid, xL, yL, zL)
# Center (fixation) Position
fPos = [[0, 0, 0] for x in range(len(cPos))]
fr = range(1, nGrid**3 + 1)
# break up into multiple directories with <maxFilesPerDir> files each
nDirs = int(math.ceil((nGrid**3) / float(maxFilesPerDir)))
# Loop over objects to render each <maxFilesPerDir> files
for Obj in ObToAdd:
# Loop to create a separate scene for each <maxFilesPerDir> files
for d in range(nDirs):
# Create Scene
ObCt += 1
if Obj.pose or Obj.pose == 0:
pNum = Obj.pose + 1
else:
pNum = 1
# Frame range
FR = (d * maxFilesPerDir + 1,
min(nGrid**3, maxFilesPerDir * (d + 1)))
Cam = bvp.Camera(location=cPos[FR[0] - 1:FR[1]],
fixPos=fPos[FR[0] - 1:FR[1]],
frames=fr[FR[0] - 1:FR[1]])
fpath = '%s_%s_p%d_res%d_f%09d/vox%s' % (
Obj.semantic_category[0], Obj.name, pNum, nGrid,
d * maxFilesPerDir, '#' * len(str(nGrid**3)))
ScnL.append(
bvp.Scene(Num=ObCt,
Obj=(Obj, ),
BG=BG,
Sky=Sky,
Shadow=None,
Cam=Cam,
FrameRange=FR,
fpath=fpath))
# Convert list of scenes to SceneList
SL = bvp.SceneList(ScnList=ScnL, RenderOptions=RO)
#return SL
jIDs = SL.RenderSlurm(RenderGroupSize=1,
RenderType=('Voxels', ),
Is_Overwrite=True)
ConcatCmd = """import bvp, pickle, os
fD = '{fD}'
vox = bvp.utils.math.concatVoxels(fD)
sName = fD+'.pik'
bvp.utils.basics.savePik(vox, sName)
for f in os.listdir(fD): os.unlink(os.path.join(fD, f))
os.rmdir(fD)
"""
cjIDs = []
for jID, S in zip(jIDs, ScnL):
DepStr = 'afterok:%s' % jID # Dependencies for job
fD, xx = S.fpath.split('/') # last part of filepath from scene
fD = RO.BVPopts['BasePath'] % fD
cjID = bvp.utils.basics.pySlurm(
ConcatCmd.format(fD=fD),
dep=DepStr,
memory=2000,
)
cjIDs.append(cjID)
# Up priority... (base is 5000 as of 2012.12.31)
subprocess.call([
'sudo', 'scontrol', 'update', 'jobID=' + jID,
'Priority=5010'
])
# Final concatenation and clean-up
ConcatAllCmd = """import bvp, os
import numpy as np
I = np.zeros({res}**3)
fD = "{fD}"
fNm = sorted([os.path.join(fD, f) for f in os.listdir(fD) if 'pik' in f and "{key}" in f])
for f in fNm:
I+=np.array(bvp.utils.basics.loadPik(f))
I = bvp.utils.basics.make_blender_safe(I, 'float')
sName = os.path.join(fD, "{key}"+".pik")
bvp.utils.basics.savePik(I, sName)
for f in fNm:
os.unlink(f)
"""
DepStr = ('afterok' + ':%s' * len(cjIDs)) % tuple(
cjIDs) # Depends on all other concat jobs
fD, key = os.path.split(fD)
key = key[:-11] # Cut "_f000000000" part
bvp.utils.basics.pySlurm(
ConcatAllCmd.format(res=nGrid, fD=fD, key=key),
dep=DepStr,
memory=2000,
)
def RenderObjects(self,
Type=('Image', ),
subCat=None,
rotList=(0, ),
render_Pose=True,
renderGroupSize=1,
Is_Overwrite=False,
scaleObj=None):
"""
Render (all) objects in bvpLibrary
TODO: longer help!
ScaleObj = optional scale object to render along with this object (NOT FINISHED!)
"""
RO = bvp.RenderOptions()
RO.BVPopts['BasePath'] = os.path.join(self.LibDir, 'Images', 'Objects',
'Scenes', '%s')
RO.resolution_x = RO.resolution_y = 256 # smaller images
if subCat:
ToRender = self.getSCL(subCat, 'objects')
else:
ToRender = self.objects # all objects
ObCt = 0
ScnL = []
for o in ToRender:
# Get all object variations to add as separate scenes
ObToAdd = []
for rotZ in rotList:
if o['nPoses'] and render_Pose:
for p in range(o['nPoses']):
O = bvp.Object(obID=o['name'],
Lib=self,
pos3D=(0, 0, 0),
size3D=10,
rot3D=(0, 0, rotZ),
pose=p)
ObToAdd.append(O)
if scaleObj:
ScObSz = 10. * scaleObj.size3D / O.size3D
ScObToAdd.append
else:
O = bvp.Object(obID=o['name'],
Lib=self,
pos3D=(0, 0, 0),
size3D=10,
rot3D=(0, 0, rotZ))
ObToAdd.append(O)
# Add scale object in here somehwhere... Scale for each object!
if scaleObj:
ScObSz = 10. * scaleObj.size3D / O.size3D
ScObToAdd.append
# Camera, Lights (Sky), Background
Cam = bvp.Camera()
Sky = bvp.Sky()
BG = bvp.Background()
# Objects
for Obj in ObToAdd:
# Create Scene
ObCt += 1
if Obj.pose or Obj.pose == 0:
pNum = Obj.pose + 1
else:
pNum = 1
fpath = '%s_%s_p%d_r%d_fr##' % (Obj.semantic_category[0],
Obj.name, pNum, Obj.rot3D[2])
ScnL.append(
bvp.Scene(Num=ObCt,
Obj=(Obj, ),
BG=BG,
Sky=Sky,
Shadow=None,
Cam=Cam,
FrameRange=(1, 1),
fpath=fpath,
FrameRate=15))
# Convert list of scenes to SceneList
SL = bvp.SceneList(ScnList=ScnL, RenderOptions=RO)
SL.RenderSlurm(RenderGroupSize=renderGroupSize, RenderType=Type)