add webassembly for client decoder

This commit is contained in:
Xuan Sang LE 2018-09-21 20:00:55 +02:00
parent 7e536d6c0b
commit 63028a15e5
6 changed files with 260 additions and 1 deletions

3
.gitignore vendored
View File

@ -1,5 +1,6 @@
# Prerequisites
plugins
libjpeg
zlib
build
*._*
*.d

View File

@ -9,6 +9,7 @@ PLUGINSDEP = $(OBJS) \
wvnc.o
DEP =
WEB_BUILD_PATH = /home/mrsang/myws/antd-web-apps/apps/assets/scripts
ifeq ($(UNAME_S),Darwin)
DEP= -I/usr/local/opt/jpeg-turbo/include -L/usr/local/opt/jpeg-turbo/lib
endif
@ -26,6 +27,12 @@ main: $(PLUGINSDEP) $(PLUGINS) #lib
-ln -s $(PBUILDIRD)/libantd.$(EXT) .
$(CC) $(PCFLAGS) $(PLUGINSDEP) $(PLUGINLIBS) -shared -o $(PBUILDIRD)/$(basename $@).$(EXT)
web:
emcc -o $(WEB_BUILD_PATH)/wvnc_asm.js -I wasm/libjpeg/ -I wasm/zlib wasm/decoder.c \
wasm/libjpeg/.libs/libjpeg.a wasm/zlib/libz.a \
-O3 -s ALLOW_MEMORY_GROWTH=1 -s WASM=1 -s NO_EXIT_RUNTIME=1 -s \
'EXTRA_EXPORTED_RUNTIME_METHODS=["cwrap"]'
clean: #libclean
-rm -f *.o *.$(EXT) $(PBUILDIRD)/$(PLUGINS)

213
wasm/decoder.c Normal file
View File

