/*-----------------------------------------------------------------------

File  : ccl_formulafunc.c

Author: Stephan Schulz

Contents

  Higher level formula functions that need to know about sets (and
  CNFing).

  Copyright 2004 by the author.
  This code is released under the GNU General Public Licence and
  the GNU Lesser General Public License.
  See the file COPYING in the main E directory for details..
  Run "eprover -h" for contact information.

Changes

<1> Sun Apr  4 14:12:05 CEST 2004
    New

-----------------------------------------------------------------------*/

#include "ccl_formulafunc.h"
#include "ccl_clausefunc.h"
#include "cte_lambda.h"

#define MAX_RW_STEPS 500

/*---------------------------------------------------------------------*/
/*                        Global Variables                             */
/*---------------------------------------------------------------------*/
extern bool app_encode;

/*---------------------------------------------------------------------*/
/*                      Forward Declarations                           */
/*---------------------------------------------------------------------*/

typedef TFormula_p (*FormulaMapper)(TFormula_p, TB_p);

/*---------------------------------------------------------------------*/
/*                         Internal Functions                          */
/*---------------------------------------------------------------------*/

/*-----------------------------------------------------------------------
//
// Function: FlattenApps_driver()
//
//   Apply additional arguments to hd assuming hd needs to be flattened.
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/

Term_p FlattenApps_driver(TB_p terms, Term_p t)
{
   if(TermIsPhonyApp(t) && !TermIsPhonyAppTarget(t->args[0]))
   {
      return FlattenApps(terms, t->args[0], t->args+1, t->arity-1, t->type);
   }
   else
   {
      return t;
   }
}

/*-----------------------------------------------------------------------
//
// Function: close_let_def()
//
//   For each defined symbol f(bound vars) = s, finds what are free variables
//   in s and creates a fresh symbol f_fresh(free_vars, bound_vars)
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

static void close_let_def(TB_p bank, NumTree_p *closed_defs, Term_p def)
{
   assert(def->f_code == bank->sig->eqn_code);
   PTree_p free_vars = NULL;
   Term_p lhs = def->args[0], rhs = def->args[1];
   TFormulaCollectFreeVars(bank, rhs, &free_vars);

   for (int i = 0; i < lhs->arity; i++)
   {
      Term_p arg = lhs->args[i];
      assert(TermIsFreeVar(arg));
      PTreeDeleteEntry(&free_vars, arg);
   }

   PStack_p all_vars = PStackAlloc();
   PTreeToPStack(all_vars, free_vars);
   for (int i = 0; i < lhs->arity; i++)
   {
      PStackPushP(all_vars, lhs->args[i]);
   }
   Term_p fresh_sym = TermAllocNewSkolem(bank->sig, all_vars, lhs->type);
   fresh_sym = TBTermTopInsert(bank, fresh_sym);

   IntOrP orig_def = {.p_val = lhs}, new_def = {.p_val = fresh_sym};
   NumTreeStore(closed_defs, lhs->f_code, orig_def, new_def);

   PTreeFree(free_vars);
   PStackFree(all_vars);
}

/*-----------------------------------------------------------------------
//
// Function: replace_body()
//
//   Replace all occurences of old symbols with new definitions.
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

static Term_p replace_body(TB_p bank, NumTree_p *closed_defs, Term_p t)
{
   NumTree_p node = NumTreeFind(closed_defs, t->f_code);
   Term_p new = TermTopCopyWithoutArgs(t);
   bool changed = false;
   for (long i = 0; i < t->arity; i++)
   {
      new->args[i] = replace_body(bank, closed_defs, t->args[i]);
      changed = changed || new->args[i] != t->args[i];
   }

   if (!changed)
   {
      TermTopFree(new);
      new = t;
   }
   else
   {
      new = TBTermTopInsert(bank, new);
   }

   if (node)
   {
      Term_p old_def = node->val1.p_val;
      Term_p new_def = node->val2.p_val;

      Subst_p subst = SubstAlloc();
      for (long i = 0; i < new->arity; i++)
      {
         SubstAddBinding(subst, old_def->args[i], new->args[i]);
      }

      new = TBInsertInstantiated(bank, new_def);
      SubstDelete(subst);
   }

   return new;
}

/*-----------------------------------------------------------------------
//
// Function: make_fresh_defs()
//
//   Make a formula introducing a new name for a local let definition
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

void make_fresh_defs(TB_p bank, Term_p let_t, NumTree_p *defs, PStack_p res)
{
   assert(let_t->f_code == SIG_LET_CODE);
   long num_def = let_t->arity - 1;
   Sig_p sig = bank->sig;
   for (long i = 0; i < num_def; i++)
   {
      assert(let_t->args[i]->f_code == bank->sig->eqn_code);
      FunCode old_lhs_fc = let_t->args[i]->args[0]->f_code;
      Term_p rhs = let_t->args[i]->args[1];
      NumTree_p node = NumTreeFind(defs, old_lhs_fc);
      assert(node);
      Term_p new_lhs = node->val2.p_val;
      TFormula_p matrix;

      if (TypeIsBool(rhs->type))
      {
         matrix = TFormulaFCodeAlloc(bank, sig->equiv_code,
                                     EncodePredicateAsEqn(bank, new_lhs),
                                     EncodePredicateAsEqn(bank, rhs));
      }
      else
      {
         matrix = TFormulaFCodeAlloc(bank, sig->eqn_code, new_lhs, rhs);
      }

      PStackPushP(res, TFormulaClosure(bank, matrix, true));
   }
}

/*-----------------------------------------------------------------------
//
// Function: lift_lets()
//
//   Does the actual lifting of let terms
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

TFormula_p lift_lets(TB_p terms, TFormula_p t, PStack_p fresh_defs)
{
   if (TermIsAnyVar(t))
   {
      return t;
   }
   else if (t->f_code == SIG_LET_CODE)
   {
      Term_p new = TermTopCopyWithoutArgs(t);
      NumTree_p closed_defs = NULL;
      long num_defs = t->arity - 1;
      for (long i = 0; i < num_defs; i++)
      {
         new->args[i] = lift_lets(terms, t->args[i], fresh_defs);
         close_let_def(terms, &closed_defs, new->args[i]);
      }
      make_fresh_defs(terms, new, &closed_defs, fresh_defs);
      TermTopFree(new);
      Term_p res = replace_body(terms, &closed_defs, t->args[num_defs]);
      NumTreeFree(closed_defs);
      return lift_lets(terms, res, fresh_defs);
   }
   else
   {
      Term_p new = TermTopCopyWithoutArgs(t);
      bool changed = false;
      for (int i = 0; i < new->arity; i++)
      {
         new->args[i] = lift_lets(terms, t->args[i], fresh_defs);
         changed = changed || new->args[i] != t->args[i];
      }

      if (changed)
      {
         if(TermIsPhonyApp(new) && !TermIsPhonyAppTarget(new->args[0]))
         {
            Term_p flat = FlattenApps(terms, new->args[0], new->args+1, new->arity-1, new->type);
            TermTopFree(new);
            new = flat;
         }
         new = TBTermTopInsert(terms, new);

      }
      else
      {
         TermTopFree(new);
         new = t;
      }
      return new;
   }
}

/*-----------------------------------------------------------------------
//
// Function: unencode_eqns()
//
//   Undo encoding of the form **formula** = $true to **formula**
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

TFormula_p unencode_eqns(TB_p terms, TFormula_p t)
{
   Term_p res = t;
   if(!TermHasEqNeq(t))
   {
      res = NULL;
   }
   else if (t->f_code == terms->sig->eqn_code
            && t->arity == 2
            && t->args[1] == terms->true_term
            && !TermIsAnyVar(t->args[0])
            && (SigQueryFuncProp(terms->sig, t->args[0]->f_code, FPFOFOp)
                || t->args[0]->f_code == terms->sig->qex_code
                || t->args[0]->f_code == terms->sig->qall_code
                || t->args[0]->f_code == terms->sig->eqn_code
                || t->args[0]->f_code == terms->sig->neqn_code))
   {
      res = t->args[0];
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: refresh_qvars()
//
//   Bind all quantified variables in form to fresh free variables.
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

Term_p refresh_qvars(TB_p terms, Term_p form)
{
   Term_p res;
   Sig_p sig = terms->sig;
   if (TermIsGround(form))
   {
      res = form;
   }
   else if (TermIsFreeVar(form))
   {
      assert(form->binding);
      res = form->binding;
   }
   else if (TermIsDBVar(form) || form->arity == 0)
   {
      res = form;
   }
   else if ((form->f_code == sig->qall_code || form->f_code == sig->qex_code) && form->arity == 2)
   {
      Term_p prev_binding = form->args[0]->binding;
      Term_p fresh_var = VarBankGetFreshVar(terms->vars, form->args[0]->type);
      form->args[0]->binding = fresh_var;
      res = TFormulaFCodeAlloc(terms, form->f_code,
                               fresh_var, refresh_qvars(terms, form->args[1]));
      form->args[0]->binding = prev_binding;
   }
   else
   {
      res = TermTopCopyWithoutArgs(form);
      bool changed = false;

      for (int i = 0; i < form->arity; i++)
      {
         res->args[i] = refresh_qvars(terms, form->args[i]);
         changed = changed || form->args[i] != res->args[i];
      }

      if (changed)
      {
         res = TBTermTopInsert(terms, res);
      }
      else
      {
         TermTopFree(res);
         res = form;
      }
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: do_rw_with_defs()
//
//   Actually performs rewriting on a term using definition map.
//   Stores used formulas in used_defs.
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

Term_p do_rw_with_defs(TB_p terms, Term_p t, IntMap_p def_map,
                       PTree_p *used_defs, int *steps, bool snf)
{
   if (*steps <= 0)
   {
      return t;
   }

   bool changed = false;
   Term_p new = TermTopAlloc(t->f_code, t->arity);
   for (long i = 0; i < t->arity; i++)
   {
      new->args[i] = do_rw_with_defs(terms, t->args[i], def_map, used_defs, steps, snf);
      changed = changed || new->args[i] != t->args[i];
   }

   if (!changed)
   {
      TermTopFree(new);
      new = t;
   }
   else
   {
      new = TBTermTopInsert(terms, new);
   }

   WFormula_p wform = IntMapGetVal(def_map, new->f_code);
   if (wform)
   {
      Term_p rhs = refresh_qvars(terms, wform->tformula->args[1]);
      PStack_p args = PStackAlloc();
      for (long i = 0; i < new->arity; i++)
      {
         PStackPushP(args, new->args[i]);
      }
      new = snf ? BetaNormalizeDB(terms, ApplyTerms(terms, rhs, args))
                : ApplyTerms(terms, rhs, args);
      PTreeStore(used_defs, wform);
      new = do_rw_with_defs(terms, new,
                            def_map, used_defs, steps, snf);
      *steps = *steps - 1; //forgot the precedence :)
      PStackFree(args);
   }

   return new;
}

/*-----------------------------------------------------------------------
//
// Function: create_sym_map()
//
//   Creates a map from symbol to WFormula describing the simplified
//   definition f = \xyz. body
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

PTree_p create_sym_map(FormulaSet_p set, IntMap_p sym_def_map, bool unfold_only_forms)
{
   PTree_p recognized_definitions = NULL;
   for (WFormula_p form = set->anchor->succ; form != set->anchor; form = form->succ)
   {
      if (!(FormulaQueryProp(form, CPIsLambdaDef)))
      {
         continue;
      }

      TB_p bank = form->terms;
      Sig_p sig = form->terms->sig;
      Term_p lhs = NULL, rhs = NULL;
      Term_p tform = form->tformula;

      while (tform->f_code == sig->qall_code && tform->arity == 2)
      {
         tform = tform->args[1];
      }

      if (tform->f_code == sig->eqn_code)
      {
         lhs = tform->args[0];
         rhs = tform->args[1];
      }
      else if (tform->f_code == sig->equiv_code &&
               tform->args[0]->f_code == sig->eqn_code &&
               tform->args[0]->args[1] == bank->true_term)
      {
         lhs = tform->args[0]->args[0];
         rhs = tform->args[1];
      }
      else
      {
         continue;
      }

      PStack_p bvars = PStackAlloc();
      Term_p lhs_body = UnfoldLambda(lhs, bvars);
      for (long i = 0; i < PStackGetSP(bvars); i++)
      {
         Type_p ty = ((Term_p)PStackElementP(bvars, i))->type;
         Term_p tmp = VarBankGetFreshVar(bank->vars, ty);
         PStackAssignP(bvars, i, tmp);
      }
      lhs_body = BetaNormalizeDB(bank, ApplyTerms(bank, lhs, bvars));
      Term_p rhs_applied = BetaNormalizeDB(bank, ApplyTerms(bank, rhs, bvars));

      // now the definition is of the form f @ ..terms.. = \xyz. body
      // and we need to check if terms are distinct variables
      // and if \terms\xyz.body has no free variables, and if lhs does
      // not appear in
      bool is_def = (!unfold_only_forms || TypeIsPredicate(lhs->type)) &&
                    lhs->f_code > sig->internal_symbols && rhs != bank->true_term;
      PStackReset(bvars);
      for (long i = 0; is_def && i < lhs_body->arity; i++)
      {
         Term_p arg = lhs_body->args[i];
         if (arg->f_code == sig->eqn_code &&
             arg->arity == 2 &&
             arg->args[1] == bank->true_term)
         {
            arg = arg->args[0];
         }
         if (!TermIsFreeVar(arg) || TermCellQueryProp(arg, TPIsSpecialVar))
         {
            is_def = false;
         }
         else
         {
            TermCellSetProp(arg, TPIsSpecialVar);
            PStackPushP(bvars, arg);
         }
      }

      if (is_def && !TermHasFCode(rhs_applied, lhs_body->f_code))
      {
         rhs = AbstractVars(bank, rhs_applied, bvars);
         if (!TFormulaHasFreeVars(bank, rhs))
         {
            lhs = TermTopAlloc(lhs_body->f_code, 0);
#ifdef NDEBUG
            // optimization
            lhs->type = SigGetType(sig, lhs_body->f_code);
#endif
            lhs = TBTermTopInsert(bank, lhs);

            TFormula_p def = TFormulaFCodeAlloc(bank, sig->eqn_code, lhs, rhs);
            WFormula_p def_wform = WFormulaFlatCopy(form);
            def_wform->tformula = def;
            WFormulaPushDerivation(def_wform, DCFofQuote, form, NULL);
            IntMapAssign(sym_def_map, lhs->f_code, def_wform);

            PTreeStore(&recognized_definitions, form);
         }
      }

      for (PStackPointer i = 0; i < PStackGetSP(bvars); i++)
      {
         TermCellDelProp(((Term_p)PStackElementP(bvars, i)), TPIsSpecialVar);
      }
      PStackFree(bvars);
   }

   return recognized_definitions;
}

/*-----------------------------------------------------------------------
//
// Function: intersimplify_definitions()
//
//   Make sure that the definitions themselves are rewritten using
//   terms in sym_def_map.
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/
void intersimplify_definitions(TB_p terms, IntMap_p sym_def_map)
{
   IntMapIter_p iter = IntMapIterAlloc(sym_def_map, 0, LONG_MAX);
   long i;
   WFormula_p next = NULL;
   while ((next = IntMapIterNext(iter, &i)))
   {
      PTree_p used_defs = NULL;
      int max_steps = MAX_RW_STEPS;
      Term_p new_rhs = do_rw_with_defs(terms, next->tformula->args[1],
                                       sym_def_map, &used_defs,
                                       &max_steps, true);
      if (new_rhs != next->tformula->args[1])
      {
         assert(used_defs);
         next->tformula->args[1] = new_rhs;
         PStack_p ptiter = PTreeTraverseInit(used_defs);
         PTree_p node = NULL;
         while ((node = PTreeTraverseNext(ptiter)))
         {
            WFormulaPushDerivation(next, DCApplyDef, node->key, NULL);
         }
         PTreeTraverseExit(ptiter);
      }
      PTreeFree(used_defs);
   }
   IntMapIterFree(iter);
}

/*-----------------------------------------------------------------------
//
// Function: map_formula()
//
//   Applies processor to form. If formula is changed it alters
//   the proof object by applying the correct derivation code.
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

bool map_formula(WFormula_p form, TB_p terms, FormulaMapper processor, DerivationCode dc)
{
   TFormula_p original = form->tformula;
   bool changed = false;

   form->tformula = processor(original, terms);

   if (form->tformula != original)
   {
      WFormulaPushDerivation(form, dc, NULL, NULL);
      changed = true;
   }

   return changed;
}

/*-----------------------------------------------------------------------
//
// Function: ignore_include()
//
//   Ignore includes and echoes the ignored declaration. Used for
//   app encoding only.
//
// Global Variables: -
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

void ignore_include(Scanner_p in)
{
   assert(app_encode);

   AcceptInpId(in, "include");
   AcceptInpTok(in, OpenBracket);
   CheckInpTok(in, SQString);
   char *name = DStrCopyCore(AktToken(in)->literal);
   NextToken(in);
   AcceptInpTok(in, CloseBracket);
   AcceptInpTok(in, Fullstop);

   fprintf(stdout, "include('%s').\n", name);
   FREE(name);
}

/*-----------------------------------------------------------------------
//
// Function: answer_lit_alloc()
//
//   Allocate a FOF literal of the form ~$answer(skn(x1, ... xn)),
//   where the xi are the variables on varstack and skn is a new
//   skolem symbol.
//
// Global Variables: -
//
// Side Effects    : Memory operations, changes the term bank.
//
/----------------------------------------------------------------------*/

