//DESCRIPTION: Sort a table
// Peter kahrel -- www.kahrel.plus.com

// Details at www.kahrel.plus.com/indesign/tablesort.html

#target indesign;
#targetengine 'table_sort';

var TSORT = get_data ();
var keys = build_keys ();

if (TSORT.formatted)
	{
	TSORT.table.rows.add();
	sort_formatted_sub_a (keys, 0, keys.length-1, TSORT.headers, TSORT.table.rows.everyItem().getElements(), TSORT.table.rows[-1], TSORT.ascend, TSORT.table.rows[0].cells.length);
	TSORT.table.rows[-1].remove();
	}
else
	{
	TSORT.ascend ? keys.sort (sort_keys_ascend) : keys.sort (sort_keys_descend);
	for (var i = 0; i < keys.length; i++)
		TSORT.table.rows[i+TSORT.headers].contents = keys[i].contents;
	}


function sort_keys_ascend (a, b) {return a.key > b.key;}
function sort_keys_descend (a, b) {return a.key < b.key;}


function build_keys ()
	{
	var temp = "", str = "", keys = [],
			rows = TSORT.table.rows.everyItem().getElements();
	var dummy = TSORT.ascend ? "~" : ".";
	
	var last = rows.length - TSORT.footers
	for (var i = TSORT.headers; i < last; i++)
		{
		str = "";
		for (var j = 0; j < TSORT.column_order.length; j++)
			{
			temp = TSORT.table.columns[TSORT.column_order[j]].cells[i].contents;
			if (!TSORT.anumeric[j])
				{
				if (TSORT.ignore_case) temp = temp.toUpperCase();
				temp = strip_accents (temp);
				}
			str += pad (temp, TSORT.anumeric[j], TSORT.ascend);
			}
		TSORT.formatted ? keys.push (str) : keys.push ({key: str, contents: rows[i].cells.everyItem().contents})
		}
		return keys;
	} // build_keys_unformatted


function pad (s, numeric, ascend)
	{
	if (numeric)
		{
		if (s == "")
			{
			return ascend ? "9" : "0";
			}
		return ("0000000000"+s).slice(-10);
		}
	
	var filler = ascend ? "ZZZZZZZZZZ" : "..............................";
	return (s+filler).slice(0,30);
	}


function strip_accents (s) {
	return s.replace (/[ÁÀÂÄÅĀĄĂÆ]/g, "A").
		replace (/[ÇĆČĊ]/g, "C").
		replace (/[ĎĐ]/g, "D").
		replace (/[ÉÈÊËĘĒĔĖĚ]/g, "E").
		replace (/[ĢĜĞĠ]/g, "G").
		replace (/[ĤĦ]/g, "H").
		replace (/[ÍÌÎÏĪĨĬĮİ]/g, "I").
		replace (/[ĵ]/g, "J").
		replace (/[ķ]/g, "K").
		replace (/[ŁĹĻĽ]/g, "L").
		replace (/[ÑŃŇŅŊ]/g, "N").
		replace (/[ÓÒÔÖŌŎŐØŒ]/g, "O").
		replace (/[ŔŘŖ]/g, "R").
		replace (/[ŚŠŜŞȘß]/g, "S").
		replace (/[ŢȚŤŦ]/g, "T").
		replace (/[ÚÙÛÜŮŪŲŨŬŰŲ]/g, "U").
		replace (/[Ŵ]/g, "W").
		replace (/[ŸÝŶ]/g, "Y").
		replace (/[ŹŻŽ]/g, "Z")
}


function sort_formatted_sub_a (keys, first, last, head, rows, last_row, dir, row_length)
	{
	var i = first; var j = last;
	var pivot = keys[Math.floor((first+last)/2)];
	while ( i < j )
		{
		while (larger_than (pivot, keys[i], dir))
			i++;
		while (larger_than (keys[j], pivot, dir))
			j--;
		if ( i <= j )
			{
			swap (keys, i, j);
			swap_rows (rows, i+head, j+head, last_row, row_length);
			i++;
			j--;
			}
		} // while i < j
	if (first < j) sort_formatted_sub_a (keys, first, j, head, rows, last_row, dir, row_length);
	if (i < last) sort_formatted_sub_a (keys, i, last, head, rows, last_row, dir, row_length);
	}


function swap (array, x, y)
	{
	var temp = array[x];
	array[x] = array[y];
	array[y] = temp;
	}


/* Swap_rows swaps rows by moving a row x into the dummy row,
row y into row x, and the dummy into row y. As we can't move rows,
we have to move the contents of the cells. */

function swap_rows (r, x, y, last_row, le)
	{
	if (x != y)
	{
		for( var i = 0; i < le; i++ )
			r[x].cells[i].texts[0].move (LocationOptions.after, last_row.cells[i].insertionPoints[0]);
		for( var i = 0; i < le; i++)
			r[y].cells[i].texts[0].move (LocationOptions.after, r[x].cells[i].insertionPoints[0]);
		for( var i = 0; i < le; i++ )
			last_row.cells[i].texts[0].move (LocationOptions.after, r[y].cells[i].insertionPoints[0]);
		}
	}


function larger_than (a, b, ascend)
	{
	if (ascend) return a > b;
		else return a < b;
	}


// Interface =============================================================

