initial_conditions.py

import numpy as np

def initial_conditions(mesh_pos, geometry, paramMVG):
    """Calcule les conditions initiales du modèle hydro
    p est un tableau de coordonnees x,y, type, charge
    où le type decrit le type de noeud, charge est la charge imposée initiale
    """

    h0 = paramMVG.h0()
    
    p = np.zeros((len(mesh_pos), 4), dtype = 'float') #matrice vide
    p[:,:2] = np.array(mesh_pos[:,:2]) #coordonnées xy des noeuds du maillage

    for i in range(0,len(p)) :
        # charge au fond du trou
        if p[i,1]==geometry.etrou and p[i,0]<=geometry.r :
            p[i,2]=1 # type de neoud qui ne peut pas évoluer
            p[i,3]=geometry.h_eau
        # charge au fond du modèle    
        elif p[i,1]==0 :
            p[i,2]=0 #type free drainage?
            p[i,3]=h0
        # partout ailleurs    
        else :
            p[i,3]=h0
                
    #Interpolation lineaire pour la valeur de charge au bord du trou sous l'eau
    #On considère que la charge est nulle à la surface de l'eau
    rows = np.array(np.where((p[:,1] <= geometry.etrou + geometry.h_eau) & (p[:,1] >= geometry.etrou) & (p[:,0]==geometry.r))).T
    for i in range(1,len(rows)):
        p[rows[i,0],2] = 1
        p[rows[i,0],3] = geometry.h_eau-(p[rows[i,0],1]-geometry.etrou)
     
    return p