aboutsummaryrefslogtreecommitdiff
path: root/fly-tools/derive-background
diff options
context:
space:
mode:
Diffstat (limited to 'fly-tools/derive-background')
-rw-r--r--fly-tools/derive-background/pixel.c200
1 files changed, 110 insertions, 90 deletions
diff --git a/fly-tools/derive-background/pixel.c b/fly-tools/derive-background/pixel.c
index 9316e0d..7fde6ae 100644
--- a/fly-tools/derive-background/pixel.c
+++ b/fly-tools/derive-background/pixel.c
@@ -11,77 +11,72 @@
int main(int argc, char **argv ) {
- MagickWand *magick_wand;
- MagickWand *first;
- double green = 0;
- double red = 0;
- double blue = 0;
- int image = 0;
+ MagickWand *magick_wand;
+ MagickWand *output_wand;
+ MagickWand *first_wand;
- int i = 0;
- int j = 0;
- int k = 0;
- int l = 0;
+ double green, red, blue;
+ int i,j, k;
+ int image = 0;
int height = 0;
int width = 0;
int nImages = 0;
+ unsigned long number_wands;
+
+ char filename[256];
+ char *temp;
+
+ uint8_t ****array;
+
MagickWandGenesis();
- first = NewMagickWand();
- // Check the height and width of the first image
- MagickReadImage(first, "Background.png");
- height = MagickGetImageHeight(first);
- width = MagickGetImageWidth(first);
+ // open the first image and get the height and width
+ first_wand = NewMagickWand();
+ if(MagickReadImage(first_wand, "Background.png") == MagickFalse) {
+ ThrowWandException(first_wand);
+ }
+ height = MagickGetImageHeight(first_wand);
+ width = MagickGetImageWidth(first_wand);
- printf("%d %d \n", height, width);
+ first_wand = DestroyMagickWand(first_wand);
+ printf("height: %d width:%d \n", height, width);
- // Count how many images there are in the input file
- char fname[256];
+ // count how many images there are in the input file
FILE *count = fopen( argv[1], "r");
if(count != NULL) {
- while((fgets(fname, sizeof(fname), count)) != NULL) {
+ while((fgets(filename, sizeof(filename), count)) != NULL) {
nImages++;
}
fclose(count);
}
printf("%d \n", nImages);
- //this will store the number of pixels in each row
- unsigned long number_wands;
-
- char filename[256];
- char *temp;
- FILE *file = fopen ( argv[1], "r" );
- // initialize this crazy array.
- uint8_t ****array;
- int dim1 = nImages;
- int dim2 = height;
- int dim3 = width;
- int dim4 = 3;
-
- array = calloc(dim1, sizeof(array[0]));
- for(i = 0; i < dim1; i++)
+ // initialize the storage array.
+ array = calloc(nImages, sizeof(array[0]));
+ for(i = 0; i < nImages; i++)
{
- array[i] = calloc(dim2, sizeof(array[0][0]));
- for(j = 0; j < dim2; j++)
+ array[i] = calloc(height, sizeof(array[0][0]));
+ for(j = 0; j < height; j++)
{
- array[i][j] = calloc(dim3, sizeof(array[0][0][0]));
- for(k = 0; k < dim3; k++)
+ array[i][j] = calloc(width, sizeof(array[0][0][0]));
+ for(k = 0; k < width; k++)
{
- array[i][j][k] = calloc(dim4, sizeof(array[0][0][0][0]));
+ array[i][j][k] = calloc(3, sizeof(array[0][0][0][0]));
}
}
}
- if ( file != NULL ) {
- while ((fgets(filename, sizeof(filename), file)) != NULL ) {
+ // store each pixel in the storage array. array[nImage][height][width][ { R, G, B} ]
+ FILE *input_file = fopen ( argv[1], "r" );
+ if ( input_file != NULL ) {
+ while ((fgets(filename, sizeof(filename), input_file)) != NULL ) {
temp = strchr(filename, '\n');
if (temp != NULL) *temp = '\0';
@@ -114,7 +109,7 @@ int main(int argc, char **argv ) {
image++;
}
- fclose ( file );
+ fclose ( input_file );
}
else {
exit(0);
@@ -142,65 +137,90 @@ int main(int argc, char **argv ) {
gsl_histogram_increment (r, array[i][j][k][0]);
gsl_histogram_increment (g, array[i][j][k][1]);
gsl_histogram_increment (b, array[i][j][k][2]);
- // printf("height %d width %d image %d %hhu %hhu %hhu \n", j, k, i, array[i][j][k][0], array[i][j][k][1], array[i][j][k][2]);
+ //printf("height %d width %d image %d %hhu %hhu %hhu \n", j, k, i, array[i][j][k][0], array[i][j][k][1], array[i][j][k][2]);
}
- size_t red = gsl_histogram_max_val(r);
- size_t green = gsl_histogram_max_val(r);
- size_t blue = gsl_histogram_max_val(r);
- output[j][k][0] = array[red][j][k][0];
- output[j][k][1] = array[green][j][k][1];
- output[j][k][2] = array[blue][j][k][2];
-
- printf("calculated histogram for (%d,%d) \n ", j, k);
- /* int red_val = array[red - 1 ][j][k][0];
- int blue_val = array[blue - 1 ][j][k][1];
- int green_val = array[green - 1][j][k][2];
- printf("h:%d w:%d rm: %d gm: %d bm: %d rv: %lu bv: %lu gv:%lu\n",
- j, k, red_val, green_val, blue_val, red, blue, green);
- */
+ size_t red = gsl_histogram_max_bin(r);
+ size_t green = gsl_histogram_max_bin(g);
+ size_t blue = gsl_histogram_max_bin(b);
+ int red_index = gsl_histogram_get(r, red);
+ int green_index = gsl_histogram_get(g, green);
+ int blue_index = gsl_histogram_get(b, blue);
+ output[j][k][0] = array[red_index - 1 ][j][k][0];
+ output[j][k][1] = array[blue_index - 1 ][j][k][1];
+ output[j][k][2] = array[green_index - 1 ][j][k][1];
+
+ printf("i (%d,%d) rgb(%d,%d,%d) \n ", j, k, output[j][k][0], output[j][k][1], output[j][k][2]);
+
gsl_histogram_free (r);
gsl_histogram_free (g);
gsl_histogram_free (b);
}
}
- MagickWand *m_wand = NULL;
- PixelWand *p_wand = NULL;
- PixelIterator *iterator = NULL;
- PixelWand **pixels = NULL;
-
- int x,y;
- int grey;
- MagickWandGenesis();
-
- p_wand = NewPixelWand();
- PixelSetColor(p_wand,"white");
- m_wand = NewMagickWand();
-
- MagickNewImage(m_wand,height,width,p_wand);
-
- // Get a new pixel iterator
- iterator=NewPixelIterator(m_wand);
+ MagickWand *m_wand = NULL;
+ PixelWand *p_wand = NULL;
+ PixelIterator *iterator = NULL;
+ PixelWand **pixels = NULL;
+ int x,y;
+ char rgb[128];
+
+ MagickWandGenesis();
+
+ p_wand = NewPixelWand();
+ PixelSetColor(p_wand,"white");
+ m_wand = NewMagickWand();
+ // Create a 100x100 image with a default of white
+ MagickNewImage(m_wand,height,width,p_wand);
+ // Get a new pixel iterator
+ iterator=NewPixelIterator(m_wand);
+ for(y=0;y<100;y++) {
+ // Get the next row of the image as an array of PixelWands
+ pixels=PixelGetNextIteratorRow(iterator,&x);
+ // Set the row of wands to a simple gray scale gradient
+ for(x=0;x<100;x++) {
+ sprintf(rgb, "rgb(%d,%d,%d)", array[0][i][j][0], array[0][i][j][1], array[0][i][j][2]);
+ PixelSetColor(pixels[x],rgb);
+ }
+ // Sync writes the pixels back to the m_wand
+ PixelSyncIterator(iterator);
+ }
+ // Clean up
+ iterator=DestroyPixelIterator(iterator);
+ MagickWriteImage(m_wand,"output.png");
+ DestroyMagickWand(m_wand);
+ MagickWandTerminus();
+
+
+ // write the histogram data to an image
+ // output_wand = NewMagickWand();
+ // PixelWand *p = NewPixelWand();
+ // PixelSetColor(p, "white");
+ // if(MagickNewImage(output_wand, height, width, p) == MagickFalse) {
+ // ThrowWandException(output_wand);
+ // }
+
+ // PixelIterator* iterator = NewPixelIterator(output_wand);
+ // PixelWand **pixels = PixelGetNextIteratorRow(iterator,&number_wands);
+
+ // printf("Image number:%d Filename: %s \n", image, filename);
+ // for (i=0; i<=height; i++) {
+ // for (j=0; j<width; j++) {
+
+ // char rgb[24];
+
+ // //sprintf(rgb, "rgb(0,20,0)");
+ // printf("o (%d,%d) %s \n", i, j, rgb);
+ // if( (PixelSetColor(pixels[j], rgb)) == MagickFalse ) {
+ // ThrowWandException(output_wand);
+ // }
+ // }
+ // PixelSyncIterator(iterator);
+ // pixels=PixelGetNextIteratorRow(iterator,&number_wands);
+ // }
- for(y=0; y < height;y++) {
- // Get the next row of the image as an array of PixelWands
- pixels=PixelGetNextIteratorRow(iterator,&x);
- for(x=0; x < width; x++) {
- char rgb[17];
- sprintf(rgb, "rgb(%d,%d,%d)", output[x][y][0], output[x][y][1], output[x][y][2]);
- printf("(%d,%d) %s \n", y, x, rgb);
- PixelSetColor(pixels[x], rgb);
- }
- // Sync writes the pixels back to the m_wand
- PixelSyncIterator(iterator);
- }
- // Clean up
- iterator=DestroyPixelIterator(iterator);
- MagickWriteImage(m_wand,"output.png");
- DestroyMagickWand(m_wand);
MagickWandTerminus();
return 0;