如何读写ppm文件?

时间:2015-11-06 10:07:51

标签: c++ ppm

我尝试读取ppm文件并创建一个相同的新文件。但是当我用GIMP2打开它们时,图像并不相同。

我的代码出现问题在哪里?

int main()
{
    FILE *in, *out;
    in = fopen("parrots.ppm","r");
    if( in == NULL )
    {
        std::cout<<"Error.\n";
        return 0;
    }

    unsigned char *buffer = NULL;

    long size = 0;
    fseek(in, 0, 2);
    size = ftell(in);
    fseek(in, 0, 0);

    buffer = new unsigned char[size];
    if( buffer == NULL )
    {
        std::cout<<"Error\n";
        return 0;
    }

    if( fread(buffer, size, 1, in) < 0 )
    {
          std::cout<<"Error.\n";
          return 0 ; 
    }

    out = fopen("out.ppm","w");
    if( in == NULL )
    {
         std::cout<<"Error.\n";
         return 0;
    }

    if( fwrite(buffer, size, 1, out) < 0 )
    {
         std::cout<<"Error.\n";
         return 0;
    }

    delete[] buffer;

    fcloseall();

    return 0;
}

在此之前我在一个结构中读取ppm文件,当我写它时,我得到相同的图像,但绿色比原始图片更强烈。然后我尝试了这个简单的阅读和写作,但我得到了相同的结果。

1 个答案:

答案 0 :(得分:1)

int main()

缺少包括。

FILE *in, *out;

C ++程序中的C风格I / O,为什么?此外,在初始化时声明,接近第一次使用。

in = fopen("parrots.ppm","r");

这是在文本模式下打开文件,这肯定不是你想要的。使用"rb"作为模式。

unsigned char *buffer = NULL;

在初始化时声明,接近第一次使用。

fseek(in, 0, 2);

您应该使用SEEK_END,但不保证将其定义为2

fseek(in, 0, 0);

见上文,SEEK_SET不能保证定义为0。

buffer = new unsigned char[size];
if( buffer == NULL )

默认情况下,new不会返回NULL指针,但会抛出std::bad_alloc异常。 (由于在大多数当前操作系统上的分配是常态,检查NULL即使使用malloc()也无法保护您免受内存不足的影响,但很高兴看到您养成了检查的习惯。 )

C ++ 11为我们带来了smart pointers。使用它们。它们是避免内存泄漏的绝佳工具(C ++的极少数弱点之一)。

if( fread(buffer, size, 1, in) < 0 )

成功使用fread应该返回写入的对象数量,应该检查它们与第三个参数(!= 1)相等,而不是< 0

out = fopen("out.ppm","w");
再次

文字模式,您需要"wb"

if( fwrite(buffer, size, 1, out) < 0 )

请参阅上面有关fread返回值的说明。同样适用于此。

fcloseall();

不是标准功能。使用fclose( in );fclose( out );

一个C ++ 11-ified解决方案(省略错误检查以简化)看起来有点像这样:

#include <iostream>
#include <fstream>
#include <memory>

int main()
{
    std::ifstream in( "parrots.ppm", std::ios::binary );
    std::ofstream out( "out.ppm", std::ios::binary );

    in.seekg( 0, std::ios::end );
    auto size = in.tellg();
    in.seekg( 0 );

    std::unique_ptr< char[] > buffer( new char[ size ] );

    in.read( buffer.get(), size );
    out.write( buffer.get(), size );

    in.close();
    out.close();

    return 0;
}

当然,智能解决方案可以通过Boost.Filesystemstandard functionality进行实际的文件系统复制(在撰写本文时进行实验)。