@ -0,0 +1,213 @@
#include <stdio.h>
#include <stdlib.h>
#include <stdint.h>
#include <zlib.h>
#include <jpeglib.h>
#include <string.h>
#include "emscripten.h"
/*
* This file is the decoder for the wvnc data
* received on the client side, it will be complied
* to web assembly to be used on the browser
*/
/*alloc a new frame buffer*/
typedef struct{
uint8_t r;
uint8_t g;
uint8_t b;
} raw_pixel_t;
EMSCRIPTEN_KEEPALIVE
uint8_t* create_buffer(int size) {
return malloc(size);
}
/*destroy buffer*/
EMSCRIPTEN_KEEPALIVE
void destroy_buffer(uint8_t* p) {
free(p);
}
int decode_zlib(uint8_t* in,int size_in, uint8_t* out, int size_out)
{
z_stream dstream;
dstream.zalloc = Z_NULL;
dstream.zfree = Z_NULL;
dstream.opaque = Z_NULL;
dstream.next_in = in;
dstream.avail_in = size_in;
dstream.avail_out = size_out;
dstream.next_out = out;
inflateInit(&dstream);
inflate(&dstream, Z_FINISH);
inflateEnd(&dstream);
return dstream.total_out;
}
int decode_jpeg(uint8_t* in,int size_in, uint8_t* out)
{
struct jpeg_decompress_struct cinfo = {0};
struct jpeg_error_mgr jerror = {0};
cinfo.err = jpeg_std_error(&jerror);
jpeg_create_decompress(&cinfo);
jpeg_mem_src(&cinfo, in, size_in);
// check if the jpeg is valid
int rc = jpeg_read_header(&cinfo, TRUE);
if (rc != 1) {
printf("Cannot read JPEG header");
return 0;
}
//cinfo.out_color_space = JCS_RGB;
//cinfo.dct_method = JDCT_ISLOW;
jpeg_start_decompress(&cinfo);
int width = cinfo.output_width;
int height = cinfo.output_height;
int osize = width*height*3;
int row_stride = width * 3;
//printf("width %d, height %d %d\n", width, height, cinfo.output_components);
uint8_t * tmp = (uint8_t*) malloc(osize);
uint8_t * line_buffer[1];
int index;
while( cinfo.output_scanline < height ){
line_buffer[0] = tmp +(cinfo.output_scanline) * row_stride;
jpeg_read_scanlines(&cinfo, line_buffer, 1);
}
jpeg_finish_decompress(&cinfo);
jpeg_destroy_decompress(&cinfo);
for(int j = 0; j < height; j++)
{
for(int i = 0; i < width; i++)
{
index = j*width*4 + i*4;
memcpy(out + index, tmp + j*width*3 + i*3, 3);
*(out + index + 3) = 255;
}
}
//copy
//memcpy(out, tmp, osize);
//free
free(tmp);
return osize;
}
int decode_raw(uint8_t* in,int size, int depth, uint8_t* out)
{
if(depth != 24 && depth != 16)
{
printf("Depth % d is not supported\n", depth);
return 0;
}
int components = depth/8;
int npixels = size / components;
int outsize = npixels*4;
unsigned int value = 0;
uint8_t* tmp = (uint8_t*) malloc(outsize);
raw_pixel_t px;
for(int i = 0; i < npixels; i++)
{
value = 0;
for(int j=0; j < components; j++ )
value = (in[i * components + j] << (j*8)) | value ;
// lookup for pixel
switch (depth)
{
case 24:
px.r = value & 0xFF;
px.g = (value >> 8) & 0xFF;
px.b = (value >> 16) & 0xFF;
break;
case 16:
px.r = (value & 0x1F) * (255 / 31);
px.g = ((value >> 5) & 0x3F) * (255 / 63);
px.b = ((value >> 11) & 0x1F) * (255 / 31);
break;
default:
break;
}
tmp[i*4] = px.r;
tmp[i*4+1] = px.g;
tmp[i*4+2] = px.b;
tmp[i*4+3] = 255;
}
memcpy(out, tmp, outsize);
free(tmp);
return outsize;
}
void update_buffer(uint8_t* fb, uint8_t* data, int x, int y, int w, int h, int bw)
{
//printf("update buffer\n");
//copy line by line
uint8_t *src_ptr, *dest_ptr;
for(int j = 0; j < h; j++)
{
src_ptr = data + (j*w*4);
dest_ptr = fb + ((j+y)*bw*4 + x*4);
memcpy(dest_ptr, src_ptr, w*4);
}
}
EMSCRIPTEN_KEEPALIVE
uint8_t* decode(uint8_t* data, int data_size, int depth, int size_out)
{
uint8_t flag = data[9];
//printf("x %d y %d w %d h %d flag %d\n",x,y,w,h,flag);
uint8_t* encoded_data = data + 10;
uint8_t* decoded_data = NULL;
int ret = 0;
if(flag == 0x0 && depth == 32)
{
return encoded_data;
}
else
{
decoded_data = (uint8_t*) malloc(size_out);
switch (flag)
{
case 0x0: // raw 24 or 16 bits
ret = decode_raw(encoded_data,data_size,depth, decoded_data);
break;
case 0x1: // jpeg data
ret = decode_jpeg(encoded_data, data_size, decoded_data);
break;
case 0x2: // zlib data
ret = decode_zlib(encoded_data, data_size, decoded_data, size_out);
if(ret > 0 && depth != 32)
ret = decode_raw(decoded_data,ret,depth, decoded_data);
break;
case 0x3: // jpeg and zlib data
ret = decode_zlib(encoded_data, data_size, decoded_data, size_out);
if(ret > 0)
ret = decode_jpeg(decoded_data, ret, decoded_data);
break;
default:
if(decoded_data) free(decoded_data);
return NULL;
}
// write decoded data to buffer
return decoded_data;
}
}
/*decode the data format*/
EMSCRIPTEN_KEEPALIVE
int update(uint8_t* fb, uint8_t* data, int data_size, int bw, int bh, int depth)
{
int x,y,w,h;
uint8_t flag;
// data [0] is commad
x = data[1] | (data[2] << 8);
y = data[3] | (data[4] << 8);
w = data[5] | (data[6] << 8);
h = data[7] | (data[8] << 8);
flag = data[9];
uint8_t* decoded_data = decode(data, data_size, depth, w*h*4);
if(decoded_data)
update_buffer(fb,decoded_data,x,y,w,h,bw);
if(flag != 0x0 || depth != 32) free(decoded_data);
else
printf("Cannot decode data\n");
return 1;
}

34
wasm/dep.sh Executable file
View File

@ -0,0 +1,34 @@
#! /bin/bash
# Make sure the emsdk is installed and worked
echo "Preparing libjpeg..."
if [ -d "libjpeg" ]; then
rm -r libjpeg
fi
if [ -d "zlib" ]; then
rm -r zlib
fi
wget http://www.ijg.org/files/jpegsrc.v9c.tar.gz
mkdir libjpeg
tar xvzf jpegsrc.v9c.tar.gz -C ./libjpeg --strip-components=1
rm jpegsrc.v9c.tar.gz
cd libjpeg
emconfigure ./configure
emmake make
## Using libjpeg-turbo
#git clone https://github.com/libjpeg-turbo/libjpeg-turbo
#mv libjpeg-turbo libjpeg
#cd libjpeg
#mkdir build
#cd build
#emcmake cmake ../
#emmake make
cd ../
echo "Preparinf Zlib..."
wget https://zlib.net/zlib-1.2.11.tar.gz
mkdir zlib
tar xvzf zlib-1.2.11.tar.gz -C ./zlib --strip-components=1
rm zlib-1.2.11.tar.gz
cd zlib
emconfigure ./configure
# TODO modify make file using sed if macos
emmake make

4
wasm/wvnc_asm.js Normal file

File diff suppressed because one or more lines are too long

BIN
wasm/wvnc_asm.wasm Normal file

Binary file not shown.