961b54180ba5a24952dd42671180edfb35fdebd5
[nepi.git] / src / nepi / resources / ns3 / ns3errorratemodel.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 NS3BaseErrorRateModel(NS3Base):
25     _rtype = "abstract::ns3::ErrorRateModel"
26
27     @property
28     def node(self):
29         return self.phy.node
30
31     @property
32     def phy(self):
33         from nepi.resources.ns3.ns3wifiphy import NS3BaseWifiPhy
34         phys = self.get_connected(NS3BaseWifiPhy.get_rtype())
35
36         if not phys: 
37             msg = "ErrorRateModel not connected to phy"
38             self.error(msg)
39             raise RuntimeError, msg
40
41         return phys[0]
42
43     @property
44     def _rms_to_wait(self):
45         rms = set()
46         rms.add(self.phy)
47         return rms
48
49     def _connect_object(self):
50         phy = self.phy
51         if phy.uuid not in self.connected:
52             self.simulation.invoke(phy.uuid, "SetErrorRateModel", self.uuid)
53             self._connected.add(phy.uuid)
54