Turbulence-Resolving Integral Simulations (TRIS) applies a moment-of-momentum integral approach, derived from Navier-Stokes, to run the time evolution of the integrated velocity and pressure fields. The Python code here provides an animation of a chosen field depending on the user input.
unlisted
ubuntu2204o
�m6g . � @ s� d dl Zd dlmZ d dlmZ d dlZd dlZdZ ej
Z
ej Z ej Z G dd� d�Z
G dd� d�Zdd � Zd
d � Zd d
� Zdd� Zdd� Zdd� Zdd� Zdd� Zdd� ZdS )� N)�cmg�����|�=c @ s6 e Zd ZdZdej dej ddfdd�Zdd� ZdS ) �Mshz9Base grid class containing physical and wavenumber grids.� �@ c C s || _ || _|| _|| _|| | _|| | _t�d|| j |�| _t�d|| j |�| _ tj
| j| j dd�\| _ | _ tj
| j | _tj
| j | _dtj
| | _dtj
| | _|| _|d d | _d| j tj�|� | _t�d| j| j�| _tj
| j| jdd�\| _| _d S )Nr �ij)�indexingr � )�Lx�Lz�nx�nz�dx�fz�np�linspace�x�z�meshgrid�X�Z�pi�kx_max�kz_max�dkx�dkz�nkx�nkz�fft�fftfreq�kx�kz�KX�KZ)�selfr r
r r � r$ �3/home/user/S002211202200903X-Figure-1/periodic2d.py�__init__ s$
z Msh.__init__c C s t �| j| jf�S �N)r �onesr r �r# r$ r$ r% �one& s zMsh.oneN)�__name__�
__module__� __qualname__�__doc__r r r& r* r$ r$ r$ r% r s r c @ s" e Zd ZdZdKdd�ZdLd d
�Zd d � Zd
d� Zdd� ZdMdd�Z dd� Z
dd� Z dd� Z dd� Z
d d � Zd d � Zd d!� Zd"d#� Zd$d%� Zd&d'� Zd(d)� Zd*d+� Zd,d-� Zd.d/� Zd0d1� Zd2d3� Zd4d5� Zd6d7� Zd8d9� Z d:d;� Z d<d=� Z d>d?� Z dNdAdB�Z dCdD� Z!dEdF� Z"dGdH� Z#dOdIdJ�Z$dS )P�Varz Base solution variable class.N�var�None�o�kc C s� || _ || _|| _|| _|| _t�|�|j|jfkr!| � |� d S t�|�|j
|j fkr3| � |� d S | � t�
|j|jf�� | � tj
|j
|j ftd�� d| _d| _d S )N��dtypeF)�msh�name�ls�mrk�clrr �shaper r �setvalr r �sethat�zeros�complex�hasval�hashat)r# r6 �val�hatr7 r8 r9 r: r$ r$ r% r& , s
z Var.__init__c C s~ t | j| j| j| j| j�}| jrt�| j �|_ d|_| j
r)tj| j t d�|_ d|_
|r.||_|r3||_|r8||_|r=||_|S )NTr4 )
r/ r6 r7 r8 r9 r: r@ r �arrayrB rA rC r? )r# r7 r8 r9 r: �newr$ r$ r% �copy= s zVar.copyc C s t �|�| _d| _d| _d S �NTF)r rD rB r@ rA )r# � val_arrayr$ r$ r% r< O s
z
Var.setvalc C s t j|td�| _d| _d| _d S )Nr4 TF)r rD r? rC rA r@ )r# � hat_arrayr$ r$ r% r= T s
z
Var.sethatc C s: t �|t �| jj| jjf� �| _d| _d| _| � � d S rG )
r rD r( r6 r r rB r@ rA r )r# �constr$ r$ r% � set_constantY s " zVar.set_constant� �?c C sF |t j�| jj| jj � }| �t �|| jj| jjf�� | �� d S r' ) r �random�randnr6 r r r< �reshaper )r# �sigma�randnsr$ r$ r% rN _ s z Var.randnc C sB | j s | jrtj�| j�| jj| jj | _ d| _ d S t
d��d S )NTz+Called fft for variable with hasval = False) rA r@ r r �rfftnrB r6 r r rC � Exceptionr) r$ r$ r% r d �
�zVar.fftc C sB | j s | jrtj�| j�| jj| jj | _ d| _ d S t
d��d S )NTz,Called ifft for variable with hashat = False) r@ rA r r �irfftnrC r6 r r rB rS r) r$ r$ r% �ifftm rT zVar.ifftc C s< | j r
t| j|| j d�S | jrt| j|| j d�S td��)N�rC �rB z+Called smult with neither hashat nor hasval)rA r/ r6 rC r@ rB rS )r# �scalarr$ r$ r% �smultv s
z Var.smultc C s� | j r|j rt| j| j|j d�S | jr3|jr3t| j�}| �� |�� |�| j|j � |�� |S td| j� td| j � td|j� td|j � t d��)NrX �self.hashat = �self.hasval = �other.hashat = �other.hasval = z*mult called with neither hashat nor hasval)
r@ r/ r6 rB rA rV r< r �printrS )r# �otherrE r$ r$ r% �mult~ s
zVar.multc C s` | j rt| jt�t�| j��d�S | jr,| �� t| jt�t�| j��d�}|� � |S t
d��)NrX z*sqrt called with neither hashat nor hasval) r@ r/ r6 r �sqrt�absrB rA rV r rS )r# rE r$ r$ r% rb � s zVar.sqrtc C sx | j r t| j|t�| jj| jjf� d�}|�� | �|�S | j r8t| j|t�| jj| jjf� d�}| �|�S t
d��)NrX z*sadd called with neither hashat nor hasval) rA r/ r6 r r( r r r �addr@ rS )r# rY rE r$ r$ r% �sadd� s $
$
zVar.saddc C s| | j r|j rt| j| j|j d�S | jr"|jr"t| j| j|j d�S td| j � td| j� td|j � td|j� td��)NrW rX r[ r\ r] r^ z)add called with neither hashat nor hasval)rA r/ r6 rC r@ rB r_ rS �r# r` r$ r$ r% rd � s zVar.addc C �\ | j s t�d� | �� t| j| jd | j| j| j �}d| jj
| j |_ d|_ |� � |S )Nz Had to call fft within diff_x�_dx� �?T)
rA �warnings�warnr r/ r6 r7 r8 r9 r: r! rC � filter_kxmax�r# �derivr$ r$ r% �diff_x� �
z
Var.diff_xc C rg )Nz Had to call fft within diff_z�_fzri T)
rA rj rk r r/ r6 r7 r8 r9 r: r" rC � filter_kzmaxrm r$ r$ r% �diff_z� rp z
Var.diff_zc C sj | j s t�d� | �� t| j| jd | j| j| j �}| jj
d | jj d | j |_ d|_ |�
� |S )NzHad to call fft within lapl�_laplr T)rA rj rk r r/ r6 r7 r8 r9 r: r! r" rC �
filter_max)r# �laplr$ r$ r% � laplacian� s
"z
Var.laplacianc C s< | j jd | j jd }| jt�|d | | � | _d S �Nr )r6 r! r" rC r �exp)r# �nu�dt�k2r$ r$ r% � viscous_damp� s $zVar.viscous_dampc C s� | j rZ| jrZtj�| j�| jj| jj }tj �
| j | �t kr/t
dttj �
| j | �� ��tj�| j �| jj| jj }tj �
| j| �t krXt
dttj �
| j| �� ��d S t
dt| j � d t| j� ��)Nz Check zero: zCheck failed: hashat = z ; hasval = )rA r@ r r rU rC r6 r r �linalg�normrB � check_tolrS �strrR )r# rB rC r$ r$ r% �check� s ���z Var.checkc C sl | j j|j jkrR| j j|j jkrR|jd | j jd �d | j j�f | jd | j jd �d d �f< |j|j j| j jd d �d | j j�f | j| j jd d �d d �f< d S | j j|j jkr�| j j|j jkr�tj| j j| j jftd�| _|jd |j jd �d d �f | jd |j jd �d |j j�f< |j|j jd d �d d �f | j| j j|j jd d �d |j j�f< d S td��)Nr r4 z Not implemented yet: cp_hat())r6 r r rC r r> r? rS rf r$ r$ r% �cp_hat� s �( � � ,�z
Var.cp_hatc C sF d}d| j t�| jj�|| k< d| j t�| jj�|| k< d| _d S )N��&� .>� F)rC r rc r6 r! r" r@ )r# �kx_cut�kz_cut�k_tolr$ r$ r% �filter� s
z
Var.filterc C s | � | jj| jj� d S r' �r� r6 r r r) r$ r$ r% ru � s zVar.filter_maxc C s | � | jjd| jj � d S rx r� r) r$ r$ r% rl � � zVar.filter_kxmaxc C s | � d| jj | jj� d S rx r� r) r$ r$ r% rr � r� zVar.filter_kzmaxc C s( | � d| jj d d| jj d � d S )Nr � r� r) r$ r$ r% � filter_third� s (zVar.filter_thirdc C s | � | jjd | jjd � d S rx r� r) r$ r$ r% � filter_half � zVar.filter_halfc C s | � | jjd | jjd � d S )Nr )r� r6 r r r) r$ r$ r% �
filter_all r� zVar.filter_allc C s, | j r t�| jd �S | jrt�| j�S d S )N�r r )rA r �realrC r@ �meanrB r) r$ r$ r% r� s
�zVar.meanc C s6 | j r| �� }d|d< t�|�S | jrt�| j�S d S )N� r� )rA �spectrumr �sumr@ r0 rB �r# r� r$ r$ r% r0 s
�zVar.var�d c C sL | j r
| js
| �� tj| j|dd�\}}|d d� t�|�d }||fS )NT)�bins�density�����r )rA r@ rV r � histogramrB �diff)r# �nbins�pdf�edgesr� r$ r$ r% r� s
zVar.pdfc C s | j s| �� t�| j�d S rx )rA r r rc rC r) r$ r$ r% r� s z Var.spectrumc C � | � � }tj|dd�S )Nr ��axis�r� r r� r� r$ r$ r% �
spectrum_x# � zVar.spectrum_xc C r� )Nr r� r� r� r$ r$ r% �
spectrum_z' r� zVar.spectrum_zc C sL | j r
| js
| �� t�| j�t�| j�t�| j�f}td| d |� d S )Nzmin/mean/max of z is:) rA r@ rV r �minrB r� �maxr_ )r# r7 �reportr$ r$ r% r� + s "z
Var.report)NNr0 r1 r2 r3 )NNNN)rL )r� )r0 )%r+ r, r- r. r& rF r< r= rK rN r rV rZ ra rb re rd ro rs rw r} r� r� r� ru rl rr r� r� r� r� r0 r� r� r� r� r� r$ r$ r$ r% r/ ) sF
r/ c C sp | j s t�d� | �� |j st�d� |�� t| j�}|�d| jj |j d| jj | j � |�
� |S )Nz"Had to call fft within vorticity()ri ) rA rj rk r r/ r6 r= r! rC r" ru )�u�w�vortr$ r$ r% � vorticity1 s
*r� c C sn | j s| �� || }| jj}| jj}|| || d }t| j�}|�| jt� d| | � � |�
� |S )N�>��N}a+� rA r r6 r! r" r/ r= rC r ry ru ��f�ell�ell2r r r| �gr$ r$ r% � filter_Gauss= s
r� c C sr | j s| �� || }| jj}| jj}|| || d }t| j�}|�| jdt� d| | � � |�
� |S )Nr� r r� r� r� r$ r$ r% �filter_Gauss_highpassI s
"r� c C sB | j s| �� d}t| j| jd�}d|jt�|jj�|| k< |S )Nr� rW r� )rA r r/ r6 rC r rc r" )r� r� r� r� r� r$ r$ r% �filter_highpassU s r� c C s� | j s| �� |j s|�� | j}|j}|j}|| || d }|| | }|| | }|| | }d||dk< d||dk< d||dk< t|�} t|�}
| �d| | j ||j � |
�d| |j || j � | �� |
�� | |
fS )Nr� r� rL ) rA r r6 r! r" r/ r= rC ru ) �f_x�f_zr6 r r r| �Pxx�Pxz�Pzz�r_x�r_zr$ r$ r% �
projection] s* r� c C sd | j r |j r t| |�S | jr|jrt| |�S td| j � td| j� td|j � td|j� td��)Nz
f_x.hashat = z
f_x.hasval = z
f_z.hashat = z
f_z.hasval = z)Called div with neither hashat nor hasval)rA �div_hatr@ �div_valr_ rS �r� r� r$ r$ r% �divu s
r� c C s. | j r|j r| �� |�� t| |�S td��)Nz Called div_val with hasval=False)r@ r r� rS r� r$ r$ r% r� � s
r� c C sV | j r'|j r'| j}|j}|j}t|d| | j d| |j d�}|�� |S td��)Nri rW z Called div_hat with hashat=False)rA r6 r! r" r/ rC ru rS )r� r� r6 r r r� r$ r$ r% r� � s $r� c
C s� t | j�}t |j�}| �� |�� t�| jd |jd �d }| j| }|j| }|| | } |�| | � |�| | � |�� |�� |�� |�� ||fS )Nr r� ) r/ r6 rV r rb rB r< r r� )
r� r� �u_avg�pwr�fxr �u_mag�e_x�e_z�frictionr$ r$ r% r� � s
r� )�numpyr �matplotlib.pyplot�pyplot�plt�
matplotlibr rj �
subprocessr� r �cos�sinr r/ r� r� r� r� r� r� r� r� r� r$ r$ r$ r% �<module> s,