236 lines
6.3 KiB
C++
236 lines
6.3 KiB
C++
// Copyright (C) 1996-1997 Microsoft Corporation. All rights reserved.
|
|
//
|
|
// FTS.H
|
|
//
|
|
#if _MSC_VER > 1000
|
|
#pragma once
|
|
#endif
|
|
|
|
#ifndef __FTS_H__
|
|
#define __FTS_H__
|
|
|
|
#ifndef __SITEMAP_H__
|
|
#include "sitemap.h"
|
|
#endif
|
|
|
|
// Shroom header files
|
|
//
|
|
#include "itquery.h"
|
|
#include "itgroup.h"
|
|
#include "itcc.h"
|
|
#include "itrs.h"
|
|
#include "itdb.h"
|
|
#include "itww.h"
|
|
|
|
class CExCollection;
|
|
class CExTitle;
|
|
class CCombinedFTS;
|
|
class CSubSet;
|
|
class CUWait;
|
|
|
|
#define FTS_TITLE_ONLY 0x0001
|
|
#define FTS_ENABLE_STEMMING 0x0002
|
|
#define FTS_SEARCH_PREVIOUS 0x0004
|
|
|
|
|
|
#define FTS_NO_INDEX MAKE_HRESULT(SEVERITY_ERROR,FACILITY_ITF,100)
|
|
#define FTS_NOT_INITIALIZED MAKE_HRESULT(SEVERITY_ERROR,FACILITY_ITF,101)
|
|
#define FTS_E_SKIP_TITLE MAKE_HRESULT(SEVERITY_ERROR,FACILITY_ITF,102)
|
|
#define FTS_E_SKIP_VOLUME MAKE_HRESULT(SEVERITY_ERROR,FACILITY_ITF,103)
|
|
#define FTS_E_SKIP_ALL MAKE_HRESULT(SEVERITY_ERROR,FACILITY_ITF,104)
|
|
#define FTS_INVALID_SYNTAX MAKE_HRESULT(SEVERITY_ERROR,FACILITY_ITF,105)
|
|
#define FTS_CANCELED MAKE_HRESULT(SEVERITY_ERROR,FACILITY_ITF,106)
|
|
|
|
#define MAX_HIGHLIGHT_TERMS 128
|
|
|
|
typedef struct CHM_MapEntry
|
|
{
|
|
char szChmName[50];
|
|
WORD iIndex;
|
|
FILETIME versioninfo;
|
|
LCID language;
|
|
DWORD dwTopicCount;
|
|
} CHM_MAP_ENTRY;
|
|
|
|
DWORD Han2Zen(unsigned char *lpInBuffer, unsigned char *lpOutBuffer, UINT codepage);
|
|
WCHAR* PreProcessQuery(WCHAR *pwcQuery, UINT codepage);
|
|
BOOL compareOperator(char *pszQuery, char *pszTerm);
|
|
int IsJOperator(char *pszQuery);
|
|
int FASTCALL CompareIntValues(const void *pval1, const void *pval2);
|
|
|
|
ERR SearchMessageFunc(DWORD dwFlag, LPVOID pUserData, LPVOID pMessage);
|
|
|
|
BOOL IsQuery(char *pszQuery);
|
|
|
|
// FTS Results structure
|
|
//
|
|
typedef struct _search_result
|
|
{
|
|
DWORD dwTopicNumber;
|
|
DWORD dwRank;
|
|
CExTitle *pTitle;
|
|
} SEARCH_RESULT;
|
|
|
|
class CTitleFTS;
|
|
|
|
//
|
|
// Topic numbers in IVQ files will be reduced from 32bit numbers to 22bit numbers.
|
|
// The high 10 bits will be used as an CHM ID.
|
|
//
|
|
// Conversion from DWORD to CHM_ID and Topic Number.
|
|
//
|
|
#define CHM_ID(exp) (0x000003ff & (exp >> 22))
|
|
#define TOPIC_NUM(exp) (0x003fffff & exp)
|
|
|
|
// Conversion from CHM_ID and Topic Number to DWORD.
|
|
//
|
|
#define PACKED_TOPICNUM(iTopNum, iChmID) ((iChmID << 22) | iTopNum)
|
|
|
|
// Title array structure
|
|
//
|
|
typedef struct _titleArray
|
|
{
|
|
CExTitle *pExTitle;
|
|
CCombinedFTS *pCombinedFTI;
|
|
BOOL bSearch;
|
|
BOOL bCombinedIndex;
|
|
WORD iTitleIndex;
|
|
FILETIME versioninfo;
|
|
char *pszQueryName;
|
|
char *pszIndexName;
|
|
char *pszShortName;
|
|
LCID language;
|
|
DWORD dwTopicCount;
|
|
UINT uiVolumeOrder;
|
|
BOOL bHasResults;
|
|
BOOL bAlreadyQueried;
|
|
} TITLE_ENTRY;
|
|
|
|
// CFullTextSearch class
|
|
//
|
|
class CFullTextSearch
|
|
{
|
|
public:
|
|
CFullTextSearch(CExCollection *pTitleCollection);
|
|
~CFullTextSearch();
|
|
HRESULT SimpleQuery(WCHAR *pszQuery, int *cResultCount, SEARCH_RESULT **);
|
|
HRESULT ComplexQuery(WCHAR *pszQuery, DWORD dwFlags, int *cResultCount, SEARCH_RESULT **, CSubSet *pSubSet);
|
|
HRESULT AbortQuery();
|
|
HRESULT SetProximity(WORD wNear);
|
|
HRESULT SetResultCount(LONG cRows);
|
|
HRESULT SetOptions(DWORD dwFlag);
|
|
void InitTitleArray(void);
|
|
BOOL LoadCombinedIndex(DWORD);
|
|
CCombinedFTS * GetPreviousInstance(char *pszQueryName);
|
|
CExTitle *LookupTitle(CCombinedFTS *, DWORD);
|
|
HRESULT AddHLTerm(WCHAR *, int len);
|
|
HRESULT CFullTextSearch::AddQueryToTermList(WCHAR *pwsBuffer);
|
|
HRESULT TermListRemoveAll(void);
|
|
WCHAR * GetHLTermAt(int index);
|
|
INT GetHLTermCount(void);
|
|
VOID FreeResults(SEARCH_RESULT *);
|
|
long ComputeResultCount(IITResultSet *pResultSet);
|
|
PCSTR m_pszITSSFile;
|
|
BOOL Initialize();
|
|
long m_lMaxRowCount;
|
|
WORD m_wQueryProximity;
|
|
DWORD m_dwQueryFlags;
|
|
int m_iLastResultCount;
|
|
TITLE_ENTRY *m_pTitleArray;
|
|
BOOL m_bMergedChmSetWithCHQ; // When true, we are running a merged chm set that has
|
|
// a combined index (NT5 Help).
|
|
protected:
|
|
LANGID m_SystemLangID;
|
|
WCHAR *m_HLTermArray[MAX_HIGHLIGHT_TERMS];
|
|
int m_iHLIndex;
|
|
BOOL m_bInit;
|
|
BOOL m_InitFailed;
|
|
BOOL m_SearchActive;
|
|
CExCollection *m_pTitleCollection;
|
|
BOOL m_bTitleArrayInit;
|
|
INT m_TitleArraySize;
|
|
};
|
|
|
|
// CTitleFTS class
|
|
//
|
|
class CTitleFTS
|
|
{
|
|
public:
|
|
CTitleFTS(PCSTR pwszTitlePath, LCID lcid, CExTitle *);
|
|
void ReleaseObjects();
|
|
~CTitleFTS();
|
|
HRESULT Query(WCHAR *pszQuery, DWORD dwFlags, IITResultSet **, CFullTextSearch *pFullTextSearch, CUWait *, int);
|
|
HRESULT AbortQuery();
|
|
IITResultSet * GetResultsSet(void) { return m_pITResultSet; }
|
|
HRESULT SetProximity(WORD wNear);
|
|
HRESULT SetResultCount(LONG cRows);
|
|
HRESULT SetOptions(DWORD dwFlag);
|
|
VOID FreeResults(SEARCH_RESULT *);
|
|
HRESULT Initialize();
|
|
HRESULT UpdateOptions(WORD wNear, LONG cRows);
|
|
protected:
|
|
UINT m_codepage;
|
|
long m_iLastResultCount;
|
|
WCHAR *m_pPrevQuery;
|
|
BOOL m_bInit;
|
|
CExTitle *m_pTitle;
|
|
HRESULT m_InitError;
|
|
BOOL m_InitFailed;
|
|
BOOL m_SearchActive;
|
|
DWORD m_dwQueryFlags;
|
|
LCID m_lcid;
|
|
IITIndex *m_pIndex;
|
|
IITQuery *m_pQuery;
|
|
IITResultSet *m_pITResultSet;
|
|
IITDatabase *m_pITDB;
|
|
LANGID m_SystemLangID;
|
|
LANGID m_langid;
|
|
BOOL m_fDBCS;
|
|
WCHAR m_tcTitlePath[MAX_PATH];
|
|
long m_lMaxRowCount;
|
|
WORD m_wQueryProximity;
|
|
|
|
inline BOOL Init() { if( !m_bInit ) Initialize(); return m_bInit; }
|
|
};
|
|
|
|
// CCombinedFTS class
|
|
//
|
|
class CCombinedFTS
|
|
{
|
|
public:
|
|
CCombinedFTS(CExTitle *, LCID lcid, CFullTextSearch *);
|
|
~CCombinedFTS();
|
|
HRESULT Query(WCHAR *pszQuery, DWORD dwFlags, IITResultSet **, CFullTextSearch *pFullTextSearch, CUWait *, int);
|
|
HRESULT AbortQuery();
|
|
IITResultSet * GetResultsSet(void) { return m_pITResultSet; }
|
|
VOID FreeResults(SEARCH_RESULT *);
|
|
HRESULT Initialize();
|
|
HRESULT UpdateOptions(WORD wNear, LONG cRows);
|
|
void ReleaseObjects();
|
|
protected:
|
|
UINT m_codepage;
|
|
long m_iLastResultCount;
|
|
HRESULT SetProximity(WORD wNear);
|
|
HRESULT SetResultCount(LONG cRows);
|
|
HRESULT SetOptions(DWORD dwFlag);
|
|
CExTitle *m_pTitle;
|
|
CFullTextSearch *m_pFullTextSearch;
|
|
BOOL m_SearchActive;
|
|
DWORD m_dwQueryFlags;
|
|
LCID m_lcid;
|
|
WCHAR *m_pPrevQuery;
|
|
IITIndex *m_pIndex;
|
|
IITQuery *m_pQuery;
|
|
IITResultSet *m_pITResultSet;
|
|
IITDatabase *m_pITDB;
|
|
LANGID m_SystemLangID;
|
|
LANGID m_langid;
|
|
BOOL m_fDBCS;
|
|
WCHAR m_tcTitlePath[MAX_PATH];
|
|
long m_lMaxRowCount;
|
|
WORD m_wQueryProximity;
|
|
};
|
|
|
|
|
|
#endif // __FTS_H__
|