cinelerra/framecache.C

   1 #include "bcsignals.h"
   2 #include "clip.h"
   3 #include "framecache.h"
   4 #include "mutex.h"
   5 #include "vframe.h"
   6
   7
   8 #include <math.h>
   9 #include <string.h>
  10
  11
  12
  13 FrameCacheItem::FrameCacheItem()
  14  : CacheItemBase()
  15 {
  16         data = 0;
  17         position = 0;
  18         frame_rate = (double)30000.0 / 1001;
  19 }
  20
  21 FrameCacheItem::~FrameCacheItem()
  22 {
  23         delete data;
  24 }
  25
  26 int FrameCacheItem::get_size()
  27 {
  28         if(data) return data->get_data_size() + (path ? strlen(path) : 0);
  29         return 0;
  30 }
  31
  32
  33
  34
  35
  36
  37
  38
  39
  40
  41
  42
  43
  44
  45
  46 FrameCache::FrameCache()
  47  : CacheBase()
  48 {
  49 }
  50
  51 FrameCache::~FrameCache()
  52 {
  53 }
  54
  55
  56 // Returns 1 if frame exists in cache and copies it to the frame argument.
  57 int FrameCache::get_frame(VFrame *frame,
  58         int64_t position,
  59         double frame_rate)
  60 {
  61         lock->lock("FrameCache::get_frame");
  62         FrameCacheItem *result = 0;
  63
  64         if(frame_exists(frame,
  65                 position,
  66                 frame_rate,
  67                 &result))
  68         {
  69                 if(result->data)
  70                 {
  71                         frame->copy_from(result->data);
  72                         frame->copy_stacks(result->data);
  73                 }
  74                 result->age = get_age();
  75         }
  76
  77         lock->unlock();
  78         if(result) return 1;
  79         return 0;
  80 }
  81
  82
  83 VFrame* FrameCache::get_frame_ptr(int64_t position,
  84         double frame_rate,
  85         int color_model,
  86         int w,
  87         int h)
  88 {
  89         lock->lock("FrameCache::get_frame_ptr");
  90         FrameCacheItem *result = 0;
  91         if(frame_exists(position,
  92                 frame_rate,
  93                 color_model,
  94                 w,
  95                 h,
  96                 &result))
  97         {
  98                 result->age = get_age();
  99                 return result->data;
 100         }
 101
 102
 103         lock->unlock();
 104         return 0;
 105 }
 106
 107 // Puts frame in cache if enough space exists and the frame doesn't already
 108 // exist.
 109 void FrameCache::put_frame(VFrame *frame,
 110         int64_t position,
 111         double frame_rate,
 112         int use_copy)
 113 {
 114         lock->lock("FrameCache::put_frame");
 115         FrameCacheItem *item = 0;
 116         if(frame_exists(frame,
 117                 position,
 118                 frame_rate,
 119                 &item))
 120         {
 121                 item->age = get_age();
 122                 lock->unlock();
 123                 return;
 124         }
 125
 126
 127         item = new FrameCacheItem;
 128
 129         if(use_copy)
 130         {
 131                 item->data = new VFrame(*frame);
 132         }
 133         else
 134         {
 135                 item->data = frame;
 136         }
 137
 138         item->position = position;
 139         item->frame_rate = frame_rate;
 140         item->age = get_age();
 141
 142         put_item(item);
 143         lock->unlock();
 144 }
 145
 146
 147
 148
 149 int FrameCache::frame_exists(VFrame *format,
 150         int64_t position,
 151         double frame_rate,
 152         FrameCacheItem **item_return)
 153 {
 154         FrameCacheItem *item = (FrameCacheItem*)get_item(position);
 155         while(item && item->position == position)
 156         {
 157                 if(EQUIV(item->frame_rate, frame_rate) &&
 158                         format->equivalent(item->data, 1))
 159                 {
 160                         *item_return = item;
 161                         return 1;
 162                 }
 163                 else
 164                         item = (FrameCacheItem*)item->next;
 165         }
 166         return 0;
 167 }
 168
 169 int FrameCache::frame_exists(int64_t position,
 170         double frame_rate,
 171         int color_model,
 172         int w,
 173         int h,
 174         FrameCacheItem **item_return)
 175 {
 176         FrameCacheItem *item = (FrameCacheItem*)get_item(position);
 177         while(item && item->position == position)
 178         {
 179                 if(EQUIV(item->frame_rate, frame_rate) &&
 180                         color_model == item->data->get_color_model() &&
 181                         w == item->data->get_w() &&
 182                         h == item->data->get_h())
 183                 {
 184                         *item_return = item;
 185                         return 1;
 186                 }
 187                 else
 188                         item = (FrameCacheItem*)item->next;
 189         }
 190         return 0;
 191 }
 192
 193
 194 void FrameCache::dump()
 195 {
 196 //      lock->lock("FrameCache::dump");
 197 //      printf("FrameCache::dump 1 %d\n", items.total);
 198 //      for(int i = 0; i < items.total; i++)
 199 //      {
 200 //              FrameCacheItem *item = (FrameCacheItem*)items.values[i];
 201 //              printf("  position=%lld frame_rate=%f age=%d size=%d\n",
 202 //                      item->position,
 203 //                      item->frame_rate,
 204 //                      item->age,
 205 //                      item->data->get_data_size());
 206 //      }
 207 //      lock->unlock();
 208 }
 209
 210
 211
 212