TFormula_p answer_lit_alloc(TB_p terms, PStack_p varstack)
{
   TFormula_p res;
   Term_p handle;

   handle = TBAllocNewSkolem(terms, varstack, NULL);
   res = TermTopAlloc(terms->sig->answer_code, 1);
   res->args[0] = handle;
   res->type = terms->sig->type_bank->bool_type;
   res = TBTermTopInsert(terms, res);
   res = EqnTermsTBTermEncode(terms, res, terms->true_term, false, PENormal);

   return res;
}

/*-----------------------------------------------------------------------
//
// Function: verify_name()
//
//   If name_selector is NULL, return true. Otherwise, check if
//   info->name is in name_selector. Return true if yes, false
//   otherwise.
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/

static bool verify_name(StrTree_p *name_selector, ClauseInfo_p info)
{
   StrTree_p handle;

   if (!(*name_selector))
   {
      return true;
   }
   handle = StrTreeFind(name_selector, info->name);
   if (!handle)
   {
      return false;
   }
   handle->val1.i_val = 1; /* Mark as found */
   return true;
}

/*-----------------------------------------------------------------------
//
// Function: check_all_found()
//
//   Check if all names in name_selector are marked as found. Print a
//   useful error message and terminate otherwise.
//
// Global Variables: -
//
// Side Effects    : Possibly exit, memory operations.
//
/----------------------------------------------------------------------*/

static void check_all_found(Scanner_p in, StrTree_p name_selector)
{
   PStack_p trav_stack, err_stack = PStackAlloc();
   StrTree_p handle;

   trav_stack = StrTreeTraverseInit(name_selector);

   while ((handle = StrTreeTraverseNext(trav_stack)))
   {
      if (!handle->val1.i_val)
      {
         PStackPushP(err_stack, handle->key);
      }
   }
   StrTreeTraverseExit(trav_stack);

   if (!PStackEmpty(err_stack))
   {
      DStr_p err_str = DStrAlloc();
      char *sep = "";
      PStackPointer i;

      if (in->include_pos)
      {
         DStrSet(err_str, in->include_pos);
         DStrAppendStr(err_str, " ");
      }
      DStrAppendStr(err_str, "\"include\" statement cannot "
                             "find the following requested clauses/formulae in ");
      DStrAppendDStr(err_str, Source(in));
      DStrAppendStr(err_str, ": ");

      for (i = 0; i < PStackGetSP(err_stack); i++)
      {
         DStrAppendStr(err_str, sep);
         DStrAppendStr(err_str, PStackElementP(err_stack, i));
         sep = ", ";
      }
      Error(DStrView(err_str), INPUT_SEMANTIC_ERROR);
      DStrFree(err_str);
   }

   PStackFree(err_stack);
}

/*-----------------------------------------------------------------------
//
// Function: fool_should_ignore()
//
//   Is the term a variable encoded as X = true, X!=true or a negation
//   thereof. Or is it of the form $eq(true, true) or $eq(false, false).
//
// Global Variables: -
//
// Side Effects    : Input, Memory operations
//
/----------------------------------------------------------------------*/

static bool fool_should_ignore(Term_p t, TB_p bank)
{
   Sig_p sig = bank->sig;
   if(!TypeIsBool(t->type))
   {
      return false;
   }

   if(t->f_code == sig->eqn_code || t->f_code == sig->neqn_code)
   {
      assert(t->arity == 2);
      if(t->args[1] == bank->true_term)
      {
         return TermIsFreeVar(t->args[0]) || t->args[0] == bank->true_term;
      }
   }
   return TermIsFreeVar(t);
}


/*-----------------------------------------------------------------------
//
// Function: find_fool_subterm()
//
//   Returns true if it finds a formula subterm in t. pos is the position
//   corresponding to this subterm if it is found, empty otherwise.
//
// Global Variables:
//
// Side Effects    :
//
/----------------------------------------------------------------------*/

