#!/usr/bin/env python
#
#    NEPI, a framework to manage network experiments
#    Copyright (C) 2015 INRIA
#
#    This program is free software: you can redistribute it and/or modify
#    it under the terms of the GNU General Public License version 2 as
#    published by the Free Software Foundation;
#
#    This program is distributed in the hope that it will be useful,
#    but WITHOUT ANY WARRANTY; without even the implied warranty of
#    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
#    GNU General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
# Author: Damien Saucez <damien.saucez@inria.fr>
#         Alina Quereilhac <alina.quereilhac@inria.fr>
#

#
# Note: To run this experiment you need to have a PlanetLab account.
#
# This experiment consists of a simulated wireless sensor network (ns-3)
# with one fixed access point (AP), running an agent application, and several
# mobile stations that run a transmitter application to send messages to
# the AP.
#
# The same experiment described above is run in parallel with different
# number of mobile stations in 2 PlanetLab hosts.
#

#
# command line:
#
# PYTHONPATH=$PYTHONPATH:src python examples/ns3/multi_host/parallel.py
#

import os

from topology import *

from nepi.execution.ec import ExperimentController
from nepi.execution.resource import ResourceState, ResourceManager

# tunning
os.environ["NEPI_NTHREADS"] = "1"
ResourceManager._reschedule_delay = "0s"

# list of hosts for running the experiment on
hostname1 = "onelab4.warsaw.rd.tp.pl"
hostname2 = "planet2.servers.ua.pt"

(username, pl_user, pl_password, ssh_key, node_count) = get_options()

ec = ExperimentController(exp_id="parallel")
counts = [node_count, 10]
hosts = [hostname1, hostname2]

simulations = []
agents = []

for hostname in hosts:
    host, simu = add_host_simu(ec, hostname, username, pl_user, pl_password, 
            ssh_key)
    simulations.append(simu)

    node_count = counts.pop()
    ap, agent = build_ns3_topology(ec, simu, node_count, network="10.1.0.0", 
            prefixlen="24", agent_ip="10.1.0.1")
    agents.append(agent)

ec.deploy()

ec.wait_finished(simulations)

for agent in agents:
    stdout = ec.trace(agent, "stdout")
    print " Agent says:"
    print stdout

ec.shutdown()