From: Mark Thom <markjordanthom@gmail.com>
Date: Sun, 7 Apr 2019 18:14:20 +0000 (-0600)
Subject: correct conformity test #234
X-Git-Tag: v0.8.110~106
X-Git-Url: https://git.sagredo.dev/?a=commitdiff_plain;h=e9c8d6adb94b77dbe9507a73fc7d6113f676eb5a;p=scryer-prolog.git

correct conformity test #234
---

diff --git a/Cargo.toml b/Cargo.toml
index 5c5238ac..5f2ef357 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -1,6 +1,6 @@
 [package]
 name = "scryer-prolog"
-version = "0.8.49"
+version = "0.8.50"
 authors = ["Mark Thom <markjordanthom@gmail.com>"]
 repository = "https://github.com/mthom/scryer-prolog"
 description = "A modern Prolog implementation written mostly in Rust."
diff --git a/src/prolog/heap_print.rs b/src/prolog/heap_print.rs
index 80e8db42..e472fa2e 100644
--- a/src/prolog/heap_print.rs
+++ b/src/prolog/heap_print.rs
@@ -281,8 +281,8 @@ macro_rules! push_space_if_amb {
 fn requires_space(atom: &str, op: &str) -> bool {
     match atom.chars().last() {
         Some(ac) => op.chars().next().map(|oc| {
-            if alpha_char!(ac) {
-                alpha_numeric_char!(oc)
+            if alpha_numeric_char!(ac) {
+                oc == '(' || alpha_numeric_char!(oc)
             } else if graphic_token_char!(ac) {
                 graphic_token_char!(oc)
             } else if variable_indicator_char!(ac) {
@@ -423,14 +423,14 @@ impl<'a, Outputter: HCValueOutputter> HCPrinter<'a, Outputter>
             self.state_stack.push(TokenOrRedirect::CompositeRedirect(left_directed_op));
             self.state_stack.push(TokenOrRedirect::Op(ct.name(), spec));
         } else { // if is_infix!(spec.assoc())
-            match ct.name().as_str() {                
+            match ct.name().as_str() {
                 "|" => {
                     self.format_infix_op_with_space(ct.name(), spec);
                     return;
                 },
                 _ => {}
             };
-            
+
             let left_directed_op  = DirectedOp::Left(ct.name(), spec.clone());
             let right_directed_op = DirectedOp::Right(ct.name(), spec.clone());
 
@@ -677,19 +677,25 @@ impl<'a, Outputter: HCValueOutputter> HCPrinter<'a, Outputter>
     fn print_constant(&mut self, c: Constant, op: &Option<DirectedOp>) {
         match c {
             Constant::Atom(ref atom, Some(ref spec)) if spec.prec() > 0 => {
+                let mut result = String::new();
+
                 if let Some(ref op) = op {
                     if self.outputter.ends_with(&format!(" {}", op.as_str())) {
-                        self.push_char(' ');
+                        result.push(' ');
                     }
 
-                    self.push_char('(');
+                    result.push('(');
                 }
 
-                self.print_atom(atom);
+                result += &self.print_op_addendum(atom.as_str());
 
                 if op.is_some() {
-                    self.push_char(')');
+                    result.push(')');
                 }
+
+                push_space_if_amb!(self, &result, {
+                    self.append_str(&result);
+                });
             },
             Constant::Atom(ref atom, _) =>
                 push_space_if_amb!(self, atom.as_str(), {
@@ -800,6 +806,12 @@ impl<'a, Outputter: HCValueOutputter> HCPrinter<'a, Outputter>
 
                 if add_brackets {
                     self.state_stack.push(TokenOrRedirect::Open);
+
+                    if let Some(ref op) = &op {
+                        if requires_space(op.as_str(), "(") {
+                            self.state_stack.push(TokenOrRedirect::Space);
+                        }
+                    }
                 }
             },
             HeapCellValue::NamedStr(0, name, fixity) =>