package com.dlj.ir.analysis.ru;


import java.util.Map;
import java.util.WeakHashMap;



/**
 * Russian stemmer-  
 * Done by Ljiljana Dolamic(University of Neuchatel, www.unine.ch/info/clef/)
 *
 * @author Ljiljana Doalmic. Email: ljiljana.dolamic@unine.ch
 * -removes case endings form nouns and adjectives and certain derivational suffixes
 */
public class RussianStemmerAggressive {
	/**
	 * A cache of words and their stems
	 */
	static private Map<String, String> cache = new WeakHashMap<String, String>();

	/**
	 * A buffer of the current word being stemmed
	 */
	private StringBuffer sb=new StringBuffer();

	

	/**
	 * Default constructor
	 */
	public RussianStemmerLight(){
	} // constructor

	public String stem(String input){
		String result = cache.get(input);
		if (result != null)
			return result;

		//reset string buffer
		sb.delete(0,sb.length());
		sb.insert(0,input);
		// stemming...
		//removes case endings from nouns and adjectives
		removeCase(sb);
		removeDerivationalSuffixes(sb);
		normalize(sb);
		result = sb.toString();
		cache.put(input, result);

		return result;
	}

	private void normalize(StringBuffer buffer) {
		int len=buffer.length();
		if( len>3){

			if(buffer.substring( len - 1, len).equals("ь")){
				
				buffer.delete( len - 1 , len);
				return;
			}
		
			if(buffer.substring( len - 2, len).equals("нн")){
				
				buffer.delete( len - 1 , len);
				return;
			}
		
			if(buffer.substring( len -1,len).equals("и")){
				
				buffer.delete( len - 1 , len);
				return;
			}


		}//len>3		

	}

