from cis import read_data_list


# -------- This breaks with the GASSP plugin, but not the (identical) NCAR_NetCDF_RAF ------------

#from pympler import tracker, classtracker

# tr = classtracker.ClassTracker()
#
# tr.track_class(UngriddedData, trace=1, resolution_level=2)
#
# tr.start_periodic_snapshots()


c_data = read_data_list("ACE1/ACE1_CPC*.nc",
                        ["N3", "N10", "N15_NONVOL", "N15_VOLATILE", "N3_10"], product='GASSP')

# tr.create_snapshot("After creating combined data")
#
# tr.stats.print_stats()
import pdb

pdb.set_trace()


# Try writing out the original data
c_data.save_data("abcdefghijklmnop")
# c_data.save_data("ACE1_aircraft.nc")


# -------- This doesn't reproduce the problem.... ------------

# air_pressure = []
# longitude = []
# latitude = []
# time = []
# data_1 = []
# data_2 = []
# data_3 = []
# data_4 = []
# data_5 = []
#
# for f in expand_filelist("ACE1/ACE1_CPC*.nc"):
#     d = Dataset(f, mode='r')
#
#     lon = d['LONGITUDE'][:]
#     lat = d['LATITUDE'][:]
#     ap = d['AIR_PRESSURE'][:]
#     t = d['TIME'][:]
#     data_1.append(['N3'][:])
#     data_2.append(['N10'][:])
#     data_3.append(['N15_NONVOL'][:])
#     data_4.append(['N15_VOLATILE'][:])
#     data_5.append(['N3_10'][:])
#     ap = Unit('mbar').convert(ap, 'hPa')
#     air_pressure.append(ap)
#     longitude.append(lon)
#     latitude.append(lat)
#     time.append(t)
#
#     d.close()
#
# all_air_pressure = concatenate(air_pressure)
# all_longitude = concatenate(longitude)
# all_latitude = concatenate(latitude)
# all_time = concatenate(time)
# all_data_1 = concatenate(data_1)
# all_data_2 = concatenate(data_2)
# all_data_3 = concatenate(data_3)
# all_data_4 = concatenate(data_4)
# all_data_5 = concatenate(data_5)
#
#
# out = Dataset("abcdefghijklmnop", mode='w')
# dim = out.createDimension('obs')
#
# ap_var = out.createVariable('air_pressure', 'f4', dimensions='obs')
# lon_var = out.createVariable('longitude', 'f4', dimensions='obs')
# lat_var = out.createVariable('latitude', 'f4', dimensions='obs')
# t_var = out.createVariable('time', 'f4', dimensions='obs')
# dat_var_1 = out.createVariable('N3', 'f4', dimensions='obs')
# dat_var_2 = out.createVariable('N10', 'f4', dimensions='obs')
# dat_var_3 = out.createVariable('N15_NONVOL', 'f4', dimensions='obs')
# dat_var_4 = out.createVariable('N15_VOLATILE', 'f4', dimensions='obs')
# dat_var_5 = out.createVariable('N3_10', 'f4', dimensions='obs')
#
# ap_var[:] = all_air_pressure
# lon_var[:] = all_longitude
# lat_var[:] = all_latitude
# t_var[:] = all_time
# dat_var_1[:] = dat_var_1
# dat_var_2[:] = dat_var_2
# dat_var_3[:] = dat_var_3
# dat_var_4[:] = dat_var_4
# dat_var_5[:] = dat_var_5
#
#
# out.close()