Как можно перевернуть часть массива? - PullRequest
0 голосов
/ 04 октября 2019

Я пишу программу, которая берет растровый файл для чтения в память. Но когда я читаю это в памяти, я делаю некоторые изменения. Сначала я инвертирую цвета пикселей. Мне удалось заставить это работать. Сейчас я пытаюсь перевернуть изображение по оси Y. Я пробовал использовать два цикла for, но в итоге получал ошибки сегментации, а также мне не нравилось, как это выглядит грязно. Со второй попытки я нашел другой подход, который чище, потому что он использует только один цикл и одно условие против двух циклов и двух условий. Мой код теперь не выдает ошибок, но не выполняет запланированную операцию. Есть ли другой алгоритм, который я мог бы попробовать?

Ниже приведен код для части моей программы. Я пытаюсь обратить пиксель, когда я читаю их построчно.

    fseek(fin, bfh.offset, SEEK_SET);

    Pixel *p = new Pixel[bih.height * bih.width];

    for (uint32_t i = 0; i < bih.height; i++) {

        for (uint32_t j = 0; j < bih.width; j++) {

            uint32_t index = i * bih.width + j;

            fread(&p[index], 1, sizeof(p[0]), fin);

            p[index].blue = 255 - p[index].blue;
            p[index].green = 255 - p[index].green;
            p[index].red = 255 - p[index].red;

        }

        uint32_t k = (bih.width * i) - 1;
        uint32_t c = 0 + (i * bih.width);

        if ( i == 0) {
            k = bih.width - 1;
        }

        while( c < k) 
        {
            temp = p[c];
            p[c] = p[k];
            p[k] = temp;
            c++;
            k--;

        }

        fseek(fin, padding_bytes, SEEK_CUR);

    }

    fclose(fin);

Ниже приведена вся моя программа, если это необходимо.

#include <cstdint>
#include <cstdio>

#pragma pack(push, 2)
struct BitmapFileHeader {

    uint16_t type;
    uint32_t size;
    uint16_t reserved_1;
    uint16_t reserved_2;
    uint32_t offset;

};
struct BitmapInfoHeader {

    uint32_t size;
    uint32_t width;
    uint32_t height;
    uint16_t planes;
    uint16_t bitcount;
    uint32_t compression;
    uint32_t imagesize;
    uint32_t x_pixels_per_meter;
    uint32_t y_pixels_per_meter;
    uint32_t color_used;
    uint32_t color_important;
};
#pragma pack(pop)

struct Pixel {
    uint8_t blue;
    uint8_t green;
    uint8_t red;
};

int main(int argc, char* argv[])
{
    if(argc != 3) {
        printf("Usage : %s input_file output_file\n", argv[0]);
        return 1;
    }

    FILE *fin;
    FILE *fout;
    BitmapFileHeader bfh;
    BitmapInfoHeader bih;

    Pixel temp;

    fin = fopen(argv[1], "rb");

    if (nullptr == fin) {
        perror(argv[1]);
        return -1;
    }

    if (sizeof(BitmapFileHeader) != fread(
        &bfh,
        1,
        sizeof(bfh),
        fin
    )) {
        printf("Unable to read bitmap file header. \n");
        return -2;
    }

    if (sizeof(BitmapInfoHeader) != fread(
        &bih,
        1,
        sizeof(bih),
        fin
    )) {
        printf("Unable to read bitmap info header. \n");
        return -3;
    }

    printf("Size of File Header = %lu\n", sizeof(BitmapFileHeader));

    int8_t first = (bfh.type >> 8) & 0xff;
    int8_t second = bfh.type & 0xff;

    if ( (first != 'M') && (second != 'B') ){
        printf("Input file is not a Bitmap file. \n");
        return -4;
    }

    printf("File type = %c%c\n", first, second);
    printf("File size = %u\n", bfh.size);
    printf("File offset = %u\n", bfh.offset);
    printf("File width = %u\n", bih.width);
    printf("Info size = %u\n", bih.size);

    uint32_t padding_bytes = 0;
    uint32_t row_bytes_final = bih.width * sizeof(Pixel);
    uint32_t row_bytes_initial = row_bytes_final;

    do{
        uint32_t rem = row_bytes_final % 4;

        if (rem != 0) {
            row_bytes_final += 1;
        }

        padding_bytes = row_bytes_final - row_bytes_initial;

    } while( (row_bytes_final % 4) != 0);


    fseek(fin, bfh.offset, SEEK_SET);

    Pixel *p = new Pixel[bih.height * bih.width];

    for (uint32_t i = 0; i < bih.height; i++) {

        for (uint32_t j = 0; j < bih.width; j++) {

            uint32_t index = i * bih.width + j;

            fread(&p[index], 1, sizeof(p[0]), fin);

            p[index].blue = 255 - p[index].blue;
            p[index].green = 255 - p[index].green;
            p[index].red = 255 - p[index].red;

        }

        uint32_t k = (bih.width * i) - 1;
        uint32_t c = 0 + (i * bih.width);

        if ( i == 0) {
            k = bih.width - 1;
        }

        while( (c * bih.width) < (k * bih.width)) 
        {
            temp = p[c];
            p[c] = p[k];
            p[k] = temp;
            c++;
            k--;

        }

        fseek(fin, padding_bytes, SEEK_CUR);

    }

    fclose(fin);


    fout = fopen(argv[2], "wb");

    if(nullptr == fout) {
        perror(argv[2]);
        return -5;
    }

    if( sizeof(BitmapFileHeader) != fwrite(
    &bfh, 
    1, 
    sizeof(bfh), 
    fout
    )) {
        printf("Unable to write bitmap file header.\n");
        return -6;
    }

    if( sizeof(BitmapInfoHeader) != fwrite(
        &bih, 
        1, 
        sizeof(bih), 
        fout
        )) {
            printf("Unable to write bitmap info header.\n");
            return -7;
        }

    fseek(fout, bfh.offset, SEEK_SET);

    for (uint32_t i = 0; i < bih.height; i++) {

        for (uint32_t j = 0; j  < bih.width; j++) {

            uint32_t index = i * bih.width + j;

            fwrite(&p[index], 1, sizeof(p[0]), fout);

        }

        fseek(fout, padding_bytes, SEEK_CUR);


    }

    if (padding_bytes > 0) {
        fseek(fout, -1, SEEK_CUR);
        fputc('\0', fout);
    }


    fclose(fout);
    delete[] p;

    return 0;

}

1 Ответ

0 голосов
/ 04 октября 2019

Вы получили неправильные границы, это должно быть c = i * bih.width; k = (i + 1) * bih.width - 1;.

Вы также можете использовать std::reverse, чтобы сделать это:

std::reverse(p + i * bih.width, p + (i + 1) * bih.width);  // Exclusive end, so no -1
...