Merge CV, ver=5.1; ops/methods from HV, and interface from CV where possible

[goodguy/history.git] / cinelerra-5.1 / libzmpeg3 / video / cache.C

diff --git a/cinelerra-5.1/libzmpeg3/video/cache.C b/cinelerra-5.1/libzmpeg3/video/cache.C
new file mode 100644 (file)
index 0000000..2a5857c
--- /dev/null
+++ b/cinelerra-5.1/libzmpeg3/video/cache.C
@@ -0,0 +1,139 @@
+#include "../libzmpeg3.h"
+
+zcache_t::
+~cache_t()
+{
+  clear();
+}
+
+void zcache_t::
+clear()
+{
+  if( frames ) {
+    for( int i=0; i<allocation; ++i ) {
+      cacheframe_t *frame = &frames[i];
+      if( frame->y ) delete [] frame->y;
+      if( frame->u ) delete [] frame->u;
+      if( frame->v ) delete [] frame->v;
+    }
+    delete [] frames;
+    total = 0;
+    allocation = 0;
+    frames = 0;
+    seq = 0;
+  }
+}
+
+int zcache_t::
+extend_cache()
+{
+  if( total >= MAX_CACHE_FRAMES ) {
+    int i = 0, n = 0;
+    uint32_t vtim = frames[0].age;
+    while( ++i < total ) {
+      if( frames[i].age < vtim ) {
+        n = i;  vtim = frames[i].age;
+      }
+    }
+    frames[n].age = seq++;
+    return n;
+  }
+  if( total >= allocation ) {
+    int new_allocation = ZMAX(allocation*2,8);
+    cacheframe_t *new_frames = new cacheframe_t[new_allocation];
+    for( int i=0; i<total; ++i )
+      new_frames[i] = frames[i];
+    delete [] frames;
+    frames = new_frames;
+    allocation = new_allocation;
+//zmsgs("%d %d %d\n", new_allocation, allocation, total);
+  }
+  frames[total].age = seq++;
+  return total++;
+}
+
+void zcache_t::
+put_frame(int64_t zframe_number,
+  uint8_t *zy, uint8_t *zu, uint8_t *zv,
+  int zy_size, int zu_size, int zv_size)
+{
+  cacheframe_t *frame = 0;
+  int i;
+//zmsgs("save %d\n",zframe_number);
+
+  for( i=0; i<total; ++i ) { /* Get existing frame */
+    if( frames[i].frame_number == zframe_number ) {
+      frame = &frames[i];
+      break;
+    }
+  }
+
+  if( !frame ) {
+    int i = extend_cache();
+    frame = &frames[i];
+//zmsgs("30 %d %p %p %p\n", ptr->total, frame->y, frame->u, frame->v);
+    if( zy ) {
+      if( zy_size > frame->y_alloc ) {
+        delete [] frame->y;
+        frame->y = new uint8_t[frame->y_alloc=zy_size];
+      }
+      memcpy(frame->y, zy, frame->y_size=zy_size);
+    }
+    if( zu ) {
+      if( zu_size > frame->u_alloc ) {
+        delete [] frame->u;
+        frame->u = new uint8_t[frame->u_alloc=zu_size];
+      }
+      memcpy(frame->u, zu, frame->u_size=zu_size);
+    }
+    if( zv ) {
+      if( zv_size > frame->v_alloc ) {
+        delete [] frame->v;
+        frame->v = new uint8_t[frame->v_alloc=zv_size];
+      }
+      memcpy(frame->v, zv, frame->v_size=zv_size);
+    }
+    frame->frame_number = zframe_number;
+  }
+}
+
+int zcache_t::
+get_frame(int64_t zframe_number,
+  uint8_t **zy, uint8_t **zu, uint8_t **zv)
+{
+  for( int i=0; i<total; ++i ) {
+    cacheframe_t *frame = &frames[i];
+    if( frame->frame_number == zframe_number ) {
+      frame->age = seq++;
+      *zy = frame->y;
+      *zu = frame->u;
+      *zv = frame->v;
+//zmsgs("hit %d\n",zframe_number);
+      return 1;
+    }
+  }
+  clear();
+//zmsgs("missed %d\n",zframe_number);
+  return 0;
+}
+
+int zcache_t::
+has_frame(int64_t zframe_number)
+{
+  for( int i=0; i<total; ++i )
+    if( frames[i].frame_number == zframe_number ) return 1;
+
+  return 0;
+}
+
+int64_t zcache_t::
+memory_usage()
+{
+  int64_t result = 0;
+  for( int i=0; i<allocation; ++i ) {
+    cacheframe_t *frame = &frames[i];
+    result += frame->y_alloc + frame->u_alloc + frame->v_alloc;
+  }
+  return result;
+}
+