apps/simple_precoded_srs.ipynb

{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "from bokeh.plotting import output_notebook\n",
    "output_notebook()"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": true
   },
   "outputs": [],
   "source": [
    "from simple_precoded_srs import *\n",
    "from pyphysim.reference_signals.zadoffchu import get_shifted_root_seq"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Scenario Description xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# 3 Base Stations, each sending data to its own user while interfering\n",
    "# with the other users.\n",
    "\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Configuration xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "num_prbs = 25  # Number of PRBs to simulate\n",
    "Nsc = 12 * num_prbs  # Number of subcarriers\n",
    "Nzc = 149  # Size of the sequence\n",
    "u1 = 1  # Root sequence index of the first user\n",
    "u2 = 2  # Root sequence index of the first user\n",
    "u3 = 3  # Root sequence index of the first user\n",
    "numAnAnt = 4  # Number of Base station antennas\n",
    "numUeAnt = 2  # Number of UE antennas\n",
    "\n",
    "num_samples = 1  # Number of simulated channel samples (from\n",
    "# Jakes process)\n",
    "\n",
    "# Channel configuration\n",
    "speedTerminal = 0 / 3.6  # Speed in m/s\n",
    "fcDbl = 2.6e9  # Central carrier frequency (in Hz)\n",
    "timeTTIDbl = 1e-3  # Time of a single TTI\n",
    "subcarrierBandDbl = 15e3  # Subcarrier bandwidth (in Hz)\n",
    "numOfSubcarriersPRBInt = 12  # Number of subcarriers in each PRB\n",
    "L = 16  # The number of rays for the Jakes model.\n",
    "\n",
    "# Dependent parameters\n",
    "lambdaDbl = 3e8 / fcDbl  # Carrier wave length\n",
    "Fd = speedTerminal / lambdaDbl  # Doppler Frequency\n",
    "Ts = 1. / (Nsc * subcarrierBandDbl)  # Sampling time\n",
    "\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Generate the root sequence xxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "a_u1 = get_extended_ZF(calcBaseZC(Nzc, u1), Nsc / 2)\n",
    "a_u2 = get_extended_ZF(calcBaseZC(Nzc, u2), Nsc / 2)\n",
    "a_u3 = get_extended_ZF(calcBaseZC(Nzc, u3), Nsc / 2)\n",
    "\n",
    "print(\"Nsc: {0}\".format(Nsc))\n",
    "print(\"a_u.shape: {0}\".format(a_u1.shape))\n",
    "\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Create shifted sequences for 3 users xxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# We arbitrarely choose some cyclic shift index and then we call\n",
    "# zadoffchu.getShiftedZF to get the shifted sequence.\n",
    "shift_index = 4\n",
    "r1 = get_shifted_root_seq(a_u1, shift_index, 8)\n",
    "r2 = get_shifted_root_seq(a_u2, shift_index, 8)\n",
    "r3 = get_shifted_root_seq(a_u3, shift_index, 8)\n",
    "\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Generate channels from users to the BS xxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "jakes_all_links = np.empty([3, 3], dtype=object)\n",
    "tdlchannels_all_links = np.empty([3, 3], dtype=object)\n",
    "impulse_responses = np.empty([3, 3], dtype=object)\n",
    "# Dimension: `UEs x ANs x num_subcarriers x numUeAnt x numAnAnt`\n",
    "freq_responses = np.empty([3, 3, Nsc, numUeAnt, numAnAnt], dtype=complex)\n",
    "\n",
    "for ueIdx in range(3):\n",
    "    for anIdx in range(3):\n",
    "        jakes_all_links[ueIdx, anIdx] = JakesSampleGenerator(\n",
    "            Fd, Ts, L, shape=(numUeAnt, numAnAnt))\n",
    "\n",
    "        tdlchannels_all_links[ueIdx, anIdx] = TdlChannel(\n",
    "            jakes_all_links[ueIdx, anIdx],\n",
    "            tap_powers_dB=COST259_TUx.tap_powers_dB,\n",
    "            tap_delays=COST259_TUx.tap_delays)\n",
    "\n",
    "        tdlchannels_all_links[ueIdx, anIdx]._generate_impulse_response(\n",
    "            num_samples)\n",
    "\n",
    "        impulse_responses[ueIdx, anIdx] \\\n",
    "            = tdlchannels_all_links[ueIdx, anIdx].get_last_impulse_response()\n",
    "\n",
    "        freq_responses[ueIdx, anIdx] = \\\n",
    "            impulse_responses[ueIdx, anIdx].get_freq_response(Nsc)[:, :, :, 0]\n",
    "\n",
    "# xxxxxxxxxx Channels in downlink direction xxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# Dimension: `Nsc x numUeAnt x numAnAnt`\n",
    "dH11 = freq_responses[0, 0]\n",
    "dH12 = freq_responses[0, 1]\n",
    "dH13 = freq_responses[0, 2]\n",
    "dH21 = freq_responses[1, 0]\n",
    "dH22 = freq_responses[1, 1]\n",
    "dH23 = freq_responses[1, 2]\n",
    "dH31 = freq_responses[2, 0]\n",
    "dH32 = freq_responses[2, 1]\n",
    "dH33 = freq_responses[2, 2]\n",
    "\n",
    "# xxxxxxxxxx Principal dimension in downlink direction xxxxxxxxxxxxxxxx\n",
    "sc_idx = 124  # Index of the subcarrier we are interested in\n",
    "[dU11, dS11, dV11_H] = np.linalg.svd(dH11[sc_idx])\n",
    "[dU22, dS22, dV22_H] = np.linalg.svd(dH22[sc_idx])\n",
    "[dU33, dS33, dV33_H] = np.linalg.svd(dH33[sc_idx])\n",
    "\n",
    "# xxxxxxxxxx Users precoders xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# Users' precoders are the main column of the U matrix\n",
    "F11 = dU11[:, 0].conj()\n",
    "F22 = dU22[:, 0].conj()\n",
    "F33 = dU33[:, 0].conj()\n",
    "\n",
    "# xxxxxxxxxx Channels in uplink direction xxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "int_path_loss = 0.05  # Path loss of the interfering links\n",
    "int_g = math.sqrt(int_path_loss)  # Gain of interfering links\n",
    "dir_d = 1.0                   #  Gain of direct links\n",
    "\n",
    "pl = np.array([[  2.21e-08,   2.14e-09,   1.88e-08],\n",
    "               [  3.45e-10,   2.17e-08,   4.53e-10],\n",
    "               [  4.38e-10,   8.04e-10,   4.75e-08]])\n",
    "# pl = np.array([[  1,   0.1,   0.1],\n",
    "#                [  0.1,   1,   0.1],\n",
    "#                [  0.1,   0.1,   1]])\n",
    "\n",
    "\n",
    "# Dimension: `Nsc x numAnAnt x numUeAnt`\n",
    "uH11 = math.sqrt(pl[0, 0]) * np.transpose(dH11, axes=[0, 2, 1])\n",
    "uH12 = math.sqrt(pl[0, 1]) * np.transpose(dH12, axes=[0, 2, 1])\n",
    "uH13 = math.sqrt(pl[0, 2]) * np.transpose(dH13, axes=[0, 2, 1])\n",
    "uH21 = math.sqrt(pl[1, 0]) * np.transpose(dH21, axes=[0, 2, 1])\n",
    "uH22 = math.sqrt(pl[1, 1]) * np.transpose(dH22, axes=[0, 2, 1])\n",
    "uH23 = math.sqrt(pl[1, 2]) * np.transpose(dH23, axes=[0, 2, 1])\n",
    "uH31 = math.sqrt(pl[2, 0]) * np.transpose(dH31, axes=[0, 2, 1])\n",
    "uH32 = math.sqrt(pl[2, 1]) * np.transpose(dH32, axes=[0, 2, 1])\n",
    "uH33 = math.sqrt(pl[2, 2]) * np.transpose(dH33, axes=[0, 2, 1])\n",
    "\n",
    "# Compute the equivalent uplink channels\n",
    "uH11_eq = uH11.dot(F11)\n",
    "uH12_eq = uH12.dot(F22)\n",
    "uH13_eq = uH13.dot(F33)\n",
    "uH21_eq = uH21.dot(F11)\n",
    "uH22_eq = uH22.dot(F22)\n",
    "uH23_eq = uH23.dot(F33)\n",
    "uH31_eq = uH31.dot(F11)\n",
    "uH32_eq = uH32.dot(F22)\n",
    "uH33_eq = uH33.dot(F33)\n",
    "\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Compute Received Signals xxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# Calculate the received signals\n",
    "comb_indexes = np.r_[0:Nsc:2]\n",
    "Y1_term11 = uH11_eq[comb_indexes] * r1[:, np.newaxis]\n",
    "Y1_term12 = uH12_eq[comb_indexes] * r2[:, np.newaxis]\n",
    "Y1_term13 = uH13_eq[comb_indexes] * r3[:, np.newaxis]\n",
    "Y1 = Y1_term11 + Y1_term12 + Y1_term13\n",
    "\n",
    "Y2_term21 = uH21_eq[comb_indexes] * r1[:, np.newaxis]\n",
    "Y2_term22 = uH22_eq[comb_indexes] * r2[:, np.newaxis]\n",
    "Y2_term23 = uH23_eq[comb_indexes] * r3[:, np.newaxis]\n",
    "Y2 = Y2_term21 + Y2_term22 + Y2_term23\n",
    "\n",
    "Y3_term31 = uH31_eq[comb_indexes] * r1[:, np.newaxis]\n",
    "Y3_term32 = uH32_eq[comb_indexes] * r2[:, np.newaxis]\n",
    "Y3_term33 = uH33_eq[comb_indexes] * r3[:, np.newaxis]\n",
    "Y3 = Y3_term31 + Y3_term32 + Y3_term33"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Estimate the equivalent channel xxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "(uH11_eq_est, uH12_eq_est, uH13_eq_est, uH21_eq_est, uH22_eq_est,\n",
    "     uH23_eq_est, uH31_eq_est, uH32_eq_est, uH33_eq_est\n",
    "    ) = estimate_channels_remove_only_direct(Y1, Y2, Y3, r1, r2, r3, Nsc, comb_indexes)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false,
    "scrolled": false
   },
   "outputs": [],
   "source": [
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxx Plot the true and estimated channels xxxxxxxxxxxxxxxx\n",
    "# xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx\n",
    "#session = bp.Session(load_from_config=False)\n",
    "#bp.output_server(docname='simple_precoded_srs_AN1', session=session)\n",
    "\n",
    "p1 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH11_eq, uH11_eq_est,\n",
    "    title='Direct Channel from UE1 to AN1')\n",
    "#bp.show(p1)\n",
    "\n",
    "p2 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH12_eq, uH12_eq_est,\n",
    "    title='Interfering Channel from UE2 to AN1')\n",
    "#bp.show(p2)\n",
    "\n",
    "p3 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH13_eq, uH13_eq_est,\n",
    "    title='Interfering Channel from UE3 to AN1')\n",
    "#bp.show(p3)\n",
    "\n",
    "tab1 = bw.Panel(child=p1, title=\"UE1 to AN1\")\n",
    "tab2 = bw.Panel(child=p2, title=\"UE2 to AN1\")\n",
    "tab3 = bw.Panel(child=p3, title=\"UE3 to AN1\")\n",
    "tabs_an1 = bw.Tabs(tabs=[tab1, tab2, tab3])\n",
    "#bp.show(tabs_an1)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#bp.output_server(docname='simple_precoded_srs_AN2', session=session)\n",
    "p1 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH21_eq, uH21_eq_est,\n",
    "    title='Interfering Channel from UE2 to AN2')\n",
    "#bp.show(p1)\n",
    "\n",
    "p2 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH22_eq, uH22_eq_est,\n",
    "    title='Direct Channel from UE2 to AN2')\n",
    "#bp.show(p2)\n",
    "\n",
    "p3 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH23_eq, uH23_eq_est,\n",
    "    title='Interfering Channel from UE3 to AN2')\n",
    "#bp.show(p3)\n",
    "\n",
    "tab1 = bw.Panel(child=p1, title=\"UE1 to AN2\")\n",
    "tab2 = bw.Panel(child=p2, title=\"UE2 to AN2\")\n",
    "tab3 = bw.Panel(child=p3, title=\"UE3 to AN2\")\n",
    "tabs_an2 = bw.Tabs(tabs=[tab1, tab2, tab3])\n",
    "#bp.show(tabs_an2)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "#bp.output_server(docname='simple_precoded_srs_AN3', session=session)\n",
    "p1 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH31_eq, uH31_eq_est,\n",
    "    title='Interfering Channel from UE1 to AN3')\n",
    "p2 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH32_eq, uH32_eq_est,\n",
    "    title='Interfering Channel from UE2 to AN3')\n",
    "p3 = plot_true_and_estimated_channel_with_bokeh_all_antennas(\n",
    "    uH33_eq, uH33_eq_est,\n",
    "    title='Direct Channel from UE3 to AN3')\n",
    "#bp.show(p3)\n",
    "\n",
    "tab1 = bw.Panel(child=p1, title=\"UE1 to AN3\")\n",
    "tab2 = bw.Panel(child=p2, title=\"UE2 to AN3\")\n",
    "tab3 = bw.Panel(child=p3, title=\"UE3 to AN3\")\n",
    "tabs_an3 = bw.Tabs(tabs=[tab1, tab2, tab3])\n",
    "#bp.show(tabs_an3)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": [
    "tabs1 = bw.Panel(child=tabs_an1, title=\"AN1\")\n",
    "tabs2 = bw.Panel(child=tabs_an2, title=\"AN2\")\n",
    "tabs3 = bw.Panel(child=tabs_an3, title=\"AN3\")\n",
    "tabs_all = bw.Tabs(tabs=[tabs1, tabs2, tabs3])\n",
    "bp.show(tabs_all)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {
    "collapsed": false
   },
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.5.0+"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 0
}