Fix for print_visitor.

Signed-off-by: James Goppert <james.goppert@gmail.com>

Author: jgoppert

src/main.rs

@@ -19,6 +19,10 @@ fn main() -> Result<(), Box<dyn std::error::Error>> {
     let args = Args::parse();
     let def = rumoca_parser::s1_parser::parse_file(&args.model_file);
 
+    if args.verbose {
+        println!("{:#?}", def);
+    }
+
     let mut visitor = PrintVisitor::default();
     Walker::walk_stored_definition(&mut visitor, &def);
     Ok(())

src/s1_parser/print_visitor.rs

@@ -52,8 +52,8 @@ impl Visitor for PrintVisitor {
             ast::Expression::Unary { op, .. } => {
                 self.print(&format!("{:?}", op));
             }
-            ast::Expression::Ref { comp } => {
-                self.print(&format!("{:?}", comp));
+            ast::Expression::Ref { .. } => {
+                self.print("ref");
             }
             ast::Expression::UnsignedInteger(val) => {
                 self.print(&format!("{:?}", val));
@@ -70,8 +70,8 @@ impl Visitor for PrintVisitor {
             ast::Expression::ArrayArguments { .. } => {
                 self.print("array_args");
             }
-            ast::Expression::FunctionCall { comp, .. } => {
-                self.print(&format!("{:?}", comp));
+            ast::Expression::FunctionCall { .. } => {
+                self.print("function call");
             }
             ast::Expression::Der { .. } => {
                 self.print("der");
@@ -108,4 +108,15 @@ impl Visitor for PrintVisitor {
             ast::Element::ExtendsClause { .. } => {}
         }
     }
+
+    fn exit_component_reference(&mut self, comp: &ast::ComponentReference) {
+        let mut s: String = "".to_string();
+        for (index, part) in comp.parts.iter().enumerate() {
+             if index != 0 || comp.local {
+                 s += ".";
+             }
+             s += &part.name;
+         }
+         self.print(&s);
+    }
 }