qse/qse/lib/awk/tree.c

/*
 * $Id: tree.c 466 2008-12-09 09:50:40Z baconevi $
 *
   Copyright 2006-2008 Chung, Hyung-Hwan.

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

       http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License.
 */

#include "awk.h"

static const qse_char_t* assop_str[] =
{
	QSE_T("="),
	QSE_T("+="),
	QSE_T("-="),
	QSE_T("*="),
	QSE_T("/="),
	QSE_T("//="),
	QSE_T("%="),
	QSE_T("**="),
	QSE_T(">>="),
	QSE_T("<<="),
	QSE_T("&="),
	QSE_T("^="),
	QSE_T("|=")
};

static const qse_char_t* binop_str[][2] =
{
	{ QSE_T("||"), QSE_T("||") },
	{ QSE_T("&&"), QSE_T("&&") },
	{ QSE_T("in"), QSE_T("in") },

	{ QSE_T("|"),  QSE_T("|") },
	{ QSE_T("^"),  QSE_T("^") },
	{ QSE_T("&"),  QSE_T("&") },

	{ QSE_T("=="), QSE_T("==") },
	{ QSE_T("!="), QSE_T("!=") },
	{ QSE_T(">"),  QSE_T(">") },
	{ QSE_T(">="), QSE_T(">=") },
	{ QSE_T("<"),  QSE_T("<") },
	{ QSE_T("<="), QSE_T("<=") },

	{ QSE_T("<<"), QSE_T("<<") },
	{ QSE_T(">>"), QSE_T(">>") },

	{ QSE_T("+"),  QSE_T("+") },
	{ QSE_T("-"),  QSE_T("-") },
	{ QSE_T("*"),  QSE_T("*") },
	{ QSE_T("/"),  QSE_T("/") },
	{ QSE_T("//"), QSE_T("//") },
	{ QSE_T("%"),  QSE_T("%") },
	{ QSE_T("**"), QSE_T("**") },

	{ QSE_T(" "),  QSE_T(".") },
	{ QSE_T("~"),  QSE_T("~") },
	{ QSE_T("!~"), QSE_T("!~") }
};

static const qse_char_t* unrop_str[] =
{
	QSE_T("+"),
	QSE_T("-"),
	QSE_T("!"),
	QSE_T("~")
};

static const qse_char_t* incop_str[] =
{
	QSE_T("++"),
	QSE_T("--"),
	QSE_T("++"),
	QSE_T("--")
};

static const qse_char_t* getline_inop_str[] =
{
	QSE_T("|"),
	QSE_T("||"),
	QSE_T("<"),
	QSE_T("")
};

static const qse_char_t* print_outop_str[] =
{
	QSE_T("|"),
	QSE_T("||"),
	QSE_T(">"),
	QSE_T(">>"),
	QSE_T("")
};

#define PUT_SRCSTR(awk,str) \
	do { if (qse_awk_putsrcstr (awk, str) == -1) return -1; } while (0)

#define PUT_NEWLINE(awk) \
	do { \
		if (awk->option & QSE_AWK_CRLF) PUT_SRCSTR (awk, QSE_T("\r")); \
		PUT_SRCSTR (awk, QSE_T("\n")); \
	} while (0)

#define PUT_SRCSTRX(awk,str,len) \
	do { if (qse_awk_putsrcstrx (awk, str, len) == -1) return -1; } while (0)

#define PRINT_TABS(awk,depth) \
	do { if (print_tabs(awk,depth) == -1) return -1; } while (0)

#define PRINT_EXPRESSION(awk,nde) \
	do { if (print_expression(awk,nde) == -1) return -1; } while (0)

#define PRINT_EXPRESSION_LIST(awk,nde) \
	do { if (print_expression_list(awk,nde) == -1) return -1; } while (0)

#define PRINT_STATEMENTS(awk,nde,depth) \
	do { if (print_statements(awk,nde,depth) == -1) return -1; } while (0)

static int print_tabs (qse_awk_t* awk, int depth);
static int print_expression (qse_awk_t* awk, qse_awk_nde_t* nde);
static int print_expression_list (qse_awk_t* awk, qse_awk_nde_t* tree);
static int print_statements (qse_awk_t* awk, qse_awk_nde_t* tree, int depth);

static int print_tabs (qse_awk_t* awk, int depth)
{
	while (depth > 0) 
	{
		PUT_SRCSTR (awk, QSE_T("\t"));
		depth--;
	}

	return 0;
}

