X-Git-Url: https://www.ginac.de/ginac.git//ginac.git?p=ginac.git;a=blobdiff_plain;f=ginac%2Finput_lexer.ll;h=4be043a55738cd6787a5be0643ab96dda57838d7;hp=e4757b9d8181584e48f6f91de150e5b889fb74dd;hb=f4f7f966d313edf65d46216c7eb48d0791c6537c;hpb=6bff4f32e5cbe9f7e51837f392a2d0bf0d5b721d

diff --git a/ginac/input_lexer.ll b/ginac/input_lexer.ll
index e4757b9d..4be043a5 100644
--- a/ginac/input_lexer.ll
+++ b/ginac/input_lexer.ll
@@ -4,7 +4,7 @@
  *  This file must be processed with flex. */
 
 /*
- *  GiNaC Copyright (C) 1999-2001 Johannes Gutenberg University Mainz, Germany
+ *  GiNaC Copyright (C) 1999-2004 Johannes Gutenberg University Mainz, Germany
  *
  *  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
@@ -39,6 +39,7 @@
 #include "numeric.h"
 #include "symbol.h"
 #include "lst.h"
+#include "idx.h"
 
 using namespace GiNaC;
 namespace GiNaC {
@@ -47,7 +48,7 @@ namespace GiNaC {
 
 } // namespace GiNaC
 
-// Table of all used symbols
+// Table of all used symbols/indices
 struct sym_def {
 	sym_def() : predefined(false) {}
 	sym_def(const ex &s, bool predef) : sym(s), predefined(predef) {}
@@ -86,7 +87,7 @@ AN	[0-9a-zA-Z_]
  */
 
 %%
-[ \t]+			/* skip whitespace */
+[ \t\n]+		/* skip whitespace */
 
 			/* special values */
 Pi			ginac_yylval = Pi; return T_LITERAL;
@@ -122,6 +123,9 @@ Digits			ginac_yylval = (long)Digits; return T_DIGITS;
 				return T_SYMBOL;
 			}
 
+			/* end of input */
+<<EOF>>			return T_EOF;
+
 			/* everything else */
 .			return *yytext;
 
@@ -166,22 +170,34 @@ void set_lexer_string(const std::string &s)
 	curr_pos = 0;
 }
 
-// Set the list of predefined symbols
+// Get name of symbol/index
+std::string get_symbol_name(const ex & s)
+{
+	if (is_a<symbol>(s))
+		return ex_to<symbol>(s).get_name();
+	else if (is_a<idx>(s) && is_a<symbol>(s.op(0)))
+		return ex_to<symbol>(s.op(0)).get_name();
+	else
+		throw (std::runtime_error("get_symbol_name(): unexpected expression type"));
+}
+
+// Set the list of predefined symbols/indices
 void set_lexer_symbols(ex l)
 {
 	syms.clear();
 	if (!is_exactly_a<lst>(l))
 		return;
 	for (unsigned i=0; i<l.nops(); i++) {
-		if (is_exactly_a<symbol>(l.op(i)))
-			syms[ex_to<symbol>(l.op(i)).get_name()] = sym_def(l.op(i), true);
+		const ex &o = l.op(i);
+		if (is_a<symbol>(o) || (is_a<idx>(o) && is_a<symbol>(o.op(0))))
+			syms[get_symbol_name(o)] = sym_def(o, true);
 	}
 }
 
-// Check whether symbol was predefined
+// Check whether symbol/index was predefined
 bool is_lexer_symbol_predefined(const ex &s)
 {
-	sym_tab::const_iterator i = syms.find(ex_to<symbol>(s).get_name());
+	sym_tab::const_iterator i = syms.find(get_symbol_name(s));
 	if (i == syms.end())
 		return false;
 	else