Home | History | Annotate | Download | only in src
      1 /*
      2  * Copyright (C) 2011 The Android Open Source Project
      3  *
      4  * Licensed under the Apache License, Version 2.0 (the "License");
      5  * you may not use this file except in compliance with the License.
      6  * You may obtain a copy of the License at
      7  *
      8  *      http://www.apache.org/licenses/LICENSE-2.0
      9  *
     10  * Unless required by applicable law or agreed to in writing, software
     11  * distributed under the License is distributed on an "AS IS" BASIS,
     12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
     13  * See the License for the specific language governing permissions and
     14  * limitations under the License.
     15  */
     16 #include    "M4VIFI_FiltersAPI.h"
     17 
     18 #include    "M4VIFI_Defines.h"
     19 
     20 #include    "M4VIFI_Clip.h"
     21 
     22 /***************************************************************************
     23 Proto:
     24 M4VIFI_UInt8    M4VIFI_RGB888toYUV420(void *pUserData, M4VIFI_ImagePlane *PlaneIn,
     25                                      M4VIFI_ImagePlane PlaneOut[3]);
     26 Purpose:    filling of the YUV420 plane from a BGR24 plane
     27 Abstract:    Loop on each row ( 2 rows by 2 rows )
     28                 Loop on each column ( 2 col by 2 col )
     29                     Get 4 BGR samples from input data and build 4 output Y samples and
     30                     each single U & V data
     31                 end loop on col
     32             end loop on row
     33 
     34 In:            RGB24 plane
     35 InOut:        none
     36 Out:        array of 3 M4VIFI_ImagePlane structures
     37 Modified:    ML: RGB function modified to BGR.
     38 ***************************************************************************/
     39 M4VIFI_UInt8 M4VIFI_RGB888toYUV420(void *pUserData, M4VIFI_ImagePlane *PlaneIn,
     40                                     M4VIFI_ImagePlane PlaneOut[3])
     41 {
     42     M4VIFI_UInt32    u32_width, u32_height;
     43     M4VIFI_UInt32    u32_stride_Y, u32_stride2_Y, u32_stride_U, u32_stride_V, u32_stride_rgb,\
     44                      u32_stride_2rgb;
     45     M4VIFI_UInt32    u32_col, u32_row;
     46 
     47     M4VIFI_Int32    i32_r00, i32_r01, i32_r10, i32_r11;
     48     M4VIFI_Int32    i32_g00, i32_g01, i32_g10, i32_g11;
     49     M4VIFI_Int32    i32_b00, i32_b01, i32_b10, i32_b11;
     50     M4VIFI_Int32    i32_y00, i32_y01, i32_y10, i32_y11;
     51     M4VIFI_Int32    i32_u00, i32_u01, i32_u10, i32_u11;
     52     M4VIFI_Int32    i32_v00, i32_v01, i32_v10, i32_v11;
     53     M4VIFI_UInt8    *pu8_yn, *pu8_ys, *pu8_u, *pu8_v;
     54     M4VIFI_UInt8    *pu8_y_data, *pu8_u_data, *pu8_v_data;
     55     M4VIFI_UInt8    *pu8_rgbn_data, *pu8_rgbn;
     56 
     57     /* check sizes */
     58     if( (PlaneIn->u_height != PlaneOut[0].u_height)            ||
     59         (PlaneOut[0].u_height != (PlaneOut[1].u_height<<1))    ||
     60         (PlaneOut[0].u_height != (PlaneOut[2].u_height<<1)))
     61         return M4VIFI_ILLEGAL_FRAME_HEIGHT;
     62 
     63     if( (PlaneIn->u_width != PlaneOut[0].u_width)        ||
     64         (PlaneOut[0].u_width != (PlaneOut[1].u_width<<1))    ||
     65         (PlaneOut[0].u_width != (PlaneOut[2].u_width<<1)))
     66         return M4VIFI_ILLEGAL_FRAME_WIDTH;
     67 
     68 
     69     /* set the pointer to the beginning of the output data buffers */
     70     pu8_y_data    = PlaneOut[0].pac_data + PlaneOut[0].u_topleft;
     71     pu8_u_data    = PlaneOut[1].pac_data + PlaneOut[1].u_topleft;
     72     pu8_v_data    = PlaneOut[2].pac_data + PlaneOut[2].u_topleft;
     73 
     74     /* idem for input buffer */
     75     pu8_rgbn_data    = PlaneIn->pac_data + PlaneIn->u_topleft;
     76 
     77     /* get the size of the output image */
     78     u32_width    = PlaneOut[0].u_width;
     79     u32_height    = PlaneOut[0].u_height;
     80 
     81     /* set the size of the memory jumps corresponding to row jump in each output plane */
     82     u32_stride_Y = PlaneOut[0].u_stride;
     83     u32_stride2_Y= u32_stride_Y << 1;
     84     u32_stride_U = PlaneOut[1].u_stride;
     85     u32_stride_V = PlaneOut[2].u_stride;
     86 
     87     /* idem for input plane */
     88     u32_stride_rgb = PlaneIn->u_stride;
     89     u32_stride_2rgb = u32_stride_rgb << 1;
     90 
     91     /* loop on each row of the output image, input coordinates are estimated from output ones */
     92     /* two YUV rows are computed at each pass */
     93     for    (u32_row = u32_height ;u32_row != 0; u32_row -=2)
     94     {
     95         /* update working pointers */
     96         pu8_yn    = pu8_y_data;
     97         pu8_ys    = pu8_yn + u32_stride_Y;
     98 
     99         pu8_u    = pu8_u_data;
    100         pu8_v    = pu8_v_data;
    101 
    102         pu8_rgbn= pu8_rgbn_data;
    103 
    104         /* loop on each column of the output image*/
    105         for    (u32_col = u32_width; u32_col != 0 ; u32_col -=2)
    106         {
    107             /* get RGB samples of 4 pixels */
    108             GET_RGB24(i32_r00, i32_g00, i32_b00, pu8_rgbn, 0);
    109             GET_RGB24(i32_r10, i32_g10, i32_b10, pu8_rgbn, CST_RGB_24_SIZE);
    110             GET_RGB24(i32_r01, i32_g01, i32_b01, pu8_rgbn, u32_stride_rgb);
    111             GET_RGB24(i32_r11, i32_g11, i32_b11, pu8_rgbn, u32_stride_rgb + CST_RGB_24_SIZE);
    112 
    113             i32_u00    = U24(i32_r00, i32_g00, i32_b00);
    114             i32_v00    = V24(i32_r00, i32_g00, i32_b00);
    115             i32_y00    = Y24(i32_r00, i32_g00, i32_b00);        /* matrix luminance */
    116             pu8_yn[0]= (M4VIFI_UInt8)i32_y00;
    117 
    118             i32_u10    = U24(i32_r10, i32_g10, i32_b10);
    119             i32_v10    = V24(i32_r10, i32_g10, i32_b10);
    120             i32_y10    = Y24(i32_r10, i32_g10, i32_b10);
    121             pu8_yn[1]= (M4VIFI_UInt8)i32_y10;
    122 
    123             i32_u01    = U24(i32_r01, i32_g01, i32_b01);
    124             i32_v01    = V24(i32_r01, i32_g01, i32_b01);
    125             i32_y01    = Y24(i32_r01, i32_g01, i32_b01);
    126             pu8_ys[0]= (M4VIFI_UInt8)i32_y01;
    127 
    128             i32_u11    = U24(i32_r11, i32_g11, i32_b11);
    129             i32_v11    = V24(i32_r11, i32_g11, i32_b11);
    130             i32_y11    = Y24(i32_r11, i32_g11, i32_b11);
    131             pu8_ys[1] = (M4VIFI_UInt8)i32_y11;
    132 
    133             *pu8_u    = (M4VIFI_UInt8)((i32_u00 + i32_u01 + i32_u10 + i32_u11 + 2) >> 2);
    134             *pu8_v    = (M4VIFI_UInt8)((i32_v00 + i32_v01 + i32_v10 + i32_v11 + 2) >> 2);
    135 
    136             pu8_rgbn    +=  (CST_RGB_24_SIZE<<1);
    137             pu8_yn        += 2;
    138             pu8_ys        += 2;
    139 
    140             pu8_u ++;
    141             pu8_v ++;
    142         } /* end of horizontal scanning */
    143 
    144         pu8_y_data        += u32_stride2_Y;
    145         pu8_u_data        += u32_stride_U;
    146         pu8_v_data        += u32_stride_V;
    147         pu8_rgbn_data    += u32_stride_2rgb;
    148 
    149 
    150     } /* End of vertical scanning */
    151 
    152     return M4VIFI_OK;
    153 }
    154