| Server IP : 217.160.0.135 / Your IP : 216.73.217.85 Web Server : Apache System : Linux www 6.18.52-i1-ampere #1203 SMP Mon Sep 14 18:29:59 CEST 2026 aarch64 User : sws1074145052 ( 1074145052) PHP Version : 8.3.32 Disable Function : NONE MySQL : OFF | cURL : ON | WGET : ON | Perl : ON | Python : OFF | Sudo : OFF | Pkexec : OFF Directory : /lib/python3/dist-packages/scipy/io/__pycache__/ |
Upload File : |
�
d�c�* � �x � d Z ddlZddlZg d�Z G d� dee� � Z G d� dee� � Z G d� d � � Z dS )
z�
Module to read / write Fortran unformatted sequential files.
This is in the spirit of code written by Neil Martinsen-Burrell and Joe Zuntz.
� N)� FortranFile�FortranEOFError�FortranFormattingErrorc � � e Zd ZdZdS )r z�Indicates that the file ended properly.
This error descends from TypeError because the code used to raise
TypeError (and this was the only way to know that the file had
ended) so users might have ``except TypeError:``.
N��__name__�
__module__� __qualname__�__doc__� � �3/usr/lib/python3/dist-packages/scipy/io/_fortran.pyr r
s � � � � � �� � �Dr
r c � � e Zd ZdZdS )r zhIndicates that the file ended mid-record.
Descends from TypeError for backward compatibility.
Nr r r
r r r s � � � � � �� �
�Dr
r c �^ � e Zd ZdZdej fd�Zdd�Zd� Zd� Z dd �Z
dd �Z d � Z d
� Z
d� ZdS )r a�
A file object for unformatted sequential files from Fortran code.
Parameters
----------
filename : file or str
Open file object or filename.
mode : {'r', 'w'}, optional
Read-write mode, default is 'r'.
header_dtype : dtype, optional
Data type of the header. Size and endiness must match the input/output file.
Notes
-----
These files are broken up into records of unspecified types. The size of
each record is given at the start (although the size of this header is not
standard) and the data is written onto disk without any formatting. Fortran
compilers supporting the BACKSPACE statement will write a second copy of
the size to facilitate backwards seeking.
This class only supports files written with both sizes for the record.
It also does not support the subrecords used in Intel and gfortran compilers
for records which are greater than 2GB with a 4-byte header.
An example of an unformatted sequential file in Fortran would be written as::
OPEN(1, FILE=myfilename, FORM='unformatted')
WRITE(1) myvariable
Since this is a non-standard file format, whose contents depend on the
compiler and the endianness of the machine, caution is advised. Files from
gfortran 4.8.0 and gfortran 4.1.2 on x86_64 are known to work.
Consider using Fortran direct-access files or files from the newer Stream
I/O, which can be easily read by `numpy.fromfile`.
Examples
--------
To create an unformatted sequential Fortran file:
>>> from scipy.io import FortranFile
>>> import numpy as np
>>> f = FortranFile('test.unf', 'w')
>>> f.write_record(np.array([1,2,3,4,5], dtype=np.int32))
>>> f.write_record(np.linspace(0,1,20).reshape((5,4)).T)
>>> f.close()
To read this file:
>>> f = FortranFile('test.unf', 'r')
>>> print(f.read_ints(np.int32))
[1 2 3 4 5]
>>> print(f.read_reals(float).reshape((5,4), order="F"))
[[0. 0.05263158 0.10526316 0.15789474]
[0.21052632 0.26315789 0.31578947 0.36842105]
[0.42105263 0.47368421 0.52631579 0.57894737]
[0.63157895 0.68421053 0.73684211 0.78947368]
[0.84210526 0.89473684 0.94736842 1. ]]
>>> f.close()
Or, in Fortran::
integer :: a(5), i
double precision :: b(5,4)
open(1, file='test.unf', form='unformatted')
read(1) a
read(1) b
close(1)
write(*,*) a
do i = 1, 5
write(*,*) b(i,:)
end do
�rc �H � |�t d� � �t j |� � }|j dk rt j d� � |dvst
|� � dk rt d� � �t |d� � r|| _ nt |d|z � � | _ || _
d S ) NzMust specify dtype�uz$Given a dtype which is not unsigned.�rw� zmode must be either r or w�seekz%sb) �
ValueError�np�dtype�kind�warnings�warn�len�hasattr�_fp�open�
_header_dtype)�self�filename�mode� header_dtypes r �__init__zFortranFile.__init__m s� � � � � � 1�2�2� 2��x� �-�-� � � �� #� #� �M�@� A� A� A� �t� � �s�4�y�y�A�~�~� � 9�:�:� :� �8�V� $� $� 4� �D�H�H��H�e�d�l�3�3�D�H� )����r
Fc � � | j j }| j � |� � }|s|rt d� � �t |� � |k rt
d� � �t t j || j d�� � d � � S )Nz%End of file occurred at end of recordz,End of file in the middle of the record sizer �r �countr )
r! �itemsizer �readr r r �intr �
frombuffer)r" �eof_ok�n�bs r �
_read_sizezFortranFile._read_size s� � � � � '�� �H�M�M�!� � ��� @�v� @�!�"I�J�J� J�
��V�V�a�Z�Z�(�>�@� @�
@��2�=��$�*<�A�F�F�F�q�I�J�J�Jr
c �H � t d� |D � � � � }t d� |D � � � � }t j |g| j �� � }|� | j � � |D ] }|� | j � � � |� | j � � dS )a
Write a record (including sizes) to the file.
Parameters
----------
*items : array_like
The data arrays to write.
Notes
-----
Writes data items to a file::
write_record(a.T, b.T, c.T, ...)
write(1) a, b, c, ...
Note that data in multidimensional arrays is written in
row-major order --- to make them read correctly by Fortran
programs, you need to transpose the arrays yourself when
writing them.
c 3 �> K � | ]}t j |� � V � �d S �N)r �asarray��.0�items r � <genexpr>z+FortranFile.write_record.<locals>.<genexpr>� s, � � � �9�9�4�b�j��&�&�9�9�9�9�9�9r
c 3 �$ K � | ] }|j V � � d S r4 )�nbytesr6 s r r9 z+FortranFile.write_record.<locals>.<genexpr>� s$ � � � �7�7����7�7�7�7�7�7r
)r N)�tuple�sumr �arrayr! �tofiler )r" �items�
total_size�nbr8 s r � write_recordzFortranFile.write_record� s� � �. �9�9�5�9�9�9�9�9���7�7��7�7�7�7�7�
�
�X�z�l�$�*<�
=�
=�
=��
� � �$�(����� "� "�D� �K�K��� !� !� !� !�
� � �$�(�����r
c � � |� dd� � }|rAt d� t |� � � � � � � � � �|�||fz }n|st d� � �| � d�� � }t d� |D � � � � }t
d� |D � � � � }t ||� � \ }}|d k r#t d
� ||� � � � �t |� � d k r)||k r#t d � ||� � � � �g }|D ]�}t j
| j ||�
� � } t | � � |k rt d� � �|j
dk r#|d k r | j
d|j
z k sJ �| d } |� | � � ��| � � � }
||
k rt d� � �t |� � d k r|d S t |� � S )a*
Reads a record of a given type from the file.
Parameters
----------
*dtypes : dtypes, optional
Data type(s) specifying the size and endiness of the data.
Returns
-------
data : ndarray
A 1-D array object.
Raises
------
FortranEOFError
To signal that no further records are available
FortranFormattingError
To signal that the end of the file was encountered
part-way through a record
Notes
-----
If the record contains a multidimensional array, you can specify
the size in the dtype. For example::
INTEGER var(5,4)
can be read with::
read_record('(4,5)i4').T
Note that this function does **not** assume the file data is in Fortran
column major order, so you need to (i) swap the order of dimensions
when reading and (ii) transpose the resulting array.
Alternatively, you can read the data as a 1-D array and handle the
ordering yourself. For example::
read_record('i4').reshape(5, 4, order='F')
For records that contain several variables or mixed types (as opposed
to single scalar or array types), give them as separate arguments::
double precision :: a
integer :: b
write(1) a, b
record = f.read_record('<f4', '<i4')
a = record[0] # first number
b = record[1] # second number
and if any of the variables are arrays, the shape can be specified as
the third item in the relevant dtype::
double precision :: a
integer :: b(3,4)
write(1) a, b
record = f.read_record('<f4', np.dtype(('<i4', (4, 3))))
a = record[0]
b = record[1].T
NumPy also supports a short syntax for this kind of type::
record = f.read_record('<f4', '(3,3)<i4')
See Also
--------
read_reals
read_ints
r Nz Unknown keyword arguments {}z Must specify at least one dtypeT)r. c 3 �> K � | ]}t j |� � V � �d S r4 )r r �r7 r s r r9 z*FortranFile.read_record.<locals>.<genexpr>� s* � � � �;�;�5�r�x����;�;�;�;�;�;r
c 3 �$ K � | ] }|j V � � d S r4 )r* rF s r r9 z*FortranFile.read_record.<locals>.<genexpr> s$ � � � �<�<�E�� �<�<�<�<�<�<r
r z@Size obtained ({0}) is not a multiple of the dtypes given ({1}).r zTSize obtained ({0}) does not match with the expected size ({1}) of multi-item recordr( z%End of file in the middle of a recordr )r zPSizes do not agree in the header and footer for this record - check header dtype)�popr �formatr< �keysr1 r= �divmodr r �fromfiler r �shape�append) r" �dtypes�kwargsr �
first_size�
block_size�
num_blocks� remainder�datar � second_sizes r � read_recordzFortranFile.read_record� sO � �T �
�
�7�D�)�)�� � Z� � ;� B� B�5���� � �CW�CW� X� X�Y�Y� Y� � ��u�h�&�F�F�� @� � >�?�?� ?��_�_�D�_�1�1�
��;�;�F�;�;�;�;�;���<�<�V�<�<�<�<�<�
� &�z�:� >� >� �
�I� ��>�>� � 3�39�6�*�j�3Q�3Q�S� S�
S� �v�;�;�!� � �
�j� 8� 8� � ?�?E�v�j�R\�?]�?]�_� _�
_� ��� � �E�� �D�H�E��D�D�D�A��1�v�v��#�#�,�;�=� =� =��{�b� � � ��?�?� �7�d�U�[�&8�8�8�8�8��!��A� �K�K��N�N�N�N��o�o�'�'� � �� $� $� � @� A� A�
A� �v�;�;�!� � ���7�N���;�;� r
�i4c �, � | � |� � S )a�
Reads a record of a given type from the file, defaulting to an integer
type (``INTEGER*4`` in Fortran).
Parameters
----------
dtype : dtype, optional
Data type specifying the size and endiness of the data.
Returns
-------
data : ndarray
A 1-D array object.
See Also
--------
read_reals
read_record
�rW �r" r s r � read_intszFortranFile.read_ints' � � �* � � ��&�&�&r
�f8c �, � | � |� � S )a�
Reads a record of a given type from the file, defaulting to a floating
point number (``real*8`` in Fortran).
Parameters
----------
dtype : dtype, optional
Data type specifying the size and endiness of the data.
Returns
-------
data : ndarray
A 1-D array object.
See Also
--------
read_ints
read_record
rZ r[ s r �
read_realszFortranFile.read_reals> r] r
c �8 � | j � � � dS )z�
Closes the file. It is unsupported to call any other methods off this
object after closing it. Note that this class supports the 'with'
statement in modern versions of Python, to call this automatically
N)r �close�r" s r rb zFortranFile.closeU s � �
��������r
c � � | S r4 r rc s r � __enter__zFortranFile.__enter__^ s � �� r
c �. � | � � � d S r4 )rb )r" �type�value�tbs r �__exit__zFortranFile.__exit__a s � � �
�
� � � � � r
N)F)rX )r^ )r r r
r r �uint32r& r1 rC rW r\ r` rb re rj r r
r r r ! s� � � � � � �J� J�V '*�� � *� *� *� *�$K� K� K� K� � � �B{ � { � { �z'� '� '� '�.'� '� '� '�.� � �� � �� � � � r
r )
r r �numpyr �__all__� TypeError�OSErrorr r r r r
r �<module>rp s� ��� � ���� � � � �
F�
F�
F�� � � � � �i�� � � � � � � � �Y�� � � �A� A� A� A� A� A� A� A� A� Ar