Color Management, Stage 2: Switch color pipeline to use OpenColorIO
[blender.git] / source / blender / editors / space_clip / clip_editor.c
index b57b16707d42b7c76e061239e1341b7a0de64990..f2a48f299cfd1d50c8324d150ffa352fe13313b1 100644 (file)
@@ -44,6 +44,7 @@
 
 #include "BLI_utildefines.h"
 #include "BLI_math.h"
+#include "BLI_rect.h"
 
 #include "GPU_extensions.h"
 
@@ -115,7 +116,7 @@ int ED_space_clip_maskedit_mask_poll(bContext *C)
                if (clip) {
                        SpaceClip *sc = CTX_wm_space_clip(C);
 
-                       return sc->mask != NULL;
+                       return sc->mask_info.mask != NULL;
                }
        }
 
@@ -124,27 +125,32 @@ int ED_space_clip_maskedit_mask_poll(bContext *C)
 
 /* ******** common editing functions ******** */
 
-void ED_space_clip_get_size(const bContext *C, int *width, int *height)
+void ED_space_clip_get_size(SpaceClip *sc, int *width, int *height)
 {
-       SpaceClip *sc = CTX_wm_space_clip(C);
-
-       if (!sc->clip) {
-               *width = *height = 0;
+       if (sc->clip) {
+               BKE_movieclip_get_size(sc->clip, &sc->user, width, height);
        }
        else {
-               BKE_movieclip_get_size(sc->clip, &sc->user, width, height);
+               *width = *height = IMG_SIZE_FALLBACK;
        }
 }
 
-void ED_space_clip_get_zoom(const bContext *C, float *zoomx, float *zoomy)
+void ED_space_clip_get_size_fl(SpaceClip *sc, float size[2])
+{
+       int size_i[2];
+       ED_space_clip_get_size(sc, &size_i[0], &size_i[1]);
+       size[0] = size_i[0];
+       size[1] = size_i[1];
+}
+
+void ED_space_clip_get_zoom(SpaceClip *sc, ARegion *ar, float *zoomx, float *zoomy)
 {
-       ARegion *ar = CTX_wm_region(C);
        int width, height;
 
-       ED_space_clip_get_size(C, &width, &height);
+       ED_space_clip_get_size(sc, &width, &height);
 
-       *zoomx = (float)(ar->winrct.xmax - ar->winrct.xmin + 1) / (float)((ar->v2d.cur.xmax - ar->v2d.cur.xmin) * width);
-       *zoomy = (float)(ar->winrct.ymax - ar->winrct.ymin + 1) / (float)((ar->v2d.cur.ymax - ar->v2d.cur.ymin) * height);
+       *zoomx = (float)(BLI_RCT_SIZE_X(&ar->winrct) + 1) / (float)(BLI_RCT_SIZE_X(&ar->v2d.cur) * width);
+       *zoomy = (float)(BLI_RCT_SIZE_Y(&ar->winrct) + 1) / (float)(BLI_RCT_SIZE_Y(&ar->v2d.cur) * height);
 }
 
 void ED_space_clip_get_aspect(SpaceClip *sc, float *aspx, float *aspy)
@@ -152,7 +158,7 @@ void ED_space_clip_get_aspect(SpaceClip *sc, float *aspx, float *aspy)
        MovieClip *clip = ED_space_clip_get_clip(sc);
 
        if (clip)
-               BKE_movieclip_aspect(clip, aspx, aspy);
+               BKE_movieclip_get_aspect(clip, aspx, aspy);
        else
                *aspx = *aspy = 1.0f;
 
@@ -177,6 +183,13 @@ void ED_space_clip_get_aspect_dimension_aware(SpaceClip *sc, float *aspx, float
         * mainly this is sued for transformation stuff
         */
 
+       if (!sc->clip) {
+               *aspx = 1.0f;
+               *aspy = 1.0f;
+
+               return;
+       }
+
        ED_space_clip_get_aspect(sc, aspx, aspy);
        BKE_movieclip_get_size(sc->clip, &sc->user, &w, &h);
 
@@ -235,6 +248,48 @@ ImBuf *ED_space_clip_get_stable_buffer(SpaceClip *sc, float loc[2], float *scale
        return NULL;
 }
 
+/* returns color in SRGB */
+/* matching ED_space_image_color_sample() */
+int ED_space_clip_color_sample(SpaceClip *sc, ARegion *ar, int mval[2], float r_col[3])
+{
+       ImBuf *ibuf;
+       float fx, fy, co[2];
+       int ret = FALSE;
+
+       ibuf = ED_space_clip_get_buffer(sc);
+       if (!ibuf) {
+               return FALSE;
+       }
+
+       /* map the mouse coords to the backdrop image space */
+       ED_clip_mouse_pos(sc, ar, mval, co);
+
+       fx = co[0];
+       fy = co[1];
+
+       if (fx >= 0.0f && fy >= 0.0f && fx < 1.0f && fy < 1.0f) {
+               float *fp;
+               unsigned char *cp;
+               int x = (int)(fx * ibuf->x), y = (int)(fy * ibuf->y);
+
+               CLAMP(x, 0, ibuf->x - 1);
+               CLAMP(y, 0, ibuf->y - 1);
+
+               if (ibuf->rect_float) {
+                       fp = (ibuf->rect_float + (ibuf->channels) * (y * ibuf->x + x));
+                       linearrgb_to_srgb_v3_v3(r_col, fp);
+                       ret = TRUE;
+               }
+               else if (ibuf->rect) {
+                       cp = (unsigned char *)(ibuf->rect + y * ibuf->x + x);
+                       rgb_uchar_to_float(r_col, cp);
+                       ret = TRUE;
+               }
+       }
+
+       return ret;
+}
+
 void ED_clip_update_frame(const Main *mainp, int cfra)
 {
        wmWindowManager *wm;
@@ -258,9 +313,8 @@ void ED_clip_update_frame(const Main *mainp, int cfra)
        }
 }
 
-static int selected_boundbox(const bContext *C, float min[2], float max[2])
+static int selected_boundbox(SpaceClip *sc, float min[2], float max[2])
 {
-       SpaceClip *sc = CTX_wm_space_clip(C);
        MovieClip *clip = ED_space_clip_get_clip(sc);
        MovieTrackingTrack *track;
        int width, height, ok = FALSE;
@@ -269,7 +323,7 @@ static int selected_boundbox(const bContext *C, float min[2], float max[2])
 
        INIT_MINMAX2(min, max);
 
-       ED_space_clip_get_size(C, &width, &height);
+       ED_space_clip_get_size(sc, &width, &height);
 
        track = tracksbase->first;
        while (track) {
@@ -312,17 +366,17 @@ int ED_clip_view_selection(const bContext *C, ARegion *ar, int fit)
        int w, h, frame_width, frame_height;
        float min[2], max[2];
 
-       ED_space_clip_get_size(C, &frame_width, &frame_height);
+       ED_space_clip_get_size(sc, &frame_width, &frame_height);
 
        if (frame_width == 0 || frame_height == 0)
                return FALSE;
 
-       if (!selected_boundbox(C, min, max))
+       if (!selected_boundbox(sc, min, max))
                return FALSE;
 
        /* center view */
-       clip_view_center_to_point(C, (max[0] + min[0]) / (2 * frame_width),
-                                    (max[1] + min[1]) / (2 * frame_height));
+       clip_view_center_to_point(sc, (max[0] + min[0]) / (2 * frame_width),
+                                     (max[1] + min[1]) / (2 * frame_height));
 
        w = max[0] - min[0];
        h = max[1] - min[1];
@@ -334,13 +388,13 @@ int ED_clip_view_selection(const bContext *C, ARegion *ar, int fit)
 
                ED_space_clip_get_aspect(sc, &aspx, &aspy);
 
-               width = ar->winrct.xmax - ar->winrct.xmin + 1;
-               height = ar->winrct.ymax - ar->winrct.ymin + 1;
+               width  = BLI_RCT_SIZE_X(&ar->winrct) + 1;
+               height = BLI_RCT_SIZE_Y(&ar->winrct) + 1;
 
                zoomx = (float)width / w / aspx;
                zoomy = (float)height / h / aspy;
 
-               newzoom = 1.0f / power_of_2(1.0f / MIN2(zoomx, zoomy));
+               newzoom = 1.0f / power_of_2(1.0f / minf(zoomx, zoomy));
 
                if (fit || sc->zoom > newzoom)
                        sc->zoom = newzoom;
@@ -370,15 +424,13 @@ void ED_clip_point_undistorted_pos(SpaceClip *sc, const float co[2], float r_co[
        }
 }
 
-void ED_clip_point_stable_pos(const bContext *C, float x, float y, float *xr, float *yr)
+void ED_clip_point_stable_pos(SpaceClip *sc, ARegion *ar, float x, float y, float *xr, float *yr)
 {
-       ARegion *ar = CTX_wm_region(C);
-       SpaceClip *sc = CTX_wm_space_clip(C);
        int sx, sy, width, height;
        float zoomx, zoomy, pos[3], imat[4][4];
 
-       ED_space_clip_get_zoom(C, &zoomx, &zoomy);
-       ED_space_clip_get_size(C, &width, &height);
+       ED_space_clip_get_zoom(sc, ar, &zoomx, &zoomy);
+       ED_space_clip_get_size(sc, &width, &height);
 
        UI_view2d_to_region_no_clip(&ar->v2d, 0.0f, 0.0f, &sx, &sy);
 
@@ -409,18 +461,16 @@ void ED_clip_point_stable_pos(const bContext *C, float x, float y, float *xr, fl
  * \brief the reverse of ED_clip_point_stable_pos(), gets the marker region coords.
  * better name here? view_to_track / track_to_view or so?
  */
-void ED_clip_point_stable_pos__reverse(const bContext *C, const float co[2], float r_co[2])
+void ED_clip_point_stable_pos__reverse(SpaceClip *sc, ARegion *ar, const float co[2], float r_co[2])
 {
-       SpaceClip *sc = CTX_wm_space_clip(C);
-       ARegion *ar = CTX_wm_region(C);
        float zoomx, zoomy;
        float pos[3];
        int width, height;
        int sx, sy;
 
        UI_view2d_to_region_no_clip(&ar->v2d, 0.0f, 0.0f, &sx, &sy);
-       ED_space_clip_get_size(C, &width, &height);
-       ED_space_clip_get_zoom(C, &zoomx, &zoomy);
+       ED_space_clip_get_size(sc, &width, &height);
+       ED_space_clip_get_zoom(sc, ar, &zoomx, &zoomy);
 
        ED_clip_point_undistorted_pos(sc, co, pos);
        pos[2] = 0.0f;
@@ -432,9 +482,10 @@ void ED_clip_point_stable_pos__reverse(const bContext *C, const float co[2], flo
        r_co[1] = (pos[1] * height * zoomy) + (float)sy;
 }
 
-void ED_clip_mouse_pos(const bContext *C, wmEvent *event, float co[2])
+/* takes event->mval */
+void ED_clip_mouse_pos(SpaceClip *sc, ARegion *ar, const int mval[2], float co[2])
 {
-       ED_clip_point_stable_pos(C, event->mval[0], event->mval[1], &co[0], &co[1]);
+       ED_clip_point_stable_pos(sc, ar, mval[0], mval[1], &co[0], &co[1]);
 }
 
 int ED_space_clip_check_show_trackedit(SpaceClip *sc)
@@ -502,15 +553,15 @@ void ED_space_clip_set_clip(bContext *C, bScreen *screen, SpaceClip *sc, MovieCl
 
 Mask *ED_space_clip_get_mask(SpaceClip *sc)
 {
-       return sc->mask;
+       return sc->mask_info.mask;
 }
 
 void ED_space_clip_set_mask(bContext *C, SpaceClip *sc, Mask *mask)
 {
-       sc->mask = mask;
+       sc->mask_info.mask = mask;
 
-       if (sc->mask && sc->mask->id.us == 0) {
-               sc->clip->id.us = 1;
+       if (sc->mask_info.mask && sc->mask_info.mask->id.us == 0) {
+               sc->mask_info.mask->id.us = 1;
        }
 
        if (C) {
@@ -526,12 +577,15 @@ typedef struct SpaceClipDrawContext {
        GLuint texture;                 /* OGL texture ID */
        short texture_allocated;        /* flag if texture was allocated by glGenTextures */
        struct ImBuf *texture_ibuf;     /* image buffer for which texture was created */
+       const unsigned char *display_buffer; /* display buffer for which texture was created */
        int image_width, image_height;  /* image width and height for which texture was created */
        unsigned last_texture;          /* ID of previously used texture, so it'll be restored after clip drawing */
 
        /* fields to check if cache is still valid */
        int framenr, start_frame, frame_offset;
        short render_size, render_flag;
+
+       char colorspace[64];
 } SpaceClipDrawContext;
 
 int ED_space_clip_texture_buffer_supported(SpaceClip *sc)
@@ -556,7 +610,7 @@ int ED_space_clip_texture_buffer_supported(SpaceClip *sc)
        return context->buffers_supported;
 }
 
-int ED_space_clip_load_movieclip_buffer(SpaceClip *sc, ImBuf *ibuf)
+int ED_space_clip_load_movieclip_buffer(SpaceClip *sc, ImBuf *ibuf, const unsigned char *display_buffer)
 {
        SpaceClipDrawContext *context = sc->draw_context;
        MovieClip *clip = ED_space_clip_get_clip(sc);
@@ -568,12 +622,22 @@ int ED_space_clip_load_movieclip_buffer(SpaceClip *sc, ImBuf *ibuf)
         * assuming displaying happens of footage frames only on which painting doesn't heppen.
         * so not changed image buffer pointer means unchanged image content */
        need_rebind |= context->texture_ibuf != ibuf;
+       need_rebind |= context->display_buffer != display_buffer;
        need_rebind |= context->framenr != sc->user.framenr;
        need_rebind |= context->render_size != sc->user.render_size;
        need_rebind |= context->render_flag != sc->user.render_flag;
        need_rebind |= context->start_frame != clip->start_frame;
        need_rebind |= context->frame_offset != clip->frame_offset;
 
+       if (!need_rebind) {
+               /* OCIO_TODO: not entirely nice, but currently it seems to be easiest way
+                *            to deal with changing input color space settings
+                *            pointer-based check could fail due to new buffers could be
+                *            be allocated on on old memory
+                */
+               need_rebind = strcmp(context->colorspace, clip->colorspace_settings.name) != 0;
+       }
+
        if (need_rebind) {
                int width = ibuf->x, height = ibuf->y;
                int need_recreate = 0;
@@ -613,16 +677,12 @@ int ED_space_clip_load_movieclip_buffer(SpaceClip *sc, ImBuf *ibuf)
                        glBindTexture(GL_TEXTURE_2D, context->texture);
                }
 
-               if (ibuf->rect_float) {
-                       if (ibuf->rect == NULL)
-                               IMB_rect_from_float(ibuf);
-               }
-
-               if (ibuf->rect)
-                       glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA8, width, height, 0, GL_RGBA, GL_UNSIGNED_BYTE, ibuf->rect);
+               if (display_buffer)
+                       glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA8, width, height, 0, GL_RGBA, GL_UNSIGNED_BYTE, display_buffer);
 
                /* store settings */
                context->texture_allocated = 1;
+               context->display_buffer = display_buffer;
                context->texture_ibuf = ibuf;
                context->image_width = ibuf->x;
                context->image_height = ibuf->y;
@@ -631,6 +691,8 @@ int ED_space_clip_load_movieclip_buffer(SpaceClip *sc, ImBuf *ibuf)
                context->render_flag = sc->user.render_flag;
                context->start_frame = clip->start_frame;
                context->frame_offset = clip->frame_offset;
+
+               strcpy(context->colorspace, clip->colorspace_settings.name);
        }
        else {
                /* displaying exactly the same image which was loaded t oa texture,