tst

import jinja2

# import jinja2

#this is test message 0523

def test():

    try:

import numpy as np

import matplotlib.pyplot as plt

import ephem

        template = jinja2.Template("Hello {{ name }}!")

        assert template.render(name="World") == "Hello World!"

        print("✅ Jinja2 模板渲染测试通过")

    except Exception as e:

        print("❌ Jinja2 测试失败:", e)

# def test():

#     try:

#         template = jinja2.Template("Hello {{ name }}!")

#         assert template.render(name="World") == "Hello World!"

#         print("✅ Jinja2 模板渲染测试通过")

#     except Exception as e:

#         print("❌ Jinja2 测试失败:", e)

def simulate_star_trajectory(semi_major_axis, eccentricity, period, start_date, end_date, step_days):

    # Initialize observer location (e.g., Earth location)

    observer = ephem.Observer()

    observer.lat, observer.lon = '0', '0'

    # Create a body using orbital elements

    star = ephem.EllipticalBody()

    star._a = semi_major_axis         # Semi-major axis in AU

    star._e = eccentricity            # Eccentricity of the orbit

    star._Om = 0.0                    # Longitude of ascending node

    star._om = 0.0                    # Argument of perihelion

    star._inc = 0.0                   # Inclination to the ecliptic

    star._M = 0.0                     # Mean anomaly

    star._epoch = ephem.Date(start_date)  # Epoch of perihelion

    # Date range

    date_range = np.arange(ephem.Date(start_date), ephem.Date(end_date), step_days)

    # Store positions

    positions = []

    for date in date_range:

        observer.date = date

        star.compute(observer)

        #将天体坐标转换为黄道坐标

        ecliptic_coords = ephem.Ecliptic(star)

        # 获取黄道坐标

        x, y = ecliptic_coords.lon, ecliptic_coords.lat # lon 对应黄道经度，lat 对应黄道纬度

        # x, y = ephem.Ecliptic(star).suun()

        positions.append((x, y))

    # Extract x and y coordinates

    x_coords, y_coords = zip(*positions)

    # Visualization

    plt.plot(x_coords, y_coords, label='Star trajectory')

    plt.scatter(x_coords[0], y_coords[0], color='red', label='Start')

    plt.scatter(x_coords[-1], y_coords[-1], color='green', label='End')

    plt.xlabel('X (AU)')

    plt.ylabel('Y (AU)')

    plt.title('Simulated Star Trajectory')

    plt.legend()

    plt.grid()

    plt.show()

    plt.savefig("star_trajectory.png")

    print("图像已保存为 star_trajectory.png")

# try:

#     import jsonschema

import os

import shutil

#from codesp import test

from codesp import simulate_star_trajectory

def main():

    # test()

    # 定义初始路径和目标路径

    source_dir = '/workspace/input/'

    target_dir = '/workspace/output/'

    # 确保源路径存在

    if not os.path.exists(source_dir):

        print(f"错误：源路径 {source_dir} 不存在。")

    else:

        # 创建目标路径（如果不存在）

        os.makedirs(target_dir, exist_ok=True)

        # 获取源路径下的所有文件（不递归子目录）

        for file_name in os.listdir(source_dir):

            source_file = os.path.join(source_dir, file_name)

            target_file = os.path.join(target_dir, file_name)

            # 只拷贝文件，忽略子目录

            if os.path.isfile(source_file):

                shutil.copy2(source_file, target_file)  # copy2 保留元数据（如修改时间）

                print(f"已拷贝: {source_file} -> {target_file}")

        print("拷贝完成。")

    simulate_star_trajectory(semi_major_axis=1.0, eccentricity=0.1, period=365.25, start_date='2023/1/1', end_date='2024/1/1', step_days=10)

    # # 定义初始路径和目标路径

    # source_dir = '/workspace/input/'

    # target_dir = '/workspace/output/'

    # # 确保源路径存在

    # if not os.path.exists(source_dir):

    #     print(f"错误：源路径 {source_dir} 不存在。")

    # else:

    #     # 创建目标路径（如果不存在）

    #     os.makedirs(target_dir, exist_ok=True)

    #     # 获取源路径下的所有文件（不递归子目录）

    #     for file_name in os.listdir(source_dir):

    #         source_file = os.path.join(source_dir, file_name)

    #         target_file = os.path.join(target_dir, file_name)

    #         # 只拷贝文件，忽略子目录

    #         if os.path.isfile(source_file):

    #             shutil.copy2(source_file, target_file)  # copy2 保留元数据（如修改时间）

    #             print(f"已拷贝: {source_file} -> {target_file}")

    #     print("拷贝完成。")

if __name__ == '__main__':

    main()