/******************************************************************************

 * Top contributors (to current version):

 *   Andrew Reynolds, Tianyi Liang, Mathias Preiner

 *

 * This file is part of the cvc5 project.

 *

 * Copyright (c) 2009-2021 by the authors listed in the file AUTHORS

 * in the top-level source directory and their institutional affiliations.

 * All rights reserved.  See the file COPYING in the top-level source

 * directory for licensing information.

 * ****************************************************************************

 *

 * Theory of strings.

 */

#include "cvc5_private.h"

#ifndef CVC5__THEORY__STRINGS__THEORY_STRINGS_H

#define CVC5__THEORY__STRINGS__THEORY_STRINGS_H

#include <climits>

#include <deque>

#include "context/cdhashset.h"

#include "context/cdlist.h"

#include "expr/node_trie.h"

#include "theory/ext_theory.h"

#include "theory/strings/base_solver.h"

#include "theory/strings/core_solver.h"

#include "theory/strings/eager_solver.h"

#include "theory/strings/extf_solver.h"

#include "theory/strings/infer_info.h"

#include "theory/strings/inference_manager.h"

#include "theory/strings/normal_form.h"

#include "theory/strings/proof_checker.h"

#include "theory/strings/regexp_elim.h"

#include "theory/strings/regexp_operation.h"

#include "theory/strings/regexp_solver.h"

#include "theory/strings/sequences_stats.h"

#include "theory/strings/solver_state.h"

#include "theory/strings/strategy.h"

#include "theory/strings/strings_fmf.h"

#include "theory/strings/strings_rewriter.h"

#include "theory/strings/term_registry.h"

#include "theory/theory.h"

#include "theory/uf/equality_engine.h"

namespace cvc5 {

namespace theory {

namespace strings {

/**

 * A theory solver for strings. At a high level, the solver implements

 * techniques described in:

 * - Liang et al, CAV 2014,

 * - Reynolds et al, CAV 2017,

 * - Reynolds et al, IJCAR 2020.

 * Its rewriter is described in:

 * - Reynolds et al, CAV 2019.

 */

class TheoryStrings : public Theory {

  friend class InferenceManager;

  typedef context::CDHashSet<Node> NodeSet;

  typedef context::CDHashSet<TypeNode, std::hash<TypeNode>> TypeNodeSet;

 public:

  TheoryStrings(context::Context* c,

                context::UserContext* u,

                OutputChannel& out,

                Valuation valuation,

                const LogicInfo& logicInfo,

                ProofNodeManager* pnm);

  ~TheoryStrings();

  //--------------------------------- initialization

  /** get the official theory rewriter of this theory */

  TheoryRewriter* getTheoryRewriter() override;

  /** get the proof checker of this theory */

  ProofRuleChecker* getProofChecker() override;

  /**

   * Returns true if we need an equality engine. If so, we initialize the

   * information regarding how it should be setup. For details, see the

   * documentation in Theory::needsEqualityEngine.

   */

  bool needsEqualityEngine(EeSetupInfo& esi) override;

  /** finish initialization */

  void finishInit() override;

  //--------------------------------- end initialization

  /** Identify this theory */

  std::string identify() const override;

  /** Explain */

  TrustNode explain(TNode literal) override;

  /** presolve */

  void presolve() override;

  /** shutdown */

  /** preregister term */

  void preRegisterTerm(TNode n) override;

  //--------------------------------- standard check

  /** Do we need a check call at last call effort? */

  bool needsCheckLastEffort() override;

  bool preNotifyFact(TNode atom,

                     bool pol,

                     TNode fact,

                     bool isPrereg,

                     bool isInternal) override;

  void notifyFact(TNode atom, bool pol, TNode fact, bool isInternal) override;

  /** Post-check, called after the fact queue of the theory is processed. */

  void postCheck(Effort level) override;

  //--------------------------------- end standard check

  /** propagate method */

  bool propagateLit(TNode literal);

  /** Conflict when merging two constants */

  void conflict(TNode a, TNode b);

  /** called when a new equivalence class is created */

  void eqNotifyNewClass(TNode t);

  /** preprocess rewrite */

  TrustNode ppRewrite(TNode atom, std::vector<SkolemLemma>& lems) override;

  /** Collect model values in m based on the relevant terms given by termSet */

  bool collectModelValues(TheoryModel* m,

                          const std::set<Node>& termSet) override;

 private:

  /** NotifyClass for equality engine */

  public:

   {

   {

     {

     }

    }

                                     TNode t1,

                                     TNode t2,

                                     bool value) override

    {

      }

    }

