unlisted
ubuntu2404a
=bi� � @ sn d Z ddlZddlmZ ddlmZ ddlmZ ddd�Z d d
� Z
d d � Z d
d� Z dd� Z
dd� Zdd� ZdS )z6
Created on Fri Nov 7 19:15:57 2025
@author: davide
� N)� interpolate)�odeint)� curve_fit� c
C sd t || |d�\}}}} t|||| |||| d�\}
} t|
dd�df |
dd�df |
dd�df | �S )zsreturns rotational dinamics from initial condition IC along time at Reynolds Rep for the spheroid with aspect ratio)�req�Rep)�IC�time�gammadot�r�b1�b2�b3�b4Nr � � )�
load_betas�Einarsson_interface�split_orbits_and_get_C) r r �dirr r r
r r
r r ZnnnZttt� r �_/Users/davide/Downloads/Joint_paper_figures/jupyter_notebook_submission/figure_5/Einarsson15.py�deltaC_einarsson15 s r c C s� t �| d �}t �| d �}t �| d �}t �| d �}t�|dd�df |dd�df �}t�|dd�df |dd�df �}t�|dd�df |dd�df �} t�|dd�df |dd�df �}
t �||�||�| |�|
|�g�| } | S )zWLoads the graphically extrapolated coefficients from Einarsson et al., PoF 2015, Fig. 2zB1.txtzB2.txtzB3.txtzB4.txtNr r )�np�loadtxtr Zinterp1d�array) �folderr r ZB1dataZB2dataZB3dataZB4data�f1�f2�f3�f4�betasr r r r s $$$$&r c
C s< t �g d�g d�g d�g�| }t �g d�g d�g d�g�| } t �| t �| | ��}
t �| t �|t �| | ���} t �| t �| t �| | ���} || }|| }|| }|| }t �||| | �||
| ||
t �| | �| |
||
t �|| � |t �|t �| | ��| | |t �| t �| | ��| | S )zMCalculates the rotation rate according to Einarsson et al., PoF, 2015: eq. 39)r � �?r )g �r r )r r r )r" r r )r r �dot)
�p�t�betar
r r
r r �Omega�EZpEpZpOEpZpEEpr r r �rotation_Einarsson( s r) c
C s\ |d d |d d }|} t �| d | d | d g�}
tt|
| ||||||fd�} | | fS )zQManages the integration of the Einarsson rotation rate from a given initial_valuer r r )�args)r r r r) ) r r r
r r r
r r r& ZsimtimeZpinitZptheoryr r r r 2 s
r c C s2 || t �t �| �d |d t �| �d � S )Nr )r �sqrt�cos�sin)�phi�Cr r r r �
jeffery_model< s r0 c s� � fdd�}zRt ||t�| �dtjgdgd�\}}|jrLt�t�|��d ntj}|d |fW S ty| tjtjf Y S 0 d S )Nc s t | |� �S )N)r0 )r. r/ �r r r �model@ s zfit_C.<locals>.modelr � �?)�bounds�p0) r r �tan�inf�sizer+ �diag�nan� Exception)�thetar. r r2 ZC_fit�covZC_stdr r1 r �fit_C? s $ r> c C s� |dkr t �t �| ��dk}nt �t �|��dk}t j| td�}t �|�|dd�< |�� d }g }t|�D ]N}t �|dd� |k�} t � | | || �}
t �
|| �} |� t | |
|�� qjt �
|�jS )z&Method to split orbits in semi-periodsr3 r )�dtyper N)r �diff�sign�
zeros_like�int�cumsum�max�range�where�arctan2�arccos�appendr> r �T) �n1�n2�n3r � sign_changeZ orbit_labels�
num_orbitsr/ �iZ
orbit_indsr. r< r r r r J s r )r NNN)�__doc__�numpyr �scipyr Zscipy.integrater �scipy.optimizer r r r) r r0 r> r r r r r �<module> s