Cyber: Add segmentation args for python record writer.

cyber/py_wrapper/py_record.h

@@ -108,6 +108,18 @@ class PyRecordReader {
 class PyRecordWriter {
  public:
   PyRecordWriter() {}
+
+  PyRecordWriter(const uint64_t file_segmentation_size_kb,
+                 const uint64_t file_segmentation_interval_sec) {
+    if (file_segmentation_size_kb > 0) {
+      recored_writer_.SetSizeOfFileSegmentation(file_segmentation_size_kb);
+    }
+    if (file_segmentation_interval_sec > 0) {
+      recored_writer_.SetIntervalOfFileSegmentation(
+          file_segmentation_interval_sec);
+    }
+  }
+
   ~PyRecordWriter() {}
 
   bool Open(const std::string& path) { return recored_writer_.Open(path); }

cyber/python/cyber_py/record.py

@@ -112,6 +112,11 @@ class RecordWriter(object):
     def __init__(self):
         self.record_writer = _CYBER_RECORD.new_PyRecordWriter()
 
+    def __init__(self, file_segmentation_size_kb,
+                 file_segmentation_interval_sec):
+        self.record_writer = _CYBER_RECORD.new_PyRecordWriter(
+            file_segmentation_size_kb, file_segmentation_interval_sec)
+
     def __del__(self):
         _CYBER_RECORD.delete_PyRecordWriter(self.record_writer)
 

cyber/python/examples/py_record.cc

@@ -59,6 +59,7 @@ void test_read(const std::string &readfile) {
 
   AINFO << "reader msg count = " << count;
 }
+
 int main(int argc, char *argv[]) {
   apollo::cyber::Init("cyber_python");
   test_write(TEST_RECORD_FILE);