c - 如何用 C 将二进制位写入二进制文件？

Question

我正在尝试用 C 语言实现霍夫曼编码。我完成了树的构造，并随着算法的进行获得了每个符号的码字。但现在我不得不将代码字插入相应符号的二进制文件中。有人可以建议如何将代码字或二进制位写入二进制文件，以便我可以获得压缩文件。

码字的长度可变。

将这些位写入文件或从文件中读取这些位的函数会很有帮助。

这是我写的代码

void create_compressed_file()
{
    char str[20], ch, *str2, str1[10], str_arr[6], str3[10];
    FILE *fp, *fp2, *fp3;
    int i, array[20], j = 0;
    fp2 = fopen("newfile.txt", "r"); // contains the original text file
    fp3 = fopen("codeword.txt", "r"); // contains the symbol and codeword
    while (fscanf(fp2, "%s", &str) == 1) {
        rewind(fp3);
        str2 = strtok(str, "-");
        while (str2 != NULL) {
            strcpy(str_arr, str2);
            printf("str2= %s ", str_arr); //str2 stores the symbol(not char but a string)
            printf(" %s-", str2);
            while (fscanf(fp3, "%s", &str1) == 1) {
                if (strcmp(str1, str_arr) == 0) {
                    fscanf(fp3, "%s", &str1); // extracted corresponding codeword(1s and 0s) of   the symbol and stored it into str1
                    printf("%s\n", str1);
                    write_codeword_to_binaryfile(); // function that i want to create with is   incomplete and need your help.
                }
            }
            str2 = strtok(NULL, "-");
            rewind(fp3);
        }
        printf("\nspace:");
        strcpy(str_arr, "space");
        while (fscanf(fp3, "%s", &str1) == 1) {
            if (strcmp(str1, str_arr) == 0) {
                fscanf(fp3, "\n%s", &str1); // extract the codeword for(space)character  
                printf("%s\n", str1);
            }
        }
    }
    fclose(fp2);
    fclose(fp3);    
}

codeword.txt:

is  0000
por 00010
Plain   000110
most    0001110
the 0001111
ted 00100 
text    00101
ly  0011000
near    0011001
pli 0011010
ap  0011011
ble 0011100
ta  0011101
by  0011110
sup 0011111
cryp    0100000
In  0100001
ra  0100010
tog 0100011
ting    0100100
tain    0100101
mands   0100110
com 0100111
mes 0101000
to  0101001
ge  0101010
sa  0101011
plain   0101100
phy 0101101

我尝试了上面的代码，如下所示，但它没有写任何东西...执行后的文件大小为0字节:

#include <stdio.h>
#include <conio.h>
#include <stdint.h>

void write_codeword_to_binaryfile(
    const char *codeword, // codeword to write, in ASCII format
    FILE *file,           // destination file
    uint8_t *buffer,
    int *fullness)
{
    char c;
    //  fullness = ;
    *buffer = 0;
    for (c = *codeword++; c != '\0'; c = *codeword++) // iterate
    {
        int bit = c - '0'; // convert from ASCII to binary 0/1
        *buffer |= bit << (7 - fullness);
        ++fullness;
    }
    fputc(*buffer, file);
}

int main() {
    FILE *fp;
    uint8_t *buffer = 0;
    char *c = "10101010";
    char b = 0;
    int i;
    fp = fopen("myfile.bin", "wb");
    write_codeword_to_binaryfile(c, fp, buffer, 8);
    fclose(fp);
    getch();
}

Answer 1

首先，您应该以二进制模式打开文件:

fp = fopen("myfile", "wb"); // "b" means "binary"

这在 Windows 中是必须的，但在大多数其他平台上不是必需的(您不需要做任何特殊的事情来区分平台；只需使用“wb”)。

要将位写入文件，您应该使用缓冲区 - 部分填充的字节。当缓冲区填满时将缓冲区写入文件(恰好包含 8 个填充位)。

uint8_t buffer = 0;

您应该使用一个计数器来跟踪填充了多少位。

int fullness = 0;

写入文件的函数应该接收缓冲区及其填充度。因为它会改变它们，所以你实际上必须发送指针:

void write_codeword_to_binaryfile(
    const char *codeword, // codeword to write, in ASCII format
    FILE *file,           // destination file
    uint8_t *buffer,
    int *fullness)
{
    for (char c = *codeword++; c != '\0'; c = *codeword++) // iterate
    {
        int bit = c - '0'; // convert from ASCII to binary 0/1
        ...
    }
}

有两种方式来排列字节中的位 - 小端(第一位是最低有效位)或大端(第一位是最高有效位)。习惯的方式是使用大端排序。

如果你的缓冲区已填充一定数量的位，如何填充下一位？以下示例显示了填充了 5 位的缓冲区:

011011...
      ^
next bit to fill (its position, starting from the left, is 2)

从这个例子中可以看出，下一位的位置是7 - fullness。因此，对于每一位，请执行以下操作:

*buffer |= bit << (7 - *fullness);
++fullness;

参见How do you set, clear and toggle a single bit in C/C++?了解更多信息。

当缓冲区满时(fullness等于8)，将其写入文件:

fputc(*buffer, file);
*fullness = 0;
*buffer = 0;

完成消息编码后，您还应该“刷新”缓冲区(即将其写入文件):

if (*fullness > 0)
    fputc(*buffer, file);

<小时/>

顺便说一句，消息末尾发生的情况对于位级编码器来说是一个常见的重要问题。您应该从解码器的角度考虑它:您需要了解应该在文件的最后一个字节中解码多少位。有几种解决方案:

• 对消息进行编码后，再编码一个 1 位，然后编码零位，直到缓冲区已满。解码器需要反向解码零位和1位。这是 MPEG 使用的。
• 在文件头中写入消息的长度(以位为单位)。这可能是最简单的解决方案，尽管它需要在完成编码后更新文件的开头。
• 有一个特殊的代码字来表示“消息结束”(也经常使用)