コード例 #1
0
    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()
コード例 #2
0
 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)
コード例 #3
0
ファイル: def_intermediarios.py プロジェクト: jpra2/ADM_mod_2
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
コード例 #4
0
ファイル: mad_mesh.py プロジェクト: jpra2/MAD_modularizado
        # 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']