def test_break_quad4_to_quad9_to_quad9(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) doGenSideSets = True if p_size <= 3: n = 12 nx = n ny = n fixture = QuadFixture_4(pm, nx, ny, doGenSideSets) isCommited = False eMesh = PerceptMesh(fixture.meta_data, fixture.bulk_data, isCommited) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, QUAD4_QUAD9_1, proc_rank_field) eMesh.commit() fixture.generate_mesh() breaker.doBreak() eMesh.save_as("quad_fixture_quad9_quad9_0.e") em1 = PerceptMesh(2) em1.open("quad_fixture_quad9_quad9_0.e") scalarDimension = 0 proc_rank_field = em1.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(em1, QUAD9_QUAD9_4, proc_rank_field) em1.commit() breaker.doBreak() em1.save_as("quad_fixture_quad9_quad9_1.e")
def test_break_tri3_to_tri6_sierra(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) print p_size, "++++++++++++++++++++" for i in range(100): print "-------------------" if p_size <= 3: n = 12 nx = n ny = n createEdgeSets = True fixture = QuadFixture_3(pm, nx, ny, createEdgeSets) isCommited = False eMesh = PerceptMesh(fixture.meta_data, fixture.bulk_data, isCommited) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) eMesh.add_field("proc_rank_ede", eMesh.edge_rank(), scalarDimension) breaker = Refiner(eMesh, TRI3_TRI6_1, proc_rank_field) eMesh.commit() fixture.generate_mesh() eMesh.print_info("tri mesh tri6") eMesh.save_as("quad_fixture_tri3_tri6_0.e") breaker.doBreak() eMesh.print_info("tri mesh enriched") eMesh.save_as("quad_fixture_tri6_tri6_0.e") eMesh.save_as("quad_fixture_tri3_tri6_1.e")
def test_hex27_hex27_0(self): fixture_setup() scalarDimension = 0 eMesh = PerceptMesh() p_size = eMesh.get_parallel_size() gmesh_spec = "1x1x" + str(p_size) + "|bbox:0,0,0,1,1," + str(p_size) eMesh.new_mesh(GMeshSpec(gmesh_spec)) proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX8_HEX27_1, proc_rank_field) eMesh.commit() eMesh.print_info() eMesh.save_as("hex27_hex27_cube1x1x" + str(p_size) + "-orig.e") breaker.setRemoveOldElements(True) breaker.doBreak() eMesh.save_as("hex27_hex27_cube1x1x" + str(p_size) + "_0.e") em1 = PerceptMesh(3) p_size = em1.get_parallel_size() em1.open("hex27_hex27_cube1x1x" + str(p_size) + "_0.e") proc_rank_field = em1.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(em1, HEX27_HEX27_8, proc_rank_field) em1.commit() breaker.setIgnoreSideSets(True) breaker.setRemoveOldElements(True) breaker.doBreak() em1.save_as("hex27_hex27_cube1x1x" + str(p_size) + "_1.e")
def test_break_tet4_tet10_tet10_1(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size == 1 or p_size == 3: eMesh = PerceptMesh(3) eMesh.open("tet_from_hex_fixture_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, TET4_TET10_1, proc_rank_field) eMesh.commit() eMesh.print_info("tet mesh") breaker.doBreak() eMesh.save_as("tet10_1.e") eMesh.print_info("tet10_1") if p_size == 1 or p_size == 3: eMesh = PerceptMesh(3) eMesh.open("tet10_1.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, TET10_TET10_8, proc_rank_field) eMesh.commit() breaker.doBreak() eMesh.save_as("tet10_tet10_1.e")
def test_wedge6_enrich_refine(self): fixture_setup() p_size = parallel_machine_size(MPI.COMM_WORLD) if p_size == 1: eMesh = PerceptMesh(3) wedgeFixture = WedgeFixture() wedgeFixture.createMesh(MPI.COMM_WORLD, 4, 2, 2, 0, 1, 0, 1, 0, 1, "tmp-swept-wedge_enrich_0.e") eMesh.open("tmp-swept-wedge_enrich_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, WEDGE6_WEDGE15_1, proc_rank_field) eMesh.commit() breaker.doBreak() eMesh.save_as("swept-wedge_2_enrich_refine_0.e") if p_size == 1: eMesh = PerceptMesh(3) eMesh.open("swept-wedge_2_enrich_refine_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, WEDGE15_WEDGE15_8, proc_rank_field) eMesh.commit() breaker.setIgnoreSideSets(True) breaker.doBreak() eMesh.save_as("swept-wedge_2_enrich_refine_1.e")
def test_hex27_hex27_1(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size == 1 or p_size == 3: eMesh = PerceptMesh(3) eMesh.open("hex_fixture.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX8_HEX27_1, proc_rank_field) eMesh.commit() eMesh.print_info() eMesh.save_as("hex8_hex27_0.e") breaker.doBreak() eMesh.save_as("hex8_27_1.e") if p_size == 1 or p_size == 3: eMesh = PerceptMesh(3) eMesh.open("hex8_27_1.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX27_HEX27_8, proc_rank_field) eMesh.commit() breaker.setRemoveOldElements(True) breaker.doBreak() eMesh.save_as("hex8_hex27_hex27_1.e")
def test_fieldFunction_demo_2(self): eMesh = PerceptMesh() eMesh.new_mesh(GMeshSpec("3x3x3|bbox:0,0,0,1,1,1")) vectorDimension = 0 eMesh.add_field("coords_mag_field", FEMMetaData.NODE_RANK, vectorDimension) eMesh.commit() f_coords = eMesh.get_field("coordinates") coords_mag_field = eMesh.get_field("coords_mag_field") ff_coords = FieldFunction("ff_coords", f_coords, eMesh, 3, 3) eval_vec3_print(0.1,0.1,0.1,0.0,ff_coords) coords_mag_sf = StringFunction("sqrt(x*x + y*y + z*z)" , "coords_mag_sf", 3, 1) x = 0.123 y = 0.234 z = 0.345 vv = sqrt(x*x + y*y + z*z) v1 = eval_func(x,y,z,0,coords_mag_sf) print "vv = ", vv, "== v1 = ", v1 self.assertEqual(vv, v1) coords_mag_field_function = FieldFunction("coords_mag_field_function", coords_mag_field, eMesh, 3, 1) coords_mag_field_function.interpolateFrom(coords_mag_sf) eMesh.save_as("./cubehex8_withCoordMag_out.e") ff_coords.add_alias("mc") sfcm = StringFunction("sqrt(mc[0]*mc[0]+mc[1]*mc[1]+mc[2]*mc[2])", "sfcm", 3, 1)
def test_fieldFunction_readMesh_createField_interpolateFrom(self): num_x = 3 num_y = 3 num_z = 3 config_mesh = str(num_x) + "x" + str(num_y) + "x" + str(num_z) + "|bbox:0,0,0,1,1,1" eMesh = PerceptMesh() eMesh.new_mesh(GMeshSpec(config_mesh)) vectorDimension = 0 eMesh.add_field("coords_mag_field", FEMMetaData.NODE_RANK, vectorDimension) eMesh.commit() #p_rank = eMesh.get_bulk_data().parallel_rank() #setRank(p_rank) #from Util f_coords = eMesh.get_field("coordinates") coords_mag_field = eMesh.get_field("coords_mag_field") #VERIFY_OP_ON Here the unit test does something ff_coords = FieldFunction("ff_coords", f_coords, eMesh, 3, 3, FieldFunction.SIMPLE_SEARCH) #here we could evaluate the function #eval_vec3_print(0.1,0.2,0.3,0.0,ff_coords) coords_mag_sf = StringFunction("sqrt(x*x + y*y + z*z)", "coords_mag_sf", 3, 1) coords_mag_field_function = FieldFunction("coords_mag_field_function", coords_mag_field, eMesh, 3, 3, FieldFunction.SIMPLE_SEARCH) coords_mag_field_function.interpolateFrom(coords_mag_sf) #The following is not doable from Python checkCoordMag = CheckCoordMag() #eMesh.nodalOpLoop(checkCoordMag, coords_mag_field) print checkCoordMag.error ff_coords.add_alias("mc") sfcm = StringFunction("sqrt(mc[0]*mc[0]+mc[1]*mc[1]+mc[2]*mc[2])", "sfcm", Dimensions(3), Dimensions(1)) tol1 = 1.e-12 vv = eval_vec3(0.1, 0.2, 0.3, 0.0, ff_coords) print print "0.1 == vv[0] = ", vv[0], "passed" print "0.2 == vv[1] = ", vv[1], "passed" print "0.3 == vv[2] = ", vv[2], "passed" self.assertAlmostEqual(.1, vv[0], delta=tol1) self.assertAlmostEqual(.2, vv[1], delta=tol1) self.assertAlmostEqual(.3, vv[2], delta=tol1) vv = eval_func(0.1, 0.2, 0.3, 0.0, sfcm) v_expect = sqrt(0.1*0.1+0.2*0.2+0.3*0.3) if ((vv-v_expect) < tol1): print "vv = ", vv, " == v_expect = ", v_expect, "passed" coords_mag_field_function.interpolateFrom(sfcm)
def test_fieldFunction_readMesh_createField_interpolateFrom(self): num_x = 3 num_y = 3 num_z = 3 config_mesh = str(num_x) + "x" + str(num_y) + "x" + str(num_z) + "|bbox:0,0,0,1,1,1" eMesh = PerceptMesh() eMesh.new_mesh(GMeshSpec(config_mesh)) vectorDimension = 0 eMesh.add_field("coords_mag_field", FEMMetaData.NODE_RANK, vectorDimension) eMesh.commit() #p_rank = eMesh.get_bulk_data().parallel_rank() #setRank(p_rank) #from Util f_coords = eMesh.get_field(FEMMetaData.NODE_RANK, "coordinates") coords_mag_field = eMesh.get_field(FEMMetaData.NODE_RANK, "coords_mag_field") #VERIFY_OP_ON Here the unit test does something ff_coords = FieldFunction("ff_coords", f_coords, eMesh, 3, 3, FieldFunction.SIMPLE_SEARCH) #here we could evaluate the function #eval_vec3_print(0.1,0.2,0.3,0.0,ff_coords) coords_mag_sf = StringFunction("sqrt(x*x + y*y + z*z)", "coords_mag_sf", 3, 1) coords_mag_field_function = FieldFunction("coords_mag_field_function", coords_mag_field, eMesh, 3, 3, FieldFunction.SIMPLE_SEARCH) coords_mag_field_function.interpolateFrom(coords_mag_sf) #The following is not doable from Python checkCoordMag = CheckCoordMag() #eMesh.nodalOpLoop(checkCoordMag, coords_mag_field) print checkCoordMag.error ff_coords.add_alias("mc") sfcm = StringFunction("sqrt(mc[0]*mc[0]+mc[1]*mc[1]+mc[2]*mc[2])", "sfcm", Dimensions(3), Dimensions(1)) tol1 = 1.e-12 vv = eval_vec3(0.1, 0.2, 0.3, 0.0, ff_coords) print print "0.1 == vv[0] = ", vv[0], "passed" print "0.2 == vv[1] = ", vv[1], "passed" print "0.3 == vv[2] = ", vv[2], "passed" self.assertAlmostEqual(.1, vv[0], delta=tol1) self.assertAlmostEqual(.2, vv[1], delta=tol1) self.assertAlmostEqual(.3, vv[2], delta=tol1) vv = eval_func(0.1, 0.2, 0.3, 0.0, sfcm) v_expect = sqrt(0.1*0.1+0.2*0.2+0.3*0.3) if ((vv-v_expect) < tol1): print "vv = ", vv, " == v_expect = ", v_expect, "passed" coords_mag_field_function.interpolateFrom(sfcm)
def test_break_quad_to_quad_sierra_2(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) doGenSideSets = True if p_size <= 3: n = 12 nx = n ny = n fixture = QuadFixture_4(pm, nx, ny, doGenSideSets) isCommited = False eMesh = PerceptMesh(fixture.meta_data, fixture.bulk_data, isCommited) eMesh.commit() fixture.generate_mesh() eMesh.save_as("quad_fixture_mbreak_0.e") eMesh.close() eMesh1 = PerceptMesh(2) eMesh1.open("quad_fixture_mbreak_0.e") scalarDimension = 0 proc_rank_field = eMesh1.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh1, QUAD4_QUAD4_4_SIERRA, proc_rank_field) eMesh1.commit() i = 0 while i < 2: print "\n\n\n ================ tmp Refine Pass = "******"quad_fixture_mbreak_" + str(i) + ".e") i = i + 1
def test_break_quad_to_quad_sierra_2(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) doGenSideSets = True if p_size <= 3: n = 12 nx = n ny = n fixture = QuadFixture_4(pm,nx,ny,doGenSideSets) isCommited = False eMesh = PerceptMesh(fixture.meta_data, fixture.bulk_data, isCommited) eMesh.commit() fixture.generate_mesh() eMesh.save_as("quad_fixture_mbreak_0.e") eMesh.close() eMesh1 = PerceptMesh(2) eMesh1.open("quad_fixture_mbreak_0.e") scalarDimension = 0 proc_rank_field = eMesh1.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh1, QUAD4_QUAD4_4_SIERRA, proc_rank_field) eMesh1.commit() i = 0 while i < 2: print "\n\n\n ================ tmp Refine Pass = "******"quad_fixture_mbreak_" + str(i) + ".e") i = i + 1
def test_use_case_1(self): pMesh = PerceptMesh() pMesh.new_mesh(GMeshSpec("3x3x3|bbox:0,0,0,2,2,2")) field = pMesh.add_field("coordinates", 1) pMesh.commit() input_array = array([1.0, 0.5, 0.5]) input_array_2 = array([1.0, 1.5, 1.5]) ff = FieldFunction("ff", field, pMesh, 3, 3) ff.add_alias("myalias") ff_output = ff.evaluate(input_array) f2 = FieldFunction("f2", field, pMesh, 3, 3) f2_output = f2.evaluate(input_array_2) sf = StringFunction("x+y+z", "myname", 3, 1) sf_output = sf.evaluate(input_array) sf_diff = StringFunction("ff-f2", "myname") norm = L1Norm(pMesh.get_bulk_data()) value = norm.evaluate(ff) diffnorm = norm.evaluate(sf_diff)
def test_fieldFunction_demo_2(self): eMesh = PerceptMesh() eMesh.new_mesh(GMeshSpec("3x3x3|bbox:0,0,0,1,1,1") ) # use a fixture to generate a 3x3x3 hex mesh vectorDimension = 0 # add a field eMesh.add_field("coords_mag_field", FEMMetaData.NODE_RANK, vectorDimension) eMesh.commit() f_coords = eMesh.get_field("coordinates") # get pre-existing field coords_mag_field = eMesh.get_field( "coords_mag_field") # get the field we just created ff_coords = FieldFunction("ff_coords", f_coords, eMesh, 3, 3) # define a field function eval_vec3_print( 0.1, 0.1, 0.1, 0.0, ff_coords ) # evaluate and print the field function a point {0.1, 0.1, 0.1} time=0.0 coords_mag_sf = StringFunction( "sqrt(x*x + y*y + z*z)", "coords_mag_sf", 3, 1) # define coordinate magnitude function x = 0.123 y = 0.234 z = 0.345 vv = sqrt(x * x + y * y + z * z) v1 = eval_func(x, y, z, 0, coords_mag_sf) print "vv = ", vv, "== v1 = ", v1 self.assertEqual(vv, v1) # ensure correctness of string function # define a field function coords_mag_field_function = FieldFunction("coords_mag_field_function", coords_mag_field, eMesh, 3, 1) # interpolate the function onto the mesh coords_mag_field_function.interpolateFrom(coords_mag_sf) eMesh.save_as("./cubehex8_withCoordMag_out.e") # demonstrate how to usa an alias ff_coords.add_alias("mc") sfcm = StringFunction("sqrt(mc[0]*mc[0]+mc[1]*mc[1]+mc[2]*mc[2])", "sfcm", 3, 1)
def test_break_tri3_to_tri6_to_tri6_sierra(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size <= 3: eMesh = PerceptMesh(2) eMesh.open("quad_fixture_tri6_tri6_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) eMesh.add_field("proc_rank_ede", eMesh.edge_rank(), scalarDimension) breaker = Refiner(eMesh, TRI6_TRI6_4, proc_rank_field) eMesh.commit() eMesh.print_info("tri mesh tri6") eMesh.save_as("quad_fixture_tri6_tri6_0.e") breaker.doBreak() eMesh.print_info("tri mesh refined") eMesh.save_as("quad_fixture_tri6_tri6_1.e")
def test_hex8_hex8_8_1(self): fixture_setup() eMesh = PerceptMesh() p_size = eMesh.get_parallel_size() gmesh_spec = "4x4x"+str(4*p_size)+"|bbox:0,0,0,1,1,1" eMesh.new_mesh(GMeshSpec(gmesh_spec)) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX8_HEX8_8, proc_rank_field) eMesh.commit() breaker.doBreak
def test_break_tri_to_tri_sierra_1_test(self): pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size <= 2: n = 2 nx = n ny = n createEdgeSets = False fixture = QuadFixture_4(pm,nx,ny,createEdgeSets) isCommited = False eMesh = PerceptMesh(fixture.meta_data, fixture.bulk_data, isCommited) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) eMesh.add_field("proc_rank_edge", eMesh.edge_rank(), scalarDimension) breaker = Refiner(eMesh, TRI3_TRI3_4, proc_rank_field) eMesh.commit() fixture.generate_mesh() breaker.setRemoveOldElements(False) breaker.doBreak() eMesh.dump_elements_compact()
def test_wedge6_wedge18_enrich(self): pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size == 1: wedgeFixture = WedgeFixture() bulk = wedgeFixture.createMesh(MPI.COMM_WORLD, 4,3,2,0,1,0,1,0,1,"") eMesh = PerceptMesh(wedgeFixture.getMetaData(), bulk, False) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, WEDGE6_WEDGE18_1, proc_rank_field) eMesh.commit() wedgeFixture.createBulkAfterMetaCommit(MPI.COMM_WORLD) breaker.doBreak()
def test_hex8_tet4_6_12_2(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) eMesh = PerceptMesh(3) eMesh.open("hex_fixture.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX8_TET4_6_12, proc_rank_field) eMesh.commit() eMesh.print_info() breaker.doBreak() eMesh.save_as("hex_tet_6_12_1.e")
def test_hex20_hex20_1(self): fixture_setup() eMesh = PerceptMesh(3) p_size = eMesh.get_parallel_size() if p_size <= 3: eMesh.open("hex20_hex20_cube1x1x"+str(p_size) + "_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX20_HEX20_8, proc_rank_field) eMesh.commit() breaker.setRemoveOldElements(True) breaker.doBreak() eMesh.save_as("hex20_hex20_cube1x1x" + str(p_size) + "_1.e")
def test_fieldFunction_demo_2(self): eMesh = PerceptMesh() eMesh.new_mesh(GMeshSpec("3x3x3|bbox:0,0,0,1,1,1")) vectorDimension = 0 eMesh.add_field("coords_mag_field", FEMMetaData.NODE_RANK, vectorDimension) eMesh.commit() f_coords = eMesh.get_field(FEMMetaData.NODE_RANK, "coordinates") coords_mag_field = eMesh.get_field(FEMMetaData.NODE_RANK, "coords_mag_field") ff_coords = FieldFunction("ff_coords", f_coords, eMesh, 3, 3) eval_vec3_print(0.1,0.1,0.1,0.0,ff_coords) coords_mag_sf = StringFunction("sqrt(x*x + y*y + z*z)" , "coords_mag_sf", 3, 1) x = 0.123 y = 0.234 z = 0.345 vv = sqrt(x*x + y*y + z*z) v1 = eval_func(x,y,z,0,coords_mag_sf) print "vv = ", vv, "== v1 = ", v1 self.assertEqual(vv, v1) coords_mag_field_function = FieldFunction("coords_mag_field_function", coords_mag_field, eMesh, 3, 1) coords_mag_field_function.interpolateFrom(coords_mag_sf) eMesh.save_as("./cubehex8_withCoordMag_out.e") ff_coords.add_alias("mc") sfcm = StringFunction("sqrt(mc[0]*mc[0]+mc[1]*mc[1]+mc[2]*mc[2])", "sfcm", 3, 1) vv = eval_func(0.1,0.1,0.1,0.0, sfcm) print "expected = ", sqrt(3*0.1*0.1), " actual= " , vv sfcm = StringFunction("sqrt(ff_coords[0]*ff_coords[0]+ff_coords[1]*ff_coords[1]+ff_coords[2]*ff_coords[2])", "sfcm", 3, 1) vv = eval_func(0.1,0.1,0.1,0.0, sfcm) print "expected = ", sqrt(3*0.1*0.1), " actual= " , vv
def test_break_quad_to_quad_sierra_sidesets(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size == 1 or p_size == 2: eMesh = PerceptMesh(2) eMesh.open("quad_fixture.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, QUAD4_QUAD4_4_SIERRA, proc_rank_field) eMesh.commit() eMesh.print_info("after refinement break_quad_to_quad_sierra_sidesets") breaker.doBreak() eMesh.save_as("quad_sidesets_sierra_out.e")
def test_hex8_tet4_6_12_1(self): fixture_setup() eMesh = PerceptMesh(3) p_size = eMesh.get_parallel_size() gmesh_spec = "1x1x" + str(p_size) + "|bbox:0,0,0,1,1," + str(p_size) eMesh.new_mesh(GMeshSpec(gmesh_spec)) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX8_TET4_6_12, proc_rank_field) eMesh.commit() eMesh.print_info() breaker.setRemoveOldElements(True) breaker.doBreak() eMesh.save_as("hex_tet_6_12_cube1x1x1.e")
def test_quad8_to_quad8(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size <= 3: eMesh = PerceptMesh(2) eMesh.open("quad_fixture_quad8_quad8_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, QUAD8_QUAD8_4, proc_rank_field) eMesh.commit() breaker.setIgnoreSideSets(False) breaker.doBreak() eMesh.save_as("quad_fixture_quad8_quad8_1.e")
def test_hex20_hex20_1_2(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size == 1 or p_size == 3: eMesh = PerceptMesh(3) eMesh.open("hex20_hex20_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX20_HEX20_8, proc_rank_field) eMesh.commit() eMesh.save_as("hex20_hex20_0.e") breaker.doBreak() eMesh.save_as("hex20_hex20_1.e")
def test_wedge6_wedge18_enrich(self): pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size == 1: wedgeFixture = WedgeFixture() bulk = wedgeFixture.createMesh(MPI.COMM_WORLD, 4,3,2,0,1,0,1,0,1,"") # create stk::mesh::BulkData from wedge fixture eMesh = PerceptMesh(wedgeFixture.getMetaData(), bulk, False) # adopt bulk data scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, WEDGE6_WEDGE15_1, proc_rank_field) eMesh.commit() wedgeFixture.createBulkAfterMetaCommit(MPI.COMM_WORLD) # generate the mesh breaker.doBreak() # refine eMesh.save_as("./wedge6-15.e") # save
def test_wedge6_2(self): fixture_setup() eMesh = PerceptMesh(3) p_size = eMesh.get_parallel_size() if p_size == 1: wedgeFixture = WedgeFixture() wedgeFixture.createMesh(MPI.COMM_WORLD, 4,3,2,0,1,0,1,0,1, "swept_wedge_0.e") eMesh.open("swept_wedge_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, WEDGE6_WEDGE6_8, proc_rank_field) eMesh.commit() breaker.doBreak() eMesh.save_as("swept-wedge_1.e")
def test_break_tri_to_tri_sierra_1(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size <= 3: n = 12 nx = n ny = n createEdgeSets = True fixture = QuadFixture_3(pm, nx, ny, createEdgeSets) isCommited = False eMesh = PerceptMesh(fixture.meta_data, fixture.bulk_data, isCommited) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", stk::topology::ELEMENT_RANK, scalarDimension) eMesh.add_field("proc_rank_ede", eMesh.edge_rank(), scalarDimension) breaker = Refiner(eMesh, TRI3_TRI3_4, proc_rank_field) eMesh.commit() fixture.generate_mesh() eMesh.print_info("tri mesh") eMesh.save_as("quad_fixture_tri3_0.e") breaker.doBreak() eMesh.print_info("tri mesh refined") eMesh.save_as("quad_fixture_tri3_1.e")
def test_break_quad_to_quad(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size == 1 or p_size == 3: eMesh = PerceptMesh(2) eMesh.open("quad_fixture_no_sidesets.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, QUAD4_QUAD4_4, proc_rank_field) eMesh.commit() eMesh.print_info("quad mesh") breaker.setIgnoreSideSets(True) breaker.doBreak() eMesh.save_as("square_quad4_ref_out.e")
def test_fieldFunction_demo_2(self): eMesh = PerceptMesh() eMesh.new_mesh(GMeshSpec("3x3x3|bbox:0,0,0,1,1,1")) # use a fixture to generate a 3x3x3 hex mesh vectorDimension = 0 # add a field eMesh.add_field("coords_mag_field", FEMMetaData.NODE_RANK, vectorDimension) eMesh.commit() f_coords = eMesh.get_field("coordinates") # get pre-existing field coords_mag_field = eMesh.get_field("coords_mag_field") # get the field we just created ff_coords = FieldFunction("ff_coords", f_coords, eMesh, 3, 3) # define a field function eval_vec3_print(0.1,0.1,0.1,0.0,ff_coords) # evaluate and print the field function a point {0.1, 0.1, 0.1} time=0.0 coords_mag_sf = StringFunction("sqrt(x*x + y*y + z*z)" , "coords_mag_sf", 3, 1) # define coordinate magnitude function x = 0.123 y = 0.234 z = 0.345 vv = sqrt(x*x + y*y + z*z) v1 = eval_func(x,y,z,0,coords_mag_sf) print "vv = ", vv, "== v1 = ", v1 self.assertEqual(vv, v1) # ensure correctness of string function # define a field function coords_mag_field_function = FieldFunction("coords_mag_field_function", coords_mag_field, eMesh, 3, 1) # interpolate the function onto the mesh coords_mag_field_function.interpolateFrom(coords_mag_sf) eMesh.save_as("./cubehex8_withCoordMag_out.e") # demonstrate how to usa an alias ff_coords.add_alias("mc") sfcm = StringFunction("sqrt(mc[0]*mc[0]+mc[1]*mc[1]+mc[2]*mc[2])", "sfcm", 3, 1)
def test_wedge6_enrich_refine(self): fixture_setup() p_size = parallel_machine_size(MPI.COMM_WORLD) if p_size == 1: eMesh = PerceptMesh(3) wedgeFixture = WedgeFixture() wedgeFixture.createMesh(MPI.COMM_WORLD, 4,2,2,0,1,0,1,0,1, "tmp-swept-wedge_enrich_0.e") eMesh.open("tmp-swept-wedge_enrich_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, WEDGE6_WEDGE15_1, proc_rank_field) eMesh.commit() breaker.doBreak() eMesh.save_as("swept-wedge_2_enrich_refine_0.e") if p_size == 1: eMesh = PerceptMesh(3) eMesh.open("swept-wedge_2_enrich_refine_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, WEDGE15_WEDGE15_8, proc_rank_field) eMesh.commit() breaker.setIgnoreSideSets(True) breaker.doBreak() eMesh.save_as("swept-wedge_2_enrich_refine_1.e")
def fixture_setup_0(): eMesh = PerceptMesh() p_size = eMesh.get_parallel_size() gmesh_spec = "4x4x"+str(4*p_size)+"|bbox:0,0,0,1,1,1" eMesh.new_mesh(GMeshSpec(gmesh_spec)) eMesh.commit() eMesh.save_as("hex_fixture.e") eMesh = PerceptMesh() eMesh.open("exodus_files/"+input_files_loc+"hex_fixture.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, HEX8_TET4_24, proc_rank_field) eMesh.commit() breaker.doBreak() eMesh.save_as("tet_fixture.e")
def test_beam_enrich(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size <= 1: eMesh = PerceptMesh() eMesh.open("exodus_files/beam.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) eMesh.save_as("beam_enrich_0.e") breaker = Refiner(eMesh, BEAM2_BEAM3_1, proc_rank_field) eMesh.commit() breaker.setIgnoreSideSets(True) breaker.doBreak() eMesh.save_as("beam_enrich.e")
def test_break_quad_to_quad_sierra(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size <= 3: n = 12 nx = n ny = n fixture = QuadFixture_4(pm, nx, ny, True) eMesh = PerceptMesh(fixture.meta_data, fixture.bulk_data, False) scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, QUAD4_QUAD4_4_SIERRA, proc_rank_field) eMesh.commit() fixture.generate_mesh() eMesh.print_info("quad mesh") breaker.doBreak()
def test_heterogeneous_mesh(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(MPI.COMM_WORLD) if p_size <= 1: mesh = HeterogeneousFixture(MPI.COMM_WORLD, False) #put_io_part_attribute(mesh.m_block_hex) #put_io_part_attribute(mesh.m_block_wedge) #put_io_part_attribute(mesh.m_block_tet) mesh.m_metaData.commit() mesh.populate() isCommited = True em1 = PerceptMesh(mesh.m_metaData, mesh.m_bulkData, isCommited) em1.save_as("heterogeneous_0.e") em1.close() eMesh = PerceptMesh(3) eMesh.open("heterogeneous_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension)
def test_beam_refine(self): fixture_setup() pm = MPI.COMM_WORLD p_size = parallel_machine_size(pm) if p_size <= 1: mesh = BeamFixture(pm, False) mesh.m_metaData.commit() mesh.populate() isCommited = True em1 = PerceptMesh(mesh.m_metaData, mesh.m_bulkData, isCommited) em1.save_as("beam_0.e") eMesh = PerceptMesh() eMesh.open("beam_0.e") scalarDimension = 0 proc_rank_field = eMesh.add_field("proc_rank", eMesh.element_rank(), scalarDimension) breaker = Refiner(eMesh, BEAM2_BEAM2_2, proc_rank_field) eMesh.commit() breaker.setIgnoreSideSets(True) breaker.doBreak()
def test_high_level_interface(self): self.fixture_setup() p_size = parallel_machine_size(self.pm) if p_size <= 2: eMesh = PerceptMesh(2) eMesh.open("./exodus_files/quad_fixture.e") vectorDimension = 0 eMesh.add_field("coords_mag_field", FEMMetaData.NODE_RANK, vectorDimension) eMesh.commit() f_coords = eMesh.get_field("coordinates") coords_mag_field = eMesh.get_field("coords_mag_field") ff_coords = FieldFunction("ff_coords", f_coords, eMesh, 2, 2) #eval_vec3_print(0.1,0.1,0.1,0.0,ff_coords) coords_mag_sf = StringFunction("sqrt(x*x + y*y )" , "coords_mag_sf", 2, 1) x = 0.123 y = 0.234 vv = sqrt(x*x + y*y ) v1 = eval_func2(x,y,0,coords_mag_sf) print "vv = ", vv, "== v1 = ", v1 self.assertEqual(vv, v1) coords_mag_field_function = FieldFunction("coords_mag_field_function", coords_mag_field, eMesh, 2, 1) coords_mag_field_function.interpolateFrom(coords_mag_sf) eMesh.save_as("./exodus_files/quad_fixture_with_coords_mag.e") ff_coords.add_alias("mc") sfcm = StringFunction("sqrt(mc[0]*mc[0]+mc[1]*mc[1]+mc[2]*mc[2])", "sfcm", 3, 1) add_newlines = True eMesh.print_info("quad fixture", 2, add_newlines) self.assertTrue(eMesh.get_spatial_dim() == 2) self.assertTrue(eMesh.get_number_elements() == 12*12) self.assertTrue(eMesh.get_number_nodes() == 13*13) self.assertTrue(eMesh.get_parallel_size() == p_size) self.assertTrue(eMesh.get_bulk_data() != 0) self.assertTrue(eMesh.get_fem_meta_data() != 0) # // entity data setter/getters node = eMesh.get_node(1) self.assertTrue(node != 0) cm1 = eMesh.get_field_data(coords_mag_field, node) co1 = [0,0] co1[0] = eMesh.get_field_data(f_coords, node, 0) co1[1] = eMesh.get_field_data(f_coords, node, 1) print "cm1= ", cm1, " co1= ", co1 eMesh.set_field_data(123.0, f_coords, node, 0) co1[0] = eMesh.get_field_data(f_coords, node, 0) print " co1= ", co1 element = eMesh.get_element(1) self.assertTrue(element != 0) element1 = eMesh.get_entity(eMesh.element_rank(), 1) self.assertTrue(element == element1) #/// find node closest to given point node = eMesh.get_node(0,0) self.assertTrue(node != 0) #/// find element that contains given point element = eMesh.get_element(0.01, 0.01) self.assertTrue(element != 0)