bool find_fool_subterm(TB_p bank, Term_p t, TermPos_p pos)
{
   int i;
   PStackPushP(pos, t);
   bool found = false;

   if(!TermIsLambda(t) && TermHasBoolSubterm(t))
   {
      for(i=0; i<t->arity; i++)
      {
         PStackPushInt(pos, i);

         if(TypeIsBool(t->args[i]->type))
         {
            if(!(fool_should_ignore(t->args[i], bank) ||
                  t->args[i]->f_code == SIG_TRUE_CODE
               || t->args[i]->f_code == SIG_FALSE_CODE))
            {
               found = true;
               break;
            }
         }
         else if(find_fool_subterm(bank, t->args[i], pos))
         {
            found = true;
            break;
         }

         PStackDiscardTop(pos);
      }
   }


   if(!found)
   {
      // did not find formula subterm
      PStackDiscardTop(pos);
      return false;
   }
   else
   {
      return true;
   }
}


/*-----------------------------------------------------------------------
//
// Function: do_fool_unroll()
//
//   Unroll boolean arguments of terms. For example, subformula
//   "f(a, p&q) = a" is replaced with "(~(p&q)|f(a,$true)=a) &
//   (p&q)|f(a, $false)=a".
//
// Global Variables: -
//
// Side Effects    : Changes enclosed formula
//
/----------------------------------------------------------------------*/

TFormula_p do_fool_unroll(TFormula_p form, TB_p terms)
{
   TFormula_p unrolled1 = NULL;
   TFormula_p unrolled2 = NULL;

   if (TFormulaIsLiteral(terms->sig, form))
   {
      /* printf("f_code: %ld  arity: %d   sig->eqn_code: %ld  sig->negn_code: %ld\n", */
      /*        form->f_code, form->arity, terms->sig->eqn_code, terms->sig->neqn_code); */
      /* printf("This is a literal %p %p: ", form->args[0], form->args[1]); */
      /* TermPrettyPrintSimple(stdout, form, terms->sig, 0); */
      /* printf("\n"); */
      /* printf("arg[0]: %p\n", form->args[0]); */
      /* TBPrintTerm(stdout, terms,  form->args[0], true); */
      /* printf("\n"); */
      /* printf("arg[1]: "); */
      /* TBPrintTerm(stdout, terms,  form->args[1], true); */
      /* printf("\n"); */
      form = LambdaEtaReduceDB(terms, form);
      TermPos_p pos = PStackAlloc();
      PStackPushP(pos, form);
      PStackPushInt(pos, 0);
      if (!find_fool_subterm(terms, form->args[0], pos))
      {
         PStackDiscardTop(pos);
         PStackPushInt(pos, 1);
         if (!find_fool_subterm(terms, form->args[1], pos))
         {
            PStackReset(pos);
         }
      }

      if (!PStackEmpty(pos))
      {
         TFormula_p subform =
             ((Term_p)PStackBelowTopP(pos))->args[PStackTopInt(pos)];
         assert(TypeIsBool(subform->type));

            // This is a Skolem symbol that is not yet encoded as literal

         subform = EncodePredicateAsEqn(terms, subform);

         Term_p subform_t = TBTermPosReplace(terms, terms->true_term, pos,
                                             DEREF_NEVER, 0, subform);
         Term_p subform_f = TBTermPosReplace(terms, terms->false_term, pos,
                                             DEREF_NEVER, 0, subform);

         TFormula_p neg_subf = TFormulaNegate(subform, terms);

         // ~(p&q)|f(a,$true)=a from the above example
         TFormula_p fst_impl = TFormulaFCodeAlloc(terms, terms->sig->or_code,
                                                  neg_subf, subform_t);
         // (p&q)|f(a, $false)=a
         TFormula_p snd_impl = TFormulaFCodeAlloc(terms, terms->sig->or_code,
                                                  subform, subform_f);

         // the whole formula
         form = TFormulaFCodeAlloc(terms, terms->sig->and_code,
                                   do_fool_unroll(fst_impl, terms),
                                   do_fool_unroll(snd_impl, terms));
      }
      PStackFree(pos);
   }
   else
   {
      if (TFormulaIsQuantified(terms->sig, form) && !TermIsLambda(form))
      {
         unrolled1 = do_fool_unroll(form->args[1], terms);
         if (form->args[1] != unrolled1)
         {
            form = TFormulaQuantorAlloc(terms, form->f_code,
                                        form->args[0], unrolled1);
         }
      }
      else if (!TermIsLambda(form))
      {
         if (TFormulaHasSubForm1(terms->sig, form))
         {
            unrolled1 = do_fool_unroll(form->args[0], terms);
         }
         if (TFormulaHasSubForm2(terms->sig, form))
         {
            unrolled2 = do_fool_unroll(form->args[1], terms);
         }

         if ((unrolled1 && unrolled1 != form->args[0]) ||
             (unrolled2 && unrolled2 != form->args[1]))
         {
            form = TFormulaFCodeAlloc(terms, form->f_code, unrolled1, unrolled2);
         }
      }
   }
   return form;
}


/*-----------------------------------------------------------------------
//
// Function: do_ite_unroll()
//
//   If $ite(c, it, if) occurs at formula position p, replace f|_p
//   with f[c -> it /\ ~c -> if]_p. If it occurs at subterm position p,
//   find the first above formula position q and do the replacement
//   f[c -> f[it]_p /\ ~c -> f[if]_p]_q.
//
// Global Variables: -
//
// Side Effects    : Changes enclosed formula
//
/----------------------------------------------------------------------*/

TFormula_p do_ite_unroll(TFormula_p form, TB_p terms)
{
   //TFormula_p safe = form;

   if (form->f_code == SIG_ITE_CODE)
   {
      assert(form->arity == 3);
      TFormula_p cond = form->args[0];
      TFormula_p not_cond = TFormulaNegate(cond, terms);

      TFormula_p true_part = TermTopAlloc(terms->sig->or_code, 2);
      true_part->args[0] = not_cond;
      true_part->args[1] = form->args[1];

      TFormula_p false_part = TermTopAlloc(terms->sig->or_code, 2);
      false_part->args[0] = cond;
      false_part->args[1] = form->args[2];

      true_part = TBTermTopInsert(terms, true_part);
      false_part = TBTermTopInsert(terms, false_part);

      TFormula_p unrolled =
          TFormulaFCodeAlloc(terms, terms->sig->and_code,
                             true_part,
                             false_part);

      /* printf("# ITE-Form case: "); */
      /* TermPrint(stdout, safe, terms->sig, DEREF_NEVER); */
      /* printf("\n# =>             "); */
      /* TermPrint(stdout, form, terms->sig, DEREF_NEVER); */
      /* printf("\n"); */
      form = do_ite_unroll(TermMap(terms, unrolled, FlattenApps_driver),
                           terms);
   }
   else if (TFormulaIsLiteral(terms->sig, form))
   {
      TermPos_p pos = PStackAlloc();
      PStackPushP(pos, form);
      PStackPushInt(pos, 0);
      if (form->args[0]->f_code != SIG_ITE_CODE
          && !TermFindIteSubterm(form->args[0], pos))
      {
         PStackDiscardTop(pos);
         PStackPushInt(pos, 1);
         if (form->args[1]->f_code != SIG_ITE_CODE
             && !TermFindIteSubterm(form->args[1], pos))
         {
            PStackReset(pos);
         }
      }

      if (!PStackEmpty(pos))
      {
         TFormula_p ite_term =
             ((Term_p)PStackBelowTopP(pos))->args[PStackTopInt(pos)];
         assert(ite_term->f_code == SIG_ITE_CODE);

         Term_p repl_t = TBTermPosReplace(terms, ite_term->args[1], pos,
                                          DEREF_NEVER, 0, ite_term);

         /* printf("# ite_term: "); */
         /* TermPrint(stdout, ite_term, terms->sig, DEREF_NEVER); */
         /* printf("\n"); */
         /* printf("# ite_term->args[0]: "); */
         /* TermPrint(stdout, ite_term->args[0], terms->sig, DEREF_NEVER); */
         /* printf("\n"); */
         /* printf("# ite_term->args[1]: "); */
         /* TermPrint(stdout, ite_term->args[1], terms->sig, DEREF_NEVER); */
         /* printf("\n"); */
         /* printf("# ite_term->args[2]: "); */
         /* TermPrint(stdout, ite_term->args[2], terms->sig, DEREF_NEVER); */
         /* printf("\n"); */
         Term_p repl_f = TBTermPosReplace(terms, ite_term->args[2], pos,
                                          DEREF_NEVER, 0, ite_term);

         TFormula_p cond = ite_term->args[0];
         TFormula_p neg_cond = TFormulaNegate(cond, terms);

         TFormula_p if_true_impl =
            TFormulaFCodeAlloc(terms, terms->sig->or_code,
                               neg_cond, TermMap(terms, repl_t, FlattenApps_driver));
         TFormula_p if_false_impl =
            TFormulaFCodeAlloc(terms, terms->sig->or_code,
                               cond, TermMap(terms, repl_f, FlattenApps_driver));

         // the whole formula
         form = TFormulaFCodeAlloc(terms, terms->sig->and_code,
                                   do_ite_unroll(if_true_impl, terms),
                                   do_ite_unroll(if_false_impl, terms));
         /* printf("# ite_term: "); */
         /* TermPrint(stdout, ite_term, terms->sig, DEREF_NEVER); */
         /* printf("\n"); */
         /* printf("# ITE-Term case: "); */
         /* TermPrint(stdout, safe, terms->sig, DEREF_NEVER); */
         /* printf("\n# =>             "); */
         /* TermPrint(stdout, form, terms->sig, DEREF_NEVER); */
         /* printf("\n"); */
         /* printf("# repl_t: "); */
         /* TermPrint(stdout, repl_t, terms->sig, DEREF_NEVER); */
         /* printf("\n# repl_f: "); */
         /* TermPrint(stdout, repl_f, terms->sig, DEREF_NEVER); */
         /* printf("\n"); */
      }
      PStackFree(pos);
   }
   else
   {
      Term_p new = TermTopCopyWithoutArgs(form);
      bool changed = false;
      for (long i = 0; i < new->arity; i++)
      {
         new->args[i] = do_ite_unroll(form->args[i], terms);
         changed = changed || new->args[i] != form->args[i];
      }

      if (changed)
      {
         form = TBTermTopInsert(terms, new);
      }
      else
      {
         TermTopFree(new);
      }
   }
   /* printf("# do_ite_unroll: "); */
   /* TermPrint(stdout, safe, terms->sig, DEREF_NEVER); */
   /* printf("\n# =>             "); */
   /* TermPrint(stdout, form, terms->sig, DEREF_NEVER); */
   /* printf("\n"); */

   return form;
}

/*-----------------------------------------------------------------------
//
// Function: do_bool_eqn_replace1 ()
//
//   Replace boolean equations with equivalences. Goes inside literals
//   as well. For example, "f(a, p = q) = b" will be translated to
//   "f(a, p <=> q) = b".
//
// Global Variables: -
//
// Side Effects    : Changes enclosed formula
//
/----------------------------------------------------------------------*/

