===================================================================
RCS file: /home/cvs/OpenXM_contrib2/asir2000/builtin/strobj.c,v
retrieving revision 1.46
retrieving revision 1.55
diff -u -p -r1.46 -r1.55
--- OpenXM_contrib2/asir2000/builtin/strobj.c	2004/03/25 01:56:00	1.46
+++ OpenXM_contrib2/asir2000/builtin/strobj.c	2005/07/26 01:33:46	1.55
@@ -45,7 +45,7 @@
  * DEVELOPER SHALL HAVE NO LIABILITY IN CONNECTION WITH THE USE,
  * PERFORMANCE OR NON-PERFORMANCE OF THE SOFTWARE.
  *
- * $OpenXM: OpenXM_contrib2/asir2000/builtin/strobj.c,v 1.45 2004/03/25 01:31:03 noro Exp $
+ * $OpenXM: OpenXM_contrib2/asir2000/builtin/strobj.c,v 1.54 2005/07/15 00:23:26 noro Exp $
 */
 #include "ca.h"
 #include "parse.h"
@@ -65,6 +65,7 @@ struct TeXSymbol {
 
 extern char *parse_strp;
 
+void Psprintf();
 void Prtostr(), Pstrtov(), Peval_str();
 void Pstrtoascii(), Pasciitostr();
 void Pstr_len(), Pstr_chr(), Psub_str();
@@ -75,6 +76,9 @@ void Pstring_to_tb();
 void Pquotetotex_tb();
 void Pquotetotex();
 void Pquotetotex_env();
+void Pflatten_quote();
+void Pquote_to_funargs(),Pfunargs_to_quote(),Pget_function_name();
+void Pquote_unify();
 void fnodetotex_tb(FNODE f,TB tb);
 char *symbol_name(char *name);
 char *conv_rule(char *name);
@@ -84,8 +88,10 @@ void tb_to_string(TB tb,STRING *rp);
 void fnodenodetotex_tb(NODE n,TB tb);
 void fargstotex_tb(char *opname,FNODE f,TB tb);
 int top_is_minus(FNODE f);
+NODE quote_unify(Obj f,Obj pat);
 
 struct ftab str_tab[] = {
+	{"sprintf",Psprintf,-99999999},
 	{"rtostr",Prtostr,1},
 	{"strtov",Pstrtov,1},
 	{"eval_str",Peval_str,1},
@@ -101,6 +107,11 @@ struct ftab str_tab[] = {
 	{"quotetotex_tb",Pquotetotex_tb,2},
 	{"quotetotex",Pquotetotex,1},
 	{"quotetotex_env",Pquotetotex_env,-99999999},
+	{"flatten_quote",Pflatten_quote,2},
+	{"quote_to_funargs",Pquote_to_funargs,1},
+	{"quote_unify",Pquote_unify,2},
+	{"funargs_to_quote",Pfunargs_to_quote,1},
+	{"get_function_name",Pget_function_name,1},
 	{0,0,0},
 };
 
@@ -496,6 +507,128 @@ void Pwrite_to_tb(NODE arg,Q *rp)
 	*rp = 0;	
 }
 
+void Pquote_unify(NODE arg,LIST *rp)
+{
+	NODE r;
+
+	r = quote_unify((Obj)ARG0(arg),(Obj)ARG1(arg));
+	MKLIST(*rp,r);
+}
+
+/* 
+/* consistency check and merge */
+
+NODE merge_matching_node(NODE n,NODE a)
+{
+	NODE ta,ba,tn,bn;
+	QUOTE pa,va,pn,vn;
+
+	if ( !n )
+		return a;
+	for ( ta = a; ta; ta = NEXT(ta) ) {
+		ba = BDY((LIST)BDY(ta));
+		if ( !ba ) continue;
+		pa = (QUOTE)BDY(ba); va = (QUOTE)BDY(NEXT(ba));
+		for ( tn = n; tn; tn = NEXT(tn) ) {
+			bn = BDY((LIST)BDY(tn));	
+			if ( !bn ) continue;
+			pn = (QUOTE)BDY(bn); vn = (QUOTE)BDY(NEXT(bn));
+			if ( !compquote(CO,pa,pn) ) {
+				if ( !compquote(CO,va,vn) ) break;
+				else return 0;
+			}
+		}
+		if ( !tn ) {
+			MKNODE(tn,(pointer)BDY(ta),n);
+			n = tn;
+		}
+	}
+	return n;
+}
+
+NODE quote_unify_node(NODE f,NODE pat) {
+	NODE r,a,tf,tp;
+
+	if ( length(f) != length(pat) ) return 0;
+	r = 0;
+	for ( tf = f, tp = pat; tf; tf = NEXT(tf), tp = NEXT(tp) ) {
+		a = quote_unify((Obj)BDY(tf),(Obj)BDY(tp));
+		if ( !a ) return 0;
+		r = merge_matching_node(r,a);
+		if ( !r ) return 0;
+	}
+	return r;
+}
+
+void get_quote_id_arg(QUOTE f,int *id,NODE *r)
+{
+	LIST fa;
+	NODE arg,fab;
+
+	arg = mknode(1,f); Pquote_to_funargs(arg,&fa); fab = BDY((LIST)fa);
+	*id = QTOS((Q)BDY(fab)); *r = NEXT(fab);
+}
+
+/* ret : [[quote(A),quote(1)],...] */
+
+NODE quote_unify(Obj f, Obj pat)
+{
+	NODE tf,tp,head,body;
+	NODE parg,farg,r;
+	LIST fa,l;
+	int pid,id;
+	FUNC ff,pf;
+
+	if ( OID(pat) == O_LIST ) {
+		if ( OID(f) == O_LIST )
+			return quote_unify_node(BDY((LIST)f),BDY((LIST)pat));
+		else
+			return 0;
+	} else if ( OID(pat) == O_QUOTE ) {
+		if ( OID(f) != O_QUOTE ) return 0;
+		get_quote_id_arg((QUOTE)pat,&pid,&parg);
+		get_quote_id_arg((QUOTE)f,&id,&farg);
+		switch ( pid ) {
+			case I_FORMULA:
+				if ( compquote(CO,f,pat) )
+					return 0;
+				else {
+					MKLIST(l,0);
+					return mknode(1,l);
+				}
+				break;
+			case I_LIST:
+				return quote_unify_node(BDY((LIST)BDY(farg)),BDY((LIST)BDY(parg)));
+			case I_PVAR:
+				/* [[pat,f]] */
+				r = mknode(2,pat,f); MKLIST(l,r);
+				return mknode(1,l);
+			case I_IFUNC:
+				/* F(X,Y,...) = ... */
+				if ( id == I_FUNC ) {
+					r = mknode(2,BDY(parg),BDY(farg)); MKLIST(l,r);
+					head = mknode(1,l);
+					body = quote_unify(BDY(NEXT(farg)),BDY(NEXT(parg)));
+					if ( !body ) return 0;
+					return merge_matching_node(head,body);
+				} else
+					return 0;
+			case I_BOP:
+			case I_FUNC:
+				/* X+Y = ... */
+				/* f(...) = ... */
+				if ( compqa(CO,BDY(farg),BDY(parg)) ) return 0;
+				return quote_unify_node(NEXT(farg),NEXT(parg));
+				break;
+			default:
+				if ( pid == id )
+					return quote_unify_node(farg,parg);
+				else
+					return 0;
+		}
+	}
+}
+
 void Pquotetotex(NODE arg,STRING *rp)
 {
 	TB tb;
@@ -837,6 +970,24 @@ char *symbol_name(char *name)
 	return 0;
 }
 
+void Pget_function_name(NODE arg,STRING *rp)
+{
+		QUOTEARG qa;
+		ARF f;
+		char *opname;
+
+		qa = (QUOTEARG)BDY(arg);
+		if ( !qa || OID(qa) != O_QUOTEARG || qa->type != A_arf )
+			*rp = 0;
+		else {
+			f = (ARF)BDY(qa);
+			opname = f->name;
+			MKSTR(*rp,opname);
+		}
+}
+
+FNODE strip_paren(FNODE);
+
 void fnodetotex_tb(FNODE f,TB tb)
 {
 	NODE n,t,t0;
@@ -909,7 +1060,7 @@ void fnodetotex_tb(FNODE f,TB tb)
 			} else if ( !strcmp(opname,"^") ) {
 				fnodetotex_tb((FNODE)FA1(f),tb);
 				write_tb("^{",tb);
-				fnodetotex_tb((FNODE)FA2(f),tb);
+				fnodetotex_tb(strip_paren((FNODE)FA2(f)),tb);
 				write_tb("} ",tb);
 			} else if ( !strcmp(opname,"%") ) {
 				fnodetotex_tb((FNODE)FA1(f),tb);
@@ -1195,6 +1346,42 @@ char *objtostr(Obj obj)
 	return r;
 }
 
+void Psprintf(NODE arg,STRING *rp)
+{
+    STRING string;
+    char *s,*t,*r;
+    int argc,n,len;
+    NODE node;
+
+    string = (STRING)ARG0(arg);
+    asir_assert(string,O_STR,"sprintf");
+    s = BDY(string);
+    for(n = 0, t = s; *t; t++) {
+        if (*t=='%' && *(t+1)=='a') {
+            n++;
+        }
+    }
+    for(node = NEXT(arg), argc = 0, len = strlen(s); node; node = NEXT(node), argc++) {
+        len += estimate_length(CO,BDY(node));
+    }
+    if (argc < n) {
+        error("sprintf: invalid argument");
+    }
+    r = (char *)MALLOC_ATOMIC(len);
+    for(node = NEXT(arg), t = r; *s; s++) {
+        if (*s=='%' && *(s+1)=='a') {
+            strcpy(t,objtostr(BDY(node)));
+            node = NEXT(node);
+            t = strchr(t,0);
+            s++;
+        }else {
+            *t++ = *s;
+        }
+    }
+    *t = 0;
+    MKSTR(*rp,r);
+}
+
 void fnodenodetotex_tb(NODE n,TB tb)
 {
 	for ( ; n; n = NEXT(n) ) {
@@ -1277,4 +1464,160 @@ int top_is_minus(FNODE f)
 		default:
 			return 0;
 	}
+}
+
+FNODE flatten_fnode(FNODE,char *);
+
+void Pflatten_quote(NODE arg,Obj *rp)
+{
+	FNODE f;
+	QUOTE q;
+
+	if ( !ARG0(arg) || OID((Obj)ARG0(arg)) != O_QUOTE )
+		*rp = (Obj)ARG0(arg);
+	else {
+		f = flatten_fnode(BDY((QUOTE)ARG0(arg)),BDY((STRING)ARG1(arg)));
+		MKQUOTE(q,f);
+		*rp = (Obj)q;
+	}
+}
+
+void Pquote_to_funargs(NODE arg,LIST *rp)
+{
+	fid_spec_p spec;
+	QUOTE q;
+	QUOTEARG qa;
+	FNODE f;
+	STRING s;
+	QUOTE r;
+	int i;
+	Q id,a;
+	LIST l;
+	NODE t0,t,w,u,u0;
+
+	q = (QUOTE)ARG0(arg);
+	if ( !q || OID(q) != O_QUOTE )
+		error("quote_to_funargs : invalid argument");
+	f = BDY(q);
+	if ( !f ) {
+		MKLIST(*rp,0);
+		return;
+	}
+	get_fid_spec(f->id,&spec);
+	if ( !spec )
+		error("quote_to_funargs : not supported yet");
+	t0 = 0;
+	STOQ((int)f->id,id);
+	NEXTNODE(t0,t);
+	BDY(t) = (pointer)id;
+	for ( i = 0; spec->type[i] != A_end; i++ ) {
+		NEXTNODE(t0,t);
+		switch ( spec->type[i] ) {
+			case A_fnode:
+				MKQUOTE(r,(FNODE)f->arg[i]);
+				BDY(t) = (pointer)r;
+				break;
+			case A_int:
+				STOQ((int)f->arg[i],a);
+				BDY(t) = (pointer)a;
+				break;
+			case A_str:
+				MKSTR(s,(char *)f->arg[i]);
+				BDY(t) = (pointer)s;
+				break;
+			case A_internal:
+				BDY(t) = (pointer)f->arg[i];
+				break;
+			case A_node:
+				w = (NODE)f->arg[i];
+				for ( u0 = 0; w; w = NEXT(w) ){
+					NEXTNODE(u0,u);
+					MKQUOTE(r,(FNODE)BDY(w));
+					BDY(u) = (pointer)r;
+				}
+				if ( u0 ) NEXT(u) = 0;
+				MKLIST(l,u0);
+				BDY(t) = (pointer)l;
+				break;
+			default:
+				MKQUOTEARG(qa,spec->type[i],f->arg[i]);	
+				BDY(t) = (pointer)qa;
+				break;
+		}
+	}
+	if ( t0 ) NEXT(t) = 0;
+	MKLIST(*rp,t0);
+}
+
+void Pfunargs_to_quote(NODE arg,QUOTE *rp)
+{
+	fid_spec_p spec;
+	QUOTE q;
+	QUOTEARG qa;
+	FNODE f;
+	STRING s;
+	QUOTE r,b;
+	int i;
+	LIST l;
+	fid id;
+	Obj a;
+	NODE t0,t,u0,u,w;
+
+	l = (LIST)ARG0(arg);
+	if ( !l || OID(l) != O_LIST || !(t=BDY(l)) )
+		error("funargs_to_quote : invalid argument");
+	t = BDY(l);
+	id = (fid)QTOS((Q)BDY(t)); t = NEXT(t);
+	get_fid_spec(id,&spec);
+	if ( !spec )
+		error("funargs_to_quote : not supported yet");
+	for ( i = 0; spec->type[i] != A_end; i++ );
+	NEWFNODE(f,i);
+	f->id = id;
+	for ( i = 0; spec->type[i] != A_end; i++, t = NEXT(t) ) {
+		if ( !t )
+			error("funargs_to_quote : argument mismatch");
+		a = (Obj)BDY(t);
+		switch ( spec->type[i] ) {
+			case A_fnode:
+				if ( !a || OID(a) != O_QUOTE )
+					error("funargs_to_quote : invalid argument");
+				f->arg[i] = BDY((QUOTE)a);
+				break;
+			case A_int:
+				if ( !INT(a) )
+					error("funargs_to_quote : invalid argument");
+				f->arg[i] = (pointer)QTOS((Q)a);
+				break;
+			case A_str:
+				if ( !a || OID(a) != O_STR )
+					error("funargs_to_quote : invalid argument");
+				f->arg[i] = (pointer)BDY((STRING)a);
+				break;
+			case A_internal:
+				f->arg[i] = (pointer)a;
+				break;
+			case A_node:
+				if ( !a || OID(a) != O_LIST )
+					error("funargs_to_quote : invalid argument");
+				u0 = 0;
+				for ( w = BDY((LIST)a); w; w = NEXT(w) ) {
+					NEXTNODE(u0,u);
+					b = (QUOTE)BDY(w);
+					if ( !b || OID(b) != O_QUOTE )
+						error("funargs_to_quote : invalid argument");
+					BDY(u) = BDY(b);
+				}
+				if ( u0 ) NEXT(u) = 0;
+				f->arg[i] = (pointer)u0;
+				break;
+			default:
+				if ( !a || OID(a) != O_QUOTEARG || 
+					((QUOTEARG)a)->type != spec->type[i] )
+					error("funargs_to_quote : invalid argument");
+				f->arg[i] = BDY((QUOTEARG)a);
+				break;
+		}
+	}
+	MKQUOTE(*rp,f);
 }