#!/usr/bin/python3

from run_experiments import run_exp

LOG_FOLDER = "Experiments/"

NUM_EPOCHS = 10
PRIV_OUT = 1
PRIV_IN = 1
PUB_OUT = 1
PUB_IN = 1
# B = message size (bytes)
B = 256

## Figure 5 Public
PRIVATE_ROUTE = False
N = [1<<15, 1<<16, 1<<17, 1<<18, 1<<19, 1<<20]
M = [4]
T = [4]

run_exp(LOG_FOLDER, PRIVATE_ROUTE, NUM_EPOCHS, N, M, T, B, PRIV_OUT, PRIV_IN, PUB_OUT, PUB_IN)

## Figure 6 Public
PRIVATE_ROUTE = False
N = [1<<20]
M = [72, 64, 48, 36, 32, 24, 16, 8, 6, 4]
T = [1]

run_exp(LOG_FOLDER, PRIVATE_ROUTE, NUM_EPOCHS, N, M, T, B, PRIV_OUT, PRIV_IN, PUB_OUT, PUB_IN)

## Figure 5 Private
PRIVATE_ROUTE = True
N = [1<<15, 1<<16, 1<<17, 1<<18, 1<<19, 1<<20]
M = [4]
T = [4]

run_exp(LOG_FOLDER, PRIVATE_ROUTE, NUM_EPOCHS, N, M, T, B, PRIV_OUT, PRIV_IN, PUB_OUT, PUB_IN)

## Figure 6 Private
PRIVATE_ROUTE = True
N = [1<<20]
M = [72, 64, 48, 36, 32, 24, 16, 8, 6, 4]
T = [1]

run_exp(LOG_FOLDER, PRIVATE_ROUTE, NUM_EPOCHS, N, M, T, B, PRIV_OUT, PRIV_IN, PUB_OUT, PUB_IN)