static int print_expression (qse_awk_t* awk, qse_awk_nde_t* nde)
{
	qse_cstr_t kw;
	
	switch (nde->type) 
	{
		case QSE_AWK_NDE_GRP:
		{	
			qse_awk_nde_t* p = ((qse_awk_nde_grp_t*)nde)->body;

			PUT_SRCSTR (awk, QSE_T("("));
			while (p != QSE_NULL) 
			{
				PRINT_EXPRESSION (awk, p);
				if (p->next != QSE_NULL) 
					PUT_SRCSTR (awk, QSE_T(","));
				p = p->next;
			}
			PUT_SRCSTR (awk, QSE_T(")"));
			break;
		}

		case QSE_AWK_NDE_ASS:
		{
			qse_awk_nde_ass_t* px = (qse_awk_nde_ass_t*)nde;

			PRINT_EXPRESSION (awk, px->left);
			PUT_SRCSTR (awk, QSE_T(" "));
			PUT_SRCSTR (awk, assop_str[px->opcode]);
			PUT_SRCSTR (awk, QSE_T(" "));
			PRINT_EXPRESSION (awk, px->right);

			QSE_ASSERT (px->right->next == QSE_NULL);
			break;
		}

		case QSE_AWK_NDE_EXP_BIN:
		{
			qse_awk_nde_exp_t* px = (qse_awk_nde_exp_t*)nde;

			PUT_SRCSTR (awk, QSE_T("("));
			PRINT_EXPRESSION (awk, px->left);
			QSE_ASSERT (px->left->next == QSE_NULL);

			PUT_SRCSTR (awk, QSE_T(" "));
			PUT_SRCSTR (awk, binop_str[px->opcode][(awk->option & QSE_AWK_IMPLICIT)? 0: 1]);
			PUT_SRCSTR (awk, QSE_T(" "));

			if (px->right->type == QSE_AWK_NDE_ASS) 
				PUT_SRCSTR (awk, QSE_T("("));
			PRINT_EXPRESSION (awk, px->right);
			if (px->right->type == QSE_AWK_NDE_ASS) 
				PUT_SRCSTR (awk, QSE_T(")"));
			QSE_ASSERT (px->right->next == QSE_NULL); 
			PUT_SRCSTR (awk, QSE_T(")"));
			break;
		}

		case QSE_AWK_NDE_EXP_UNR:
		{
			qse_awk_nde_exp_t* px = (qse_awk_nde_exp_t*)nde;
			QSE_ASSERT (px->right == QSE_NULL);

			PUT_SRCSTR (awk, QSE_T("("));
			PUT_SRCSTR (awk, unrop_str[px->opcode]);
			PUT_SRCSTR (awk, QSE_T("("));
			PRINT_EXPRESSION (awk, px->left);
			PUT_SRCSTR (awk, QSE_T(")"));
			PUT_SRCSTR (awk, QSE_T(")"));
			break;
		}

		case QSE_AWK_NDE_EXP_INCPRE:
		{
			qse_awk_nde_exp_t* px = (qse_awk_nde_exp_t*)nde;
			QSE_ASSERT (px->right == QSE_NULL);

			PUT_SRCSTR (awk, incop_str[px->opcode]);
			PUT_SRCSTR (awk, QSE_T("("));
			PRINT_EXPRESSION (awk, px->left);
			PUT_SRCSTR (awk, QSE_T(")"));
			break;
		}

		case QSE_AWK_NDE_EXP_INCPST:
		{
			qse_awk_nde_exp_t* px = (qse_awk_nde_exp_t*)nde;
			QSE_ASSERT (px->right == QSE_NULL);

			PUT_SRCSTR (awk, QSE_T("("));
			PRINT_EXPRESSION (awk, px->left);
			PUT_SRCSTR (awk, QSE_T(")"));
			PUT_SRCSTR (awk, incop_str[px->opcode]);
			break;
		}

		case QSE_AWK_NDE_CND:
		{
			qse_awk_nde_cnd_t* px = (qse_awk_nde_cnd_t*)nde;

			PUT_SRCSTR (awk, QSE_T("("));
			PRINT_EXPRESSION (awk, px->test);
			PUT_SRCSTR (awk, QSE_T(")?"));

			PRINT_EXPRESSION (awk, px->left);
			PUT_SRCSTR (awk, QSE_T(":"));
			PRINT_EXPRESSION (awk, px->right);
			break;
		}

		case QSE_AWK_NDE_INT:
		{
			if (((qse_awk_nde_int_t*)nde)->str == QSE_NULL)
			{
				qse_size_t n;

				n = qse_awk_longtostr (
					((qse_awk_nde_int_t*)nde)->val, 10, QSE_NULL, 
					awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt));

				PUT_SRCSTRX (awk, awk->tmp.fmt, n);
			}
			else
			{
				PUT_SRCSTRX (awk,
					((qse_awk_nde_int_t*)nde)->str,
					((qse_awk_nde_int_t*)nde)->len);
			}
			break;
		}

		case QSE_AWK_NDE_REAL:
		{
			if (((qse_awk_nde_real_t*)nde)->str == QSE_NULL)
			{
			#if (QSE_SIZEOF_LONG_DOUBLE != 0) && !defined(__MINGW32__)
				awk->prmfns->sprintf (
					awk->prmfns->data,
					awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt), QSE_T("%Lf"), 
					(long double)((qse_awk_nde_real_t*)nde)->val);
			#elif (QSE_SIZEOF_DOUBLE != 0)
				awk->prmfns->sprintf (
					awk->prmfns->data,
					awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt), QSE_T("%f"), 
					(double)((qse_awk_nde_real_t*)nde)->val);
			#else
				#error unsupported floating-point data type
			#endif
				PUT_SRCSTR (awk, awk->tmp.fmt);
			}
			else
			{
				PUT_SRCSTRX (awk,
					((qse_awk_nde_real_t*)nde)->str,
					((qse_awk_nde_real_t*)nde)->len);
			}
			break;
		}

		case QSE_AWK_NDE_STR:
		{
			qse_char_t* ptr;
			qse_size_t len, i;

			PUT_SRCSTR (awk, QSE_T("\""));

			ptr = ((qse_awk_nde_str_t*)nde)->ptr;
			len = ((qse_awk_nde_str_t*)nde)->len;
			for (i = 0; i < len; i++)
			{
				/* TODO: maybe more de-escaping?? */
				if (ptr[i] == QSE_T('\n'))
					PUT_SRCSTR (awk, QSE_T("\\n"));
				else if (ptr[i] == QSE_T('\r'))
					PUT_SRCSTR (awk, QSE_T("\\r"));
				else if (ptr[i] == QSE_T('\f'))
					PUT_SRCSTR (awk, QSE_T("\\f"));
				else if (ptr[i] == QSE_T('\b'))
					PUT_SRCSTR (awk, QSE_T("\\b"));
				else if (ptr[i] == QSE_T('\v'))
					PUT_SRCSTR (awk, QSE_T("\\v"));
				else if (ptr[i] == QSE_T('\a'))
					PUT_SRCSTR (awk, QSE_T("\\a"));
				else if (ptr[i] == QSE_T('\0'))
					PUT_SRCSTR (awk, QSE_T("\\0"));
				else
					PUT_SRCSTRX (awk, &ptr[i], 1);
			}
			PUT_SRCSTR (awk, QSE_T("\""));
			break;
		}

		case QSE_AWK_NDE_REX:
		{
			PUT_SRCSTR (awk, QSE_T("/"));
			PUT_SRCSTRX (awk,
				((qse_awk_nde_rex_t*)nde)->ptr, 
				((qse_awk_nde_rex_t*)nde)->len);
			PUT_SRCSTR (awk, QSE_T("/"));
			break;
		}

		case QSE_AWK_NDE_ARG:
		{
			qse_char_t tmp[QSE_SIZEOF(qse_long_t)*8+2]; 
			qse_size_t n;
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;
			QSE_ASSERT (px->id.idxa != (qse_size_t)-1);

			n = qse_awk_longtostr (
				px->id.idxa, 10, QSE_NULL, tmp, QSE_COUNTOF(tmp));

			PUT_SRCSTR (awk, QSE_T("__p"));
			PUT_SRCSTRX (awk, tmp, n);

			QSE_ASSERT (px->idx == QSE_NULL);
			break;
		}

		case QSE_AWK_NDE_ARGIDX:
		{
			qse_size_t n;
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;
			QSE_ASSERT (px->id.idxa != (qse_size_t)-1);
			QSE_ASSERT (px->idx != QSE_NULL);

			PUT_SRCSTR (awk, QSE_T("__p"));
			n = qse_awk_longtostr (
				px->id.idxa, 10, QSE_NULL,
				awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt));
			PUT_SRCSTRX (awk, awk->tmp.fmt, n);
			PUT_SRCSTR (awk, QSE_T("["));
			PRINT_EXPRESSION_LIST (awk, px->idx);
			PUT_SRCSTR (awk, QSE_T("]"));
			break;
		}

		case QSE_AWK_NDE_NAMED:
		{
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;
			QSE_ASSERT (px->id.idxa == (qse_size_t)-1);
			QSE_ASSERT (px->idx == QSE_NULL);

			PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
			break;
		}

		case QSE_AWK_NDE_NAMEDIDX:
		{
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;
			QSE_ASSERT (px->id.idxa == (qse_size_t)-1);
			QSE_ASSERT (px->idx != QSE_NULL);

			PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
			PUT_SRCSTR (awk, QSE_T("["));
			PRINT_EXPRESSION_LIST (awk, px->idx);
			PUT_SRCSTR (awk, QSE_T("]"));
			break;
		}

		case QSE_AWK_NDE_GBL:
		{
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;

			if (px->id.idxa != (qse_size_t)-1) 
			{

				if ((awk->option & QSE_AWK_EXPLICIT) && 
				    !(awk->option & QSE_AWK_IMPLICIT))
				{
					/* no implicit(named) variable is allowed.
					 * use the actual name */
					PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
				}
				else if (px->id.idxa < awk->tree.nbgbls)
				{
					/* static global variables */
					PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
				}
				else
				{
					qse_char_t tmp[QSE_SIZEOF(qse_long_t)*8+2]; 
					qse_size_t n;

					PUT_SRCSTR (awk, QSE_T("__g"));
					n = qse_awk_longtostr (
						px->id.idxa, 10, 
						QSE_NULL, tmp, QSE_COUNTOF(tmp));
					PUT_SRCSTRX (awk, tmp, n);
				}
			}
			else 
			{
				PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
			}
			QSE_ASSERT (px->idx == QSE_NULL);
			break;
		}

		case QSE_AWK_NDE_GBLIDX:
		{
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;

			if (px->id.idxa != (qse_size_t)-1) 
			{
				if ((awk->option & QSE_AWK_EXPLICIT) && 
				    !(awk->option & QSE_AWK_IMPLICIT))
				{
					/* no implicit(named) variable is allowed.
					 * use the actual name */
					PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
				}
				else if (px->id.idxa < awk->tree.nbgbls)
				{
					/* static global variables */
					PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
				}
				else
				{
					qse_char_t tmp[QSE_SIZEOF(qse_long_t)*8+2]; 
					qse_size_t n;

					PUT_SRCSTR (awk, QSE_T("__g"));
					n = qse_awk_longtostr (
						px->id.idxa, 10, 
						QSE_NULL, tmp, QSE_COUNTOF(tmp));
					PUT_SRCSTRX (awk, tmp, n);
				}
				PUT_SRCSTR (awk, QSE_T("["));
			}
			else 
			{
				PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
				PUT_SRCSTR (awk, QSE_T("["));
			}
			QSE_ASSERT (px->idx != QSE_NULL);
			PRINT_EXPRESSION_LIST (awk, px->idx);
			PUT_SRCSTR (awk, QSE_T("]"));
			break;
		}

		case QSE_AWK_NDE_LCL:
		{
			qse_size_t n;
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;

			if (px->id.idxa != (qse_size_t)-1) 
			{
				PUT_SRCSTR (awk, QSE_T("__l"));
				n = qse_awk_longtostr (
					px->id.idxa, 10, QSE_NULL, 
					awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt));
				PUT_SRCSTRX (awk, awk->tmp.fmt, n);
			}
			else 
			{
				PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
			}
			QSE_ASSERT (px->idx == QSE_NULL);
			break;
		}

		case QSE_AWK_NDE_LCLIDX:
		{
			qse_size_t n;
			qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)nde;

			if (px->id.idxa != (qse_size_t)-1) 
			{
				PUT_SRCSTR (awk, QSE_T("__l"));
				n = qse_awk_longtostr (
					px->id.idxa, 10, QSE_NULL, 
					awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt));
				PUT_SRCSTRX (awk, awk->tmp.fmt, n);
				PUT_SRCSTR (awk, QSE_T("["));
			}
			else 
			{
				PUT_SRCSTRX (awk, px->id.name.ptr, px->id.name.len);
				PUT_SRCSTR (awk, QSE_T("["));
			}
			QSE_ASSERT (px->idx != QSE_NULL);
			PRINT_EXPRESSION_LIST (awk, px->idx);
			PUT_SRCSTR (awk, QSE_T("]"));
			break;
		}

		case QSE_AWK_NDE_POS:
		{
			PUT_SRCSTR (awk, QSE_T("$"));
			PRINT_EXPRESSION (awk, ((qse_awk_nde_pos_t*)nde)->val);
			break;
		}

		case QSE_AWK_NDE_FNC:
		{
			qse_awk_nde_call_t* px = (qse_awk_nde_call_t*)nde;
			PUT_SRCSTRX (awk, 
				px->what.fnc.name.ptr, px->what.fnc.name.len);
			PUT_SRCSTR (awk, QSE_T(" ("));
			PRINT_EXPRESSION_LIST (awk, px->args);
			PUT_SRCSTR (awk, QSE_T(")"));
			break;
		}

		case QSE_AWK_NDE_FUN:
		{
			qse_awk_nde_call_t* px = (qse_awk_nde_call_t*)nde;
			PUT_SRCSTRX (awk, 
				px->what.fun.name.ptr, px->what.fun.name.len);
			PUT_SRCSTR (awk, QSE_T(" ("));
			PRINT_EXPRESSION_LIST (awk, px->args);
			PUT_SRCSTR (awk, QSE_T(")"));
			break;
		}

		case QSE_AWK_NDE_GETLINE:
		{
			qse_awk_nde_getline_t* px = (qse_awk_nde_getline_t*)nde;
			if (px->in != QSE_NULL &&
			    (px->in_type == QSE_AWK_IN_PIPE ||
			     px->in_type == QSE_AWK_IN_RWPIPE))
			{
				PRINT_EXPRESSION (awk, px->in);
				PUT_SRCSTR (awk, QSE_T(" "));
				PUT_SRCSTR (awk, getline_inop_str[px->in_type]);
				PUT_SRCSTR (awk, QSE_T(" "));
			}

			qse_awk_getkw (awk, KW_GETLINE, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			if (px->var != QSE_NULL)
			{
				PUT_SRCSTR (awk, QSE_T(" "));
				PRINT_EXPRESSION (awk, px->var);
			}

			if (px->in != QSE_NULL &&
			    px->in_type == QSE_AWK_IN_FILE)
			{
				PUT_SRCSTR (awk, QSE_T(" "));
				PUT_SRCSTR (awk, getline_inop_str[px->in_type]);
				PUT_SRCSTR (awk, QSE_T(" "));
				PRINT_EXPRESSION (awk, px->in);
			}	  
			break;
		}

		default:
		{
			return -1;
		}
	}

	return 0;
}

