92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
|
"""0.fix import problem"""
|
|||
|
|
import sys
|
|||
|
|
from pathlib import Path
|
|||
|
|
|
|||
|
|
# 获取项目根目录（自动适配 Windows/Linux/Mac）
|
|||
|
|
sys.path.append(str(Path(__file__).parent.parent))
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""1.import libs"""
|
|||
|
|
import numpy as np
|
|||
|
|
import matplotlib
|
|||
|
|
matplotlib.use('Agg')
|
|||
|
|
# Agg 是非交互式后端，不需要 display，图片可通过 plt.savefig() 保存为文件。现在可以恢复 matplotlib import 并正常调试了。
|
|||
|
|
import matplotlib.pyplot as plt
|
|||
|
|
# uv pip install construct
|
|||
|
|
from construct import *
|
|||
|
|
# uv pip install scipy
|
|||
|
|
import scipy.signal
|
|||
|
|
|
|||
|
|
import Tianwen.ccsds as ccsds
|
|||
|
|
import Tianwen.tianwen1_tm as tianwen1_tm
|
|||
|
|
|
|||
|
|
import struct
|
|||
|
|
import collections
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""2.define some functions"""
|
|||
|
|
def get_packet(p):
|
|||
|
|
return p[0] if type(p) is tuple else p
|
|||
|
|
|
|||
|
|
def packets_asarray(packets):
|
|||
|
|
return np.array([np.frombuffer(get_packet(p)[ccsds.SpacePacketPrimaryHeader.sizeof():], 'uint8')
|
|||
|
|
for p in packets])
|
|||
|
|
|
|||
|
|
def plot_apids(apids, sc, vc):
|
|||
|
|
for apid in sorted(apids.keys()):
|
|||
|
|
plt.figure(figsize = (16,16), facecolor = 'w')
|
|||
|
|
ps = packets_asarray(apids[apid])
|
|||
|
|
plt.imshow(ps, aspect = ps.shape[1]/ps.shape[0])
|
|||
|
|
plt.title(f'APID {apid} Spacecraft {sc} Virtual channel {vc}')
|
|||
|
|
|
|||
|
|
def get_timestamps(aos):
|
|||
|
|
return np.array([tianwen1_tm.parse_timestamp_datetime64(a.insert_zone.timestamp) for a in aos])
|
|||
|
|
|
|||
|
|
def get_packet_timestamps(packets):
|
|||
|
|
return np.array([tianwen1_tm.parse_timestamp_datetime64(p[1]) for p in packets])
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""3.set some frame vars"""
|
|||
|
|
frame_size = 220
|
|||
|
|
frames = np.fromfile('/home/zjz/CCSDS_study/Tianwen/tianwen1_frames_20200730.u8', dtype = 'uint8')
|
|||
|
|
frames = frames[:frames.size//frame_size*frame_size].reshape((-1, frame_size))
|
|||
|
|
print("frames.shape[0]: ", frames.shape[0])
|
|||
|
|
print("\n")
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""4.AOS header"""
|
|||
|
|
aos = [ccsds.AOSFrame.parse(f) for f in frames]
|
|||
|
|
|
|||
|
|
collections.Counter([a.primary_header.transfer_frame_version_number for a in aos])
|
|||
|
|
|
|||
|
|
collections.Counter([a.primary_header.spacecraft_id for a in aos])
|
|||
|
|
|
|||
|
|
collections.Counter([a.primary_header.virtual_channel_id for a in aos
|
|||
|
|
if a.primary_header.spacecraft_id == 245])
|
|||
|
|
|
|||
|
|
collections.Counter([a.primary_header.virtual_channel_id for a in aos
|
|||
|
|
if a.primary_header.spacecraft_id == 82])
|
|||
|
|
|
|||
|
|
print([(a.primary_header.spacecraft_id, a.primary_header.virtual_channel_id) for a in aos[:20]])
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""5.Spacecraft 82 Virtual channel 1"""
|
|||
|
|
print([a.primary_header for a in aos if a.primary_header.spacecraft_id == 82][:10])
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
"""for test"""
|
|||
|
|
# def main():
|
|||
|
|
# frame_size = 220
|
|||
|
|
# frames = np.fromfile('/home/zjz/CCSDS_study/Tianwen/tianwen1_frames_20200730.u8', dtype = 'uint8')
|
|||
|
|
# frames = frames[:frames.size//frame_size*frame_size].reshape((-1, frame_size))
|
|||
|
|
# print("frames.shape[0]: ", frames.shape[0])
|
|||
|
|
# print("\n")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# if __name__ == "__main__":
|
|||
|
|
# print("hello world!")
|
|||
|
|
# main()
|
|||
|
|
|
|||
|
|
|