Repo created

libs/tracking/pytracking/CMakeLists.txt

@@ -0,0 +1,25 @@
+project(pytracking)
+
+set(
+  SRC
+  bindings.cpp
+)
+
+include_directories(${CMAKE_BINARY_DIR})
+
+omim_add_library(${PROJECT_NAME} MODULE ${SRC})
+
+omim_link_libraries(
+    ${PROJECT_NAME} 
+    ${Boost_LIBRARIES}
+    tracking
+    coding
+    geometry
+    base
+)
+
+if (PLATFORM_MAC)
+    omim_link_libraries(${PROJECT_NAME} "-Wl,-undefined,dynamic_lookup")
+endif()
+
+set_target_properties(${PROJECT_NAME} PROPERTIES PREFIX "")

libs/tracking/pytracking/bindings.cpp

@@ -0,0 +1,61 @@
+#include "tracking/protocol.hpp"
+
+#include "coding/traffic.hpp"
+
+#include "pyhelpers/module_version.hpp"
+#include "pyhelpers/pair.hpp"
+#include "pyhelpers/vector_uint8.hpp"
+
+#include <vector>
+
+#include <boost/python.hpp>
+#include <boost/python/suite/indexing/vector_indexing_suite.hpp>
+
+BOOST_PYTHON_MODULE(pytracking)
+{
+  using namespace boost::python;
+  scope().attr("__version__") = PYBINDINGS_VERSION;
+
+  using tracking::Protocol;
+
+  // Register the to-python converters.
+  pair_to_python_converter<Protocol::PacketType, size_t>();
+  to_python_converter<std::vector<uint8_t>, vector_uint8t_to_str>();
+  vector_uint8t_from_python_str();
+
+  class_<Protocol::DataElementsVec>("DataElementsVec").def(vector_indexing_suite<Protocol::DataElementsVec>());
+
+  class_<ms::LatLon>("LatLon").def_readwrite("lat", &ms::LatLon::m_lat).def_readwrite("lon", &ms::LatLon::m_lon);
+
+  class_<coding::TrafficGPSEncoder::DataPoint>("DataPoint")
+      .def(init<uint64_t, ms::LatLon const &, uint8_t>())
+      .def_readwrite("timestamp", &coding::TrafficGPSEncoder::DataPoint::m_timestamp)
+      .def_readwrite("coords", &coding::TrafficGPSEncoder::DataPoint::m_latLon)
+      .def_readwrite("traffic", &coding::TrafficGPSEncoder::DataPoint::m_traffic);
+
+  enum_<Protocol::PacketType>("PacketType")
+      .value("Error", Protocol::PacketType::Error)
+      .value("AuthV0", Protocol::PacketType::AuthV0)
+      .value("DataV0", Protocol::PacketType::DataV0)
+      .value("DataV1", Protocol::PacketType::DataV1)
+      .value("CurrentAuth", Protocol::PacketType::CurrentAuth)
+      .value("CurrentData", Protocol::PacketType::CurrentData);
+
+  std::vector<uint8_t> (*CreateDataPacket1)(Protocol::DataElementsCirc const &, tracking::Protocol::PacketType) =
+      &Protocol::CreateDataPacket;
+  std::vector<uint8_t> (*CreateDataPacket2)(Protocol::DataElementsVec const &, tracking::Protocol::PacketType) =
+      &Protocol::CreateDataPacket;
+
+  class_<Protocol>("Protocol")
+      .def("CreateAuthPacket", &Protocol::CreateAuthPacket)
+      .staticmethod("CreateAuthPacket")
+      .def("CreateDataPacket", CreateDataPacket1)
+      .def("CreateDataPacket", CreateDataPacket2)
+      .staticmethod("CreateDataPacket")
+      .def("CreateHeader", &Protocol::CreateHeader)
+      .staticmethod("CreateHeader")
+      .def("DecodeHeader", &Protocol::DecodeHeader)
+      .staticmethod("DecodeHeader")
+      .def("DecodeDataPacket", &Protocol::DecodeDataPacket)
+      .staticmethod("DecodeDataPacket");
+}

libs/tracking/pytracking/setup.py

@@ -0,0 +1,13 @@
+#!/usr/bin/env python3
+# -*- coding: utf-8 -*-
+import os
+import sys
+module_dir = os.path.abspath(os.path.dirname(__file__))
+sys.path.insert(0, os.path.join(module_dir, '..', '..'))
+
+from pyhelpers.setup import setup_omim_pybinding
+
+
+NAME = "pytracking"
+
+setup_omim_pybinding(name=NAME)