import com.aliasi.chunk.Chunk;
import com.aliasi.chunk.Chunking;

import com.aliasi.sentences.MedlineSentenceModel;
import com.aliasi.sentences.SentenceChunker;
import com.aliasi.sentences.SentenceModel;

import com.aliasi.tokenizer.IndoEuropeanTokenizerFactory;
import com.aliasi.tokenizer.TokenizerFactory;

import com.aliasi.util.Files;

import java.io.File;
import java.io.IOException;

import java.util.Iterator;
import java.util.Set;

/** Use SentenceModel to find sentence boundaries in text */
public class SentenceChunkerDemo {

    static final TokenizerFactory TOKENIZER_FACTORY = new IndoEuropeanTokenizerFactory();
    static final SentenceModel SENTENCE_MODEL  = new MedlineSentenceModel();
    static final SentenceChunker SENTENCE_CHUNKER 
	= new SentenceChunker(TOKENIZER_FACTORY,SENTENCE_MODEL);

    public static void main(String[] args) throws IOException {
	File file = new File(args[0]);
	String text = Files.readFromFile(file);
	System.out.println("INPUT TEXT: ");
	System.out.println(text);

	Chunking chunking 
	    = SENTENCE_CHUNKER.chunk(text.toCharArray(),0,text.length());
	Set sentences = chunking.chunkSet();
	if (sentences.size() < 1) {
	    System.out.println("No sentence chunks found.");
	    return;
	}
	String slice = chunking.charSequence().toString();
	int i = 1;
	for (Iterator it = sentences.iterator(); it.hasNext(); ) {
	    Chunk sentence = (Chunk)it.next();
	    int start = sentence.start();
	    int end = sentence.end();
	    System.out.println("SENTENCE "+(i++)+":");
	    System.out.println(slice.substring(start,end));
	}
    }
}