unlisted
ubuntu2404�
=bi� � �b � d Z ddlZddlmZ ddlmZ ddlmZ dd�Z d� Z
d � Z d
� Z d � Z
d � Zd
� ZdS )z6
Created on Fri Nov 7 19:15:57 2025
@author: davide
� N)� interpolate)�odeint)� curve_fit� c
�� � t || |�� � \ }}}} t |||| |||| �� � \ }
} 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 �nnn�ttts �R/Users/ziqiwang/PD/MP_PD/2025-JFM/Jupyter_notebook_figures/figure_4/Einarsson15.py�deltaC_einarsson15r s} � � �S�Q�3�/�/�/�K�B�r�"�R�!�R�T�8�a�2�QS�WY�]_�`�`�`�G�C�� !�#�a�a�a��c�(�3�q�q�q��s�8�C����!��H�Q� ?� ?�?� c � � t j | dz � � }t j | dz � � }t j | dz � � }t j | dz � � }t j |dd�df |dd�df � � }t j |dd�df |dd�df � � }t j |dd�df |dd�df � � } t j |dd�df |dd�df � � }
t j ||� � ||� � | |� � |
|� � g� � |z } | S )zWLoads the graphically extrapolated coefficients from Einarsson et al., PoF 2015, Fig. 2zB1.txtzB2.txtzB3.txtzB4.txtNr r )�np�loadtxtr �interp1d�array) �folderr r �B1data�B2data�B3data�B4data�f1�f2�f3�f4�betass r r r s8 � �
�Z��x��
(�
(�F�
�Z��x��
(�
(�F�
�Z��x��
(�
(�F�
�Z��x��
(�
(�F� � �f�Q�Q�Q�q�S�k�6�!�!�!�A�#�;� 7� 7�B� � �f�Q�Q�Q�q�S�k�6�!�!�!�A�#�;� 7� 7�B� � �f�Q�Q�Q�q�S�k�6�!�!�!�A�#�;� 7� 7�B� � �f�Q�Q�Q�q�S�k�6�!�!�!�A�#�;� 7� 7�B� �H�b�b��g�g�b�b��g�g�b�b��g�g�b�b��g�g�6� 7� 7�� <�E� �Lr c
�r � t j g d�g d�g d�g� � |z }t j g d�g d�g d�g� � |z } t j | t j | | � � � � }
t j | t j |t j | | � � � � � � } t j | t j | t j | | � � � � � � } ||z }||z }||z }||z }t j ||| z z | � � ||
z | z z
||
z t j | | � � | |
z z
z z ||
z t j || � � z z |t j |t j | | � � � � | | z z
z z |t j | t j | | � � � � | | z z
z z 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�E�pEp�pOEp�pEEps
r �rotation_Einarssonr7 ( s � �
�X�{�{�{�<�<�<�
�
�
�;�
<�
<�x�
G�E�
��+�+�+�k�k�k�*�*�*�5�6�6��A�A�
��q����!��� � �C� ���"�&��r�v�a��{�{�+�+� ,� ,�D� ���"�&��2�6�!�A�;�;�'�'� (� (�D���L�B��X� ��b�(�l�b�2�x�<�2�
�6�%��Q��,�q� !� !�$�s�(�1�*� ,�R��V�R�V�A�a�[�[��3��5F�-G� G��3��r�v�V[�\]���H^� ^�_a�ce�ci�jo�pr�pv�wx�yz�p{�p{�c|�c|� ~B� CD� ~D� dD� `E� E� FH� JL� JP� QR� SU� SY� Z[� \]� S^� S^� J_� J_� `d� ef� `f� Jf� Fg� g� gr c � � |dz dz
|dz dz z }|} t j | d | d | d g� � }
t t |
| ||||||f�� � } | | fS )zQManages the integration of the Einarsson rotation rate from a given initial_valuer r r )�args)r r! r r7 ) r
r r r
r r r r r1 �simtime�pinit�ptheorys r r r 2 ss � �
�q�D��F�Q��T�!�V� �D��G�
�(�B�q�E�"�Q�%��1��&�
'�
'�E��'��g�D��"�R�PR�SU�;V�W�W�W�G� �7�?�r c � � ||z t j t j | � � dz |dz t j | � � dz z z � � z S )Nr )r �sqrt�cos�sin��phi�Cr
s r �
jeffery_modelrD < sA � � �q�5�2�7�2�6�#�;�;��>�A�q�D�2�6�#�;�;��>�,A�A�B�B� B�Br c �^ �� �fd�} t ||t j | � � dt j gdg�� � \ }}|j r,t j t j |� � � � d n t j }|d |fS # t $ r t j t j fcY S w xY w)Nc �& �� t | |�� � S )N)rD rA s �r �modelzfit_C.<locals>.model@ s �� � �S�!�Q�'�'�'r r � �?)�bounds�p0) r r �tan�inf�sizer> �diag�nan� Exception)�thetarB r
rG �C_fit�cov�C_stds ` r �fit_CrU ? s� �� �(� (� (� (� (�"� �u�c�2�6�%�=�=�!�R�V��RU�QV�W�W�W�
��s�,/�H�@������ � �%�%�a�(�(�"�&���Q�x��� �� � "� "� "��6�2�6�>� !� !� !�"���s �A>B �"B,�+B,c � � |dk r+t j t j | � � � � dk }n*t j t j |� � � � dk }t j | t �� � }t j |� � |dd�< |� � � dz }g }t |� � D ]�}t j |dd� |k � � } t j | | || � � }
t j
|| � � } |� t | |
|� � � � ��t j
|� � j S )z&Method to split orbits in semi-periodsrH r )�dtyper N)r �diff�sign�
zeros_like�int�cumsum�max�range�where�arctan2�arccos�appendrU r! �T) �n1�n2�n3r
� sign_change� orbit_labels�
num_orbitsrC �i�
orbit_indsrB rQ s r r r J s � � �3�w�w��g�b�g�b�k�k�*�*�a�/� � ��g�b�g�b�k�k�*�*�a�/� ��=��3�/�/�/�L��y��-�-�L����� �!�!�#�#�a�'�J�
�A�
�:�
�
� %� %���X�l�1�1�1�o�q� 0�1�1�
��j��J� ��:��7�7��� �"�Z�.�)�)�� ����u�S��#�#�$�$�$�$�
�8�A�;�;�=�r )r NNN)�__doc__�numpyr �scipyr �scipy.integrater �scipy.optimizer r r r7 r rD rU r � r r �<module>rr s� ��� � � � � � � � � � � � "� "� "� "� "� "� $� $� $� $� $� $�@� @� @� @�
�
�
� g� g� g�� � �C� C� C� "� "� "�� � � � r