#!/usr/bin/env python3
"""
简单检查fast_results.pkl文件的结构
"""

import pickle
import sys

def check_file(results_file):
    try:
        print(f"检查文件: {results_file}")

        # 只读取文件头，不完全加载
        with open(results_file, 'rb') as f:
            # 尝试读取文件大小
            f.seek(0, 2)  # 移到文件末尾
            file_size = f.tell()
            print(f"文件大小: {file_size / (1024**3):.2f} GB")

            # 回到文件开头
            f.seek(0)

            # 尝试读取pickle头
            try:
                # 使用最高协议读取
                results = pickle.load(f)
                print(f"数据类型: {type(results)}")

                if isinstance(results, list):
                    print(f"列表长度: {len(results)}")
                    if len(results) > 0:
                        first_item = results[0]
                        print(f"第一个元素类型: {type(first_item)}")
                        if isinstance(first_item, dict):
                            print(f"字典键: {list(first_item.keys())}")
                elif isinstance(results, dict):
                    print(f"字典键: {list(results.keys())}")

            except Exception as e:
                print(f"Pickle读取错误: {e}")

    except Exception as e:
        print(f"文件操作错误: {e}")

if __name__ == "__main__":
    if len(sys.argv) != 2:
        print("用法: python CHECK_RESULTS.py <results_file>")
        sys.exit(1)

    check_file(sys.argv[1])