def calculateModeVolume(arguments):
# .. todo:: Finish this
# NOTE: Add way to specify snapshots, epsilon/frequency snapshot pairs
# read in mesh
if arguments.meshfile is None:
print('ERROR: No meshfile specified.', file=sys.stderr)
sys.exit(-1)
sim_mesh = bfdtd.readBristolFDTD(arguments.meshfile, arguments.verbosity)
# read in snapshot files from the various input files
if len(arguments.infile) <= 0:
print('ERROR: No infile(s) specified.', file=sys.stderr)
sys.exit(-1)
sim_in = bfdtd.BFDTDobject()
sim_in.verbosity = arguments.verbosity
for infile in arguments.infile:
sim_in.readBristolFDTD(infile)
# .. todo:: add path of file based on where it was read from
# .. todo:: read in .prn files
# calculate mode volume
snaplist = sim_in.getFrequencySnapshots()
for numID in range(len(snaplist)):
snapshot = snaplist[numID]
#print(['x','y','z'][snapshot.plane-1])
#print(sim_in.flag.id_string)
fsnap_filename, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_FrequencySnapshot(
numID + 1, ['x', 'y', 'z'][snapshot.plane - 1],
sim_in.flag.id_string.strip('"'),
snap_time_number=1)
print(fsnap_filename)
esnap_filename, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_EpsilonSnapshot(
numID + 1, ['x', 'y', 'z'][snapshot.plane - 1],
sim_in.flag.id_string.strip('"'),
snap_time_number=1)
print(esnap_filename)
#if arguments.fsnapfiles is None:
#arguments.fsnapfiles = sim_in.getFrequencySnapshots():
#if arguments.esnapfiles is None:
#arguments.esnapfiles = sim_in.getEpsilonSnapshots()
#if len(arguments.fsnapfiles) != len(arguments.esnapfiles):
#print('ERROR: number of frequency snapshots and epsilon snapshots do not match', file=sys.stderr)
#sys.exit(-1)
#else:
#print('OK')
#print(arguments.fsnapfiles)
#print(arguments.esnapfiles)
return
def calculateModeVolume(arguments):
# TODO: Finish this
# NOTE: Add way to specify snapshots, epsilon/frequency snapshot pairs
# read in mesh
if arguments.meshfile is None:
print("ERROR: No meshfile specified.", file=sys.stderr)
sys.exit(-1)
sim_mesh = bfdtd.readBristolFDTD(arguments.meshfile, arguments.verbosity)
# read in snapshot files from the various input files
if len(arguments.infile) <= 0:
print("ERROR: No infile(s) specified.", file=sys.stderr)
sys.exit(-1)
sim_in = bfdtd.BFDTDobject()
sim_in.verbosity = arguments.verbosity
for infile in arguments.infile:
sim_in.readBristolFDTD(infile)
# TODO: add path of file based on where it was read from
# TODO: read in .prn files
# calculate mode volume
snaplist = sim_in.getFrequencySnapshots()
for numID in range(len(snaplist)):
snapshot = snaplist[numID]
# print(['x','y','z'][snapshot.plane-1])
# print(sim_in.flag.id_string)
fsnap_filename, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_FrequencySnapshot(
numID + 1, ["x", "y", "z"][snapshot.plane - 1], sim_in.flag.id_string.strip('"'), snap_time_number=1
)
print(fsnap_filename)
esnap_filename, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_EpsilonSnapshot(
numID + 1, ["x", "y", "z"][snapshot.plane - 1], sim_in.flag.id_string.strip('"'), snap_time_number=1
)
print(esnap_filename)
# if arguments.fsnapfiles is None:
# arguments.fsnapfiles = sim_in.getFrequencySnapshots():
# if arguments.esnapfiles is None:
# arguments.esnapfiles = sim_in.getEpsilonSnapshots()
# if len(arguments.fsnapfiles) != len(arguments.esnapfiles):
# print('ERROR: number of frequency snapshots and epsilon snapshots do not match', file=sys.stderr)
# sys.exit(-1)
# else:
# print('OK')
# print(arguments.fsnapfiles)
# print(arguments.esnapfiles)
return
def importBristolFDTD(filename):
''' import BristolFDTD geometry from .in,.geo or .inp and create corresponding structure in Blender '''
print('----->Importing bristol FDTD geometry: ' + filename)
#Blender.Window.WaitCursor(1);
# save import path
# Blender.Set('tempdir',os.path.dirname(filename));
#FILE = open(, 'w');
#pickle.dump(, FILE);
#FILE.close();
with open(cfgfile, 'wb') as f:
# Pickle the 'data' dictionary using the highest protocol available.
pickle.dump(filename, f, pickle.HIGHEST_PROTOCOL)
# create structured_entries
structured_entries = readBristolFDTD(filename)
##################
# GROUP SETUP
##################
# create group corresponding to this file
# deselect all
bpy.ops.object.select_all(action='DESELECT')
# Truncated to 63 characters because that seems to be the maximum string length Blender accepts for group names.
# (allows keeping part of the path for better identification if multiple files use the same basename)
group_name = filename[-63:]
bpy.ops.group.create(name=group_name)
if not 'meshes' in bpy.data.groups: bpy.ops.group.create(name='meshes')
if not 'boxes' in bpy.data.groups: bpy.ops.group.create(name='boxes')
if not 'excitations' in bpy.data.groups:
bpy.ops.group.create(name='excitations')
if not 'frequencySnapshots' in bpy.data.groups:
bpy.ops.group.create(name='frequencySnapshots')
if not 'timeSnapshots' in bpy.data.groups:
bpy.ops.group.create(name='timeSnapshots')
if not 'epsilonSnapshots' in bpy.data.groups:
bpy.ops.group.create(name='epsilonSnapshots')
if not 'spheres' in bpy.data.groups: bpy.ops.group.create(name='spheres')
if not 'distorted' in bpy.data.groups:
bpy.ops.group.create(name='distorted')
if not 'blocks' in bpy.data.groups: bpy.ops.group.create(name='blocks')
if not 'cylinders' in bpy.data.groups:
bpy.ops.group.create(name='cylinders')
if not 'probes' in bpy.data.groups: bpy.ops.group.create(name='probes')
##################
FDTDGeometryObjects_obj = FDTDGeometryObjects()
# Blender.Window.RedrawAll(); # This must be called before any SetActiveLayer calls!
layerManager = LayerManagerObjects()
# Box
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('box'));
obj = FDTDGeometryObjects_obj.GEObox(structured_entries.box.name,
Vector(structured_entries.box.lower),
Vector(structured_entries.box.upper))
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='boxes')
# mesh
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('mesh'));
#FDTDGeometryObjects_obj.GEOmesh('mesh', False, structured_entries.delta_X_vector,structured_entries.delta_Y_vector,structured_entries.delta_Z_vector);
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('mesh'));
obj = FDTDGeometryObjects_obj.GEOmesh(
'mesh', False, structured_entries.mesh.getXmeshDelta(),
structured_entries.mesh.getYmeshDelta(),
structured_entries.mesh.getZmeshDelta())
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='meshes')
# Time_snapshot (time or EPS)
Ntsnaps = 0
for time_snapshot in structured_entries.time_snapshot_list:
Ntsnaps += 1
snap_plane = ['x', 'y', 'z'][time_snapshot.plane - 1]
probe_ident = structured_entries.flag.id_string.replace('\"', '')
snap_time_number = 1
TimeSnapshotFileName, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_EpsilonSnapshot(
Ntsnaps, snap_plane, probe_ident, snap_time_number)
if time_snapshot.eps == 0:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('time_snapshots_'+planeNumberName(time_snapshot.plane)[1]))
obj = FDTDGeometryObjects_obj.GEOtime_snapshot(
time_snapshot.name, time_snapshot.plane, time_snapshot.P1,
time_snapshot.P2)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='timeSnapshots')
else:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('eps_snapshots_'+planeNumberName(time_snapshot.plane)[1]))
obj = FDTDGeometryObjects_obj.GEOeps_snapshot(
TimeSnapshotFileName, time_snapshot.plane, time_snapshot.P1,
time_snapshot.P2)
#obj = FDTDGeometryObjects_obj.GEOeps_snapshot(time_snapshot.name, time_snapshot.plane, time_snapshot.P1, time_snapshot.P2)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='epsilonSnapshots')
# Frequency_snapshot
# TODO: Finally get a correct system for filenames/comment names/etc implemented. getfilename() or something...
Nfsnaps = 0
for frequency_snapshot in structured_entries.frequency_snapshot_list:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('frequency_snapshots_'+planeNumberName(frequency_snapshot.plane)[1]));
Nfsnaps += 1
snap_plane = ['x', 'y', 'z'][frequency_snapshot.plane - 1]
probe_ident = structured_entries.flag.id_string.replace('\"', '')
snap_time_number = 0
FrequencySnapshotFileName, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_FrequencySnapshot(
Nfsnaps, snap_plane, probe_ident, snap_time_number)
obj = FDTDGeometryObjects_obj.GEOfrequency_snapshot(
FrequencySnapshotFileName, frequency_snapshot.plane,
frequency_snapshot.P1, frequency_snapshot.P2)
#obj = FDTDGeometryObjects_obj.GEOfrequency_snapshot(frequency_snapshot.name, frequency_snapshot.plane, frequency_snapshot.P1, frequency_snapshot.P2)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='frequencySnapshots')
# Excitation
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('excitations'));
for excitation in structured_entries.excitation_list:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('excitations'));
#print(Blender.Window.GetActiveLayer())
#print(excitation)
obj = FDTDGeometryObjects_obj.GEOexcitation(excitation)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='excitations')
#FDTDGeometryObjects_obj.GEOexcitation(excitation.name, Vector(excitation.P1), Vector(excitation.P2));
# Probe
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('probes'));
Nprobes = 0
for probe in structured_entries.probe_list:
# print('probe = ',Vector(probe.position))
Nprobes += 1
ProbeFileName = 'p' + str(Nprobes).zfill(
2) + structured_entries.flag.id_string.replace('\"', '') + '.prn'
#FDTDGeometryObjects_obj.GEOprobe(probe.name+' ('+ProbeFileName+')', Vector(probe.position));
obj = FDTDGeometryObjects_obj.GEOprobe(ProbeFileName,
Vector(probe.position))
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='probes')
# Sphere
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('spheres'));
for sphere in structured_entries.sphere_list:
# variables
centre = Vector(sphere.centre)
# initialise rotation_matrix
rotation_matrix = Matrix()
rotation_matrix.identity()
# scale object
#Sx=Blender.Mathutils.ScaleMatrix(abs(2*sphere.outer_radius),4,Blender.Mathutils.Vector(1,0,0))
#Sy=Blender.Mathutils.ScaleMatrix(abs(2*sphere.outer_radius),4,Blender.Mathutils.Vector(0,1,0))
#Sz=Blender.Mathutils.ScaleMatrix(abs(2*sphere.outer_radius),4,Blender.Mathutils.Vector(0,0,1))
#rotation_matrix *= Sx*Sy*Sz;
# position object
#T = Blender.Mathutils.TranslationMatrix(centre)
#rotation_matrix *= T;
# add rotations
#for r in sphere.rotation_list:
#rotation_matrix *= rotationMatrix(r.axis_point, r.axis_direction, r.angle_degrees);
# create object
obj = FDTDGeometryObjects_obj.GEOsphere(sphere.name, sphere.centre,
sphere.outer_radius,
sphere.inner_radius,
sphere.permittivity,
sphere.conductivity)
#FDTDGeometryObjects_obj.GEOsphere_matrix(sphere.name, rotation_matrix, sphere.outer_radius, sphere.inner_radius, sphere.permittivity, sphere.conductivity);
#FDTDGeometryObjects_obj.GEOblock_matrix(sphere.name, rotation_matrix, sphere.permittivity, sphere.conductivity);
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='spheres')
# Block
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('blocks'))
for block in structured_entries.block_list:
# variables
lower = Vector(block.lower)
upper = Vector(block.upper)
pos = 0.5 * (lower + upper)
diag = 0.5 * (upper - lower)
# initialise rotation_matrix
rotation_matrix = Matrix()
rotation_matrix.identity()
# add rotations
for r in block.rotation_list:
rotation_matrix = rotationMatrix(r.axis_point, r.axis_direction,
r.angle_degrees) * rotation_matrix
# position object
T = Matrix.Translation(pos)
rotation_matrix *= T
## scale object
Sx = Matrix.Scale(abs(diag[0]), 4, Vector((1, 0, 0)))
Sy = Matrix.Scale(abs(diag[1]), 4, Vector((0, 1, 0)))
Sz = Matrix.Scale(abs(diag[2]), 4, Vector((0, 0, 1)))
rotation_matrix *= Sx * Sy * Sz
# create object
obj = FDTDGeometryObjects_obj.GEOblock_matrix(block.name,
rotation_matrix,
block.permittivity,
block.conductivity)
#FDTDGeometryObjects_obj.GEOblock(block.name, block.lower, block.upper, block.permittivity, block.conductivity)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='blocks')
# Distorted
for distorted in structured_entries.distorted_list:
# create object
#print(distorted)
obj = FDTDGeometryObjects_obj.GEOdistorted(distorted.name,
distorted.vertices,
distorted.permittivity,
distorted.conductivity)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='distorted')
# Cylinder
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('cylinders'));
for cylinder in structured_entries.cylinder_list:
# initialise rotation_matrix
rotation_matrix = Matrix()
rotation_matrix.identity()
# add rotations
for r in cylinder.rotation_list:
rotation_matrix = rotationMatrix(r.axis_point, r.axis_direction,
r.angle_degrees) * rotation_matrix
# position object
T = Matrix.Translation(
Vector(
[cylinder.centre[0], cylinder.centre[1], cylinder.centre[2]]))
rotation_matrix *= T
# because FDTD cylinders are aligned with the Y axis by default
rotation_matrix *= rotationMatrix(Vector([0, 0, 0]), Vector([1, 0, 0]),
-90)
# create object
obj = FDTDGeometryObjects_obj.GEOcylinder_matrix(
cylinder.name, rotation_matrix, cylinder.inner_radius,
cylinder.outer_radius, cylinder.height, cylinder.permittivity,
cylinder.conductivity)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='cylinders')
#angle_X = numpy.deg2rad(-90)
#angle_X = -0.5*numpy.pi
#angle_Y = 0
#angle_Z = 0
#FDTDGeometryObjects_obj.GEOcylinder(cylinder.name, cylinder.centre, cylinder.inner_radius, cylinder.outer_radius, cylinder.height, cylinder.permittivity, cylinder.conductivity, angle_X, angle_Y, angle_Z)
#########################
# Not yet implemented:
# Flag
# structured_entries.flag;
# Boundaries
# structured_entries.boundaries;
#########################
# TODO: Save the layer settings somewhere for reuse
#scene = Blender.Scene.GetCurrent()
scene = bpy.context.scene
layersOn = [
layerManager.DefaultLayers.index('spheres') + 1,
layerManager.DefaultLayers.index('blocks') + 1,
layerManager.DefaultLayers.index('cylinders') + 1
]
print(layersOn)
#layersOn = [1,2,3]
#print layersOn
#Blender.Scene.GetCurrent().setLayers(layersOn)
#scene.update(0);
#Blender.Window.RedrawAll();
#Blender.Window.WaitCursor(0);
#Blender.Scene.GetCurrent().setLayers([1,3,4,5,6,7,8,9,10]);
print('...done')
def importBristolFDTD(filename):
''' import BristolFDTD geometry from .in,.geo or .inp and create corresponding structure in Blender '''
print('----->Importing bristol FDTD geometry: '+filename)
#Blender.Window.WaitCursor(1);
# save import path
# Blender.Set('tempdir',os.path.dirname(filename));
#FILE = open(, 'w');
#pickle.dump(, FILE);
#FILE.close();
with open(cfgfile, 'wb') as f:
# Pickle the 'data' dictionary using the highest protocol available.
pickle.dump(filename, f, pickle.HIGHEST_PROTOCOL)
# create structured_entries
structured_entries = readBristolFDTD(filename);
##################
# GROUP SETUP
##################
# create group corresponding to this file
# deselect all
bpy.ops.object.select_all(action='DESELECT')
# Truncated to 63 characters because that seems to be the maximum string length Blender accepts for group names.
# (allows keeping part of the path for better identification if multiple files use the same basename)
group_name = filename[-63:]
bpy.ops.group.create(name=group_name)
if not 'meshes' in bpy.data.groups: bpy.ops.group.create(name='meshes')
if not 'boxes' in bpy.data.groups: bpy.ops.group.create(name='boxes')
if not 'excitations' in bpy.data.groups: bpy.ops.group.create(name='excitations')
if not 'frequencySnapshots' in bpy.data.groups: bpy.ops.group.create(name='frequencySnapshots')
if not 'timeSnapshots' in bpy.data.groups: bpy.ops.group.create(name='timeSnapshots')
if not 'epsilonSnapshots' in bpy.data.groups: bpy.ops.group.create(name='epsilonSnapshots')
if not 'spheres' in bpy.data.groups: bpy.ops.group.create(name='spheres')
if not 'distorted' in bpy.data.groups: bpy.ops.group.create(name='distorted')
if not 'blocks' in bpy.data.groups: bpy.ops.group.create(name='blocks')
if not 'cylinders' in bpy.data.groups: bpy.ops.group.create(name='cylinders')
if not 'probes' in bpy.data.groups: bpy.ops.group.create(name='probes')
##################
FDTDGeometryObjects_obj = FDTDGeometryObjects()
# Blender.Window.RedrawAll(); # This must be called before any SetActiveLayer calls!
layerManager = LayerManagerObjects()
# Box
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('box'));
obj = FDTDGeometryObjects_obj.GEObox(structured_entries.box.name, Vector(structured_entries.box.lower), Vector(structured_entries.box.upper));
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='boxes')
# mesh
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('mesh'));
#FDTDGeometryObjects_obj.GEOmesh('mesh', False, structured_entries.delta_X_vector,structured_entries.delta_Y_vector,structured_entries.delta_Z_vector);
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('mesh'));
obj = FDTDGeometryObjects_obj.GEOmesh('mesh', False, structured_entries.mesh.getXmeshDelta(),structured_entries.mesh.getYmeshDelta(),structured_entries.mesh.getZmeshDelta());
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='meshes')
# Time_snapshot (time or EPS)
Ntsnaps = 0
for time_snapshot in structured_entries.time_snapshot_list:
Ntsnaps += 1
snap_plane = ['x','y','z'][time_snapshot.plane - 1]
probe_ident = structured_entries.flag.id_string.replace('\"','')
snap_time_number = 1
TimeSnapshotFileName, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_EpsilonSnapshot(Ntsnaps, snap_plane, probe_ident, snap_time_number)
if time_snapshot.eps == 0:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('time_snapshots_'+planeNumberName(time_snapshot.plane)[1]))
obj = FDTDGeometryObjects_obj.GEOtime_snapshot(time_snapshot.name, time_snapshot.plane, time_snapshot.P1, time_snapshot.P2)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='timeSnapshots')
else:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('eps_snapshots_'+planeNumberName(time_snapshot.plane)[1]))
obj = FDTDGeometryObjects_obj.GEOeps_snapshot(TimeSnapshotFileName, time_snapshot.plane, time_snapshot.P1, time_snapshot.P2)
#obj = FDTDGeometryObjects_obj.GEOeps_snapshot(time_snapshot.name, time_snapshot.plane, time_snapshot.P1, time_snapshot.P2)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='epsilonSnapshots')
# Frequency_snapshot
# TODO: Finally get a correct system for filenames/comment names/etc implemented. getfilename() or something...
Nfsnaps = 0
for frequency_snapshot in structured_entries.frequency_snapshot_list:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('frequency_snapshots_'+planeNumberName(frequency_snapshot.plane)[1]));
Nfsnaps += 1
snap_plane = ['x','y','z'][frequency_snapshot.plane - 1]
probe_ident = structured_entries.flag.id_string.replace('\"','')
snap_time_number = 0
FrequencySnapshotFileName, alphaID, pair = brisFDTD_ID_info.numID_to_alphaID_FrequencySnapshot(Nfsnaps, snap_plane, probe_ident, snap_time_number)
obj = FDTDGeometryObjects_obj.GEOfrequency_snapshot(FrequencySnapshotFileName, frequency_snapshot.plane, frequency_snapshot.P1, frequency_snapshot.P2)
#obj = FDTDGeometryObjects_obj.GEOfrequency_snapshot(frequency_snapshot.name, frequency_snapshot.plane, frequency_snapshot.P1, frequency_snapshot.P2)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='frequencySnapshots')
# Excitation
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('excitations'));
for excitation in structured_entries.excitation_list:
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('excitations'));
#print(Blender.Window.GetActiveLayer())
#print(excitation)
obj = FDTDGeometryObjects_obj.GEOexcitation(excitation);
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='excitations')
#FDTDGeometryObjects_obj.GEOexcitation(excitation.name, Vector(excitation.P1), Vector(excitation.P2));
# Probe
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('probes'));
Nprobes = 0
for probe in structured_entries.probe_list:
# print('probe = ',Vector(probe.position))
Nprobes += 1
ProbeFileName = 'p' + str(Nprobes).zfill(2) + structured_entries.flag.id_string.replace('\"','') + '.prn'
#FDTDGeometryObjects_obj.GEOprobe(probe.name+' ('+ProbeFileName+')', Vector(probe.position));
obj = FDTDGeometryObjects_obj.GEOprobe(ProbeFileName, Vector(probe.position));
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='probes')
# Sphere
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('spheres'));
for sphere in structured_entries.sphere_list:
# variables
centre = Vector(sphere.centre)
# initialise rotation_matrix
rotation_matrix = Matrix()
rotation_matrix.identity();
# scale object
#Sx=Blender.Mathutils.ScaleMatrix(abs(2*sphere.outer_radius),4,Blender.Mathutils.Vector(1,0,0))
#Sy=Blender.Mathutils.ScaleMatrix(abs(2*sphere.outer_radius),4,Blender.Mathutils.Vector(0,1,0))
#Sz=Blender.Mathutils.ScaleMatrix(abs(2*sphere.outer_radius),4,Blender.Mathutils.Vector(0,0,1))
#rotation_matrix *= Sx*Sy*Sz;
# position object
#T = Blender.Mathutils.TranslationMatrix(centre)
#rotation_matrix *= T;
# add rotations
#for r in sphere.rotation_list:
#rotation_matrix *= rotationMatrix(r.axis_point, r.axis_direction, r.angle_degrees);
# create object
obj = FDTDGeometryObjects_obj.GEOsphere(sphere.name, sphere.centre, sphere.outer_radius, sphere.inner_radius, sphere.permittivity, sphere.conductivity)
#FDTDGeometryObjects_obj.GEOsphere_matrix(sphere.name, rotation_matrix, sphere.outer_radius, sphere.inner_radius, sphere.permittivity, sphere.conductivity);
#FDTDGeometryObjects_obj.GEOblock_matrix(sphere.name, rotation_matrix, sphere.permittivity, sphere.conductivity);
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='spheres')
# Block
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('blocks'))
for block in structured_entries.block_list:
# variables
lower = Vector(block.lower)
upper = Vector(block.upper)
pos = 0.5*(lower+upper)
diag = 0.5*(upper-lower)
# initialise rotation_matrix
rotation_matrix = Matrix()
rotation_matrix.identity()
# add rotations
for r in block.rotation_list:
rotation_matrix = rotationMatrix(r.axis_point, r.axis_direction, r.angle_degrees)*rotation_matrix
# position object
T = Matrix.Translation(pos)
rotation_matrix *= T
## scale object
Sx = Matrix.Scale(abs(diag[0]), 4, Vector((1,0,0)) )
Sy = Matrix.Scale(abs(diag[1]), 4, Vector((0,1,0)) )
Sz = Matrix.Scale(abs(diag[2]), 4, Vector((0,0,1)) )
rotation_matrix *= Sx*Sy*Sz;
# create object
obj = FDTDGeometryObjects_obj.GEOblock_matrix(block.name, rotation_matrix, block.permittivity, block.conductivity)
#FDTDGeometryObjects_obj.GEOblock(block.name, block.lower, block.upper, block.permittivity, block.conductivity)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='blocks')
# Distorted
for distorted in structured_entries.distorted_list:
# create object
#print(distorted)
obj = FDTDGeometryObjects_obj.GEOdistorted(distorted.name, distorted.vertices, distorted.permittivity, distorted.conductivity);
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='distorted')
# Cylinder
#Blender.Window.SetActiveLayer(1<<layerManager.DefaultLayers.index('cylinders'));
for cylinder in structured_entries.cylinder_list:
# initialise rotation_matrix
rotation_matrix = Matrix()
rotation_matrix.identity()
# add rotations
for r in cylinder.rotation_list:
rotation_matrix = rotationMatrix(r.axis_point, r.axis_direction, r.angle_degrees)*rotation_matrix
# position object
T = Matrix.Translation(Vector([cylinder.centre[0],cylinder.centre[1],cylinder.centre[2]]))
rotation_matrix *= T;
# because FDTD cylinders are aligned with the Y axis by default
rotation_matrix *= rotationMatrix(Vector([0,0,0]), Vector([1,0,0]), -90)
# create object
obj = FDTDGeometryObjects_obj.GEOcylinder_matrix(cylinder.name, rotation_matrix, cylinder.inner_radius, cylinder.outer_radius, cylinder.height, cylinder.permittivity, cylinder.conductivity)
bpy.context.scene.objects.active = obj
bpy.ops.object.group_link(group=group_name)
bpy.ops.object.group_link(group='cylinders')
#angle_X = numpy.deg2rad(-90)
#angle_X = -0.5*numpy.pi
#angle_Y = 0
#angle_Z = 0
#FDTDGeometryObjects_obj.GEOcylinder(cylinder.name, cylinder.centre, cylinder.inner_radius, cylinder.outer_radius, cylinder.height, cylinder.permittivity, cylinder.conductivity, angle_X, angle_Y, angle_Z)
#########################
# Not yet implemented:
# Flag
# structured_entries.flag;
# Boundaries
# structured_entries.boundaries;
#########################
# TODO: Save the layer settings somewhere for reuse
#scene = Blender.Scene.GetCurrent()
scene = bpy.context.scene
layersOn = [layerManager.DefaultLayers.index('spheres')+1,layerManager.DefaultLayers.index('blocks')+1,layerManager.DefaultLayers.index('cylinders')+1]
print(layersOn)
#layersOn = [1,2,3]
#print layersOn
#Blender.Scene.GetCurrent().setLayers(layersOn)
#scene.update(0);
#Blender.Window.RedrawAll();
#Blender.Window.WaitCursor(0);
#Blender.Scene.GetCurrent().setLayers([1,3,4,5,6,7,8,9,10]);
print('...done')