    {

    {

    {

    {

   private:

    /** The theory of strings object to notify */

    TheoryStrings& d_str;

    /** The eager solver of the theory of strings */

    EagerSolver& d_eagerSolver;

  };/* class TheoryStrings::NotifyClass */

  /** compute care graph */

  void computeCareGraph() override;

  /**

   * Are x and y shared terms that are not equal? This is used for constructing

   * the care graph in the above function.

   */

  bool areCareDisequal(TNode x, TNode y);

  /** Add care pairs */

  void addCarePairs(TNodeTrie* t1,

                    TNodeTrie* t2,

                    unsigned arity,

                    unsigned depth);

  /** Collect model info for type tn

   *

   * Assigns model values (in m) to all relevant terms of the string-like type

   * tn in the current context, which are stored in repSet. Furthermore,

   * col is a partition of repSet where equivalence classes are grouped into

   * sets having equal length, where these lengths are stored in lts.

   *

   * Returns false if a conflict is discovered while doing this assignment.

   */

  bool collectModelInfoType(TypeNode tn,

                            const std::unordered_set<Node>& repSet,

                            std::vector<std::vector<Node>>& col,

                            std::vector<Node>& lts,

                            TheoryModel* m);

  /** assert pending fact

   *

   * This asserts atom with polarity to the equality engine of this class,

   * where exp is the explanation of why (~) atom holds.

   *

   * This call may trigger further initialization steps involving the terms

   * of atom, including calls to registerTerm.

   */

  void assertPendingFact(Node atom, bool polarity, Node exp);

  //-----------------------inference steps

  /** check register terms pre-normal forms

   *

   * This calls registerTerm(n,2) on all non-congruent strings in the

   * equality engine of this class.

   */

  void checkRegisterTermsPreNormalForm();

  /** check codes

   *

   * This inference schema ensures that constraints between str.code terms

   * are satisfied by models that correspond to extensions of the current

   * assignment. In particular, this method ensures that str.code can be

   * given an interpretation that is injective for string arguments with length

   * one. It may add lemmas of the form:

   *   str.code(x) == -1 V str.code(x) != str.code(y) V x == y

   */

  void checkCodes();

  /** check register terms for normal forms

   *

   * This calls registerTerm(str.++(t1, ..., tn ), 3) on the normal forms

   * (t1, ..., tn) of all string equivalence classes { s1, ..., sm } such that

   * there does not exist a term of the form str.len(si) in the current context.

   */

  void checkRegisterTermsNormalForms();

  //-----------------------end inference steps

  /** run the given inference step */

  void runInferStep(InferStep s, int effort);

  /** run strategy for effort e */

  void runStrategy(Theory::Effort e);

  /** print strings equivalence classes for debugging */

  std::string debugPrintStringsEqc();

  /** Commonly used constants */

  Node d_true;

  Node d_false;

  Node d_zero;

  Node d_one;

  Node d_neg_one;

  /** the cardinality of the alphabet */

  uint32_t d_cardSize;

  /** The notify class */

  NotifyClass d_notify;

  /**

   * Statistics for the theory of strings/sequences. All statistics for these

   * theories is collected in this object.

   */

  SequencesStatistics d_statistics;

  /** The solver state object */

  SolverState d_state;

  /** The eager solver */

  EagerSolver d_eagerSolver;

  /** The term registry for this theory */

  TermRegistry d_termReg;

  /** The extended theory callback */

  StringsExtfCallback d_extTheoryCb;

  /** Extended theory, responsible for context-dependent simplification. */

  ExtTheory d_extTheory;

  /** The (custom) output channel of the theory of strings */

  InferenceManager d_im;

  /** The theory rewriter for this theory. */

  StringsRewriter d_rewriter;

  /** The proof rule checker */

  StringProofRuleChecker d_checker;

  /**

   * The base solver, responsible for reasoning about congruent terms and

   * inferring constants for equivalence classes.

   */

  BaseSolver d_bsolver;

  /**

   * The core solver, responsible for reasoning about string concatenation

   * with length constraints.

   */

  CoreSolver d_csolver;

  /**

   * Extended function solver, responsible for reductions and simplifications

   * involving extended string functions.

   */

  ExtfSolver d_esolver;

  /** regular expression solver module */

  RegExpSolver d_rsolver;

  /** regular expression elimination module */

  RegExpElimination d_regexp_elim;

  /** Strings finite model finding decision strategy */

  StringsFmf d_stringsFmf;

  /** The representation of the strategy */

  Strategy d_strat;

};/* class TheoryStrings */

}  // namespace strings

}  // namespace theory

}  // namespace cvc5

#endif /* CVC5__THEORY__STRINGS__THEORY_STRINGS_H */