在 python 中读取 *.mhd/*.raw 格式

在Python中读取`.mhd`和`.raw`格式的文件通常涉及到两个主要步骤：解析MHD文件以获取数据的基本信息和结构，然后根据这些信息读取对应的RAW数据文件。以下是一个简单的示例说明如何进行这一过程。

### 1. 解析MHD文件

首先，我们需要编写一个函数来解析`.mhd`文件内容，提取出数据的基本信息，比如图像的尺寸、像素间距等信息。

```python
import re

def parse_mhd(filename):
    with open(filename, 'r') as file:
        meta_data = {}
        for line in file:
            key, value = line.strip().split('=')  # 假设MHD文件的键值对都是等号分隔的
            meta_data[key] = value

    # 将元数据转换为整数或浮点数（如果需要的话）
    if 'DimSize' in meta_data:
        meta_data['DimSize'] = list(map(int, meta_data['DimSize'].split()))
    if 'ElementSpacing' in meta_data:
        meta_data['ElementSpacing'] = tuple(map(float, meta_data['ElementSpacing'].split()))

    return meta_data
```

### 2. 读取RAW数据文件

接下来，我们根据解析得到的元数据来读取RAW数据。对于`.raw`格式的数据，通常是连续的字节流，需要根据元数据中的尺寸和数据类型进行解码。

```python
import numpy as np

def read_raw(filename, metadata):
    # 计算总字节数
    total_bytes = np.prod(metadata['DimSize']) * np.dtype(metadata['ElementType']).itemsize

    # 读取文件并转换为NumPy数组
    with open(filename, 'rb') as file:
        data = np.frombuffer(file.read(), dtype=metadata['ElementType'], count=-1, offset=0)

    # 如果数据不是C语言的行序，需要进行转置
    if metadata['DataByteOrder'] == 'LittleEndian' and sys.byteorder != 'little':
        data = data.byteswap().newbyteorder()

    return data.reshape(metadata['DimSize'])
```

### 完整示例

结合上述两个函数，我们可以编写一个完整的读取`.mhd/.raw`文件的函数：

```python
import numpy as np

def load_mhd_data(mhd_filename):
    meta = parse_mhd(mhd_filename)
    raw_data = read_raw(re.sub('\.mhd$', '.raw', mhd_filename), meta)
    return raw_data, meta

# 使用示例：
mhd_file = 'image.mhd'
data, metadata = load_mhd_data(mhd_file)
print("Image shape:", data.shape)
```

请注意，这个示例假设MHD文件中的数据类型是支持NumPy直接读取的类型（如'int32'、'float32'等），并且RAW文件的字节顺序与系统默认字节序一致或不需要交换。在实际应用中，可能需要根据实际情况调整处理细节和错误处理。