def __init__(self): os.chdir(input_dir) with open("input_mesh_generator.yaml", 'r') as stream: data_loaded_mesh = yaml.load(stream) # data_loaded = yaml.load(stream, Loader=yaml.FullLoader) # data_loaded = yaml.full_load(stream) with open("input_wells.yaml", 'r') as stream: data_loaded_wells = yaml.load(stream) # data_loaded = yaml.load(stream, Loader=yaml.FullLoader) # data_loaded = yaml.full_load(stream) self.data_loaded = data_loaded_mesh file_name = data_loaded_mesh['file_name'] self.file_name = file_name # self.Lz = float(data_loaded_mesh['mesh_size'][2]) name_mesh = file_name + '_wells.h5m' self.mb = core.Core() self.mtu = topo_util.MeshTopoUtil(self.mb) mesh_file = name_mesh os.chdir(flying_dir) self.mb.load_file(mesh_file) names_tags = np.load('tags_wells.npy') self.tags = utpy.get_all_tags_2(self.mb, names_tags) self.all_nodes, self.all_edges, self.all_faces, self.all_volumes = utpy.get_all_entities( self.mb) import pdb pdb.set_trace()
def __init__(self, mesh_file): self.mb = core.Core() self.mtu = topo_util.MeshTopoUtil(self.mb) self.mb.load_file(mesh_file) self.all_nodes, self.all_edges, self.all_faces, self.all_volumes = utpy.get_all_entities( self.mb) self.tags = LoadMesh.load_tags(self.mb) self.matrices = LoadMesh.load_matrices() self.ADJs = np.array( [self.mb.get_adjacencies(face, 3) for face in self.all_faces]) self.all_centroids = np.array( [self.mtu.get_average_position([v]) for v in self.all_volumes]) self.wirebasket_elems, self.wirebasket_numbers, self.mv = LoadMesh.load_wirebasket_elems( self.mb, self.tags) self.meshset_levels, self.L2_meshset = LoadMesh.load_meshsets_level( self.mb, self.tags) self.all_boundary_faces = LoadMesh.load_all_bound_faces( self.mb, self.tags)
def carregar_dados_anterior(data_loaded, loop): from utils import pymoab_utils as utpy parent_dir = os.path.dirname(os.path.abspath(__file__)) parent_parent_dir = os.path.dirname(parent_dir) input_dir = os.path.join(parent_parent_dir, 'input') flying_dir = os.path.join(parent_parent_dir, 'flying') bifasico_dir = os.path.join(flying_dir, 'bifasico') bifasico_sol_direta_dir = os.path.join(bifasico_dir, 'sol_direta') bifasico_sol_multiescala_dir = os.path.join(bifasico_dir, 'sol_multiescala') utils_dir = os.path.join(parent_parent_dir, 'utils') ADM = data_loaded['ADM'] input_file = data_loaded['input_file'] if ADM: caminho = bifasico_sol_multiescala_dir ext_h5m = input_file + 'sol_multiescala_' + str(loop) + '.h5m' else: caminho = bifasico_sol_direta_dir ext_h5m = input_file + 'sol_direta_' + str(loop) + '.h5m' os.chdir(caminho) mb = core.Core() mtu = topo_util.MeshTopoUtil(mb) mb.load_file(ext_h5m) name_historico = 'historico_' + str(loop) + '.npy' historico = np.load(name_historico) t_loop = historico[1] vpi = historico[0] os.chdir(flying_dir) list_names_tags = np.load('list_names_tags.npy') tags_1 = utpy.get_all_tags_2(mb, list_names_tags) tempos_impr = data_loaded['tempos_vpi_impressao'] contar_loop = data_loaded['contar_loop'] contar_tempo = data_loaded['contar_tempo'] imprimir_sempre = data_loaded['imprimir_sempre'] all_nodes, all_edges, all_faces, all_volumes = utpy.get_all_entities(mb) return mb, mtu, tags_1, input_file, ADM, tempos_impr, contar_loop, contar_tempo, imprimir_sempre, all_nodes, all_edges, all_faces, all_volumes, t_loop, vpi
# data_loaded = yaml.full_load(stream) input_file = data_loaded['input_file'] ext_h5m = input_file + '_dual_primal.h5m' os.chdir(flying_dir) mesh_file = ext_h5m mb = core.Core() root_set = mb.get_root_set() mtu = topo_util.MeshTopoUtil(mb) mb.load_file(mesh_file) intern_adjs_by_dual = np.load('intern_adjs_by_dual.npy') faces_adjs_by_dual = np.load('faces_adjs_by_dual.npy') all_nodes, all_edges, all_faces, all_volumes = utpy.get_all_entities(mb) cent_tag = mb.tag_get_handle('CENT') all_centroids = mb.tag_get_data(cent_tag, all_volumes) press_value_tag = mb.tag_get_handle('P') vazao_value_tag = mb.tag_get_handle('Q') wells_injector_tag = mb.tag_get_handle('WELLS_INJECTOR', 1, types.MB_TYPE_HANDLE, types.MB_TAG_MESH, True) wells_producer_tag = mb.tag_get_handle('WELLS_PRODUCER', 1, types.MB_TYPE_HANDLE, types.MB_TAG_MESH, True) tags_criadas_aqui += ['WELLS_INJECTOR', 'WELLS_PRODUCER'] wells_injector = mb.create_meshset() wells_producer = mb.create_meshset() mb.tag_set_data(wells_injector_tag, 0, wells_injector) mb.tag_set_data(wells_producer_tag, 0, wells_producer) bifasico = data_loaded['bifasico'] r0 = data_loaded['rs']['r0']