A theme color extracting library implemented by C.
This library is used in a related company for years.
TODO: Using KD-Tree to find some colors.
่ ็ฝ่ฎ๏ผใฒใใ ใณใใใใ๏ผHiziri Byakuren๏ผๆฏ็ณปๅไฝๅใไธๆนprojectใไธญ็่ง่ฒ๏ผ้ฆๆฌก็ปๅบไบใไธๆนๆ่ฒ่นใใ
- ็งๆ๏ผ้ญๆณไฝฟ
- ่ฝๅ๏ผไฝฟ็จ้ญๆณ็จๅบฆ็่ฝๅ๏ผ้ ฃ็ ๅขๅผบ่บซไฝ่ฝๅ็๏ผ
- ๅฑ้ฉๅบฆ๏ผไธๆ
- ไบบ็ฑปๅๅฅฝๅบฆ๏ผไธญ
- ไธป่ฆๆดปๅจๅบๆ๏ผๅฝ่ฒๅฏบไน็ฑป
ๅฝ่ฒๅฏบ็ไฝๆใ่ฝ็ถๅๆฌๆฏไบบ็ฑป๏ผไธ่ฟ็ฑไบๅธธๅนด็ไฟฎ่กๅทฒ็ปๅฎๅ จ่ถ ่ถไบไบบ็ฑปใ็ฐๅจๅทฒ็ปๅฑไบไบบไปฌๅธธ่ฏด็้ญๆณไฝฟไบใ
่ฝ็ถๅทฒ็ปๅ ฅไบไฝ้จ๏ผไฝๆฏไธ็ฅ้ไปไนๅๅ ๅด่ขซๅฆๆชๆฌไปฐ็ใๅฅนไปๆฅๆฒกๆๅ็ซฅ่ฏๆ ไบไธญ็้ญๆณไฝฟ้ฃๆ ท๏ผๅฟต่ฏต็ๅ่ฏญๆฒป้ๅฆๆชใไฝฟ็จ็ๅ้ๅฎๅ จๆฏ้ชๆถ็๏ผไธ็น้ฝไธๅๆฏๅฃไบบ๏ผ่ฝ็ถๅนถๆฒกๆไบบ็ฎๅปๅฐๅฅนไธไบบ็ฑปไธบๆ๏ผไฝๅ ถๅฎๅทฒๅฝปๅบๆไธบๅฆๆช็ๅไผดไบใ
Clone the project first.
$ git clone --recurse-submodules https://github.com/XadillaX/byakuren.gitThis project can be compiled to a static libary (byakuren.a) for using.
$ make byakurenAfter compiling, you may use this library just by including byakuren.h in your project.
#include "byakuren.h"- octree algorithm
- min-diff algorithm
- mix-in algorithm
typedef struct bkr_rgb {
uint8_t red;
uint8_t green;
uint8_t blue
} bkr_rgb;RGB pixel structure.
| name | type | description |
|---|---|---|
| red | uint8_t | the RED value (0-255) |
| green | uint8_t | the GREEN value (0-255) |
| blue | uint8_t | the BLUE value (0-255) |
typedef struct bkr_color_stats {
bkr_rgb color;
uint32_t value;
uint32_t count;
} bkr_color_stats;Stats of theme color result.
| name | type | description |
|---|---|---|
| color | bkr_rgb | a color pixel to indicate a theme color |
| value | uint32_t | a color pixel's INT32 value to indicates the theme color |
| count | uint32_t | stats of this theme color in the picture |
typedef struct bkr_palette_array {
uint32_t count;
bkr_rgb* colors;
} bkr_palette_array;A crowd of colors to indicate a theme color palette.
| name | type | description |
|---|---|---|
| count | uint32_t | color count in this palette |
| colors | bkr_rgb* | each color in this palette |
typedef struct bkr_mindiff_parameter {
bkr_palette_array* palette;
int16_t gray_offset;
} bkr_mindiff_parameter;A parameter passes to Min-diff Algorithm.
| name | type | description |
|---|---|---|
| palette | bkr_palette_array* | a palette to calculate the theme color, left for NULL to indicate the default palette |
| gray_offset | int16_t | the offset to judge whether a color is gray, recommand to be 5 |
Before the whole work you should initialize the Byakuren environment:
int bkr_init();And after all the work you should release the environment:
void bkr_destroy();bkr_octree_node* bkr_build_octree(
bkr_rgb* pixels,
uint32_t pixel_count,
uint32_t max_colors);| parameter | type | description |
|---|---|---|
| pixels | bkr_rgb* | the RGB pixels of a picture |
| pixel_count | uint32_t | pixel count of the picture |
| max_colors | uint32_t | maximum theme color count this octree will have |
- Return an octree
int bkr_octree_calculate_color_stats(
bkr_octree_node* node,
bkr_color_stats stats[]);| parameter | type | description |
|---|---|---|
| node | bkr_octree_node* | the octree which bkr_build_octree returned |
| stats | bkr_color_stats | an array to receive each theme color stats |
- Return the count of theme colors.
void bkr_release_octree(bkr_octree_node* node);| parameter | type | description |
|---|---|---|
| node | bkr_octree_node* | the octree to be released |
bkr_rgb* rgb = GET_PICTURE_RGB(); // implement by yourself
uint32_t color_count = GET_PICTURE_PIXEL_COUNT(); // implement by yourself
bkr_color_stats stats[256];
bkr_octree_node* root = bkr_build_octree(rgb, color_count, 256);
int colors = bkr_octree_calculate_color_stats(root, stats);int bkr_mindiff_calculate_color_stats(
bkr_rgb* pixels,
uint32_t pixel_count,
bkr_color_stats stats[],
bkr_mindiff_parameter* param);| parameter | type | description |
|---|---|---|
| pixels | bkr_rgb* | the RGB pixels of a picture |
| pixel_count | uint32_t | pixel count of the picture |
| stats | bkr_color_stats | an array to receive each theme color stats |
| param | bkr_mindiff_parameter* | the parameter passes to Min-diff Algorithm for calculating |
- Return the count of theme colors.
bkr_rgb* rgb = GET_PICTURE_RGB(); // implement by yourself
uint32_t color_count = GET_PICTURE_PIXEL_COUNT(); // implement by yourself
bkr_color_stats stats[256];
bkr_mindiff_parameter param;
param.gray_offset = 5;
param.palette = NULL;
int colors = bkr_mindiff_calculate_color_stats(rgb, color_count, stats, ¶m);Mix Octree and Min-diff up.
int bkr_mix_calculate_color_stats(
bkr_rgb* pixels,
uint32_t pixel_count,
uint32_t octree_max_colors,
bkr_mindiff_parameter* mindiff_param,
bkr_color_stats stats[]);| parameter | type | description |
|---|---|---|
| pixels | bkr_rgb* | the RGB pixels of a picture |
| pixel_count | uint32_t | pixel count of the picture |
| octree_max_colors | uint32_t | maximum theme color count this octree will have |
| param | bkr_mindiff_parameter* | the parameter passes to Min-diff Algorithm for calculating |
| stats | bkr_color_stats | an array to receive each theme color stats |
- Return the count of theme colors.
bkr_rgb* rgb = GET_PICTURE_RGB(); // implement by yourself
uint32_t color_count = GET_PICTURE_PIXEL_COUNT(); // implement by yourself
bkr_color_stats stats[256];
bkr_mindiff_parameter param;
param.gray_offset = -1;
param.palette = NULL;
int colors = bkr_mix_calculate_color_stats(rgb, color_count, 256, ¶m, stats);$ make ./test/bkr_testAfter make command, you should generate a binary file named test.rgb. Then run:
$ cd test && ./bkr_test ALGORITHMYou may create your own
.rgbfile by referring test/run.js or test/test.c. Or you may have a look at Test Helper.
ALGORITHMis a parameter means algorithm you want to test.support
octreeใmindiffandmixso far.
If you want to test quickly (no *.rgb), you may use a simple script.
Install Node.js first and come to test folder. Then run:
$ cd test
$ npm installNOTICE: You can start test helper directly under OSX. Otherwise, you should compile a test binary executor before testing, that is
$ make ./test/bkr_test.
After installing, run --help.
$ ./run.js --helpYou will see some introduction. -u means a URL.
For an example:
$ ./run.js -u http://cdn.duitang.com/uploads/item/201205/22/20120522224448_43nFu.thumb.600_0.jpeg -a octreeAfter command above, a browser will be open to display the result.
If you want to test three algorithm at one time, you should make -a be all.
For an example:
$ ./run.js -u http://cdn.duitang.com/uploads/item/201205/22/20120522224448_43nFu.thumb.600_0.jpeg -a allWith
octree_max_colorsbe 16 in Octree and Mix algorithm.
You're welcome to make Pull Requests.
ใ้็ถๆ่ฆบๅพไธๆ้บผๅฏ่ฝๆไบบๆ้ๆณจๆใ