stfpy: Added support to get header comments

Author: furuame

stfpy/stfpy/stf_inst_reader.pxd

@@ -1,9 +1,31 @@
 # distutils: language = c++
 
+from cython.operator cimport dereference
 from stfpy.stf_lib.stf_inst_reader cimport STFInstReader as _STFInstReader
+from stfpy.stf_lib.stf_inst_reader cimport StringVector as _StringVector, StringVectorIterator as _StringVectorIterator
 
 ctypedef _STFInstReader.iterator _STFInstReaderIterator
 
+cdef class StringVectorIterator:
+    cdef _StringVectorIterator c_it
+    cdef _StringVectorIterator c_end_it
+
+    @staticmethod
+    cdef inline StringVectorIterator _construct(const _StringVector* vec):
+        it = StringVectorIterator()
+        it.c_it = dereference(vec).begin()
+        it.c_end_it = dereference(vec).end()
+        return it
+
+cdef class StringVector:
+    cdef const _StringVector* c_vec
+
+    @staticmethod
+    cdef inline StringVector _construct(const _StringVector& vec):
+        new_vec = StringVector()
+        new_vec.c_vec = &vec
+        return new_vec
+
 cdef class STFInstReaderIterator:
     cdef _STFInstReaderIterator c_it
     cdef _STFInstReaderIterator c_end_it

stfpy/stfpy/stf_inst_reader.pyx

@@ -7,6 +7,27 @@ from stfpy.stf_inst import STFInst
 from cython.operator cimport dereference, preincrement
 include "stfpy/stf_lib/stf_reader_constants.pxi"
 
+cdef class StringVectorIterator:
+    def __next__(self):
+        if self.c_it == self.c_end_it:
+            raise StopIteration
+        value = dereference(self.c_it)
+        preincrement(self.c_it)
+        return value
+
+cdef class StringVector:
+    def __iter__(self):
+        return StringVectorIterator._construct(self.c_vec)
+
+    def __len__(self):
+        return dereference(self.c_vec).size()
+
+    def __getitem__(self, idx):
+        return dereference(self.c_vec).at(idx)
+
+    def __bool__(self):
+        return not dereference(self.c_vec).empty()
+
 cdef class STFInstReaderIterator:
     def __next__(self):
         if self.c_it == self.c_end_it:
@@ -24,11 +45,11 @@ cdef class STFInstReader:
                   size_t buffer_size = __DEFAULT_BUFFER_SIZE,
                   bint force_single_threaded_stream = False):
         self.c_reader = new _STFInstReader(filename,
-                                          only_user_mode,
-                                          enable_address_translation,
-                                          filter_mode_change_events,
-                                          buffer_size,
-                                          force_single_threaded_stream)
+                                           only_user_mode,
+                                           enable_address_translation,
+                                           filter_mode_change_events,
+                                           buffer_size,
+                                           force_single_threaded_stream)
 
     def __dealloc__(self):
         del self.c_reader
@@ -44,3 +65,12 @@ cdef class STFInstReader:
 
     def close(self):
         dereference(self.c_reader).close()
+
+    def getMajorVersion(self):
+        return self.c_reader.major()
+
+    def getMinorVersion(self):
+        return self.c_reader.minor()
+
+    def getHeaderComments(self):
+        return StringVector._construct(dereference(self.c_reader).getHeaderCommentsString())

stfpy/stfpy/stf_lib/stf_inst_reader.pxd

@@ -1,7 +1,13 @@
 # distutils: language = c++
 
+from libc.stdint cimport *
+from libcpp.vector cimport vector
+from cython.cimports.libcpp.string import string
 from stfpy.stf_lib.stf_inst cimport STFInst
 
+ctypedef vector[string] StringVector
+ctypedef vector[string].const_iterator StringVectorIterator
+
 cdef extern from "stf_inst_reader.hpp" namespace "stf":
     cdef cppclass STFInstReader:
         cppclass iterator:
@@ -17,3 +23,6 @@ cdef extern from "stf_inst_reader.hpp" namespace "stf":
         void close()
         iterator begin()
         iterator end()
+        uint32_t major()
+        uint32_t minor()
+        const StringVector& getHeaderCommentsString()