/////////////////////////////////////////////////////////////////
// BIG5與GB2312編碼轉(zhuǎn)換程序
// 作者: Janhail Luo
// 最后整理: 2003-03-03
/////////////////////////////////////////////////////////////////
#include    <stdio.h>
#include    "LanTranform.h"

// 構(gòu)造函數(shù)
CLanTransform::CLanTransform()
{
    pBIG5 = 0;
    pGB2312 = 0;
}

// 析構(gòu)函數(shù)
CLanTransform::~CLanTransform()
{
    // 釋放碼表
    FreeTableBuf(true);
    FreeTableBuf(false);
}

// 初始化轉(zhuǎn)換程序，讀入轉(zhuǎn)換碼表文件
bool CLanTransform::Init(const char* lpszBig2GBFile, const char* lpszGB2BigFile)
{
    // 讀入BIG5轉(zhuǎn)換GB2312的碼表文件
    if (lpszBig2GBFile)
    {
        if (false==this->LoadTable(lpszBig2GBFile, true))
        {
            return false;   // 讀入失敗
        }
    }

    // 讀入GB2312轉(zhuǎn)換BIG5的碼表文件
    if (lpszGB2BigFile)
    {
        if (false==this->LoadTable(lpszGB2BigFile, false))
        {
            return false;   // 讀入失敗
        }
    }

    // 讀入成功
    return true;
}


// 轉(zhuǎn)換一個(gè)指定的BIG5編碼的字符串到GB2312編碼的字符串
//   lpBuf 是指定字符串的指針
//   如果不能轉(zhuǎn)換就返回0, 否則返回lpBuf指向的地址
char* CLanTransform::BIG2GB(char* lpBuf)
{
    unsigned char *pCur, *pDest, bFirst, bSecond, bIndex ;

    // 如果沒有初始化數(shù)據(jù)或者沒有需要轉(zhuǎn)換的數(shù)據(jù)，就直接返回
    if (pBIG5==0 || lpBuf==0)
    {
        return 0;
    }

    pCur = (unsigned char*)lpBuf;
    while(*pCur)   // 處理到字符串結(jié)束
    {
        bFirst = *pCur;         // 取得第一個(gè)字節(jié)的值
        bSecond = *(pCur+1);    // 取得第二個(gè)字節(jié)的值
        if (bFirst<0xA1 || bFirst>0xFE)   // 如果是英文字符
        {
            // 不轉(zhuǎn)換,直接跳到下一個(gè)字符
            pCur++;
        }else if (bSecond<0x40 // 如果是非BIG5的其他字符
            || bSecond>0xFE || (bSecond>0X7E&&bSecond<0XA1))
        {
            // 不轉(zhuǎn)換,直接跳到下一個(gè)字符
            pCur++;
        }else   // 是BIG5字符,需要轉(zhuǎn)換
        {
            bIndex = bFirst-0xA1;

            // 由
            // if (bSecond<0x7F)
            // {
            //     d = pBIG5 + ((temp*0xA0 + (bSecond-0x40))<<1);
            // }else
            // {
            //     d = pBIG5 + ((temp*0xA0 +
            //                (bSecond+(0x7F-0x40)-0xA1))<<1);
            // }
            // 可以優(yōu)化成為下面語句
            //pDest = pBIG5 + ((bIndex<<7)+(bIndex<<5))<<1;
            pDest = pBIG5 + (((bIndex<<7)+(bIndex<<5))<<1);
            if (bSecond<0x7F)
            {
                pDest += (bSecond-0x40)<<1;
            }else
            {
                pDest += (bSecond+0x3F-0xA1)<<1;
            }

            *pCur++ = *pDest++;    // 第一個(gè)字節(jié)
            *pCur++ = *pDest;      // 第二個(gè)字節(jié)
        }
    }
    return lpBuf;
}