static int print_expression_list (qse_awk_t* awk, qse_awk_nde_t* tree)
{
	qse_awk_nde_t* p = tree;

	while (p != QSE_NULL) 
	{
		PRINT_EXPRESSION (awk, p);
		p = p->next;
		if (p != QSE_NULL) PUT_SRCSTR (awk, QSE_T(","));
	}

	return 0;
}

static int print_statement (qse_awk_t* awk, qse_awk_nde_t* p, int depth)
{
	qse_size_t i;
	qse_cstr_t kw;

	switch (p->type) 
	{
		case QSE_AWK_NDE_NULL:
		{
			PRINT_TABS (awk, depth);
			PUT_SRCSTR (awk, QSE_T(";"));
			PUT_NEWLINE (awk);
			break;
		}

		case QSE_AWK_NDE_BLK:
		{
			qse_size_t n;
			qse_awk_nde_blk_t* px = (qse_awk_nde_blk_t*)p;

			PRINT_TABS (awk, depth);
			PUT_SRCSTR (awk, QSE_T("{"));
			PUT_NEWLINE (awk);

			if (px->nlcls > 0) 
			{
				PRINT_TABS (awk, depth + 1);
				qse_awk_getkw (awk, KW_LOCAL, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
				PUT_SRCSTR (awk, QSE_T(" "));

				for (i = 0; i < px->nlcls - 1; i++) 
				{
					PUT_SRCSTR (awk, QSE_T("__l"));
					n = qse_awk_longtostr (
						i, 10, QSE_NULL, 
						awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt));
					PUT_SRCSTRX (awk, awk->tmp.fmt, n);
					PUT_SRCSTR (awk, QSE_T(", "));
				}

				PUT_SRCSTR (awk, QSE_T("__l"));
				n = qse_awk_longtostr (
					i, 10, QSE_NULL, 
					awk->tmp.fmt, QSE_COUNTOF(awk->tmp.fmt));
				PUT_SRCSTRX (awk, awk->tmp.fmt, n);
				PUT_SRCSTR (awk, QSE_T(";"));
				PUT_NEWLINE (awk);
			}

			PRINT_STATEMENTS (awk, px->body, depth + 1);	
			PRINT_TABS (awk, depth);
			PUT_SRCSTR (awk, QSE_T("}"));
			PUT_NEWLINE (awk);
			break;
		}

		case QSE_AWK_NDE_IF: 
		{
			qse_awk_nde_if_t* px = (qse_awk_nde_if_t*)p;

			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_IF, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(" ("));	
			PRINT_EXPRESSION (awk, px->test);
			PUT_SRCSTR (awk, QSE_T(")"));
			PUT_NEWLINE (awk);

			QSE_ASSERT (px->then_part != QSE_NULL);
			if (px->then_part->type == QSE_AWK_NDE_BLK)
				PRINT_STATEMENTS (awk, px->then_part, depth);
			else
				PRINT_STATEMENTS (awk, px->then_part, depth + 1);

			if (px->else_part != QSE_NULL) 
			{
				PRINT_TABS (awk, depth);
				qse_awk_getkw (awk, KW_ELSE, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
				PUT_NEWLINE (awk);
				if (px->else_part->type == QSE_AWK_NDE_BLK)
					PRINT_STATEMENTS (awk, px->else_part, depth);
				else
					PRINT_STATEMENTS (awk, px->else_part, depth + 1);
			}
			break;
		}

		case QSE_AWK_NDE_WHILE: 
		{
			qse_awk_nde_while_t* px = (qse_awk_nde_while_t*)p;

			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_WHILE, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(" ("));	
			PRINT_EXPRESSION (awk, px->test);
			PUT_SRCSTR (awk, QSE_T(")"));
			PUT_NEWLINE (awk);
			if (px->body->type == QSE_AWK_NDE_BLK) 
			{
				PRINT_STATEMENTS (awk, px->body, depth);
			}
			else 
			{
				PRINT_STATEMENTS (awk, px->body, depth + 1);
			}
			break;
		}

		case QSE_AWK_NDE_DOWHILE: 
		{
			qse_awk_nde_while_t* px = (qse_awk_nde_while_t*)p;

			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_DO, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_NEWLINE (awk);
			if (px->body->type == QSE_AWK_NDE_BLK) 
			{
				PRINT_STATEMENTS (awk, px->body, depth);
			}
			else 
			{
				PRINT_STATEMENTS (awk, px->body, depth + 1);
			}

			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_WHILE, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(" ("));	
			PRINT_EXPRESSION (awk, px->test);
			PUT_SRCSTR (awk, QSE_T(");"));
			PUT_NEWLINE (awk);
			break;
		}

		case QSE_AWK_NDE_FOR:
		{
			qse_awk_nde_for_t* px = (qse_awk_nde_for_t*)p;

			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_FOR, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(" ("));
			if (px->init != QSE_NULL) 
			{
				PRINT_EXPRESSION (awk, px->init);
			}
			PUT_SRCSTR (awk, QSE_T("; "));
			if (px->test != QSE_NULL) 
			{
				PRINT_EXPRESSION (awk, px->test);
			}
			PUT_SRCSTR (awk, QSE_T("; "));
			if (px->incr != QSE_NULL) 
			{
				PRINT_EXPRESSION (awk, px->incr);
			}
			PUT_SRCSTR (awk, QSE_T(")"));
			PUT_NEWLINE (awk);

			if (px->body->type == QSE_AWK_NDE_BLK) 
			{
				PRINT_STATEMENTS (awk, px->body, depth);
			}
			else 
			{
				PRINT_STATEMENTS (awk, px->body, depth + 1);
			}
			break;
		}

		case QSE_AWK_NDE_FOREACH:
		{
			qse_awk_nde_foreach_t* px = (qse_awk_nde_foreach_t*)p;

			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_FOR, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(" "));
			PRINT_EXPRESSION (awk, px->test);
			PUT_NEWLINE (awk);
			if (px->body->type == QSE_AWK_NDE_BLK) 
			{
				PRINT_STATEMENTS (awk, px->body, depth);
			}
			else 
			{
				PRINT_STATEMENTS (awk, px->body, depth + 1);
			}
			break;
		}

		case QSE_AWK_NDE_BREAK:
		{
			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_BREAK, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(";"));
			PUT_NEWLINE (awk);
			break;
		}

		case QSE_AWK_NDE_CONTINUE:
		{
			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_CONTINUE, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(";"));
			PUT_NEWLINE (awk);
			break;
		}

		case QSE_AWK_NDE_RETURN:
		{
			PRINT_TABS (awk, depth);
			if (((qse_awk_nde_return_t*)p)->val == QSE_NULL) 
			{
				qse_awk_getkw (awk, KW_RETURN, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
				PUT_SRCSTR (awk, QSE_T(";"));
				PUT_NEWLINE (awk);
			}
			else 
			{
				qse_awk_getkw (awk, KW_RETURN, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
				PUT_SRCSTR (awk, QSE_T(" "));
				QSE_ASSERT (((qse_awk_nde_return_t*)p)->val->next == QSE_NULL);

				PRINT_EXPRESSION (awk, ((qse_awk_nde_return_t*)p)->val);
				PUT_SRCSTR (awk, QSE_T(";"));
				PUT_NEWLINE (awk);
			}
			break;
		}

		case QSE_AWK_NDE_EXIT:
		{
			qse_awk_nde_exit_t* px = (qse_awk_nde_exit_t*)p;
			PRINT_TABS (awk, depth);

			if (px->val == QSE_NULL) 
			{
				qse_awk_getkw (awk, KW_EXIT, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
				PUT_SRCSTR (awk, QSE_T(";"));
				PUT_NEWLINE (awk);
			}
			else 
			{
				qse_awk_getkw (awk, KW_EXIT, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
				PUT_SRCSTR (awk, QSE_T(" "));
				QSE_ASSERT (px->val->next == QSE_NULL);
				PRINT_EXPRESSION (awk, px->val);
				PUT_SRCSTR (awk, QSE_T(";"));
				PUT_NEWLINE (awk);
			}
			break;
		}

		case QSE_AWK_NDE_NEXT:
		{
			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_NEXT, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(";"));
			PUT_NEWLINE (awk);
			break;
		}

		case QSE_AWK_NDE_NEXTFILE:
		{
			PRINT_TABS (awk, depth);
			if (((qse_awk_nde_nextfile_t*)p)->out)
			{
				qse_awk_getkw (awk, KW_NEXTOFILE, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
			}
			else
			{
				qse_awk_getkw (awk, KW_NEXTFILE, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
			}
			PUT_SRCSTR (awk, QSE_T(";"));
			PUT_NEWLINE (awk);
			break;
		}

		case QSE_AWK_NDE_DELETE:
		{
			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_DELETE, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(" "));
			qse_awk_prnpt (awk, ((qse_awk_nde_delete_t*)p)->var);
			break;
		}

		case QSE_AWK_NDE_RESET:
		{
			PRINT_TABS (awk, depth);
			qse_awk_getkw (awk, KW_RESET, &kw);
			PUT_SRCSTRX (awk, kw.ptr, kw.len);
			PUT_SRCSTR (awk, QSE_T(" "));
			qse_awk_prnpt (awk, ((qse_awk_nde_reset_t*)p)->var);
			break;
		}

		case QSE_AWK_NDE_PRINT:
		case QSE_AWK_NDE_PRINTF:
		{
			qse_awk_nde_print_t* px = (qse_awk_nde_print_t*)p;

			PRINT_TABS (awk, depth);

			if (p->type == QSE_AWK_NDE_PRINT) 
			{
				qse_awk_getkw (awk, KW_PRINT, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
			}
			else
			{
				qse_awk_getkw (awk, KW_PRINTF, &kw);
				PUT_SRCSTRX (awk, kw.ptr, kw.len);
			}

			if (px->args != QSE_NULL)
			{
				PUT_SRCSTR (awk, QSE_T(" "));
				PRINT_EXPRESSION_LIST (awk, px->args);
			}

			if (px->out != QSE_NULL)
			{
				PUT_SRCSTR (awk, QSE_T(" "));
				PUT_SRCSTR (awk, print_outop_str[px->out_type]);
				PUT_SRCSTR (awk, QSE_T(" "));
				PRINT_EXPRESSION (awk, px->out);
			}

			PUT_SRCSTR (awk, QSE_T(";"));
			PUT_NEWLINE (awk);
			break;
		}

		default:
		{
			PRINT_TABS (awk, depth);
			PRINT_EXPRESSION (awk, p);
			PUT_SRCSTR (awk, QSE_T(";"));
			PUT_NEWLINE (awk);
		}
	}

	return 0;
}

static int print_statements (qse_awk_t* awk, qse_awk_nde_t* tree, int depth)
{
	qse_awk_nde_t* p = tree;

	while (p != QSE_NULL) 
	{
		if (print_statement (awk, p, depth) == -1) return -1;
		p = p->next;
	}

	return 0;
}

int qse_awk_prnpt (qse_awk_t* awk, qse_awk_nde_t* tree)
{
	return print_statements (awk, tree, 0);
}

int qse_awk_prnnde (qse_awk_t* awk, qse_awk_nde_t* tree)
{
	return print_statement (awk, tree, 0);
}

int qse_awk_prnptnpt (qse_awk_t* awk, qse_awk_nde_t* tree)
{
	qse_awk_nde_t* nde = tree;

	while (nde != QSE_NULL)
	{
		if (print_expression (awk, nde) == -1) return -1;
		if (nde->next == QSE_NULL) break;

		PUT_SRCSTR (awk, QSE_T(","));
		nde = nde->next;
	}

	return 0;
}

void qse_awk_clrpt (qse_awk_t* awk, qse_awk_nde_t* tree)
{
	qse_awk_nde_t* p = tree;
	qse_awk_nde_t* next;

	while (p != QSE_NULL) 
	{
		next = p->next;

		switch (p->type) 
		{
			case QSE_AWK_NDE_NULL:
			{
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_BLK:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_blk_t*)p)->body);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_IF:
			{
				qse_awk_nde_if_t* px = (qse_awk_nde_if_t*)p;
				qse_awk_clrpt (awk, px->test);
				qse_awk_clrpt (awk, px->then_part);

				if (px->else_part != QSE_NULL)
					qse_awk_clrpt (awk, px->else_part);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_WHILE:
			case QSE_AWK_NDE_DOWHILE:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_while_t*)p)->test);
				qse_awk_clrpt (awk, ((qse_awk_nde_while_t*)p)->body);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_FOR:
			{
				qse_awk_nde_for_t* px = (qse_awk_nde_for_t*)p;

				if (px->init != QSE_NULL)
					qse_awk_clrpt (awk, px->init);
				if (px->test != QSE_NULL)
					qse_awk_clrpt (awk, px->test);
				if (px->incr != QSE_NULL)
					qse_awk_clrpt (awk, px->incr);
				qse_awk_clrpt (awk, px->body);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_FOREACH:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_foreach_t*)p)->test);
				qse_awk_clrpt (awk, ((qse_awk_nde_foreach_t*)p)->body);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_BREAK:
			{
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_CONTINUE:
			{
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_RETURN:
			{
				qse_awk_nde_return_t* px =
					(qse_awk_nde_return_t*)p;
				if (px->val != QSE_NULL) 
					qse_awk_clrpt (awk, px->val);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_EXIT:
			{
				if (((qse_awk_nde_exit_t*)p)->val != QSE_NULL) 
					qse_awk_clrpt (awk, ((qse_awk_nde_exit_t*)p)->val);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_NEXT:
			case QSE_AWK_NDE_NEXTFILE:
			{
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_DELETE:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_delete_t*)p)->var);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_RESET:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_reset_t*)p)->var);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_PRINT:
			case QSE_AWK_NDE_PRINTF:
			{
				qse_awk_nde_print_t* px = 
					(qse_awk_nde_print_t*)p;
				if (px->args != QSE_NULL) 
					qse_awk_clrpt (awk, px->args);
				if (px->out != QSE_NULL) 
					qse_awk_clrpt (awk, px->out);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_GRP:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_grp_t*)p)->body);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_ASS:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_ass_t*)p)->left);
				qse_awk_clrpt (awk, ((qse_awk_nde_ass_t*)p)->right);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_EXP_BIN:
			{
				qse_awk_nde_exp_t* px = (qse_awk_nde_exp_t*)p;
				QSE_ASSERT (px->left->next == QSE_NULL);
				QSE_ASSERT (px->right->next == QSE_NULL);

				qse_awk_clrpt (awk, px->left);
				qse_awk_clrpt (awk, px->right);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_EXP_UNR:
			case QSE_AWK_NDE_EXP_INCPRE:
			case QSE_AWK_NDE_EXP_INCPST:
			{
				qse_awk_nde_exp_t* px = (qse_awk_nde_exp_t*)p;
				QSE_ASSERT (px->right == QSE_NULL);
				qse_awk_clrpt (awk, px->left);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_CND:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_cnd_t*)p)->test);
				qse_awk_clrpt (awk, ((qse_awk_nde_cnd_t*)p)->left);
				qse_awk_clrpt (awk, ((qse_awk_nde_cnd_t*)p)->right);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_INT:
			{
				if (((qse_awk_nde_int_t*)p)->str != QSE_NULL)
					QSE_AWK_FREE (awk, ((qse_awk_nde_int_t*)p)->str);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_REAL:
			{
				if (((qse_awk_nde_real_t*)p)->str != QSE_NULL)
					QSE_AWK_FREE (awk, ((qse_awk_nde_real_t*)p)->str);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_STR:
			{
				QSE_AWK_FREE (awk, ((qse_awk_nde_str_t*)p)->ptr);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_REX:
			{
				QSE_AWK_FREE (awk, ((qse_awk_nde_rex_t*)p)->ptr);
				QSE_AWK_FREE (awk, ((qse_awk_nde_rex_t*)p)->code);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_NAMED:
			case QSE_AWK_NDE_GBL:
			case QSE_AWK_NDE_LCL:
			case QSE_AWK_NDE_ARG:
			{
				qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)p;
				QSE_ASSERT (px->idx == QSE_NULL);
				if (px->id.name.ptr != QSE_NULL)
					QSE_AWK_FREE (awk, px->id.name.ptr);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_NAMEDIDX:
			case QSE_AWK_NDE_GBLIDX:
			case QSE_AWK_NDE_LCLIDX:
			case QSE_AWK_NDE_ARGIDX:
			{
				qse_awk_nde_var_t* px = (qse_awk_nde_var_t*)p;
				QSE_ASSERT (px->idx != QSE_NULL);
				qse_awk_clrpt (awk, px->idx);
				if (px->id.name.ptr != QSE_NULL)
					QSE_AWK_FREE (awk, px->id.name.ptr);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_POS:
			{
				qse_awk_clrpt (awk, ((qse_awk_nde_pos_t*)p)->val);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_FNC:
			{
				qse_awk_nde_call_t* px = (qse_awk_nde_call_t*)p;
				/* QSE_AWK_FREE (awk, px->what.fnc); */
				QSE_AWK_FREE (awk, px->what.fnc.name.ptr);
				qse_awk_clrpt (awk, px->args);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_FUN:
			{
				qse_awk_nde_call_t* px = (qse_awk_nde_call_t*)p;
				QSE_AWK_FREE (awk, px->what.fun.name.ptr);
				qse_awk_clrpt (awk, px->args);
				QSE_AWK_FREE (awk, p);
				break;
			}

			case QSE_AWK_NDE_GETLINE:
			{
				qse_awk_nde_getline_t* px = 
					(qse_awk_nde_getline_t*)p;
				if (px->var != QSE_NULL) 
					qse_awk_clrpt (awk, px->var);
				if (px->in != QSE_NULL) 
					qse_awk_clrpt (awk, px->in);
				QSE_AWK_FREE (awk, p);
				break;
			}

			default:
			{
				QSE_ASSERT (!"should never happen - invalid node type");
				QSE_AWK_FREE (awk, p);
				break;
			}
		}

		p = next;
	}
}