function get_data ()
	{
	var table = get_table();
	var location = get_previous();
	var w = new Window ('dialog {text: "Table sort", orientation: "row", alignChildren: "top"}');
	
	if (location != null)
		w.location = location.split(',');
	
	var main = w.add ('panel {alignChildren: "fill", orientation: "column"}');

	var order = main.add ('panel {text: "Sort order", alignChildren: "right"}');
		var columns = [];
		var anum = [false, false, false];
		var g1 = order.add ("group");
			g1.add ('statictext {text: "First sort on column"}');
			columns[0] = g1.add ('edittext {text: ""}');
			anum[0] = g1.add ('checkbox {text: "\u00A0Numeric"}');
		var g2 = order.add ("group");
			g2.add ('statictext {text: "Then on column"}');
			columns[1] = g2.add ('edittext {text: ""}');
			anum[1] = g2.add ('checkbox {text: "\u00A0Numeric"}');
		var g3 = order.add ("group");
			g3.add ('statictext {text: "Then on column"}');
			columns[2] = g3.add ('edittext {text: ""}');
			anum[2] = g3.add ('checkbox {text: "\u00A0Numeric"}');
		columns[0].characters = columns[1].characters = columns[2].characters = 2;

	var headers = main.add ('panel {text: "Headers/footers", alignChildren: "right"}');
		headers.margins.right = 65;
		var h1 = headers.add ("group");
			h1.add ('statictext {text: "Number of header rows:"}');
			var head = h1.add ('edittext {text: "0"}');
		var h2 = headers.add ("group");
			h2.add ('statictext {text: "Number of footer rows:"}');
			var foot = h2.add ('edittext {text: "0"}');
		head.characters = foot.characters = 2;

	var direction = main.add ('panel {text: "Direction", orientation: "row"}');
		var ascend = direction.add ("radiobutton", undefined, "\u00A0Ascending");
		var descend = direction.add ("radiobutton", undefined, "\u00A0Descending");
		ascend.value = true;

	var misc = main.add ('panel {alignChildren: "left"}');
		var ignore_case = misc.add ('checkbox {text: "\u00A0Ignore case"}');
		var formatted = misc.add ('checkbox {text: "\u00A0Formatted text"}');
		ignore_case.value = true;
		formatted.value = false;
	
	var buttons = w.add ('group {orientation: "column"}');
		buttons.add ("button", undefined, "Ok", {name: "ok"});
		buttons.add ("button", undefined, "Cancel", {name: "cancel"});
	
	// Check if some rows are selected. If so, calculate the number of headers and foorter
	// and disable the header/footer section
	var head_foot = headers_footers (table);
	head.text = head_foot [1];
	foot.text = head_foot [2];
	if (head_foot[0] == true)
		headers.enabled = false;
	
	// Check which columns are selected
	var cols = find_columns (table);
	for (var i = 0; i < cols.length; i++)
		columns[i].text = cols[i];
	
	columns[0].active = true;
	
	if (w.show () == 1)
		{
		var f = File (script_dir()+'/table_sort.txt');
		f.open('w'); f.write (w.location); f.close();
		
		var obj = {
			table: table,
			column_order: col_order (columns, table.columns.length),
			anumeric: [anum[0].value, anum[1].value, anum[2].value],  // array of booleans
			ascend: ascend.value,
			headers: Number (head.text),
			footers: Number (foot.text),
			formatted: formatted.value,
			ignore_case: ignore_case.value
			}
		w.close ();
		return obj
		}
	else
		{
		w.close ();
		exit ()
		}
	} // get_date


function col_order (array, le)
	{
	var num, temp = [];
	for (var i = 0; i < array.length; i++)
		{
		num = Number (array[i].text);
		if (isNaN (num) || num > le)
			errorM ("Illegal column number for column " + String (i+1));
		if (num > 0)
			temp.push (num-1);
		}
	return temp;
	}


function headers_footers (table)
	{
	if (app.selection[0] instanceof Cell && app.selection[0].rows.length > 1)
		{
		var c = app.selection[0].cells;
		var header = c[0].parentRow.index;
		var footer = c[-1].parentColumn.cells.length - c[-1].parentRow.index - 1;
		// Add a boolean that dims the header/footer panel
		return [true, header, footer];
		}
	else
		return [false, table.headerRowCount, table.footerRowCount];
	}



function find_columns ()
	{ // Get the number of the selected column so we can plug that into the dialog
	// insertion point selected
	if (app.selection[0].parent instanceof Cell)
		return [String (app.selection[0].parent.parentColumn.index + 1)];
	// several columns selected
	if (app.selection[0] instanceof Cell && app.selection[0].columns.length > 1)
		{
		var temp = [];
		for (var i = 0; i < app.selection[0].columns.length; i++)
			if (i < 3) temp.push (String (app.selection[0].columns[i].index + 1));
		return temp
		}
	// More than one cell selected in the same column
	if (app.selection[0] instanceof Cell && app.selection[0].rows.length > 0)
		return [String (app.selection[0].parentColumn.index + 1)];
	// If nothing else works, this will
	return ["1"]
	}


// General ==============================================================


function get_table ()
	{
	try
		{
		if (app.selection[0].parent instanceof Cell)
			return app.selection[0].parent.parent;
		if (app.selection[0].parent instanceof Table)
			return app.selection[0].parent;
		}
	catch(_)
		{
		errorM ("Select an insertion point in a table or some rows.");
		}
	}


function errorM (m)
	{
	alert (m, "Table sort", true);
	exit ();
	}


function get_time () {return new Date ().getTime ()}
function time_diff (start) {return (new Date () - start) / 1000}



function script_dir()
	{
	try {return File (app.activeScript).path}
	catch(e) {return File (e.fileName).path}
	}


function get_previous ()
	{
	var temp = null, f = File (script_dir()+'/table_sort.txt');
	if (f.exists)
		{
		f.open('r'); temp = f.read(); f.close();
		}
	return temp;
	}