cvc5-cvc5-1.2.0.src.theory.bags.theory_bags.h Maven / Gradle / Ivy
The newest version!
/******************************************************************************
* Top contributors (to current version):
* Mudathir Mohamed, Andrew Reynolds, Aina Niemetz
*
* This file is part of the cvc5 project.
*
* Copyright (c) 2009-2024 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.
* ****************************************************************************
*
* Bags theory.
*/
#include "cvc5_private.h"
#ifndef CVC5__THEORY__BAGS__THEORY_BAGS_H
#define CVC5__THEORY__BAGS__THEORY_BAGS_H
#include "theory/bags/bag_reduction.h"
#include "theory/bags/bag_solver.h"
#include "theory/bags/bags_rewriter.h"
#include "theory/bags/bags_statistics.h"
#include "theory/bags/inference_generator.h"
#include "theory/bags/inference_manager.h"
#include "theory/bags/solver_state.h"
#include "theory/bags/strategy.h"
#include "theory/bags/term_registry.h"
#include "theory/care_pair_argument_callback.h"
#include "theory/theory.h"
#include "theory/theory_eq_notify.h"
namespace cvc5::internal {
namespace theory {
namespace bags {
class TheoryBags : public Theory
{
public:
/** Constructs a new instance of TheoryBags w.r.t. the provided contexts. */
TheoryBags(Env& env, OutputChannel& out, Valuation valuation);
~TheoryBags() override;
//--------------------------------- 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;
/** preprocess rewrite */
TrustNode ppRewrite(TNode atom, std::vector& lems) override;
//--------------------------------- end initialization
/**
* initialize bag and count terms
*/
void initialize();
/**
* collect bags' representatives and all count terms.
*/
void collectBagsAndCountTerms();
//--------------------------------- standard check
/** Post-check, called after the fact queue of the theory is processed. */
void postCheck(Effort effort) override;
/** Notify fact */
void notifyFact(TNode atom, bool pol, TNode fact, bool isInternal) override;
//--------------------------------- end standard check
/** Collect model values in m based on the relevant terms given by termSet */
bool collectModelValues(TheoryModel* m,
const std::set& termSet) override;
TrustNode explain(TNode) override;
Node getCandidateModelValue(TNode) override;
std::string identify() const override { return "THEORY_BAGS"; }
void preRegisterTerm(TNode n) override;
void presolve() override;
void computeCareGraph() override;
void processCarePairArgs(TNode a, TNode b) override;
bool isCareArg(Node n, unsigned a);
/** run strategy for effort e */
void runStrategy(Theory::Effort e);
/** run the given inference step */
bool runInferStep(InferStep s, int effort);
private:
/** Functions to handle callbacks from equality engine */
class NotifyClass : public TheoryEqNotifyClass
{
public:
NotifyClass(TheoryBags& theory, TheoryInferenceManager& inferenceManager)
: TheoryEqNotifyClass(inferenceManager), d_theory(theory)
{
}
void eqNotifyNewClass(TNode n) override;
void eqNotifyMerge(TNode n1, TNode n2) override;
void eqNotifyDisequal(TNode n1, TNode n2, TNode reason) override;
private:
TheoryBags& d_theory;
};
/** expand the definition of the bag.choose operator */
TrustNode expandChooseOperator(const Node& node,
std::vector& lems);
/** The state of the bags solver at full effort */
SolverState d_state;
/** The inference manager */
InferenceManager d_im;
/** The inference generator */
InferenceGenerator d_ig;
/** Instance of the above class */
NotifyClass d_notify;
/** Statistics for the theory of bags. */
BagsStatistics d_statistics;
/** The theory rewriter for this theory. */
BagsRewriter d_rewriter;
/** The term registry for this theory */
TermRegistry d_termReg;
/** the main solver for bags */
BagSolver d_solver;
/** The care pair argument callback, used for theory combination */
CarePairArgumentCallback d_cpacb;
/** map kinds to their terms. It is cleared during post check */
std::map> d_opMap;
/** The representation of the strategy */
Strategy d_strat;
void eqNotifyNewClass(TNode n);
void eqNotifyMerge(TNode n1, TNode n2);
void eqNotifyDisequal(TNode t1, TNode t2, TNode reason);
}; /* class TheoryBags */
} // namespace bags
} // namespace theory
} // namespace cvc5::internal
#endif /* CVC5__THEORY__BAGS__THEORY_BAGS_H */