Ignore:
Timestamp:
May 5, 2011, 12:36:33 PM (11 years ago)
Author:
Sam Hocevar
Message:

imlib: refactor the Filter and ColorFilter? classes.

File:
1 edited

Legend:

Unmodified
Added
Removed
  • abuse/trunk/src/imlib/filter.cpp

    r555 r579  
    1818#include "filter.h"
    1919
    20 filter::filter(palette *from, palette *to)   // creates a conversion filter from one palette to another
     20Filter::Filter(int colors)
    2121{
    22   nc=from->pal_size() > to->pal_size() ? from->pal_size() : to->pal_size();
    23   unsigned char *p=fdat=(unsigned char *)malloc(nc);
    24   unsigned char *r,*g,*b;
    25   r=g=b=(unsigned char *)from->addr();
    26   g++;
    27   b+=2;
    28 
    29   int dk=to->darkest(1);
    30   for (int i=0; i<nc; i++,p++,r+=3,g+=3,b+=3)
    31   {
    32     *p=to->find_closest(*r,*g,*b);
    33 
    34     // make sure non-blacks don't get remapped to the transparency
    35     if ((*r!=0 || *g!=0 || *b!=0) && (to->red(*p)==0 && to->green(*p)==0 && to->blue(*p)==0))
    36       *p=dk;
    37   }
    38 
     22    CONDITION(colors >= 0 && colors <= 256, "bad colors value");
     23    m_size = colors;
     24    m_table = (uint8_t *)malloc(m_size);
     25    memset(m_table, 0, m_size * sizeof(*m_table));
    3926}
    4027
    41 void filter::clear()
     28// Creates a conversion filter from one palette to another
     29Filter::Filter(palette *from, palette *to)
    4230{
    43   int i;
    44   for (i=0; i<nc; i++)
    45     fdat[i]=i;
     31    m_size = Max(from->pal_size(), to->pal_size());
     32    m_table = (uint8_t *)malloc(m_size);
     33
     34    uint8_t *dst = m_table;
     35    uint8_t *src = (uint8_t *)from->addr();
     36    int dk = to->darkest(1);
     37
     38    for (int i = 0; i < m_size; i++)
     39    {
     40       int r = *src++;
     41       int g = *src++;
     42       int b = *src++;
     43       int color = to->find_closest(r, g, b);
     44
     45       // Make sure non-blacks don't get remapped to the transparency
     46       if ((r || g || b) && to->red(color) == 0
     47            && to->green(color) == 0 && to->blue(color) == 0)
     48           color = dk;
     49
     50       *dst++ = color;
     51    }
    4652}
    4753
    48 void filter::max_threshold(int minv, char blank)
     54Filter::~Filter()
    4955{
    50   int i;
    51   CONDITION(minv>=0 && minv<nc,"Bad minv");
    52   for (i=0; i<minv; i++)
    53     fdat[i]=blank;
     56    free(m_table);
    5457}
    5558
    56 void filter::min_threshold(int maxv, char blank)
     59void Filter::Set(int color_num, int change_to)
    5760{
    58   int i;
    59   CONDITION(maxv>=0 && maxv<nc,"bad maxv value in filter::max_thresh");
    60   for (i=nc-1; i>=maxv; i--)
    61     fdat[i]=(unsigned) blank;
     61    CONDITION(color_num >= 0 && color_num < m_size, "Bad colors_num");
     62    m_table[color_num] = change_to;
    6263}
    6364
    64 
    65 void filter::set(int color_num, char change_to)
     65void Filter::Apply(image *im)
    6666{
    67   CONDITION(color_num>=0 && color_num<nc,"Bad colors_num");
    68   fdat[color_num]=(unsigned) change_to;
     67    im->Lock();
     68    uint8_t *dst = im->scan_line(0);
     69    int npixels = im->Size().x * im->Size().y;
     70    while (npixels--)
     71    {
     72        CONDITION(*dst < m_size, "not enough filter colors");
     73        *dst = m_table[*dst];
     74        dst++;
     75    }
     76    im->Unlock();
    6977}
    7078
    71 
    72 filter::filter(int colors)
    73 {
    74   CONDITION(colors>=0 && colors<=256,"bad colors value");
    75   nc=colors;
    76   fdat=(unsigned char *)malloc(nc);
    77   clear();
    78 }
    79 
    80 void filter::apply(image *im)
    81 {
    82   int x,y;
    83   unsigned char *c;
    84   CONDITION(im,"null image passed in filter::apply\n");
    85   im->Lock();
    86   for (y=im->Size().y-1; y>=0; y--)
    87   {
    88     c=im->scan_line(y);
    89     for (x=im->Size().x-1; x>=0; x--)
    90     {
    91       CONDITION((unsigned) c[x]<nc,"not enough filter colors");
    92       c[x]=fdat[(unsigned) c[x]];
    93     }
    94   }
    95   im->Unlock();
    96 }
    97 
    98 
    99 palette *compare_pal;
    100 
    101 int color_compare(void *c1, void *c2)
    102 {
    103   long v1,v2;
    104   unsigned char r1,g1,b1,r2,g2,b2;
    105   compare_pal->get(  *((unsigned char *)c1),r1,g1,b1);
    106   compare_pal->get(  *((unsigned char *)c2),r2,g2,b2);
    107   v1=(int)r1*(int)r1+(int)g1*(int)g1+(int)b1*(int)b1;
    108   v2=(int)r2*(int)r2+(int)g2*(int)g2+(int)b2*(int)b2;
    109   if (v1<v2) return -1;
    110   else if (v1>v2) return 1;
    111   else return 0;
    112 }
    113 
    114 color_filter::color_filter(palette *pal, int color_bits, void (*stat_fun)(int))
    115 {
    116   color_bits=5;      // hard code 5 for now
    117   int r,g,b,rv,gv,bv,
    118       c=0,i,max=pal->pal_size(),
    119       lshift=8-color_bits;
    120   unsigned char *pp;
    121 
    122   long dist_sqr,best;
    123   int colors=1<<color_bits;
    124   color_table=(unsigned char *)malloc(colors*colors*colors);
    125   for (r=0; r<colors; r++)
    126   {
    127     if (stat_fun) stat_fun(r);
    128     rv=r<<lshift;
    129     for (g=0; g<colors; g++)
    130     {
    131       gv=g<<lshift;
    132       for (b=0; b<colors; b++)
    133       {
    134     bv=b<<lshift;
    135         best=0x7fffffff;
    136         for (i=0,pp=(unsigned char *)pal->addr(); i<max; i++)
    137         {
    138           register long rd=*(pp++)-rv,
    139                         gd=*(pp++)-gv,
    140                         bd=*(pp++)-bv;
    141 
    142           dist_sqr=(long)rd*rd+(long)bd*bd+(long)gd*gd;
    143           if (dist_sqr<best)
    144           { best=dist_sqr;
    145             c=i;
    146           }
    147         }
    148         color_table[r*colors*colors+g*colors+b]=c;
    149       }
    150     }
    151   }
    152 }
    153 
    154 color_filter::color_filter(spec_entry *e, bFILE *fp)
    155 {
    156   fp->seek(e->offset,0);
    157   fp->read_uint16();
    158   int colors=32;
    159   color_table=(unsigned char *)malloc(colors*colors*colors);
    160   fp->read(color_table,colors*colors*colors);
    161 }
    162 
    163 int color_filter::size()
    164 {
    165   int colors=32;
    166   return 2+colors*colors*colors;
    167 }
    168 
    169 int color_filter::write(bFILE *fp)
    170 {
    171   int colors=32;
    172   fp->write_uint16(colors);
    173   return fp->write(color_table,colors*colors*colors)==colors*colors*colors;
    174 }
    175 
    176 
    177 void filter::put_image(image *screen, image *im, short x, short y,
    178                        char transparent)
     79/* This is only ever used in the editor, when showing the toolbar. It
     80 * does not look like it's very useful. */
     81void Filter::PutImage(image *screen, image *im, vec2i pos)
    17982{
    18083    int cx1, cy1, cx2, cy2, x1 = 0, y1 = 0,
    181           x2 = im->Size().x, y2 = im->Size().y;
     84        x2 = im->Size().x, y2 = im->Size().y;
    18285    screen->GetClip(cx1, cy1, cx2, cy2);
    18386
    184     // see if the image gets clipped off the screen
    185     if(x >= cx2 || y >= cy2 || x + (x2 - x1) <= cx1 || y + (y2 - y1) <= cy1)
     87    // See if the image gets clipped off the screen
     88    if(pos.x >= cx2 || pos.y >= cy2 ||
     89       pos.x + (x2 - x1) <= cx1 || pos.y + (y2 - y1) <= cy1)
    18690        return;
    18791
    188     if(x < cx1)
    189     {
    190         x1 += (cx1 - x);
    191         x = cx1;
    192     }
    193     if(y < cy1)
    194     {
    195         y1 += (cy1 - y);
    196          y = cy1;
    197     }
    198 
    199     if(x + x2 - x1 >= cx2)
    200         x2 = cx2 - x + x1;
    201 
    202     if(y + y2 - y1 >= cy2)
    203         y2 = cy2 - y + y1;
     92    x1 += Max(cx1 - pos.x, 0);
     93    y1 += Max(cy1 - pos.y, 0);
     94    pos.x = Max(pos.x, cx1);
     95    pos.y = Max(pos.y, cy1);
     96    x2 = Min(x2, cx2 - pos.x + x1);
     97    y2 = Min(y2, cy2 - pos.y + y1);
    20498
    20599    if(x1 >= x2 || y1 >= y2)
     
    209103    int yl = y2 - y1;
    210104
    211     screen->AddDirty(x, y, x + xl, y + yl);
     105    screen->AddDirty(pos.x, pos.y, pos.x + xl, pos.y + yl);
    212106
    213107    screen->Lock();
    214108    im->Lock();
    215109
    216     uint8_t *pg1 = screen->scan_line(y), *source, *dest;
    217     uint8_t *pg2 = im->scan_line(y1);
    218     int i;
    219110    for(int j = 0; j < yl; j++)
    220111    {
    221         for(i = 0, source = &pg2[x1], dest = &pg1[x];
    222             i < xl;
    223             i++, source++, dest++)
    224         {
    225             if (!transparent || *source)
    226                 *dest=fdat[*source];
    227         }
    228         pg1 = screen->next_line(y + j, pg1);
    229         pg2 = im->next_line(y1 + j, pg2);
     112        uint8_t *source = im->scan_line(y1 + j) + x1;
     113        uint8_t *dest = screen->scan_line(pos.y + j) + pos.x;
     114
     115        for(int i = 0; i < xl; i++, source++, dest++)
     116            if (*source)
     117                *dest = m_table[*source];
    230118    }
    231119
     
    234122}
    235123
     124ColorFilter::ColorFilter(palette *pal, int color_bits)
     125{
     126    int max = pal->pal_size();
     127    int mul = 1 << (8 - color_bits);
     128    m_size = 1 << color_bits;
     129    m_table = (uint8_t *)malloc(m_size * m_size * m_size);
     130
     131    /* For each colour in the RGB cube, find the nearest palette element. */
     132    for (int r = 0; r < m_size; r++)
     133    for (int g = 0; g < m_size; g++)
     134    for (int b = 0; b < m_size; b++)
     135    {
     136        int best = 256 * 256 * 3;
     137        int color = 0;
     138        uint8_t *pp = (uint8_t *)pal->addr();
     139
     140        for (int i = 0; i < max; i++)
     141        {
     142            int rd = *pp++ - r * mul,
     143                gd = *pp++ - g * mul,
     144                bd = *pp++ - b * mul;
     145
     146            int dist = rd * rd + bd * bd + gd * gd;
     147            if (dist < best)
     148            {
     149                best = dist;
     150                color = i;
     151            }
     152        }
     153        m_table[(r * m_size + g) * m_size + b] = color;
     154    }
     155}
     156
     157ColorFilter::ColorFilter(spec_entry *e, bFILE *fp)
     158{
     159    fp->seek(e->offset, 0);
     160    m_size = fp->read_uint16();
     161    m_table = (uint8_t *)malloc(m_size * m_size * m_size);
     162    fp->read(m_table, m_size * m_size * m_size);
     163}
     164
     165ColorFilter::~ColorFilter()
     166{
     167    free(m_table);
     168}
     169
     170size_t ColorFilter::DiskUsage()
     171{
     172    return sizeof(uint16_t) + m_size * m_size * m_size;
     173}
     174
     175int ColorFilter::Write(bFILE *fp)
     176{
     177    fp->write_uint16(m_size);
     178    int bytes = m_size * m_size * m_size;
     179    return fp->write(m_table, bytes) == bytes;
     180}
     181
Note: See TracChangeset for help on using the changeset viewer.