/* OTGrammar_ex_tongueRoot.c
 *
 * Copyright (C) 1997-2007 Paul Boersma
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or (at
 * your option) any later version.
 *
 * This program is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU General Public License for more details.
 *
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.
 */

/*
 * pb 2002/07/16 GPL
 * pb 2007/07/23 constraint plasticity
 * pb 2007/08/12 wchar_t
 */

#include "OTGrammar.h"

static const wchar_t *vowels [] = { L"i", L"e", L"\\sw", L"\\ic", L"\\ep", L"a" };
#define i  0
#define e  1
#define schwa  2
#undef I
#define I  3
#define E  4
#define a  5
#define isatr(v)  ((v) <= schwa)
#define isrtr(v)  ((v) >= I)
#define fliptr(v)  (((v) + 3) % 6)

static void countVowelViolations (int *marks, int ncons, int v) {
	if (v == I) marks [1] ++;
	if (v == schwa) marks [2] ++;
	if (ncons == 9) {
		if (v == E) marks [6] ++;
		if (v == a) marks [7] ++;
		if (v == e) marks [8] ++;
		if (v == i) marks [9] ++;
	}
}

static void OTGrammarCandidate_init (OTGrammarCandidate me, int ncons, int v1, int v2) {
	wchar_t buffer [100];
	swprintf (buffer, 100, L"%lst%ls", vowels [v1], vowels [v2]);
	my output = Melder_wcsdup_e (buffer); iferror return;
	my marks = NUMivector (1, my numberOfConstraints = ncons); iferror return;
	/*
	 * Count vowel-gesture violations.
	 */
	countVowelViolations (my marks, ncons, v1);
	countVowelViolations (my marks, ncons, v2);
	/*
	 * Count contour-gesture violations.
	 */
	if (isatr (v1) != isatr (v2)) my marks [5] ++;
}

OTGrammar OTGrammar_create_tongueRoot_grammar (int small_large, int equal_random_infant_Wolof) {
	int icons, ncons = small_large == 1 ? 5 : 9, itab, v1, v2;
	OTGrammar me = new (OTGrammar); cherror
	my constraints = NUMstructvector (OTGrammarConstraint, 1, my numberOfConstraints = ncons); cherror
	my constraints [1]. name = Melder_wcsdup_e (L"*[rtr / hi]"); cherror
	my constraints [2]. name = Melder_wcsdup_e (L"*[atr / lo]"); cherror
	my constraints [3]. name = Melder_wcsdup_e (L"P\\s{ARSE}\n(rtr)"); cherror
	my constraints [4]. name = Melder_wcsdup_e (L"P\\s{ARSE}\n(atr)"); cherror
	my constraints [5]. name = Melder_wcsdup_e (L"*G\\s{ESTURE}\n(contour)"); cherror
	if (ncons == 9) {
		my constraints [6]. name = Melder_wcsdup_e (L"*[rtr / mid]"); cherror
		my constraints [7]. name = Melder_wcsdup_e (L"*[rtr / lo]"); cherror
		my constraints [8]. name = Melder_wcsdup_e (L"*[atr / mid]"); cherror
		my constraints [9]. name = Melder_wcsdup_e (L"*[atr / hi]"); cherror
	}
	if (equal_random_infant_Wolof == 1) {   /* equal? */
		for (icons = 1; icons <= ncons; icons ++)
			my constraints [icons]. ranking = 100.0;
	} else if (equal_random_infant_Wolof == 2) {   /* random? */
		for (icons = 1; icons <= ncons; icons ++)
			my constraints [icons]. ranking = NUMrandomGauss (100.0, 10.0);
	} else if (equal_random_infant_Wolof == 3) {   /* infant (= cannot speak) ? */
		for (icons = 1; icons <= ncons; icons ++)
			my constraints [icons]. ranking = 100.0;   /* Structural constraints. */
		my constraints [3]. ranking = 50.0;   /* Faithfulness constraints. */
		my constraints [4]. ranking = 50.0;
	} else {   /* adult Wolof */
		my constraints [1]. ranking = 100.0;
		my constraints [2]. ranking =  10.0;
		my constraints [3]. ranking =  50.0;
		my constraints [4]. ranking =  20.0;
		my constraints [5]. ranking =  30.0;
		if (ncons == 9) {
			my constraints [6]. ranking =   0.0;
			my constraints [7]. ranking = -10.0;
			my constraints [8]. ranking =   0.0;
			my constraints [9]. ranking = -10.0;
		}
	}
	if (ncons == 9) {
		my fixedRankings = NUMstructvector (OTGrammarFixedRanking, 1, my numberOfFixedRankings = 4); cherror
		my fixedRankings [1]. higher = 1, my fixedRankings [1]. lower = 6;
		my fixedRankings [2]. higher = 6, my fixedRankings [2]. lower = 7;
		my fixedRankings [3]. higher = 2, my fixedRankings [3]. lower = 8;
		my fixedRankings [4]. higher = 8, my fixedRankings [4]. lower = 9;
	}
	my tableaus = NUMstructvector (OTGrammarTableau, 1, my numberOfTableaus = 36); cherror
	itab = 1;
	for (v1 = 0; v1 < 6; v1 ++) for (v2 = 0; v2 < 6; v2 ++) {
		OTGrammarTableau tableau = & my tableaus [itab];
		wchar_t buffer [100];
		swprintf (buffer, 100, L"%lst%ls", vowels [v1], vowels [v2]);
		tableau -> input = Melder_wcsdup_e (buffer); cherror
		tableau -> candidates = NUMstructvector (OTGrammarCandidate, 1, tableau -> numberOfCandidates = 4); cherror
		/*
		 * Generate the four tongue-root variants as output candidates.
		 */
		OTGrammarCandidate_init (& tableau -> candidates [1], ncons, v1, v2); cherror
			/* Faithful: no PARSE constraints violated. */
		OTGrammarCandidate_init (& tableau -> candidates [2], ncons, fliptr (v1), v2); cherror
			/* First vowel flipped: violated one PARSE constraint. */
		OTGrammarCandidate_init (& tableau -> candidates [3], ncons, v1, fliptr (v2)); cherror
			/* Second vowel flipped. */
		OTGrammarCandidate_init (& tableau -> candidates [4], ncons, fliptr (v1), fliptr (v2)); cherror
			/* Both vowels flipped. */
		/*
		 * Count PARSE violations.
		 */
		if (isatr (v1)) {
			tableau -> candidates [2]. marks [4] ++;
			tableau -> candidates [4]. marks [4] ++;
		} else {
			tableau -> candidates [2]. marks [3] ++;
			tableau -> candidates [4]. marks [3] ++;
		}
		if (isatr (v2)) {
			tableau -> candidates [3]. marks [4] ++;
			tableau -> candidates [4]. marks [4] ++;
		} else {
			tableau -> candidates [3]. marks [3] ++;
			tableau -> candidates [4]. marks [3] ++;
		}
		itab ++;
	}
	OTGrammar_checkIndex (me);
	OTGrammar_newDisharmonies (me, 0.0);
	for (icons = 1; icons <= my numberOfConstraints; icons ++)
		my constraints [icons]. plasticity = 1.0;
end:
	iferror forget (me);
	return me;
}

/* End of file OTGrammar_ex_tongueRoot.c */
