package edu.stanford.nlp.semgraph.semgrex.ssurgeon;

import static org.junit.Assert.*;

import java.util.Collection;

import org.junit.Test;

import edu.stanford.nlp.ling.CoreAnnotations;
import edu.stanford.nlp.ling.IndexedWord;
import edu.stanford.nlp.semgraph.SemanticGraph;
import edu.stanford.nlp.semgraph.semgrex.SemgrexPattern;
import edu.stanford.nlp.trees.EnglishGrammaticalRelations;

public class SsurgeonTest {

	/**
	 * Simple test of an Ssurgeon edit script.  This instances a simple semantic graph,
	 * a semgrex pattern, and then the resulting actions over the named nodes in the
	 * semgrex match.
	 */
	@Test
	public void simpleTest() throws Exception {
		SemanticGraph sg = SemanticGraph.valueOf("[mixed/VBN nsubj>[Joe/NNP appos>[bartender/NN det>the/DT]]  dobj>[drink/NN det>a/DT]]");
		SemgrexPattern semgrexPattern = SemgrexPattern.compile("{}=a1 >appos=e1 {}=a2 <nsubj=e2 {}=a3");
		SsurgeonPattern pattern = new SsurgeonPattern(semgrexPattern);

		System.out.println("Start = "+sg.toCompactString());

		// Find and snip the appos and root to nsubj links
		SsurgeonEdit apposSnip = new RemoveNamedEdge("e1", "a1", "a2");
		pattern.addEdit(apposSnip);

		SsurgeonEdit nsubjSnip = new RemoveNamedEdge("e2", "a3", "a1");
		pattern.addEdit(nsubjSnip);

		// Attach Joe to be the nsubj of bartender
		SsurgeonEdit reattachSubj = new AddEdge("a2", "a1", EnglishGrammaticalRelations.NOMINAL_SUBJECT);
		pattern.addEdit(reattachSubj);

		// Attach copula
		IndexedWord isNode = new IndexedWord();
		isNode.set(CoreAnnotations.TextAnnotation.class, "is");
		isNode.set(CoreAnnotations.LemmaAnnotation.class, "is");
		isNode.set(CoreAnnotations.OriginalTextAnnotation.class, "is");
		isNode.set(CoreAnnotations.PartOfSpeechAnnotation.class, "VBN");
		SsurgeonEdit addCopula = new AddDep("a2", EnglishGrammaticalRelations.COPULA, isNode);
		pattern.addEdit(addCopula);

		// Destroy subgraph
		SsurgeonEdit destroySubgraph = new DeleteGraphFromNode("a3");
		pattern.addEdit(destroySubgraph);

		// Process and output modified
		Collection<SemanticGraph> newSgs = pattern.execute(sg);
		for (SemanticGraph newSg : newSgs)
			System.out.println("Modified = "+newSg.toCompactString());
		String firstGraphString = newSgs.iterator().next().toCompactString().trim();
		assertEquals(firstGraphString, "[bartender cop>is nsubj>Joe det>the]");
	}
}