/*
* 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: 
*
*/
#include "itk.h"

#include "thaianalysis.h"

#include "CLucene.h"

#include <iostream>

#include "testutils.h"

#include "evaluationtool.h"
#include "analysisunittest.h"
#include "cjkanalyzer.h"
#include "ngram.h"

using namespace analysis; 
using namespace evaluationtool; 
using namespace lucene::util;

using namespace lucene::analysis; 

template<int N>
void KoreanNgramAnalyzerTest(Itk::TestMgr* /*testMgr*/) {
	CjkNGramAnalyzer analyzer(N);
	
	Corpus corpus(KOREAN_TEXTCORPUS); 
	
	for (int i = 0; i < corpus.size(); i++) {
		printTokens(analyzer, corpus[i]);
	}
}

template<int N>
void JamuNgramAnalyzerTest(Itk::TestMgr* /*testMgr*/) {
    JamuNGramAnalyzer analyzer(N);
    
    Corpus corpus(KOREAN_TEXTCORPUS); 
    
    for (int i = 0; i < corpus.size(); i++) {
        printTokens(analyzer, corpus[i]);
    }
}

template<int N>
void ChinesePrcNgramAnalyzerTest(Itk::TestMgr* /*testMgr*/) {
	CjkNGramAnalyzer analyzer(N);
	
	Corpus corpus(CHINESE_PRC_TEXTCORPUS); 
	
	for (int i = 0; i < corpus.size(); i++) {
		printTokens(analyzer, corpus[i]);
	}
}

Itk::TesterBase * CreateNgramAnalyzerUnitTest() 
{
	using namespace Itk;
	SuiteTester
		* testSuite = 
			new SuiteTester( "ngram" );
	
	testSuite->add( "korean_1gram", KoreanNgramAnalyzerTest<1>, "korean_1gram" );
    testSuite->add( "korean_2gram", KoreanNgramAnalyzerTest<2>, "korean_2gram" );
	testSuite->add( "chinese_prc_1gram", ChinesePrcNgramAnalyzerTest<1>, "chinese_prc_1gram" );
    testSuite->add( "chinese_prc_2gram", ChinesePrcNgramAnalyzerTest<2>, "chinese_prc_2gram" );
    testSuite->add( "jamu_1gram", JamuNgramAnalyzerTest<1>, "jamu_1gram" );
    testSuite->add( "jamu_2gram", JamuNgramAnalyzerTest<2>, "jamu_2gram" );
    
	return testSuite;
}