1/*******************************************************
  2*    這是配合我的部落格《[email protected] DIY視訊會議系統》
  3*    而寫的一個階段性實驗。
  4*    作者：馮富秋 tinnal
  5*    郵箱：[email protected]
  6*    部落格：www.cnitblog.com/tinnal/
  7*    目期：2009-01-03
  8*    版本：1.00
  9*********************************************************/
 10
 11#include "stdio.h"
 12#include "string.h"
 13#include "types.h"
 14#include "g711.h"
 15#include "wav.h"
 16
 17struct RIFF_HEADER    riff_header;
 18struct FMT_BLOCK    fmt_block;
 19char   fack_block_buffer[20];        //20 should be enough
 20struct FACT_BLOCK    fact_block;
 21struct DATA_BLOCK    data_block;
 22
 23int main(int argc, char **argv)
 24{
 25    FILE *wav_in;
 26    FILE *wav_out;
 27    U32 i;
 28    U8    has_fact_block =0;
 29
 30    unsigned char pcm_bytes[2];
 31    short pcm;
 32    unsigned char a_law;    
 33
 34    long file_pos;
 35
 36    if(argc != 3 )
 37    {
 38        printf("Usage:\n\t%s <intput file> <output file>\n", argv[0]);
 39        exit(-1);
 40    }
 41
 42    wav_in = fopen(argv[1],"rb");
 43    if(wav_in == NULL)
 44    {
 45        printf("Can't open input file %s\n", argv[1]);
 46        return (-1);
 47    }
 48
 49    wav_out = fopen(argv[2], "wb");
 50    if( wav_out == NULL)
 51    {
 52        printf("Can't open output file %s\n",argv[2]);
 53        fclose(wav_in);
 54        return(-1);
 55    }
 56    
 57    file_pos = ftell(wav_in);
 58
 59    //Read RIFF_HEADER
 60    fread(&riff_header, sizeof(struct RIFF_HEADER), 1, wav_in);
 61    if(    memcmp(riff_header.szRiffID, "RIFF", 4) != 0 ||
 62        memcmp(riff_header.szRiffFormat, "WAVE", 4) != 0 )
 63    {
 64        printf("No a vaild wave file!\n");
 65        fclose(wav_in);
 66        fclose(wav_out);
 67        return(-1);
 68    }
 69    file_pos = ftell(wav_in);
 70
 71    //Read FMT_BLOCK
 72    fread(&fmt_block, sizeof(struct FMT_BLOCK), 1, wav_in);
 73    if(    memcmp(fmt_block.szFmtID, "fmt ", 4) !=0 ||
 74        fmt_block.dwFmtSize != 18 ||
 75        fmt_block.wavFormat.wFormatTag != 0x1 ||
 76        fmt_block.wavFormat.wChannels != 0x1 ||
 77        fmt_block.wavFormat.dwSamplesPerSec != 8000 ||
 78        fmt_block.wavFormat.wBitsPerSample != 16)
 79    {
 80        printf("Sorry this is only test program,\n"
 81            "we only support follow format,\n"
 82            "\t 1. Format:        linear PCM \n"
 83            "\t 2. Samples Rate:  8000 KHz \n"
 84            "\t 3. Channels:      one channel \n"
 85            "\t 4. BitsPerSample: 16 \n");
 86        fclose(wav_in);
 87        fclose(wav_out);
 88        return(-1);
 89    }
 90    
 91    file_pos = ftell(wav_in);
 92
 93    //Try to read FACT_BLOCK
 94    file_pos = ftell(wav_in);
 95    fread(&fact_block, sizeof(struct FACT_BLOCK), 1, wav_in);
 96    if( memcmp(fact_block.szFactID, "fact", 4) == 0 )
 97    {    
 98        has_fact_block =1;
 99        fread(&fack_block_buffer, fact_block.dwFactSize, 1, wav_in);
100    }
101    else
102        fseek(wav_in, file_pos, SEEK_SET);
103    
104    fread(&data_block, sizeof(struct DATA_BLOCK), 1, wav_in);
105    if (memcmp(data_block.szDataID, "data", 4) != 0)
106    {
107        printf("OOh what error?\n");
108        fclose(wav_in);
109        fclose(wav_out);
110        return(-1);
111    }
112
113    //Change the wave header to write
114    riff_header.dwRiffSize                    -=    data_block.dwDataSize/2 ;
115    
116    fmt_block.wavFormat.wFormatTag            =    0x06;
117    fmt_block.wavFormat.wChannels            =    0x01;
118    fmt_block.wavFormat.dwSamplesPerSec        =    8000;
119    fmt_block.wavFormat.dwAvgBytesPerSec    =    8000;
120    fmt_block.wavFormat.wBlockAlign            =    0x01;
121    fmt_block.wavFormat.wBitsPerSample        =    0x08;
122
123    data_block.dwDataSize                    -=    data_block.dwDataSize/2 ;
124
125    //Write wave file header
126    fwrite(&riff_header, sizeof(struct RIFF_HEADER), 1, wav_out);
127    fwrite(&fmt_block, sizeof(struct FMT_BLOCK), 1, wav_out);
128    if(has_fact_block == 1) 
129    {
130        fwrite(&fact_block, sizeof(struct FACT_BLOCK), 1, wav_out);
131        fwrite(&fack_block_buffer, fact_block.dwFactSize, 1, wav_out);
132    }
133    fwrite(&data_block, sizeof(struct DATA_BLOCK), 1, wav_out);
134
135    //Convert pcm data to a-low data  and  write wav file.
136    for(i =0; i< data_block.dwDataSize; i++)
137    {
138        pcm_bytes[0] = (U8) fgetc(wav_in);
139        pcm_bytes[1] = (U8) fgetc(wav_in);
140        pcm = *(short *)&pcm_bytes;
141
142        a_law     = ALawEncode((int)pcm);
143//        a_law     = linear2alaw((int)pcm);
144        fputc(a_law, wav_out);
145    }
146    fclose(wav_in);
147    fclose(wav_out);
148    
149    printf("Finish!\n");
150    return 0;
151}
152