#!/usr/bin/env python3

from jinja2 import Template
import os

# Values: 6 (1.4 MHz), 15 (3MHz), 25 (5MHz), 50 (10MHz), 75 (15MHz), 100 (20MHz)
LTE_N_RB_DL=100
LTE_TX_GAIN=78
LTE_RX_GAIN=40
LTE_IMSI="001010123456789"
LTE_K="00112233445566778899aabbccddeeff"

NR_BANDWIDTH=40
NR_TX_GAIN=78
NR_RX_GAIN=40
NR_IMSI="001010123456789"
NR_K="00112233445566778899aabbccddeeff"

N_ANTENNA_DL=2
N_ANTENNA_UL=2

global_context = {
        'generated_file_message': "This file was generated using a jinja2 template and the render-templates script, don't modify directly."
}

rf_mode_context_list = [
        {
            'rf_mode': 'tdd1900',

            'default_dl_earfcn': 38350,
            'default_lte_dl_freq': 1890.0,
            'default_lte_band': 39,
            'default_lte_n_rb_dl': LTE_N_RB_DL,
            'default_lte_tx_gain': LTE_TX_GAIN,
            'default_lte_rx_gain': LTE_RX_GAIN,
            'default_lte_imsi': LTE_IMSI,
            'default_lte_k': LTE_K,

            'default_dl_nr_arfcn': 378000,
            'default_nr_band': 39,
            'default_nr_dl_freq': 1890.0,
            'default_nr_bandwidth': NR_BANDWIDTH,
            'default_nr_tx_gain': NR_TX_GAIN,
            'default_nr_rx_gain': NR_RX_GAIN,
            'default_nr_ssb_pos_bitmap': "1000",
            'default_ssb_nr_arfcn': 377790,
            'default_nr_imsi': NR_IMSI,
            'default_nr_k': NR_K,

            'min_frequency': 1880.0,
            'max_frequency': 1920,
            'default_n_antenna_dl': N_ANTENNA_DL,
            'default_n_antenna_ul': N_ANTENNA_UL,
        }, {
            'rf_mode': 'tdd2600',

            'default_dl_earfcn': 38050,
            'default_lte_dl_freq': 2600.0,
            'default_lte_band': 38,
            'default_lte_n_rb_dl': LTE_N_RB_DL,
            'default_lte_tx_gain': LTE_TX_GAIN,
            'default_lte_rx_gain': LTE_RX_GAIN,
            'default_lte_imsi': LTE_IMSI,
            'default_lte_k': LTE_K,

            'default_dl_nr_arfcn': 520000,
            'default_nr_band': 38,
            'default_nr_dl_freq': 2600.0,
            'default_nr_bandwidth': NR_BANDWIDTH,
            'default_nr_tx_gain': NR_TX_GAIN,
            'default_nr_rx_gain': NR_RX_GAIN,
            'default_nr_ssb_pos_bitmap': "10000000",
            'default_ssb_nr_arfcn': 517930,
            'default_nr_imsi': NR_IMSI,
            'default_nr_k': NR_K,

            'min_frequency': 2570,
            'max_frequency': 2620,
            'default_n_antenna_dl': N_ANTENNA_DL,
            'default_n_antenna_ul': N_ANTENNA_UL,
        }, {
            'rf_mode': 'tdd3500',

            'default_dl_earfcn': 42590,
            'default_lte_dl_freq': 3500.0,
            'default_lte_band': 42,
            'default_lte_n_rb_dl': LTE_N_RB_DL,
            'default_lte_tx_gain': LTE_TX_GAIN,
            'default_lte_rx_gain': LTE_RX_GAIN,
            'default_lte_imsi': LTE_IMSI,
            'default_lte_k': LTE_K,

            'default_dl_nr_arfcn': 632628,
            'default_nr_band': 78,
            'default_nr_dl_freq': 3489.42,
            'default_nr_bandwidth': NR_BANDWIDTH,
            'default_nr_tx_gain': NR_TX_GAIN,
            'default_nr_rx_gain': NR_RX_GAIN,
            'default_nr_ssb_pos_bitmap': "10000000",
            'default_ssb_nr_arfcn': 632544,
            'default_nr_imsi': NR_IMSI,
            'default_nr_k': NR_K,

            'min_frequency': 3400,
            'max_frequency': 3600,
            'default_n_antenna_dl': N_ANTENNA_DL,
            'default_n_antenna_ul': N_ANTENNA_UL,
        }, {
            'rf_mode': 'tdd3700',

            'default_dl_earfcn': 44590,
            'default_lte_dl_freq': 3700.0,
            'default_lte_band': 43,
            'default_lte_n_rb_dl': LTE_N_RB_DL,
            'default_lte_tx_gain': LTE_TX_GAIN,
            'default_lte_rx_gain': LTE_RX_GAIN,
            'default_lte_imsi': LTE_IMSI,
            'default_lte_k': LTE_K,

            'default_dl_nr_arfcn': 646666,
            'default_nr_band': 78,
            'default_nr_dl_freq': 3699.99,
            'default_nr_bandwidth': NR_BANDWIDTH,
            'default_nr_tx_gain': NR_TX_GAIN,
            'default_nr_rx_gain': NR_RX_GAIN,
            'default_nr_ssb_pos_bitmap': "10000000",
            'default_ssb_nr_arfcn': 646656,
            'default_nr_imsi': NR_IMSI,
            'default_nr_k': NR_K,

            'min_frequency': 3600,
            'max_frequency': 3800,
            'default_n_antenna_dl': N_ANTENNA_DL,
            'default_n_antenna_ul': N_ANTENNA_UL,
        }
]