TFormula_p do_bool_eqn_replace1(TFormula_p form, TB_p terms)
{
   const Sig_p sig = terms->sig;
   assert(sig);
   bool changed = false;

   if (TermIsDBVar(form) || !TermHasEqNeq(form))
   {
      return form;
   }

   if ((form->f_code == sig->eqn_code || form->f_code == sig->neqn_code) && form->arity == 2)
   {
      if (!TermIsAnyVar(form->args[0]) && !TermIsAnyVar(form->args[1]) &&
          SigIsLogicalSymbol(terms->sig, form->args[0]->f_code) &&
          SigIsLogicalSymbol(terms->sig, form->args[1]->f_code) &&
          TypeIsBool(form->args[0]) &&
          form->args[1] != terms->true_term)
      {
         // DAS literal is encoded as <predicate> = TRUE.
         // Our boolean equalities are <formula> = <formula>
         form = TFormulaFCodeAlloc(terms,
                                   (form->f_code == terms->sig->eqn_code ?
                                    terms->sig->equiv_code : terms->sig->xor_code),
                                   do_bool_eqn_replace1(form->args[0], terms),
                                   do_bool_eqn_replace1(form->args[1], terms));
         changed = true;
      }
   }
   if (!TermIsAnyVar(form) && !changed)
   {
      TFormula_p tmp = TermTopAlloc(form->f_code, form->arity);
      tmp->type = form->type;
      for (int i = 0; i < form->arity; i++)
      {
         tmp->args[i] = do_bool_eqn_replace1(form->args[i], terms);
      }
      form = TBTermTopInsert(terms, tmp);
   }
   return form;
}

/*-----------------------------------------------------------------------
//
// Function: do_bool_eqn_replace()
//
//   Replace boolean equations with equivalences. Goes inside literals
//   as well. For example, "f(a, p = q) = b" will be translated to
//   "f(a, p <=> q) = b".
//
//   We don't want to lift "true" atoms, but we do want to lift
//   proper Boolean formulas. So with t as a non-logical term, f as a
//   non-trivial formula:
//   eq(t, $true) => eq(t, $true)
//   eq(f, $true) => f
//
//   We also don't want to lift equations of the form (n)eq(f, var) (why
//   not?)
//
// Global Variables: -
//
// Side Effects    : Changes enclosed formula
//
/----------------------------------------------------------------------*/

TFormula_p do_bool_eqn_replace(TFormula_p form, TB_p terms)
{
   const Sig_p sig = terms->sig;
   //Term_p orig = form;
   assert(sig);

   /* printf("# do_bool_eqn_replace %p: ", form); */
   /* TermPrintDbg(stdout, form, terms->sig, DEREF_NEVER); */
   /* printf("\n"); */


   if (TermIsDBVar(form) || !TermHasEqNeq(form) || TermIsAnyVar(form))
   {
      //printf("# exit %p\n", form);
      return form;
   }

   TFormula_p tmp = TermTopAlloc(form->f_code, form->arity);
   tmp->type = form->type;
   for (int i = 0; i < form->arity; i++)
   {
      tmp->args[i] = do_bool_eqn_replace(form->args[i], terms);
   }
   form = TBTermTopInsert(terms, tmp);

   /* printf("# returned from recursion %p: ", orig); */
   /* TermPrintDbg(stdout, form, terms->sig, DEREF_NEVER); */
   /* printf("\n"); */

   if ((form->f_code == sig->eqn_code) && form->arity == 2)
   {  // Case $eqn(t1, t2) (may be terms, may be formulas)
      //printf("# case $eqn %p\n", orig);
      if (TFormulaIsComplexBool(terms->sig, form->args[0]) &&
          TFormulaIsComplexBool(terms->sig, form->args[1]))
      { // Case $eqn(f1,f2) (two proper formulas)
         if(form->args[1] != terms->true_term)
         { // Case $eqn(f1,f2) and f2 is not $true
            // DAS literal is encoded as <predicate> = $true
            // Our boolean equalities are <formula> = <formula>
            // What is DAS?!? (StS)
            form = TFormulaFCodeAlloc(terms,
                                      terms->sig->equiv_code,
                                      form->args[0],
                                      form->args[1]);
         }
         else if(form->args[0] != terms->true_term)
         { // It's $eqn(f1, $true) and f1 is a complex formula
            form = form->args[0];
         }
      }
   }
   else if ((form->f_code == sig->neqn_code) && form->arity == 2)
   { // Case $neqn(t1, t2) (may be terms, may be formulas)
      if (TFormulaIsComplexBool(terms->sig, form->args[0]) &&
          TFormulaIsComplexBool(terms->sig, form->args[1]))
      { // Case $neqn(f1,f2) (two proper formulas)
         if(form->args[1] != terms->true_term)
         {  // Case $neqn(f1,f2) and f2 is not $true
            form = TFormulaFCodeAlloc(terms,
                                      terms->sig->xor_code,
                                      form->args[0],
                                      form->args[1]);
         }
         else if(form->args[0] != terms->true_term)
         { // It's $neqn(f1, $true) and f1 is a complex formula
            form = TFormulaFCodeAlloc(terms,
                                      terms->sig->not_code,
                                      form->args[0],
                                      NULL);
         }
      }
   }
   return form;
}


/*---------------------------------------------------------------------*/
/*                         Exported Functions                          */
/*---------------------------------------------------------------------*/

/*-----------------------------------------------------------------------
//
// Function: TformulaCollectClause()
//
//   Given a term-encoded formula that is a disjunction of literals,
//   transform it into a clause.
//
// Global Variables: -
//
// Side Effects    : Same as in TFormulaConjunctiveToCNF() below.
//
/----------------------------------------------------------------------*/

/* Clause_p TformulaCollectClause(TFormula_p form, TB_p terms, */
/*                                VarBank_p fresh_vars) */
/* { */
/*    Clause_p res; */
/*    Eqn_p lit_list = NULL, tmp_list = NULL, lit; */
/*    PStack_p stack, lit_stack; */
/*    Subst_p  normsubst = SubstAlloc(); */

/*    /\*printf("tformula_collect_clause(): "); */
/*      TFormulaTPTPPrint(GlobalOut, terms, form, true); */
/*      printf("\n"); *\/ */

/*    litstack = PStackAlloc(); */
/*    stack = PStackAlloc(); */

/*    PStackPushP(stack, form); */
/*    while(!PStackEmpty(stack)) */
/*    { */
/*       form = PStackPopP(stack); */
/*       if(form->f_code == terms->sig->or_code) */
/*       { */
/*          PStackPushP(stack, form->args[0]); */
/*          PStackPushP(stack, form->args[1]); */
/*       } */
/*       else */
/*       { */
/*          assert(TFormulaIsLiteral(terms->sig, form)); */
/*          lit = EqnTBTermDecode(terms, form); */
/*          PStackPushP(lit_stack, lit); */

/*       } */
/*    } */
/*    PStackFree(stack); */
/*    while(!PStackEmpty(lit_stack)) */
/*    { */
/*       lit = PStackPopP(lit_stack); */
/*       EqnListInsertFirst(&lit_list, lit); */
/*    } */
/*    PStackFree(lit_stack); */

/*    VarBankResetVCounts(fresh_vars); */
/*    NormSubstEqnList(lit_list, normsubst, fresh_vars); */
/*    tmp_list = EqnListCopy(lit_list, terms); */
/*    res = ClauseAlloc(tmp_list); */
/*    EqnListFree(lit_list); /\* Created just for this *\/ */
/*    SubstDelete(normsubst); */
/*    return res; */
/* } */

/*-----------------------------------------------------------------------
//
// Function: WFormulaConjectureNegate()
//
//   If formula is a conjecture, negate it and delete that property
//   (but set WPInitialConjecture). Returns true if formula was a
//   conjecture.
//
// Global Variables: -
//
// Side Effects    : Changes formula
//
/----------------------------------------------------------------------*/

