import pickle
from Unpack import unpack
from read_files import read_radar_data, read_motion_data
from data_align import align_data
motion_data = read_motion_data("../Raw_Data/UASSAR4_rail_diagonal.csv","UASSAR4")
radar_data = read_radar_data("../Raw_Data/railTestDiagonal.pkl")
aligned_data = align_data(radar_data,motion_data,272,1844,5368)
Pulses = aligned_data[0]
Final_Motion = aligned_data[1]
print(Final_Motion)
aligned_data[2] = radar_data[2]
with open('parrot.pkl', 'wb') as f:
pickle.dump(aligned_data,f)
f.close()
#Performs RCS correction
radar_data = rcs(radar_data)
#Plot RTI for radar_data
plotRTI(radar_data)
#Set parameters
#Take motion_start and motion_end from .tak file video
#Estimate radar_start from the plotRTI above, initially comment out code below
radar_start = 300 #Initial 500 380 for flight 6 300 for flight 7
motion_start = 3850 #5200 flight 6 3900 flight 7 2411 flight 8
motion_end = 34129 #16400 flight 6 34129 flight 7 29126 flight 8
#Aligns data
aligned_data = align_data(radar_data, motion_data, radar_start, motion_start,
motion_end, 200, 4600) #100, 3500
#Plots aligned graph with known data point
AlignedGraph(
aligned_data, radar_data, [[-0.254065627, 0.252586546, 2.662674582],
[-0.264097265, 0.237835515, -3.115790777]]
) #[[.942713,.1,1.019]] Final: [[-0.254065627, 0.252586546, 2.662674582], [-0.264097265, 0.237835515, -3.115790777]]
#Calculates and plots BackProjected Image
#IntensityList = BackProjection(aligned_data,radar_data,[-4,0],[4,4],0.1)
IntensityList2 = FastBackProjection(aligned_data, radar_data, [-2.0, 0.5],
[1.5, 1.75], 0.002)
#Deconvolutes image and plots images
Image = deconvolute(IntensityList2,
IterationNumber=3,
#Finds the first point of the motion data that it starts to move
#motion_start = find_i_of_first_motion(motion_data)
#Finds the last point of the motion data that it moves
#motion_end = find_i_of_last_motion(motion_data)
#Finds the first point in the radar data that it starts to move
#radar_start = find_point_one_radar(radar_data)
radar_start = 600
motion_start = 6600
motion_end = 29545
#Aligns data, currently using frames given in function definition
aligned_data = align_data(radar_data,motion_data,radar_start,motion_start, motion_end)
#Plots aligned graph
#AlignedGraph(aligned_data,radar_data)
#Calculates and plots BackProjected Image
IntensityList = BackProjection(aligned_data,radar_data,[-5,-5],[5,5],0.5)
#IntensityList = read_intensity('../Raw_Data/intensity2.csv')
#Deconvolutes image and plots images
#deconvolute(IntensityList, IterationNumber = 3, PercentageMin = 1/5.5)
'''
#Currently unused code
#Calculates backprojected image using Ramu's algorithm