编写PNM P6的问题

Question

我正在编写一个程序，它接收两个重复的PNM P6文件，将第一个文件的内存放入缓冲区，在缓冲区上创建一条黄色对角线，并将结果写入第二个文件。运行它时，输出文件已损坏，无法显示。在查看输出时，我注意到它缺少了应该位于顶部的三行：

P6
1786 1344
255

我不知道如何以编程的方式确保这些行保留在代码中--我不知道为什么它们一开始就被覆盖了。

当我手动将这些行添加到输出时，文件不再损坏。然而，没有出现黄色对角线。这是同一个问题的一部分，还是还有什么需要我研究解决的问题？

我的代码：

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

typedef struct
{
    unsigned char red, green, blue;
} iPixel;

typedef struct
{
    int x, y;
    iPixel *data;

} Image;

Image *
ReadImage(char *filename)
{
    FILE *f = fopen(filename, "rb");
    Image *img = NULL;
    char magicNum[128];
    int  width, height, maxval;

    if (f == NULL)
    {
        fprintf(stderr, "Unable to open file %s\n", filename);
        return NULL;
    }

    fscanf(f, "%s\n%d %d\n%d\n", magicNum, &width, &height, &maxval);
/*
    printf("Magic num = %s width = %d, height = %d, maxval = %d\n",
            magicNum, width, height, maxval)
; 
*/
    if (strcmp(magicNum, "P6") != 0)
    {
        fprintf(stderr, "Unable to read from file %s, because it is not a PNM file of type P6\n", filename);
        return NULL;
    }

    img = (Image *) malloc(sizeof(Image));
    img -> x = width;
    img -> y = height;
    img -> data = (iPixel*) malloc(img -> x * img -> y * sizeof(iPixel));

    fread(img -> data, sizeof(iPixel), width*height, f);

    fclose(f);
    return img;

}

void WriteImage(Image *img, char *filename)
{
    FILE *f = fopen(filename, "wb");

    fwrite(img->data, sizeof(iPixel), img-> x * img-> y, f);
    fclose(f);
}

Image *
YellowDiagonal(Image *input)
{

    int i, j;

    for (i = 0; i < input->x; i++)
    {
        for (j=0; j < input->y; j++)
        {
            if (i==j)
            {
                input->data[i].red=255;
                input->data[i].green=255;
                input->data[i].blue=0;
            }
        }
    }

    return input;   
}

int main(int argc, char *argv[])
{
    if (argc != 3)
    {
        fprintf(stderr, "Usage: ./3A_c.c <input image file> <output image file>\n");
        exit(EXIT_FAILURE);
    }

        Image *img;

        img = ReadImage(argv[1]);
        fprintf(stderr, "Read.\n");
        YellowDiagonal(img);
        fprintf(stderr, "Diagonal line.\n");
        WriteImage(img, argv[2]);
        fprintf(stderr, "Write.\n");
}

Answer 1

小心地写完全相同的格式，你正在阅读。PNM格式是明确定义，您正在正确地阅读它。然而，在写作过程中有几个错误：

1. 用"w"或"wb"打开文件，将其截断为0字节；
2. 最佳实践总是检查fopen是否成功；
3. 读取实际的ASCII数据可以用fscanf完成，二进制数据可以用fread完成。类似地，应该使用fprintf完成ASCII的编写，并再次使用fwrite编写二进制数据。
4. 如果要确保写入与前面读取的数据相同的数据，则需要将其保存在某个地方。maxval变量被读取，但没有保存，因此我无法将它写回。但是，这并不是一个大问题，因为您的代码的其余部分都假设图像是R8G8B8，所以maxval应该始终是255。

这是一个经过调整的WriteImage，可以工作。

void WriteImage(Image *img, char *filename)
{
    FILE *f = fopen(filename, "wb");

    if (f == NULL)
    {
        printf ("Unable to open '%s' for writing!\n", filename);
     /* better would be: "return -1" to indicate an error, 0 otherwise */
        return;
    }

    fprintf (f, "P6\n");
    fprintf (f, "%d %d\n", img->x, img->y);
/*  actually you need to write 'maxval' here */
    fprintf (f, "%d\n", 255);

    fwrite(img->data, sizeof(iPixel), img->x * img->y, f);
    fclose(f);
}

用以上的方式，你现在可以看到你的‘对角线’是不正确的！我不打算解决这个问题(我想我看不出发生了什么事情阻止了你的前进)，但以下是一些指导你的建议：

• 如果您在原地更改数据，则不需要返回Image *。
• 不需要检查每一个像素
• 检查哪些像素被更改的坐标..。