// 轉(zhuǎn)換一個(gè)指定的GB2312編碼的字符串到BIG5編碼的字符串
//   lpBuf 是指定字符串的指針
//   如果不能轉(zhuǎn)換就返回0, 否則返回lpBuf指向的地址
char* CLanTransform::GB2BIG(char* lpBuf)
{
    unsigned char *pCur, *pDest, bFirst, bSecond, bIndex ;
 
    // 如果沒有初始化數(shù)據(jù)或者沒有需要轉(zhuǎn)換的數(shù)據(jù)，就直接返回
    if (pGB2312==0 || lpBuf==0)
    {
        return 0;
    }

    pCur = (unsigned char*)lpBuf;
    while(*pCur)   // 如果字符串結(jié)束
    {
        bFirst = *pCur;         // 取得第一個(gè)字節(jié)的值
        bSecond = *(pCur+1);    // 取得第二個(gè)字節(jié)的值
        if (bFirst<0xA1 || bFirst>0xF7)   // 如果是英文字符
        {
            pCur++;    // 不轉(zhuǎn)換,直接跳到下一個(gè)字符
        }else if (bSecond<0xA1 || bSecond>0xFE) // 如果是非GB2312的其他字符
        {
            pCur++;    // 不轉(zhuǎn)換,直接跳到下一個(gè)字符
        }else   // 是GB2312字符,需要轉(zhuǎn)換
        {
            bIndex = bFirst-0xA1;

            // 由 d = pGB2312 + ((temp*0x5E + (bSecond-0xA1))<<1);
            // 可以優(yōu)化成為下面語句
            pDest = pGB2312 + (((bIndex<<6)+(bIndex<<5)
                + (bSecond-0xA1))<<1);

            *pCur++ = *pDest++;    // 第一個(gè)字節(jié)
            *pCur++ = *pDest;      // 第二個(gè)字節(jié)
        }
    }
    return lpBuf;
}

// 從指定文件中讀入字符轉(zhuǎn)換表。
//   lpFileName 字符轉(zhuǎn)換表的文件名
//   bIsBIG5 是否是BIG5的轉(zhuǎn)換表
//   返回true讀入成功,否則失敗
bool CLanTransform::LoadTable(const char* lpFileName, const bool bIsBIG5)
{
    FILE*           fp;
    fpos_t          pos;
    unsigned long   fileSize;
    unsigned char*  pData;

    if (bIsBIG5)
    {
        // 如果數(shù)據(jù)已經(jīng)有讀入
        if (pBIG5)
        {
            // 釋放原來讀入的數(shù)據(jù)準(zhǔn)備重新讀入
            this->FreeTableBuf(bIsBIG5);
            pBIG5 = 0;
        }
    }else
    {
        // 如果數(shù)據(jù)已經(jīng)有讀入
        if (pGB2312)
        {
            // 釋放原來讀入的數(shù)據(jù)準(zhǔn)備重新讀入
            this->FreeTableBuf(bIsBIG5);
            pGB2312 = 0;
        }
    }

    // 打開文件
    if ((fp=fopen(lpFileName, "rb"))==NULL)
    {
        return false;   // 文件打開失敗
    }

    // 取得文件大小
    fseek(fp, 0, SEEK_END);
    fgetpos( fp, &pos );
    fileSize = (unsigned long)(pos&0x00000000FFFFFFFF);

    // 跳回到文件頭部
    fseek(fp, 0, SEEK_SET);

    // 在這里跳過文件頭部信息
    // ...

    // 分配內(nèi)存
    pData = new unsigned char[fileSize];
    if (pData==NULL)    // 如果內(nèi)存分配失敗
    {
        fclose(fp);     // 關(guān)閉文件
        return false;   // 內(nèi)存分配失敗
    }

    // 讀取文件數(shù)據(jù)到內(nèi)存中
    fread(pData, fileSize, 1, fp);

    // 關(guān)閉文件
    fclose(fp);

    if (bIsBIG5)
    {
        pBIG5 = pData;
    }else
    {
        pGB2312 = pData;
    }
    return true;    // 完成
}

// 釋放字符表
//   bIsBIG5 是否是BIG5的字符表
bool CLanTransform::FreeTableBuf(const bool bIsBIG5)
{
    if (bIsBIG5)
    {
        if (pBIG5)      // 如果有數(shù)據(jù)
        {
            delete[] pBIG5; // 刪除數(shù)據(jù)
            pBIG5 = 0 ;     // 清空指針
        }
    }else
    {
        if (pGB2312)        // 如果有數(shù)據(jù)
        {
            delete[] pGB2312;   // 刪除數(shù)據(jù)
            pGB2312 = 0 ;       // 清空指針
        }
    }
    return true;
}