图像恢复程序存在问题

时间:2016-09-06 14:08:55

标签: c jpeg cs50

基本上,我为我的计算机课程(shoutout CS50)编写了这个程序,可以从.raw文件中恢复图像。我已经设法让程序恢复该文件中50个文件中的48个。

我现在使用该程序的问题是程序无法恢复位于.raw上的第一个和第二个文件。它可以读取和写入第一个文件(这个女孩在白雪皑皑的背景中)或第二个文件在.raw(书籍背后的人)。

出于某种原因,如果我将fopen从写入改为追加,我可以在女孩的照片和那个人之间切换,但我似乎无法打开它们。

https://github.com/CoreData/cs50/blob/master/pset4/jpg/card.raw 这是card.raw的链接,遗憾的是它与我使用的不一样,但即使使用这个,你也可以获得两个不同的image1.jpg图像,具体取决于你是否打开“a”或“w”。

任何想法???

如果你们想要任何其他信息,请告诉我

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

#include "bmp2.h"

int main(void)
{
    /*OPEN CARD FILE*/
    char* infile = "card.raw";;

    FILE* card = fopen(infile, "r");
    if (card == NULL)
    {
        printf("Could not open %s.\n", "card.raw");
        return 2;
    }

    int f = 0, c = 0, l = 0, x = 128, imageno = 1;
    // c signals that a jpg is being written
    // l size control, 0 means 0 jpgs
    FILE* images;
    char* title = (char*)malloc(15);

    /*repeat until end of card*/
    do
    {
        //read one block into buffer
        INTROJPG *buffer = (INTROJPG*)malloc(sizeof(INTROJPG)*x);

        for (int i = 0; i < 128; i++)
        {
            fread(&buffer[i], sizeof(INTROJPG), 1, card);
        }

        if (buffer[0].first == 0xff && buffer[0].second == 0xd8 && buffer[0].third == 0xff)
        {
            sprintf(title, "image%d.jpg", imageno); //change jpg title

            if (f == 1) //close previous jpg
            {
                fclose(images);
                imageno++;
            }

            images = fopen(title, "w");
            f = 1; //very first jpg has been opened
            c = 1; //jpg open
            l++; //jpg count + 1
        }

        //jpg already open?
        if (c == 1)
        {
            for (int i = 0; i < 128; i++)
            {
                fwrite(&buffer[i], sizeof(INTROJPG), 1, images);
            }
        }
        free(buffer);
    }    
    while (l < 50);
    free(title);
    return 5;
        //close any remaining files
}


and this is my bmp2.h file

    #include <stdint.h>

/**
 * Common Data Types 
 *
 * The data types in this section are essentially aliases for C/C++ 
 * primitive data types.
 *
 * Adapted from http://msdn.microsoft.com/en-us/library/cc230309.aspx.
 * See http://en.wikipedia.org/wiki/Stdint.h for more on stdint.h.
 */
typedef uint8_t  BYTE;
typedef uint32_t DWORD;
typedef int32_t  LONG;
typedef uint16_t WORD;

/**
 * BITMAPFILEHEADER
 *
 * The BITMAPFILEHEADER structure contains information about the type, size,
 * and layout of a file that contains a DIB [device-independent bitmap].
 *
 * Adapted from http://msdn.microsoft.com/en-us/library/dd183374(VS.85).aspx.
 */
typedef struct 
{ 
    WORD   bfType; 
    DWORD  bfSize; 
    WORD   bfReserved1; 
    WORD   bfReserved2; 
    DWORD  bfOffBits; 
} __attribute__((__packed__)) 
BITMAPFILEHEADER; 

/**
 * BITMAPINFOHEADER
 *
 * The BITMAPINFOHEADER structure contains information about the 
 * dimensions and color format of a DIB [device-independent bitmap].
 *
 * Adapted from http://msdn.microsoft.com/en-us/library/dd183376(VS.85).aspx.
 */
