searchengine/cpix/tsrc/cpixunittest/src/pdftests.cpp
author Dremov Kirill (Nokia-D-MSW/Tampere) <kirill.dremov@nokia.com>
Mon, 19 Apr 2010 14:40:16 +0300
changeset 0 671dee74050a
child 1 6f2c1c46032b
permissions -rw-r--r--
Revision: 201011 Kit: 201015

/*
* Copyright (c) 2010 Nokia Corporation and/or its subsidiary(-ies).
* All rights reserved.
* This component and the accompanying materials are made available
* under the terms of "Eclipse Public License v1.0"
* which accompanies this distribution, and is available
* at the URL "http://www.eclipse.org/legal/epl-v10.html".
*
* Initial Contributors:
* Nokia Corporation - initial contribution.
*
* Contributors:
*
* Description: 
*
*/
/*
 * terms.cpp
 *
 *  Created on: 3.3.2010
 *      Author: anide
 */


#include <wchar.h>
#include <stddef.h>

#include <iostream>
#include <sstream>
#include "indevicecfg.h"

#include "cpixidxdb.h"

#include "itk.h"

#include "config.h"
#include "testutils.h"
#include "suggestion.h"

const char * PdfDocsToIndex[5] = {
	FILE_TEST_CORPUS_PATH "\\pdf\\ctutor.pdf",
//	FILE_TEST_CORPUS_PATH "\\pdf\\geology.pdf",
//	FILE_TEST_CORPUS_PATH "\\pdf\\samplepdf.pdf",
//	FILE_TEST_CORPUS_PATH "\\pdf\\windjack.pdf",
    NULL
};


const wchar_t * PdfSearchParameters[5] = {
	L"inline",
//	L"CALDEBA",
//	L"sample",
//	L"reset",
    NULL
};

void pdfTestAppclassFilteredTermSearch(Itk::TestMgr * testMgr, const wchar_t* appclassPrefix)
{
    cpix_Result
        result;

    cpix_IdxDb_dbgScrapAll(&result);

    ITK_ASSERT(testMgr,
               cpix_Succeeded(&result),
               "Could not get rid of all test qbac-idx pairs");

    std::auto_ptr<FileIdxUtil> util( new FileIdxUtil ); 
    
    util->init(); 
    
    cpix_Analyzer* analyzer = cpix_CreateSimpleAnalyzer(&result); 
    
    if ( cpix_Failed( &result) ) ITK_PANIC("Analyzer could not be created");
    
    for (int i = 0; PdfDocsToIndex[i]; i++) 
    {
        util->indexFile( PdfDocsToIndex[i], analyzer, testMgr ); 
    }

    for (int i = 0; Mp3TestCorpus[i]; i++) 
    {
        util->indexFile( Mp3TestCorpus[i], analyzer, testMgr ); 
    }

        util->flush();
    
    for (int i = 0; PdfSearchParameters[i]; i++) 
    {
        cpix_QueryParser
            * queryParser = cpix_QueryParser_create(&result,
                                                    LCPIX_DEFAULT_FIELD,
                                                    analyzer );
        if (queryParser == NULL)
            {
                cpix_Analyzer_destroy( analyzer );
                ITK_PANIC("Could not create query parser");
            }
        
        std::wostringstream queryString;
        if ( appclassPrefix ) {
            queryString<<L"$terms<5,'"<<appclassPrefix<<L"'>("<<PdfSearchParameters[i]<<L")";
        } else {
            queryString<<L"$terms<5>("<<PdfSearchParameters[i]<<L")";
        }
    
        cpix_Query* query = cpix_QueryParser_parse(queryParser,
                                                   queryString.str().c_str());
        if (cpix_Failed(queryParser)
            || query == NULL)
            {
                cpix_Analyzer_destroy(analyzer);
                cpix_ClearError(queryParser);
                cpix_QueryParser_destroy(queryParser);
                ITK_PANIC("Could not parse query string");
            }
        cpix_QueryParser_destroy(queryParser);

        cpix_Hits
            * hits = cpix_IdxDb_search(util->idxDb(),
                                       query );
        
        int32_t hitsLength = cpix_Hits_length(hits);
        cpix_Query_destroy( query ); 
        
        wprintf(L"Results for %S:\n", PdfSearchParameters[i]);
        
        Suggestion::printSuggestions(hits,
                                     testMgr);
        
        printf("\n"); 
                
        cpix_Hits_destroy( hits ); 
    }

}
// int32_t hitsLength = cpix_Hits_length(hits);
void CreateSimplePdfSearch(Itk::TestMgr * testMgr) 
{
    pdfTestAppclassFilteredTermSearch(testMgr, LPDFAPPCLASS);
}

Itk::TesterBase * CreatePdfSearchTests()
{
    using namespace Itk;

    ContextTester
        * pdfTests = new ContextTester("pdfTests", NULL);

    pdfTests->add("pdfterms", &CreateSimplePdfSearch);
    
    return pdfTests;
}