X-Git-Url: https://gerrit.opnfv.org/gerrit/gitweb?a=blobdiff_plain;f=nfvbench%2Ftraffic_gen%2Ftraffic_base.py;h=03605918fce7966f34d1b38b644498db5722cf87;hb=40d716a6fca17fcbba066769b503f8aeeea400dc;hp=81537b32721000a0a43a1a124335286fa227a683;hpb=5d582aada32e1c5f43601015660b7936013f5b30;p=nfvbench.git

diff --git a/nfvbench/traffic_gen/traffic_base.py b/nfvbench/traffic_gen/traffic_base.py
index 81537b3..0360591 100644
--- a/nfvbench/traffic_gen/traffic_base.py
+++ b/nfvbench/traffic_gen/traffic_base.py
@@ -13,30 +13,52 @@
 #    under the License.
 
 import abc
+import sys
 
 from nfvbench.log import LOG
 import traffic_utils
 
 
+class Latency(object):
+    """A class to hold latency data."""
+
+    def __init__(self, latency_list=None):
+        """Create a latency instance.
+
+        latency_list: aggregate all latency values from list if not None
+        """
+        self.min_usec = sys.maxint
+        self.max_usec = 0
+        self.avg_usec = 0
+        if latency_list:
+            for lat in latency_list:
+                if lat.available():
+                    self.min_usec = min(self.min_usec, lat.min_usec)
+                    self.max_usec = max(self.max_usec, lat.max_usec)
+                    self.avg_usec += lat.avg_usec
+            # round to nearest usec
+            self.avg_usec = int(round(float(self.avg_usec) / len(latency_list)))
+
+    def available(self):
+        """Return True if latency information is available."""
+        return self.min_usec != sys.maxint
+
+
 class TrafficGeneratorException(Exception):
+    """Exception for traffic generator."""
+
     pass
 
 
 class AbstractTrafficGenerator(object):
-    def __init__(self, config):
-        self.config = config
-        self.imix_l2_sizes = [64, 594, 1518]
-        self.imix_ratios = [7, 4, 1]
-        self.imix_avg_l2_size = 0
-        self.adjust_imix_min_size(64)
 
-    @abc.abstractmethod
-    def get_version(self):
-        # Must be implemented by sub classes
-        return None
+    def __init__(self, traffic_client):
+        self.traffic_client = traffic_client
+        self.generator_config = traffic_client.generator_config
+        self.config = traffic_client.config
 
     @abc.abstractmethod
-    def init(self):
+    def get_version(self):
         # Must be implemented by sub classes
         return None
 
@@ -45,36 +67,27 @@ class AbstractTrafficGenerator(object):
         # Must be implemented by sub classes
         return None
 
-    @abc.abstractmethod
-    def config_interface(self):
-        # Must be implemented by sub classes
-        return None
-
     @abc.abstractmethod
     def create_traffic(self, l2frame_size, rates, bidirectional, latency=True):
         # Must be implemented by sub classes
         return None
 
     def modify_rate(self, rate, reverse):
+        """Change the rate per port.
+
+        rate: new rate in % (0 to 100)
+        reverse: 0 for port 0, 1 for port 1
+        """
         port_index = int(reverse)
         port = self.port_handle[port_index]
         self.rates[port_index] = traffic_utils.to_rate_str(rate)
-        LOG.info('Modified traffic stream for %s, new rate=%s.', port,
-                 traffic_utils.to_rate_str(rate))
-
-    def modify_traffic(self):
-        # Must be implemented by sub classes
-        return None
+        LOG.info('Modified traffic stream for port %s, new rate=%s.', port, self.rates[port_index])
 
     @abc.abstractmethod
     def get_stats(self):
         # Must be implemented by sub classes
         return None
 
-    def clear_traffic(self):
-        # Must be implemented by sub classes
-        return None
-
     @abc.abstractmethod
     def start_traffic(self):
         # Must be implemented by sub classes
@@ -87,13 +100,35 @@ class AbstractTrafficGenerator(object):
 
     @abc.abstractmethod
     def cleanup(self):
-        # Must be implemented by sub classes
+        """Cleanup the traffic generator."""
         return None
 
-    def adjust_imix_min_size(self, min_size):
-        # assume the min size is always the first entry
-        self.imix_l2_sizes[0] = min_size
-        self.imix_avg_l2_size = sum(
-            [1.0 * imix[0] * imix[1] for imix in zip(self.imix_l2_sizes, self.imix_ratios)]) / sum(
-                self.imix_ratios)
-        traffic_utils.imix_avg_l2_size = self.imix_avg_l2_size
+    def clear_streamblock(self):
+        """Clear all streams from the traffic generator."""
+        pass
+
+    @abc.abstractmethod
+    def resolve_arp(self):
+        """Resolve all configured remote IP addresses.
+
+        return: None if ARP failed to resolve for all IP addresses
+                else a dict of list of dest macs indexed by port#
+                the dest macs in the list are indexed by the chain id
+        """
+        pass
+
+    @abc.abstractmethod
+    def get_macs(self):
+        """Return the local port MAC addresses.
+
+        return: a list of MAC addresses indexed by the port#
+        """
+        pass
+
+    @abc.abstractmethod
+    def get_port_speed_gbps(self):
+        """Return the local port speeds.
+
+        return: a list of speed in Gbps indexed by the port#
+        """
+        pass