d46293f3ee0885a8c766a652e12eaa156bd23abc
[nepi.git] / src / nepi / resources / ns3 / ns3errormodel.py
1 #
2 #    NEPI, a framework to manage network experiments
3 #    Copyright (C) 2014 INRIA
4 #
5 #    This program is free software: you can redistribute it and/or modify
6 #    it under the terms of the GNU General Public License as published by
7 #    the Free Software Foundation, either version 3 of the License, or
8 #    (at your option) any later version.
9 #
10 #    This program is distributed in the hope that it will be useful,
11 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
12 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13 #    GNU General Public License for more details.
14 #
15 #    You should have received a copy of the GNU General Public License
16 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
17 #
18 # Author: Alina Quereilhac <alina.quereilhac@inria.fr>
19
20 from nepi.execution.resource import clsinit_copy
21 from nepi.resources.ns3.ns3base import NS3Base
22
23 @clsinit_copy
24 class NS3BaseErrorModel(NS3Base):
25     _rtype = "abstract::ns3::ErrorModel"
26
27     @property
28     def device(self):
29         from nepi.resources.ns3.ns3netdevice import NS3BaseNetDevice
30         devices = self.get_connected(NS3BaseNetDevice.get_rtype())
31
32         if not devices: 
33             msg = "ErrorModel not connected to device"
34             self.error(msg)
35             raise RuntimeError, msg
36
37         return devices[0]
38
39     @property
40     def _rms_to_wait(self):
41         rms = set()
42         rms.add(self.device)
43         return rms
44
45     def _connect_object(self):
46         device = self.device
47         if device.uuid not in self.connected:
48             self.simulation.invoke(device.uuid, "SetReceiveErrorModel", self.uuid)
49             self._connected.add(device.uuid)
50