with open('software.cfg.jinja2.json', 'r') as f:
    software_json_template = Template(f.read())
with open('instance-enb-input-schema.jinja2.json', 'r') as f:
    instance_enb_json_template = Template(f.read())
with open('instance-gnb-input-schema.jinja2.json', 'r') as f:
    instance_gnb_json_template = Template(f.read())
with open('instance-ue-lte-input-schema.jinja2.json', 'r') as f:
    instance_ue_lte_json_template = Template(f.read())
with open('instance-ue-nr-input-schema.jinja2.json', 'r') as f:
    instance_ue_nr_json_template = Template(f.read())
with open('software.jinja2.cfg', 'r') as f:
    software_template = Template(f.read())
with open('test/test.jinja2.py', 'r') as f:
    test_template = Template(f.read())

for rf_mode_context in rf_mode_context_list:
    with open('software-{}.cfg.json'.format(rf_mode_context['rf_mode']),
              'w+') as f:
        f.write(software_json_template.render(**rf_mode_context, **global_context) + '\n')
    with open('test/test{}.py'.format(rf_mode_context['rf_mode'].upper()),
              'w+') as f:
        f.write(test_template.render(**rf_mode_context, **global_context) + '\n')
    with open('software-{}.cfg'.format(rf_mode_context['rf_mode']),
              'w+') as f:
        f.write(software_template.render(**rf_mode_context, **global_context) + '\n')
    for software_type in ['enb', 'enb-epc', 'gnb', 'gnb-epc', 'ue-lte', 'ue-nr']:
        with open('instance-{}-{}-input-schema.json'.format(
                    rf_mode_context['rf_mode'],
                    software_type),
                  'w+') as f:
            if software_type in ['enb', 'enb-epc']:
                f.write(instance_enb_json_template.render(**rf_mode_context, **global_context, epc='epc' in software_type) + '\n')
            elif software_type in ['gnb', 'gnb-epc']: 
                f.write(instance_gnb_json_template.render(**rf_mode_context, **global_context, epc='epc' in software_type) + '\n')
            elif software_type in ['ue-lte']:
                f.write(instance_ue_lte_json_template.render(**rf_mode_context, **global_context) + '\n')
            else: 
                f.write(instance_ue_nr_json_template.render(**rf_mode_context, **global_context) + '\n')