/* Distributions_and_Strings.c
 *
 * Copyright (C) 1997-2002 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 1998/06/21
 * pb 2002/07/16 GPL
 */

#include "Distributions_and_Strings.h"

Strings Distributions_to_Strings (Distributions me, long column, long numberOfStrings) {
	Strings thee = NULL;
	double total = 0.0;
	long istring, irow;
	if (column > my numberOfColumns) { Melder_error ("No column %ld.", column); goto end; }
	if (my numberOfRows < 1) { Melder_error ("No candidates."); goto end; }
	for (irow = 1; irow <= my numberOfRows; irow ++)
		total += my data [irow] [column];
	if (total <= 0.0) { Melder_error ("Column total not positive."); goto end; }
	thee = new (Strings);
	thy numberOfStrings = numberOfStrings;
	thy strings = NUMpvector (1, numberOfStrings); cherror
	for (istring = 1; istring <= numberOfStrings; istring ++) {
		char *string;
		do {
			double rand = NUMrandomUniform (0, total), sum = 0.0;
			for (irow = 1; irow <= my numberOfRows; irow ++) {
				sum += my data [irow] [column];
				if (rand <= sum) break;
			}
		} while (irow > my numberOfRows);   /* Guard against rounding errors. */
		string = my rowLabels [irow];
		if (! string) { Melder_error ("No string in row %ld.", irow); goto end; }
		thy strings [istring] = Melder_strdup (string); cherror
	}
end:
	iferror { forget (thee); return Melder_errorp ("(Distributions_to_Strings:) Not performed."); }
	return thee;
}

Strings Distributions_to_Strings_exact (Distributions me, long column) {
	Strings thee = NULL;
	long total = 0;
	long istring = 0, irow, i;
	if (column > my numberOfColumns) { Melder_error ("No column %ld.", column); goto end; }
	if (my numberOfRows < 1) { Melder_error ("No candidates."); goto end; }
	for (irow = 1; irow <= my numberOfRows; irow ++) {
		double value = my data [irow] [column];
		if (value != floor (value)) { Melder_error ("Non-integer value %.17g in row %ld.", value, irow); goto end; }
		if (value < 0.0) { Melder_error ("Found a negative value %g in row %ld.", value, irow); goto end; }
		total += value;
	}
	if (total <= 0) { Melder_error ("Column total not positive."); goto end; }
	thee = new (Strings);
	thy numberOfStrings = total;
	thy strings = NUMpvector (1, total); cherror
	for (irow = 1; irow <= my numberOfRows; irow ++) {
		long number = my data [irow] [column];
		char *string = my rowLabels [irow];
		if (! string) { Melder_error ("No string in row %ld.", irow); goto end; }
		for (i = 1; i <= number; i ++)
			thy strings [++ istring] = Melder_strdup (string); cherror
	}
	Strings_randomize (thee);
end:
	iferror { forget (thee); return Melder_errorp ("(Distributions_to_Strings_exact:) Not performed."); }
	return thee;
}

Distributions Strings_to_Distributions (Strings me) {
	Distributions thee = NULL;
	long i, idist = 0, j;
	thee = Distributions_create (my numberOfStrings, 1); cherror
	for (i = 1; i <= my numberOfStrings; i ++) {
		char *string = my strings [i];
		long where = 0;
		for (j = 1; j <= idist; j ++)
			if (strequ (thy rowLabels [j], string))
				{ where = j; break; }
		if (where) {
			thy data [j] [1] += 1.0;
		} else {
			thy rowLabels [++ idist] = Melder_strdup (string); cherror
			thy data [idist] [1] = 1.0;
		}
	}
	thy numberOfRows = idist;
	TableOfReal_sortByLabel (thee, 1, 0);
end:
	iferror forget (thee);
	return thee;
}

/* End of file Distributions_and_Strings.c */