61 lines
1.7 KiB
C++
61 lines
1.7 KiB
C++
|
#include <precomp.h>
|
||
|
|
|
||
|
|
#include "convolve.h"
|
||
|
|
|
||
|
|
#define RED(a) (((a)>>16)&0xff)
|
||
|
|
#define GRN(a) (((a)>>8)&0xff)
|
||
|
|
#define BLU(a) (((a)&0xff))
|
||
|
|
#define ALP(a) (((a)>>24))
|
||
|
|
|
||
|
|
Convolve3x3::Convolve3x3(ARGB32 *_bits, int _w, int _h) : bits(_bits), w(_w), h(_h) {
|
||
|
|
ZERO(vals);
|
||
|
|
multiplier = 0;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Convolve3x3::setPos(int x, int y, float v) {
|
||
|
|
ASSERT(x >= -1 && x <= 1 && y >= -1 && y <= 1);
|
||
|
|
vals[y+1][x+1] = v;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Convolve3x3::setMultiplier(int m) {
|
||
|
|
multiplier = m;
|
||
|
|
}
|
||
|
|
|
||
|
|
void Convolve3x3::convolve() {
|
||
|
|
if (bits == NULL || w <= 0 || h <= 0) return; // nothin'
|
||
|
|
MemMatrix<ARGB32> temp(w, h, bits); // make a copy
|
||
|
|
for (int y = 0; y < h; y++) {
|
||
|
|
for (int x = 0; x < w; x++) {
|
||
|
|
if (ALP(temp(x, y))<=1) continue;
|
||
|
|
float ra=0, rg=0, rb=0;
|
||
|
|
for (int a = -1; a <= 1; a++) {
|
||
|
|
for (int b = -1; b <= 1; b++) {
|
||
|
|
int px = x + a, py = y + b;
|
||
|
|
if (px < 0 || px >= w || py < 0 || py >= h) continue;
|
||
|
|
ARGB32 c = temp(px, py);
|
||
|
|
if (ALP(c) <= 1) continue;
|
||
|
|
ra += (float)RED(c) * vals[b][a] * multiplier;
|
||
|
|
rg += (float)GRN(c) * vals[b][a] * multiplier;
|
||
|
|
rb += (float)BLU(c) * vals[b][a] * multiplier;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
unsigned int r = MINMAX((int)ra, 0, 255);
|
||
|
|
unsigned int g = MINMAX((int)rg, 0, 255);
|
||
|
|
unsigned int b = MINMAX((int)rb, 0, 255);
|
||
|
|
unsigned int lum = MAX(MAX(r, g), b);
|
||
|
|
if (lum < 64) lum = 0;
|
||
|
|
else if (lum > 192) lum = 255;
|
||
|
|
//bits[x+y*w] = (ALP(*temp.m(x, y))<<24)|(r<<16)|(g<<8)|(b);
|
||
|
|
bits[x+y*w] &= 0xffffff;
|
||
|
|
bits[x+y*w] |= (255-lum) << 24;
|
||
|
|
//if (lum < 64) {
|
||
|
|
// bits[x+y*w] &= 0xffff00ff;
|
||
|
|
// bits[x+y*w] |= lum << 8;
|
||
|
|
//} else {
|
||
|
|
// bits[x+y*w] &= 0xff00ffff;
|
||
|
|
// bits[x+y*w] |= lum << 16;
|
||
|
|
//}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|