Files
libreoffice/xmlhelp/source/cxxhelp/provider/resultsetforquery.cxx
Luboš Luňák 1946794ae0 mass removal of rtl:: prefixes for O(U)String*
Modules sal, salhelper, cppu, cppuhelper, codemaker (selectively) and odk
have kept them, in order not to break external API (the automatic using declaration
is LO-internal).

Change-Id: I588fc9e0c45b914f824f91c0376980621d730f09
2013-04-07 14:23:11 +02:00

354 lines
12 KiB
C++

/* -*- Mode: C++; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 4 -*- */
/*
* This file is part of the LibreOffice project.
*
* This Source Code Form is subject to the terms of the Mozilla Public
* License, v. 2.0. If a copy of the MPL was not distributed with this
* file, You can obtain one at http://mozilla.org/MPL/2.0/.
*
* This file incorporates work covered by the following license notice:
*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed
* with this work for additional information regarding copyright
* ownership. The ASF licenses this file to you under the Apache
* License, Version 2.0 (the "License"); you may not use this file
* except in compliance with the License. You may obtain a copy of
* the License at http://www.apache.org/licenses/LICENSE-2.0 .
*/
#include <comphelper/processfactory.hxx>
#include <com/sun/star/ucb/Command.hpp>
#include <com/sun/star/ucb/XCommandEnvironment.hpp>
#include <com/sun/star/i18n/Transliteration.hpp>
#include <com/sun/star/ucb/XCommandProcessor.hpp>
#include <com/sun/star/lang/Locale.hpp>
#include <com/sun/star/script/XInvocation.hpp>
#include <helpcompiler/HelpSearch.hxx>
#if defined _MSC_VER
#pragma warning(push)
#pragma warning(disable : 4068 4263 4264 4266)
#endif
#if defined(__GNUC__) && defined(HAVE_GCC_VISIBILITY_FEATURE)
# pragma GCC visibility push (default)
#endif
#include <CLucene.h>
#if defined(__GNUC__) && defined(HAVE_GCC_VISIBILITY_FEATURE)
# pragma GCC visibility pop
#endif
#if defined _MSC_VER
#pragma warning(pop)
#endif
#include <rtl/ustring.hxx>
#include <algorithm>
#include <set>
#include <qe/Query.hxx>
#include <qe/DocGenerator.hxx>
#include "resultsetforquery.hxx"
#include "databases.hxx"
using namespace std;
using namespace chelp;
using namespace xmlsearch::excep;
using namespace xmlsearch::qe;
using namespace com::sun::star;
using namespace com::sun::star::ucb;
using namespace com::sun::star::i18n;
using namespace com::sun::star::uno;
using namespace com::sun::star::lang;
struct HitItem
{
OUString m_aURL;
float m_fScore;
HitItem( void ) {}
HitItem( const OUString& aURL, float fScore )
: m_aURL( aURL )
, m_fScore( fScore )
{}
bool operator < ( const HitItem& rHitItem ) const
{
return rHitItem.m_fScore < m_fScore;
}
};
ResultSetForQuery::ResultSetForQuery( const uno::Reference< uno::XComponentContext >& rxContext,
const uno::Reference< XContentProvider >& xProvider,
sal_Int32 nOpenMode,
const uno::Sequence< beans::Property >& seq,
const uno::Sequence< NumberedSortingInfo >& seqSort,
URLParameter& aURLParameter,
Databases* pDatabases )
: ResultSetBase( rxContext,xProvider,nOpenMode,seq,seqSort ),
m_aURLParameter( aURLParameter )
{
Reference< XExtendedTransliteration > xTrans = Transliteration::create( rxContext );
Locale aLocale( aURLParameter.get_language(),
OUString(),
OUString() );
xTrans->loadModule(TransliterationModules_UPPERCASE_LOWERCASE,
aLocale );
vector< vector< OUString > > queryList;
{
sal_Int32 idx;
OUString query = m_aURLParameter.get_query();
while( !query.isEmpty() )
{
idx = query.indexOf( sal_Unicode( ' ' ) );
if( idx == -1 )
idx = query.getLength();
vector< OUString > currentQuery;
OUString tmp(query.copy( 0,idx ));
rtl:: OUString toliterate = tmp;
Sequence<sal_Int32> aSeq;
toliterate = xTrans->transliterate(
tmp,0,tmp.getLength(),aSeq);
currentQuery.push_back( toliterate );
queryList.push_back( currentQuery );
int nCpy = 1 + idx;
if( nCpy >= query.getLength() )
query = OUString();
else
query = query.copy( 1 + idx );
}
}
vector< OUString > aCompleteResultVector;
OUString scope = m_aURLParameter.get_scope();
bool bCaptionsOnly = ( scope.compareToAscii( "Heading" ) == 0 );
sal_Int32 hitCount = m_aURLParameter.get_hitCount();
IndexFolderIterator aIndexFolderIt( *pDatabases, m_aURLParameter.get_module(), m_aURLParameter.get_language() );
OUString idxDir;
bool bExtension = false;
int iDir = 0;
vector< vector<HitItem>* > aIndexFolderResultVectorVector;
bool bTemporary;
while( !(idxDir = aIndexFolderIt.nextIndexFolder( bExtension, bTemporary )).isEmpty() )
{
vector<HitItem> aIndexFolderResultVector;
try
{
vector< vector<HitItem>* > aQueryListResultVectorVector;
set< OUString > aSet,aCurrent,aResultSet;
int nQueryListSize = queryList.size();
if( nQueryListSize > 1 )
hitCount = 2000;
for( int i = 0; i < nQueryListSize; ++i )
{
vector<HitItem>* pQueryResultVector;
if( nQueryListSize > 1 )
{
pQueryResultVector = new vector<HitItem>();
aQueryListResultVectorVector.push_back( pQueryResultVector );
}
else
{
pQueryResultVector = &aIndexFolderResultVector;
}
pQueryResultVector->reserve( hitCount );
OUString aLang = m_aURLParameter.get_language();
const std::vector< OUString >& aListItem = queryList[i];
OUString aNewQueryStr = aListItem[0];
vector<float> aScoreVector;
vector<OUString> aPathVector;
try
{
HelpSearch searcher(aLang, idxDir);
searcher.query(aNewQueryStr, bCaptionsOnly, aPathVector, aScoreVector);
}
catch (CLuceneError &e)
{
SAL_WARN("xmlhelp", "CLuceneError: " << e.what());
}
if( nQueryListSize > 1 )
aSet.clear();
for (unsigned j = 0; j < aPathVector.size(); ++j) {
pQueryResultVector->push_back(HitItem(aPathVector[j], aScoreVector[j]));
if (nQueryListSize > 1)
aSet.insert(aPathVector[j]);
}
// intersect
if( nQueryListSize > 1 )
{
if( i == 0 )
{
aResultSet = aSet;
}
else
{
aCurrent = aResultSet;
aResultSet.clear();
set_intersection( aSet.begin(),aSet.end(),
aCurrent.begin(),aCurrent.end(),
inserter(aResultSet,aResultSet.begin()));
}
}
}
// Combine results in aIndexFolderResultVector
if( nQueryListSize > 1 )
{
for( int n = 0 ; n < nQueryListSize ; ++n )
{
vector<HitItem>* pQueryResultVector = aQueryListResultVectorVector[n];
vector<HitItem>& rQueryResultVector = *pQueryResultVector;
int nItemCount = rQueryResultVector.size();
for( int i = 0 ; i < nItemCount ; ++i )
{
const HitItem& rItem = rQueryResultVector[ i ];
if( (aResultSet.find( rItem.m_aURL )) != aResultSet.end() )
{
HitItem aItemCopy( rItem );
aItemCopy.m_fScore /= nQueryListSize; // To get average score
if( n == 0 )
{
// Use first pass to create entry
aIndexFolderResultVector.push_back( aItemCopy );
}
else
{
// Find entry in vector
int nCount = aIndexFolderResultVector.size();
for( int j = 0 ; j < nCount ; ++j )
{
HitItem& rFindItem = aIndexFolderResultVector[ j ];
if( rFindItem.m_aURL.equals( aItemCopy.m_aURL ) )
{
rFindItem.m_fScore += aItemCopy.m_fScore;
break;
}
}
}
}
}
delete pQueryResultVector;
}
sort( aIndexFolderResultVector.begin(), aIndexFolderResultVector.end() );
}
vector<HitItem>* pIndexFolderHitItemVector = new vector<HitItem>( aIndexFolderResultVector );
aIndexFolderResultVectorVector.push_back( pIndexFolderHitItemVector );
aIndexFolderResultVector.clear();
}
catch (const Exception &e)
{
SAL_WARN("xmlhelp", "Exception: " << e.Message);
}
++iDir;
if( bTemporary )
aIndexFolderIt.deleteTempIndexFolder( idxDir );
} // Iterator
int nVectorCount = aIndexFolderResultVectorVector.size();
vector<HitItem>::size_type* pCurrentVectorIndex = new vector<HitItem>::size_type[nVectorCount];
for( int j = 0 ; j < nVectorCount ; ++j )
pCurrentVectorIndex[j] = 0;
sal_Int32 nTotalHitCount = m_aURLParameter.get_hitCount();
sal_Int32 nHitCount = 0;
while( nHitCount < nTotalHitCount )
{
int iVectorWithBestScore = -1;
float fBestScore = 0.0;
for( int k = 0 ; k < nVectorCount ; ++k )
{
vector<HitItem>& rIndexFolderVector = *aIndexFolderResultVectorVector[k];
if( pCurrentVectorIndex[k] < rIndexFolderVector.size() )
{
const HitItem& rItem = rIndexFolderVector[ pCurrentVectorIndex[k] ];
if( fBestScore < rItem.m_fScore )
{
fBestScore = rItem.m_fScore;
iVectorWithBestScore = k;
}
}
}
if( iVectorWithBestScore == -1 ) // No item left at all
break;
vector<HitItem>& rIndexFolderVector = *aIndexFolderResultVectorVector[iVectorWithBestScore];
const HitItem& rItem = rIndexFolderVector[ pCurrentVectorIndex[iVectorWithBestScore] ];
pCurrentVectorIndex[iVectorWithBestScore]++;
aCompleteResultVector.push_back( rItem.m_aURL );
++nHitCount;
}
delete[] pCurrentVectorIndex;
for( int n = 0 ; n < nVectorCount ; ++n )
{
vector<HitItem>* pIndexFolderVector = aIndexFolderResultVectorVector[n];
delete pIndexFolderVector;
}
sal_Int32 replIdx = OUString( "#HLP#" ).getLength();
OUString replWith = OUString( "vnd.sun.star.help://" );
int nResultCount = aCompleteResultVector.size();
for( int r = 0 ; r < nResultCount ; ++r )
{
OUString aURL = aCompleteResultVector[r];
OUString aResultStr = replWith + aURL.copy(replIdx);
m_aPath.push_back( aResultStr );
}
m_aItems.resize( m_aPath.size() );
m_aIdents.resize( m_aPath.size() );
Command aCommand;
aCommand.Name = OUString( "getPropertyValues" );
aCommand.Argument <<= m_sProperty;
for( m_nRow = 0; sal::static_int_cast<sal_uInt32>( m_nRow ) < m_aPath.size(); ++m_nRow )
{
m_aPath[m_nRow] =
m_aPath[m_nRow] +
OUString( "?Language=" ) +
m_aURLParameter.get_language() +
OUString( "&System=" ) +
m_aURLParameter.get_system();
uno::Reference< XContent > content = queryContent();
if( content.is() )
{
uno::Reference< XCommandProcessor > cmd( content,uno::UNO_QUERY );
cmd->execute( aCommand,0,uno::Reference< XCommandEnvironment >( 0 ) ) >>= m_aItems[m_nRow]; //TODO: check return value of operator >>=
}
}
m_nRow = 0xffffffff;
}
/* vim:set shiftwidth=4 softtabstop=4 expandtab: */