compiler missing files

2 files changed, 149 insertions, 0 deletions

diff --git a/src/stateless/cp/trex_streams_compiler.cpp b/src/stateless/cp/trex_streams_compiler.cpp
new file mode 100644
index 00000000..6c77ad0f
--- /dev/null
+++ b/src/stateless/cp/trex_streams_compiler.cpp
@@ -0,0 +1,81 @@
+/*
+ Itay Marom
+ Cisco Systems, Inc.
+*/
+
+/*
+Copyright (c) 2015-2015 Cisco Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+#include <string.h>
+#include <trex_streams_compiler.h>
+#include <trex_stream.h>
+
+/**************************************
+ * stream compiled object
+ *************************************/
+TrexStreamsCompiledObj::~TrexStreamsCompiledObj() {
+    for (auto &obj : m_objs) {
+        delete obj.m_pkt;
+    }
+    m_objs.clear();
+}
+
+void 
+TrexStreamsCompiledObj::add_compiled_stream(double pps, uint8_t *pkt, uint16_t pkt_len) {
+    obj_st obj;
+
+    obj.m_pps = pps;
+    obj.m_pkt_len = pkt_len;
+
+    obj.m_pkt = new uint8_t[pkt_len];
+    memcpy(obj.m_pkt, pkt, pkt_len);
+
+    m_objs.push_back(obj);
+}
+
+/**************************************
+ * stream compiler
+ *************************************/
+bool 
+TrexStreamsCompiler::compile(const std::vector<TrexStream *> &streams, TrexStreamsCompiledObj &obj) {
+    /* for now we do something trivial, */
+    for (auto stream : streams) {
+
+        /* skip non-enabled streams */
+        if (!stream->m_enabled) {
+            continue;
+        }
+
+        /* for now skip also non self started streams */
+        if (!stream->m_self_start) {
+            continue;
+        }
+
+        /* for now support only continous ... */
+        TrexStreamContinuous *cont_stream = dynamic_cast<TrexStreamContinuous *>(stream);
+        if (!cont_stream) {
+            continue;
+        }
+
+        /* add it */
+        obj.add_compiled_stream(cont_stream->get_pps(),
+                                cont_stream->m_pkt.binary,
+                                cont_stream->m_pkt.len);
+    }
+
+    return true;
+}
+
diff --git a/src/stateless/cp/trex_streams_compiler.h b/src/stateless/cp/trex_streams_compiler.h
new file mode 100644
index 00000000..90253cdf
--- /dev/null
+++ b/src/stateless/cp/trex_streams_compiler.h
@@ -0,0 +1,68 @@
+/*
+ Itay Marom
+ Cisco Systems, Inc.
+*/
+
+/*
+Copyright (c) 2015-2015 Cisco Systems, Inc.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+#ifndef __TREX_STREAMS_COMPILER_H__
+#define __TREX_STREAMS_COMPILER_H__
+
+#include <stdint.h>
+#include <vector>
+
+class TrexStreamsCompiler;
+class TrexStream;
+
+/**
+ * compiled object for a table of streams
+ * 
+ * @author imarom (28-Oct-15)
+ */
+class TrexStreamsCompiledObj {
+    friend class TrexStreamsCompiler;
+public:
+
+    TrexStreamsCompiledObj() {}
+    ~TrexStreamsCompiledObj();
+
+    struct obj_st {
+        double   m_pps;
+        uint8_t *m_pkt;
+        uint16_t m_pkt_len;
+    };
+
+    const std::vector<obj_st> & get_objects() {
+        return m_objs;
+    }
+
+private:
+    void add_compiled_stream(double pps, uint8_t *pkt, uint16_t pkt_len);
+    std::vector<obj_st> m_objs;
+};
+
+class TrexStreamsCompiler {
+public:
+    /**
+     * compiles a vector of streams to an object passable to the DP
+     * 
+     * @author imarom (28-Oct-15)
+     * 
+     */
+    bool compile(const std::vector<TrexStream *> &streams, TrexStreamsCompiledObj &obj);
+};
+
+#endif /* __TREX_STREAMS_COMPILER_H__ */