ascii.c

Ugrás a fájl dokumentációjához.

#include <stdio.h>
#include <stdlib.h>

enum {
    MAX_BRIGHT = 36,    
    ASCII_WIDTH = 79,   
    MAX_FNAME = 1000    
};

char ASCII[MAX_BRIGHT] = {
    ' ', '.', '\'', '`', '\'', '-', '^', '\"', '!',
    '>', ';', '=', '0', 'i', 'j', 'l', 't', 'c',
    'f', 'o', 'a', 'V', '5', 'e', 'b', 'D', 'p',
    'g', 'B', 'm', 'H', 'H', 'H', 'M', 'M', '@'
};

typedef unsigned char Byte; 
typedef struct {
    Byte r; 
    Byte g; 
    Byte b; 
} Pixel;

typedef struct {
    unsigned height;    
    unsigned width;     
    Pixel *pixels;      
} Image;

Image *image_read(char *fname)
{
    Image *image;
    FILE *fp;
    unsigned depth;

    /* memóriafoglalás a struktúrának */
    image = (Image *)malloc(sizeof(Image));
    if (image == NULL) {
        fprintf(stderr, "Error allocating memory for image\n");
        return NULL;
    }

    fp = fopen(fname, "rb");
    if (fp == NULL) {
        fprintf(stderr, "Error opening file %s\n", fname);
        free(image);
        return NULL;
    }
    
    /* fejléc beolvasása a fájlból */
    if (fscanf(fp, "P6\n%u%u%u%*c", &image->width, &image->height, &depth) != 3 || depth != 255u) {
        fprintf(stderr, "Error reading image\n");
        free(image);
        fclose(fp);
        return NULL;
    }
    
    /* pixeltömb foglalása */
    image->pixels = (Pixel *)malloc(image->height*image->width*sizeof(Pixel));
    if (image->pixels == NULL) {
        fprintf(stderr, "Error allocating memory to image data\n");
        free(image);
        fclose(fp);
        return NULL;
    }
    
    /* pixeladatok beolvasása */
    if (fread(image->pixels, sizeof(Pixel), image->height*image->width, fp) != image->height*image->width) {
        fprintf(stderr, "Error reading image data\n");
        free(image->pixels);
        free(image);
        fclose(fp);
        return NULL;
    }

    fclose(fp);
    return image;
}

void image_close(Image *image)
{
    free(image->pixels);
    free(image);
}

Pixel get_pixel(Image *image, int h, int w)
{
    int idx = h*image->width+w;
    return image->pixels[idx];
}

Byte pixel_brightness(Pixel p)
{
    return (Byte) (p.r/3. + p.g/3. + p.b/3.);
}

void image_to_ascii(Image *image)
{
    unsigned W = image->width/ASCII_WIDTH + 1;  /* egy ASCII karakter ennyi pixel széles lesz */
    unsigned H = W * 1.7;               /* és ennyi magas */

    unsigned h, w;
    for (h = 0; h < image->height/H; ++h) { /* minden ascii sorra és oszlopra */
        for (w = 0; w < image->width/W; ++w) {
            unsigned i, j, b = 0;   /* b a fényesség */
            for (i = 0; i < H; ++i) /* a blokk minden pixelére */
                for (j = 0; j < W; ++j) {
                    Pixel p = get_pixel(image, h*H+i, w*W+j);
                    b += pixel_brightness(p);
                }
            b *= MAX_BRIGHT / 256.0 / (W * H);
            printf("%c", ASCII[b]);     /* karakter kikeresése és kiírása */
        }
        printf("\n");
    }
}

int main(void)
{
    char fname[MAX_FNAME];
    Image *image;   /* A képstruktúrára mutató pointer */

    printf("fajlnev: "); fflush(stdout);
    scanf("%s", fname);
    image = image_read(fname);  /* kép beolvasása */
    if (image == NULL)
        return 1;

    image_to_ascii(image);  /* kép feldolgozása */
    image_close(image);     /* kép felszabadítása */

    return 0;
}