@@ -1,7 +1,10 @@ -CFLAGS= -Wunused -O3 -funroll-loops -fno-rtti -fomit-frame-pointer -ffast-math -fno-stack-protector -ffunction-sections -LFLAGS= -lpthread -lX11 +CFLAGS= -w -O3 -funroll-loops -fno-rtti -fomit-frame-pointer -ffast-math -fno-stack-protector -ffunction-sections +LFLAGS= all: g++ *.cpp -o kerbyd $(CFLAGS) $(LFLAGS) +install: + #For testing purpose only, do not run this ! + scp -r ../kerby root@ganryu:/root clean: rm -f *.o kerbyd Makefile @@ -20,119 +20,62 @@ ***************************************************************************/ #include "analyzer.h" -void Analyzer::convert( kb_video_buffer_t *frame, CImg<byte> *image ){ - int x,y; - - for( x = 0; x < frame->width; x++ ){ - for( y = 0; y < frame->height; y++ ){ - *image->ptr( x, y, 0 ) = frame->data[ (y * frame->width + x) * frame->depth + 0 ]; - *image->ptr( x, y, 1 ) = frame->data[ (y * frame->width + x) * frame->depth + 1 ]; - *image->ptr( x, y, 2 ) = frame->data[ (y * frame->width + x) * frame->depth + 2 ]; - } - } -} - -void Analyzer::convert( kb_video_buffer_t* frame ){ - this->convert( frame, this->m_masterframe ); -} - -CImg<byte> Analyzer::quadrant( CImg<byte> *image, int xt, int yt, int xb, int yb ){ - CImg<byte> quad( xb - xt, yb - yt, 1, 3 ); - - int xq,yq,xi,yi; - - for( xi = xt, xq = 0; xi < xb; xi++, xq++ ){ - for( yi = yt, yq = 0; yi < yb; yi++, yq++ ){ - *quad.ptr( xq, yq, 0 ) = *image->ptr( xi, yi, 0 ); - *quad.ptr( xq, yq, 1 ) = *image->ptr( xi, yi, 1 ); - *quad.ptr( xq, yq, 2 ) = *image->ptr( xi, yi, 2 ); - } - } - - return quad; -} - -CImg<byte> Analyzer::quadrant( int xt, int yt, int xb, int yb ){ - return this->quadrant( this->m_masterframe, xt, yt, xb, yb ); -} - -double Analyzer::delta( short total, short value ){ - double a = total, - b = value; - - if( total < value ){ - a = value; - b = total; - } - - return ((a - b) / b) * 100.0f; -} - -double Analyzer::delta( CImg<byte> *quadrant, int xt, int yt, int xb, int yb ){ - CImg<byte> masterquadrant = this->quadrant( xt, yt, xb, yb ); - double delta = 0.0f; - - int h = yb - yt, - w = xb - xt, - y, x; - - double rdelta,gdelta,bdelta; - - for( y = 0; y < h; y++ ){ - for( x = 0; x < w; x++ ){ - rdelta = this->delta( *masterquadrant.ptr( x, y, 0 ), *quadrant->ptr( x, y, 0 ) ); - gdelta = this->delta( *masterquadrant.ptr( x, y, 1 ), *quadrant->ptr( x, y, 1 ) ); - bdelta = this->delta( *masterquadrant.ptr( x, y, 2 ), *quadrant->ptr( x, y, 2 ) ); - - delta += ((rdelta + gdelta + bdelta) / 3.0f); - } - } - - return (delta / (w * h)); -} - -Analyzer::Analyzer( uint width, uint height, uint depth, uint quadrant_size, double threshold ){ - this->m_masterframe = new CImg<byte>( width, height, 1, depth ); - this->m_frame = new CImg<byte>( width, height, 1, depth ); - this->m_quadrant_size = quadrant_size; - this->m_threshold = threshold; - this->m_quadrants = (width * height) / this->m_quadrant_size; +Analyzer::Analyzer( kb_video_driver_t *driver, uint width, uint height, uint depth, uint quadrant_size, double threshold ){ + m_driver = driver; + m_masterframe = driver->create( width, height, depth ); + m_quadrant_size = quadrant_size; + m_threshold = threshold; + m_quadrants = (width * height) / this->m_quadrant_size; } Analyzer::~Analyzer(){ - delete this->m_masterframe; - delete this->m_frame; + free( m_masterframe->data ); + free( m_masterframe ); } double Analyzer::update( kb_video_buffer_t *frame ){ - CImg<byte> iframe( frame->width, frame->height, 1, frame->depth ), - quadrant; - int x, y, xt, yt, xb, yb, count = 0; - double qdelta; - - this->convert( frame, &iframe ); + int x, y, // main coordinates + xt, yt, xb, yb, // top-x, top-y, bottom-x and bottom-x of a quadrant + xq, yq, // quadrants coordinates + count = 0; // relevant quadrants counter - this->m_frame->swap(iframe); + kb_rgb_t mpixel, // master pixel + fpixel; // frame pixel - for( x = 0; x < iframe.width; x += this->m_quadrant_size ){ - for( y = 0; y < iframe.height; y += this->m_quadrant_size ){ + double rd, gd, bd, // red, green and blue delta values + qdelta; // quadrant delta value + + for( x = 0; x < frame->width; x += this->m_quadrant_size ){ + for( y = 0; y < frame->height; y += this->m_quadrant_size ){ xt = x; yt = y; xb = x + this->m_quadrant_size; yb = y + this->m_quadrant_size; - quadrant = this->quadrant( &iframe, xt, yt, xb, yb ); - - qdelta = this->delta( &quadrant, xt, yt, xb, yb ); - + // compute delta value between frame and m_masterframe (the previous frame) + qdelta = 0; + for( xq = xt; xq < xb; xq++ ){ + for( yq = yt; yq < yb; yq++ ){ + // acquire pixels + m_driver->pixel( m_masterframe, xq, yq, &mpixel ); + m_driver->pixel( frame, xq, yq, &fpixel ); + // compute delta + rd = this->delta( mpixel.red, fpixel.red ); + gd = this->delta( mpixel.green, fpixel.green ); + bd = this->delta( mpixel.blue, fpixel.blue ); + + qdelta += ((rd + gd + bd) / 3.0f); + } + } + qdelta /= m_quadrants; + if( qdelta >= this->m_threshold ){ count++; } } } - this->m_masterframe->swap(iframe); - - - return (count * 100.0f) / (double)this->m_quadrants; + memcpy( m_masterframe->data, frame->data, frame->size ); + + return (count * 100.0f) / (double)m_quadrants; } analyzer.cpp @@ -22,57 +22,32 @@ # define __analyzer_h__ #include "driver.h" -#include <vector> - -#include <CImg.h> - -using std::vector; -using namespace cimg_library; - -typedef struct _kb_quadrant{ - int xt; - int yt; - int xb; - int yb; - - double delta; - - _kb_quadrant( int xt, int yt, int xb, int yb, double delta ){ - this->xt = xt; - this->yt = yt; - this->xb = xb; - this->yb = yb; - this->delta = delta; - } -} -kb_quadrant_t; - -typedef vector<kb_quadrant_t *> kb_delta_quadrants_t; class Analyzer { private : - - double m_threshold; - - uint m_quadrant_size; - uint m_quadrants; - - CImg<byte> * m_masterframe; - CImg<byte> * m_frame; + kb_video_driver_t *m_driver; + + double m_threshold; - void convert( kb_video_buffer_t *frame, CImg<byte> *image ); - void convert( kb_video_buffer_t *frame ); + uint m_quadrant_size; + uint m_quadrants; - CImg<byte> quadrant( CImg<byte> *image, int xt, int yt, int xb, int yb ); - CImg<byte> quadrant( int xt, int yt, int xb, int yb ); + kb_video_buffer_t *m_masterframe; - double delta( short total, short value ); - double delta( CImg<byte> *quadrant, int xt, int yt, int xb, int yb ); - + inline double delta( short total, short value ){ + double a = total, + b = value; + if( total < value ){ + a = value; + b = total; + } + return (b > 0 ? (((a - b) / b) * 100.0f) : 0); + } + public : - Analyzer( uint width, uint height, uint depth, uint quadrant_size, double threshold ); + Analyzer( kb_video_driver_t *driver, uint width, uint height, uint depth, uint quadrant_size, double threshold ); ~Analyzer(); double update( kb_video_buffer_t *frame ); analyzer.h @@ -77,7 +77,7 @@ int v4l_capture( kb_device_t * dev, kb_video_buffer_t * vbuffer ){ perror("VIDIOCSYNC"); return -1; } - + vbuffer->data = dev->mmap; return 1; @@ -87,3 +87,27 @@ void v4l_close( kb_device_t * dev ){ munmap( dev->mmap, dev->width * dev->height * dev->depth ); close( dev->fd); } + +void v4l_pixel( kb_video_buffer_t * vbuffer, uint x, uint y, kb_rgb_t *pixel ){ + uint offset = (y * vbuffer->width + x) * vbuffer->depth; + + pixel->red = vbuffer->data[ offset + 0 ]; + pixel->green = vbuffer->data[ offset + 1 ]; + pixel->blue = vbuffer->data[ offset + 2 ]; +} + +void v4l_swap( kb_video_buffer_t * src, kb_video_buffer_t * dst ){ + memcpy( dst->data, src->data, dst->size ); +} + +kb_video_buffer_t * v4l_create( uint width, uint height, uint depth ){ + kb_video_buffer_t *buffer = (kb_video_buffer_t *)malloc(sizeof(kb_video_buffer_t)); + + buffer->width = width; + buffer->height = height; + buffer->depth = depth; + buffer->size = width * height * depth; + buffer->data = (byte *)malloc( buffer->size ); + + return buffer; +} driver.cpp @@ -32,6 +32,7 @@ #include <signal.h> #include <malloc.h> #include <linux/videodev.h> +#include <stdlib.h> typedef unsigned char byte; typedef unsigned int uint; @@ -57,9 +58,19 @@ typedef struct{ } kb_video_buffer_t; -typedef int (* kb_open_t) ( char *devname, kb_device_t * dev ); -typedef int (* kb_capture_t)( kb_device_t * dev, kb_video_buffer_t * vbuffer ); -typedef void (* kb_close_t) ( kb_device_t * dev ); +typedef struct{ + byte red; + byte green; + byte blue; +} +kb_rgb_t; + +typedef int (* kb_open_t) ( char *devname, kb_device_t * dev ); +typedef int (* kb_capture_t) ( kb_device_t * dev, kb_video_buffer_t * vbuffer ); +typedef void (* kb_close_t) ( kb_device_t * dev ); +typedef void (* kb_pixel_t) ( kb_video_buffer_t * vbuffer, uint x, uint y, kb_rgb_t *pixel ); +typedef void (* kb_swap_t) ( kb_video_buffer_t * src, kb_video_buffer_t * dst ); +typedef kb_video_buffer_t * (* kb_create_t) ( uint width, uint height, uint depth ); typedef struct{ char name[0xFF]; @@ -67,16 +78,22 @@ typedef struct{ kb_open_t open; kb_capture_t capture; kb_close_t close; + kb_pixel_t pixel; + kb_swap_t swap; + kb_create_t create; } kb_video_driver_t; -int v4l_open ( char *devname, kb_device_t * dev ); -int v4l_capture( kb_device_t * dev, kb_video_buffer_t * vbuffer ); -void v4l_close ( kb_device_t * dev ); +int v4l_open ( char *devname, kb_device_t * dev ); +int v4l_capture ( kb_device_t * dev, kb_video_buffer_t * vbuffer ); +void v4l_close ( kb_device_t * dev ); +void v4l_pixel ( kb_video_buffer_t * vbuffer, uint x, uint y, kb_rgb_t *pixel ); +void v4l_swap ( kb_video_buffer_t * src, kb_video_buffer_t * dst ); +kb_video_buffer_t * v4l_create ( uint width, uint height, uint depth ); static const kb_video_driver_t kb_video_drivers[] = { - { "Video 4 Linux v1.0", v4l_open, v4l_capture, v4l_close } + { "Video 4 Linux v1.0", v4l_open, v4l_capture, v4l_close, v4l_pixel, v4l_swap, v4l_create } }; #define SSVIDEODRIVERSN sizeof(kb_video_drivers) / sizeof(kb_video_drivers[0]) driver.h @@ -20,8 +20,12 @@ ***************************************************************************/ #include "analyzer.h" +// TODO: Those values should be read from a configuration file . #define QUADRANTSIZE 8 -#define THRESHOLD 5.0f +#define THRESHOLD 0.2f +#define RECORDPATH "/media/pendrive/kerbyd" + +#define DEBUG 1 void usage( char *app ){ printf( "\nUsage %s -d <device> -t <threshold> -q <quadrantsize> -a <alarm percentage>\n\n", app ); @@ -33,26 +37,30 @@ int main( int argc, char *argv[] ){ kb_video_buffer_t frame; Analyzer *analyzer; - v4l->open( argv[1], &device ); - - analyzer = new Analyzer( device.width, device.height, 3, QUADRANTSIZE, THRESHOLD ); + if( v4l->open( argv[1], &device ) < 0 ){ + exit(-1); + } - printf( "Device :\n" - "\tfile descriptor : %d\n" - "\tmemory map : 0x%X\n" - "\tdevice : %s\n" - "\tname : %s\n" - "\tmax width : %d\n" - "\tmax height : %d\n" - "\tdepth : %d\n", - device.fd, - device.mmap, - device.device, - device.name, - device.width, - device.height, - device.depth ); - + analyzer = new Analyzer( v4l, device.width, device.height, 3, QUADRANTSIZE, THRESHOLD ); + + #ifdef DEBUG + printf( "Device :\n" + "\tfile descriptor : %d\n" + "\tmemory map : 0x%X\n" + "\tdevice : %s\n" + "\tname : %s\n" + "\tmax width : %d\n" + "\tmax height : %d\n" + "\tdepth : %d\n", + device.fd, + device.mmap, + device.device, + device.name, + device.width, + device.height, + device.depth ); + #endif + double delta; while( 1 ){ @@ -60,7 +68,7 @@ int main( int argc, char *argv[] ){ delta = analyzer->update( &frame ); - printf( "DELTA : %f%%\n", delta ); + printf( "DELTA : %f\n", delta ); } v4l->close( &device ); main.cpp b1c589a910f8e578b666c0ae9b51d43d643b6017 evilsocket evilsocket@gmail.com http://github.com/evilsocket/Kerby/commit/813f067a4df6ddd2044d888daae313cbe40e734b 813f067a4df6ddd2044d888daae313cbe40e734b 2009-06-17T10:17:48-07:00 2009-06-17T10:17:48-07:00 General optimization / removed dependencies to CImg, libx11, libpthread, etc ee1e8a99059ca3254967e2b5333f9c7d334b6d13 evilsocket evilsocket@gmail.com