######################################## OFDM setting ###########################################
size_after_compress = (opt.size // (2**opt.n_downsample))**2 * (opt.C_channel // 2)
opt.N = opt.batchSize # Batch size
opt.P = 1 # Number of symbols
opt.M = 64 # Number of subcarriers per symbol
opt.K = 16 # Length of CP
opt.L = 8 # Number of paths
opt.decay = 4
opt.S = size_after_compress // opt.M # Number of packets
opt.is_cfo = False
opt.is_trick = True
opt.is_cfo_random = False
opt.max_ang = 1.7
opt.ang = 1.7
if opt.CE not in ['LS', 'MMSE', 'TRUE', 'IMPLICIT']:
raise Exception("Channel estimation method not implemented")
if opt.EQ not in ['ZF', 'MMSE', 'IMPLICIT']:
raise Exception("Equalization method not implemented")
if opt.feedforward not in ['IMPLICIT', 'EXPLICIT-CE', 'EXPLICIT-CE-EQ', 'EXPLICIT-RES']:
raise Exception("Forward method not implemented")
# Display setting
opt.checkpoints_dir = './Checkpoints/' + opt.dataset_mode + '_OFDM'
opt.name = opt.gan_mode + '_C' + str(opt.C_channel) + '_' + opt.feedforward + '_feed_' + str(opt.is_feedback) + '_clip_' + str(opt.CR) + '_SNR_' + str(opt.SNR)
opt.display_env = opt.dataset_mode + '_OFDM_' + opt.name
