diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/api/cpp/cvc5.cpp | 206 | ||||
-rw-r--r-- | src/api/cpp/cvc5.h | 23 |
2 files changed, 228 insertions, 1 deletions
diff --git a/src/api/cpp/cvc5.cpp b/src/api/cpp/cvc5.cpp index 4e14b8e34..9b920b875 100644 --- a/src/api/cpp/cvc5.cpp +++ b/src/api/cpp/cvc5.cpp @@ -570,6 +570,7 @@ const static std::unordered_map<cvc5::Kind, Kind, cvc5::kind::KindHashFunction> {cvc5::Kind::APPLY_UPDATER, APPLY_UPDATER}, {cvc5::Kind::DT_SIZE, DT_SIZE}, {cvc5::Kind::TUPLE_PROJECT, TUPLE_PROJECT}, + {cvc5::Kind::TUPLE_PROJECT_OP, TUPLE_PROJECT}, /* Separation Logic ------------------------------------------------ */ {cvc5::Kind::SEP_NIL, SEP_NIL}, {cvc5::Kind::SEP_EMP, SEP_EMP}, @@ -1865,6 +1866,14 @@ size_t Op::getNumIndices() const { CVC5_API_TRY_CATCH_BEGIN; CVC5_API_CHECK_NOT_NULL; + //////// all checks before this line + return getNumIndicesHelper(); + //////// + CVC5_API_TRY_CATCH_END; +} + +size_t Op::getNumIndicesHelper() const +{ if (!isIndexedHelper()) { return 0; @@ -1898,9 +1907,185 @@ size_t Op::getNumIndices() const break; default: CVC5_API_CHECK(false) << "Unhandled kind " << kindToString(k); } + return size; +} + +Term Op::operator[](size_t index) const +{ + return getIndexHelper(index); +} + +Term Op::getIndexHelper(size_t index) const +{ + CVC5_API_TRY_CATCH_BEGIN; + CVC5_API_CHECK_NOT_NULL; + CVC5_API_CHECK(!d_node->isNull()) + << "Expecting a non-null internal expression. This Op is not indexed."; + CVC5_API_CHECK(index < getNumIndicesHelper()) << "index out of bound"; + Kind k = intToExtKind(d_node->getKind()); + Term t; + switch (k) + { + case DIVISIBLE: + { + t = d_solver->mkValHelper<Rational>( + Rational(d_node->getConst<Divisible>().k)); + break; + } + case BITVECTOR_REPEAT: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<BitVectorRepeat>().d_repeatAmount); + break; + } + case BITVECTOR_ZERO_EXTEND: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<BitVectorZeroExtend>().d_zeroExtendAmount); + break; + } + case BITVECTOR_SIGN_EXTEND: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<BitVectorSignExtend>().d_signExtendAmount); + break; + } + case BITVECTOR_ROTATE_LEFT: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<BitVectorRotateLeft>().d_rotateLeftAmount); + break; + } + case BITVECTOR_ROTATE_RIGHT: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<BitVectorRotateRight>().d_rotateRightAmount); + break; + } + case INT_TO_BITVECTOR: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<IntToBitVector>().d_size); + break; + } + case IAND: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<IntAnd>().d_size); + break; + } + case FLOATINGPOINT_TO_UBV: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<FloatingPointToUBV>().d_bv_size.d_size); + break; + } + case FLOATINGPOINT_TO_SBV: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<FloatingPointToSBV>().d_bv_size.d_size); + break; + } + case REGEXP_REPEAT: + { + t = d_solver->mkValHelper<cvc5::Rational>( + d_node->getConst<RegExpRepeat>().d_repeatAmount); + break; + } + case BITVECTOR_EXTRACT: + { + cvc5::BitVectorExtract ext = d_node->getConst<BitVectorExtract>(); + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>(ext.d_high) + : d_solver->mkValHelper<cvc5::Rational>(ext.d_low); + break; + } + case FLOATINGPOINT_TO_FP_IEEE_BITVECTOR: + { + cvc5::FloatingPointToFPIEEEBitVector ext = + d_node->getConst<FloatingPointToFPIEEEBitVector>(); + + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().exponentWidth()) + : d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().significandWidth()); + break; + } + case FLOATINGPOINT_TO_FP_FLOATINGPOINT: + { + cvc5::FloatingPointToFPFloatingPoint ext = + d_node->getConst<FloatingPointToFPFloatingPoint>(); + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().exponentWidth()) + : d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().significandWidth()); + break; + } + case FLOATINGPOINT_TO_FP_REAL: + { + cvc5::FloatingPointToFPReal ext = + d_node->getConst<FloatingPointToFPReal>(); + + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().exponentWidth()) + : d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().significandWidth()); + break; + } + case FLOATINGPOINT_TO_FP_SIGNED_BITVECTOR: + { + cvc5::FloatingPointToFPSignedBitVector ext = + d_node->getConst<FloatingPointToFPSignedBitVector>(); + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().exponentWidth()) + : d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().significandWidth()); + break; + } + case FLOATINGPOINT_TO_FP_UNSIGNED_BITVECTOR: + { + cvc5::FloatingPointToFPUnsignedBitVector ext = + d_node->getConst<FloatingPointToFPUnsignedBitVector>(); + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().exponentWidth()) + : d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().significandWidth()); + break; + } + case FLOATINGPOINT_TO_FP_GENERIC: + { + cvc5::FloatingPointToFPGeneric ext = + d_node->getConst<FloatingPointToFPGeneric>(); + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().exponentWidth()) + : d_solver->mkValHelper<cvc5::Rational>( + ext.getSize().significandWidth()); + break; + } + case REGEXP_LOOP: + { + cvc5::RegExpLoop ext = d_node->getConst<RegExpLoop>(); + t = index == 0 ? d_solver->mkValHelper<cvc5::Rational>(ext.d_loopMinOcc) + : d_solver->mkValHelper<cvc5::Rational>(ext.d_loopMaxOcc); + + break; + } + + case TUPLE_PROJECT: + { + const std::vector<uint32_t>& projectionIndices = + d_node->getConst<TupleProjectOp>().getIndices(); + t = d_solver->mkValHelper<cvc5::Rational>(projectionIndices[index]); + break; + } + default: + { + CVC5_API_CHECK(false) << "Unhandled kind " << kindToString(k); + break; + } + } //////// all checks before this line - return size; + return t; //////// CVC5_API_TRY_CATCH_END; } @@ -2043,6 +2228,25 @@ std::pair<uint32_t, uint32_t> Op::getIndices() const CVC5_API_TRY_CATCH_END; } +template <> +std::vector<api::Term> Op::getIndices() const +{ + CVC5_API_TRY_CATCH_BEGIN; + CVC5_API_CHECK_NOT_NULL; + CVC5_API_CHECK(!d_node->isNull()) + << "Expecting a non-null internal expression. This Op is not indexed."; + size_t size = getNumIndicesHelper(); + std::vector<Term> terms(getNumIndices()); + for (size_t i = 0; i < size; i++) + { + terms[i] = getIndexHelper(i); + } + //////// all checks before this line + return terms; + //////// + CVC5_API_TRY_CATCH_END; +} + std::string Op::toString() const { CVC5_API_TRY_CATCH_BEGIN; diff --git a/src/api/cpp/cvc5.h b/src/api/cpp/cvc5.h index 0ee5990da..672ccf8f8 100644 --- a/src/api/cpp/cvc5.h +++ b/src/api/cpp/cvc5.h @@ -782,6 +782,8 @@ namespace cvc5::api { /* Op */ /* -------------------------------------------------------------------------- */ +class Term; + /** * A cvc5 operator. * An operator is a term that represents certain operators, instantiated @@ -843,6 +845,14 @@ class CVC5_EXPORT Op size_t getNumIndices() const; /** + * Get the index at position i. + * @param i the position of the index to return + * @return the index at position i + */ + + Term operator[](size_t i) const; + + /** * Get the indices used to create this Op. * Supports the following template arguments: * - string @@ -892,6 +902,19 @@ class CVC5_EXPORT Op bool isIndexedHelper() const; /** + * Helper for getNumIndices + * @return the number of indices of this op + */ + size_t getNumIndicesHelper() const; + + /** + * Helper for operator[](size_t i). + * @param i position of the index. Should be less than getNumIndicesHelper(). + * @return the index at position i + */ + Term getIndexHelper(size_t index) const; + + /** * The associated solver object. */ const Solver* d_solver; |