使用C语言扩展Python(三)

上一篇中我们已经了解如何在Python程序和C模块之间进行值的相互传递，现在我们来进入实作阶段，看看如何将一个C语言开发的开源mp3编解码库LAME包装为一个Python下可以使用的扩展模块。

首先去http://lame.sourceforge.net/download.php下载LAME的源代码，然后切换到root用户编译源代码，

./configure
make
make install

安装完成后你可以在/usr/local/include/lame目录下找到lame.h头文件，我们在后面的demo程序中会include它的，下面就是一个非常简单的lame示例程序lame_test.c:

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

#define INBUFSIZE 4096
#define MP3BUFSIZE (int) (1.25 * INBUFSIZE) + 7200

int encode(char* inPath, char* outPath) {
    int status = 0;
    lame_global_flags* gfp;
    int ret_code;
    FILE* infp;
    FILE* outfp;
    short* input_buffer;
    int input_samples;
    char* mp3_buffer;
    int mp3_bytes;
    
    gfp = lame_init();
    if (gfp == NULL) {
        printf("lame_init failed\n");
        status = -1;
        goto exit;
    }
    
    ret_code = lame_init_params(gfp);
    if (ret_code < 0) {
        printf("lame_init_params returned %d\n",ret_code);
        status = -1;
        goto close_lame;
    }

    infp = fopen(inPath, "rb");
    outfp = fopen(outPath, "wb");
    
    input_buffer = (short*)malloc(INBUFSIZE*2);
    mp3_buffer = (char*)malloc(MP3BUFSIZE);
    
    do{
        input_samples = fread(input_buffer, 2, INBUFSIZE, infp);
        mp3_bytes = lame_encode_buffer_interleaved(gfp, input_buffer,input_samples/2, mp3_buffer, MP3BUFSIZE);
        if (mp3_bytes < 0) {
            printf("lame_encode_buffer_interleaved returned %d\n", mp3_bytes);
            status = -1;
            goto free_buffers;
        } else if(mp3_bytes > 0) {
            fwrite(mp3_buffer, 1, mp3_bytes, outfp);
        }
    }while (input_samples == INBUFSIZE);
    
    mp3_bytes = lame_encode_flush(gfp, mp3_buffer, sizeof(mp3_buffer));
    if (mp3_bytes > 0) {
        printf("writing %d mp3 bytes\n", mp3_bytes);
        fwrite(mp3_buffer, 1, mp3_bytes, outfp);
    }
free_buffers:
    free(mp3_buffer);
    free(input_buffer);
    
    fclose(outfp);
    fclose(infp);
close_lame:
    lame_close(gfp);
exit:
    return status;
}

int main(int argc, char** argv) {
    if (argc < 3) {
        printf("usage: lame_test rawinfile mp3outfile\n");
    }
    encode(argv[1], argv[2]);
    return 0;
}