Ejemplo n.º 1
0
    Ns = np.array([2, 2, 2])  # np.ones(K) * 2

    multiuserchannel = MultiUserChannelMatrix()
    modulator = PSK(4)

    SNR = 40
    noise_var = 1 / dB2Linear(SNR)
    print("SNR: {0}".format(SNR))
    print("noise_var: {0}".format(noise_var))

    multiuserchannel.randomize(Nr, Nt, K)
    multiuserchannel.noise_var = noise_var

    ia_solver = algorithms.AlternatingMinIASolver(multiuserchannel)
    ia_solver2 = algorithms.MMSEIASolver(multiuserchannel)
    ia_solver3 = algorithms.MaxSinrIASolver(multiuserchannel)

    ia_solver4 = algorithms.AlternatingMinIASolver(multiuserchannel)

    # ia_solver.initialize_with_closed_form = True
    # ia_solver2.initialize_with_closed_form = True
    # ia_solver3.initialize_with_closed_form = True

    ia_solver.randomizeF(Ns)
    ia_solver.max_iterations = 400
    ia_solver.solve(Ns)

    ia_solver2.randomizeF(Ns)
    ia_solver2.max_iterations = 100
    ia_solver2.solve(Ns)
Ejemplo n.º 2
0
def main():
    """Main function.
    """
    K = 3
    Nr = 4
    Nt = 4
    Ns = 2
    SNR = 30.0
    P = 1.0

    # Dependent parameters
    noise_var = 1 / dB2Linear(SNR)

    RepMax = 1
    mmse_sinrs = np.empty([RepMax, K, Ns], dtype=float)
    max_sinr_sinrs = np.empty([RepMax, K, Ns], dtype=float)
    mmse_capacity = np.empty(RepMax, dtype=float)
    max_sinr_capacity = np.empty(RepMax, dtype=float)

    pbar = ProgressbarText(RepMax, message="Simulating for SNR: {0}".format(SNR))

    for rep in range(RepMax):
        # Creat the channel
        multiUserChannel = pyphysim.channels.multiuser.MultiUserChannelMatrix()
        multiUserChannel.randomize(Nr, Nt, K)
        multiUserChannel.noise_var = noise_var

        # Creat the IA solver object
        mmse_ia_solver = algorithms.MMSEIASolver(multiUserChannel)
        max_sinr_ia_solver = algorithms.MaxSinrIASolver(multiUserChannel)

        mmse_ia_solver.randomizeF(Ns, P)

        mmse_ia_solver.initialize_with = 'fix'
        max_sinr_ia_solver.initialize_with = 'fix'
        # noinspection PyProtectedMember
        max_sinr_ia_solver._F = mmse_ia_solver._F

        #mmse_ia_solver.initialize_with = 'fix'

        # We wouldn't need to explicitly set ia_solver.noise_var
        # variable if the multiUserChannel object had the correct value at
        # this point.
        # mmse_ia_solver.noise_var = noise_var
        mmse_ia_solver.max_iterations = 200
        mmse_ia_solver.solve(Ns)

        # max_sinr_ia_solver.noise_var = noise_var
        max_sinr_ia_solver.max_iterations = 200

        max_sinr_ia_solver.solve(Ns)

        mmse_sinrs[rep] = list(map(linear2dB, mmse_ia_solver.calc_SINR()))
        max_sinr_sinrs[rep] = list(map(linear2dB, max_sinr_ia_solver.calc_SINR()))

        mmse_capacity[rep] = np.sum(calc_capacity(mmse_ia_solver.calc_SINR()))
        max_sinr_capacity[rep] = np.sum(calc_capacity(max_sinr_ia_solver.calc_SINR()))

        # print "MMSE Alt. SINRs:\n{0}".format(np.vstack(mmse_sinrs[rep]))
        # print "Max SINR Alg. SINRs:\n{0}".format(np.vstack(max_sinr_sinrs[rep]))

        # print "MMSE Alt. Capacity: {0}".format(np.sum(calc_capacity(mmse_sinrs[rep])))
        # print "Max SINR Alg. Capacity: {0}".format(np.sum(calc_capacity(max_sinr_sinrs[rep])))
        # print

        pbar.progress(rep)

    print("MMSE Average SINRs:\n{0}".format(mmse_sinrs.mean(0)))
    print("Max SINR Average SINRs:\n{0}".format(max_sinr_sinrs.mean(0)))
    print("MMSE Average Capacity: {0}".format(mmse_capacity.mean()))
    print("Max SINR Average Capacity: {0}".format(max_sinr_capacity.mean()))

    print("\nEnd!")
Ejemplo n.º 3
0
    noise_var = 1 / dB2Linear(SNR)
    M = 2
    NSymbs = 50
    rep_max = 300
    modulator = fundamental.BPSK()
    K = 3
    Nr = np.ones(K, dtype=int) * 2
    Nt = np.ones(K, dtype=int) * 2
    Ns = np.ones(K, dtype=int) * 1
    multi_user_channel = pyphysim.channels.multiuser.MultiUserChannelMatrix()
    multi_user_channel_quant = pyphysim.channels.multiuser.MultiUserChannelMatrix(
    )
    multi_user_channel.noise_var = noise_var
    multi_user_channel_quant.noise_var = noise_var

    ia_solver = algorithms.MaxSinrIASolver(multi_user_channel_quant)
    ia_solver.max_iterations = 120

    pb = ProgressbarText2(rep_max,
                          '*',
                          message="SNR {0} - {{elapsed_time}}".format(SNR))

    symbolErrors = 0
    bitErrors = 0
    numSymbols = 0
    numBits = 0

    # xxxxxxxxxx Quantization paramters xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx
    codebook_size = 512
    dimension = 4
    codebook = gen_codebook(codebook_size, dimension)