mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
724 lines
20 KiB
724 lines
20 KiB
//+---------------------------------------------------------------------------
|
|
//
|
|
// Microsoft Windows
|
|
// Copyright (C) Microsoft Corporation, 1997
|
|
//
|
|
// File: IWBreak.cxx
|
|
//
|
|
// Contents: Thai Word Breaker glue code
|
|
//
|
|
// History: weibz, 10-Nov-1997 created
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
#include <pch.cxx>
|
|
|
|
#include "iwbreak.hxx"
|
|
|
|
extern long gulcInstances;
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CWordBreaker::CWordBreaker
|
|
//
|
|
// Synopsis: Constructor for the CWordBreaker class.
|
|
//
|
|
// Arguments: [lcid] -- locale id
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
CWordBreaker::CWordBreaker( LCID lcid )
|
|
: _cRefs(1),
|
|
_lcid(lcid)
|
|
{
|
|
|
|
InterlockedIncrement( &gulcInstances );
|
|
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CWordBreaker::~CWordBreaker
|
|
//
|
|
// Synopsis: Destructor for the CWordBreaker class.
|
|
//
|
|
// Notes: All termination/deallocation is done by embedded smart pointers
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
CWordBreaker::~CWordBreaker()
|
|
{
|
|
InterlockedDecrement( &gulcInstances );
|
|
|
|
|
|
}
|
|
|
|
//+-------------------------------------------------------------------------
|
|
//
|
|
// Method: CWordBreaker::QueryInterface
|
|
//
|
|
// Synopsis: Rebind to other interface
|
|
//
|
|
// Arguments: [riid] -- IID of new interface
|
|
// [ppvObject] -- New interface * returned here
|
|
//
|
|
// Returns: S_OK if bind succeeded, E_NOINTERFACE if bind failed
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
SCODE STDMETHODCALLTYPE
|
|
CWordBreaker::QueryInterface( REFIID riid, void ** ppvObject)
|
|
{
|
|
//
|
|
// Optimize QueryInterface by only checking minimal number of bytes.
|
|
//
|
|
// IID_IUnknown = 00000000-0000-0000-C000-000000000046
|
|
// IID_IWordBreaker = D53552C8-77E3-101A-B552-08002B33B0E6
|
|
// --------
|
|
// |
|
|
// +--- Unique!
|
|
//
|
|
|
|
Assert( (IID_IUnknown.Data1 & 0x000000FF) == 0x00 );
|
|
Assert( (IID_IWordBreaker.Data1 & 0x000000FF) == 0xC8 );
|
|
|
|
IUnknown *pUnkTemp;
|
|
SCODE sc = S_OK;
|
|
|
|
switch( riid.Data1 )
|
|
{
|
|
case 0x00000000:
|
|
if ( memcmp( &IID_IUnknown, &riid, sizeof(riid) ) == 0 )
|
|
pUnkTemp = (IUnknown *)this;
|
|
else
|
|
sc = E_NOINTERFACE;
|
|
break;
|
|
|
|
case 0xD53552C8:
|
|
if ( memcmp( &IID_IWordBreaker, &riid, sizeof(riid) ) == 0 )
|
|
pUnkTemp = (IUnknown *)(IWordBreaker *)this;
|
|
else
|
|
sc = E_NOINTERFACE;
|
|
break;
|
|
|
|
default:
|
|
pUnkTemp = 0;
|
|
sc = E_NOINTERFACE;
|
|
break;
|
|
}
|
|
|
|
if( 0 != pUnkTemp )
|
|
{
|
|
*ppvObject = (void * )pUnkTemp;
|
|
pUnkTemp->AddRef();
|
|
}
|
|
else
|
|
*ppvObject = 0;
|
|
return(sc);
|
|
}
|
|
|
|
|
|
//+-------------------------------------------------------------------------
|
|
//
|
|
// Method: CWordBreaker::AddRef
|
|
//
|
|
// Synopsis: Increments refcount
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
ULONG STDMETHODCALLTYPE
|
|
CWordBreaker::AddRef()
|
|
{
|
|
return InterlockedIncrement( &_cRefs );
|
|
}
|
|
|
|
//+-------------------------------------------------------------------------
|
|
//
|
|
// Method: CWordBreaker::Release
|
|
//
|
|
// Synopsis: Decrement refcount. Delete if necessary.
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
ULONG STDMETHODCALLTYPE
|
|
CWordBreaker::Release()
|
|
{
|
|
unsigned long uTmp = InterlockedDecrement( &_cRefs );
|
|
|
|
if ( 0 == uTmp )
|
|
delete this;
|
|
|
|
|
|
return(uTmp);
|
|
}
|
|
|
|
//+-------------------------------------------------------------------------
|
|
//
|
|
// Method: CWordBreaker::Init
|
|
//
|
|
// Synopsis: Initialize word-breaker
|
|
//
|
|
// Arguments: [fQuery] -- TRUE if query-time
|
|
// [ulMaxTokenSize] -- Maximum size token stored by caller
|
|
// [pfLicense] -- Set to true if use restricted
|
|
//
|
|
// Returns: Status code
|
|
//
|
|
//--------------------------------------------------------------------------
|
|
|
|
SCODE STDMETHODCALLTYPE
|
|
CWordBreaker::Init(
|
|
BOOL fQuery,
|
|
ULONG ulMaxTokenSize,
|
|
BOOL *pfLicense )
|
|
{
|
|
if ( NULL == pfLicense ) {
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
|
|
if (IsBadWritePtr(pfLicense, sizeof(DWORD))) {
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
*pfLicense = TRUE;
|
|
_fQuery = fQuery;
|
|
_ulMaxTokenSize = ulMaxTokenSize;
|
|
|
|
|
|
return S_OK;
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CWordBreaker::ComposePhrase
|
|
//
|
|
// Synopsis: Convert a noun and a modifier into a phrase.
|
|
//
|
|
// Arguments: [pwcNoun] -- pointer to noun.
|
|
// [cwcNoun] -- count of chars in pwcNoun
|
|
// [pwcModifier] -- pointer to word modifying pwcNoun
|
|
// [cwcModifier] -- count of chars in pwcModifier
|
|
// [ulAttachmentType] -- relationship between pwcNoun &pwcModifier
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
SCODE STDMETHODCALLTYPE
|
|
CWordBreaker::ComposePhrase(
|
|
WCHAR const *pwcNoun,
|
|
ULONG cwcNoun,
|
|
WCHAR const *pwcModifier,
|
|
ULONG cwcModifier,
|
|
ULONG ulAttachmentType,
|
|
WCHAR *pwcPhrase,
|
|
ULONG *pcwcPhrase )
|
|
{
|
|
//
|
|
// Need to code in later
|
|
//
|
|
if ( _fQuery )
|
|
return( E_NOTIMPL );
|
|
else
|
|
return ( WBREAK_E_QUERY_ONLY );
|
|
}
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CWordBreaker::GetLicenseToUse
|
|
//
|
|
// Synopsis: Returns a pointer to vendors license information
|
|
//
|
|
// Arguments: [ppwcsLicense] -- ptr to ptr to which license info is returned
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
SCODE STDMETHODCALLTYPE
|
|
CWordBreaker::GetLicenseToUse(
|
|
const WCHAR **ppwcsLicense )
|
|
{
|
|
|
|
static WCHAR const * wcsCopyright = L"Copyright Microsoft, 1991-1998";
|
|
|
|
|
|
if ( NULL == ppwcsLicense ) {
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if (IsBadWritePtr(ppwcsLicense, sizeof(DWORD))) {
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
*ppwcsLicense = wcsCopyright;
|
|
return( S_OK );
|
|
}
|
|
|
|
|
|
//+---------------------------------------------------------------------------
|
|
//
|
|
// Member: CWordBreaker::BreakText
|
|
//
|
|
// Synopsis: Break input stream into words.
|
|
//
|
|
// Arguments: [pTextSource] -- source of Unicode text
|
|
// [pWordSink] -- sink for collecting words
|
|
// [pPhraseSink] -- sink for collecting phrases
|
|
//
|
|
// History: 10-Nov-1997, WeibZ, Created.
|
|
//
|
|
// Notes: Since the input buffer may be greater than MAX_II_BUFFER_LEN
|
|
// we process the buffer in chunks of length MAX_II_BUFFER_LEN.
|
|
//
|
|
//----------------------------------------------------------------------------
|
|
|
|
SCODE STDMETHODCALLTYPE CWordBreaker::BreakText( TEXT_SOURCE *pTextSource,
|
|
IWordSink *pWordSink,
|
|
IPhraseSink *pPhraseSink )
|
|
{
|
|
SCODE sc = S_OK;
|
|
ULONG cwc;
|
|
SCRIPT_ITEM *pItems, *pItem_Next, *pItem_org;
|
|
SCRIPT_ANALYSIS *psa;
|
|
PCWSTR pwcInChars;
|
|
INT iItems;
|
|
BOOL bItemProc;
|
|
PCWSTR pwcChars;
|
|
INT cChars;
|
|
HRESULT retUSP;
|
|
|
|
|
|
if ( NULL == pTextSource ) {
|
|
return E_INVALIDARG;
|
|
}
|
|
|
|
if ( NULL == pWordSink )
|
|
{
|
|
// BUGBUG, propagate the null word sink error code
|
|
return sc;
|
|
}
|
|
|
|
|
|
if ( 0 != pPhraseSink )
|
|
{
|
|
// ignore the phrase sink for now
|
|
// return sc;
|
|
}
|
|
|
|
if (pTextSource->iEnd == pTextSource->iCur) {
|
|
return S_OK;
|
|
}
|
|
|
|
Assert( pTextSource->iCur < pTextSource->iEnd );
|
|
|
|
|
|
__try
|
|
{
|
|
do {
|
|
|
|
if ( pTextSource->iCur >= pTextSource->iEnd )
|
|
continue;
|
|
|
|
cwc = pTextSource->iEnd - pTextSource->iCur;
|
|
pwcInChars = pTextSource->awcBuffer + pTextSource->iCur;
|
|
|
|
|
|
pItems = (SCRIPT_ITEM *)LocalAlloc(LPTR,sizeof(SCRIPT_ITEM)*(cwc+1));
|
|
|
|
if ( !pItems) {
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
pItem_org = pItems;
|
|
|
|
|
|
iItems = 0;
|
|
retUSP = ScriptItemize(pwcInChars,cwc,cwc+1, NULL, NULL,
|
|
pItems, &iItems);
|
|
|
|
if (retUSP != S_OK) {
|
|
LocalFree(pItem_org);
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
while ( iItems > 1 ) {
|
|
|
|
pItem_Next = pItems + 1;
|
|
pwcChars = pwcInChars + pItems->iCharPos;
|
|
cChars = pItem_Next->iCharPos - pItems->iCharPos;
|
|
|
|
bItemProc = ProcessItem(pwcChars,
|
|
cChars,
|
|
pItems,
|
|
FALSE, // no need to keep chars
|
|
pTextSource,
|
|
pWordSink,
|
|
pPhraseSink);
|
|
|
|
if ( bItemProc == FALSE ) {
|
|
LocalFree(pItem_org);
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
pItems++;
|
|
iItems--;
|
|
|
|
}
|
|
|
|
// special handle for the last item
|
|
if ( iItems == 1 ) {
|
|
|
|
pwcChars = pwcInChars + pItems->iCharPos;
|
|
cChars = pTextSource->iEnd - pTextSource->iCur;
|
|
|
|
bItemProc = ProcessItem(pwcChars,
|
|
cChars,
|
|
pItems,
|
|
TRUE, // need to keep chars
|
|
pTextSource,
|
|
pWordSink,
|
|
pPhraseSink);
|
|
|
|
if ( bItemProc == FALSE ) {
|
|
LocalFree(pItem_org);
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
if (pItem_org)
|
|
LocalFree(pItem_org);
|
|
|
|
} while ( SUCCEEDED(pTextSource->pfnFillTextBuffer(pTextSource)) );
|
|
|
|
|
|
if ( pTextSource->iCur < pTextSource->iEnd ) {
|
|
|
|
cwc = pTextSource->iEnd - pTextSource->iCur;
|
|
pwcInChars = pTextSource->awcBuffer + pTextSource->iCur;
|
|
|
|
pItems = (SCRIPT_ITEM *)LocalAlloc(LPTR,sizeof(SCRIPT_ITEM)*(cwc+1));
|
|
|
|
if ( !pItems ) {
|
|
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
pItem_org = pItems;
|
|
|
|
|
|
iItems = 0;
|
|
retUSP = ScriptItemize(pwcInChars,cwc,cwc+1, NULL, NULL,
|
|
pItems, &iItems);
|
|
|
|
if (retUSP != S_OK) {
|
|
LocalFree(pItem_org);
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
while ( iItems > 1 ) {
|
|
|
|
pItem_Next = pItems + 1;
|
|
pwcChars = pwcInChars + pItems->iCharPos;
|
|
cChars = pItem_Next->iCharPos - pItems->iCharPos;
|
|
|
|
bItemProc = ProcessItem(pwcChars,
|
|
cChars,
|
|
pItems,
|
|
FALSE, // no need to keep chars
|
|
pTextSource,
|
|
pWordSink,
|
|
pPhraseSink);
|
|
|
|
if ( bItemProc == FALSE ) {
|
|
LocalFree(pItem_org);
|
|
return E_UNEXPECTED;
|
|
}
|
|
|
|
pItems++;
|
|
iItems--;
|
|
|
|
}
|
|
|
|
if ( iItems == 1 ) {
|
|
|
|
pwcChars = pwcInChars + pItems->iCharPos;
|
|
cChars = pTextSource->iEnd - pTextSource->iCur;
|
|
|
|
bItemProc = ProcessItem(pwcChars,
|
|
cChars,
|
|
pItems,
|
|
FALSE, // no need to keep chars
|
|
pTextSource,
|
|
pWordSink,
|
|
pPhraseSink);
|
|
|
|
if ( bItemProc == FALSE ) {
|
|
LocalFree(pItem_org);
|
|
return E_UNEXPECTED;
|
|
}
|
|
}
|
|
|
|
if ( pItem_org )
|
|
LocalFree(pItem_org);
|
|
}
|
|
|
|
} __except(1) {
|
|
|
|
sc = E_UNEXPECTED;
|
|
}
|
|
|
|
return sc;
|
|
}
|
|
|
|
BOOL CWordBreaker::ProcessItem(
|
|
PCWSTR pwcChars,
|
|
INT cChars,
|
|
SCRIPT_ITEM *pItems,
|
|
BOOL fKeep,
|
|
TEXT_SOURCE *pTextSource,
|
|
IWordSink *pWordSink,
|
|
IPhraseSink *pPhraseSink )
|
|
{
|
|
|
|
SCRIPT_LOGATTR *psla, *psla_org, *pslatmp;
|
|
INT iChar,i;
|
|
INT iWord, iWordStart, iWordLen;
|
|
PTHAIWORD pThaiWord, pThaiWordTmp;
|
|
BOOL fLastIsWhiteSpace=FALSE;
|
|
|
|
const SCRIPT_PROPERTIES **pScript_Properties;
|
|
DWORD LangID;
|
|
WORD iScript;
|
|
|
|
HRESULT retUSP;
|
|
|
|
ScriptGetProperties(&pScript_Properties, NULL);
|
|
|
|
iScript = pItems->a.eScript;
|
|
|
|
LangID = (pScript_Properties[iScript])->langid;
|
|
|
|
switch (LangID) {
|
|
|
|
case LANG_THAI: //Thai script, there are SCRIPT_THAI & SCRIPT_THAI_NUM
|
|
|
|
// if (pScript_Properties[iScript].fNeedsWordBreaking == FALSE) {
|
|
// // this is SCRIPT_THAI_NUM,
|
|
//
|
|
// for ( iChar=0; iChar < cChars; iChar++ ) {
|
|
// pWordSink->PutWord(1,
|
|
// pwcChars+iChar,
|
|
// 1,
|
|
// pTextSource->iCur);
|
|
// pTextSource->iCur ++;
|
|
// }
|
|
//
|
|
// break;
|
|
//
|
|
// }
|
|
//
|
|
// // Following code is for SCRIPT_THAI
|
|
|
|
psla = (SCRIPT_LOGATTR *)LocalAlloc(LPTR,
|
|
(sizeof(SCRIPT_LOGATTR)+sizeof(THAIWORD))*cChars);
|
|
psla_org = psla;
|
|
|
|
pThaiWord = (PTHAIWORD)&psla[cChars];
|
|
|
|
if ( psla == NULL )
|
|
return FALSE;
|
|
|
|
retUSP = ScriptBreak(pwcChars, cChars, &(pItems->a), psla);
|
|
|
|
if (retUSP != S_OK) {
|
|
LocalFree(psla_org);
|
|
return FALSE;
|
|
}
|
|
|
|
pThaiWordTmp = pThaiWord;
|
|
|
|
|
|
// check if the last char is a white space.
|
|
if ( psla[cChars-1].fWhiteSpace )
|
|
fLastIsWhiteSpace=TRUE;
|
|
|
|
|
|
iWordStart = -1;
|
|
iWord=0;
|
|
for (iChar=0; iChar < cChars; iChar++) {
|
|
if (psla->fWordStop) {
|
|
if ( iWordStart != -1 ) {
|
|
|
|
iWordLen = iChar - iWordStart;
|
|
if (iWordLen > 0) {
|
|
pThaiWordTmp->offset = iWordStart;
|
|
pThaiWordTmp->len = iWordLen;
|
|
pThaiWordTmp++;
|
|
iWord++;
|
|
}
|
|
|
|
}
|
|
iWordStart = iChar;
|
|
}
|
|
|
|
|
|
if ( psla->fWhiteSpace ) {
|
|
|
|
if ( iWordStart != -1 ) {
|
|
|
|
iWordLen = iChar - iWordStart;
|
|
if (iWordLen > 0 ) {
|
|
pThaiWordTmp->offset = iWordStart;
|
|
pThaiWordTmp->len = iWordLen;
|
|
pThaiWordTmp++;
|
|
iWord++;
|
|
}
|
|
|
|
}
|
|
iWordStart = iChar;
|
|
|
|
// skip all the consequent white spaces.
|
|
while ( (psla->fWhiteSpace) && (iChar < cChars) ) {
|
|
iWordStart = iChar;
|
|
iChar ++;
|
|
psla++;
|
|
}
|
|
|
|
if ( iChar < cChars)
|
|
iWordStart = iChar;
|
|
|
|
}
|
|
|
|
psla++;
|
|
}
|
|
|
|
if ( !fLastIsWhiteSpace ) {
|
|
iWordLen = cChars - iWordStart;
|
|
if ( iWordLen > 0 ) {
|
|
pThaiWordTmp->offset = iWordStart;
|
|
pThaiWordTmp->len = iWordLen;
|
|
iWord++;
|
|
}
|
|
}
|
|
|
|
|
|
// if this is the last item, and the last char is a Thai char,
|
|
// we will discard the last 4 words and integrate into the next
|
|
// filled buffer and do again!!!
|
|
|
|
if (fKeep && !fLastIsWhiteSpace)
|
|
iWord = iWord - 4;
|
|
|
|
pThaiWordTmp = pThaiWord;
|
|
for (i=0; i<iWord; i++) {
|
|
|
|
iWordLen = pThaiWordTmp->len;
|
|
iWordStart = pThaiWordTmp->offset;
|
|
pWordSink->PutWord(iWordLen,
|
|
pwcChars+iWordStart,
|
|
iWordLen,
|
|
pTextSource->iCur+iWordStart);
|
|
|
|
pThaiWordTmp++;
|
|
}
|
|
|
|
|
|
if ( fKeep && !fLastIsWhiteSpace)
|
|
pTextSource->iCur += iWordStart + iWordLen;
|
|
else
|
|
pTextSource->iCur += cChars;
|
|
|
|
LocalFree(psla_org);
|
|
|
|
break;
|
|
|
|
case LANG_ENGLISH : // handle English chars
|
|
|
|
{
|
|
|
|
BYTE ct;
|
|
BOOL fRomanWord = FALSE;
|
|
CONST WCHAR *pwcInput;
|
|
WT Type;
|
|
|
|
Type = WT_START;
|
|
|
|
pwcInput = pwcChars;
|
|
iWordStart = 0;
|
|
|
|
for (iChar=0; iChar< cChars; iChar++, pwcInput++) {
|
|
|
|
ct = GetCharType(*pwcInput);
|
|
|
|
if ( (ct != WS) && (ct != PS) )
|
|
ct = CH;
|
|
|
|
|
|
switch (ct) {
|
|
case CH :
|
|
if (!fRomanWord) {
|
|
iWordStart = iChar;
|
|
fRomanWord = TRUE;
|
|
Type = WT_ROMAJI;
|
|
}
|
|
break;
|
|
case WS :
|
|
if (fRomanWord) {
|
|
iWordLen = iChar - iWordStart;
|
|
pWordSink->PutWord(iWordLen,
|
|
pwcChars+iWordStart,
|
|
iWordLen,
|
|
pTextSource->iCur);
|
|
|
|
pTextSource->iCur += iWordLen;
|
|
fRomanWord = FALSE;
|
|
}
|
|
|
|
Type = WT_WORD_SEP;
|
|
pTextSource->iCur++;
|
|
break;
|
|
|
|
case PS :
|
|
if (fRomanWord) {
|
|
iWordLen = iChar - iWordStart;
|
|
pWordSink->PutWord(iWordLen,
|
|
pwcChars+iWordStart,
|
|
iWordLen,
|
|
pTextSource->iCur);
|
|
|
|
pTextSource->iCur += iWordLen;
|
|
fRomanWord = FALSE;
|
|
}
|
|
|
|
Type = WT_PHRASE_SEP;
|
|
pWordSink->PutBreak(WORDREP_BREAK_EOS);
|
|
pTextSource->iCur++;
|
|
break;
|
|
|
|
}
|
|
}
|
|
|
|
if ((Type == WT_WORD_SEP) || (Type == WT_PHRASE_SEP))
|
|
break;
|
|
|
|
if ( fKeep )
|
|
break;
|
|
|
|
iWordLen =cChars - iWordStart;
|
|
pWordSink->PutWord(iWordLen,
|
|
pwcChars+iWordStart,
|
|
iWordLen,
|
|
pTextSource->iCur);
|
|
|
|
pTextSource->iCur += iWordLen;
|
|
}
|
|
|
|
break;
|
|
|
|
|
|
default :
|
|
|
|
pTextSource->iCur += cChars;
|
|
break;
|
|
}
|
|
|
|
return TRUE;
|
|
|
|
}
|