@@ -67,8 +67,10 @@ def __init__(self, num_nodes=None, num_elements=None, mesh=None, server=None):
67
67
self ._server = server_module .get_or_create_server (server )
68
68
69
69
# step 2: get api
70
- self ._api = self ._server .get_api_for_type (capi = meshed_region_capi .MeshedRegionCAPI ,
71
- grpcapi = meshed_region_grpcapi .MeshedRegionGRPCAPI )
70
+ self ._api = self ._server .get_api_for_type (
71
+ capi = meshed_region_capi .MeshedRegionCAPI ,
72
+ grpcapi = meshed_region_grpcapi .MeshedRegionGRPCAPI ,
73
+ )
72
74
73
75
# step3: init environment
74
76
self ._api .init_meshed_region_environment (self ) # creates stub when gRPC
@@ -80,7 +82,9 @@ def __init__(self, num_nodes=None, num_elements=None, mesh=None, server=None):
80
82
else :
81
83
# if no mesh object, create one
82
84
if self ._server .has_client ():
83
- self ._internal_obj = self ._api .meshed_region_new_on_client (self ._server .client )
85
+ self ._internal_obj = self ._api .meshed_region_new_on_client (
86
+ self ._server .client
87
+ )
84
88
else :
85
89
self ._internal_obj = self ._api .meshed_region_new ()
86
90
@@ -223,10 +227,13 @@ def available_property_fields(self):
223
227
available_property_fields : list str
224
228
"""
225
229
available_property_fields = []
226
- n_property_field = self ._api .meshed_region_get_num_available_property_field (self )
230
+ n_property_field = self ._api .meshed_region_get_num_available_property_field (
231
+ self
232
+ )
227
233
for index in range (n_property_field ):
228
- available_property_fields .append (self ._api .meshed_region_get_property_field_name (self ,
229
- index ))
234
+ available_property_fields .append (
235
+ self ._api .meshed_region_get_property_field_name (self , index )
236
+ )
230
237
return available_property_fields
231
238
232
239
def property_field (self , property_name ):
@@ -290,7 +297,9 @@ def _get_available_named_selections(self):
290
297
named_selections = []
291
298
n_selections = self ._api .meshed_region_get_num_available_named_selection (self )
292
299
for index in range (n_selections ):
293
- named_selections .append (self ._api .meshed_region_get_named_selection_name (self , index ))
300
+ named_selections .append (
301
+ self ._api .meshed_region_get_named_selection_name (self , index )
302
+ )
294
303
return named_selections
295
304
296
305
def named_selection (self , named_selection ):
@@ -307,7 +316,9 @@ def named_selection(self, named_selection):
307
316
named_selection : Scoping
308
317
"""
309
318
if server_meet_version ("2.1" , self ._server ):
310
- out = self ._api .meshed_region_get_named_selection_scoping (self , named_selection )
319
+ out = self ._api .meshed_region_get_named_selection_scoping (
320
+ self , named_selection
321
+ )
311
322
return scoping .Scoping (scoping = out , server = self ._server )
312
323
else :
313
324
if hasattr (self , "_stream_provider" ):
@@ -335,8 +346,9 @@ def set_named_selection_scoping(self, named_selection_name, scoping):
335
346
named selection name
336
347
scoping : Scoping
337
348
"""
338
- return self ._api .meshed_region_set_named_selection_scoping (self ,
339
- named_selection_name , scoping )
349
+ return self ._api .meshed_region_set_named_selection_scoping (
350
+ self , named_selection_name , scoping
351
+ )
340
352
341
353
def _set_stream_provider (self , stream_provider ):
342
354
self ._stream_provider = stream_provider
@@ -382,7 +394,7 @@ def _set_stream_provider(self, stream_provider):
382
394
# self._internal_obj = skin.get_output(0, types.meshed_region)
383
395
# return MeshedRegion(self._server.channel, skin, self._model, name)
384
396
385
- def deform_by (self , deform_by , scale_factor = 1. ):
397
+ def deform_by (self , deform_by , scale_factor = 1.0 ):
386
398
"""
387
399
Deforms the mesh according to a 3D vector field and an additional scale factor.
388
400
@@ -399,10 +411,11 @@ def deform_by(self, deform_by, scale_factor=1.):
399
411
400
412
"""
401
413
from ansys .dpf .core .operators .math import add , scale
414
+
402
415
scale_op = scale (field = deform_by , ponderation = scale_factor )
403
- return add (fieldA = self . nodes . coordinates_field ,
404
- fieldB = scale_op .outputs .field
405
- ).eval ()
416
+ return add (
417
+ fieldA = self . nodes . coordinates_field , fieldB = scale_op .outputs .field
418
+ ).eval ()
406
419
407
420
def _as_vtk (self , coordinates = None , as_linear = True , include_ids = False ):
408
421
"""Convert DPF mesh to a PyVista unstructured grid."""
@@ -465,12 +478,12 @@ def grid(self):
465
478
return self ._full_grid
466
479
467
480
def plot (
468
- self ,
469
- field_or_fields_container = None ,
470
- shell_layers = None ,
471
- deform_by = None ,
472
- scale_factor = 1.0 ,
473
- ** kwargs
481
+ self ,
482
+ field_or_fields_container = None ,
483
+ shell_layers = None ,
484
+ deform_by = None ,
485
+ scale_factor = 1.0 ,
486
+ ** kwargs ,
474
487
):
475
488
"""
476
489
Plot the field or fields container on the mesh.
@@ -504,15 +517,24 @@ def plot(
504
517
"""
505
518
if field_or_fields_container is not None :
506
519
pl = Plotter (self , ** kwargs )
507
- return pl .plot_contour (field_or_fields_container , shell_layers ,
508
- show_axes = kwargs .pop ("show_axes" , True ),
509
- deform_by = deform_by ,
510
- scale_factor = scale_factor , ** kwargs )
520
+ return pl .plot_contour (
521
+ field_or_fields_container ,
522
+ shell_layers ,
523
+ show_axes = kwargs .pop ("show_axes" , True ),
524
+ deform_by = deform_by ,
525
+ scale_factor = scale_factor ,
526
+ ** kwargs ,
527
+ )
511
528
512
529
# otherwise, simply plot the mesh
513
530
pl = DpfPlotter (** kwargs )
514
- pl .add_mesh (self , deform_by = deform_by , scale_factor = scale_factor ,
515
- show_axes = kwargs .pop ("show_axes" , True ), ** kwargs )
531
+ pl .add_mesh (
532
+ self ,
533
+ deform_by = deform_by ,
534
+ scale_factor = scale_factor ,
535
+ show_axes = kwargs .pop ("show_axes" , True ),
536
+ ** kwargs ,
537
+ )
516
538
return pl .show_figure (** kwargs )
517
539
518
540
def deep_copy (self , server = None ):
@@ -595,12 +617,15 @@ def field_of_properties(self, property_name):
595
617
else :
596
618
field_out = self ._api .meshed_region_get_property_field (self , property_name )
597
619
if isinstance (field_out , int ):
598
- res = property_field .PropertyField (server = self ._server , property_field = field_out )
620
+ res = property_field .PropertyField (
621
+ server = self ._server , property_field = field_out
622
+ )
599
623
return res
600
624
else :
601
625
if field_out .datatype == "int" :
602
- return property_field .PropertyField (server = self ._server ,
603
- property_field = field_out )
626
+ return property_field .PropertyField (
627
+ server = self ._server , property_field = field_out
628
+ )
604
629
else :
605
630
# Not sure we go through here since the only datatype not int is coordinates,
606
631
# which is already dealt with previously.
0 commit comments