typedef struct
{
    DWORD  biSize; 
    LONG   biWidth; 
    LONG   biHeight; 
    WORD   biPlanes; 
    WORD   biBitCount; 
    DWORD  biCompression; 
    DWORD  biSizeImage; 
    LONG   biXPelsPerMeter; 
    LONG   biYPelsPerMeter; 
    DWORD  biClrUsed; 
    DWORD  biClrImportant; 
} __attribute__((__packed__))
BITMAPINFOHEADER; 

/**
 * RGBTRIPLE
 *
 * This structure describes a color consisting of relative intensities of
 * red, green, and blue.
 *
 * Adapted from http://msdn.microsoft.com/en-us/library/aa922590.aspx.
 */
typedef struct
{
    BYTE  rgbtBlue;
    BYTE  rgbtGreen;
    BYTE  rgbtRed;
} __attribute__((__packed__))
RGBTRIPLE;


typedef struct
{
    BYTE first;
    BYTE second;
    BYTE third;
    BYTE fourth;
} __attribute__((__packed__))
INTROJPG;

typedef struct
{
    BYTE image;
}
BYTEIMAGE;

1 个答案:

答案 0 :(得分:0)

首先,我会尝试改进代码中的一些内容。我也做了这个pset,很高兴帮助别人。

INTROJPG *buffer = (INTROJPG*)malloc(sizeof(INTROJPG)*x);

在这一部分,您知道INTROJPG和x的大小都是常量,因此不需要在每次迭代时不断地分配和释放内存,这比简单地创建普通数组需要更多的时间。另外,为什么缓冲区是指向INTROJPG的指针?如果只是在每次迭代时测试一个标题,我不认为值得,你可以简单地访问普通BYTE数组的前4个字节。

我创建了一个512个BYTE(库上的结构)的静态数组,因为这是你不断分配和释放的大小,而且你使用的是BYTE,而不是INTROJPG。

其次,在本节和另一个类似的部分:

for (int i = 0; i < 128; i++)
{
    fread(&buffer[i], sizeof(INTROJPG), 1, card);
}

完全不需要这个循环,或者甚至使用INTROJPG。您总是在读取和写入512个字节,您可以使用:

fread(buffer, 4, 128, card);
// or even better
fread(buffer, 512, 1, card);

现在关于你的问题,我已多次测试你的代码(没有任何修改),并发现image1.jpg和image2.jpg没有任何问题。是的,我改变了#34; w&#34;模式到&#34; a&#34;反之亦然。

然而,你的代码在最后一张图片方面有问题,你的最后一张图片是image49.jpg,当它应该是image50.jpg,你的image49.jpg甚至没有打开,那是因为在存储image49.jpg的其余部分之前完成循环,即,您只存储image49.jpg的前512个字节。

为了解决这个问题,我已经改变了do-while循环的条件,一直到卡片文件结束,IIRC问题保证最后一个块是最后一个图像的一部分或类似的东西,如果没有,你可以解决这个小问题!

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

#include "bmp2.h"

int main(void)
{
    /*OPEN CARD FILE*/
    char* infile = "card.raw";;

    FILE* card = fopen(infile, "r");
    if (card == NULL)
    {
        printf("Could not open %s.\n", "card.raw");
        return 2;
    }

    int f = 0, c = 0, imageno = 1;
    // c signals that a jpg is being written
    // l size control, 0 means 0 jpgs
    FILE* images;
    char title[25];
    BYTE buffer[512];

    /*repeat until end of card*/
    do
    {

        fread(buffer,  512, 1, card);

        if (buffer[0] == 0xff && buffer[1] == 0xd8 && buffer[2] == 0xff)
        {
            sprintf(title, "image%d.jpg", imageno); //change jpg title

            if (f == 1) //close previous jpg
            {
                fclose(images);
                imageno++;
            }

            images = fopen(title, "w");
            f = 1; //very first jpg has been opened
            c = 1; //jpg open
        }

        //jpg already open?
        if (c == 1) fwrite(buffer, 512, 1, images);
    }    
    while (!feof(card));
    return 5;
        //close any remaining files
}

最后一件事,为什么你在节目结束时返回5?好奇。