	private void removeCase(StringBuffer buffer) {
		int len=buffer.length();
		
		// 
		if( len> 6){
			if(buffer.substring( len - 4 , len).equals("иями")||					
					buffer.substring( len - 4, len).equals("оями")){

				buffer.delete( len - 4 , len);
				return;
			}
		}//len>6
		if( len>5 ){
			if(buffer.substring( len - 3 , len).equals("иям")||
					buffer.substring( len - 3 , len).equals("иях")||
					buffer.substring( len - 3 , len).equals("оях")||
					buffer.substring( len - 3 , len).equals("ями")||
					buffer.substring( len - 3 , en).equals("oям")||
					buffer.substring( len - 3 , en).equals("oьв")||
					buffer.substring( len - 3 , len).equals("ами")||
					buffer.substring( len - 3 , len).equals("его")||  
					buffer.substring( len - 3 , len).equals("емu")||
					buffer.substring( len - 3 , len).equals("ери")||
					buffer.substring( len - 3 , len).equals("ими")||
					buffer.substring( len - 3 , len).equals("ого")|| 
					buffer.substring( len  -3 , len).equals("ому")||
					buffer.substring( len - 3 , len).equals("ыми")||
					buffer.substring( len - 3 , len).equals("оев"))}
					){

				buffer.delete( len - 3 , len);
				return;
			}
			
		}//len>5

		if( len> 4 ){
			if(buffer.substring( len - 2 , len).equals("ая")|| 
					buffer.substring( len - 2 , len).equals("яя")||
					buffer.substring( len - 2 , len).equals("ях")||
					buffer.substring( len - 2, len).equals("юю")||
					buffer.substring( len - 2 , len).equals("ах")|| 
					buffer.substring( len - 2 , len).equals("ею")|| 					  
					buffer.substring( len - 2 , len).equals("их")||
					buffer.substring( len - 2, len).equals("ия")||
					buffer.substring( len - 2, len).equals("ию")||					  
					buffer.substring( len - 2, len).equals("ьв")|| 
					buffer.substring( len - 2, len).equals("ою")||
					buffer.substring( len - 2 , len).equals("ую")||
					buffer.substring( len - 2 , len).equals("ям")||   
					buffer.substring( len - 2 , len).equals("ых")||					
					buffer.substring( len - 2 , len).equals("ея")||
					buffer.substring( len - 2 , len).equals("ам")||
					buffer.substring( len - 2 , len).equals("ем")|| 
					buffer.substring( len - 2 , len).equals("ей")||    
					buffer.substring( len - 2 , len).equals("ем")||    
					buffer.substring( len - 2 , len).equals("ев")||					 
					buffer.substring( len - 2 , len).equals("ий")||   
					buffer.substring( len - 2 , len).equals("им")||
					buffer.substring( len - 2 , len).equals("ое")||   
					buffer.substring( len - 2 , len).equals("ой")||     
					buffer.substring( len - 2 , len).equals("ом")||   
					buffer.substring( len - 2 , len).equals("ов")||
					buffer.substring( len - 2 , len).equals("ыe")|| 
					buffer.substring( len - 2 , len).equals("ыj")||   
					buffer.substring( len - 2 , len).equals("ым")||
					buffer.substring( len - 2 , len).equals("ми")){ 

				buffer.delete( len - 2 , len);
				return;
			}
			
		}  
		
		if( len> 3){
			if( buffer.substring( len - 1 , len).equals("а")||
					buffer.substring( len - 1 , len).equals("е")||
					buffer.substring( len - 1 , len).equals("и")||  
					buffer.substring( len - 1 , len).equals("о")||  
					buffer.substring( len - 1 , len).equals("у")||
					buffer.substring( len - 1 , len).equals("й")||
					buffer.substring( len - 1 , len).equals("ы")||
					buffer.substring( len - 1 , len).equals("я")||
					buffer.substring( len - 1 , len).equals("ь")||){   
				
				buffer.delete( len - 1 , len);
				return;
			}
		}//len>3
	}

	private void removeDerivationalSuffixes(StringBuffer buffer) {
		int len=buffer.length();
		// 
		if( len> 7 ){
			if(buffer.substring( len - 5 , len).equals("ичeсk")||
					buffer.substring( len - 5 , len).equals("тeльн")){

				buffer.delete( len  5, len);
				return;
			}
		
			
		}//len>7
		if( len> 6 ){
			if(buffer.substring( len - 4 , len).equals("oвсk")||
					buffer.substring( len - 4 , len).equals("инсk")||
					buffer.substring( len - 4 , len).equals("eнсk")||
					buffer.substring( len - 4 , len).equals("ийсk")||
					buffer.substring( len - 4 , len).equals("eйсk")||
					buffer.substring( len - 4 , len).equals("oвat")||
					){

				buffer.delete( len - 4 , len);
				return;
			}
			
		}//len>6	
		if( len> 5 ){
			
			if(buffer.substring( len - 3 , len).equals("ebн")|| 
					buffer.substring( len - 3 , len).equals("ств")|| 
					buffer.substring( len - 3 , len).equals("oст")|| 
					buffer.substring( len - 3 , len).equals("ист")||
					buffer.substring( len - 3 , len).equals("лив")||					
					buffer.substring( len - 3 , len).equals("aст")||					
					buffer.substring( len - 3 , len).equals("eмс")||
					buffer.substring( len - 3 , len).equals("чив")||
					buffer.substring( len - 3 , len).equals("ичн")||
					buffer.substring( len - 3 , len).equals("oчн")||
					buffer.substring( len - 3 , len).equals("eцн")||
					buffer.substring( len - 3 , len).equals("льн")||
					buffer.substring( len - 3 , len).equals("eнн")|| 
					buffer.substring( len - 3 , len).equals("oнн")|| 
					buffer.substring( len - 3 , len).equals("oвн")
					){

				buffer.delete( len - 3, len);
				return;
			}
		}  //len>5
		if( len> 4){

			if( buffer.substring( len - 2 , len).equals("сk")|| 
					buffer.substring( len - 2 , len).equals("oв")||  
					buffer.substring( len - 2 , len).equals("ok")||
					buffer.substring( len - 2 , len).equals("aт")||
					buffer.substring( len - 2 , len).equals("aв")||
					buffer.substring( len - 2 , len).equals("ин")||
					buffer.substring( len - 2 , len).equals("eн")||
					buffer.substring( len - 2 , len).equals("eт")||
					buffer.substring( len - 2 , len).equals("eв")||
					buffer.substring( len - 2 , len).equals("ут")||
					buffer.substring( len - 2 , len).equals("ьв")|| 
					buffer.substring( len - 2 , len).equals("eц")||
					buffer.substring( len - 2 , len).equals("шн")||
					buffer.substring( len - 2 , len).equals("ив")){

				buffer.delete( len - 2 , len);
				return;
			}

		}//len>4
		if( len> 3){

			if(buffer.substring( len - 1, len).equals("k")||
					buffer.substring( len - 1 , len).equals("ц")){
				
				buffer.delete( len - 1 , len);
				return;
			}
		}

	}

}