more codegen tweaks.

src/prolog/codegen.rs

@@ -768,18 +768,18 @@ impl<'a> CodeGenerator<'a> {
             }
         }
 
-        for (i, term) in clauses.iter().enumerate() {
-            if let &TermOrCut::Term(ref term) = term {
-                Self::mark_vars_in_term(term.breadth_first_iter(), &mut vs, i + 1)
-            }
-        }
-
         let mut deep_cuts = false;
 
-        for term in clauses {
+        for (i, term) in clauses.iter().enumerate() {
-            if let &TermOrCut::Cut = term {
+            match term {
+                &TermOrCut::Cut => {
                     deep_cuts = true;
-                break;
+                },
+                &TermOrCut::Term(ref term) => {
+                    Self::mark_vars_in_term(term.breadth_first_iter(),
+                                            &mut vs,
+                                            i + 1);
+                }
             }
         }
 
@@ -834,9 +834,7 @@ impl<'a> CodeGenerator<'a> {
         dealloc_index
     }
 
-    fn mark_unsafe_query_vars(head: &Term,
+    fn mark_unsafe_query_vars(head: &Term, vs: &VariableFixtures, query: &mut CompiledQuery)
-                              vs: &VariableFixtures,
-                              query: &mut CompiledQuery)
     {
         let mut unsafe_vars = HashMap::new();
 
@@ -909,7 +907,6 @@ impl<'a> CodeGenerator<'a> {
             }
         };
 
-        
         self.marker.advance(p0);
 
         if p0.is_clause() {
@@ -1003,7 +1000,8 @@ impl<'a> CodeGenerator<'a> {
         }
     }
 
-    pub fn compile_fact(&mut self, term: &'a Term) -> Code {
+    pub fn compile_fact(&mut self, term: &'a Term) -> Code
+    {
         self.marker.advance(term);
         self.update_var_count(term.breadth_first_iter());
 
@@ -1038,7 +1036,8 @@ impl<'a> CodeGenerator<'a> {
         code
     }
 
-    pub fn compile_query(&mut self, term: &'a Term) -> Code {
+    pub fn compile_query(&mut self, term: &'a Term) -> Code
+    {
         self.marker.advance(term);
         self.update_var_count(term.breadth_first_iter());