def coords_map(elements, node_set1): lsize = nums[2] * map_dofs_coords ind_coords = compute_ind_extr(nums, map_dofs_coords, nelems, layers, mesh2d, dofs_coords, A, wedges, elems2nodes, lsize) return op2.Map(elements, node_set1, map_dofs_coords, ind_coords, "elem_dofs", off1)
def field_map(elements, elem_set1): lsize = nums[2] * map_dofs_field ind_field = compute_ind_extr(nums, map_dofs_field, nelems, layers, mesh2d, dofs_field, A, wedges, elems2elems, lsize) return op2.Map(elements, elem_set1, map_dofs_field, ind_field, "elem_elem", off2)
def field_map(elements, elem_set1): lsize = nums[2] * map_dofs_field ind_field = compute_ind_extr( nums, map_dofs_field, nelems, layers, mesh2d, dofs_field, A, wedges, elems2elems, lsize) return op2.Map(elements, elem_set1, map_dofs_field, ind_field, "elem_elem", off2)
def coords_map(elements, node_set1): lsize = nums[2] * map_dofs_coords ind_coords = compute_ind_extr( nums, map_dofs_coords, nelems, layers, mesh2d, dofs_coords, A, wedges, elems2nodes, lsize) return op2.Map(elements, node_set1, map_dofs_coords, ind_coords, "elem_dofs", off1)
# DECLARE OP2 STRUCTURES coords_dofsSet = op2.Set(nums[0] * layers, "coords_dofsSet") coords = op2.Dat(coords_dofsSet ** 2, coords_dat, np.float64, "coords") wedges_dofsSet = op2.Set(nums[2] * wedges, "wedges_dofsSet") field = op2.Dat(wedges_dofsSet, field_dat, np.float64, "field") p1_dofsSet = op2.Set(nums[0] * layers, "p1_dofsSet") res = op2.Dat(p1_dofsSet, res_dat, np.float64, "res") # THE MAP from the ind # create the map from element to dofs for each element in the 2D mesh lsize = nums[2] * map_dofs_coords ind_coords = compute_ind_extr(nums, map_dofs_coords, lins, layers, mesh2d, dofs_coords, A, wedges, mapp_coords, lsize) lsize = nums[2] * map_dofs_field ind_field = compute_ind_extr(nums, map_dofs_field, lins, layers, mesh2d, dofs_field, A, wedges, mapp_field, lsize) lsize = nums[2] * map_dofs_res ind_res = compute_ind_extr(nums, map_dofs_res, lins, layers, mesh2d, dofs_res, A, wedges, mapp_res, lsize) elem_dofs = op2.Map(elements, coords_dofsSet, map_dofs_coords, ind_coords, "elem_dofs", off_coords) elem_elem = op2.Map(elements, wedges_dofsSet, map_dofs_field, ind_field, "elem_elem", off_field) elem_p1_dofs = op2.Map(elements, p1_dofsSet, map_dofs_res, ind_res, "elem_p1_dofs", off_res)
# DECLARE OP2 STRUCTURES coords_dofsSet = op2.Set(nums[0] * layers, "coords_dofsSet") coords = op2.Dat(coords_dofsSet**2, coords_dat, np.float64, "coords") wedges_dofsSet = op2.Set(nums[2] * wedges, "wedges_dofsSet") field = op2.Dat(wedges_dofsSet, field_dat, np.float64, "field") p1_dofsSet = op2.Set(nums[0] * layers, "p1_dofsSet") res = op2.Dat(p1_dofsSet, res_dat, np.float64, "res") # THE MAP from the ind # create the map from element to dofs for each element in the 2D mesh lsize = nums[2] * map_dofs_coords ind_coords = compute_ind_extr(nums, map_dofs_coords, lins, layers, mesh2d, dofs_coords, A, wedges, mapp_coords, lsize) lsize = nums[2] * map_dofs_field ind_field = compute_ind_extr(nums, map_dofs_field, lins, layers, mesh2d, dofs_field, A, wedges, mapp_field, lsize) lsize = nums[2] * map_dofs_res ind_res = compute_ind_extr(nums, map_dofs_res, lins, layers, mesh2d, dofs_res, A, wedges, mapp_res, lsize) elem_dofs = op2.Map(elements, coords_dofsSet, map_dofs_coords, ind_coords, "elem_dofs", off_coords) elem_elem = op2.Map(elements, wedges_dofsSet, map_dofs_field, ind_field, "elem_elem", off_field) elem_p1_dofs = op2.Map(elements, p1_dofsSet, map_dofs_res, ind_res, "elem_p1_dofs", off_res)