git.sesse.net Git - ffmpeg/blob - postproc/rgb2rgb.c

   1 #include <inttypes.h>
   2 #include "../config.h"
   3 #include "rgb2rgb.h"
   4 #include "mmx.h"
   5
   6 #ifdef HAVE_3DNOW
   7 #define PREFETCH "prefetch"
   8 #define PREFETCHW "prefetchw"
   9 #elif HAVE_MMX2
  10 #define PREFETCH "prefetchnta"
  11 #define PREFETCHW "prefetcht0"
  12 #endif
  13
  14 #ifdef HAVE_3DNOW
  15 #define EMMS "femms"
  16 #else
  17 #define EMMS "emms"
  18 #endif
  19
  20 #ifdef HAVE_MMX2
  21 #define MOVNTQ "movntq"
  22 #else
  23 #define MOVNTQ "movq"
  24 #endif
  25
  26 void rgb24to32(uint8_t *src,uint8_t *dst,uint32_t src_size)
  27 {
  28   uint8_t *dest = dst;
  29   uint8_t *s = src;
  30   uint8_t *end;
  31 #ifdef HAVE_MMX
  32   const uint64_t mask32 = 0x00FFFFFF00FFFFFFULL;
  33   uint8_t *mm_end;
  34 #endif
  35   end = s + src_size;
  36 #ifdef HAVE_MMX
  37 #ifdef PREFETCH
  38   __asm __volatile(
  39     PREFETCH" %0\n\t"
  40     PREFETCH" 64%0\n\t"
  41     PREFETCHW" %1\n\t"
  42     PREFETCHW" 64%1\n\t"::"m"(*s),"m"(*dest):"memory");
  43 #endif
  44   mm_end = (uint8_t*)((((unsigned long)end)/16)*16);
  45   __asm __volatile("movq %0, %%mm7"::"m"(mask32):"memory");
  46   while(s < mm_end)
  47   {
  48 #ifdef PREFETCH
  49     __asm __volatile(
  50         PREFETCH" 128%0\n\t"
  51         PREFETCHW" 128%1"
  52         ::"m"(*s),"m"(*dest):"memory");
  53 #endif
  54     __asm __volatile(
  55         "movd   %1, %%mm0\n\t"
  56         "movd   3%1, %%mm1\n\t"
  57         "movd   6%1, %%mm2\n\t"
  58         "movd   9%1, %%mm3\n\t"
  59         "punpckldq %%mm1, %%mm0\n\t"
  60         "punpckldq %%mm3, %%mm2\n\t"
  61         "pand   %%mm7, %%mm0\n\t"
  62         "pand   %%mm7, %%mm2\n\t"
  63         MOVNTQ" %%mm0, %0\n\t"
  64         MOVNTQ" %%mm2, 8%0"
  65         :"=m"(*dest)
  66         :"m"(*s)
  67         :"memory");
  68     dest += 16;
  69     s += 12;
  70   }
  71   __asm __volatile(EMMS:::"memory");
  72 #endif
  73   while(s < end)
  74   {
  75     *dest++ = *s++;
  76     *dest++ = *s++;
  77     *dest++ = *s++;
  78     *dest++ = 0;
  79   }
  80 }
  81
  82 /* TODO: MMX optimization */
  83 void rgb32to24(uint8_t *src,uint8_t *dst,uint32_t src_size)
  84 {
  85   uint8_t *dest = dst;
  86   uint8_t *s = src;
  87   uint8_t *end;
  88   end = s + src_size;
  89   while(s < end)
  90   {
  91     *dest++ = *s++;
  92     *dest++ = *s++;
  93     *dest++ = *s++;
  94     s++;
  95   }
  96 }
  97
  98 /* Original by Strepto/Astral
  99  ported to gcc & bugfixed : A'rpi */
 100 void rgb15to16(uint8_t *src,uint8_t *dst,uint32_t src_size)
 101 {
 102 #ifdef HAVE_MMX
 103   static uint64_t mask_b  = 0x001F001F001F001FLL; // 00000000 00011111  xxB
 104   static uint64_t mask_rg = 0x7FE07FE07FE07FE0LL; // 01111111 11100000  RGx
 105   register char* s=src+src_size;
 106   register char* d=dst+src_size;
 107   register int offs=-src_size;
 108   movq_m2r (mask_b,  mm4);
 109   movq_m2r (mask_rg, mm5);
 110   while(offs<0){
 111     movq_m2r (*(s+offs), mm0);
 112     movq_r2r (mm0, mm1);
 113
 114     movq_m2r (*(s+8+offs), mm2);
 115     movq_r2r (mm2, mm3);
 116
 117     pand_r2r (mm4, mm0);
 118     pand_r2r (mm5, mm1);
 119
 120     psllq_i2r(1,mm1);
 121     pand_r2r (mm4, mm2);
 122
 123     pand_r2r (mm5, mm3);
 124     por_r2r  (mm1, mm0);
 125
 126     psllq_i2r(1,mm3);
 127     movq_r2m (mm0,*(d+offs));
 128
 129     por_r2r  (mm3,mm2);
 130     movq_r2m (mm2,*(d+8+offs));
 131
 132     offs+=16;
 133   }
 134   emms();
 135 #else
 136    uint16_t *s1=( uint16_t * )src;
 137    uint16_t *d1=( uint16_t * )dst;
 138    uint16_t *e=((uint8_t *)s1)+src_size;
 139    while( s1<e ){
 140      register int x=*( s1++ );
 141      /* rrrrrggggggbbbbb
 142         0rrrrrgggggbbbbb
 143         0111 1111 1110 0000=0x7FE0
 144         00000000000001 1111=0x001F */
 145      *( d1++ )=( x&0x001F )|( ( x&0x7FE0 )<<1 );
 146    }
 147 #endif
 148 }