bool WFormulaConjectureNegate(WFormula_p wform)
{
   FormulaProperties ftype = FormulaQueryType(wform);

   if (ftype == CPTypeConjecture)
   {
      wform->tformula = TFormulaFCodeAlloc(wform->terms,
                                           wform->terms->sig->not_code,
                                           wform->tformula,
                                           NULL);
      FormulaSetType(wform, CPTypeNegConjecture);
      DocFormulaModificationDefault(wform, inf_neg_conjecture);
      WFormulaPushDerivation(wform, DCNegateConjecture, NULL, NULL);
      return true;
   }
   return false;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaAnnotateQuestion()
//
//   Take a formula of the form ((\exists X)*.F) and convert it to
//   ((\exists Xi)*.(F&~$answer(skn(X1,...Xn))), i.e. add an answer
//   literal encoding all leading existentially quantified variables.
//
// Global Variables: -
//
// Side Effects    : Memory operations.
//
/----------------------------------------------------------------------*/

TFormula_p TFormulaAnnotateQuestion(TB_p terms,
                                    TFormula_p form,
                                    NumTree_p *question_assoc)
{
   TFormula_p res, handle, tmp;
   PStack_p varstack = PStackAlloc();

   handle = form;
   while(handle->f_code == terms->sig->qex_code)
   {
      PStackPushP(varstack, handle->args[0]);
      handle = handle->args[1];
   }
   if(PStackEmpty(varstack))
   {
      /* Not a "real" question, nothing to do */
      res = form;
   }
   else
   {
      tmp = answer_lit_alloc(terms, varstack);
      res = TFormulaFCodeAlloc(terms, terms->sig->and_code, handle, tmp);
      while(!PStackEmpty(varstack))
      {
         handle = PStackPopP(varstack);
         res = TFormulaFCodeAlloc(terms, terms->sig->qex_code, handle, res);
      }
   }
   PStackFree(varstack);
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: WFormulaAnnotateQuestion()
//
//   If formula is a question, convert it into the equivalent
//   conjecture with answer annotation. Returns true if formula was a
//   question. Add the association of the new skolem symbol in the
//   answer literal to the clause id.
//
// Global Variables: -
//
// Side Effects    : Changes formula
//
/----------------------------------------------------------------------*/

bool WFormulaAnnotateQuestion(WFormula_p wform, bool add_answer_lits,
                              bool conjectures_are_questions,
                              NumTree_p *question_assoc)
{
   if (FormulaQueryProp(wform, CPTypeQuestion) ||
       (FormulaQueryProp(wform, CPTypeConjecture) && conjectures_are_questions))
   {
      if (add_answer_lits)
      {
         wform->tformula = TFormulaAnnotateQuestion(wform->terms,
                                                    wform->tformula,
                                                    question_assoc);
      }
      FormulaSetType(wform, CPTypeConjecture);
      DocFormulaModificationDefault(wform, inf_annotate_question);
      WFormulaPushDerivation(wform, DCAnnoQuestion, NULL, NULL);
      return true;
   }
   return false;
}

/*-----------------------------------------------------------------------
//
// Function: FormulaSetPreprocConjectures()
//
//   Negate all conjectures to make the implication to prove into an
//   formula set that is inconsistent if the implication is true. Note
//   that multiple conjectures are implicitely disjunctively
//   connected! Returns number of conjectures.
//
// Global Variables: -
//
// Side Effects    : Changes formula, may print warning if number of
//                   conjectures is different from 1.
//
/----------------------------------------------------------------------*/

long FormulaSetPreprocConjectures(FormulaSet_p set,
                                  FormulaSet_p archive,
                                  bool add_answer_lits,
                                  bool conjectures_are_questions)
{
   long res = 0;
   WFormula_p handle;

   handle = set->anchor->succ;

   // fprintf(stderr, "sine formulas: ");
   while (handle != set->anchor)
   {
      // DBG_PRINT(stderr, WFormulaGetId(handle), DerivationDebugPrint(stderr, handle->derivation), "\n");
      WFormulaAnnotateQuestion(handle, add_answer_lits,
                               conjectures_are_questions,
                               NULL);

      if (WFormulaConjectureNegate(handle))
      {
         res++;
      }
      handle = handle->succ;
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: WFormulaSimplify()
//
//   Apply standard simplifications to the wrapped formula. Return
//   true if the formula has changed.
//
// Global Variables: -
//
// Side Effects    : Memory operations.
//
/----------------------------------------------------------------------*/

bool WFormulaSimplify(WFormula_p form, TB_p terms)
{
   TFormula_p simplified;
   bool res = false;

   //assert(!terms->freevarsets);
   simplified = TFormulaSimplify(terms, form->tformula, 0);
   // TBVarSetStoreFree(terms);

   if (simplified != form->tformula)
   {
      form->tformula = simplified;
      DocFormulaModificationDefault(form, inf_fof_simpl);
      WFormulaPushDerivation(form, DCFofSimplify, NULL, NULL);
      res = true;
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: WFormulaCNF()
//
//   Transform the formula of a wrapped formula into CNF.
//
// Global Variables: -
//
// Side Effects    : Changes formula, memory operations
//
/----------------------------------------------------------------------*/

long WFormulaCNF(WFormula_p form, ClauseSet_p set,
                 TB_p terms, VarBank_p fresh_vars)
{
   if (form->is_clause)
   {
      Clause_p clause = WFormClauseToClause(form);
      ClausePushDerivation(clause, DCFofQuote, form, NULL);
      ClauseSetInsert(set, clause);
      return 1;
   }
   WTFormulaConjunctiveNF(form, terms);
   return TFormulaToCNF(form, FormulaQueryType(form),
                        set, terms, fresh_vars);
}

/*-----------------------------------------------------------------------
//
// Function: WFormulaCNF2()
//
//   Transform the formula of a wrapped formula into CNF.
//
// Global Variables: -
//
// Side Effects    : Changes formula, memory operations
//
/----------------------------------------------------------------------*/

long WFormulaCNF2(WFormula_p form, ClauseSet_p set,
                  TB_p terms, VarBank_p fresh_vars,
                  long miniscope_limit,
                  bool fool_unroll)
{
   DBGTermCheckUnownedSubterm(stdout, form->tformula, "WFormulaCNF2_0");
   form->tformula = LambdaNormalizeDB(terms, form->tformula);
   if (form->is_clause)
   {
      Clause_p clause = WFormClauseToClause(form);
      ClausePushDerivation(clause, DCFofQuote, form, NULL);
      if(problemType == PROBLEM_HO)
      {
         EqnListMapTerms(clause->literals,
            (TermMapper_p)PostCNFEncodeFormulas, terms);
      }
      ClauseSetInsert(set, clause);

      return 1;
   }
   WTFormulaConjunctiveNF3(form, terms, miniscope_limit, fool_unroll);
   return TFormulaToCNF(form, FormulaQueryType(form),
                        set, terms, fresh_vars);
}

/*-----------------------------------------------------------------------
//
// Function: FormulaSetSimplify()
//
//   Apply standard FOF simplification rules to all formulae in the
//   set. Returns number of changed formulas.
//
// Global Variables: -
//
// Side Effects    : Changes formulas, output of inference steps.
//
/----------------------------------------------------------------------*/

long FormulaSetSimplify(FormulaSet_p set, TB_p terms, bool do_garbage_collect)
{
   WFormula_p handle;
   long res = 0;
   long old_nodes = TBNonVarTermNodes(terms);
   long gc_threshold = old_nodes * TFORMULA_GC_LIMIT;
   bool changed;

   handle = set->anchor->succ;
   while (handle != set->anchor)
   {
      //printf("Simplifying: \n");
      //WFormulaPrint(stdout, handle, true);
      //printf("\n");
      changed = WFormulaSimplify(handle, terms);
      //printf("Simplified %d\n", changed);
      //WFormulaPrint(stdout, handle, true);
      //printf("\n");
      if (changed)
      {
         res++;
         if (do_garbage_collect && TBNonVarTermNodes(terms) > gc_threshold)
         {
            assert(terms == handle->terms);
            TBGCCollect(terms);
            old_nodes = TBNonVarTermNodes(terms);
            gc_threshold = old_nodes * TFORMULA_GC_LIMIT;
         }
      }
      handle = handle->succ;
   }
   // printf("All simplified\n");
   if (do_garbage_collect && TBNonVarTermNodes(terms) != old_nodes)
   {
      TBGCCollect(terms);
   }
   // printf("Garbage collected\n");
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: FormulaSetCNF()
//
//   Transform all formulae in set into CNF. Return number of clauses
//   generated.
//
// Global Variables: -
//
// Side Effects    : Plenty of memory stuff.
//
/----------------------------------------------------------------------*/

long FormulaSetCNF(FormulaSet_p set, FormulaSet_p archive,
                   ClauseSet_p clauseset, TB_p terms,
                   VarBank_p fresh_vars, long def_limit)
{
   WFormula_p form, handle;
   long res = 0;
   long old_nodes = TBNonVarTermNodes(terms);
   long gc_threshold = old_nodes * TFORMULA_GC_LIMIT;

   FormulaSetSimplify(set, terms, true);
   // printf("FormulaSetSimplify done\n");
   TFormulaSetIntroduceDefs(set, archive, terms, def_limit);
   // printf("Definitions introduced\n");

   while (!FormulaSetEmpty(set))
   {
      handle = FormulaSetExtractFirst(set);
      // WFormulaPrint(stdout, handle, true);
      // fprintf(stdout, "\n");
      form = WFormulaFlatCopy(handle);
      FormulaSetInsert(archive, handle);
      WFormulaPushDerivation(form, DCFofQuote, handle, NULL);
      handle = form;
      res += WFormulaCNF(handle, clauseset, terms, fresh_vars);
      FormulaSetInsert(archive, handle);
      if (handle->tformula &&
          (TBNonVarTermNodes(terms) > gc_threshold))
      {
         assert(terms == handle->terms);
         TBGCCollect(terms);
         old_nodes = TBNonVarTermNodes(terms);
         gc_threshold = old_nodes * TFORMULA_GC_LIMIT;
      }
   }
   if (TBNonVarTermNodes(terms) != old_nodes)
   {
      TBGCCollect(terms);
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: FormulaSetCNF2()
//
//   Transform all formulae in set into CNF. Return number of clauses
//   generated.
//
// Global Variables: -
//
// Side Effects    : Plenty of memory stuff.
//
/----------------------------------------------------------------------*/

long FormulaSetCNF2(FormulaSet_p set, FormulaSet_p archive,
                    ClauseSet_p clauseset, TB_p terms,
                    VarBank_p fresh_vars,
                    long miniscope_limit,
                    long def_limit,
                    bool lift_lambdas,
                    bool lambda_to_forall,
                    bool unfold_only_forms,
                    bool fool_unroll)
{
   WFormula_p form, handle;
   long res = 0;
   long old_nodes = TBNonVarTermNodes(terms);
   long gc_threshold = old_nodes * TFORMULA_GC_LIMIT;
#ifdef ENABLE_LFHO
   if(problemType == PROBLEM_HO)
   {
      TFormulaSetNamedToDBLambdas(set, archive, terms);
      TFormulaSetLiftItes(set, archive, terms);
      TFormulaSetLiftLets(set, archive, terms);
      TFormulaSetUnfoldDefSymbols(set, archive, terms, unfold_only_forms);
      if(lambda_to_forall)
      {
         TFormulaSetLambdaNormalize(set, archive, terms);
      }
   }
#endif
   if(fool_unroll)
   {
      TFormulaSetUnrollFOOL(set, archive, terms);
   }
   FormulaSetSimplify(set, terms, true);

   //printf("# Introducing definitions\n");
   TFormulaSetIntroduceDefs(set, archive, terms, def_limit);
   //printf("# Definitions introduced\n");
   while (!FormulaSetEmpty(set))
   {
      handle = FormulaSetExtractFirst(set);
      DBGTermCheckUnownedSubterm(stdout, handle->tformula, "UnownedCNF");
      form = WFormulaFlatCopy(handle);
      FormulaSetInsert(archive, handle);
      WFormulaPushDerivation(form, DCFofQuote, handle, NULL);
      handle = form;
      res += WFormulaCNF2(handle, clauseset, terms, fresh_vars,
                          miniscope_limit, fool_unroll);
      FormulaSetInsert(archive, handle);
      if (handle->tformula &&
          (TBNonVarTermNodes(terms) > gc_threshold))
      {
         assert(terms == handle->terms);
         TBGCCollect(terms);
         old_nodes = TBNonVarTermNodes(terms);
         gc_threshold = old_nodes * TFORMULA_GC_LIMIT;
      }
   }
   if(lift_lambdas)
   {
      ClauseSetLiftLambdas(clauseset, archive, terms, fresh_vars, fool_unroll);
   }
   if (TBNonVarTermNodes(terms) != old_nodes)
   {
      TBGCCollect(terms);
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: FormulaAndClauseSetParse()
//
//   Parse a mixture of clauses and formulas (if the syntax supports
//   it). Return number of elements parsed (even if discarded by
//   filter). Watch list clauses are parsed as clauses in wlset,
//   everything else (even clauses) is parsed as a formula and put
//   into fset.
//
// Global Variables: -
//
// Side Effects    : Input, changes termbank and sets.
//
/----------------------------------------------------------------------*/

long FormulaAndClauseSetParse(Scanner_p in, FormulaSet_p fset,
                              ClauseSet_p wlset, TB_p terms,
                              StrTree_p *name_selector,
                              StrTree_p *skip_includes)
{
   long res = 0;
   WFormula_p form, nextform;
   Clause_p clause, nextclause;
   StrTree_p stand_in = NULL;

   if(!name_selector)
   {
      name_selector = &stand_in;
   }

   switch(ScannerGetFormat(in))
   {
   case LOPFormat:
      //* LOP does not at the moment support full FOF, or inline watchlists */
      SetProblemType(PROBLEM_FO);
      while(ClauseStartsMaybe(in))
      {
         form = WFormClauseParse(in, terms);
         // fprintf(stdout, "Parsed: ");
         // WFormulaPrint(stdout, form, true);
         // fprintf(stdout, "\n");
         FormulaSetInsert(fset, form);
         res++;
      }
      break;
   default:
#ifndef ENABLE_LFHO
         if(TestInpId(in, "thf"))
         {
            Error("To support HOL reasoning, recompile E"
                  " using \'./configure --enable-ho && make rebuild\' \n",
                  SYNTAX_ERROR);
         }
#endif
         while(TestInpId(in, "input_formula|input_clause|fof|cnf|tff|thf|tcf|include"))
         {
            if(TestInpId(in, "include"))
            {
               if (app_encode)
               {
                  ignore_include(in);
                  continue;
               }

               StrTree_p new_limit = NULL;
               Scanner_p new_in;
               FormulaSet_p nfset = FormulaSetAlloc();
               ClauseSet_p nwlset = ClauseSetAlloc();
               new_in = ScannerParseInclude(in, &new_limit, skip_includes);

               if (new_in)
               {
                  res += FormulaAndClauseSetParse(new_in,
                                                  nfset,
                                                  nwlset,
                                                  terms,
                                                  &new_limit,
                                                  skip_includes);
                  DestroyScanner(new_in);
               }
               StrTreeFree(new_limit);
               FormulaSetInsertSet(fset, nfset);
               ClauseSetInsertSet(wlset, nwlset);
               assert(ClauseSetEmpty(nfset));
               assert(ClauseSetEmpty(nwlset));
               FormulaSetFree(nfset);
               ClauseSetFree(nwlset);
            }
            else
            {
               if(TestInpId(in, "input_formula|fof|tff|thf|tcf"))
               {
                  if(TestInpId(in, "tff|thf|tcf"))
                  {
                     terms->sig->typed_symbols = true;
                  }
                  //printf("It's a formula\n");
                  form = WFormulaParse(in, terms);
                  //fprintf(stdout, "Parsed: ");
                  //WFormulaPrint(stdout, form, true);
               }
               else
               {
                  assert(TestInpId(in, "input_clause|cnf"));
                  //clause = ClauseParse(in, terms);
                  //ClauseSetInsert(cset, clause);
                  SetProblemType(PROBLEM_FO);
                  form = WFormClauseParse(in, terms);
               }
               if(FormulaQueryType(form) == CPTypeWatchClause)
               {
                  assert(form->is_clause);
                  clause = WFormClauseToClause(form);
                  ClauseSetInsert(wlset, clause);
                  WFormulaFree(form);
               }
               else
               {
                  FormulaSetInsert(fset, form);
               }
               res++;
            }
         }
         break;
   }
   if(*name_selector)
   {
      form = fset->anchor->succ;
      while(form != fset->anchor)
      {
         nextform = form->succ;
         if(!verify_name(name_selector, form->info))
         {
            FormulaSetDeleteEntry(form);
         }
         form = nextform;
      }
      clause = wlset->anchor->succ;
      while(clause != wlset->anchor)
      {
         nextclause = clause->succ;
         if(!verify_name(name_selector, clause->info))
         {
            ClauseSetDeleteEntry(clause);
         }
         clause = nextclause;
      }
      check_all_found(in, *name_selector);
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaToCNF()
//
//   Convert a term-encoded formula from conjunctive normal form into
//   a set of (variable-normalized) clauses. Return number of clauses
//   generated.
//
// Global Variables: -
//
// Side Effects    : Memory operations, may allocate new variables in
//                   fresh_vars, may create new terms in the term
//                   bank.
//
/----------------------------------------------------------------------*/

long TFormulaToCNF(WFormula_p form, FormulaProperties type, ClauseSet_p set,
                   TB_p terms, VarBank_p fresh_vars)
{
   TFormula_p handle;
   long old_clause_number = set->members;
   PStack_p stack = PStackAlloc();
   Clause_p clause;

   /* Skip quantors */
   for (handle = form->tformula;
        handle->f_code == terms->sig->qall_code;
        handle = handle->args[1])
   {
      assert(handle);
   }
   PStackPushP(stack, handle);
   while (!PStackEmpty(stack))
   {
      handle = PStackPopP(stack);
      if (handle->f_code == terms->sig->and_code)
      {
         PStackPushP(stack, handle->args[0]);
         PStackPushP(stack, handle->args[1]);
      }
      else
      {
         clause = TFormulaCollectClause(handle, terms, fresh_vars);
         ClauseSetTPTPType(clause, type);
         DocClauseFromForm(GlobalOut, OutputLevel, clause, form);
         ClausePushDerivation(clause, DCSplitConjunct, form, NULL);

         if (ClauseEliminateNakedBooleanVariables(clause))
         {
            ClausePushDerivation(clause, DCEliminateBVar, NULL, NULL);
         }

         if(problemType == PROBLEM_HO)
         {
            EqnListMapTerms(clause->literals, (TermMapper_p)PostCNFEncodeFormulas, terms);
         }

         ClauseSetInsert(set, clause);
      }
   }
   PStackFree(stack);
   return set->members - old_clause_number;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetDelTermpProp()
//
//   Go through a set of term-encoded formulas and delete prop in all
//   term and formula cells.
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/

void TFormulaSetDelTermpProp(FormulaSet_p set, TermProperties prop)
{
   WFormula_p handle;

   for (handle = set->anchor->succ; handle != set->anchor; handle = handle->succ)
   {
      if (handle->tformula)
      {
         TermDelProp(handle->tformula, DEREF_NEVER, prop);
      }
   }
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetFindDefs()
//
//   Go through a set of formulas and generate and record all
//   necessary definitions. Assumes that the formulas are simplified!
//
// Global Variables: -
//
// Side Effects    : Memory operations
//
/----------------------------------------------------------------------*/

void TFormulaSetFindDefs(FormulaSet_p set, TB_p terms, NumXTree_p *defs,
                         PStack_p renamed_forms, long formula_def_limit)
{
   WFormula_p handle;

   // printf("TFormulaSetFindDefs()...\n");
   for (handle = set->anchor->succ; handle != set->anchor; handle = handle->succ)
   {
      assert(handle->tformula);

      if (handle->tformula && !handle->is_clause && formula_def_limit)
      {
         TFormulaFindDefs(terms, handle->tformula, 1,
                          formula_def_limit, defs, renamed_forms);
      }
   }
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaApplyDefs()
//
//   Given a formula and a number of definitions represented by defs
//   and tags in bank, apply all apropriate definitions to simplify
//   the formula. Return the number of definitions used. Note that
//   defs has to contain the defined atoms in val2 and the ident of
//   the corresponding definition in val1 of its cells.
//
// Global Variables: -
//
// Side Effects    : Simplifies set, may print simplification steps.
//
/----------------------------------------------------------------------*/

long TFormulaApplyDefs(WFormula_p form, TB_p terms, NumXTree_p *defs)
{
   TFormula_p reduced;
   long res = 0;
   PStack_p defs_used = PStackAlloc();
   PStackPointer i;

   reduced = TFormulaCopyDef(terms, form->tformula, form->ident,
                             defs, defs_used);
   if (!PStackEmpty(defs_used))
   {
      assert(form->tformula != reduced);
      form->tformula = reduced; /* Old one will be picked up by gc */
      DocFormulaIntroDefsDefault(form, defs_used);
      res = PStackGetSP(defs_used);
      for (i = 0; i < res; i++)
      {
         WFormulaPushDerivation(form,
                                DCApplyDef,
                                PStackElementP(defs_used, i),
                                NULL);
      }
   }
   else
   {
      assert(form->tformula == reduced);
   }

   PStackFree(defs_used);
   /* if(TFormulaHasFreeVars(terms, form->tformula)) */
   /* { */
   /*    printf("# ApplyDefs Free Variable: "); */
   /*    WFormulaPrint(stdout, form, true); */
   /*    printf("\n"); */
   /* } */
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaUnrollFOOL()
//
//   Translate FOOL features into FOL. Performs following translations:
//      - Takes formulas as arguments out of the term, leaving
//        only $true, $false and boolean vars as the argument of the term
//      - TODO: Unfolds ite expressions used as terms
//      - TODO: Unfolds ite expressions used as formulas
//
// Global Variables: -
//
// Side Effects    : Changes enclosed formulas and proof objects
//
/----------------------------------------------------------------------*/

bool TFormulaUnrollFOOL(WFormula_p form, TB_p terms)
{
   /* Term_p free_var; */
   /* if((free_var = TFormulaHasFreeVars(terms, form->tformula))) */
   /* { */
   /*    printf("# Before UnrollFOOL Free Variable "); */
   /*    TermPrintSimple(stdout, free_var, terms->sig); */
   /*    printf(" :"); */
   /*    WFormulaPrint(stdout, form, true); */
   /*    printf("\n"); */
   /*    TermPrettyPrintSimple(stdout, form->tformula, form->terms->sig, 0); */
   /* } */
   /* printf("# Before UnrollFOOL: "); */
   /* WFormulaPrint(stdout, form, true); */
   /* printf("\n"); */
   /* TermPrettyPrintSimple(stdout, form->tformula, form->terms->sig, 0); */
   /* printf("\n"); */
   form->tformula = TFormulaExpandLiterals(terms, form->tformula);
   /* printf("# Expanded:\n# "); */
   /* WFormulaPrint(stdout, form, true); */
   /* printf("\n"); */
   /* TermPrettyPrintSimple(stdout, form->tformula, form->terms->sig, 0); */
   /* printf("\n"); */

   bool res = map_formula(form, terms, do_fool_unroll, DCFoolUnroll);

   /* if(TFormulaHasFreeVars(terms, form->tformula)) */
   /* { */
   /*    printf("# UnrollFOOL Free Variable: "); */
   /*    WFormulaPrint(stdout, form, true); */
   /*    printf("\n"); */
   /* } */
   /* printf("# Unrolled: \n# "); */
   /* WFormulaPrint(stdout, form, true); */
   /* printf("\n"); */
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaReplaceEqnWithEquiv()
//
//   If input formula contains subformulas of type \alpha = \beta,
//   replace those subformulas with \alpha <=> \beta and alter
//   proof object accordingly.
//
// Global Variables: -
//
// Side Effects    : Changes enclosed formulas and proof objects
//
/----------------------------------------------------------------------*/

bool TFormulaReplaceEqnWithEquiv(WFormula_p form, TB_p terms)
{
   bool res = map_formula(form, terms, do_bool_eqn_replace, DCEqToEq);

   /* if(TFormulaHasFreeVars(terms, form->tformula)) */
   /* { */
   /*    printf("# Eqn->Equiv Free Variable: "); */
   /*    WFormulaPrint(stdout, form, true); */
   /*    printf("\n"); */
   /* } */
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetUnrollFOOL()
//
//   Unrolls FOOL features for the set of formulas.
//
// Global Variables: -
//
// Side Effects    : Simplifies set, may print simplification steps.
//
/----------------------------------------------------------------------*/

long TFormulaSetUnrollFOOL(FormulaSet_p set, FormulaSet_p archive, TB_p terms)
{
   long res = 0;
   for (WFormula_p formula = set->anchor->succ;
        formula != set->anchor;
        formula = formula->succ)
   {
      /* printf("# Before Eqn2Equiv  %p: ", formula); */
      /* WFormulaTSTPPrintDeriv(stdout, formula); */
      /* printf("\n"); */
      /* TFormulaReplaceEqnWithEquiv(formula, terms); */
      /* printf("# Eqn2Equiv  %p: ", formula); */
      /* WFormulaTSTPPrintDeriv(stdout, formula); */
      /* printf("\n"); */
      /* printf("# As term     :"); */
      /* TermPrintDbg(stdout, formula->tformula, terms->sig, DEREF_NEVER); */
      /* printf("\n"); */
      if (TFormulaUnrollFOOL(formula, terms))
      {
         res++;
      }
      /* printf("# Foolunroll %p: ", formula); */
      /* WFormulaTSTPPrintDeriv(stdout, formula); */
      /* printf("\n"); */
      /* printf("# As term     :"); */
      /* TermPrintDbg(stdout, formula->tformula, terms->sig, DEREF_NEVER); */
      /* printf("\n"); */
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetLiftLets()
//
//    Rewrites all formulas so that all occurrences of the let symbols
//    are replaced by global definitions.
//
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/

long TFormulaSetLiftLets(FormulaSet_p set, FormulaSet_p archive, TB_p terms)
{
   long res = 0;

   PStack_p lifted_lets = PStackAlloc();

   for (WFormula_p form = set->anchor->succ; form != set->anchor; form = form->succ)
   {
      TFormula_p tform = form->tformula;
      VarBankSetVCountsToUsed(terms->vars);
      tform = TFormulaVarRename(terms, tform);
      PStackPointer i = PStackGetSP(lifted_lets);

      tform = lift_lets(terms, tform, lifted_lets);
      if (i != PStackGetSP(lifted_lets))
      {
         res++;

         form->tformula = unencode_eqns(terms, tform);
         for (; i < PStackGetSP(lifted_lets); i++)
         {
            TFormula_p def = PStackElementP(lifted_lets, i);
            WFormula_p wdef = WTFormulaAlloc(terms, def);
            WFormulaPushDerivation(wdef, DCIntroDef, NULL, NULL);
            WFormulaPushDerivation(form, DCApplyDef, wdef, NULL);
            PStackAssignP(lifted_lets, i, wdef);
         }
      }
      /* printf("# Let-lifted %p: ", form); */
      /* WFormulaTSTPPrintDeriv(stdout, form); */
      /* printf("\n"); */
   }

   while (!PStackEmpty(lifted_lets))
   {
      FormulaSetInsert(set, PStackPopP(lifted_lets));
   }

   PStackFree(lifted_lets);

   return res;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetLiftItes()
//
//    Rewrites all formulas so that all occurrences of the ite symbols
//    are replaced by appropriate implications
//
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/

long TFormulaSetLiftItes(FormulaSet_p set, FormulaSet_p archive, TB_p terms)
{
   long res = 0;
   for (WFormula_p formula = set->anchor->succ;
        formula != set->anchor;
        formula = formula->succ)
   {
      if (map_formula(formula, terms, do_ite_unroll, DCLiftIte))
      {
         res++;
         /* if(TFormulaHasFreeVars(terms, formula->tformula)) */
         /* { */
         /*    printf("# LiftItes Free Variable: "); */
         /*    WFormulaPrint(stdout, formula, true); */
         /*    printf("\n"); */
         /* } */
         /* printf("Ite-expanded %p: ", formula); */
         /* WFormulaTSTPPrintDeriv(stdout, formula); */
         /* printf("\n"); */
      }
   }
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetLambdaNormalize()
//
//   Beta normalizes the input problem and turns every equation
//   (^[X]:s) = t into ![X]: (s = (t @ X))
//
// Global Variables: -
//
// Side Effects    : Simplifies set, may print simplification steps.
//
/----------------------------------------------------------------------*/

long TFormulaSetLambdaNormalize(FormulaSet_p set, FormulaSet_p archive, TB_p terms)
{
   long res = 0;
   if (problemType == PROBLEM_HO)
   {
      for (WFormula_p form = set->anchor->succ; form != set->anchor; form = form->succ)
      {
         DBGTermCheckUnownedSubterm(stdout, form->tformula, "LambdaNormUnowned1");
         TFormula_p handle = LambdaToForall(terms, BetaNormalizeDB(terms, form->tformula));

         if (handle != form->tformula)
         {
            assert(!TermIsBetaReducible(handle));
            form->tformula = handle;
            DocFormulaModificationDefault(form, inf_fof_simpl);
            WFormulaPushDerivation(form, DCFofSimplify, NULL, NULL);
            DBGTermCheckUnownedSubterm(stdout, handle, "LambdaNormUnowned2");
            res++;
            /* if(TFormulaHasFreeVars(terms, form->tformula)) */
            /* { */
            /*    printf("# LambdaNormalize Free Variable: "); */
            /*    WFormulaPrint(stdout, form, true); */
            /*    printf("\n"); */
            /* } */
         }
      }
      return res;
   }
   else
   {
      return 0;
   }
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetUnfoldDefSymbols()
//
//    Rewrites all formulas using defined symbols of the form
//    sym = \vars. body where return type of sym is Bool
//
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/

long TFormulaSetUnfoldDefSymbols(FormulaSet_p set, FormulaSet_p archive,
                                 TB_p terms, bool unfold_only_forms)
{
   long res = 0;
   if (problemType == PROBLEM_HO)
   {
      VarBankSetVCountsToUsed(terms->vars);
      IntMap_p sym_def_map = IntMapAlloc();
      PTree_p def_wforms = create_sym_map(set, sym_def_map, unfold_only_forms);
      intersimplify_definitions(terms, sym_def_map);

      for (WFormula_p form = set->anchor->succ; form != set->anchor; form = form->succ)
      {
         if (!PTreeFind(&def_wforms, form))
         {
            PTree_p used_defs = NULL;
            int max_steps = MAX_RW_STEPS;
            TFormula_p handle = do_rw_with_defs(terms, form->tformula,
                                                sym_def_map, &used_defs,
                                                &max_steps, true);

            if (handle != form->tformula)
            {
               form->tformula = TermMap(terms, handle, unencode_eqns);

               DocFormulaModificationDefault(form, inf_fof_simpl);
               PStack_p ptiter = PTreeTraverseInit(used_defs);
               PTree_p node = NULL;
               while ((node = PTreeTraverseNext(ptiter)))
               {
                  WFormulaPushDerivation(form, DCApplyDef, node->key, NULL);
               }
               PTreeTraverseExit(ptiter);
               res++;
            }
            PTreeFree(used_defs);
            /* if(TFormulaHasFreeVars(terms, form->tformula)) */
            /* { */
            /*    printf("# UnfoldDef Free Variable: "); */
            /*    WFormulaPrint(stdout, form, true); */
            /*    printf("\n"); */
            /* } */
         }
      }

      IntMapIter_p iter = IntMapIterAlloc(sym_def_map, 0, LONG_MAX);
      WFormula_p next;
      long i;
      while ((next = IntMapIterNext(iter, &i)))
      {
         FormulaSetInsert(archive, next);
      }
      IntMapIterFree(iter);

      PStack_p titer = PTreeTraverseInit(def_wforms);
      PTree_p node;
      while ((node = PTreeTraverseNext(titer)))
      {
         next = node->key;
         FormulaSetExtractEntry(next);
         FormulaSetInsert(archive, next);
      }
      PTreeTraverseExit(titer);

      IntMapFree(sym_def_map);
      PTreeFree(def_wforms);

      return res;
   }
   else
   {
      return 0;
   }
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetLiftLambdas()
//
//    Lifts lambdas from the formula set. Inserts new definitons into set.
//
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/

static void deleter(void *to_delete)
{
   EqnFree(((ClausePos_p)to_delete)->literal);
}

static void insert_to_set(void *set, void *def)
{
   FormulaSetInsert(set, def);
}

long TFormulaSetLiftLambdas(FormulaSet_p set, FormulaSet_p archive, TB_p terms)
{
   long res = 0;
   if (problemType == PROBLEM_HO)
   {
      PStack_p defs = PStackAlloc();
      PTree_p all_defs = NULL;
      PDTree_p liftings = PDTreeAllocWDeleter(terms, deleter);
      for (WFormula_p form = set->anchor->succ; form != set->anchor; form = form->succ)
      {
         TFormula_p handle = LiftLambdas(terms, form->tformula, defs, liftings);
         if (handle != form->tformula)
         {
            form->tformula = handle;
            while (!(PStackEmpty(defs)))
            {
               WFormula_p def = PStackPopP(defs);
               WFormulaPushDerivation(form, DCApplyDef, def, NULL);
               PTreeStore(&all_defs, def);
               res++;
            }
         }
      }

      PTreeVisitInOrder(all_defs, insert_to_set, set);

      PStackFree(defs);
      PTreeFree(all_defs);
      PDTreeFree(liftings);
      return res;
   }
   else
   {
      return 0;
   }
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetNamedToDBLambdas()
//
//   Convert all lambdas in the proof state from named to
//   de Bruijn representation ()
//
//
// Global Variables: -
//
// Side Effects    : -
//
/----------------------------------------------------------------------*/
long TFormulaSetNamedToDBLambdas(FormulaSet_p set, FormulaSet_p archive, TB_p terms)
{
   long res = 0;
   if (problemType == PROBLEM_HO)
   {
      for (WFormula_p form = set->anchor->succ; form != set->anchor; form = form->succ)
      {
         TFormula_p handle = NamedToDB(terms, form->tformula);
         if (handle != form->tformula)
         {
            form->tformula = handle;
            DocFormulaModificationDefault(form, inf_fof_simpl);
            WFormulaPushDerivation(form, DCFofSimplify, NULL, NULL);
            res++;
            DBGTermCheckUnownedSubterm(stdout, handle, "UnownedToDBLambdas");
         }
      }
      return res;
   }
   else
   {
      return 0;
   }
}

/*-----------------------------------------------------------------------
//
// Function: TFormulaSetIntroduceDefs()
//
//   Transform a formula set by renaming certain subformulae and
//   adding the necessary definitions. Returns the number of
//   definitions.
//
//    Note that NumXTree cells are used as follows:
//    key is the term ident of the formula to be replaced
//    vals[0].i_val starts as the polarity of that formula, but turns
//                  into the id of the "virtual" definition used for
//                  output
//    vals[1].p_val is a pointer to the defined predicate term.
//    vals[2].i_val is the id of the real definition used to protect
//                  the definition to be applied to itself.
//    vals[3].p_val is a pointer to the polarity 0 definition
//
// Global Variables: -
//
// Side Effects    : Changes set.
//
/----------------------------------------------------------------------*/

long TFormulaSetIntroduceDefs(FormulaSet_p set, FormulaSet_p archive, TB_p terms, long limit)
{
   long res = 0;
   NumXTree_p defs = NULL, cell;
   PStack_p renamed_forms = PStackAlloc();
   PStackPointer i;
   TFormula_p form, def, newdef;
   long polarity;
   WFormula_p w_def, c_def, formula, arch_form;

   //printf("TFormulaSetIntroduceDefs()...\n");
   TFormulaSetDelTermpProp(set, TPCheckFlag | TPPosPolarity | TPNegPolarity);
   //printf("Deleted properties\n");
   FormulaSetMarkPolarity(set);
   //printf("Marked polarites\n");

   //printf("About to find defs\n");
   TFormulaSetFindDefs(set, terms, &defs, renamed_forms, limit);

   res = PStackGetSP(renamed_forms);
   //printf("About to Create defs\n");

   for (i = 0; i < PStackGetSP(renamed_forms); i++)
   {
      form = PStackElementP(renamed_forms, i);
      cell = NumXTreeFind(&defs, form->entry_no);
      assert(cell);
      polarity = TFormulaDecodePolarity(terms, form);
      def = cell->vals[1].p_val;
      newdef = TFormulaCreateDef(terms, def, form, 0);
      w_def = WTFormulaAlloc(terms, newdef);
      /* printf("# New definition: "); */
      /* WFormulaTSTPPrint(stdout, w_def, true, true); */
      /* printf("\n"); */

      DocFormulaCreationDefault(w_def, inf_fof_intro_def, NULL, NULL);
      cell->vals[0].i_val = w_def->ident; /* Replace polarity with
                                           * definition id */
      arch_form = WFormulaFlatCopy(w_def);
      WFormulaPushDerivation(arch_form, DCIntroDef, NULL, NULL);
      FormulaSetInsert(archive, arch_form);
      WFormulaPushDerivation(w_def, DCFofQuote, arch_form, NULL);

      cell->vals[3].p_val = arch_form;
      if (polarity == 0)
      {
         cell->vals[2].i_val = w_def->ident; /* ..and this is the
                                                blocking id of the
                                                actual definition.*/
         FormulaSetInsert(set, w_def);
      }
      else
      {
         newdef = TFormulaCreateDef(terms, def, form,
                                    polarity);
         c_def = WTFormulaAlloc(terms, newdef);
         DocFormulaCreationDefault(c_def, inf_fof_split_equiv, w_def, NULL);
         cell->vals[2].i_val = c_def->ident; /* ..and this is the
                                                blocking id of the actual
                                                definition.*/
         WFormulaPushDerivation(c_def, DCSplitEquiv, arch_form, NULL);
         FormulaSetInsert(set, c_def);
         WFormulaFree(w_def);
      }
   }
   PStackFree(renamed_forms);

   // printf("About to apply defs\n");
   for (formula = set->anchor->succ; formula != set->anchor; formula = formula->succ)
   {
      /* printf("# Before Def-appl %p: ", formula); */
      /* WFormulaTSTPPrintDeriv(stdout, formula); */
      /* printf("\n"); */
      TFormulaApplyDefs(formula, terms, &defs);
      /* printf("# After Def-appl  %p: ", formula); */
      /* WFormulaTSTPPrint(stdout, formula, true, true); */
      /* printf("\n"); */
      /* if(TFormulaHasFreeVars(terms, formula->tformula)) */
      /* { */
      /*    printf("# ApplyDefs Free Variable: "); */
      /*    WFormulaPrint(stdout, formula, true); */
      /*    printf("\n"); */
      /* } */
   }
   NumXTreeFree(defs);
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: FormulaSetArchive()
//
//   Move each formula from set to archive, replace it by a copy that
//   quoted the archived formula as the parent.
//
// Global Variables: -
//
// Side Effects    : Memory operations.
//
/----------------------------------------------------------------------*/

void FormulaSetArchive(FormulaSet_p set, FormulaSet_p archive)
{
   FormulaSet_p tmpset;
   WFormula_p handle, newform;

   tmpset = FormulaSetAlloc();

   while ((handle = FormulaSetExtractFirst(set)))
   {
      newform = WFormulaFlatCopy(handle);
      WFormulaPushDerivation(newform, DCFofQuote, handle, NULL);
      FormulaSetInsert(tmpset, newform);
      FormulaSetInsert(archive, handle);
   }
   assert(FormulaSetEmpty(set));

   FormulaSetInsertSet(set, tmpset);
   FormulaSetFree(tmpset);
}

/*-----------------------------------------------------------------------
//
// Function: FormulaSetDocInital()
//
//   If level >= 2, print all formula as initials.
//
// Global Variables: -
//
// Side Effects    : Output
//
/----------------------------------------------------------------------*/

void FormulaSetDocInital(FILE *out, long level, FormulaSet_p set)
{
   WFormula_p handle;

   if (level >= 2)
   {
      for (handle = set->anchor->succ; handle != set->anchor; handle =
                                                                  handle->succ)
      {
         DocFormulaCreationDefault(handle, inf_initial, NULL, NULL);
      }
   }
}

/*-----------------------------------------------------------------------
//
// Function: FormulaCountNonTopLevelLambdas()
//
//   Count non-top level lambdas in the formulas
//
// Global Variables: -
//
// Side Effects    : Output
//
/----------------------------------------------------------------------*/

int FormulaCountNonTopLevelLambdas(Sig_p sig, TFormula_p form)
{
   PStack_p stack = PStackAlloc();
   PStackPushP(stack, form);
   PStackPushInt(stack, true);
   int res = 0;

   while(!PStackEmpty(stack))
   {
      bool is_at_top = PStackPopInt(stack);
      Term_p form = PStackPopP(stack);

      if(TermHasLambdaSubterm(form))
      {
         if(is_at_top)
         {
            is_at_top = is_at_top && (!TermIsFreeVar(form)
                                      && (SigIsLogicalSymbol(sig, form->f_code)
                                          || TermIsLambda(form)));
         }
         else if(TermIsLambda(form))
         {
            res++;
         }

         for(int i=TermIsPhonyApp(form) || TermIsLambda(form) ? 1 : 0; i<form->arity; i++)
         {
            if(TermHasLambdaSubterm(form->args[i]))
            {
               PStackPushP(stack, form->args[i]);
               PStackPushInt(stack, is_at_top);
            }
         }
      }
   }

   PStackFree(stack);
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: FormulaHasAppVarLit()
//
//   Does formula have a literal that is an applied variable?
//
// Global Variables: -
//
// Side Effects    : Output
//
/----------------------------------------------------------------------*/

bool FormulaHasAppVarLit(Sig_p sig, TFormula_p form)
{
   PStack_p stack = PStackAlloc();
   PStackPushP(stack, form);
   bool res = false;

   while(!PStackEmpty(stack) && !res)
   {
      TFormula_p form = PStackPopP(stack);
      if(TFormulaIsLiteral(sig, form))
      {
         res = TermIsAppliedFreeVar(form->args[0])
               || TermIsAppliedFreeVar(form->args[1]);
      }
      else if(form->f_code > 0 && SigIsLogicalSymbol(sig, form->f_code))
      {
         for(int i=0; i<form->arity; i++)
         {
            PStackPushP(stack, form->args[i]);
         }
      }
   }

   PStackFree(stack);
   return res;
}

/*-----------------------------------------------------------------------
//
// Function: ClauseSetLiftLambdas()
//
//   Lift lambdas in clauses, change them in place, modify the proof object
//   and store the lambda definitions in archive. New lambda definitions are
//   clausified in turn.
//
// Global Variables: -
//
// Side Effects    : Memory operations
//
/----------------------------------------------------------------------*/

void ClauseSetLiftLambdas(ClauseSet_p set, FormulaSet_p archive, TB_p terms,
                          VarBank_p fresh_vars, bool unroll_fool)
{
   PStack_p defs = PStackAlloc();
   PTree_p all_defs = NULL;
   PDTree_p liftings = PDTreeAllocWDeleter(terms, deleter);
   Term_p lterm, rterm;

   VarBankSetVCountsToUsed(terms->vars);

   for(Clause_p handle = set->anchor->succ; handle!=set->anchor; handle = handle->succ)
   {
      bool cl_changed = false;
      for(Eqn_p lit = handle->literals; lit; lit = lit->next)
      {
         DBGTermCheckUnownedSubterm(stdout, lit->lterm, "UnownedCSLift1L");
         DBGTermCheckUnownedSubterm(stdout, lit->rterm, "UnownedCSLift1R");
         lterm = !TermIsLambda(lit->lterm) && TermHasLambdaSubterm(lit->lterm) ?
            LiftLambdas(terms, DecodeFormulasForCNF(terms, lit->lterm), defs, liftings)
            : lit->lterm;
         rterm = !TermIsLambda(lit->rterm) && TermHasLambdaSubterm(lit->rterm) ?
            LiftLambdas(terms, DecodeFormulasForCNF(terms, lit->rterm), defs, liftings)
            : lit->rterm;
         DBGTermCheckUnownedSubterm(stdout, lit->lterm, "UnownedCSLift2L");
         DBGTermCheckUnownedSubterm(stdout, lit->rterm, "UnownedCSLift2R");
         cl_changed = cl_changed || lit->lterm != lterm || lit->rterm != rterm;
         lit->lterm = lterm;
         lit->rterm = rterm;
      }

      if(cl_changed)
      {
         while (!(PStackEmpty(defs)))
         {
            WFormula_p def = PStackPopP(defs);
            DBGTermCheckUnownedSubterm(stdout, def->tformula, "ClauseSetLiftLambdas2");
            ClausePushDerivation(handle, DCLiftLambdas, def, NULL);
            PTreeStore(&all_defs, def);
         }
      }
   }

   PStack_p def_iter = PTreeTraverseInit(all_defs);
   PTree_p node;
   while((node = PTreeTraverseNext(def_iter)))
   {
      WFormula_p handle = node->key;
      DBGTermCheckUnownedSubterm(stdout, handle->tformula, "ClauseSetLiftLambdas3");
      WFormula_p copy = WFormulaFlatCopy(handle);
      DBGTermCheckUnownedSubterm(stdout, copy->tformula, "ClauseSetLiftLambdas4");
      FormulaSetInsert(archive, handle);
      if(unroll_fool)
      {
         TFormulaUnrollFOOL(copy, terms);
      }
      WFormulaSimplify(copy, terms);
      DBGTermCheckUnownedSubterm(stdout, copy->tformula, "ClauseSetLiftLambdas5");
      WFormulaCNF2(copy, set, terms, fresh_vars, 100, unroll_fool);
      FormulaSetInsert(archive, copy);
   }
   PTreeTraverseExit(def_iter);

   PStackFree(defs);
   PTreeFree(all_defs);
   PDTreeFree(liftings);
}

/*---------------------------------------------------------------------*/
/*                        End of File                                  */
/*---------------------------------------------------------------------*/
