示例#1
0
np.seterr(divide='ignore')

if __name__ == '__main__':

    logger = logging.getLogger('LOG')
    logger.setLevel(logging.DEBUG)
    fh = logging.FileHandler('LOGS_SolAnVsNum.log')
    fh.setLevel(logging.DEBUG)
    logger.addHandler(fh)
    formatter = logging.Formatter(
        '%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    fh.setFormatter(formatter)
    logger.addHandler(fh)

    pr = gp.get_parameters()

    ##################################################################
    # Load txt file of temperature
    ##################################################################
    logger.debug("Loading txt file of temperature")
    try:
        matriz = np.loadtxt('Results/T_.txt')
    except Exception as e:
        logger.critical("No se pudo cargar el archico: ")
        sys.exit()
    size = pr["lattice_x"]  # size
    ydata = np.zeros([size])
    for i in range(size):
        ydata[i] = matriz[int(pr["lattice_y"] // 2)][i]
    xdata = np.linspace(0, size - 1, size)
示例#2
0
#! /usr/bin/env python
# -*- coding: utf-8 -*-
# Authors: Benjamín S. Noyola García, Suemi Rodriguez Romo

import local as loc
from numba import cuda, float64, float32
import numpy as np
import get_parser_parameters as gp
pr = gp.get_parameters()  # pr is a dictionary of parameters
T_h = pr["T_h"]
T_c = pr["T_c"]

# links to  local.py program
###################### velocity field #######################################
getf = cuda.jit('void (f8[:,:,:], f8[::1], i8, i8)', device=True)(loc.getf)
getux = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getux)
getuy = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getuy)
getFx = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getFx)
getFy = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getFy)
getden = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getden)
getf_l = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getf_l)
getf_2l = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getf_2l)
getT = cuda.jit('void (f8[:,:], f8[::1], i8, i8)', device=True)(loc.getT)
f2m = cuda.jit('void (f8[::1], f8[::1])', device=True)(loc.f2m)
calc_fl_PCMloc = cuda.jit('void (f8[::1], f8[::1])',
                          device=True)(loc.calc_fl_PCMloc)
calc_noruloc = cuda.jit('void (f8[::1], f8[::1], f8[::1])',
                        device=True)(loc.calc_noruloc)
calc_m_eqloc = cuda.jit(
    'void (f8[::1], f8[::1], f8[::1], f8[::1], f8[::1], f8[::1], f8[::1])',
    device=True)(loc.calc_m_eqloc)