feat: get the number of obj and req

Author: haochengxia

libCacheSim/include/python/api.h

@@ -59,6 +59,22 @@ LIBCACHESIM_C_EXPORT int LCS_ReaderFree(const ReaderHandle reader);
  */
 LIBCACHESIM_C_EXPORT int LCS_ReaderGetNextRequest(const ReaderHandle reader, RequestHandle* out);
 
+/*!
+ * \brief Get the number of requests in the reader.
+ * \param reader The reader
+ * \param[out] out The number of requests
+ * \return 0 when succeed, -1 when failure happens
+ */
+LIBCACHESIM_C_EXPORT int LCS_ReaderGetReqNum(const ReaderHandle reader, int64_t* out);
+
+/*!
+ * \brief Get the number of objects in the reader.
+ * \param reader The reader
+ * \param[out] out The number of objects
+ * \return 0 when succeed, -1 when failure happens
+ */
+LIBCACHESIM_C_EXPORT int LCS_ReaderGetObjNum(const ReaderHandle reader, int64_t* out);
+
 /*!
  * \brief Free a request.
  * \param request The request

libCacheSim/python/api.cpp

@@ -86,18 +86,17 @@ char* _extract_trace_type_str(const char* trace_path) {
   }
 }
 
-int LCS_ReaderCreateFromFile(const char* filename, const char* parameters,
-                             ReaderHandle* out) {
+int LCS_ReaderCreateFromFile(const char* filename, const char* parameters, ReaderHandle* out) {
   API_BEGIN();
   // TODO(haocheng): add more parameters
   // initialize reader by filename
   trace_type_e trace_type = trace_type_str_to_enum(_extract_trace_type_str(filename), filename);
   reader_t* reader = setup_reader(filename, trace_type, NULL);
   if (reader == NULL) {
-      printf("Failed to create reader\n");
+    printf("Failed to create reader\n");
   }
   *out = (ReaderHandle)reader;
-  API_END();  
+  API_END();
 }
 
 int LCS_ReaderFree(const ReaderHandle reader) {
@@ -121,6 +120,20 @@ int LCS_ReaderGetNextRequest(const ReaderHandle reader, RequestHandle* out) {
   API_END();
 }
 
+int LCS_ReaderGetObjNum(const ReaderHandle reader, int64_t* out) {
+  API_BEGIN();
+  int64_t wss_obj = 0, wss_byte = 0;
+  cal_working_set_size(static_cast<reader_t*>(reader), &wss_obj, &wss_byte);
+  *out = wss_obj;
+  API_END();
+}
+
+int LCS_ReaderGetReqNum(const ReaderHandle reader, int64_t* out) {
+  API_BEGIN();
+  *out = (int64_t)(static_cast<reader_t*>(reader)->n_total_req);
+  API_END();
+}
+
 int LCS_CacheCreate(const ReaderHandle reader, const char* eviction_algo, float cache_size, bool ignore_obj_size,
                     const char* eviction_parameters, CacheHandle* out) {
   API_BEGIN();

python-package/example_cache.py

@@ -1,6 +1,9 @@
 from libcachesim import simulate, FIFO, Reader
 
-reader = Reader(b"/proj/cache-PG0/haocheng/libCacheSim/data/cloudPhysicsIO.vscsi")
+reader = Reader(b"../data/cloudPhysicsIO.vscsi")
+print("req num: ", reader.get_req_num())
+print("obj num: ", reader.get_obj_num())
+
 cache = FIFO(reader, cache_size=0.1, ignore_obj_size=False)
 
 hit_cnt = 0

python-package/example_sim.py

@@ -1,5 +1,5 @@
 from libcachesim import simulate
 
-req, miss = simulate(b"/proj/cache-PG0/haocheng/libCacheSim/data/cloudPhysicsIO.vscsi", b"vscsi", b"fifo", 0.1)
+req, miss = simulate(b"../data/cloudPhysicsIO.vscsi", b"vscsi", b"fifo", 0.1)
 print(req, miss)
 print(miss / req)

python-package/libcachesim/basic.py

@@ -96,6 +96,16 @@ def read(self) -> "Request":
             raise StopIteration
         return Request(req_handle)
 
+    def get_req_num(self) -> int:
+        req_num = ctypes.c_int64(0)
+        _safe_call(_LIB.LCS_ReaderGetReqNum(self._handle, ctypes.byref(req_num)))
+        return req_num.value
+    
+    def get_obj_num(self) -> int:
+        obj_num = ctypes.c_int64(0)
+        _safe_call(_LIB.LCS_ReaderGetObjNum(self._handle, ctypes.byref(obj_num)))
+        return obj_num.value
+    
     def __iter__(self):
         while True:
             try: