sim: Remove "continue physics" code
[blender.git] / source / blender / blenkernel / intern / image.c
index d98c2bc2f8bb9dbbcb23cd940d50e309cbe0841a..c56091ead628d055fea28bf1937c219722aa4811 100644 (file)
@@ -68,7 +68,6 @@
 #include "BLI_blenlib.h"
 #include "BLI_threads.h"
 #include "BLI_utildefines.h"
-#include "BLI_bpath.h"
 
 #include "BKE_bmfont.h"
 #include "BKE_colortools.h"
@@ -313,10 +312,6 @@ static void image_assign_ibuf(Image *ima, ImBuf *ibuf, int index, int frame)
                                break;
 
                ibuf->index = index;
-               if (ima->flag & IMA_CM_PREDIVIDE)
-                       ibuf->flags |= IB_cm_predivide;
-               else
-                       ibuf->flags &= ~IB_cm_predivide;
 
                /* this function accepts (link == NULL) */
                BLI_insertlinkbefore(&ima->ibufs, link, ibuf);
@@ -553,6 +548,26 @@ int BKE_image_scale(Image *image, int width, int height)
        return (ibuf != NULL);
 }
 
+static void image_init_color_management(Image *ima)
+{
+       ImBuf *ibuf;
+       char name[FILE_MAX];
+
+       BKE_image_user_file_path(NULL, ima, name);
+
+       /* will set input color space to image format default's */
+       ibuf = IMB_loadiffname(name, IB_test | IB_alphamode_detect, ima->colorspace_settings.name);
+
+       if (ibuf) {
+               if (ibuf->flags & IB_alphamode_premul)
+                       ima->alpha_mode = IMA_ALPHA_PREMUL;
+               else
+                       ima->alpha_mode = IMA_ALPHA_STRAIGHT;
+
+               IMB_freeImBuf(ibuf);
+       }
+}
+
 Image *BKE_image_load(const char *filepath)
 {
        Image *ima;
@@ -580,6 +595,8 @@ Image *BKE_image_load(const char *filepath)
        if (BLI_testextensie_array(filepath, imb_ext_movie))
                ima->source = IMA_SRC_MOVIE;
 
+       image_init_color_management(ima);
+
        return ima;
 }
 
@@ -663,6 +680,13 @@ static ImBuf *add_ibuf_size(unsigned int width, unsigned int height, const char
                        BKE_image_buf_fill_color(rect, rect_float, width, height, color);
        }
 
+       if (rect_float) {
+               /* both byte and float buffers are filling in sRGB space, need to linearize float buffer after BKE_image_buf_fill* functions */
+
+               IMB_buffer_float_from_float(rect_float, rect_float, ibuf->channels, IB_PROFILE_LINEAR_RGB, IB_PROFILE_SRGB,
+                                           TRUE, ibuf->x, ibuf->y, ibuf->x, ibuf->x);
+       }
+
        return ibuf;
 }
 
@@ -880,7 +904,7 @@ void BKE_image_free_all_textures(void)
                                image_free_buffers(ima);
                }
        }
-       /* printf("freed total %d MB\n", totsize/(1024*1024)); */
+       /* printf("freed total %d MB\n", totsize / (1024 * 1024)); */
 }
 
 /* except_frame is weak, only works for seqs without offset... */
@@ -931,7 +955,7 @@ int BKE_imtype_to_ftype(const char imtype)
                return RADHDR;
 #endif
        else if (imtype == R_IMF_IMTYPE_PNG)
-               return PNG;
+               return PNG | 90;
 #ifdef WITH_DDS
        else if (imtype == R_IMF_IMTYPE_DDS)
                return DDS;
@@ -1113,6 +1137,8 @@ char BKE_imtype_valid_depths(const char imtype)
                        return R_IMF_CHAN_DEPTH_10;
                case R_IMF_IMTYPE_JP2:
                        return R_IMF_CHAN_DEPTH_8 | R_IMF_CHAN_DEPTH_12 | R_IMF_CHAN_DEPTH_16;
+               case R_IMF_IMTYPE_PNG:
+                       return R_IMF_CHAN_DEPTH_8 | R_IMF_CHAN_DEPTH_16;
                /* most formats are 8bit only */
                default:
                        return R_IMF_CHAN_DEPTH_8;
@@ -1159,9 +1185,10 @@ char BKE_imtype_from_arg(const char *imtype_arg)
        else return R_IMF_IMTYPE_INVALID;
 }
 
-int BKE_add_image_extension(char *string, const char imtype)
+static int do_add_image_extension(char *string, const char imtype, const ImageFormatData *im_format)
 {
        const char *extension = NULL;
+       (void)im_format;  /* may be unused, depends on build options */
 
        if (imtype == R_IMF_IMTYPE_IRIS) {
                if (!BLI_testextensie(string, ".rgb"))
@@ -1226,8 +1253,22 @@ int BKE_add_image_extension(char *string, const char imtype)
        }
 #ifdef WITH_OPENJPEG
        else if (imtype == R_IMF_IMTYPE_JP2) {
-               if (!BLI_testextensie(string, ".jp2"))
-                       extension = ".jp2";
+               if (im_format) {
+                       if (im_format->jp2_codec == R_IMF_JP2_CODEC_JP2) {
+                               if (!BLI_testextensie(string, ".jp2"))
+                                       extension = ".jp2";
+                       }
+                       else if (im_format->jp2_codec == R_IMF_JP2_CODEC_J2K) {
+                               if (!BLI_testextensie(string, ".j2c"))
+                                       extension = ".j2c";
+                       }
+                       else
+                               BLI_assert(!"Unsupported jp2 codec was specified in im_format->jp2_codec");
+               }
+               else {
+                       if (!BLI_testextensie(string, ".jp2"))
+                               extension = ".jp2";
+               }
        }
 #endif
        else { //   R_IMF_IMTYPE_AVIRAW, R_IMF_IMTYPE_AVIJPEG, R_IMF_IMTYPE_JPEG90, R_IMF_IMTYPE_QUICKTIME etc
@@ -1253,11 +1294,22 @@ int BKE_add_image_extension(char *string, const char imtype)
        }
 }
 
+int BKE_add_image_extension(char *string, const ImageFormatData *im_format)
+{
+       return do_add_image_extension(string, im_format->imtype, im_format);
+}
+
+int BKE_add_image_extension_from_type(char *string, const char imtype)
+{
+       return do_add_image_extension(string, imtype, NULL);
+}
+
 void BKE_imformat_defaults(ImageFormatData *im_format)
 {
        memset(im_format, 0, sizeof(*im_format));
        im_format->planes = R_IMF_PLANES_RGB;
        im_format->imtype = R_IMF_IMTYPE_PNG;
+       im_format->depth = R_IMF_CHAN_DEPTH_8;
        im_format->quality = 90;
        im_format->compress = 90;
 
@@ -1282,9 +1334,13 @@ void BKE_imbuf_to_image_format(struct ImageFormatData *im_format, const ImBuf *i
                im_format->imtype = R_IMF_IMTYPE_RADHDR;
 #endif
 
-       else if (ftype == PNG)
+       else if (ftype == PNG) {
                im_format->imtype = R_IMF_IMTYPE_PNG;
 
+               if (custom_flags & PNG_16BIT)
+                       im_format->depth = R_IMF_CHAN_DEPTH_16;
+       }
+
 #ifdef WITH_DDS
        else if (ftype == DDS)
                im_format->imtype = R_IMF_IMTYPE_DDS;
@@ -1345,6 +1401,13 @@ void BKE_imbuf_to_image_format(struct ImageFormatData *im_format, const ImBuf *i
                        if (ftype & JP2_CINE_48FPS)
                                im_format->jp2_flag |= R_IMF_JP2_FLAG_CINE_48;
                }
+
+               if (ftype & JP2_JP2)
+                       im_format->jp2_codec = R_IMF_JP2_CODEC_JP2;
+               else if (ftype & JP2_J2K)
+                       im_format->jp2_codec = R_IMF_JP2_CODEC_J2K;
+               else
+                       BLI_assert(!"Unsupported jp2 codec was specified in file type");
        }
 #endif
 
@@ -1393,7 +1456,9 @@ static void timecode_simple_string(char *text, size_t text_size, const int cfra,
        }
 }
 
-/* could allow access externally - 512 is for long names, 64 is for id names */
+#define STAMP_NAME_SIZE ((MAX_ID_NAME - 2) + 16)
+/* could allow access externally - 512 is for long names,
+ * STAMP_NAME_SIZE is for id names, allowing them some room for description */
 typedef struct StampData {
        char file[512];
        char note[512];
@@ -1401,12 +1466,13 @@ typedef struct StampData {
        char marker[512];
        char time[512];
        char frame[512];
-       char camera[64];
-       char cameralens[64];
-       char scene[64];
-       char strip[64];
-       char rendertime[64];
+       char camera[STAMP_NAME_SIZE];
+       char cameralens[STAMP_NAME_SIZE];
+       char scene[STAMP_NAME_SIZE];
+       char strip[STAMP_NAME_SIZE];
+       char rendertime[STAMP_NAME_SIZE];
 } StampData;
+#undef STAMP_NAME_SIZE
 
 static void stampdata(Scene *scene, Object *camera, StampData *stamp_data, int do_prefix)
 {
@@ -1806,8 +1872,12 @@ int BKE_imbuf_write(ImBuf *ibuf, const char *name, ImageFormatData *imf)
        else if (ELEM5(imtype, R_IMF_IMTYPE_PNG, R_IMF_IMTYPE_FFMPEG, R_IMF_IMTYPE_H264, R_IMF_IMTYPE_THEORA, R_IMF_IMTYPE_XVID)) {
                ibuf->ftype = PNG;
 
-               if (imtype == R_IMF_IMTYPE_PNG)
+               if (imtype == R_IMF_IMTYPE_PNG) {
+                       if (imf->depth == R_IMF_CHAN_DEPTH_16)
+                               ibuf->ftype |= PNG_16BIT;
+
                        ibuf->ftype |= compress;
+               }
 
        }
 #ifdef WITH_DDS
@@ -1897,6 +1967,13 @@ int BKE_imbuf_write(ImBuf *ibuf, const char *name, ImageFormatData *imf)
                        if (imf->jp2_flag & R_IMF_JP2_FLAG_CINE_48)
                                ibuf->ftype |= JP2_CINE_48FPS;
                }
+
+               if (imf->jp2_codec == R_IMF_JP2_CODEC_JP2)
+                       ibuf->ftype |= JP2_JP2;
+               else if (imf->jp2_codec == R_IMF_JP2_CODEC_J2K)
+                       ibuf->ftype |= JP2_J2K;
+               else
+                       BLI_assert(!"Unsupported jp2 codec was specified in im_format->jp2_codec");
        }
 #endif
        else {
@@ -1947,7 +2024,8 @@ int BKE_imbuf_write_stamp(Scene *scene, struct Object *camera, ImBuf *ibuf, cons
 }
 
 
-void BKE_makepicstring(char *string, const char *base, const char *relbase, int frame, const char imtype, const short use_ext, const short use_frames)
+static void do_makepicstring(char *string, const char *base, const char *relbase, int frame, const char imtype,
+                             const ImageFormatData *im_format, const short use_ext, const short use_frames)
 {
        if (string == NULL) return;
        BLI_strncpy(string, base, FILE_MAX - 10);   /* weak assumption */
@@ -1957,8 +2035,17 @@ void BKE_makepicstring(char *string, const char *base, const char *relbase, int
                BLI_path_frame(string, frame, 4);
 
        if (use_ext)
-               BKE_add_image_extension(string, imtype);
+               do_add_image_extension(string, imtype, im_format);
+}
 
+void BKE_makepicstring(char *string, const char *base, const char *relbase, int frame, const ImageFormatData *im_format, const short use_ext, const short use_frames)
+{
+       do_makepicstring(string, base, relbase, frame, im_format->imtype, im_format, use_ext, use_frames);
+}
+
+void BKE_makepicstring_from_type(char *string, const char *base, const char *relbase, int frame, const char imtype, const short use_ext, const short use_frames)
+{
+       do_makepicstring(string, base, relbase, frame, imtype, NULL, use_ext, use_frames);
 }
 
 /* used by sequencer too */
@@ -2275,7 +2362,7 @@ void BKE_image_backup_render(Scene *scene, Image *ima)
 static void image_create_multilayer(Image *ima, ImBuf *ibuf, int framenr)
 {
        const char *colorspace = ima->colorspace_settings.name;
-       int predivide = ima->flag & IMA_CM_PREDIVIDE;
+       int predivide = ima->alpha_mode == IMA_ALPHA_PREMUL;
 
        ima->rr = RE_MultilayerConvert(ibuf->userdata, colorspace, predivide, ibuf->x, ibuf->y);
 
@@ -2307,6 +2394,18 @@ static void image_initialize_after_load(Image *ima, ImBuf *ibuf)
 
 }
 
+static int imbuf_alpha_flags_for_image(Image *ima)
+{
+       int flag = 0;
+
+       if (ima->flag & IMA_IGNORE_ALPHA)
+               flag |= IB_ignore_alpha;
+       else if (ima->alpha_mode == IMA_ALPHA_PREMUL)
+               flag |= IB_alphamode_premul;
+
+       return flag;
+}
+
 static ImBuf *image_load_sequence_file(Image *ima, ImageUser *iuser, int frame)
 {
        struct ImBuf *ibuf;
@@ -2321,8 +2420,7 @@ static ImBuf *image_load_sequence_file(Image *ima, ImageUser *iuser, int frame)
        BKE_image_user_file_path(iuser, ima, name);
 
        flag = IB_rect | IB_multilayer;
-       if (ima->flag & IMA_DO_PREMUL)
-               flag |= IB_premul;
+       flag |= imbuf_alpha_flags_for_image(ima);
 
        /* read ibuf */
        ibuf = IMB_loadiffname(name, flag, ima->colorspace_settings.name);
@@ -2481,15 +2579,14 @@ static ImBuf *image_load_image_file(Image *ima, ImageUser *iuser, int cfra)
        /* is there a PackedFile with this image ? */
        if (ima->packedfile) {
                flag = IB_rect | IB_multilayer;
-               if (ima->flag & IMA_DO_PREMUL) flag |= IB_premul;
+               flag |= imbuf_alpha_flags_for_image(ima);
 
                ibuf = IMB_ibImageFromMemory((unsigned char *)ima->packedfile->data, ima->packedfile->size, flag,
                                             ima->colorspace_settings.name, "<packed data>");
        }
        else {
                flag = IB_rect | IB_multilayer | IB_metadata;
-               if (ima->flag & IMA_DO_PREMUL)
-                       flag |= IB_premul;
+               flag |= imbuf_alpha_flags_for_image(ima);
 
                /* get the right string */
                BKE_image_user_frame_calc(iuser, cfra, 0);
@@ -2709,20 +2806,33 @@ static ImBuf *image_get_render_result(Image *ima, ImageUser *iuser, void **lock_
 
        ibuf->dither = dither;
 
-       if (iuser->scene->r.color_mgt_flag & R_COLOR_MANAGEMENT_PREDIVIDE) {
-               ibuf->flags |= IB_cm_predivide;
-               ima->flag |= IMA_CM_PREDIVIDE;
-       }
-       else {
-               ibuf->flags &= ~IB_cm_predivide;
-               ima->flag &= ~IMA_CM_PREDIVIDE;
-       }
-
        ima->ok = IMA_OK_LOADED;
 
        return ibuf;
 }
 
+static void image_get_fame_and_index(Image *ima, ImageUser *iuser, int *frame_r, int *index_r)
+{
+       int frame = 0, index = 0;
+
+       /* see if we already have an appropriate ibuf, with image source and type */
+       if (ima->source == IMA_SRC_MOVIE) {
+               frame = iuser ? iuser->framenr : ima->lastframe;
+       }
+       else if (ima->source == IMA_SRC_SEQUENCE) {
+               if (ima->type == IMA_TYPE_IMAGE) {
+                       frame = iuser ? iuser->framenr : ima->lastframe;
+               }
+               else if (ima->type == IMA_TYPE_MULTILAYER) {
+                       frame = iuser ? iuser->framenr : ima->lastframe;
+                       index = iuser ? iuser->multi_index : IMA_NO_INDEX;
+               }
+       }
+
+       *frame_r = frame;
+       *index_r = index;
+}
+
 static ImBuf *image_get_ibuf_threadsafe(Image *ima, ImageUser *iuser, int *frame_r, int *index_r)
 {
        ImBuf *ibuf = NULL;
@@ -2778,6 +2888,21 @@ static ImBuf *image_get_ibuf_threadsafe(Image *ima, ImageUser *iuser, int *frame
        return ibuf;
 }
 
+BLI_INLINE int image_quick_test(Image *ima, ImageUser *iuser)
+{
+       if (ima == NULL)
+               return FALSE;
+
+       if (iuser) {
+               if (iuser->ok == 0)
+                       return FALSE;
+       }
+       else if (ima->ok == 0)
+               return FALSE;
+
+       return TRUE;
+}
+
 /* Checks optional ImageUser and verifies/creates ImBuf.
  *
  * not thread-safe, so callee should worry about thread locks
@@ -2792,14 +2917,7 @@ static ImBuf *image_acquire_ibuf(Image *ima, ImageUser *iuser, void **lock_r)
                *lock_r = NULL;
 
        /* quick reject tests */
-       if (ima == NULL)
-               return NULL;
-
-       if (iuser) {
-               if (iuser->ok == 0)
-                       return NULL;
-       }
-       else if (ima->ok == 0)
+       if (!image_quick_test(ima, iuser))
                return NULL;
 
        ibuf = image_get_ibuf_threadsafe(ima, iuser, &frame, &index);
@@ -2920,7 +3038,13 @@ void BKE_image_release_ibuf(Image *ima, ImBuf *ibuf, void *lock)
 /* checks whether there's an image buffer for given image and user */
 int BKE_image_has_ibuf(Image *ima, ImageUser *iuser)
 {
-       ImBuf *ibuf = image_get_ibuf_threadsafe(ima, iuser, NULL, NULL);
+       ImBuf *ibuf;
+
+       /* quick reject tests */
+       if (!image_quick_test(ima, iuser))
+               return FALSE;
+
+       ibuf = image_get_ibuf_threadsafe(ima, iuser, NULL, NULL);
 
        if (!ibuf) {
                BLI_spin_lock(&image_spin);
@@ -2936,6 +3060,122 @@ int BKE_image_has_ibuf(Image *ima, ImageUser *iuser)
        return ibuf != NULL;
 }
 
+/* ******** Pool for image buffers ********  */
+
+typedef struct ImagePoolEntry {
+       struct ImagePoolEntry *next, *prev;
+       Image *image;
+       ImBuf *ibuf;
+       int index;
+       int frame;
+} ImagePoolEntry;
+
+typedef struct ImagePool {
+       ListBase image_buffers;
+} ImagePool;
+
+ImagePool *BKE_image_pool_new(void)
+{
+       ImagePool *pool = MEM_callocN(sizeof(ImagePool), "Image Pool");
+
+       return pool;
+}
+
+void BKE_image_pool_free(ImagePool *pool)
+{
+       ImagePoolEntry *entry, *next_entry;
+
+       /* use single lock to dereference all the image buffers */
+       BLI_spin_lock(&image_spin);
+
+       for (entry = pool->image_buffers.first; entry; entry = next_entry) {
+               next_entry = entry->next;
+
+               if (entry->ibuf)
+                       IMB_freeImBuf(entry->ibuf);
+
+               MEM_freeN(entry);
+       }
+
+       BLI_spin_unlock(&image_spin);
+
+       MEM_freeN(pool);
+}
+
+BLI_INLINE ImBuf *image_pool_find_entry(ImagePool *pool, Image *image, int frame, int index, int *found)
+{
+       ImagePoolEntry *entry;
+
+       *found = FALSE;
+
+       for (entry = pool->image_buffers.first; entry; entry = entry->next) {
+               if (entry->image == image && entry->frame == frame && entry->index == index) {
+                       *found = TRUE;
+                       return entry->ibuf;
+               }
+       }
+
+       return NULL;
+}
+
+ImBuf *BKE_image_pool_acquire_ibuf(Image *ima, ImageUser *iuser, ImagePool *pool)
+{
+       ImBuf *ibuf;
+       int index, frame, found;
+
+       if (!image_quick_test(ima, iuser))
+               return NULL;
+
+       if (pool == NULL) {
+               /* pool could be NULL, in this case use general acquire function */
+               return BKE_image_acquire_ibuf(ima, iuser, NULL);
+       }
+
+       image_get_fame_and_index(ima, iuser, &frame, &index);
+
+       ibuf = image_pool_find_entry(pool, ima, frame, index, &found);
+       if (found)
+               return ibuf;
+
+       BLI_spin_lock(&image_spin);
+
+       ibuf = image_pool_find_entry(pool, ima, frame, index, &found);
+
+       /* will also create entry even in cases image buffer failed to load,
+        * prevents trying to load the same buggy file multiple times
+        */
+       if (!found) {
+               ImagePoolEntry *entry;
+
+               ibuf = image_acquire_ibuf(ima, iuser, NULL);
+
+               if (ibuf)
+                       IMB_refImBuf(ibuf);
+
+               entry = MEM_callocN(sizeof(ImagePoolEntry), "Image Pool Entry");
+               entry->image = ima;
+               entry->frame = frame;
+               entry->index = index;
+               entry->ibuf = ibuf;
+
+               BLI_addtail(&pool->image_buffers, entry);
+       }
+
+       BLI_spin_unlock(&image_spin);
+
+       return ibuf;
+}
+
+void BKE_image_pool_release_ibuf(Image *ima, ImBuf *ibuf, ImagePool *pool)
+{
+       /* if pool wasn't actually used, use general release stuff,
+        * for pools image buffers will be dereferenced on pool free
+        */
+       if (pool == NULL) {
+               BKE_image_release_ibuf(ima, ibuf, NULL);
+       }
+}
+
 int BKE_image_user_frame_get(const ImageUser *iuser, int cfra, int fieldnr, short *r_is_in_range)
 {
        const int len = (iuser->fie_ima * iuser->frames) / 2;
@@ -3098,3 +3338,57 @@ void BKE_image_get_aspect(Image *image, float *aspx, float *aspy)
        else
                *aspy = 1.0f;
 }
+
+unsigned char *BKE_image_get_pixels_for_frame(struct Image *image, int frame)
+{
+       ImageUser iuser = {0};
+       void *lock;
+       ImBuf *ibuf;
+       unsigned char *pixels = NULL;
+
+       iuser.framenr = frame;
+       iuser.ok = TRUE;
+
+       ibuf = BKE_image_acquire_ibuf(image, &iuser, &lock);
+
+       if (ibuf) {
+               pixels = (unsigned char *) ibuf->rect;
+
+               if (pixels)
+                       pixels = MEM_dupallocN(pixels);
+
+               BKE_image_release_ibuf(image, ibuf, lock);
+       }
+
+       if (!pixels)
+               return NULL;
+
+       return pixels;
+}
+
+float *BKE_image_get_float_pixels_for_frame(struct Image *image, int frame)
+{
+       ImageUser iuser = {0};
+       void *lock;
+       ImBuf *ibuf;
+       float *pixels = NULL;
+
+       iuser.framenr = frame;
+       iuser.ok = TRUE;
+
+       ibuf = BKE_image_acquire_ibuf(image, &iuser, &lock);
+
+       if (ibuf) {
+               pixels = ibuf->rect_float;
+
+               if (pixels)
+                       pixels = MEM_dupallocN(pixels);
+
+               BKE_image_release_ibuf(image, ibuf, lock);
+       }
+
+       if (!pixels)
+               return NULL;
+
+       return pixels;
+}