diff --git a/.gitmodules b/.gitmodules new file mode 100644 index 0000000..fbd8ce1 --- /dev/null +++ b/.gitmodules @@ -0,0 +1,3 @@ +[submodule "qt-openssl-encryption"] + path = qt-openssl-encryption + url = ssh://git@git.itsblue.de:2222/itsblue-development/qt-openssl-encryption.git diff --git a/Qt-Secret/GMP/.gitignore b/Qt-Secret/GMP/.gitignore deleted file mode 100644 index 84c048a..0000000 --- a/Qt-Secret/GMP/.gitignore +++ /dev/null @@ -1 +0,0 @@ -/build/ diff --git a/Qt-Secret/GMP/GMP.pri b/Qt-Secret/GMP/GMP.pri deleted file mode 100644 index 73112bb..0000000 --- a/Qt-Secret/GMP/GMP.pri +++ /dev/null @@ -1,24 +0,0 @@ -# -# Copyright (C) 2018-2019 QuasarApp. -# Distributed under the lgplv3 software license, see the accompanying -# Everyone is permitted to copy and distribute verbatim copies -# of this license document, but changing it is not allowed. -# - -!isEmpty(Qt_GMP_LIB):error("GMP.pri already included") -Qt_GMP_LIB = 1 - -DEFINES += MINIGMP_LIBRARY -DEFINES += QT_DEPRECATED_WARNINGS - -HEADERS += \ - $$PWD/bigint.h \ - $$PWD/mini-gmp.h \ - $$PWD/minigmp_global.h - -SOURCES += \ - $$PWD/bigint.cpp \ - $$PWD/mini-gmp.c - - -INCLUDEPATH += $$PWD diff --git a/Qt-Secret/GMP/LICENSE b/Qt-Secret/GMP/LICENSE deleted file mode 100644 index 450ca25..0000000 --- a/Qt-Secret/GMP/LICENSE +++ /dev/null @@ -1,21 +0,0 @@ -MIT License - -Copyright (c) 2019 QuasarApp - -Permission is hereby granted, free of charge, to any person obtaining a copy -of this software and associated documentation files (the "Software"), to deal -in the Software without restriction, including without limitation the rights -to use, copy, modify, merge, publish, distribute, sublicense, and/or sell -copies of the Software, and to permit persons to whom the Software is -furnished to do so, subject to the following conditions: - -The above copyright notice and this permission notice shall be included in all -copies or substantial portions of the Software. - -THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR -IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, -FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE -AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER -LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, -OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE -SOFTWARE. diff --git a/Qt-Secret/GMP/bigint.cpp b/Qt-Secret/GMP/bigint.cpp deleted file mode 100644 index 4287886..0000000 --- a/Qt-Secret/GMP/bigint.cpp +++ /dev/null @@ -1,618 +0,0 @@ -//# -//# Copyright (C) 2018-2019 QuasarApp. -//# Distributed under the lgplv3 software license, see the accompanying -//# Everyone is permitted to copy and distribute verbatim copies -//# of this license document, but changing it is not allowed. -//# - -#include "bigint.h" -#include -#include -#include - -// constructors -BigInt::BigInt() { - mpz_init(data); -} - -BigInt::BigInt(const BigInt &val, int bitCount) { - if (bitCount > 0) { - mpz_init2(data, static_cast(bitCount)); - } else { - mpz_init(data); - } - - mpz_set(data, val.data); - -} - -BigInt::BigInt(const std::string &str, int base): - BigInt() { - mpz_set_str(data, str.c_str(), base); -} - -BigInt::BigInt(intMpz val): - BigInt() { - mpz_set_si(data, val); -} - -BigInt::BigInt(char item, unsigned int size, int base = 2): - BigInt(std::string(size, item),base) { -} - -std::string BigInt::getString(int base) const { - char *str = mpz_get_str(nullptr, base, data); - return str; -} - -BigInt::~BigInt() { - mpz_clear(data); -} - -BigInt &BigInt::powm(const BigInt &pow, const BigInt &mod) { - mpz_powm(data, data, pow.data, mod.data); - return *this; -} - -BigInt BigInt::powm(BigInt val, const BigInt &pow, const BigInt &mod) { - return val.powm(pow, mod); -} - -BigInt &BigInt::pow(uIntMpz pow) { - mpz_pow_ui(data, data, pow); - return *this; -} - -int BigInt::sizeBits() const { - return sizeBytes() * 8; -} - -int BigInt::sizeBytes() const { - return static_cast(mpz_size(data) * sizeof ((*data->_mp_d))); -} - -int BigInt::longBits() const { - return static_cast(getString(2).size()); -} - -int BigInt::longBytes() const { - return static_cast(std::ceil(static_cast(longBits()) / 8)); -} - -#define GMP_ABS(x) ((x) >= 0 ? (x) : -(x)) -int BigInt::sizeType() const { - return static_cast(static_cast(GMP_ABS( data->_mp_alloc)) * - sizeof ((*data->_mp_d))); - -} - -bool BigInt::isPrime(bool absalut) const { - return (mpz_probab_prime_p(data, 50) - (absalut? 1: 0)) > 0; -} - -BigInt& BigInt::gcd(const BigInt &a, const BigInt &b) { - mpz_gcd(data, a.data, b.data); - return *this; - -} - -void BigInt::fromHex(const std::string &hex) { - mpz_set_str(data, hex.c_str(), 16); -} - -BigInt BigInt::bigPow10(unsigned short pow) { - return "1" + std::string(pow, '0'); -} - -BigInt &BigInt::toNegative() { - mpz_neg(data, data); - return *this; -} - -BigInt &BigInt::operator =(const BigInt &val) { - mpz_set(data, val.data); - return *this; -} - -BigInt &BigInt::operator =(const std::string &imput) { - mpz_set_str(data, imput.c_str(), 10); - return *this; -} - -BigInt &BigInt::operator =(intMpz val) { - mpz_set_si(data, val); - return *this; -} - -// add operators - -BigInt operator +(BigInt left, intMpz right) { - if (right >= 0) { - mpz_add_ui(left.data, left.data, static_cast(right)); - return left; - } - - return left -= std::abs(right); -} - -BigInt operator +(intMpz left, BigInt right) { - return right += left; -} - -BigInt operator +(BigInt left, const BigInt &right) { - mpz_add(left.data, left.data, right.data); - return left; -} - -BigInt operator +(BigInt left, const std::string &right) { - return left += BigInt(right); -} - -BigInt operator +(const std::string &left, const BigInt &right) { - return BigInt(left) + right; -} - -BigInt& operator +=(BigInt &left, intMpz right) { - if (right >= 0) { - mpz_add_ui(left.data, left.data, static_cast(right)); - return left; - } - return left -= std::abs(right); -} - -BigInt& operator +=(BigInt &left, const BigInt &right) { - mpz_add(left.data, left.data, right.data); - return left; -} - -BigInt& operator +=(BigInt &left, const std::string &right) { - return left += BigInt(right); -} - -// sub operators - -BigInt operator -(BigInt left, const BigInt &right) { - mpz_sub(left.data, left.data, right.data); - return left; -} - -BigInt operator -(BigInt left, intMpz right) { - if (right >= 0) { - mpz_sub_ui(left.data, left.data, static_cast(right)); - return left; - } - return left += std::abs(right); -} - -BigInt operator -(intMpz left, BigInt right) { - if (left >= 0) { - mpz_ui_sub(right.data, static_cast(left), right.data); - return right; - } - return right += std::abs(left); -} - -BigInt operator-(BigInt val) { - mpz_neg(val.data, val.data); - return val; -} - -BigInt operator -(BigInt left, const std::string &right) { - return left -= BigInt(right); -} - -BigInt operator -(const std::string & left, const BigInt &right) { - return BigInt(left) - right; -} - -BigInt& operator -=(BigInt &left, const BigInt &right) { - mpz_sub(left.data, left.data, right.data); - return left; -} - -BigInt& operator -=(BigInt &left, const std::string &right) { - return left -= BigInt(right); -} - -BigInt& operator -=(BigInt &left, intMpz right) { - if (right >= 0) { - mpz_sub_ui(left.data, left.data, static_cast(right)); - return left; - } - return left += std::abs(right); -} - -// div operators - -BigInt operator /(BigInt left, const BigInt &right) { - mpz_tdiv_q(left.data, left.data, right.data); - return left; -} - -BigInt operator /(BigInt left, intMpz right) { - mpz_tdiv_q_ui(left.data, left.data, static_cast(std::abs(right))); - - if (right >= 0) { - return left; - } - return -left; -} - -BigInt operator /(BigInt left, const std::string &right) { - return left /= BigInt(right); -} - -BigInt operator /(intMpz left, BigInt right) { - return BigInt(left) / right; -} - -BigInt operator /(const std::string & left, const BigInt &right) { - return BigInt(left) / right; -} - -BigInt& operator /=(BigInt &left, const BigInt &right) { - mpz_tdiv_q(left.data, left.data, right.data); - return left; -} - -BigInt& operator /=(BigInt &left, const std::string &right) { - return left /= BigInt(right); -} - -BigInt& operator /=(BigInt &left, intMpz right) { - mpz_tdiv_q_ui(left.data, left.data, static_cast(std::abs(right))); - - if (right >= 0) { - return left; - } - return left.toNegative(); -} - -// mul operators -BigInt operator *(BigInt left, const BigInt &right) { - mpz_mul(left.data, left.data, right.data); - return left; -} - -BigInt operator *(BigInt left, intMpz right) { - mpz_mul_ui(left.data, left.data, static_cast(std::abs(right))); - - if (right >= 0) { - return left; - } - return -left; -} - -BigInt operator *(intMpz left, BigInt right) { - return right *= left; -} - -BigInt operator *(const std::string & left, BigInt right) { - return right *= BigInt(left); -} - -BigInt operator *(BigInt left, const std::string &right) { - return left *= BigInt(right); -} - -BigInt& operator *=(BigInt &left, const BigInt &right) { - mpz_mul(left.data, left.data, right.data); - return left; -} - -BigInt& operator *=(BigInt &left, const std::string &right) { - return left *= BigInt(right); -} - -BigInt& operator *=(BigInt &left, intMpz right) { - mpz_mul_ui(left.data, left.data, static_cast(std::abs(right))); - - if (right >= 0) { - return left; - } - return left.toNegative(); -} - -//mod operations -BigInt operator %(BigInt left, const BigInt &right) { - mpz_tdiv_r(left.data, left.data, right.data); - return left; -} - -BigInt operator %(BigInt left, intMpz right) { - mpz_tdiv_r_ui(left.data, left.data, static_cast(std::abs(right))); - return left; -} - -BigInt operator %(intMpz left, BigInt right) { - return BigInt(left) % right; -} - -BigInt operator %(BigInt left, const std::string & right) { - return left %= BigInt(right); -} - -BigInt operator %(const std::string & left, const BigInt &right) { - return BigInt(left) % right; -} - -BigInt& operator %=(BigInt& left, const BigInt &right) { - mpz_tdiv_r(left.data, left.data, right.data); - return left; -} - -BigInt& operator %=(BigInt& left, intMpz right) { - mpz_tdiv_r_ui(left.data, left.data, static_cast(std::abs(right))); - return left; -} - -BigInt& operator %=(BigInt &left, const std::string &right) { - return left %= BigInt(right); -} - -// incriment and dicriment -BigInt &BigInt::operator--() { - *this -= 1; - return *this; -} - -BigInt &BigInt::operator++() { - *this += 1; - return *this; -} - -BigInt BigInt::operator--(int) { - BigInt temp(*this); - --*this; - return temp; -} - -BigInt BigInt::operator++(int) { - BigInt temp(*this); - ++*this; - return temp; -} - -// move operators - -BigInt operator >>(BigInt left, int right) { - if (right >= 0) { - mpn_rshift(left.data->_mp_d, - left.data->_mp_d, - left.data->_mp_size, - static_cast(std::abs(right))); - return left; - } - - return left << right; -} - -BigInt operator <<(BigInt left, int right) { - if (right >= 0) { - mpn_lshift(left.data->_mp_d, - left.data->_mp_d, - left.data->_mp_size, - static_cast(std::abs(right))); - return left; - } - - return left >> right; -} - -BigInt& operator >>=(BigInt &left, int right) { - if (right >= 0) { - mpn_rshift(left.data->_mp_d, - left.data->_mp_d, - left.data->_mp_size, - static_cast(std::abs(right))); - return left; - } - - return left <<= right; -} - -BigInt& operator <<=(BigInt &left, int right) { - if (right >= 0) { - mpn_lshift(left.data->_mp_d, - left.data->_mp_d, - left.data->_mp_size, - static_cast(std::abs(right))); - return left; - } - - return left >>= right; -} - -// other bin operators -BigInt operator ~(BigInt left) { - mpz_com(left.data, left.data); - return left; -} - -BigInt operator |(BigInt left, const BigInt &right) { - mpz_ior(left.data, left.data, right.data); - return left; -} - -BigInt operator |(const BigInt &left, intMpz right) { - return left | BigInt(right); -} - -BigInt& operator |=(BigInt &left, const BigInt &right) { - mpz_ior(left.data, left.data, right.data); - return left; -} - -BigInt& operator |=(BigInt &left, intMpz right) { - return left |= BigInt(right); -} - -BigInt operator &(BigInt left, const BigInt &right) { - mpz_and(left.data, left.data, right.data); - return left; -} - -BigInt operator &(const BigInt &left, intMpz right) { - return left & BigInt(right); -} - - -BigInt& operator &=(BigInt &left, const BigInt &right) { - mpz_and(left.data, left.data, right.data); - return left; -} - -BigInt& operator &=(BigInt &left, intMpz right) { - return left &= BigInt(right); -} - -BigInt operator ^(BigInt left, const BigInt &right) { - mpz_xor(left.data, left.data, right.data); - return left; -} - -BigInt operator ^(const BigInt &left, intMpz right) { - return left ^ BigInt(right); -} - -BigInt& operator ^=(BigInt &left, const BigInt &right) { - mpz_xor(left.data, left.data, right.data); - return left; -} - -BigInt& operator ^=(BigInt &left, intMpz right) { - return left ^= BigInt(right); -} - - -// logic operators - -bool operator!(const BigInt &val) { - return val == 0; -} - -bool operator == (const BigInt& left, const BigInt& right) { - return mpz_cmp(left.data, right.data) == 0; -} - -bool operator == (const BigInt& left, intMpz right) { - return mpz_cmp_si(left.data, right) == 0; -} - -bool operator == (const BigInt &left, const std::string &right) { - return left == BigInt(right); -} - -bool operator == ( intMpz left, const BigInt & right) { - return right == left; -} - -bool operator == ( const std::string & left, const BigInt & right) { - return right == BigInt(left); -} - -bool operator != (const BigInt &left, const BigInt& right) { - return !(left == right); -} - -bool operator != (const BigInt &left, intMpz right) { - return !(left == right); -} - -bool operator != (const BigInt &left, const std::string &right) { - return left != BigInt(right); -} - -bool operator != ( intMpz left, const BigInt & right) { - return right != left; -} - -bool operator != ( const std::string & left, const BigInt & right) { - return right != BigInt(left); -} - -bool operator < ( const BigInt &left, const BigInt& right) { - return mpz_cmp(left.data, right.data) < 0; -} - -bool operator < ( const BigInt &left, intMpz right) { - return mpz_cmp_si(left.data, right) < 0; -} - -bool operator < ( const BigInt &left, const std::string &right) { - return left < BigInt(right); -} - -bool operator < ( intMpz left, const BigInt & right) { - return right > left; -} - -bool operator < ( const std::string & left, const BigInt & right) { - return right > BigInt(left); -} - -bool operator > ( const BigInt &left, const BigInt& right) { - return mpz_cmp(left.data, right.data) > 0; -} - -bool operator > ( const BigInt &left, intMpz right) { - return mpz_cmp_si(left.data, right) > 0; -} - -bool operator > ( const BigInt &left, const std::string &right) { - return left > BigInt(right); -} - -bool operator > ( intMpz left, const BigInt & right) { - return right < left; -} - -bool operator > ( const std::string & left, const BigInt & right) { - return right < BigInt(left); -} - -bool operator <= ( const BigInt &left, const BigInt& right) { - return mpz_cmp(left.data, right.data) <= 0; -} - -bool operator <= ( const BigInt &left, intMpz right) { - return mpz_cmp_si(left.data, right) <= 0; -} - -bool operator <= ( const BigInt &left, const std::string &right) { - return left <= BigInt(right); -} - -bool operator <= ( intMpz left, const BigInt & right) { - return right >= left; -} - -bool operator <= ( const std::string & left, const BigInt & right) { - return right >= BigInt(left); -} - -bool operator >= ( const BigInt &left, const BigInt& right) { - return mpz_cmp(left.data, right.data) >= 0; -} - -bool operator >= ( const BigInt &left, intMpz right) { - return mpz_cmp_si(left.data, right) >= 0; -} - -bool operator >= ( const BigInt &left, const std::string &right) { - return left >= BigInt(right); -} - -bool operator >= ( intMpz left, const BigInt & right) { - return right <= left; -} - -bool operator >= ( const std::string & left, const BigInt & right) { - return right <= BigInt(left); -} - -//// cast operations - -//BigInt::operator bool() const { -// return *this != 0; -//} diff --git a/Qt-Secret/GMP/bigint.h b/Qt-Secret/GMP/bigint.h deleted file mode 100644 index 6a7d3a6..0000000 --- a/Qt-Secret/GMP/bigint.h +++ /dev/null @@ -1,206 +0,0 @@ -//# -//# Copyright (C) 2018-2019 QuasarApp. -//# Distributed under the lgplv3 software license, see the accompanying -//# Everyone is permitted to copy and distribute verbatim copies -//# of this license document, but changing it is not allowed. -//# - - -#ifndef BIGINT_H -#define BIGINT_H -#include "mini-gmp.h" -#include -#include -#include "minigmp_global.h" - -/** - * @brief The BigInt class - c++ minigmp wrapper - */ - -class MINIGMPSHARED_EXPORT BigInt -{ - mpz_t data; -public: - BigInt(); - BigInt(const BigInt& val, int bitCount = -1); - BigInt(const std::string &imput, int base = 10); - BigInt(intMpz val); - BigInt(char item, unsigned int size, int base); - - std::string getString(int base = 10) const; - ~BigInt(); - - BigInt& powm(const BigInt &pow, const BigInt &mod); - static BigInt powm(BigInt val, const BigInt & pow, const BigInt &mod); - - BigInt& pow(uIntMpz pow); - BigInt& log(int base); - /** - * @brief sizeBits - * @return size of bits in memory - */ - int sizeBits() const; - int sizeBytes() const; - - /** - * @brief longBits - * @return current length in Bits of number - */ - int longBits() const; - int longBytes() const; - - int sizeType() const; - bool isPrime(bool absalut = false) const; - BigInt& gcd(const BigInt &a, const BigInt &b); - void fromHex(const std::string& hex); - - /** - * @brief bigPow10 - * @param pow - * @return number 10 ^ pow - */ - static BigInt bigPow10(unsigned short pow); - - BigInt& toNegative(); - - BigInt& operator = (const BigInt& val); - BigInt& operator = (const std::string &imput); - BigInt& operator = (intMpz val); - - friend BigInt operator + ( BigInt left, const BigInt& right); - friend BigInt operator + ( BigInt left, const std::string &right); - friend BigInt operator + ( BigInt left, intMpz right); - - friend BigInt operator + ( intMpz left, BigInt right); - friend BigInt operator + ( const std::string &left, const BigInt &right); - - friend BigInt& operator += ( BigInt &left, intMpz right); - friend BigInt& operator += ( BigInt &left, const BigInt& right); - friend BigInt& operator += ( BigInt &left, const std::string &right); - - friend BigInt operator - ( BigInt left, const BigInt& right); - friend BigInt operator - ( BigInt left, intMpz right); - friend BigInt operator - ( BigInt left, const std::string &right); - - friend BigInt operator - ( intMpz right, BigInt left); - friend BigInt operator - ( const std::string &right, const BigInt &left); - - friend BigInt operator-(BigInt val); - - friend BigInt& operator -= ( BigInt &left, intMpz right); - - friend BigInt& operator -= ( BigInt &left, const BigInt& right); - friend BigInt& operator -= ( BigInt &left, const std::string &right); - - friend BigInt operator / ( BigInt left, const BigInt& right); - friend BigInt operator / ( BigInt left, const std::string &right); - friend BigInt operator / ( BigInt left, intMpz right); - friend BigInt operator / ( intMpz left, BigInt right); - friend BigInt operator / ( const std::string &left, const BigInt &right); - - friend BigInt& operator /= ( BigInt &left, intMpz right); - friend BigInt& operator /= ( BigInt &left, const std::string &right); - friend BigInt& operator /= ( BigInt &left, const BigInt& right); - - friend BigInt operator * ( BigInt left, const BigInt& right); - friend BigInt operator * ( BigInt left, const std::string &right); - friend BigInt operator * ( BigInt left, intMpz right); - - friend BigInt operator * ( intMpz left, BigInt right); - - friend BigInt& operator *= ( BigInt &left, const BigInt& right); - friend BigInt& operator *= ( BigInt &left, intMpz right); - friend BigInt& operator *= ( BigInt &left, const std::string &right); - - friend BigInt operator % ( BigInt left, const BigInt& right); - friend BigInt operator % ( BigInt left, const std::string &right); - friend BigInt operator % ( BigInt left, intMpz right); - - friend BigInt operator % ( intMpz left, BigInt right); - friend BigInt operator % ( const std::string & left, const BigInt &right); - - friend BigInt& operator %= ( BigInt &left, intMpz right); - friend BigInt& operator %= ( BigInt &left, const std::string &right); - - friend BigInt& operator %= ( BigInt &left, const BigInt& right); - - friend BigInt operator << ( BigInt left, int right); - friend BigInt operator >> ( BigInt left, int right); - - friend BigInt& operator <<= ( BigInt &left, int right); - friend BigInt& operator >>= ( BigInt &left, int right); - - friend bool operator == ( const BigInt& left, const BigInt& right); - friend bool operator == ( const BigInt& left, intMpz right); - friend bool operator == ( const BigInt& left, const std::string& right); - friend bool operator == ( const std::string& left, const BigInt& right); - friend bool operator == ( const BigInt& left, const std::string& right); - friend bool operator == ( intMpz left, const std::string& right); - - friend bool operator != ( const BigInt& left, const BigInt& right); - friend bool operator != ( const BigInt& left, intMpz right); - friend bool operator != ( const BigInt& left, const std::string& str); - friend bool operator != ( const std::string& left, const BigInt& right); - friend bool operator != ( const BigInt& left, const std::string& right); - friend bool operator != ( intMpz left, const std::string& right); - - friend bool operator < ( const BigInt& left, const BigInt& right); - friend bool operator < ( const BigInt& left, intMpz right); - friend bool operator < ( const BigInt& left, const std::string& str); - friend bool operator < ( const std::string& left, const BigInt& right); - friend bool operator < ( const BigInt& left, const std::string& right); - friend bool operator < ( intMpz left, const std::string& right); - - friend bool operator > ( const BigInt& left, const BigInt& right); - friend bool operator > ( const BigInt& left, intMpz right); - friend bool operator > ( const BigInt& left, const std::string& str); - friend bool operator > ( const std::string& left, const BigInt& right); - friend bool operator > ( const BigInt& left, const std::string& right); - friend bool operator > ( intMpz left, const std::string& right); - - friend bool operator <= ( const BigInt& left, const BigInt& right); - friend bool operator <= ( const BigInt& left, intMpz right); - friend bool operator <= ( const BigInt& left, const std::string& str); - friend bool operator <= ( const std::string& left, const BigInt& right); - friend bool operator <= ( const BigInt& left, const std::string& right); - friend bool operator <= ( intMpz left, const std::string& right); - - friend bool operator >= ( const BigInt& left, const BigInt& right); - friend bool operator >= ( const BigInt& left, intMpz right); - friend bool operator >= ( const BigInt& left, const std::string& str); - friend bool operator >= ( const std::string& left, const BigInt& right); - friend bool operator >= ( const BigInt& left, const std::string& right); - friend bool operator >= ( intMpz left, const std::string& right); - - friend bool operator!(const BigInt& val); - - BigInt& operator-- (); - BigInt& operator++ (); - - BigInt operator-- (int); - BigInt operator++ (int); - - - friend BigInt operator~ (BigInt val); - - friend BigInt operator| (BigInt left, const BigInt& right); - friend BigInt operator| (const BigInt &left, intMpz right); - - friend BigInt& operator|= (BigInt &left, const BigInt& right); - friend BigInt& operator|= (BigInt &left, intMpz right); - - friend BigInt operator& (BigInt left, const BigInt& right); - friend BigInt operator& (const BigInt &left, intMpz right); - - friend BigInt& operator&= (BigInt &left, const BigInt& right); - friend BigInt& operator&= (BigInt &left, intMpz right); - - friend BigInt operator^ (BigInt left, const BigInt& right); - friend BigInt operator^ (const BigInt &left, intMpz right); - - friend BigInt& operator^= (BigInt &left, const BigInt& right); - friend BigInt& operator^= (BigInt &left, intMpz right); - -}; - -#endif // BIGINT_H diff --git a/Qt-Secret/GMP/mini-gmp.c b/Qt-Secret/GMP/mini-gmp.c deleted file mode 100644 index f9d598b..0000000 --- a/Qt-Secret/GMP/mini-gmp.c +++ /dev/null @@ -1,4415 +0,0 @@ -/* mini-gmp, a minimalistic implementation of a GNU GMP subset. - - Contributed to the GNU project by Niels Möller - -Copyright 1991-1997, 1999-2016 Free Software Foundation, Inc. - -This file is part of the GNU MP Library. - -The GNU MP Library is free software; you can redistribute it and/or modify -it under the terms of either: - - * the GNU Lesser General Public License as published by the Free - Software Foundation; either version 3 of the License, or (at your - option) any later version. - -or - - * the GNU General Public License as published by the Free Software - Foundation; either version 2 of the License, or (at your option) any - later version. - -or both in parallel, as here. - -The GNU MP Library is distributed in the hope that it will be useful, but -WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY -or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License -for more details. - -You should have received copies of the GNU General Public License and the -GNU Lesser General Public License along with the GNU MP Library. If not, -see https://www.gnu.org/licenses/. */ - -/* NOTE: All functions in this file which are not declared in - mini-gmp.h are internal, and are not intended to be compatible - neither with GMP nor with future versions of mini-gmp. */ - -/* Much of the material copied from GMP files, including: gmp-impl.h, - longlong.h, mpn/generic/add_n.c, mpn/generic/addmul_1.c, - mpn/generic/lshift.c, mpn/generic/mul_1.c, - mpn/generic/mul_basecase.c, mpn/generic/rshift.c, - mpn/generic/sbpi1_div_qr.c, mpn/generic/sub_n.c, - mpn/generic/submul_1.c. */ - -#include -#include -#include -#include -#include -#include - -#include "mini-gmp.h" - - -/* Macros */ -#define GMP_LIMB_BITS (sizeof(mp_limb_t) * CHAR_BIT) - -#define GMP_LIMB_MAX (~ (mp_limb_t) 0) -#define GMP_LIMB_HIGHBIT ((mp_limb_t) 1 << (GMP_LIMB_BITS - 1)) - -#define GMP_HLIMB_BIT ((mp_limb_t) 1 << (GMP_LIMB_BITS / 2)) -#define GMP_LLIMB_MASK (GMP_HLIMB_BIT - 1) - -#define GMP_ULONG_BITS (sizeof(uIntMpz) * CHAR_BIT) -#define GMP_ULONG_HIGHBIT ((uIntMpz) 1 << (GMP_ULONG_BITS - 1)) - -#define GMP_ABS(x) ((x) >= 0 ? (x) : -(x)) -#define GMP_NEG_CAST(T,x) (-((T)((x) + 1) - 1)) - -#define GMP_MIN(a, b) ((a) < (b) ? (a) : (b)) -#define GMP_MAX(a, b) ((a) > (b) ? (a) : (b)) - -#define GMP_CMP(a,b) (((a) > (b)) - ((a) < (b))) - -#define gmp_assert_nocarry(x) do { \ - mp_limb_t __cy = (x); \ - assert (__cy == 0); \ - } while (0) - -#define gmp_clz(count, x) do { \ - mp_limb_t __clz_x = (x); \ - unsigned __clz_c; \ - for (__clz_c = 0; \ - (__clz_x & ((mp_limb_t) 0xff << (GMP_LIMB_BITS - 8))) == 0; \ - __clz_c += 8) \ - __clz_x <<= 8; \ - for (; (__clz_x & GMP_LIMB_HIGHBIT) == 0; __clz_c++) \ - __clz_x <<= 1; \ - (count) = __clz_c; \ - } while (0) - -#define gmp_ctz(count, x) do { \ - mp_limb_t __ctz_x = (x); \ - unsigned __ctz_c = 0; \ - gmp_clz (__ctz_c, __ctz_x & - __ctz_x); \ - (count) = GMP_LIMB_BITS - 1 - __ctz_c; \ - } while (0) - -#define gmp_add_ssaaaa(sh, sl, ah, al, bh, bl) \ - do { \ - mp_limb_t __x; \ - __x = (al) + (bl); \ - (sh) = (ah) + (bh) + (__x < (al)); \ - (sl) = __x; \ - } while (0) - -#define gmp_sub_ddmmss(sh, sl, ah, al, bh, bl) \ - do { \ - mp_limb_t __x; \ - __x = (al) - (bl); \ - (sh) = (ah) - (bh) - ((al) < (bl)); \ - (sl) = __x; \ - } while (0) - -#define gmp_umul_ppmm(w1, w0, u, v) \ - do { \ - mp_limb_t __x0, __x1, __x2, __x3; \ - unsigned __ul, __vl, __uh, __vh; \ - mp_limb_t __u = (u), __v = (v); \ - \ - __ul = __u & GMP_LLIMB_MASK; \ - __uh = __u >> (GMP_LIMB_BITS / 2); \ - __vl = __v & GMP_LLIMB_MASK; \ - __vh = __v >> (GMP_LIMB_BITS / 2); \ - \ - __x0 = (mp_limb_t) __ul * __vl; \ - __x1 = (mp_limb_t) __ul * __vh; \ - __x2 = (mp_limb_t) __uh * __vl; \ - __x3 = (mp_limb_t) __uh * __vh; \ - \ - __x1 += __x0 >> (GMP_LIMB_BITS / 2);/* this can't give carry */ \ - __x1 += __x2; /* but this indeed can */ \ - if (__x1 < __x2) /* did we get it? */ \ - __x3 += GMP_HLIMB_BIT; /* yes, add it in the proper pos. */ \ - \ - (w1) = __x3 + (__x1 >> (GMP_LIMB_BITS / 2)); \ - (w0) = (__x1 << (GMP_LIMB_BITS / 2)) + (__x0 & GMP_LLIMB_MASK); \ - } while (0) - -#define gmp_udiv_qrnnd_preinv(q, r, nh, nl, d, di) \ - do { \ - mp_limb_t _qh, _ql, _r, _mask; \ - gmp_umul_ppmm (_qh, _ql, (nh), (di)); \ - gmp_add_ssaaaa (_qh, _ql, _qh, _ql, (nh) + 1, (nl)); \ - _r = (nl) - _qh * (d); \ - _mask = -(mp_limb_t) (_r > _ql); /* both > and >= are OK */ \ - _qh += _mask; \ - _r += _mask & (d); \ - if (_r >= (d)) \ - { \ - _r -= (d); \ - _qh++; \ - } \ - \ - (r) = _r; \ - (q) = _qh; \ - } while (0) - -#define gmp_udiv_qr_3by2(q, r1, r0, n2, n1, n0, d1, d0, dinv) \ - do { \ - mp_limb_t _q0, _t1, _t0, _mask; \ - gmp_umul_ppmm ((q), _q0, (n2), (dinv)); \ - gmp_add_ssaaaa ((q), _q0, (q), _q0, (n2), (n1)); \ - \ - /* Compute the two most significant limbs of n - q'd */ \ - (r1) = (n1) - (d1) * (q); \ - gmp_sub_ddmmss ((r1), (r0), (r1), (n0), (d1), (d0)); \ - gmp_umul_ppmm (_t1, _t0, (d0), (q)); \ - gmp_sub_ddmmss ((r1), (r0), (r1), (r0), _t1, _t0); \ - (q)++; \ - \ - /* Conditionally adjust q and the remainders */ \ - _mask = - (mp_limb_t) ((r1) >= _q0); \ - (q) += _mask; \ - gmp_add_ssaaaa ((r1), (r0), (r1), (r0), _mask & (d1), _mask & (d0)); \ - if ((r1) >= (d1)) \ - { \ - if ((r1) > (d1) || (r0) >= (d0)) \ - { \ - (q)++; \ - gmp_sub_ddmmss ((r1), (r0), (r1), (r0), (d1), (d0)); \ - } \ - } \ - } while (0) - -/* Swap macros. */ -#define MP_LIMB_T_SWAP(x, y) \ - do { \ - mp_limb_t __mp_limb_t_swap__tmp = (x); \ - (x) = (y); \ - (y) = __mp_limb_t_swap__tmp; \ - } while (0) -#define MP_SIZE_T_SWAP(x, y) \ - do { \ - mp_size_t __mp_size_t_swap__tmp = (x); \ - (x) = (y); \ - (y) = __mp_size_t_swap__tmp; \ - } while (0) -#define MP_BITCNT_T_SWAP(x,y) \ - do { \ - mp_bitcnt_t __mp_bitcnt_t_swap__tmp = (x); \ - (x) = (y); \ - (y) = __mp_bitcnt_t_swap__tmp; \ - } while (0) -#define MP_PTR_SWAP(x, y) \ - do { \ - mp_ptr __mp_ptr_swap__tmp = (x); \ - (x) = (y); \ - (y) = __mp_ptr_swap__tmp; \ - } while (0) -#define MP_SRCPTR_SWAP(x, y) \ - do { \ - mp_srcptr __mp_srcptr_swap__tmp = (x); \ - (x) = (y); \ - (y) = __mp_srcptr_swap__tmp; \ - } while (0) - -#define MPN_PTR_SWAP(xp,xs, yp,ys) \ - do { \ - MP_PTR_SWAP (xp, yp); \ - MP_SIZE_T_SWAP (xs, ys); \ - } while(0) -#define MPN_SRCPTR_SWAP(xp,xs, yp,ys) \ - do { \ - MP_SRCPTR_SWAP (xp, yp); \ - MP_SIZE_T_SWAP (xs, ys); \ - } while(0) - -#define MPZ_PTR_SWAP(x, y) \ - do { \ - mpz_ptr __mpz_ptr_swap__tmp = (x); \ - (x) = (y); \ - (y) = __mpz_ptr_swap__tmp; \ - } while (0) -#define MPZ_SRCPTR_SWAP(x, y) \ - do { \ - mpz_srcptr __mpz_srcptr_swap__tmp = (x); \ - (x) = (y); \ - (y) = __mpz_srcptr_swap__tmp; \ - } while (0) - -const int mp_bits_per_limb = GMP_LIMB_BITS; - - -/* Memory allocation and other helper functions. */ -static void -gmp_die (const char *msg) -{ - fprintf (stderr, "%s\n", msg); - abort(); -} - -static void * -gmp_default_alloc (size_t size) -{ - void *p; - - assert (size > 0); - - p = malloc (size); - if (!p) - gmp_die("gmp_default_alloc: Virtual memory exhausted."); - - return p; -} - -static void * -gmp_default_realloc (void *old, size_t old_size, size_t new_size) -{ - UN_USED(old_size); - - void * p; - - p = realloc (old, new_size); - - if (!p) - gmp_die("gmp_default_realloc: Virtual memory exhausted."); - - return p; -} - -static void -gmp_default_free (void *p, size_t size) -{ - UN_USED(size); - free (p); -} - -static void * (*gmp_allocate_func) (size_t) = gmp_default_alloc; -static void * (*gmp_reallocate_func) (void *, size_t, size_t) = gmp_default_realloc; -static void (*gmp_free_func) (void *, size_t) = gmp_default_free; - -void -mp_get_memory_functions (void *(**alloc_func) (size_t), - void *(**realloc_func) (void *, size_t, size_t), - void (**free_func) (void *, size_t)) -{ - if (alloc_func) - *alloc_func = gmp_allocate_func; - - if (realloc_func) - *realloc_func = gmp_reallocate_func; - - if (free_func) - *free_func = gmp_free_func; -} - -void -mp_set_memory_functions (void *(*alloc_func) (size_t), - void *(*realloc_func) (void *, size_t, size_t), - void (*free_func) (void *, size_t)) -{ - if (!alloc_func) - alloc_func = gmp_default_alloc; - if (!realloc_func) - realloc_func = gmp_default_realloc; - if (!free_func) - free_func = gmp_default_free; - - gmp_allocate_func = alloc_func; - gmp_reallocate_func = realloc_func; - gmp_free_func = free_func; -} - -#define gmp_xalloc(size) ((*gmp_allocate_func)((size))) -#define gmp_free(p) ((*gmp_free_func) ((p), 0)) - -static mp_ptr -gmp_xalloc_limbs (mp_size_t size) -{ - return (mp_ptr) gmp_xalloc (size * sizeof (mp_limb_t)); -} - -static mp_ptr -gmp_xrealloc_limbs (mp_ptr old, mp_size_t size) -{ - assert (size > 0); - return (mp_ptr) (*gmp_reallocate_func) (old, 0, size * sizeof (mp_limb_t)); -} - - -/* MPN interface */ - -void -mpn_copyi (mp_ptr d, mp_srcptr s, mp_size_t n) -{ - mp_size_t i; - for (i = 0; i < n; i++) - d[i] = s[i]; -} - -void -mpn_copyd (mp_ptr d, mp_srcptr s, mp_size_t n) -{ - while (--n >= 0) - d[n] = s[n]; -} - -int -mpn_cmp (mp_srcptr ap, mp_srcptr bp, mp_size_t n) -{ - while (--n >= 0) - { - if (ap[n] != bp[n]) - return ap[n] > bp[n] ? 1 : -1; - } - return 0; -} - -static int -mpn_cmp4 (mp_srcptr ap, mp_size_t an, mp_srcptr bp, mp_size_t bn) -{ - if (an != bn) - return an < bn ? -1 : 1; - else - return mpn_cmp (ap, bp, an); -} - -static mp_size_t -mpn_normalized_size (mp_srcptr xp, mp_size_t n) -{ - while (n > 0 && xp[n-1] == 0) - --n; - return n; -} - -int -mpn_zero_p(mp_srcptr rp, mp_size_t n) -{ - return mpn_normalized_size (rp, n) == 0; -} - -void -mpn_zero (mp_ptr rp, mp_size_t n) -{ - while (--n >= 0) - rp[n] = 0; -} - -mp_limb_t -mpn_add_1 (mp_ptr rp, mp_srcptr ap, mp_size_t n, mp_limb_t b) -{ - mp_size_t i; - - assert (n > 0); - i = 0; - do - { - mp_limb_t r = ap[i] + b; - /* Carry out */ - b = (r < b); - rp[i] = r; - } - while (++i < n); - - return b; -} - -mp_limb_t -mpn_add_n (mp_ptr rp, mp_srcptr ap, mp_srcptr bp, mp_size_t n) -{ - mp_size_t i; - mp_limb_t cy; - - for (i = 0, cy = 0; i < n; i++) - { - mp_limb_t a, b, r; - a = ap[i]; b = bp[i]; - r = a + cy; - cy = (r < cy); - r += b; - cy += (r < b); - rp[i] = r; - } - return cy; -} - -mp_limb_t -mpn_add (mp_ptr rp, mp_srcptr ap, mp_size_t an, mp_srcptr bp, mp_size_t bn) -{ - mp_limb_t cy; - - assert (an >= bn); - - cy = mpn_add_n (rp, ap, bp, bn); - if (an > bn) - cy = mpn_add_1 (rp + bn, ap + bn, an - bn, cy); - return cy; -} - -mp_limb_t -mpn_sub_1 (mp_ptr rp, mp_srcptr ap, mp_size_t n, mp_limb_t b) -{ - mp_size_t i; - - assert (n > 0); - - i = 0; - do - { - mp_limb_t a = ap[i]; - /* Carry out */ - mp_limb_t cy = a < b; - rp[i] = a - b; - b = cy; - } - while (++i < n); - - return b; -} - -mp_limb_t -mpn_sub_n (mp_ptr rp, mp_srcptr ap, mp_srcptr bp, mp_size_t n) -{ - mp_size_t i; - mp_limb_t cy; - - for (i = 0, cy = 0; i < n; i++) - { - mp_limb_t a, b; - a = ap[i]; b = bp[i]; - b += cy; - cy = (b < cy); - cy += (a < b); - rp[i] = a - b; - } - return cy; -} - -mp_limb_t -mpn_sub (mp_ptr rp, mp_srcptr ap, mp_size_t an, mp_srcptr bp, mp_size_t bn) -{ - mp_limb_t cy; - - assert (an >= bn); - - cy = mpn_sub_n (rp, ap, bp, bn); - if (an > bn) - cy = mpn_sub_1 (rp + bn, ap + bn, an - bn, cy); - return cy; -} - -mp_limb_t -mpn_mul_1 (mp_ptr rp, mp_srcptr up, mp_size_t n, mp_limb_t vl) -{ - mp_limb_t ul, cl, hpl, lpl; - - assert (n >= 1); - - cl = 0; - do - { - ul = *up++; - gmp_umul_ppmm (hpl, lpl, ul, vl); - - lpl += cl; - cl = (lpl < cl) + hpl; - - *rp++ = lpl; - } - while (--n != 0); - - return cl; -} - -mp_limb_t -mpn_addmul_1 (mp_ptr rp, mp_srcptr up, mp_size_t n, mp_limb_t vl) -{ - mp_limb_t ul, cl, hpl, lpl, rl; - - assert (n >= 1); - - cl = 0; - do - { - ul = *up++; - gmp_umul_ppmm (hpl, lpl, ul, vl); - - lpl += cl; - cl = (lpl < cl) + hpl; - - rl = *rp; - lpl = rl + lpl; - cl += lpl < rl; - *rp++ = lpl; - } - while (--n != 0); - - return cl; -} - -mp_limb_t -mpn_submul_1 (mp_ptr rp, mp_srcptr up, mp_size_t n, mp_limb_t vl) -{ - mp_limb_t ul, cl, hpl, lpl, rl; - - assert (n >= 1); - - cl = 0; - do - { - ul = *up++; - gmp_umul_ppmm (hpl, lpl, ul, vl); - - lpl += cl; - cl = (lpl < cl) + hpl; - - rl = *rp; - lpl = rl - lpl; - cl += lpl > rl; - *rp++ = lpl; - } - while (--n != 0); - - return cl; -} - -mp_limb_t -mpn_mul (mp_ptr rp, mp_srcptr up, mp_size_t un, mp_srcptr vp, mp_size_t vn) -{ - assert (un >= vn); - assert (vn >= 1); - - /* We first multiply by the low order limb. This result can be - stored, not added, to rp. We also avoid a loop for zeroing this - way. */ - - rp[un] = mpn_mul_1 (rp, up, un, vp[0]); - - /* Now accumulate the product of up[] and the next higher limb from - vp[]. */ - - while (--vn >= 1) - { - rp += 1, vp += 1; - rp[un] = mpn_addmul_1 (rp, up, un, vp[0]); - } - return rp[un]; -} - -void -mpn_mul_n (mp_ptr rp, mp_srcptr ap, mp_srcptr bp, mp_size_t n) -{ - mpn_mul (rp, ap, n, bp, n); -} - -void -mpn_sqr (mp_ptr rp, mp_srcptr ap, mp_size_t n) -{ - mpn_mul (rp, ap, n, ap, n); -} - -mp_limb_t -mpn_lshift (mp_ptr rp, mp_srcptr up, mp_size_t n, unsigned int cnt) -{ - mp_limb_t high_limb, low_limb; - unsigned int tnc; - mp_limb_t retval; - - assert (n >= 1); - assert (cnt >= 1); - assert (cnt < GMP_LIMB_BITS); - - up += n; - rp += n; - - tnc = GMP_LIMB_BITS - cnt; - low_limb = *--up; - retval = low_limb >> tnc; - high_limb = (low_limb << cnt); - - while (--n != 0) - { - low_limb = *--up; - *--rp = high_limb | (low_limb >> tnc); - high_limb = (low_limb << cnt); - } - *--rp = high_limb; - - return retval; -} - -mp_limb_t -mpn_rshift (mp_ptr rp, mp_srcptr up, mp_size_t n, unsigned int cnt) -{ - mp_limb_t high_limb, low_limb; - unsigned int tnc; - mp_limb_t retval; - - assert (n >= 1); - assert (cnt >= 1); - assert (cnt < GMP_LIMB_BITS); - - tnc = GMP_LIMB_BITS - cnt; - high_limb = *up++; - retval = (high_limb << tnc); - low_limb = high_limb >> cnt; - - while (--n != 0) - { - high_limb = *up++; - *rp++ = low_limb | (high_limb << tnc); - low_limb = high_limb >> cnt; - } - *rp = low_limb; - - return retval; -} - -static mp_bitcnt_t -mpn_common_scan (mp_limb_t limb, mp_size_t i, mp_srcptr up, mp_size_t un, - mp_limb_t ux) -{ - unsigned cnt; - - assert (ux == 0 || ux == GMP_LIMB_MAX); - assert (0 <= i && i <= un ); - - while (limb == 0) - { - i++; - if (i == un) - return (ux == 0 ? ~(mp_bitcnt_t) 0 : un * GMP_LIMB_BITS); - limb = ux ^ up[i]; - } - gmp_ctz (cnt, limb); - return (mp_bitcnt_t) i * GMP_LIMB_BITS + cnt; -} - -mp_bitcnt_t -mpn_scan1 (mp_srcptr ptr, mp_bitcnt_t bit) -{ - mp_size_t i; - i = bit / GMP_LIMB_BITS; - - return mpn_common_scan ( ptr[i] & (GMP_LIMB_MAX << (bit % GMP_LIMB_BITS)), - i, ptr, i, 0); -} - -mp_bitcnt_t -mpn_scan0 (mp_srcptr ptr, mp_bitcnt_t bit) -{ - mp_size_t i; - i = bit / GMP_LIMB_BITS; - - return mpn_common_scan (~ptr[i] & (GMP_LIMB_MAX << (bit % GMP_LIMB_BITS)), - i, ptr, i, GMP_LIMB_MAX); -} - -void -mpn_com (mp_ptr rp, mp_srcptr up, mp_size_t n) -{ - while (--n >= 0) - *rp++ = ~ *up++; -} - -mp_limb_t -mpn_neg (mp_ptr rp, mp_srcptr up, mp_size_t n) -{ - while (*up == 0) - { - *rp = 0; - if (!--n) - return 0; - ++up; ++rp; - } - *rp = - *up; - mpn_com (++rp, ++up, --n); - return 1; -} - - -/* MPN division interface. */ - -/* The 3/2 inverse is defined as - - m = floor( (B^3-1) / (B u1 + u0)) - B -*/ -mp_limb_t -mpn_invert_3by2 (mp_limb_t u1, mp_limb_t u0) -{ - mp_limb_t r, p, m, ql; - unsigned ul, uh, qh; - - assert (u1 >= GMP_LIMB_HIGHBIT); - - /* For notation, let b denote the half-limb base, so that B = b^2. - Split u1 = b uh + ul. */ - ul = u1 & GMP_LLIMB_MASK; - uh = u1 >> (GMP_LIMB_BITS / 2); - - /* Approximation of the high half of quotient. Differs from the 2/1 - inverse of the half limb uh, since we have already subtracted - u0. */ - qh = ~u1 / uh; - - /* Adjust to get a half-limb 3/2 inverse, i.e., we want - - qh' = floor( (b^3 - 1) / u) - b = floor ((b^3 - b u - 1) / u - = floor( (b (~u) + b-1) / u), - - and the remainder - - r = b (~u) + b-1 - qh (b uh + ul) - = b (~u - qh uh) + b-1 - qh ul - - Subtraction of qh ul may underflow, which implies adjustments. - But by normalization, 2 u >= B > qh ul, so we need to adjust by - at most 2. - */ - - r = ((~u1 - (mp_limb_t) qh * uh) << (GMP_LIMB_BITS / 2)) | GMP_LLIMB_MASK; - - p = (mp_limb_t) qh * ul; - /* Adjustment steps taken from udiv_qrnnd_c */ - if (r < p) - { - qh--; - r += u1; - if (r >= u1) /* i.e. we didn't get carry when adding to r */ - if (r < p) - { - qh--; - r += u1; - } - } - r -= p; - - /* Low half of the quotient is - - ql = floor ( (b r + b-1) / u1). - - This is a 3/2 division (on half-limbs), for which qh is a - suitable inverse. */ - - p = (r >> (GMP_LIMB_BITS / 2)) * qh + r; - /* Unlike full-limb 3/2, we can add 1 without overflow. For this to - work, it is essential that ql is a full mp_limb_t. */ - ql = (p >> (GMP_LIMB_BITS / 2)) + 1; - - /* By the 3/2 trick, we don't need the high half limb. */ - r = (r << (GMP_LIMB_BITS / 2)) + GMP_LLIMB_MASK - ql * u1; - - if (r >= (p << (GMP_LIMB_BITS / 2))) - { - ql--; - r += u1; - } - m = ((mp_limb_t) qh << (GMP_LIMB_BITS / 2)) + ql; - if (r >= u1) - { - m++; - r -= u1; - } - - /* Now m is the 2/1 invers of u1. If u0 > 0, adjust it to become a - 3/2 inverse. */ - if (u0 > 0) - { - mp_limb_t th, tl; - r = ~r; - r += u0; - if (r < u0) - { - m--; - if (r >= u1) - { - m--; - r -= u1; - } - r -= u1; - } - gmp_umul_ppmm (th, tl, u0, m); - r += th; - if (r < th) - { - m--; - m -= ((r > u1) | ((r == u1) & (tl > u0))); - } - } - - return m; -} - -struct gmp_div_inverse -{ - /* Normalization shift count. */ - unsigned shift; - /* Normalized divisor (d0 unused for mpn_div_qr_1) */ - mp_limb_t d1, d0; - /* Inverse, for 2/1 or 3/2. */ - mp_limb_t di; -}; - -static void -mpn_div_qr_1_invert (struct gmp_div_inverse *inv, mp_limb_t d) -{ - unsigned shift; - - assert (d > 0); - gmp_clz (shift, d); - inv->shift = shift; - inv->d1 = d << shift; - inv->di = mpn_invert_limb (inv->d1); -} - -static void -mpn_div_qr_2_invert (struct gmp_div_inverse *inv, - mp_limb_t d1, mp_limb_t d0) -{ - unsigned shift; - - assert (d1 > 0); - gmp_clz (shift, d1); - inv->shift = shift; - if (shift > 0) - { - d1 = (d1 << shift) | (d0 >> (GMP_LIMB_BITS - shift)); - d0 <<= shift; - } - inv->d1 = d1; - inv->d0 = d0; - inv->di = mpn_invert_3by2 (d1, d0); -} - -static void -mpn_div_qr_invert (struct gmp_div_inverse *inv, - mp_srcptr dp, mp_size_t dn) -{ - assert (dn > 0); - - if (dn == 1) - mpn_div_qr_1_invert (inv, dp[0]); - else if (dn == 2) - mpn_div_qr_2_invert (inv, dp[1], dp[0]); - else - { - unsigned shift; - mp_limb_t d1, d0; - - d1 = dp[dn-1]; - d0 = dp[dn-2]; - assert (d1 > 0); - gmp_clz (shift, d1); - inv->shift = shift; - if (shift > 0) - { - d1 = (d1 << shift) | (d0 >> (GMP_LIMB_BITS - shift)); - d0 = (d0 << shift) | (dp[dn-3] >> (GMP_LIMB_BITS - shift)); - } - inv->d1 = d1; - inv->d0 = d0; - inv->di = mpn_invert_3by2 (d1, d0); - } -} - -/* Not matching current public gmp interface, rather corresponding to - the sbpi1_div_* functions. */ -static mp_limb_t -mpn_div_qr_1_preinv (mp_ptr qp, mp_srcptr np, mp_size_t nn, - const struct gmp_div_inverse *inv) -{ - mp_limb_t d, di; - mp_limb_t r; - mp_ptr tp = NULL; - - if (inv->shift > 0) - { - tp = gmp_xalloc_limbs (nn); - r = mpn_lshift (tp, np, nn, inv->shift); - np = tp; - } - else - r = 0; - - d = inv->d1; - di = inv->di; - while (--nn >= 0) - { - mp_limb_t q; - - gmp_udiv_qrnnd_preinv (q, r, r, np[nn], d, di); - if (qp) - qp[nn] = q; - } - if (inv->shift > 0) - gmp_free (tp); - - return r >> inv->shift; -} - -static mp_limb_t -mpn_div_qr_1 (mp_ptr qp, mp_srcptr np, mp_size_t nn, mp_limb_t d) -{ - assert (d > 0); - - /* Special case for powers of two. */ - if ((d & (d-1)) == 0) - { - mp_limb_t r = np[0] & (d-1); - if (qp) - { - if (d <= 1) - mpn_copyi (qp, np, nn); - else - { - unsigned shift; - gmp_ctz (shift, d); - mpn_rshift (qp, np, nn, shift); - } - } - return r; - } - else - { - struct gmp_div_inverse inv; - mpn_div_qr_1_invert (&inv, d); - return mpn_div_qr_1_preinv (qp, np, nn, &inv); - } -} - -static void -mpn_div_qr_2_preinv (mp_ptr qp, mp_ptr rp, mp_srcptr np, mp_size_t nn, - const struct gmp_div_inverse *inv) -{ - unsigned shift; - mp_size_t i; - mp_limb_t d1, d0, di, r1, r0; - mp_ptr tp; - - assert (nn >= 2); - shift = inv->shift; - d1 = inv->d1; - d0 = inv->d0; - di = inv->di; - - if (shift > 0) - { - tp = gmp_xalloc_limbs (nn); - r1 = mpn_lshift (tp, np, nn, shift); - np = tp; - } - else - r1 = 0; - - r0 = np[nn - 1]; - - i = nn - 2; - do - { - mp_limb_t n0, q; - n0 = np[i]; - gmp_udiv_qr_3by2 (q, r1, r0, r1, r0, n0, d1, d0, di); - - if (qp) - qp[i] = q; - } - while (--i >= 0); - - if (shift > 0) - { - assert ((r0 << (GMP_LIMB_BITS - shift)) == 0); - r0 = (r0 >> shift) | (r1 << (GMP_LIMB_BITS - shift)); - r1 >>= shift; - - gmp_free (tp); - } - - rp[1] = r1; - rp[0] = r0; -} - -#if 0 -static void -mpn_div_qr_2 (mp_ptr qp, mp_ptr rp, mp_srcptr np, mp_size_t nn, - mp_limb_t d1, mp_limb_t d0) -{ - struct gmp_div_inverse inv; - assert (nn >= 2); - - mpn_div_qr_2_invert (&inv, d1, d0); - mpn_div_qr_2_preinv (qp, rp, np, nn, &inv); -} -#endif - -static void -mpn_div_qr_pi1 (mp_ptr qp, - mp_ptr np, mp_size_t nn, mp_limb_t n1, - mp_srcptr dp, mp_size_t dn, - mp_limb_t dinv) -{ - mp_size_t i; - - mp_limb_t d1, d0; - mp_limb_t cy, cy1; - mp_limb_t q; - - assert (dn > 2); - assert (nn >= dn); - - d1 = dp[dn - 1]; - d0 = dp[dn - 2]; - - assert ((d1 & GMP_LIMB_HIGHBIT) != 0); - /* Iteration variable is the index of the q limb. - * - * We divide - * by - */ - - i = nn - dn; - do - { - mp_limb_t n0 = np[dn-1+i]; - - if (n1 == d1 && n0 == d0) - { - q = GMP_LIMB_MAX; - mpn_submul_1 (np+i, dp, dn, q); - n1 = np[dn-1+i]; /* update n1, last loop's value will now be invalid */ - } - else - { - gmp_udiv_qr_3by2 (q, n1, n0, n1, n0, np[dn-2+i], d1, d0, dinv); - - cy = mpn_submul_1 (np + i, dp, dn-2, q); - - cy1 = n0 < cy; - n0 = n0 - cy; - cy = n1 < cy1; - n1 = n1 - cy1; - np[dn-2+i] = n0; - - if (cy != 0) - { - n1 += d1 + mpn_add_n (np + i, np + i, dp, dn - 1); - q--; - } - } - - if (qp) - qp[i] = q; - } - while (--i >= 0); - - np[dn - 1] = n1; -} - -static void -mpn_div_qr_preinv (mp_ptr qp, mp_ptr np, mp_size_t nn, - mp_srcptr dp, mp_size_t dn, - const struct gmp_div_inverse *inv) -{ - assert (dn > 0); - assert (nn >= dn); - - if (dn == 1) - np[0] = mpn_div_qr_1_preinv (qp, np, nn, inv); - else if (dn == 2) - mpn_div_qr_2_preinv (qp, np, np, nn, inv); - else - { - mp_limb_t nh; - unsigned shift; - - assert (inv->d1 == dp[dn-1]); - assert (inv->d0 == dp[dn-2]); - assert ((inv->d1 & GMP_LIMB_HIGHBIT) != 0); - - shift = inv->shift; - if (shift > 0) - nh = mpn_lshift (np, np, nn, shift); - else - nh = 0; - - mpn_div_qr_pi1 (qp, np, nn, nh, dp, dn, inv->di); - - if (shift > 0) - gmp_assert_nocarry (mpn_rshift (np, np, dn, shift)); - } -} - -static void -mpn_div_qr (mp_ptr qp, mp_ptr np, mp_size_t nn, mp_srcptr dp, mp_size_t dn) -{ - struct gmp_div_inverse inv; - mp_ptr tp = NULL; - - assert (dn > 0); - assert (nn >= dn); - - mpn_div_qr_invert (&inv, dp, dn); - if (dn > 2 && inv.shift > 0) - { - tp = gmp_xalloc_limbs (dn); - gmp_assert_nocarry (mpn_lshift (tp, dp, dn, inv.shift)); - dp = tp; - } - mpn_div_qr_preinv (qp, np, nn, dp, dn, &inv); - if (tp) - gmp_free (tp); -} - - -/* MPN base conversion. */ -static unsigned -mpn_base_power_of_two_p (unsigned b) -{ - switch (b) - { - case 2: return 1; - case 4: return 2; - case 8: return 3; - case 16: return 4; - case 32: return 5; - case 64: return 6; - case 128: return 7; - case 256: return 8; - default: return 0; - } -} - -struct mpn_base_info -{ - /* bb is the largest power of the base which fits in one limb, and - exp is the corresponding exponent. */ - unsigned exp; - mp_limb_t bb; -}; - -static void -mpn_get_base_info (struct mpn_base_info *info, mp_limb_t b) -{ - mp_limb_t m; - mp_limb_t p; - unsigned exp; - - m = GMP_LIMB_MAX / b; - for (exp = 1, p = b; p <= m; exp++) - p *= b; - - info->exp = exp; - info->bb = p; -} - -static mp_bitcnt_t -mpn_limb_size_in_base_2 (mp_limb_t u) -{ - unsigned shift; - - assert (u > 0); - gmp_clz (shift, u); - return GMP_LIMB_BITS - shift; -} - -static size_t -mpn_get_str_bits (unsigned char *sp, unsigned bits, mp_srcptr up, mp_size_t un) -{ - unsigned char mask; - size_t sn, j; - mp_size_t i; - unsigned shift; - - sn = ((un - 1) * GMP_LIMB_BITS + mpn_limb_size_in_base_2 (up[un-1]) - + bits - 1) / bits; - - mask = (1U << bits) - 1; - - for (i = 0, j = sn, shift = 0; j-- > 0;) - { - unsigned char digit = up[i] >> shift; - - shift += bits; - - if (shift >= GMP_LIMB_BITS && ++i < un) - { - shift -= GMP_LIMB_BITS; - digit |= up[i] << (bits - shift); - } - sp[j] = digit & mask; - } - return sn; -} - -/* We generate digits from the least significant end, and reverse at - the end. */ -static size_t -mpn_limb_get_str (unsigned char *sp, mp_limb_t w, - const struct gmp_div_inverse *binv) -{ - mp_size_t i; - for (i = 0; w > 0; i++) - { - mp_limb_t h, l, r; - - h = w >> (GMP_LIMB_BITS - binv->shift); - l = w << binv->shift; - - gmp_udiv_qrnnd_preinv (w, r, h, l, binv->d1, binv->di); - assert ( (r << (GMP_LIMB_BITS - binv->shift)) == 0); - r >>= binv->shift; - - sp[i] = r; - } - return i; -} - -static size_t -mpn_get_str_other (unsigned char *sp, - int base, const struct mpn_base_info *info, - mp_ptr up, mp_size_t un) -{ - struct gmp_div_inverse binv; - size_t sn; - size_t i; - - mpn_div_qr_1_invert (&binv, base); - - sn = 0; - - if (un > 1) - { - struct gmp_div_inverse bbinv; - mpn_div_qr_1_invert (&bbinv, info->bb); - - do - { - mp_limb_t w; - size_t done; - w = mpn_div_qr_1_preinv (up, up, un, &bbinv); - un -= (up[un-1] == 0); - done = mpn_limb_get_str (sp + sn, w, &binv); - - for (sn += done; done < info->exp; done++) - sp[sn++] = 0; - } - while (un > 1); - } - sn += mpn_limb_get_str (sp + sn, up[0], &binv); - - /* Reverse order */ - for (i = 0; 2*i + 1 < sn; i++) - { - unsigned char t = sp[i]; - sp[i] = sp[sn - i - 1]; - sp[sn - i - 1] = t; - } - - return sn; -} - -size_t -mpn_get_str (unsigned char *sp, int base, mp_ptr up, mp_size_t un) -{ - unsigned bits; - - assert (un > 0); - assert (up[un-1] > 0); - - bits = mpn_base_power_of_two_p (base); - if (bits) - return mpn_get_str_bits (sp, bits, up, un); - else - { - struct mpn_base_info info; - - mpn_get_base_info (&info, base); - return mpn_get_str_other (sp, base, &info, up, un); - } -} - -static mp_size_t -mpn_set_str_bits (mp_ptr rp, const unsigned char *sp, size_t sn, - unsigned bits) -{ - mp_size_t rn; - size_t j; - unsigned shift; - - for (j = sn, rn = 0, shift = 0; j-- > 0; ) - { - if (shift == 0) - { - rp[rn++] = sp[j]; - shift += bits; - } - else - { - rp[rn-1] |= (mp_limb_t) sp[j] << shift; - shift += bits; - if (shift >= GMP_LIMB_BITS) - { - shift -= GMP_LIMB_BITS; - if (shift > 0) - rp[rn++] = (mp_limb_t) sp[j] >> (bits - shift); - } - } - } - rn = mpn_normalized_size (rp, rn); - return rn; -} - -/* Result is usually normalized, except for all-zero input, in which - case a single zero limb is written at *RP, and 1 is returned. */ -static mp_size_t -mpn_set_str_other (mp_ptr rp, const unsigned char *sp, size_t sn, - mp_limb_t b, const struct mpn_base_info *info) -{ - mp_size_t rn; - mp_limb_t w; - unsigned k; - size_t j; - - assert (sn > 0); - - k = 1 + (sn - 1) % info->exp; - - j = 0; - w = sp[j++]; - while (--k != 0) - w = w * b + sp[j++]; - - rp[0] = w; - - for (rn = 1; j < sn;) - { - mp_limb_t cy; - - w = sp[j++]; - for (k = 1; k < info->exp; k++) - w = w * b + sp[j++]; - - cy = mpn_mul_1 (rp, rp, rn, info->bb); - cy += mpn_add_1 (rp, rp, rn, w); - if (cy > 0) - rp[rn++] = cy; - } - assert (j == sn); - - return rn; -} - -mp_size_t -mpn_set_str (mp_ptr rp, const unsigned char *sp, size_t sn, int base) -{ - unsigned bits; - - if (sn == 0) - return 0; - - bits = mpn_base_power_of_two_p (base); - if (bits) - return mpn_set_str_bits (rp, sp, sn, bits); - else - { - struct mpn_base_info info; - - mpn_get_base_info (&info, base); - return mpn_set_str_other (rp, sp, sn, base, &info); - } -} - - -/* MPZ interface */ -void -mpz_init (mpz_t r) -{ - static const mp_limb_t dummy_limb = 0xc1a0; - - r->_mp_alloc = 0; - r->_mp_size = 0; - r->_mp_d = (mp_ptr) &dummy_limb; -} - -/* The utility of this function is a bit limited, since many functions - assigns the result variable using mpz_swap. */ -void -mpz_init2 (mpz_t r, mp_bitcnt_t bits) -{ - mp_size_t rn; - - bits -= (bits != 0); /* Round down, except if 0 */ - rn = 1 + bits / GMP_LIMB_BITS; - - r->_mp_alloc = rn; - r->_mp_size = 0; - r->_mp_d = gmp_xalloc_limbs (rn); -} - -void -mpz_clear (mpz_t r) -{ - if (r->_mp_alloc) - gmp_free (r->_mp_d); -} - -static mp_ptr -mpz_realloc (mpz_t r, mp_size_t size) -{ - size = GMP_MAX (size, 1); - - if (r->_mp_alloc) - r->_mp_d = gmp_xrealloc_limbs (r->_mp_d, size); - else - r->_mp_d = gmp_xalloc_limbs (size); - r->_mp_alloc = size; - - if (GMP_ABS (r->_mp_size) > size) - r->_mp_size = 0; - - return r->_mp_d; -} - -/* Realloc for an mpz_t WHAT if it has less than NEEDED limbs. */ -#define MPZ_REALLOC(z,n) ((n) > (z)->_mp_alloc \ - ? mpz_realloc(z,n) \ - : (z)->_mp_d) - -/* MPZ assignment and basic conversions. */ -void -mpz_set_si (mpz_t r, intMpz x) -{ - if (x >= 0) - mpz_set_ui (r, x); - else /* (x < 0) */ - { - r->_mp_size = -1; - MPZ_REALLOC (r, 1)[0] = GMP_NEG_CAST (uIntMpz, x); - } -} - -void -mpz_set_ui (mpz_t r, uIntMpz x) -{ - if (x > 0) - { - r->_mp_size = 1; - MPZ_REALLOC (r, 1)[0] = x; - } - else - r->_mp_size = 0; -} - -void -mpz_set (mpz_t r, const mpz_t x) -{ - /* Allow the NOP r == x */ - if (r != x) - { - mp_size_t n; - mp_ptr rp; - - n = GMP_ABS (x->_mp_size); - rp = MPZ_REALLOC (r, n); - - mpn_copyi (rp, x->_mp_d, n); - r->_mp_size = x->_mp_size; - } -} - -void -mpz_init_set_si (mpz_t r, intMpz x) -{ - mpz_init (r); - mpz_set_si (r, x); -} - -void -mpz_init_set_ui (mpz_t r, uIntMpz x) -{ - mpz_init (r); - mpz_set_ui (r, x); -} - -void -mpz_init_set (mpz_t r, const mpz_t x) -{ - mpz_init (r); - mpz_set (r, x); -} - -int -mpz_fits_slong_p (const mpz_t u) -{ - mp_size_t us = u->_mp_size; - - if (us == 1) - return u->_mp_d[0] < GMP_LIMB_HIGHBIT; - else if (us == -1) - return u->_mp_d[0] <= GMP_LIMB_HIGHBIT; - else - return (us == 0); -} - -int -mpz_fits_ulong_p (const mpz_t u) -{ - mp_size_t us = u->_mp_size; - - return (us == (us > 0)); -} - - intMpz -mpz_get_si (const mpz_t u) -{ - if (u->_mp_size < 0) - /* This expression is necessary to properly handle 0x80000000 */ - return -1 - (long) ((u->_mp_d[0] - 1) & ~GMP_LIMB_HIGHBIT); - else - return (long) (mpz_get_ui (u) & ~GMP_LIMB_HIGHBIT); -} - -uIntMpz -mpz_get_ui (const mpz_t u) -{ - return u->_mp_size == 0 ? 0 : u->_mp_d[0]; -} - -size_t -mpz_size (const mpz_t u) -{ - return GMP_ABS (u->_mp_size); -} - -mp_limb_t -mpz_getlimbn (const mpz_t u, mp_size_t n) -{ - if (n >= 0 && n < GMP_ABS (u->_mp_size)) - return u->_mp_d[n]; - else - return 0; -} - -void -mpz_realloc2 (mpz_t x, mp_bitcnt_t n) -{ - mpz_realloc (x, 1 + (n - (n != 0)) / GMP_LIMB_BITS); -} - -mp_srcptr -mpz_limbs_read (mpz_srcptr x) -{ - return x->_mp_d; -} - -mp_ptr -mpz_limbs_modify (mpz_t x, mp_size_t n) -{ - assert (n > 0); - return MPZ_REALLOC (x, n); -} - -mp_ptr -mpz_limbs_write (mpz_t x, mp_size_t n) -{ - return mpz_limbs_modify (x, n); -} - -void -mpz_limbs_finish (mpz_t x, mp_size_t xs) -{ - mp_size_t xn; - xn = mpn_normalized_size (x->_mp_d, GMP_ABS (xs)); - x->_mp_size = xs < 0 ? -xn : xn; -} - -mpz_srcptr -mpz_roinit_n (mpz_t x, mp_srcptr xp, mp_size_t xs) -{ - x->_mp_alloc = 0; - x->_mp_d = (mp_ptr) xp; - mpz_limbs_finish (x, xs); - return x; -} - - -/* Conversions and comparison to double. */ -void -mpz_set_d (mpz_t r, double x) -{ - int sign; - mp_ptr rp; - mp_size_t rn, i; - double B; - double Bi; - mp_limb_t f; - - /* x != x is true when x is a NaN, and x == x * 0.5 is true when x is - zero or infinity. */ - if (x != x || x == x * 0.5) - { - r->_mp_size = 0; - return; - } - - sign = x < 0.0 ; - if (sign) - x = - x; - - if (x < 1.0) - { - r->_mp_size = 0; - return; - } - B = 2.0 * (double) GMP_LIMB_HIGHBIT; - Bi = 1.0 / B; - for (rn = 1; x >= B; rn++) - x *= Bi; - - rp = MPZ_REALLOC (r, rn); - - f = (mp_limb_t) x; - x -= f; - assert (x < 1.0); - i = rn-1; - rp[i] = f; - while (--i >= 0) - { - x = B * x; - f = (mp_limb_t) x; - x -= f; - assert (x < 1.0); - rp[i] = f; - } - - r->_mp_size = sign ? - rn : rn; -} - -void -mpz_init_set_d (mpz_t r, double x) -{ - mpz_init (r); - mpz_set_d (r, x); -} - -double -mpz_get_d (const mpz_t u) -{ - mp_size_t un; - double x; - double B = 2.0 * (double) GMP_LIMB_HIGHBIT; - - un = GMP_ABS (u->_mp_size); - - if (un == 0) - return 0.0; - - x = u->_mp_d[--un]; - while (un > 0) - x = B*x + u->_mp_d[--un]; - - if (u->_mp_size < 0) - x = -x; - - return x; -} - -int -mpz_cmpabs_d (const mpz_t x, double d) -{ - mp_size_t xn; - double B, Bi; - mp_size_t i; - - xn = x->_mp_size; - d = GMP_ABS (d); - - if (xn != 0) - { - xn = GMP_ABS (xn); - - B = 2.0 * (double) GMP_LIMB_HIGHBIT; - Bi = 1.0 / B; - - /* Scale d so it can be compared with the top limb. */ - for (i = 1; i < xn; i++) - d *= Bi; - - if (d >= B) - return -1; - - /* Compare floor(d) to top limb, subtract and cancel when equal. */ - for (i = xn; i-- > 0;) - { - mp_limb_t f, xl; - - f = (mp_limb_t) d; - xl = x->_mp_d[i]; - if (xl > f) - return 1; - else if (xl < f) - return -1; - d = B * (d - f); - } - } - return - (d > 0.0); -} - -int -mpz_cmp_d (const mpz_t x, double d) -{ - if (x->_mp_size < 0) - { - if (d >= 0.0) - return -1; - else - return -mpz_cmpabs_d (x, d); - } - else - { - if (d < 0.0) - return 1; - else - return mpz_cmpabs_d (x, d); - } -} - - -/* MPZ comparisons and the like. */ -int -mpz_sgn (const mpz_t u) -{ - return GMP_CMP (u->_mp_size, 0); -} - -int -mpz_cmp_si (const mpz_t u, intMpz v) -{ - mp_size_t usize = u->_mp_size; - - if (usize < -1) - return -1; - else if (v >= 0) - return mpz_cmp_ui (u, v); - else if (usize >= 0) - return 1; - else /* usize == -1 */ - return GMP_CMP (GMP_NEG_CAST (mp_limb_t, v), u->_mp_d[0]); -} - -int -mpz_cmp_ui (const mpz_t u, uIntMpz v) -{ - mp_size_t usize = u->_mp_size; - - if (usize > 1) - return 1; - else if (usize < 0) - return -1; - else - return GMP_CMP (mpz_get_ui (u), v); -} - -int -mpz_cmp (const mpz_t a, const mpz_t b) -{ - mp_size_t asize = a->_mp_size; - mp_size_t bsize = b->_mp_size; - - if (asize != bsize) - return (asize < bsize) ? -1 : 1; - else if (asize >= 0) - return mpn_cmp (a->_mp_d, b->_mp_d, asize); - else - return mpn_cmp (b->_mp_d, a->_mp_d, -asize); -} - -int -mpz_cmpabs_ui (const mpz_t u, uIntMpz v) -{ - if (GMP_ABS (u->_mp_size) > 1) - return 1; - else - return GMP_CMP (mpz_get_ui (u), v); -} - -int -mpz_cmpabs (const mpz_t u, const mpz_t v) -{ - return mpn_cmp4 (u->_mp_d, GMP_ABS (u->_mp_size), - v->_mp_d, GMP_ABS (v->_mp_size)); -} - -void -mpz_abs (mpz_t r, const mpz_t u) -{ - mpz_set (r, u); - r->_mp_size = GMP_ABS (r->_mp_size); -} - -void -mpz_neg (mpz_t r, const mpz_t u) -{ - mpz_set (r, u); - r->_mp_size = -r->_mp_size; -} - -void -mpz_swap (mpz_t u, mpz_t v) -{ - MP_SIZE_T_SWAP (u->_mp_size, v->_mp_size); - MP_SIZE_T_SWAP (u->_mp_alloc, v->_mp_alloc); - MP_PTR_SWAP (u->_mp_d, v->_mp_d); -} - - -/* MPZ addition and subtraction */ - -/* Adds to the absolute value. Returns new size, but doesn't store it. */ -static mp_size_t -mpz_abs_add_ui (mpz_t r, const mpz_t a, uIntMpz b) -{ - mp_size_t an; - mp_ptr rp; - mp_limb_t cy; - - an = GMP_ABS (a->_mp_size); - if (an == 0) - { - MPZ_REALLOC (r, 1)[0] = b; - return b > 0; - } - - rp = MPZ_REALLOC (r, an + 1); - - cy = mpn_add_1 (rp, a->_mp_d, an, b); - rp[an] = cy; - an += cy; - - return an; -} - -/* Subtract from the absolute value. Returns new size, (or -1 on underflow), - but doesn't store it. */ -static mp_size_t -mpz_abs_sub_ui (mpz_t r, const mpz_t a, uIntMpz b) -{ - mp_size_t an = GMP_ABS (a->_mp_size); - mp_ptr rp; - - if (an == 0) - { - MPZ_REALLOC (r, 1)[0] = b; - return -(b > 0); - } - rp = MPZ_REALLOC (r, an); - if (an == 1 && a->_mp_d[0] < b) - { - rp[0] = b - a->_mp_d[0]; - return -1; - } - else - { - gmp_assert_nocarry (mpn_sub_1 (rp, a->_mp_d, an, b)); - return mpn_normalized_size (rp, an); - } -} - -void -mpz_add_ui (mpz_t r, const mpz_t a, uIntMpz b) -{ - if (a->_mp_size >= 0) - r->_mp_size = mpz_abs_add_ui (r, a, b); - else - r->_mp_size = -mpz_abs_sub_ui (r, a, b); -} - -void -mpz_sub_ui (mpz_t r, const mpz_t a, uIntMpz b) -{ - if (a->_mp_size < 0) - r->_mp_size = -mpz_abs_add_ui (r, a, b); - else - r->_mp_size = mpz_abs_sub_ui (r, a, b); -} - -void -mpz_ui_sub (mpz_t r, uIntMpz a, const mpz_t b) -{ - if (b->_mp_size < 0) - r->_mp_size = mpz_abs_add_ui (r, b, a); - else - r->_mp_size = -mpz_abs_sub_ui (r, b, a); -} - -static mp_size_t -mpz_abs_add (mpz_t r, const mpz_t a, const mpz_t b) -{ - mp_size_t an = GMP_ABS (a->_mp_size); - mp_size_t bn = GMP_ABS (b->_mp_size); - mp_ptr rp; - mp_limb_t cy; - - if (an < bn) - { - MPZ_SRCPTR_SWAP (a, b); - MP_SIZE_T_SWAP (an, bn); - } - - rp = MPZ_REALLOC (r, an + 1); - cy = mpn_add (rp, a->_mp_d, an, b->_mp_d, bn); - - rp[an] = cy; - - return an + cy; -} - -static mp_size_t -mpz_abs_sub (mpz_t r, const mpz_t a, const mpz_t b) -{ - mp_size_t an = GMP_ABS (a->_mp_size); - mp_size_t bn = GMP_ABS (b->_mp_size); - int cmp; - mp_ptr rp; - - cmp = mpn_cmp4 (a->_mp_d, an, b->_mp_d, bn); - if (cmp > 0) - { - rp = MPZ_REALLOC (r, an); - gmp_assert_nocarry (mpn_sub (rp, a->_mp_d, an, b->_mp_d, bn)); - return mpn_normalized_size (rp, an); - } - else if (cmp < 0) - { - rp = MPZ_REALLOC (r, bn); - gmp_assert_nocarry (mpn_sub (rp, b->_mp_d, bn, a->_mp_d, an)); - return -mpn_normalized_size (rp, bn); - } - else - return 0; -} - -void -mpz_add (mpz_t r, const mpz_t a, const mpz_t b) -{ - mp_size_t rn; - - if ( (a->_mp_size ^ b->_mp_size) >= 0) - rn = mpz_abs_add (r, a, b); - else - rn = mpz_abs_sub (r, a, b); - - r->_mp_size = a->_mp_size >= 0 ? rn : - rn; -} - -void -mpz_sub (mpz_t r, const mpz_t a, const mpz_t b) -{ - mp_size_t rn; - - if ( (a->_mp_size ^ b->_mp_size) >= 0) - rn = mpz_abs_sub (r, a, b); - else - rn = mpz_abs_add (r, a, b); - - r->_mp_size = a->_mp_size >= 0 ? rn : - rn; -} - - -/* MPZ multiplication */ -void -mpz_mul_si (mpz_t r, const mpz_t u, intMpz v) -{ - if (v < 0) - { - mpz_mul_ui (r, u, GMP_NEG_CAST (uIntMpz, v)); - mpz_neg (r, r); - } - else - mpz_mul_ui (r, u, (uIntMpz) v); -} - -void -mpz_mul_ui (mpz_t r, const mpz_t u, uIntMpz v) -{ - mp_size_t un, us; - mp_ptr tp; - mp_limb_t cy; - - us = u->_mp_size; - - if (us == 0 || v == 0) - { - r->_mp_size = 0; - return; - } - - un = GMP_ABS (us); - - tp = MPZ_REALLOC (r, un + 1); - cy = mpn_mul_1 (tp, u->_mp_d, un, v); - tp[un] = cy; - - un += (cy > 0); - r->_mp_size = (us < 0) ? - un : un; -} - -void -mpz_mul (mpz_t r, const mpz_t u, const mpz_t v) -{ - int sign; - mp_size_t un, vn, rn; - mpz_t t; - mp_ptr tp; - - un = u->_mp_size; - vn = v->_mp_size; - - if (un == 0 || vn == 0) - { - r->_mp_size = 0; - return; - } - - sign = (un ^ vn) < 0; - - un = GMP_ABS (un); - vn = GMP_ABS (vn); - - mpz_init2 (t, (un + vn) * GMP_LIMB_BITS); - - tp = t->_mp_d; - if (un >= vn) - mpn_mul (tp, u->_mp_d, un, v->_mp_d, vn); - else - mpn_mul (tp, v->_mp_d, vn, u->_mp_d, un); - - rn = un + vn; - rn -= tp[rn-1] == 0; - - t->_mp_size = sign ? - rn : rn; - mpz_swap (r, t); - mpz_clear (t); -} - -void -mpz_mul_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t bits) -{ - mp_size_t un, rn; - mp_size_t limbs; - unsigned shift; - mp_ptr rp; - - un = GMP_ABS (u->_mp_size); - if (un == 0) - { - r->_mp_size = 0; - return; - } - - limbs = bits / GMP_LIMB_BITS; - shift = bits % GMP_LIMB_BITS; - - rn = un + limbs + (shift > 0); - rp = MPZ_REALLOC (r, rn); - if (shift > 0) - { - mp_limb_t cy = mpn_lshift (rp + limbs, u->_mp_d, un, shift); - rp[rn-1] = cy; - rn -= (cy == 0); - } - else - mpn_copyd (rp + limbs, u->_mp_d, un); - - mpn_zero (rp, limbs); - - r->_mp_size = (u->_mp_size < 0) ? - rn : rn; -} - -void -mpz_addmul_ui (mpz_t r, const mpz_t u, uIntMpz v) -{ - mpz_t t; - mpz_init (t); - mpz_mul_ui (t, u, v); - mpz_add (r, r, t); - mpz_clear (t); -} - -void -mpz_submul_ui (mpz_t r, const mpz_t u, uIntMpz v) -{ - mpz_t t; - mpz_init (t); - mpz_mul_ui (t, u, v); - mpz_sub (r, r, t); - mpz_clear (t); -} - -void -mpz_addmul (mpz_t r, const mpz_t u, const mpz_t v) -{ - mpz_t t; - mpz_init (t); - mpz_mul (t, u, v); - mpz_add (r, r, t); - mpz_clear (t); -} - -void -mpz_submul (mpz_t r, const mpz_t u, const mpz_t v) -{ - mpz_t t; - mpz_init (t); - mpz_mul (t, u, v); - mpz_sub (r, r, t); - mpz_clear (t); -} - - -/* MPZ division */ -enum mpz_div_round_mode { GMP_DIV_FLOOR, GMP_DIV_CEIL, GMP_DIV_TRUNC }; - -/* Allows q or r to be zero. Returns 1 iff remainder is non-zero. */ -static int -mpz_div_qr (mpz_t q, mpz_t r, - const mpz_t n, const mpz_t d, enum mpz_div_round_mode mode) -{ - mp_size_t ns, ds, nn, dn, qs; - ns = n->_mp_size; - ds = d->_mp_size; - - if (ds == 0) - gmp_die("mpz_div_qr: Divide by zero."); - - if (ns == 0) - { - if (q) - q->_mp_size = 0; - if (r) - r->_mp_size = 0; - return 0; - } - - nn = GMP_ABS (ns); - dn = GMP_ABS (ds); - - qs = ds ^ ns; - - if (nn < dn) - { - if (mode == GMP_DIV_CEIL && qs >= 0) - { - /* q = 1, r = n - d */ - if (r) - mpz_sub (r, n, d); - if (q) - mpz_set_ui (q, 1); - } - else if (mode == GMP_DIV_FLOOR && qs < 0) - { - /* q = -1, r = n + d */ - if (r) - mpz_add (r, n, d); - if (q) - mpz_set_si (q, -1); - } - else - { - /* q = 0, r = d */ - if (r) - mpz_set (r, n); - if (q) - q->_mp_size = 0; - } - return 1; - } - else - { - mp_ptr np, qp; - mp_size_t qn, rn; - mpz_t tq, tr; - - mpz_init_set (tr, n); - np = tr->_mp_d; - - qn = nn - dn + 1; - - if (q) - { - mpz_init2 (tq, qn * GMP_LIMB_BITS); - qp = tq->_mp_d; - } - else - qp = NULL; - - mpn_div_qr (qp, np, nn, d->_mp_d, dn); - - if (qp) - { - qn -= (qp[qn-1] == 0); - - tq->_mp_size = qs < 0 ? -qn : qn; - } - rn = mpn_normalized_size (np, dn); - tr->_mp_size = ns < 0 ? - rn : rn; - - if (mode == GMP_DIV_FLOOR && qs < 0 && rn != 0) - { - if (q) - mpz_sub_ui (tq, tq, 1); - if (r) - mpz_add (tr, tr, d); - } - else if (mode == GMP_DIV_CEIL && qs >= 0 && rn != 0) - { - if (q) - mpz_add_ui (tq, tq, 1); - if (r) - mpz_sub (tr, tr, d); - } - - if (q) - { - mpz_swap (tq, q); - mpz_clear (tq); - } - if (r) - mpz_swap (tr, r); - - mpz_clear (tr); - - return rn != 0; - } -} - -void -mpz_cdiv_qr (mpz_t q, mpz_t r, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (q, r, n, d, GMP_DIV_CEIL); -} - -void -mpz_fdiv_qr (mpz_t q, mpz_t r, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (q, r, n, d, GMP_DIV_FLOOR); -} - -void -mpz_tdiv_qr (mpz_t q, mpz_t r, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (q, r, n, d, GMP_DIV_TRUNC); -} - -void -mpz_cdiv_q (mpz_t q, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (q, NULL, n, d, GMP_DIV_CEIL); -} - -void -mpz_fdiv_q (mpz_t q, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (q, NULL, n, d, GMP_DIV_FLOOR); -} - -void -mpz_tdiv_q (mpz_t q, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (q, NULL, n, d, GMP_DIV_TRUNC); -} - -void -mpz_cdiv_r (mpz_t r, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (NULL, r, n, d, GMP_DIV_CEIL); -} - -void -mpz_fdiv_r (mpz_t r, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (NULL, r, n, d, GMP_DIV_FLOOR); -} - -void -mpz_tdiv_r (mpz_t r, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (NULL, r, n, d, GMP_DIV_TRUNC); -} - -void -mpz_mod (mpz_t r, const mpz_t n, const mpz_t d) -{ - mpz_div_qr (NULL, r, n, d, d->_mp_size >= 0 ? GMP_DIV_FLOOR : GMP_DIV_CEIL); -} - -static void -mpz_div_q_2exp (mpz_t q, const mpz_t u, mp_bitcnt_t bit_index, - enum mpz_div_round_mode mode) -{ - mp_size_t un, qn; - mp_size_t limb_cnt; - mp_ptr qp; - int adjust; - - un = u->_mp_size; - if (un == 0) - { - q->_mp_size = 0; - return; - } - limb_cnt = bit_index / GMP_LIMB_BITS; - qn = GMP_ABS (un) - limb_cnt; - bit_index %= GMP_LIMB_BITS; - - if (mode == ((un > 0) ? GMP_DIV_CEIL : GMP_DIV_FLOOR)) /* un != 0 here. */ - /* Note: Below, the final indexing at limb_cnt is valid because at - that point we have qn > 0. */ - adjust = (qn <= 0 - || !mpn_zero_p (u->_mp_d, limb_cnt) - || (u->_mp_d[limb_cnt] - & (((mp_limb_t) 1 << bit_index) - 1))); - else - adjust = 0; - - if (qn <= 0) - qn = 0; - else - { - qp = MPZ_REALLOC (q, qn); - - if (bit_index != 0) - { - mpn_rshift (qp, u->_mp_d + limb_cnt, qn, bit_index); - qn -= qp[qn - 1] == 0; - } - else - { - mpn_copyi (qp, u->_mp_d + limb_cnt, qn); - } - } - - q->_mp_size = qn; - - if (adjust) - mpz_add_ui (q, q, 1); - if (un < 0) - mpz_neg (q, q); -} - -static void -mpz_div_r_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t bit_index, - enum mpz_div_round_mode mode) -{ - mp_size_t us, un, rn; - mp_ptr rp; - mp_limb_t mask; - - us = u->_mp_size; - if (us == 0 || bit_index == 0) - { - r->_mp_size = 0; - return; - } - rn = (bit_index + GMP_LIMB_BITS - 1) / GMP_LIMB_BITS; - assert (rn > 0); - - rp = MPZ_REALLOC (r, rn); - un = GMP_ABS (us); - - mask = GMP_LIMB_MAX >> (rn * GMP_LIMB_BITS - bit_index); - - if (rn > un) - { - /* Quotient (with truncation) is zero, and remainder is - non-zero */ - if (mode == ((us > 0) ? GMP_DIV_CEIL : GMP_DIV_FLOOR)) /* us != 0 here. */ - { - /* Have to negate and sign extend. */ - mp_size_t i; - - gmp_assert_nocarry (! mpn_neg (rp, u->_mp_d, un)); - for (i = un; i < rn - 1; i++) - rp[i] = GMP_LIMB_MAX; - - rp[rn-1] = mask; - us = -us; - } - else - { - /* Just copy */ - if (r != u) - mpn_copyi (rp, u->_mp_d, un); - - rn = un; - } - } - else - { - if (r != u) - mpn_copyi (rp, u->_mp_d, rn - 1); - - rp[rn-1] = u->_mp_d[rn-1] & mask; - - if (mode == ((us > 0) ? GMP_DIV_CEIL : GMP_DIV_FLOOR)) /* us != 0 here. */ - { - /* If r != 0, compute 2^{bit_count} - r. */ - mpn_neg (rp, rp, rn); - - rp[rn-1] &= mask; - - /* us is not used for anything else, so we can modify it - here to indicate flipped sign. */ - us = -us; - } - } - rn = mpn_normalized_size (rp, rn); - r->_mp_size = us < 0 ? -rn : rn; -} - -void -mpz_cdiv_q_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t cnt) -{ - mpz_div_q_2exp (r, u, cnt, GMP_DIV_CEIL); -} - -void -mpz_fdiv_q_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t cnt) -{ - mpz_div_q_2exp (r, u, cnt, GMP_DIV_FLOOR); -} - -void -mpz_tdiv_q_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t cnt) -{ - mpz_div_q_2exp (r, u, cnt, GMP_DIV_TRUNC); -} - -void -mpz_cdiv_r_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t cnt) -{ - mpz_div_r_2exp (r, u, cnt, GMP_DIV_CEIL); -} - -void -mpz_fdiv_r_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t cnt) -{ - mpz_div_r_2exp (r, u, cnt, GMP_DIV_FLOOR); -} - -void -mpz_tdiv_r_2exp (mpz_t r, const mpz_t u, mp_bitcnt_t cnt) -{ - mpz_div_r_2exp (r, u, cnt, GMP_DIV_TRUNC); -} - -void -mpz_divexact (mpz_t q, const mpz_t n, const mpz_t d) -{ - gmp_assert_nocarry (mpz_div_qr (q, NULL, n, d, GMP_DIV_TRUNC)); -} - -int -mpz_divisible_p (const mpz_t n, const mpz_t d) -{ - return mpz_div_qr (NULL, NULL, n, d, GMP_DIV_TRUNC) == 0; -} - -int -mpz_congruent_p (const mpz_t a, const mpz_t b, const mpz_t m) -{ - mpz_t t; - int res; - - /* a == b (mod 0) iff a == b */ - if (mpz_sgn (m) == 0) - return (mpz_cmp (a, b) == 0); - - mpz_init (t); - mpz_sub (t, a, b); - res = mpz_divisible_p (t, m); - mpz_clear (t); - - return res; -} - -static uIntMpz -mpz_div_qr_ui (mpz_t q, mpz_t r, - const mpz_t n, uIntMpz d, enum mpz_div_round_mode mode) -{ - mp_size_t ns, qn; - mp_ptr qp; - mp_limb_t rl; - mp_size_t rs; - - ns = n->_mp_size; - if (ns == 0) - { - if (q) - q->_mp_size = 0; - if (r) - r->_mp_size = 0; - return 0; - } - - qn = GMP_ABS (ns); - if (q) - qp = MPZ_REALLOC (q, qn); - else - qp = NULL; - - rl = mpn_div_qr_1 (qp, n->_mp_d, qn, d); - assert (rl < d); - - rs = rl > 0; - rs = (ns < 0) ? -rs : rs; - - if (rl > 0 && ( (mode == GMP_DIV_FLOOR && ns < 0) - || (mode == GMP_DIV_CEIL && ns >= 0))) - { - if (q) - gmp_assert_nocarry (mpn_add_1 (qp, qp, qn, 1)); - rl = d - rl; - rs = -rs; - } - - if (r) - { - MPZ_REALLOC (r, 1)[0] = rl; - r->_mp_size = rs; - } - if (q) - { - qn -= (qp[qn-1] == 0); - assert (qn == 0 || qp[qn-1] > 0); - - q->_mp_size = (ns < 0) ? - qn : qn; - } - - return rl; -} - -uIntMpz -mpz_cdiv_qr_ui (mpz_t q, mpz_t r, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (q, r, n, d, GMP_DIV_CEIL); -} - -uIntMpz -mpz_fdiv_qr_ui (mpz_t q, mpz_t r, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (q, r, n, d, GMP_DIV_FLOOR); -} - -uIntMpz -mpz_tdiv_qr_ui (mpz_t q, mpz_t r, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (q, r, n, d, GMP_DIV_TRUNC); -} - -uIntMpz -mpz_cdiv_q_ui (mpz_t q, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (q, NULL, n, d, GMP_DIV_CEIL); -} - -uIntMpz -mpz_fdiv_q_ui (mpz_t q, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (q, NULL, n, d, GMP_DIV_FLOOR); -} - -uIntMpz -mpz_tdiv_q_ui (mpz_t q, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (q, NULL, n, d, GMP_DIV_TRUNC); -} - -uIntMpz -mpz_cdiv_r_ui (mpz_t r, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, r, n, d, GMP_DIV_CEIL); -} -uIntMpz -mpz_fdiv_r_ui (mpz_t r, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, r, n, d, GMP_DIV_FLOOR); -} -uIntMpz -mpz_tdiv_r_ui (mpz_t r, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, r, n, d, GMP_DIV_TRUNC); -} - -uIntMpz -mpz_cdiv_ui (const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, NULL, n, d, GMP_DIV_CEIL); -} - -uIntMpz -mpz_fdiv_ui (const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, NULL, n, d, GMP_DIV_FLOOR); -} - -uIntMpz -mpz_tdiv_ui (const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, NULL, n, d, GMP_DIV_TRUNC); -} - -uIntMpz -mpz_mod_ui (mpz_t r, const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, r, n, d, GMP_DIV_FLOOR); -} - -void -mpz_divexact_ui (mpz_t q, const mpz_t n, uIntMpz d) -{ - gmp_assert_nocarry (mpz_div_qr_ui (q, NULL, n, d, GMP_DIV_TRUNC)); -} - -int -mpz_divisible_ui_p (const mpz_t n, uIntMpz d) -{ - return mpz_div_qr_ui (NULL, NULL, n, d, GMP_DIV_TRUNC) == 0; -} - - -/* GCD */ -static mp_limb_t -mpn_gcd_11 (mp_limb_t u, mp_limb_t v) -{ - unsigned shift; - - assert ( (u | v) > 0); - - if (u == 0) - return v; - else if (v == 0) - return u; - - gmp_ctz (shift, u | v); - - u >>= shift; - v >>= shift; - - if ( (u & 1) == 0) - MP_LIMB_T_SWAP (u, v); - - while ( (v & 1) == 0) - v >>= 1; - - while (u != v) - { - if (u > v) - { - u -= v; - do - u >>= 1; - while ( (u & 1) == 0); - } - else - { - v -= u; - do - v >>= 1; - while ( (v & 1) == 0); - } - } - return u << shift; -} - -uIntMpz -mpz_gcd_ui (mpz_t g, const mpz_t u, uIntMpz v) -{ - mp_size_t un; - - if (v == 0) - { - if (g) - mpz_abs (g, u); - } - else - { - un = GMP_ABS (u->_mp_size); - if (un != 0) - v = mpn_gcd_11 (mpn_div_qr_1 (NULL, u->_mp_d, un, v), v); - - if (g) - mpz_set_ui (g, v); - } - - return v; -} - -static mp_bitcnt_t -mpz_make_odd (mpz_t r) -{ - mp_bitcnt_t shift; - - assert (r->_mp_size > 0); - /* Count trailing zeros, equivalent to mpn_scan1, because we know that there is a 1 */ - shift = mpn_common_scan (r->_mp_d[0], 0, r->_mp_d, 0, 0); - mpz_tdiv_q_2exp (r, r, shift); - - return shift; -} - -void -mpz_gcd (mpz_t g, const mpz_t u, const mpz_t v) -{ - mpz_t tu, tv; - mp_bitcnt_t uz, vz, gz; - - if (u->_mp_size == 0) - { - mpz_abs (g, v); - return; - } - if (v->_mp_size == 0) - { - mpz_abs (g, u); - return; - } - - mpz_init (tu); - mpz_init (tv); - - mpz_abs (tu, u); - uz = mpz_make_odd (tu); - mpz_abs (tv, v); - vz = mpz_make_odd (tv); - gz = GMP_MIN (uz, vz); - - if (tu->_mp_size < tv->_mp_size) - mpz_swap (tu, tv); - - mpz_tdiv_r (tu, tu, tv); - if (tu->_mp_size == 0) - { - mpz_swap (g, tv); - } - else - for (;;) - { - int c; - - mpz_make_odd (tu); - c = mpz_cmp (tu, tv); - if (c == 0) - { - mpz_swap (g, tu); - break; - } - if (c < 0) - mpz_swap (tu, tv); - - if (tv->_mp_size == 1) - { - mp_limb_t vl = tv->_mp_d[0]; - mp_limb_t ul = mpz_tdiv_ui (tu, vl); - mpz_set_ui (g, mpn_gcd_11 (ul, vl)); - break; - } - mpz_sub (tu, tu, tv); - } - mpz_clear (tu); - mpz_clear (tv); - mpz_mul_2exp (g, g, gz); -} - -void -mpz_gcdext (mpz_t g, mpz_t s, mpz_t t, const mpz_t u, const mpz_t v) -{ - mpz_t tu, tv, s0, s1, t0, t1; - mp_bitcnt_t uz, vz, gz; - mp_bitcnt_t power; - - if (u->_mp_size == 0) - { - /* g = 0 u + sgn(v) v */ - intMpz sign = mpz_sgn (v); - mpz_abs (g, v); - if (s) - mpz_set_ui (s, 0); - if (t) - mpz_set_si (t, sign); - return; - } - - if (v->_mp_size == 0) - { - /* g = sgn(u) u + 0 v */ - intMpz sign = mpz_sgn (u); - mpz_abs (g, u); - if (s) - mpz_set_si (s, sign); - if (t) - mpz_set_ui (t, 0); - return; - } - - mpz_init (tu); - mpz_init (tv); - mpz_init (s0); - mpz_init (s1); - mpz_init (t0); - mpz_init (t1); - - mpz_abs (tu, u); - uz = mpz_make_odd (tu); - mpz_abs (tv, v); - vz = mpz_make_odd (tv); - gz = GMP_MIN (uz, vz); - - uz -= gz; - vz -= gz; - - /* Cofactors corresponding to odd gcd. gz handled later. */ - if (tu->_mp_size < tv->_mp_size) - { - mpz_swap (tu, tv); - MPZ_SRCPTR_SWAP (u, v); - MPZ_PTR_SWAP (s, t); - MP_BITCNT_T_SWAP (uz, vz); - } - - /* Maintain - * - * u = t0 tu + t1 tv - * v = s0 tu + s1 tv - * - * where u and v denote the inputs with common factors of two - * eliminated, and det (s0, t0; s1, t1) = 2^p. Then - * - * 2^p tu = s1 u - t1 v - * 2^p tv = -s0 u + t0 v - */ - - /* After initial division, tu = q tv + tu', we have - * - * u = 2^uz (tu' + q tv) - * v = 2^vz tv - * - * or - * - * t0 = 2^uz, t1 = 2^uz q - * s0 = 0, s1 = 2^vz - */ - - mpz_setbit (t0, uz); - mpz_tdiv_qr (t1, tu, tu, tv); - mpz_mul_2exp (t1, t1, uz); - - mpz_setbit (s1, vz); - power = uz + vz; - - if (tu->_mp_size > 0) - { - mp_bitcnt_t shift; - shift = mpz_make_odd (tu); - mpz_mul_2exp (t0, t0, shift); - mpz_mul_2exp (s0, s0, shift); - power += shift; - - for (;;) - { - int c; - c = mpz_cmp (tu, tv); - if (c == 0) - break; - - if (c < 0) - { - /* tv = tv' + tu - * - * u = t0 tu + t1 (tv' + tu) = (t0 + t1) tu + t1 tv' - * v = s0 tu + s1 (tv' + tu) = (s0 + s1) tu + s1 tv' */ - - mpz_sub (tv, tv, tu); - mpz_add (t0, t0, t1); - mpz_add (s0, s0, s1); - - shift = mpz_make_odd (tv); - mpz_mul_2exp (t1, t1, shift); - mpz_mul_2exp (s1, s1, shift); - } - else - { - mpz_sub (tu, tu, tv); - mpz_add (t1, t0, t1); - mpz_add (s1, s0, s1); - - shift = mpz_make_odd (tu); - mpz_mul_2exp (t0, t0, shift); - mpz_mul_2exp (s0, s0, shift); - } - power += shift; - } - } - - /* Now tv = odd part of gcd, and -s0 and t0 are corresponding - cofactors. */ - - mpz_mul_2exp (tv, tv, gz); - mpz_neg (s0, s0); - - /* 2^p g = s0 u + t0 v. Eliminate one factor of two at a time. To - adjust cofactors, we need u / g and v / g */ - - mpz_divexact (s1, v, tv); - mpz_abs (s1, s1); - mpz_divexact (t1, u, tv); - mpz_abs (t1, t1); - - while (power-- > 0) - { - /* s0 u + t0 v = (s0 - v/g) u - (t0 + u/g) v */ - if (mpz_odd_p (s0) || mpz_odd_p (t0)) - { - mpz_sub (s0, s0, s1); - mpz_add (t0, t0, t1); - } - mpz_divexact_ui (s0, s0, 2); - mpz_divexact_ui (t0, t0, 2); - } - - /* Arrange so that |s| < |u| / 2g */ - mpz_add (s1, s0, s1); - if (mpz_cmpabs (s0, s1) > 0) - { - mpz_swap (s0, s1); - mpz_sub (t0, t0, t1); - } - if (u->_mp_size < 0) - mpz_neg (s0, s0); - if (v->_mp_size < 0) - mpz_neg (t0, t0); - - mpz_swap (g, tv); - if (s) - mpz_swap (s, s0); - if (t) - mpz_swap (t, t0); - - mpz_clear (tu); - mpz_clear (tv); - mpz_clear (s0); - mpz_clear (s1); - mpz_clear (t0); - mpz_clear (t1); -} - -void -mpz_lcm (mpz_t r, const mpz_t u, const mpz_t v) -{ - mpz_t g; - - if (u->_mp_size == 0 || v->_mp_size == 0) - { - r->_mp_size = 0; - return; - } - - mpz_init (g); - - mpz_gcd (g, u, v); - mpz_divexact (g, u, g); - mpz_mul (r, g, v); - - mpz_clear (g); - mpz_abs (r, r); -} - -void -mpz_lcm_ui (mpz_t r, const mpz_t u, uIntMpz v) -{ - if (v == 0 || u->_mp_size == 0) - { - r->_mp_size = 0; - return; - } - - v /= mpz_gcd_ui (NULL, u, v); - mpz_mul_ui (r, u, v); - - mpz_abs (r, r); -} - -int -mpz_invert (mpz_t r, const mpz_t u, const mpz_t m) -{ - mpz_t g, tr; - int invertible; - - if (u->_mp_size == 0 || mpz_cmpabs_ui (m, 1) <= 0) - return 0; - - mpz_init (g); - mpz_init (tr); - - mpz_gcdext (g, tr, NULL, u, m); - invertible = (mpz_cmp_ui (g, 1) == 0); - - if (invertible) - { - if (tr->_mp_size < 0) - { - if (m->_mp_size >= 0) - mpz_add (tr, tr, m); - else - mpz_sub (tr, tr, m); - } - mpz_swap (r, tr); - } - - mpz_clear (g); - mpz_clear (tr); - return invertible; -} - - -/* Higher level operations (sqrt, pow and root) */ - -void -mpz_pow_ui (mpz_t r, const mpz_t b, uIntMpz e) -{ - uIntMpz bit; - mpz_t tr; - mpz_init_set_ui (tr, 1); - - bit = GMP_ULONG_HIGHBIT; - do - { - mpz_mul (tr, tr, tr); - if (e & bit) - mpz_mul (tr, tr, b); - bit >>= 1; - } - while (bit > 0); - - mpz_swap (r, tr); - mpz_clear (tr); -} - -void -mpz_ui_pow_ui (mpz_t r, uIntMpz blimb, uIntMpz e) -{ - mpz_t b; - mpz_pow_ui (r, mpz_roinit_n (b, &blimb, 1), e); -} - -void -mpz_powm (mpz_t r, const mpz_t b, const mpz_t e, const mpz_t m) -{ - mpz_t tr; - mpz_t base; - mp_size_t en, mn; - mp_srcptr mp; - struct gmp_div_inverse minv; - unsigned shift; - mp_ptr tp = NULL; - - en = GMP_ABS (e->_mp_size); - mn = GMP_ABS (m->_mp_size); - if (mn == 0) - gmp_die ("mpz_powm: Zero modulo."); - - if (en == 0) - { - mpz_set_ui (r, 1); - return; - } - - mp = m->_mp_d; - mpn_div_qr_invert (&minv, mp, mn); - shift = minv.shift; - - if (shift > 0) - { - /* To avoid shifts, we do all our reductions, except the final - one, using a *normalized* m. */ - minv.shift = 0; - - tp = gmp_xalloc_limbs (mn); - gmp_assert_nocarry (mpn_lshift (tp, mp, mn, shift)); - mp = tp; - } - - mpz_init (base); - - if (e->_mp_size < 0) - { - if (!mpz_invert (base, b, m)) - gmp_die ("mpz_powm: Negative exponent and non-invertible base."); - } - else - { - mp_size_t bn; - mpz_abs (base, b); - - bn = base->_mp_size; - if (bn >= mn) - { - mpn_div_qr_preinv (NULL, base->_mp_d, base->_mp_size, mp, mn, &minv); - bn = mn; - } - - /* We have reduced the absolute value. Now take care of the - sign. Note that we get zero represented non-canonically as - m. */ - if (b->_mp_size < 0) - { - mp_ptr bp = MPZ_REALLOC (base, mn); - gmp_assert_nocarry (mpn_sub (bp, mp, mn, bp, bn)); - bn = mn; - } - base->_mp_size = mpn_normalized_size (base->_mp_d, bn); - } - mpz_init_set_ui (tr, 1); - - while (--en >= 0) - { - mp_limb_t w = e->_mp_d[en]; - mp_limb_t bit; - - bit = GMP_LIMB_HIGHBIT; - do - { - mpz_mul (tr, tr, tr); - if (w & bit) - mpz_mul (tr, tr, base); - if (tr->_mp_size > mn) - { - mpn_div_qr_preinv (NULL, tr->_mp_d, tr->_mp_size, mp, mn, &minv); - tr->_mp_size = mpn_normalized_size (tr->_mp_d, mn); - } - bit >>= 1; - } - while (bit > 0); - } - - /* Final reduction */ - if (tr->_mp_size >= mn) - { - minv.shift = shift; - mpn_div_qr_preinv (NULL, tr->_mp_d, tr->_mp_size, mp, mn, &minv); - tr->_mp_size = mpn_normalized_size (tr->_mp_d, mn); - } - if (tp) - gmp_free (tp); - - mpz_swap (r, tr); - mpz_clear (tr); - mpz_clear (base); -} - -void -mpz_powm_ui (mpz_t r, const mpz_t b, uIntMpz elimb, const mpz_t m) -{ - mpz_t e; - mpz_powm (r, b, mpz_roinit_n (e, &elimb, 1), m); -} - -/* x=trunc(y^(1/z)), r=y-x^z */ -void -mpz_rootrem (mpz_t x, mpz_t r, const mpz_t y, uIntMpz z) -{ - int sgn; - mpz_t t, u; - - sgn = y->_mp_size < 0; - if ((~z & sgn) != 0) - gmp_die ("mpz_rootrem: Negative argument, with even root."); - if (z == 0) - gmp_die ("mpz_rootrem: Zeroth root."); - - if (mpz_cmpabs_ui (y, 1) <= 0) { - if (x) - mpz_set (x, y); - if (r) - r->_mp_size = 0; - return; - } - - mpz_init (u); - mpz_init (t); - mpz_setbit (t, mpz_sizeinbase (y, 2) / z + 1); - - if (z == 2) /* simplify sqrt loop: z-1 == 1 */ - do { - mpz_swap (u, t); /* u = x */ - mpz_tdiv_q (t, y, u); /* t = y/x */ - mpz_add (t, t, u); /* t = y/x + x */ - mpz_tdiv_q_2exp (t, t, 1); /* x'= (y/x + x)/2 */ - } while (mpz_cmpabs (t, u) < 0); /* |x'| < |x| */ - else /* z != 2 */ { - mpz_t v; - - mpz_init (v); - if (sgn) - mpz_neg (t, t); - - do { - mpz_swap (u, t); /* u = x */ - mpz_pow_ui (t, u, z - 1); /* t = x^(z-1) */ - mpz_tdiv_q (t, y, t); /* t = y/x^(z-1) */ - mpz_mul_ui (v, u, z - 1); /* v = x*(z-1) */ - mpz_add (t, t, v); /* t = y/x^(z-1) + x*(z-1) */ - mpz_tdiv_q_ui (t, t, z); /* x'=(y/x^(z-1) + x*(z-1))/z */ - } while (mpz_cmpabs (t, u) < 0); /* |x'| < |x| */ - - mpz_clear (v); - } - - if (r) { - mpz_pow_ui (t, u, z); - mpz_sub (r, y, t); - } - if (x) - mpz_swap (x, u); - mpz_clear (u); - mpz_clear (t); -} - -int -mpz_root (mpz_t x, const mpz_t y, uIntMpz z) -{ - int res; - mpz_t r; - - mpz_init (r); - mpz_rootrem (x, r, y, z); - res = r->_mp_size == 0; - mpz_clear (r); - - return res; -} - -/* Compute s = floor(sqrt(u)) and r = u - s^2. Allows r == NULL */ -void -mpz_sqrtrem (mpz_t s, mpz_t r, const mpz_t u) -{ - mpz_rootrem (s, r, u, 2); -} - -void -mpz_sqrt (mpz_t s, const mpz_t u) -{ - mpz_rootrem (s, NULL, u, 2); -} - -int -mpz_perfect_square_p (const mpz_t u) -{ - if (u->_mp_size <= 0) - return (u->_mp_size == 0); - else - return mpz_root (NULL, u, 2); -} - -int -mpn_perfect_square_p (mp_srcptr p, mp_size_t n) -{ - mpz_t t; - - assert (n > 0); - assert (p [n-1] != 0); - return mpz_root (NULL, mpz_roinit_n (t, p, n), 2); -} - -mp_size_t -mpn_sqrtrem (mp_ptr sp, mp_ptr rp, mp_srcptr p, mp_size_t n) -{ - mpz_t s, r, u; - mp_size_t res; - - assert (n > 0); - assert (p [n-1] != 0); - - mpz_init (r); - mpz_init (s); - mpz_rootrem (s, r, mpz_roinit_n (u, p, n), 2); - - assert (s->_mp_size == (n+1)/2); - mpn_copyd (sp, s->_mp_d, s->_mp_size); - mpz_clear (s); - res = r->_mp_size; - if (rp) - mpn_copyd (rp, r->_mp_d, res); - mpz_clear (r); - return res; -} - -/* Combinatorics */ - -void -mpz_fac_ui (mpz_t x, uIntMpz n) -{ - mpz_set_ui (x, n + (n == 0)); - while (n > 2) - mpz_mul_ui (x, x, --n); -} - -void -mpz_bin_uiui (mpz_t r, uIntMpz n, uIntMpz k) -{ - mpz_t t; - - mpz_set_ui (r, k <= n); - - if (k > (n >> 1)) - k = (k <= n) ? n - k : 0; - - mpz_init (t); - mpz_fac_ui (t, k); - - for (; k > 0; k--) - mpz_mul_ui (r, r, n--); - - mpz_divexact (r, r, t); - mpz_clear (t); -} - - -/* Primality testing */ -static int -gmp_millerrabin (const mpz_t n, const mpz_t nm1, mpz_t y, - const mpz_t q, mp_bitcnt_t k) -{ - assert (k > 0); - - /* Caller must initialize y to the base. */ - mpz_powm (y, y, q, n); - - if (mpz_cmp_ui (y, 1) == 0 || mpz_cmp (y, nm1) == 0) - return 1; - - while (--k > 0) - { - mpz_powm_ui (y, y, 2, n); - if (mpz_cmp (y, nm1) == 0) - return 1; - /* y == 1 means that the previous y was a non-trivial square root - of 1 (mod n). y == 0 means that n is a power of the base. - In either case, n is not prime. */ - if (mpz_cmp_ui (y, 1) <= 0) - return 0; - } - return 0; -} - -/* This product is 0xc0cfd797, and fits in 32 bits. */ -#define GMP_PRIME_PRODUCT \ - (3UL*5UL*7UL*11UL*13UL*17UL*19UL*23UL*29UL) - -/* Bit (p+1)/2 is set, for each odd prime <= 61 */ -#define GMP_PRIME_MASK 0xc96996dcUL - -int -mpz_probab_prime_p (const mpz_t n, int reps) -{ - mpz_t nm1; - mpz_t q; - mpz_t y; - mp_bitcnt_t k; - int is_prime; - int j; - - /* Note that we use the absolute value of n only, for compatibility - with the real GMP. */ - if (mpz_even_p (n)) - return (mpz_cmpabs_ui (n, 2) == 0) ? 2 : 0; - - /* Above test excludes n == 0 */ - assert (n->_mp_size != 0); - - if (mpz_cmpabs_ui (n, 64) < 0) - return (GMP_PRIME_MASK >> (n->_mp_d[0] >> 1)) & 2; - - if (mpz_gcd_ui (NULL, n, GMP_PRIME_PRODUCT) != 1) - return 0; - - /* All prime factors are >= 31. */ - if (mpz_cmpabs_ui (n, 31*31) < 0) - return 2; - - /* Use Miller-Rabin, with a deterministic sequence of bases, a[j] = - j^2 + j + 41 using Euler's polynomial. We potentially stop early, - if a[j] >= n - 1. Since n >= 31*31, this can happen only if reps > - 30 (a[30] == 971 > 31*31 == 961). */ - - mpz_init (nm1); - mpz_init (q); - mpz_init (y); - - /* Find q and k, where q is odd and n = 1 + 2**k * q. */ - nm1->_mp_size = mpz_abs_sub_ui (nm1, n, 1); - k = mpz_scan1 (nm1, 0); - mpz_tdiv_q_2exp (q, nm1, k); - - for (j = 0, is_prime = 1; is_prime & (j < reps); j++) - { - mpz_set_ui (y, (uIntMpz) j*j+j+41); - if (mpz_cmp (y, nm1) >= 0) - { - /* Don't try any further bases. This "early" break does not affect - the result for any reasonable reps value (<=5000 was tested) */ - assert (j >= 30); - break; - } - is_prime = gmp_millerrabin (n, nm1, y, q, k); - } - mpz_clear (nm1); - mpz_clear (q); - mpz_clear (y); - - return is_prime; -} - - -/* Logical operations and bit manipulation. */ - -/* Numbers are treated as if represented in two's complement (and - infinitely sign extended). For a negative values we get the two's - complement from -x = ~x + 1, where ~ is bitwise complement. - Negation transforms - - xxxx10...0 - - into - - yyyy10...0 - - where yyyy is the bitwise complement of xxxx. So least significant - bits, up to and including the first one bit, are unchanged, and - the more significant bits are all complemented. - - To change a bit from zero to one in a negative number, subtract the - corresponding power of two from the absolute value. This can never - underflow. To change a bit from one to zero, add the corresponding - power of two, and this might overflow. E.g., if x = -001111, the - two's complement is 110001. Clearing the least significant bit, we - get two's complement 110000, and -010000. */ - -int -mpz_tstbit (const mpz_t d, mp_bitcnt_t bit_index) -{ - mp_size_t limb_index; - unsigned shift; - mp_size_t ds; - mp_size_t dn; - mp_limb_t w; - int bit; - - ds = d->_mp_size; - dn = GMP_ABS (ds); - limb_index = bit_index / GMP_LIMB_BITS; - if (limb_index >= dn) - return ds < 0; - - shift = bit_index % GMP_LIMB_BITS; - w = d->_mp_d[limb_index]; - bit = (w >> shift) & 1; - - if (ds < 0) - { - /* d < 0. Check if any of the bits below is set: If so, our bit - must be complemented. */ - if (shift > 0 && (w << (GMP_LIMB_BITS - shift)) > 0) - return bit ^ 1; - while (--limb_index >= 0) - if (d->_mp_d[limb_index] > 0) - return bit ^ 1; - } - return bit; -} - -static void -mpz_abs_add_bit (mpz_t d, mp_bitcnt_t bit_index) -{ - mp_size_t dn, limb_index; - mp_limb_t bit; - mp_ptr dp; - - dn = GMP_ABS (d->_mp_size); - - limb_index = bit_index / GMP_LIMB_BITS; - bit = (mp_limb_t) 1 << (bit_index % GMP_LIMB_BITS); - - if (limb_index >= dn) - { - mp_size_t i; - /* The bit should be set outside of the end of the number. - We have to increase the size of the number. */ - dp = MPZ_REALLOC (d, limb_index + 1); - - dp[limb_index] = bit; - for (i = dn; i < limb_index; i++) - dp[i] = 0; - dn = limb_index + 1; - } - else - { - mp_limb_t cy; - - dp = d->_mp_d; - - cy = mpn_add_1 (dp + limb_index, dp + limb_index, dn - limb_index, bit); - if (cy > 0) - { - dp = MPZ_REALLOC (d, dn + 1); - dp[dn++] = cy; - } - } - - d->_mp_size = (d->_mp_size < 0) ? - dn : dn; -} - -static void -mpz_abs_sub_bit (mpz_t d, mp_bitcnt_t bit_index) -{ - mp_size_t dn, limb_index; - mp_ptr dp; - mp_limb_t bit; - - dn = GMP_ABS (d->_mp_size); - dp = d->_mp_d; - - limb_index = bit_index / GMP_LIMB_BITS; - bit = (mp_limb_t) 1 << (bit_index % GMP_LIMB_BITS); - - assert (limb_index < dn); - - gmp_assert_nocarry (mpn_sub_1 (dp + limb_index, dp + limb_index, - dn - limb_index, bit)); - dn = mpn_normalized_size (dp, dn); - d->_mp_size = (d->_mp_size < 0) ? - dn : dn; -} - -void -mpz_setbit (mpz_t d, mp_bitcnt_t bit_index) -{ - if (!mpz_tstbit (d, bit_index)) - { - if (d->_mp_size >= 0) - mpz_abs_add_bit (d, bit_index); - else - mpz_abs_sub_bit (d, bit_index); - } -} - -void -mpz_clrbit (mpz_t d, mp_bitcnt_t bit_index) -{ - if (mpz_tstbit (d, bit_index)) - { - if (d->_mp_size >= 0) - mpz_abs_sub_bit (d, bit_index); - else - mpz_abs_add_bit (d, bit_index); - } -} - -void -mpz_combit (mpz_t d, mp_bitcnt_t bit_index) -{ - if (mpz_tstbit (d, bit_index) ^ (d->_mp_size < 0)) - mpz_abs_sub_bit (d, bit_index); - else - mpz_abs_add_bit (d, bit_index); -} - -void -mpz_com (mpz_t r, const mpz_t u) -{ - mpz_neg (r, u); - mpz_sub_ui (r, r, 1); -} - -void -mpz_and (mpz_t r, const mpz_t u, const mpz_t v) -{ - mp_size_t un, vn, rn, i; - mp_ptr up, vp, rp; - - mp_limb_t ux, vx, rx; - mp_limb_t uc, vc, rc; - mp_limb_t ul, vl, rl; - - un = GMP_ABS (u->_mp_size); - vn = GMP_ABS (v->_mp_size); - if (un < vn) - { - MPZ_SRCPTR_SWAP (u, v); - MP_SIZE_T_SWAP (un, vn); - } - if (vn == 0) - { - r->_mp_size = 0; - return; - } - - uc = u->_mp_size < 0; - vc = v->_mp_size < 0; - rc = uc & vc; - - ux = -uc; - vx = -vc; - rx = -rc; - - /* If the smaller input is positive, higher limbs don't matter. */ - rn = vx ? un : vn; - - rp = MPZ_REALLOC (r, rn + (mp_size_t) rc); - - up = u->_mp_d; - vp = v->_mp_d; - - i = 0; - do - { - ul = (up[i] ^ ux) + uc; - uc = ul < uc; - - vl = (vp[i] ^ vx) + vc; - vc = vl < vc; - - rl = ( (ul & vl) ^ rx) + rc; - rc = rl < rc; - rp[i] = rl; - } - while (++i < vn); - assert (vc == 0); - - for (; i < rn; i++) - { - ul = (up[i] ^ ux) + uc; - uc = ul < uc; - - rl = ( (ul & vx) ^ rx) + rc; - rc = rl < rc; - rp[i] = rl; - } - if (rc) - rp[rn++] = rc; - else - rn = mpn_normalized_size (rp, rn); - - r->_mp_size = rx ? -rn : rn; -} - -void -mpz_ior (mpz_t r, const mpz_t u, const mpz_t v) -{ - mp_size_t un, vn, rn, i; - mp_ptr up, vp, rp; - - mp_limb_t ux, vx, rx; - mp_limb_t uc, vc, rc; - mp_limb_t ul, vl, rl; - - un = GMP_ABS (u->_mp_size); - vn = GMP_ABS (v->_mp_size); - if (un < vn) - { - MPZ_SRCPTR_SWAP (u, v); - MP_SIZE_T_SWAP (un, vn); - } - if (vn == 0) - { - mpz_set (r, u); - return; - } - - uc = u->_mp_size < 0; - vc = v->_mp_size < 0; - rc = uc | vc; - - ux = -uc; - vx = -vc; - rx = -rc; - - /* If the smaller input is negative, by sign extension higher limbs - don't matter. */ - rn = vx ? vn : un; - - rp = MPZ_REALLOC (r, rn + (mp_size_t) rc); - - up = u->_mp_d; - vp = v->_mp_d; - - i = 0; - do - { - ul = (up[i] ^ ux) + uc; - uc = ul < uc; - - vl = (vp[i] ^ vx) + vc; - vc = vl < vc; - - rl = ( (ul | vl) ^ rx) + rc; - rc = rl < rc; - rp[i] = rl; - } - while (++i < vn); - assert (vc == 0); - - for (; i < rn; i++) - { - ul = (up[i] ^ ux) + uc; - uc = ul < uc; - - rl = ( (ul | vx) ^ rx) + rc; - rc = rl < rc; - rp[i] = rl; - } - if (rc) - rp[rn++] = rc; - else - rn = mpn_normalized_size (rp, rn); - - r->_mp_size = rx ? -rn : rn; -} - -void -mpz_xor (mpz_t r, const mpz_t u, const mpz_t v) -{ - mp_size_t un, vn, i; - mp_ptr up, vp, rp; - - mp_limb_t ux, vx, rx; - mp_limb_t uc, vc, rc; - mp_limb_t ul, vl, rl; - - un = GMP_ABS (u->_mp_size); - vn = GMP_ABS (v->_mp_size); - if (un < vn) - { - MPZ_SRCPTR_SWAP (u, v); - MP_SIZE_T_SWAP (un, vn); - } - if (vn == 0) - { - mpz_set (r, u); - return; - } - - uc = u->_mp_size < 0; - vc = v->_mp_size < 0; - rc = uc ^ vc; - - ux = -uc; - vx = -vc; - rx = -rc; - - rp = MPZ_REALLOC (r, un + (mp_size_t) rc); - - up = u->_mp_d; - vp = v->_mp_d; - - i = 0; - do - { - ul = (up[i] ^ ux) + uc; - uc = ul < uc; - - vl = (vp[i] ^ vx) + vc; - vc = vl < vc; - - rl = (ul ^ vl ^ rx) + rc; - rc = rl < rc; - rp[i] = rl; - } - while (++i < vn); - assert (vc == 0); - - for (; i < un; i++) - { - ul = (up[i] ^ ux) + uc; - uc = ul < uc; - - rl = (ul ^ ux) + rc; - rc = rl < rc; - rp[i] = rl; - } - if (rc) - rp[un++] = rc; - else - un = mpn_normalized_size (rp, un); - - r->_mp_size = rx ? -un : un; -} - -static unsigned -gmp_popcount_limb (mp_limb_t x) -{ - unsigned c; - - /* Do 16 bits at a time, to avoid limb-sized constants. */ - for (c = 0; x > 0; x >>= 16) - { - unsigned w = ((x >> 1) & 0x5555) + (x & 0x5555); - w = ((w >> 2) & 0x3333) + (w & 0x3333); - w = ((w >> 4) & 0x0f0f) + (w & 0x0f0f); - w = (w >> 8) + (w & 0x00ff); - c += w; - } - return c; -} - -mp_bitcnt_t -mpn_popcount (mp_srcptr p, mp_size_t n) -{ - mp_size_t i; - mp_bitcnt_t c; - - for (c = 0, i = 0; i < n; i++) - c += gmp_popcount_limb (p[i]); - - return c; -} - -mp_bitcnt_t -mpz_popcount (const mpz_t u) -{ - mp_size_t un; - - un = u->_mp_size; - - if (un < 0) - return ~(mp_bitcnt_t) 0; - - return mpn_popcount (u->_mp_d, un); -} - -mp_bitcnt_t -mpz_hamdist (const mpz_t u, const mpz_t v) -{ - mp_size_t un, vn, i; - mp_limb_t uc, vc, ul, vl, comp; - mp_srcptr up, vp; - mp_bitcnt_t c; - - un = u->_mp_size; - vn = v->_mp_size; - - if ( (un ^ vn) < 0) - return ~(mp_bitcnt_t) 0; - - comp = - (uc = vc = (un < 0)); - if (uc) - { - assert (vn < 0); - un = -un; - vn = -vn; - } - - up = u->_mp_d; - vp = v->_mp_d; - - if (un < vn) - MPN_SRCPTR_SWAP (up, un, vp, vn); - - for (i = 0, c = 0; i < vn; i++) - { - ul = (up[i] ^ comp) + uc; - uc = ul < uc; - - vl = (vp[i] ^ comp) + vc; - vc = vl < vc; - - c += gmp_popcount_limb (ul ^ vl); - } - assert (vc == 0); - - for (; i < un; i++) - { - ul = (up[i] ^ comp) + uc; - uc = ul < uc; - - c += gmp_popcount_limb (ul ^ comp); - } - - return c; -} - -mp_bitcnt_t -mpz_scan1 (const mpz_t u, mp_bitcnt_t starting_bit) -{ - mp_ptr up; - mp_size_t us, un, i; - mp_limb_t limb, ux; - - us = u->_mp_size; - un = GMP_ABS (us); - i = starting_bit / GMP_LIMB_BITS; - - /* Past the end there's no 1 bits for u>=0, or an immediate 1 bit - for u<0. Notice this test picks up any u==0 too. */ - if (i >= un) - return (us >= 0 ? ~(mp_bitcnt_t) 0 : starting_bit); - - up = u->_mp_d; - ux = 0; - limb = up[i]; - - if (starting_bit != 0) - { - if (us < 0) - { - ux = mpn_zero_p (up, i); - limb = ~ limb + ux; - ux = - (mp_limb_t) (limb >= ux); - } - - /* Mask to 0 all bits before starting_bit, thus ignoring them. */ - limb &= (GMP_LIMB_MAX << (starting_bit % GMP_LIMB_BITS)); - } - - return mpn_common_scan (limb, i, up, un, ux); -} - -mp_bitcnt_t -mpz_scan0 (const mpz_t u, mp_bitcnt_t starting_bit) -{ - mp_ptr up; - mp_size_t us, un, i; - mp_limb_t limb, ux; - - us = u->_mp_size; - ux = - (mp_limb_t) (us >= 0); - un = GMP_ABS (us); - i = starting_bit / GMP_LIMB_BITS; - - /* When past end, there's an immediate 0 bit for u>=0, or no 0 bits for - u<0. Notice this test picks up all cases of u==0 too. */ - if (i >= un) - return (ux ? starting_bit : ~(mp_bitcnt_t) 0); - - up = u->_mp_d; - limb = up[i] ^ ux; - - if (ux == 0) - limb -= mpn_zero_p (up, i); /* limb = ~(~limb + zero_p) */ - - /* Mask all bits before starting_bit, thus ignoring them. */ - limb &= (GMP_LIMB_MAX << (starting_bit % GMP_LIMB_BITS)); - - return mpn_common_scan (limb, i, up, un, ux); -} - - -/* MPZ base conversion. */ - -size_t -mpz_sizeinbase (const mpz_t u, int base) -{ - mp_size_t un; - mp_srcptr up; - mp_ptr tp; - mp_bitcnt_t bits; - struct gmp_div_inverse bi; - size_t ndigits; - - assert (base >= 2); - assert (base <= 36); - - un = GMP_ABS (u->_mp_size); - if (un == 0) - return 1; - - up = u->_mp_d; - - bits = (un - 1) * GMP_LIMB_BITS + mpn_limb_size_in_base_2 (up[un-1]); - switch (base) - { - case 2: - return bits; - case 4: - return (bits + 1) / 2; - case 8: - return (bits + 2) / 3; - case 16: - return (bits + 3) / 4; - case 32: - return (bits + 4) / 5; - /* FIXME: Do something more clever for the common case of base - 10. */ - } - - tp = gmp_xalloc_limbs (un); - mpn_copyi (tp, up, un); - mpn_div_qr_1_invert (&bi, base); - - ndigits = 0; - do - { - ndigits++; - mpn_div_qr_1_preinv (tp, tp, un, &bi); - un -= (tp[un-1] == 0); - } - while (un > 0); - - gmp_free (tp); - return ndigits; -} - -char * -mpz_get_str (char *sp, int base, const mpz_t u) -{ - unsigned bits; - const char *digits; - mp_size_t un; - size_t i, sn; - - if (base >= 0) - { - digits = "0123456789abcdefghijklmnopqrstuvwxyz"; - } - else - { - base = -base; - digits = "0123456789ABCDEFGHIJKLMNOPQRSTUVWXYZ"; - } - if (base <= 1) - base = 10; - if (base > 36) - return NULL; - - sn = 1 + mpz_sizeinbase (u, base); - if (!sp) - sp = (char *) gmp_xalloc (1 + sn); - - un = GMP_ABS (u->_mp_size); - - if (un == 0) - { - sp[0] = '0'; - sp[1] = '\0'; - return sp; - } - - i = 0; - - if (u->_mp_size < 0) - sp[i++] = '-'; - - bits = mpn_base_power_of_two_p (base); - - if (bits) - /* Not modified in this case. */ - sn = i + mpn_get_str_bits ((unsigned char *) sp + i, bits, u->_mp_d, un); - else - { - struct mpn_base_info info; - mp_ptr tp; - - mpn_get_base_info (&info, base); - tp = gmp_xalloc_limbs (un); - mpn_copyi (tp, u->_mp_d, un); - - sn = i + mpn_get_str_other ((unsigned char *) sp + i, base, &info, tp, un); - gmp_free (tp); - } - - for (; i < sn; i++) - sp[i] = digits[(unsigned char) sp[i]]; - - sp[sn] = '\0'; - return sp; -} - -int -mpz_set_str (mpz_t r, const char *sp, int base) -{ - unsigned bits; - mp_size_t rn, alloc; - mp_ptr rp; - size_t dn; - int sign; - unsigned char *dp; - - assert (base == 0 || (base >= 2 && base <= 36)); - - while (isspace( (unsigned char) *sp)) - sp++; - - sign = (*sp == '-'); - sp += sign; - - if (base == 0) - { - if (sp[0] == '0') - { - if (sp[1] == 'x' || sp[1] == 'X') - { - base = 16; - sp += 2; - } - else if (sp[1] == 'b' || sp[1] == 'B') - { - base = 2; - sp += 2; - } - else - base = 8; - } - else - base = 10; - } - - if (!*sp) - { - r->_mp_size = 0; - return -1; - } - dp = (unsigned char *) gmp_xalloc (strlen (sp)); - - for (dn = 0; *sp; sp++) - { - unsigned digit; - - if (isspace ((unsigned char) *sp)) - continue; - else if (*sp >= '0' && *sp <= '9') - digit = *sp - '0'; - else if (*sp >= 'a' && *sp <= 'z') - digit = *sp - 'a' + 10; - else if (*sp >= 'A' && *sp <= 'Z') - digit = *sp - 'A' + 10; - else - digit = base; /* fail */ - - if (digit >= (unsigned) base) - { - gmp_free (dp); - r->_mp_size = 0; - return -1; - } - - dp[dn++] = digit; - } - - if (!dn) - { - gmp_free (dp); - r->_mp_size = 0; - return -1; - } - bits = mpn_base_power_of_two_p (base); - - if (bits > 0) - { - alloc = (dn * bits + GMP_LIMB_BITS - 1) / GMP_LIMB_BITS; - rp = MPZ_REALLOC (r, alloc); - rn = mpn_set_str_bits (rp, dp, dn, bits); - } - else - { - struct mpn_base_info info; - mpn_get_base_info (&info, base); - alloc = (dn + info.exp - 1) / info.exp; - rp = MPZ_REALLOC (r, alloc); - rn = mpn_set_str_other (rp, dp, dn, base, &info); - /* Normalization, needed for all-zero input. */ - assert (rn > 0); - rn -= rp[rn-1] == 0; - } - assert (rn <= alloc); - gmp_free (dp); - - r->_mp_size = sign ? - rn : rn; - - return 0; -} - -int -mpz_init_set_str (mpz_t r, const char *sp, int base) -{ - mpz_init (r); - return mpz_set_str (r, sp, base); -} - -size_t -mpz_out_str (FILE *stream, int base, const mpz_t x) -{ - char *str; - size_t len; - - str = mpz_get_str (NULL, base, x); - len = strlen (str); - len = fwrite (str, 1, len, stream); - gmp_free (str); - return len; -} - - -static int -gmp_detect_endian (void) -{ - static const int i = 2; - const unsigned char *p = (const unsigned char *) &i; - return 1 - *p; -} - -/* Import and export. Does not support nails. */ -void -mpz_import (mpz_t r, size_t count, int order, size_t size, int endian, - size_t nails, const void *src) -{ - const unsigned char *p; - ptrdiff_t word_step; - mp_ptr rp; - mp_size_t rn; - - /* The current (partial) limb. */ - mp_limb_t limb; - /* The number of bytes already copied to this limb (starting from - the low end). */ - size_t bytes; - /* The index where the limb should be stored, when completed. */ - mp_size_t i; - - if (nails != 0) - gmp_die ("mpz_import: Nails not supported."); - - assert (order == 1 || order == -1); - assert (endian >= -1 && endian <= 1); - - if (endian == 0) - endian = gmp_detect_endian (); - - p = (unsigned char *) src; - - word_step = (order != endian) ? 2 * size : 0; - - /* Process bytes from the least significant end, so point p at the - least significant word. */ - if (order == 1) - { - p += size * (count - 1); - word_step = - word_step; - } - - /* And at least significant byte of that word. */ - if (endian == 1) - p += (size - 1); - - rn = (size * count + sizeof(mp_limb_t) - 1) / sizeof(mp_limb_t); - rp = MPZ_REALLOC (r, rn); - - for (limb = 0, bytes = 0, i = 0; count > 0; count--, p += word_step) - { - size_t j; - for (j = 0; j < size; j++, p -= (ptrdiff_t) endian) - { - limb |= (mp_limb_t) *p << (bytes++ * CHAR_BIT); - if (bytes == sizeof(mp_limb_t)) - { - rp[i++] = limb; - bytes = 0; - limb = 0; - } - } - } - assert (i + (bytes > 0) == rn); - if (limb != 0) - rp[i++] = limb; - else - i = mpn_normalized_size (rp, i); - - r->_mp_size = i; -} - -void * -mpz_export (void *r, size_t *countp, int order, size_t size, int endian, - size_t nails, const mpz_t u) -{ - size_t count; - mp_size_t un; - - if (nails != 0) - gmp_die ("mpz_import: Nails not supported."); - - assert (order == 1 || order == -1); - assert (endian >= -1 && endian <= 1); - assert (size > 0 || u->_mp_size == 0); - - un = u->_mp_size; - count = 0; - if (un != 0) - { - size_t k; - unsigned char *p; - ptrdiff_t word_step; - /* The current (partial) limb. */ - mp_limb_t limb; - /* The number of bytes left to to in this limb. */ - size_t bytes; - /* The index where the limb was read. */ - mp_size_t i; - - un = GMP_ABS (un); - - /* Count bytes in top limb. */ - limb = u->_mp_d[un-1]; - assert (limb != 0); - - k = 0; - do { - k++; limb >>= CHAR_BIT; - } while (limb != 0); - - count = (k + (un-1) * sizeof (mp_limb_t) + size - 1) / size; - - if (!r) - r = gmp_xalloc (count * size); - - if (endian == 0) - endian = gmp_detect_endian (); - - p = (unsigned char *) r; - - word_step = (order != endian) ? 2 * size : 0; - - /* Process bytes from the least significant end, so point p at the - least significant word. */ - if (order == 1) - { - p += size * (count - 1); - word_step = - word_step; - } - - /* And at least significant byte of that word. */ - if (endian == 1) - p += (size - 1); - - for (bytes = 0, i = 0, k = 0; k < count; k++, p += word_step) - { - size_t j; - for (j = 0; j < size; j++, p -= (ptrdiff_t) endian) - { - if (bytes == 0) - { - if (i < un) - limb = u->_mp_d[i++]; - bytes = sizeof (mp_limb_t); - } - *p = limb; - limb >>= CHAR_BIT; - bytes--; - } - } - assert (i == un); - assert (k == count); - } - - if (countp) - *countp = count; - - return r; -} diff --git a/Qt-Secret/GMP/mini-gmp.h b/Qt-Secret/GMP/mini-gmp.h deleted file mode 100644 index 44ef5f5..0000000 --- a/Qt-Secret/GMP/mini-gmp.h +++ /dev/null @@ -1,303 +0,0 @@ -/* mini-gmp, a minimalistic implementation of a GNU GMP subset. - -Copyright 2011-2015 Free Software Foundation, Inc. - -This file is part of the GNU MP Library. - -The GNU MP Library is free software; you can redistribute it and/or modify -it under the terms of either: - - * the GNU Lesser General Public License as published by the Free - Software Foundation; either version 3 of the License, or (at your - option) any later version. - -or - - * the GNU General Public License as published by the Free Software - Foundation; either version 2 of the License, or (at your option) any - later version. - -or both in parallel, as here. - -The GNU MP Library is distributed in the hope that it will be useful, but -WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY -or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License -for more details. - -You should have received copies of the GNU General Public License and the -GNU Lesser General Public License along with the GNU MP Library. If not, -see https://www.gnu.org/licenses/. */ - -/* About mini-gmp: This is a minimal implementation of a subset of the - GMP interface. It is intended for inclusion into applications which - have modest bignums needs, as a fallback when the real GMP library - is not installed. - - This file defines the public interface. */ - -#ifndef MINI_GMP -#define MINI_GMP - -/* For size_t */ -#include - -#if defined (__cplusplus) -extern "C" { -#endif - -#define UN_USED(X) (void)X - -void mp_set_memory_functions (void *(*) (size_t), - void *(*) (void *, size_t, size_t), - void (*) (void *, size_t)); - -void mp_get_memory_functions (void *(**) (size_t), - void *(**) (void *, size_t, size_t), - void (**) (void *, size_t)); - -typedef unsigned long long uIntMpz; -typedef long long intMpz; - -typedef uIntMpz mp_limb_t; -typedef intMpz mp_size_t; -typedef uIntMpz mp_bitcnt_t; - -typedef mp_limb_t *mp_ptr; -typedef const mp_limb_t *mp_srcptr; - -typedef struct -{ - int _mp_alloc; /* Number of *limbs* allocated and pointed - to by the _mp_d field. */ - int _mp_size; /* abs(_mp_size) is the number of limbs the - last field points to. If _mp_size is - negative this is a negative number. */ - mp_limb_t *_mp_d; /* Pointer to the limbs. */ -} __mpz_struct; - -typedef __mpz_struct mpz_t[1]; - -typedef __mpz_struct *mpz_ptr; -typedef const __mpz_struct *mpz_srcptr; - -extern const int mp_bits_per_limb; - -void mpn_copyi (mp_ptr, mp_srcptr, mp_size_t); -void mpn_copyd (mp_ptr, mp_srcptr, mp_size_t); -void mpn_zero (mp_ptr, mp_size_t); - -int mpn_cmp (mp_srcptr, mp_srcptr, mp_size_t); -int mpn_zero_p (mp_srcptr, mp_size_t); - -mp_limb_t mpn_add_1 (mp_ptr, mp_srcptr, mp_size_t, mp_limb_t); -mp_limb_t mpn_add_n (mp_ptr, mp_srcptr, mp_srcptr, mp_size_t); -mp_limb_t mpn_add (mp_ptr, mp_srcptr, mp_size_t, mp_srcptr, mp_size_t); - -mp_limb_t mpn_sub_1 (mp_ptr, mp_srcptr, mp_size_t, mp_limb_t); -mp_limb_t mpn_sub_n (mp_ptr, mp_srcptr, mp_srcptr, mp_size_t); -mp_limb_t mpn_sub (mp_ptr, mp_srcptr, mp_size_t, mp_srcptr, mp_size_t); - -mp_limb_t mpn_mul_1 (mp_ptr, mp_srcptr, mp_size_t, mp_limb_t); -mp_limb_t mpn_addmul_1 (mp_ptr, mp_srcptr, mp_size_t, mp_limb_t); -mp_limb_t mpn_submul_1 (mp_ptr, mp_srcptr, mp_size_t, mp_limb_t); - -mp_limb_t mpn_mul (mp_ptr, mp_srcptr, mp_size_t, mp_srcptr, mp_size_t); -void mpn_mul_n (mp_ptr, mp_srcptr, mp_srcptr, mp_size_t); -void mpn_sqr (mp_ptr, mp_srcptr, mp_size_t); -int mpn_perfect_square_p (mp_srcptr, mp_size_t); -mp_size_t mpn_sqrtrem (mp_ptr, mp_ptr, mp_srcptr, mp_size_t); - -mp_limb_t mpn_lshift (mp_ptr, mp_srcptr, mp_size_t, unsigned int); -mp_limb_t mpn_rshift (mp_ptr, mp_srcptr, mp_size_t, unsigned int); - -mp_bitcnt_t mpn_scan0 (mp_srcptr, mp_bitcnt_t); -mp_bitcnt_t mpn_scan1 (mp_srcptr, mp_bitcnt_t); - -void mpn_com (mp_ptr, mp_srcptr, mp_size_t); -mp_limb_t mpn_neg (mp_ptr, mp_srcptr, mp_size_t); - -mp_bitcnt_t mpn_popcount (mp_srcptr, mp_size_t); - -mp_limb_t mpn_invert_3by2 (mp_limb_t, mp_limb_t); -#define mpn_invert_limb(x) mpn_invert_3by2 ((x), 0) - -size_t mpn_get_str (unsigned char *, int, mp_ptr, mp_size_t); -mp_size_t mpn_set_str (mp_ptr, const unsigned char *, size_t, int); - -void mpz_init (mpz_t); -void mpz_init2 (mpz_t, mp_bitcnt_t); -void mpz_clear (mpz_t); - -#define mpz_odd_p(z) (((z)->_mp_size != 0) & (int) (z)->_mp_d[0]) -#define mpz_even_p(z) (! mpz_odd_p (z)) - -int mpz_sgn (const mpz_t); -int mpz_cmp_si (const mpz_t, intMpz); -int mpz_cmp_ui (const mpz_t, uIntMpz); -int mpz_cmp (const mpz_t, const mpz_t); -int mpz_cmpabs_ui (const mpz_t, uIntMpz); -int mpz_cmpabs (const mpz_t, const mpz_t); -int mpz_cmp_d (const mpz_t, double); -int mpz_cmpabs_d (const mpz_t, double); - -void mpz_abs (mpz_t, const mpz_t); -void mpz_neg (mpz_t, const mpz_t); -void mpz_swap (mpz_t, mpz_t); - -void mpz_add_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_add (mpz_t, const mpz_t, const mpz_t); -void mpz_sub_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_ui_sub (mpz_t, uIntMpz, const mpz_t); -void mpz_sub (mpz_t, const mpz_t, const mpz_t); - -void mpz_mul_si (mpz_t, const mpz_t, intMpz); -void mpz_mul_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_mul (mpz_t, const mpz_t, const mpz_t); -void mpz_mul_2exp (mpz_t, const mpz_t, mp_bitcnt_t); -void mpz_addmul_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_addmul (mpz_t, const mpz_t, const mpz_t); -void mpz_submul_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_submul (mpz_t, const mpz_t, const mpz_t); - -void mpz_cdiv_qr (mpz_t, mpz_t, const mpz_t, const mpz_t); -void mpz_fdiv_qr (mpz_t, mpz_t, const mpz_t, const mpz_t); -void mpz_tdiv_qr (mpz_t, mpz_t, const mpz_t, const mpz_t); -void mpz_cdiv_q (mpz_t, const mpz_t, const mpz_t); -void mpz_fdiv_q (mpz_t, const mpz_t, const mpz_t); -void mpz_tdiv_q (mpz_t, const mpz_t, const mpz_t); -void mpz_cdiv_r (mpz_t, const mpz_t, const mpz_t); -void mpz_fdiv_r (mpz_t, const mpz_t, const mpz_t); -void mpz_tdiv_r (mpz_t, const mpz_t, const mpz_t); - -void mpz_cdiv_q_2exp (mpz_t, const mpz_t, mp_bitcnt_t); -void mpz_fdiv_q_2exp (mpz_t, const mpz_t, mp_bitcnt_t); -void mpz_tdiv_q_2exp (mpz_t, const mpz_t, mp_bitcnt_t); -void mpz_cdiv_r_2exp (mpz_t, const mpz_t, mp_bitcnt_t); -void mpz_fdiv_r_2exp (mpz_t, const mpz_t, mp_bitcnt_t); -void mpz_tdiv_r_2exp (mpz_t, const mpz_t, mp_bitcnt_t); - -void mpz_mod (mpz_t, const mpz_t, const mpz_t); - -void mpz_divexact (mpz_t, const mpz_t, const mpz_t); - -int mpz_divisible_p (const mpz_t, const mpz_t); -int mpz_congruent_p (const mpz_t, const mpz_t, const mpz_t); - -uIntMpz mpz_cdiv_qr_ui (mpz_t, mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_fdiv_qr_ui (mpz_t, mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_tdiv_qr_ui (mpz_t, mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_cdiv_q_ui (mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_fdiv_q_ui (mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_tdiv_q_ui (mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_cdiv_r_ui (mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_fdiv_r_ui (mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_tdiv_r_ui (mpz_t, const mpz_t, uIntMpz); -uIntMpz mpz_cdiv_ui (const mpz_t, uIntMpz); -uIntMpz mpz_fdiv_ui (const mpz_t, uIntMpz); -uIntMpz mpz_tdiv_ui (const mpz_t, uIntMpz); - -uIntMpz mpz_mod_ui (mpz_t, const mpz_t, uIntMpz); - -void mpz_divexact_ui (mpz_t, const mpz_t, uIntMpz); - -int mpz_divisible_ui_p (const mpz_t, uIntMpz); - -uIntMpz mpz_gcd_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_gcd (mpz_t, const mpz_t, const mpz_t); -void mpz_gcdext (mpz_t, mpz_t, mpz_t, const mpz_t, const mpz_t); -void mpz_lcm_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_lcm (mpz_t, const mpz_t, const mpz_t); -int mpz_invert (mpz_t, const mpz_t, const mpz_t); - -void mpz_sqrtrem (mpz_t, mpz_t, const mpz_t); -void mpz_sqrt (mpz_t, const mpz_t); -int mpz_perfect_square_p (const mpz_t); - -void mpz_pow_ui (mpz_t, const mpz_t, uIntMpz); -void mpz_ui_pow_ui (mpz_t, uIntMpz, uIntMpz); -void mpz_powm (mpz_t, const mpz_t, const mpz_t, const mpz_t); -void mpz_powm_ui (mpz_t, const mpz_t, uIntMpz, const mpz_t); - -void mpz_rootrem (mpz_t, mpz_t, const mpz_t, uIntMpz); -int mpz_root (mpz_t, const mpz_t, uIntMpz); - -void mpz_fac_ui (mpz_t, uIntMpz); -void mpz_bin_uiui (mpz_t, uIntMpz, uIntMpz); - -int mpz_probab_prime_p (const mpz_t, int); - -int mpz_tstbit (const mpz_t, mp_bitcnt_t); -void mpz_setbit (mpz_t, mp_bitcnt_t); -void mpz_clrbit (mpz_t, mp_bitcnt_t); -void mpz_combit (mpz_t, mp_bitcnt_t); - -void mpz_com (mpz_t, const mpz_t); -void mpz_and (mpz_t, const mpz_t, const mpz_t); -void mpz_ior (mpz_t, const mpz_t, const mpz_t); -void mpz_xor (mpz_t, const mpz_t, const mpz_t); - -mp_bitcnt_t mpz_popcount (const mpz_t); -mp_bitcnt_t mpz_hamdist (const mpz_t, const mpz_t); -mp_bitcnt_t mpz_scan0 (const mpz_t, mp_bitcnt_t); -mp_bitcnt_t mpz_scan1 (const mpz_t, mp_bitcnt_t); - -int mpz_fits_slong_p (const mpz_t); -int mpz_fits_ulong_p (const mpz_t); -intMpz mpz_get_si (const mpz_t); -uIntMpz mpz_get_ui (const mpz_t); -double mpz_get_d (const mpz_t); -size_t mpz_size (const mpz_t); -mp_limb_t mpz_getlimbn (const mpz_t, mp_size_t); - -void mpz_realloc2 (mpz_t, mp_bitcnt_t); -mp_srcptr mpz_limbs_read (mpz_srcptr); -mp_ptr mpz_limbs_modify (mpz_t, mp_size_t); -mp_ptr mpz_limbs_write (mpz_t, mp_size_t); -void mpz_limbs_finish (mpz_t, mp_size_t); -mpz_srcptr mpz_roinit_n (mpz_t, mp_srcptr, mp_size_t); - -#define MPZ_ROINIT_N(xp, xs) {{0, (xs),(xp) }} - -void mpz_set_si (mpz_t, intMpz); -void mpz_set_ui (mpz_t, uIntMpz); -void mpz_set (mpz_t, const mpz_t); -void mpz_set_d (mpz_t, double); - -void mpz_init_set_si (mpz_t, intMpz); -void mpz_init_set_ui (mpz_t, uIntMpz); -void mpz_init_set (mpz_t, const mpz_t); -void mpz_init_set_d (mpz_t, double); - -size_t mpz_sizeinbase (const mpz_t, int); -char *mpz_get_str (char *, int, const mpz_t); -int mpz_set_str (mpz_t, const char *, int); -int mpz_init_set_str (mpz_t, const char *, int); - -/* This long list taken from gmp.h. */ -/* For reference, "defined(EOF)" cannot be used here. In g++ 2.95.4, - defines EOF but not FILE. */ -#if defined (FILE) \ - || defined (H_STDIO) \ - || defined (_H_STDIO) /* AIX */ \ - || defined (_STDIO_H) /* glibc, Sun, SCO */ \ - || defined (_STDIO_H_) /* BSD, OSF */ \ - || defined (__STDIO_H) /* Borland */ \ - || defined (__STDIO_H__) /* IRIX */ \ - || defined (_STDIO_INCLUDED) /* HPUX */ \ - || defined (__dj_include_stdio_h_) /* DJGPP */ \ - || defined (_FILE_DEFINED) /* Microsoft */ \ - || defined (__STDIO__) /* Apple MPW MrC */ \ - || defined (_MSL_STDIO_H) /* Metrowerks */ \ - || defined (_STDIO_H_INCLUDED) /* QNX4 */ \ - || defined (_ISO_STDIO_ISO_H) /* Sun C++ */ \ - || defined (__STDIO_LOADED) /* VMS */ -size_t mpz_out_str (FILE *, int, const mpz_t); -#endif - -void mpz_import (mpz_t, size_t, int, size_t, int, size_t, const void *); -void *mpz_export (void *, size_t *, int, size_t, int, size_t, const mpz_t); - -#if defined (__cplusplus) -} -#endif -#endif /* MINI_GMP */ diff --git a/Qt-Secret/GMP/minigmp_global.h b/Qt-Secret/GMP/minigmp_global.h deleted file mode 100644 index 5757875..0000000 --- a/Qt-Secret/GMP/minigmp_global.h +++ /dev/null @@ -1,21 +0,0 @@ -//# -//# Copyright (C) 2018-2019 QuasarApp. -//# Distributed under the lgplv3 software license, see the accompanying -//# Everyone is permitted to copy and distribute verbatim copies -//# of this license document, but changing it is not allowed. -//# - -#ifndef MINIGMP_GLOBAL_H -#define MINIGMP_GLOBAL_H - -#ifdef _WIN32 -# define MINIGMPSHARED_EXPORT __declspec(dllexport) -#endif - -#ifdef linux -# define MINIGMPSHARED_EXPORT __attribute__((visibility("default"))) -#endif - - - -#endif //MINIGMP_GLOBAL_H diff --git a/Qt-Secret/LICENSE b/Qt-Secret/LICENSE deleted file mode 100644 index 65c5ca8..0000000 --- a/Qt-Secret/LICENSE +++ /dev/null @@ -1,165 +0,0 @@ - GNU LESSER GENERAL PUBLIC LICENSE - Version 3, 29 June 2007 - - Copyright (C) 2007 Free Software Foundation, Inc. - Everyone is permitted to copy and distribute verbatim copies - of this license document, but changing it is not allowed. - - - This version of the GNU Lesser General Public License incorporates -the terms and conditions of version 3 of the GNU General Public -License, supplemented by the additional permissions listed below. - - 0. Additional Definitions. - - As used herein, "this License" refers to version 3 of the GNU Lesser -General Public License, and the "GNU GPL" refers to version 3 of the GNU -General Public License. - - "The Library" refers to a covered work governed by this License, -other than an Application or a Combined Work as defined below. - - An "Application" is any work that makes use of an interface provided -by the Library, but which is not otherwise based on the Library. -Defining a subclass of a class defined by the Library is deemed a mode -of using an interface provided by the Library. - - A "Combined Work" is a work produced by combining or linking an -Application with the Library. The particular version of the Library -with which the Combined Work was made is also called the "Linked -Version". - - The "Minimal Corresponding Source" for a Combined Work means the -Corresponding Source for the Combined Work, excluding any source code -for portions of the Combined Work that, considered in isolation, are -based on the Application, and not on the Linked Version. - - The "Corresponding Application Code" for a Combined Work means the -object code and/or source code for the Application, including any data -and utility programs needed for reproducing the Combined Work from the -Application, but excluding the System Libraries of the Combined Work. - - 1. Exception to Section 3 of the GNU GPL. - - You may convey a covered work under sections 3 and 4 of this License -without being bound by section 3 of the GNU GPL. - - 2. Conveying Modified Versions. - - If you modify a copy of the Library, and, in your modifications, a -facility refers to a function or data to be supplied by an Application -that uses the facility (other than as an argument passed when the -facility is invoked), then you may convey a copy of the modified -version: - - a) under this License, provided that you make a good faith effort to - ensure that, in the event an Application does not supply the - function or data, the facility still operates, and performs - whatever part of its purpose remains meaningful, or - - b) under the GNU GPL, with none of the additional permissions of - this License applicable to that copy. - - 3. Object Code Incorporating Material from Library Header Files. - - The object code form of an Application may incorporate material from -a header file that is part of the Library. You may convey such object -code under terms of your choice, provided that, if the incorporated -material is not limited to numerical parameters, data structure -layouts and accessors, or small macros, inline functions and templates -(ten or fewer lines in length), you do both of the following: - - a) Give prominent notice with each copy of the object code that the - Library is used in it and that the Library and its use are - covered by this License. - - b) Accompany the object code with a copy of the GNU GPL and this license - document. - - 4. Combined Works. - - You may convey a Combined Work under terms of your choice that, -taken together, effectively do not restrict modification of the -portions of the Library contained in the Combined Work and reverse -engineering for debugging such modifications, if you also do each of -the following: - - a) Give prominent notice with each copy of the Combined Work that - the Library is used in it and that the Library and its use are - covered by this License. - - b) Accompany the Combined Work with a copy of the GNU GPL and this license - document. - - c) For a Combined Work that displays copyright notices during - execution, include the copyright notice for the Library among - these notices, as well as a reference directing the user to the - copies of the GNU GPL and this license document. - - d) Do one of the following: - - 0) Convey the Minimal Corresponding Source under the terms of this - License, and the Corresponding Application Code in a form - suitable for, and under terms that permit, the user to - recombine or relink the Application with a modified version of - the Linked Version to produce a modified Combined Work, in the - manner specified by section 6 of the GNU GPL for conveying - Corresponding Source. - - 1) Use a suitable shared library mechanism for linking with the - Library. A suitable mechanism is one that (a) uses at run time - a copy of the Library already present on the user's computer - system, and (b) will operate properly with a modified version - of the Library that is interface-compatible with the Linked - Version. - - e) Provide Installation Information, but only if you would otherwise - be required to provide such information under section 6 of the - GNU GPL, and only to the extent that such information is - necessary to install and execute a modified version of the - Combined Work produced by recombining or relinking the - Application with a modified version of the Linked Version. (If - you use option 4d0, the Installation Information must accompany - the Minimal Corresponding Source and Corresponding Application - Code. If you use option 4d1, you must provide the Installation - Information in the manner specified by section 6 of the GNU GPL - for conveying Corresponding Source.) - - 5. Combined Libraries. - - You may place library facilities that are a work based on the -Library side by side in a single library together with other library -facilities that are not Applications and are not covered by this -License, and convey such a combined library under terms of your -choice, if you do both of the following: - - a) Accompany the combined library with a copy of the same work based - on the Library, uncombined with any other library facilities, - conveyed under the terms of this License. - - b) Give prominent notice with the combined library that part of it - is a work based on the Library, and explaining where to find the - accompanying uncombined form of the same work. - - 6. Revised Versions of the GNU Lesser General Public License. - - The Free Software Foundation may publish revised and/or new versions -of the GNU Lesser General Public License from time to time. Such new -versions will be similar in spirit to the present version, but may -differ in detail to address new problems or concerns. - - Each version is given a distinguishing version number. If the -Library as you received it specifies that a certain numbered version -of the GNU Lesser General Public License "or any later version" -applies to it, you have the option of following the terms and -conditions either of that published version or of any later version -published by the Free Software Foundation. If the Library as you -received it does not specify a version number of the GNU Lesser -General Public License, you may choose any version of the GNU Lesser -General Public License ever published by the Free Software Foundation. - - If the Library as you received it specifies that a proxy can decide -whether future versions of the GNU Lesser General Public License shall -apply, that proxy's public statement of acceptance of any version is -permanent authorization for you to choose that version for the -Library. diff --git a/Qt-Secret/Qt-RSA/qrsaencryption.cpp b/Qt-Secret/Qt-RSA/qrsaencryption.cpp deleted file mode 100644 index 01a3344..0000000 --- a/Qt-Secret/Qt-RSA/qrsaencryption.cpp +++ /dev/null @@ -1,403 +0,0 @@ -//# -//# Copyright (C) 2018-2019 QuasarApp. -//# Distributed under the lgplv3 software license, see the accompanying -//# Everyone is permitted to copy and distribute verbatim copies -//# of this license document, but changing it is not allowed. -//# - -#include "qrsaencryption.h" - -#include -#include -#include -#include - -#define KEY_GEN_LIMIT 10 - -const QString SIGN_MARKER = "-SIGN-"; -const int signMarkerLength = SIGN_MARKER.length(); - -QRSAEncryption::INT eulerFunc(const QRSAEncryption::INT &p, const QRSAEncryption::INT &q) { - return (p - 1) * (q - 1); -} - -bool QRSAEncryption::isMutuallyPrime(const INT &a, const INT &b) { - if ( (!(a % 2) && !(b % 2)) - || (!(a % 3) && !(b % 3)) - || (!(a % 5) && !(b % 5)) - || (!(a % 7) && !(b % 7)) - ) return false; - - return INT().gcd(a, b) == 1; -} - -QRSAEncryption::Rsa QRSAEncryption::getBitsSize(const INT &i) const { - int rsaBits = RSA_64; - int intBits = i.sizeBits(); - - while (rsaBits < intBits) { - rsaBits *= 2; - } - - return static_cast(rsaBits); - -} - -QRSAEncryption::Rsa QRSAEncryption::getBitsSize(const QByteArray &key) const { - if (isValidRsaKey(key)) { - return static_cast(key.size() * 4); - } - - return QRSAEncryption::Rsa::Invalid; -} - -QRSAEncryption::INT QRSAEncryption::fromArray(const QByteArray &array) const { - INT res = 0; - res.fromHex(array.toHex().toStdString()); - return res; -} - -QByteArray QRSAEncryption::toArray(const INT &i, short sizeBlok) { - QByteArray res; - res = QByteArray::fromHex(QByteArray::fromStdString(i.getString(16))); - - if (sizeBlok < 0) { - return res; - } - - while (res.size() < sizeBlok) { - res.push_front(char(0)); - } - - return res.left(sizeBlok); -} - -QRSAEncryption::INT QRSAEncryption::randomNumber(bool fullFill) const { - - srand(std::chrono::duration_cast - (std::chrono::system_clock::now().time_since_epoch()).count() - % std::numeric_limits::max()); - - INT res{1}; - - if(fullFill) { - - while(res.longBits() < _rsa) { - res *= (rand() % (std::numeric_limits::max() - 1)) + 1; - } - } else { - - int longDiff = _rsa / (sizeof (int) * 8); - - while (longDiff > 0) { - longDiff--; - res *= (rand() % (std::numeric_limits::max() - 1)) + 1; - } - } - - return res; -} - -QRSAEncryption::INT QRSAEncryption::toPrime(INT n) const { - if (!(n % 2)) { - ++n; - } - - INT LN = n; - INT RN = n; - - while (true) { - - if (LN.isPrime(false)) return LN; - - RN+=2; - - if (RN.isPrime(false)) return RN; - LN-=2; - } -} - -QRSAEncryption::INT QRSAEncryption::randomPrimeNumber(INT no) const { - srand(static_cast(time(nullptr))); - - // max INT - INT max('1', _rsa / 2, 2); - - auto p = toPrime(randomNumber() % max); - while(p == no) p = toPrime(randomNumber() % max); - - return p; -} - -QRSAEncryption::INT QRSAEncryption::extEuclid(INT a, INT b) const { - INT x = 0, y = 1, u = 1, v = 0, gcd = b, m, n, q, r; - while (a != 0) { - q = gcd / a; - r = gcd % a; - m = x - u * q; - n = y - v * q; - gcd = a; - a = r; - x = u; - y = v; - u = m; - v = n; - } - return y; -} - -short QRSAEncryption::getBlockSize(INT i) const { - return static_cast(i.longBytes()) - 1; -} - -QByteArray QRSAEncryption::encodeBlok(const INT &block, const INT &e, const INT &m, short blockSize) { - - return toArray(INT::powm(block, e, m), blockSize); -} - -QByteArray QRSAEncryption::decodeBlok(const INT &block, const INT &d, const INT &m, short blockSize) { - - return toArray(INT::powm(block, d, m), blockSize); -} - -QRSAEncryption::QRSAEncryption(Rsa rsa) { - _rsa = rsa; -} - -bool QRSAEncryption::generatePairKeyS(QByteArray &pubKey, QByteArray &privKey, QRSAEncryption::Rsa rsa) { - qWarning() << "method " << Q_FUNC_INFO << - " will be deleted in newxt version. please use generatePairKey method"; - return generatePairKey(pubKey, privKey, rsa); -} - -QByteArray QRSAEncryption::encodeS(const QByteArray &rawData, const QByteArray &pubKey, QRSAEncryption::Rsa rsa, QRSAEncryption::BlockSize blockSizeMode) { - qWarning() << "method " << Q_FUNC_INFO << - " will be deleted in newxt version. please use encode method"; - return encode(rawData, pubKey, rsa, blockSizeMode); - -} - -QByteArray QRSAEncryption::decodeS(const QByteArray &rawData, const QByteArray &privKey, QRSAEncryption::Rsa rsa, QRSAEncryption::BlockSize blockSizeMode) { - qWarning() << "method " << Q_FUNC_INFO << - " will be deleted in newxt version. please use decode method"; - return decode(rawData, privKey, rsa, blockSizeMode); - -} - -QByteArray QRSAEncryption::signMessageS(QByteArray rawData, const QByteArray &privKey, QRSAEncryption::Rsa rsa) { - qWarning() << "method " << Q_FUNC_INFO << - " will be deleted in newxt version. please use signMessage method"; - return signMessage(rawData, privKey, rsa); - -} - -bool QRSAEncryption::checkSignMessageS(const QByteArray &rawData, const QByteArray &pubKey, QRSAEncryption::Rsa rsa) { - qWarning() << "method " << Q_FUNC_INFO << - " will be deleted in newxt version. please use signMessage method"; - return checkSignMessage(rawData, pubKey, rsa); -} - -unsigned int QRSAEncryption::getKeyBytesSize(QRSAEncryption::Rsa rsa) { - - return rsa / 4; -} - -// --- static methods --- -bool QRSAEncryption::generatePairKey(QByteArray &pubKey, QByteArray &privKey, - QRSAEncryption::Rsa rsa) { - - return QRSAEncryption(rsa).generatePairKey(pubKey, privKey); -} - -QByteArray QRSAEncryption::encode(const QByteArray &rawData, const QByteArray &pubKey, - Rsa rsa, BlockSize blockSizeMode) { - - return QRSAEncryption(rsa).encode(rawData, pubKey, blockSizeMode); -} - -QByteArray QRSAEncryption::decode(const QByteArray &rawData, const QByteArray &privKey, - Rsa rsa, BlockSize blockSizeMode) { - - return QRSAEncryption(rsa).decode(rawData, privKey, blockSizeMode); -} - -QByteArray QRSAEncryption::signMessage(QByteArray rawData, const QByteArray &privKey, Rsa rsa) { - - return QRSAEncryption(rsa).signMessage(rawData, privKey); -} - -bool QRSAEncryption::checkSignMessage(const QByteArray &rawData, const QByteArray &pubKey, Rsa rsa) { - - return QRSAEncryption(rsa).checkSignMessage(rawData, pubKey); -} - -// --- end of static methods --- - -bool QRSAEncryption::generatePairKey(QByteArray &pubKey, QByteArray &privKey) { - - int cnt{0}; - bool keyGenRes{false}; - INT p, q, modul, eilor, e, d; - do { - - pubKey.clear(); - privKey.clear(); - - p = randomPrimeNumber(); - q = randomPrimeNumber(p); - - modul = 0; - while ((modul = p * q) < 0) { - p = toPrime((p - 1) / 2); - } - - eilor = eulerFunc(p, q); - e = randomNumber() % eilor; - - if (!(e % 2)) --e; - - do { - e -= 2; - } while((!isMutuallyPrime(eilor, e))); - - d = extEuclid(eilor , e); - - while(d < 0 ) { - d += eilor; - } - - pubKey.append(toArray(e, _rsa / 8)); - pubKey.append(toArray(modul, _rsa / 8)); - privKey.append(toArray(d, _rsa / 8)); - privKey.append(toArray(modul, _rsa / 8)); - - } while (!(keyGenRes = testKeyPair(pubKey, privKey)) && (++cnt < KEY_GEN_LIMIT)); - - if(cnt >= KEY_GEN_LIMIT) qWarning() << QString("(Warning): Exceeded limit of key generation (%0)!").arg(KEY_GEN_LIMIT); - - return (keyGenRes && cnt < KEY_GEN_LIMIT); -} - -// --- non-static methods --- -QByteArray QRSAEncryption::encode(const QByteArray &rawData, const QByteArray &pubKey, BlockSize blockSizeMode) { - - if (getBitsSize(pubKey) != _rsa) { - return QByteArray(); - } - - int index = 0; - - QByteArray block; - - INT e = fromArray(pubKey.mid(0, pubKey.size() / 2)); - INT m = fromArray(pubKey.mid(pubKey.size() / 2)); - - short blockSizeOut = getBlockSize(m) + 1; // BlockSize::OneByte - short blockSizeIn = 1; // BlockSize::OneByte - - if (blockSizeMode == BlockSize::Auto) { - blockSizeIn = getBlockSize(m); - } - - if (!blockSizeIn) { - qDebug() << "module of key small! size = 1 byte, 2 byte is minimum"; - return QByteArray(); - } - - QByteArray res; - - while ((block = rawData.mid(index, blockSizeIn)).size()) { - - if (index + blockSizeIn > rawData.size() && block.size() && !block[0]) { - qWarning() << "When trying to encrypt data, problems arose, the last block contains non-significant zeros." - " These zeros will be deleted during the decryption process." - " For encode and decode data with non-significant zeros use BlockSize::OneByte"; - } - - res.append(encodeBlok(fromArray(block), e, m, blockSizeOut)); - index += blockSizeIn; - } - - return res; - -} -QByteArray QRSAEncryption::decode(const QByteArray &rawData, const QByteArray &privKey, BlockSize blockSizeMode) { - - if (getBitsSize(privKey) != _rsa) { - return QByteArray(); - } - - int index = 0; - - QByteArray block; - - INT d = fromArray(privKey.mid(0, privKey.size() / 2)); - INT m = fromArray(privKey.mid(privKey.size() / 2)); - - short blockSizeIn = getBlockSize(m) + 1; - - short blockSizeOut = 1; // BlockSize::OneByte - if (blockSizeMode == BlockSize::Auto) { - blockSizeOut = getBlockSize(m); - } - - QByteArray res; - while ((block = rawData.mid(index, blockSizeIn)).size()) { - bool isLastBlock = (index + blockSizeIn) >= rawData.size(); - - res.append(decodeBlok(fromArray(block), d, m, - (isLastBlock && blockSizeMode == BlockSize::Auto)? -1 : blockSizeOut)); - index += blockSizeIn; - } - return res; - -} -QByteArray QRSAEncryption::signMessage(QByteArray rawData, const QByteArray &privKey) { - - QByteArray hash = QCryptographicHash::hash(rawData, HashAlgorithm::Sha256); - - QByteArray signature = encode(hash, privKey, BlockSize::OneByte); - - rawData.append(SIGN_MARKER + signature.toHex() + SIGN_MARKER); - - return rawData; -} -bool QRSAEncryption::checkSignMessage(const QByteArray &rawData, const QByteArray &pubKey) { - - // start position of SIGN_MARKER in rawData - auto signStartPos = rawData.lastIndexOf(SIGN_MARKER, rawData.length() - signMarkerLength - 1); - - // length of signature in rawData - auto signLength = rawData.length() - signStartPos - signMarkerLength * 2; - - // message, that was recieved from channel - QByteArray message = rawData.left(signStartPos); - - // hash, that was decrypt from recieved signature - QByteArray recievedHash = decode(QByteArray::fromHex(rawData.mid(signStartPos + signMarkerLength, signLength)), - pubKey, BlockSize::OneByte); - - // if recievedHash == hashAlgorithm(recived message), then signed message is valid - return recievedHash == QCryptographicHash::hash(message, HashAlgorithm::Sha256); -} - -QRSAEncryption::Rsa QRSAEncryption::getRsa() const { - return _rsa; -} - -bool QRSAEncryption::testKeyPair(const QByteArray &pubKey, const QByteArray &privKey) { - - QByteArray tesVal = "Test message of encrypkey"; - - bool result = tesVal == decode(encode(tesVal, pubKey), privKey); - - if (!result) qWarning() << "(Warning): Testkey Fail, try generate new key pair!"; - - return result; -} - -// --- end of non-static methods --- - -bool QRSAEncryption::isValidRsaKey(const QByteArray &key) { - return key.size() && ((static_cast(key.size()) % getKeyBytesSize(RSA_64)) == 0); -} diff --git a/Qt-Secret/Qt-RSA/qrsaencryption.h b/Qt-Secret/Qt-RSA/qrsaencryption.h deleted file mode 100644 index 6fdffb3..0000000 --- a/Qt-Secret/Qt-RSA/qrsaencryption.h +++ /dev/null @@ -1,116 +0,0 @@ -//# -//# Copyright (C) 2018-2019 QuasarApp. -//# Distributed under the lgplv3 software license, see the accompanying -//# Everyone is permitted to copy and distribute verbatim copies -//# of this license document, but changing it is not allowed. -//# - -#ifndef QRSAENCRYPTION_H -#define QRSAENCRYPTION_H - -#include -#include -#include -#include -#include -#include // to use sha256 -#include "./../qtsecret_global.h" - -#include - - -class Qt_SECRETSHARED_EXPORT QRSAEncryption -{ -public: - typedef BigInt INT; - typedef QCryptographicHash::Algorithm HashAlgorithm; - - enum Rsa { - Invalid = 0, - RSA_64 = 64, - RSA_128 = 128, - RSA_256 = 256, - RSA_512 = 512, - RSA_1024 = 1024, - RSA_2048 = 2048, - RSA_4096 = 4096, - RSA_8192 = 8192, - - }; - - enum BlockSize { - Auto = 0, // fast but not stable. (using by default) - OneByte = 1 // stable but slow. (using for sig and check sig messages) - }; - - - QRSAEncryption(Rsa rsa = Rsa::RSA_256); - -// static methods - - // OLDMETHODS DELETE IN next Version - static bool generatePairKeyS(QByteArray &pubKey, QByteArray &privKey, - QRSAEncryption::Rsa rsa = RSA_256); - static QByteArray encodeS(const QByteArray &rawData, const QByteArray &pubKey, - Rsa rsa = RSA_256, BlockSize blockSizeMode = BlockSize::Auto); - static QByteArray decodeS(const QByteArray &rawData, const QByteArray &privKey, - Rsa rsa = RSA_256, BlockSize blockSizeMode = BlockSize::Auto); - static QByteArray signMessageS(QByteArray rawData, const QByteArray &privKey, - Rsa rsa = RSA_256); - static bool checkSignMessageS(const QByteArray &rawData, const QByteArray &pubKey, - Rsa rsa); - // OLDMETHODS END - - static bool generatePairKey(QByteArray &pubKey, QByteArray &privKey, - QRSAEncryption::Rsa rsa); - static QByteArray encode(const QByteArray &rawData, const QByteArray &pubKey, - Rsa rsa, BlockSize blockSizeMode = BlockSize::Auto); - static QByteArray decode(const QByteArray &rawData, const QByteArray &privKey, - Rsa rsa, BlockSize blockSizeMode = BlockSize::Auto); - static QByteArray signMessage(QByteArray rawData, const QByteArray &privKey, - Rsa rsa); - static bool checkSignMessage(const QByteArray &rawData, const QByteArray &pubKey, - Rsa rsa); - static bool isValidRsaKey(const QByteArray& key); - static unsigned int getKeyBytesSize(QRSAEncryption::Rsa rsa); - -// non-static methods - bool generatePairKey(QByteArray &pubKey, QByteArray &privKey); - - QByteArray encode(const QByteArray &rawData, const QByteArray &pubKey, - BlockSize blockSizeMode = BlockSize::Auto); - - QByteArray decode(const QByteArray &rawData, const QByteArray &privKey, - BlockSize blockSizeMode = BlockSize::Auto); - - QByteArray signMessage(QByteArray rawData, const QByteArray &privKey); - - bool checkSignMessage(const QByteArray &rawData, const QByteArray &pubKey); - - Rsa getRsa() const; - -private: - - Rsa _rsa; - - bool testKeyPair(const QByteArray &pubKey, const QByteArray &privKey); - bool isMutuallyPrime(const INT &a, const INT &b); - Rsa getBitsSize(const INT& i) const; - Rsa getBitsSize(const QByteArray& array) const; - - INT fromArray(const QByteArray& array) const; - QByteArray toArray(const INT &i, short sizeBlok = -1); - INT randomNumber(bool fullFilled = true) const; - INT toPrime(INT) const; - INT randomPrimeNumber(INT no = 0) const; - INT extEuclid(INT a, INT b) const; - - short getBlockSize(INT i) const; - - QByteArray encodeBlok(const INT& block, const INT& e, const INT& m, short blockSize); - QByteArray decodeBlok(const INT& block, const INT& d, const INT& m, short blockSize); - - -}; - -#endif // QRSAENCRYPTION_H diff --git a/Qt-Secret/Qt-Secret.pri b/Qt-Secret/Qt-Secret.pri deleted file mode 100644 index dfa6c3c..0000000 --- a/Qt-Secret/Qt-Secret.pri +++ /dev/null @@ -1,24 +0,0 @@ -# -# Copyright (C) 2018-2019 QuasarApp. -# Distributed under the lgplv3 software license, see the accompanying -# Everyone is permitted to copy and distribute verbatim copies -# of this license document, but changing it is not allowed. -# - -!isEmpty(Qt_SECRET_LIB):error("Qt-Secret.pri already included") -Qt_SECRET_LIB = 1 - -DEFINES += Qt_SECRET_LIBRARY -DEFINES += QT_DEPRECATED_WARNINGS - -HEADERS += \ - $$PWD/qtsecret_global.h \ - $$PWD/Qt-RSA/qrsaencryption.h - -SOURCES += \ - $$PWD/Qt-RSA/qrsaencryption.cpp - - -INCLUDEPATH += $$PWD $$PWD/Qt-RSA - -include($$PWD/GMP/GMP.pri) diff --git a/Qt-Secret/qtsecret_global.h b/Qt-Secret/qtsecret_global.h deleted file mode 100644 index 1893b41..0000000 --- a/Qt-Secret/qtsecret_global.h +++ /dev/null @@ -1,12 +0,0 @@ -#ifndef QTSECRET_GLOBAL_H -#define QTSECRET_GLOBAL_H - -#include - -#if defined(Qt_SECRET_LIBRARY) -# define Qt_SECRETSHARED_EXPORT Q_DECL_EXPORT -#else -# define Qt_SECRETSHARED_EXPORT Q_DECL_IMPORT -#endif - -#endif // QTSECRET_GLOBAL_H diff --git a/ScStwLibraries/ScStwLibraries.pri b/ScStwLibraries/ScStwLibraries.pri index 387dcd5..efeb4fd 100644 --- a/ScStwLibraries/ScStwLibraries.pri +++ b/ScStwLibraries/ScStwLibraries.pri @@ -17,31 +17,44 @@ ScStwLibraries_ClientLibs { DEFINES += ScStwLibraries_ClientLibs } +ScStwLibraries_Raspi { + LIBS += -lasound + DEFINES += ScStwLibraries_Raspi +} + INCLUDEPATH += $$PWD/headers $$PWD SOURCES += \ $$PWD/sources/ScStw.cpp \ + $$PWD/sources/scstwsettings.cpp \ $$PWD/sources/scstwsoundplayer.cpp \ $$PWD/sources/scstwlibraries.cpp \ $$PWD/sources/scstwrace.cpp \ - $$PWD/sources/scstwtimer.cpp + $$PWD/sources/scstwtimer.cpp \ + $$PWD/sources/scstwsetting.cpp HEADERS += \ $$PWD/headers/ScStw.hpp \ $$PWD/headers/ScStwLibraries_global.h \ $$PWD/headers/scstwlibraries.h \ $$PWD/headers/scstwrace.h \ + $$PWD/headers/scstwsettings.h \ $$PWD/headers/scstwsoundplayer.h \ - $$PWD/headers/scstwtimer.h + $$PWD/headers/scstwtimer.h \ + $$PWD/headers/scstwsetting.h ScStwLibraries_ClientLibs { SOURCES += \ $$PWD/sources/client/scstwclient.cpp \ - $$PWD/sources/client/scstwremotemonitorrace.cpp + $$PWD/sources/client/scstwremoterace.cpp \ + $$PWD/sources/client/scstwremotetimer.cpp \ + $$PWD/sources/client/scstwremotesettings.cpp HEADERS += \ $$PWD/headers/client/scstwclient.h \ - $$PWD/headers/client/scstwremotemonitorrace.h + $$PWD/headers/client/scstwremoterace.h \ + $$PWD/headers/client/scstwremotetimer.h \ + $$PWD/headers/client/scstwremotesettings.h INCLUDEPATH += $$PWD/headers/client } diff --git a/ScStwLibraries/headers/ScStw.hpp b/ScStwLibraries/headers/ScStw.hpp index 717aa14..a48fe03 100644 --- a/ScStwLibraries/headers/ScStw.hpp +++ b/ScStwLibraries/headers/ScStw.hpp @@ -61,28 +61,12 @@ public: RaceStateChanged = 9000, TimersChanged = 9001, ExtensionsChanged = 9002, - NextStartActionChanged = 9003 /*, ProfilesChanged*/ + CurrentStartDelayChanged = 9003, /*, ProfilesChanged*/ + SettingChanged = 9004, + RaceDetailsChanged = 9005 }; Q_ENUM(SignalKey) - /*! - * \brief The BaseStationSetting enum contains all settings of the base station that can be changed by a client - * - * \see ScStw::baseStationSettingFromInt() - * \see ScStw::baseStationSettingToString() - * \see ScStw::baseStationSettingFromString() - * \see ScStw::baseStationSettings - */ - enum BaseStationSetting { - InvalidSetting = -1, - ReadySoundEnableSetting, - ReadySoundDelaySetting, - AtYourMarksSoundEnableSetting, - AtYourMarksSoundDelaySetting, - SoundVolumeSetting - }; - Q_ENUM(BaseStationSetting) - /*! * \brief The SocketCommand enum contains all commands the base station can handle * @@ -93,15 +77,17 @@ public: InitializeSessionCommand = 1, - StartTimersCommand = 1000, - StopTimersCommand = 1001, - ResetTimersCommand = 1002, + StartRaceCommand = 1000, + StopRaceCommand = 1001, + ResetRaceCommand = 1002, + CancelRaceCommand = 1003, + SetTimerDisabledCommand = 1004, GetRaceStateCommand = 2000, - GetNextStartActionCommand = 2005, + GetRaceDetailsCommand = 2001, GetExtensionsCommand = 2006, GetTimersCommand = 2007, - GetNextStartActionDetailsCommand = 2009, + GetCurrentStartDelayCommand = 2009, WriteSettingCommand = 3000, ReadSettingCommand = 3001, @@ -126,15 +112,16 @@ public: FirmwareAlreadyUpToDateInfo = 304, - - CommandNotFoundError = 404, - ClientSessionAlreadyActiveError = 407, - UpdateSignatureInvalidError = 402, - TimestampTooSmallError = 406, - RequiredParameterNotGivenError = 405, - CurrentStateNotVaildForOperationError = 403, AccessDeniedError = 401, + UpdateSignatureInvalidError = 402, + CurrentStateNotVaildForOperationError = 403, + CommandNotFoundError = 404, + RequiredParameterNotGivenError = 405, + TimestampTooSmallError = 406, + ClientSessionAlreadyActiveError = 407, NoSessionActiveError = 408, + ItemNotFoundError = 409, + LastTimerCannotBeDisabledError = 410, UpdateFailedError = 500, @@ -143,10 +130,59 @@ public: TimeoutError = 911, SettingNotAccessibleError = 901, InternalError = 950, - InternalErrorTimerOperationFailed = 951 + InternalErrorTimerOperationFailed = 951, + ApiVersionNotSupportedError = 952, + CompetitionModeProhibitsThisError = 953, + FirmwareUpdateFormatInvalidError = 954, + TimersNotReadyError = 501 /*!< One or more timer is not ready */ }; Q_ENUM(ScStw::StatusCode) + /*! + * \brief The ExtensionType enum contains all types of extensions + */ + enum ExtensionType { + StartPad, + TopPad + }; + Q_ENUM(ExtensionType); + + + /*! + * \brief The ExtensionState enum contains all possible states of an extension + */ + enum ExtensionState { + ExtensionDisconnected = 0, + ExtensionConnecting = 1, + ExtensionInitialising = 2, + ExtensionConnected = 3 + }; + Q_ENUM(ExtensionState); + + + /*! + * \brief The ExtensionBatteryState enum contains all possible battery states of an extension + */ + enum ExtensionBatteryState { + BatteryUnknown = -1, + BatteryCritical = 0, + BatteryWarning = 1, + BatteryFine = 2, + BatteryCharging = 3, + BatteryNotCharging = 4 + }; + Q_ENUM(ExtensionBatteryState); + + /*! + * \brief The PadState enum contains whether a pad is currently pressed or not + */ + enum PadState { + PadNotPressed = 0, + PadPressed = 1 + }; + Q_ENUM(PadState); + + /*! * \brief SOCKET_MESSAGE_START_KEY contains the key, a message is supposed to start with */ @@ -157,44 +193,6 @@ public: */ static const char* SOCKET_MESSAGE_END_KEY; - /*! - * \brief baseStationSettings contains a string with reference to all BaseStationSetting values - * - * \see ScStw::BaseStationSetting - * \see ScStw::baseStationSettingToString() - * \see ScStw::baseStationSettingFromString() - */ - static const QMap baseStationSettings; - - /*! - * \brief Function to convert an int to a BaseStationSetting - * \param i the int to convert - * \return a BaseStationSetting - * - * \see ScStw::BaseStationSetting - */ - static BaseStationSetting baseStationSettingfromInt(int i); - - /*! - * \brief Function to convert a QString to a BaseStationSetting - * \param s the string to convert - * \return a BaseStationSetting - * - * \see ScStw::BaseStationSetting - * \see ScStw::baseStationSettingToString() - */ - static BaseStationSetting baseStationSettingFromString(QString s); - - /*! - * \brief Function to convert BaseStationSetting to a QString - * \param s the BaseStationSetting to convert - * \return a QString - * - * \see ScStw::BaseStationSetting - * \see ScStw::baseStationSettingFromString() - */ - Q_INVOKABLE static QString baseStationSettingToString(BaseStationSetting s); - /*! * \brief Function to convert an int to a SignalKey * \param i the int to convert @@ -213,6 +211,15 @@ public: */ static SocketCommand socketCommandFromInt(int i); + /*! + * \brief Function to convert an ExtensionType to a string + * \param t the ExtensionType to convert + * \return String + * + * \see ScStwExtensionType + */ + static QString extensionTypeToString(ExtensionType t); + /*! * \brief Function to compare to string firmware versions in .. formar * \param a version a diff --git a/ScStwLibraries/headers/client/scstwclient.h b/ScStwLibraries/headers/client/scstwclient.h index 60c3684..ed589ab 100644 --- a/ScStwLibraries/headers/client/scstwclient.h +++ b/ScStwLibraries/headers/client/scstwclient.h @@ -33,6 +33,7 @@ #include #include "ScStw.hpp" +#include "scstwsettings.h" /*! * This class is used to connect and talk to the ScStw basestation. @@ -50,7 +51,7 @@ class ScStwClient : public QObject { Q_OBJECT Q_PROPERTY(State state READ getState NOTIFY stateChanged) - Q_PROPERTY(QVariantList extensions READ getExtensions NOTIFY extensionsChanged) + Q_PROPERTY(QVariantMap extensions READ getExtensions NOTIFY extensionsChanged) Q_PROPERTY(QString ipAddress READ getIP WRITE setIP) public: @@ -59,7 +60,10 @@ public: * * \brief ScStwClient */ - explicit ScStwClient(QObject *parent = nullptr); + explicit ScStwClient( + QObject *parent = nullptr, + QList signalSubscriptions = {ScStw::ExtensionsChanged} + ); enum State {DISCONNECTED, CONNECTING, INITIALISING, CONNECTED}; Q_ENUM(State); @@ -73,7 +77,9 @@ private: int errors; const static int ERRORS_UNTIL_DISCONNECT = 4; - QVariantList extensions; + QVariantMap extensions; + + QList signalSubscriptions; //---general status values---// @@ -95,15 +101,14 @@ private: QString readBuffer; - int nextConnectionId; + unsigned int currentRequestId; - struct waitingRequest { - int id; + struct WaitingRequest { QEventLoop * loop; QJsonObject reply; }; - QList waitingRequests; + QMap waitingRequests; public slots: @@ -139,14 +144,14 @@ public slots: * \param value the value to write to * \return the status code returned by the command */ - ScStw::StatusCode writeRemoteSetting(ScStw::BaseStationSetting key, QString value); + ScStw::StatusCode writeRemoteSetting(ScStwSettings::BaseStationSetting key, QVariant value); /*! * \brief Function to read a setting on the base station * \param key the key to read from * \return the value of the key or "false" if the key is not found or an error occured */ - QString readRemoteSetting(ScStw::BaseStationSetting key); + QVariant readRemoteSetting(ScStwSettings::BaseStationSetting key, ScStw::StatusCode* status = nullptr); /*! Getter fuctions */ @@ -167,7 +172,7 @@ public slots: * \brief Function to get the extensions and their state from the base station * \return a list with all configured extensions and their state */ - QVariantList getExtensions(); + QVariantMap getExtensions(); /*! * \brief Function to get the time offset of the base station relative to the clients time @@ -199,6 +204,8 @@ public slots: */ void setIP(QString ipAdress); + void addSignalSubscription(ScStw::SignalKey key); + private slots: /*! * \brief called when timeout timer times out @@ -279,7 +286,7 @@ private slots: * \see gotSignal() * \param connections the list to set the chache to */ - void setExtensions(QVariantList extensions); + void setExtensions(QVariantMap extensions); /*! * \brief Function to set the local state. diff --git a/ScStwLibraries/headers/client/scstwremoterace.h b/ScStwLibraries/headers/client/scstwremoterace.h new file mode 100644 index 0000000..8c4a372 --- /dev/null +++ b/ScStwLibraries/headers/client/scstwremoterace.h @@ -0,0 +1,86 @@ +/**************************************************************************** + ** ScStw Libraries + ** Copyright (C) 2020 Itsblue development + ** + ** This program is free software: you can redistribute it and/or modify + ** it under the terms of the GNU General Public License as published by + ** the Free Software Foundation, either version 3 of the License, or + ** (at your option) any later version. + ** + ** This program is distributed in the hope that it will be useful, + ** but WITHOUT ANY WARRANTY; without even the implied warranty of + ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + ** GNU General Public License for more details. + ** + ** You should have received a copy of the GNU General Public License + ** along with this program. If not, see . + ****************************************************************************/ + +#ifndef SCSTWREMOTEMONITORRACE_H +#define SCSTWREMOTEMONITORRACE_H + +#include +#include "scstwrace.h" +#include "scstwclient.h" +#include "scstwremotetimer.h" +#include "scstwsoundplayer.h" + +class ScStwRemoteRace : public ScStwRace +{ + Q_OBJECT + Q_PROPERTY(ScStwClient* scStwClient READ getScStwClient WRITE setScStwClient NOTIFY scStwClientChanged) + +public: + ScStwRemoteRace(QObject *parent = nullptr); + ScStwRemoteRace(ScStwClient *scStwClient, ScStwSettings *settings = nullptr, QObject *parent = nullptr); + + enum RaceMode { + LOCAL, + REMOTE + }; + +protected: + double currentStartTotalDelay; + double currentStartDelayStartedAt; + double latestStartDelayProgress; + bool isReadyForNextState; + bool readySoundEnabled; + +private: + ScStwClient *scStwClient; + QList remoteTimers; + QList localTimers; + +public slots: + ScStw::StatusCode start(bool asyncronous = true); + ScStw::StatusCode cancel(); + ScStw::StatusCode stop(); + ScStw::StatusCode reset(); + ScStw::StatusCode setTimerDisabled(int id, bool disabled); + bool addTimer(ScStwTimer *timer); + QVariantMap getCurrentStartDelay(); + bool getIsReadyForNextState(); + bool getReadySoundEnabled(); + + ScStwClient *getScStwClient(); + void setScStwClient(ScStwClient *client); + +private slots: + void handleBaseStationSignal(ScStw::SignalKey key, QVariant data); + + bool refreshRemoteTimers(QVariantList remoteTimers); + void rebuildRemoteTimers(QVariantList remoteTimers); + void refreshDetails(QVariantMap details); + void handleClientStateChange(); + + void refreshCompetitionMode(); + RaceMode getMode(); + bool local(); + + void setTimers(QList timers, bool deleteOldTimers); + +signals: + void scStwClientChanged(); +}; + +#endif // SCSTWREMOTEMONITORRACE_H diff --git a/ScStwLibraries/headers/client/scstwremotesettings.h b/ScStwLibraries/headers/client/scstwremotesettings.h new file mode 100644 index 0000000..e1515b5 --- /dev/null +++ b/ScStwLibraries/headers/client/scstwremotesettings.h @@ -0,0 +1,62 @@ +/**************************************************************************** + ** ScStw Libraries + ** Copyright (C) 2020 Itsblue development + ** + ** This program is free software: you can redistribute it and/or modify + ** it under the terms of the GNU General Public License as published by + ** the Free Software Foundation, either version 3 of the License, or + ** (at your option) any later version. + ** + ** This program is distributed in the hope that it will be useful, + ** but WITHOUT ANY WARRANTY; without even the implied warranty of + ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + ** GNU General Public License for more details. + ** + ** You should have received a copy of the GNU General Public License + ** along with this program. If not, see . + ****************************************************************************/ + +#ifndef SCSTWREMOTESETTINGS_H +#define SCSTWREMOTESETTINGS_H + +#include +#include "scstwsettings.h" +#include "scstwclient.h" +#include + +class ScStwRemoteSettings : public ScStwSettings +{ + Q_OBJECT + Q_PROPERTY(ScStwClient* scStwClient READ getScStwClient WRITE setScStwClient NOTIFY scStwClientChanged) + +public: + ScStwRemoteSettings(QObject * parent = nullptr); + ScStwRemoteSettings(ScStwClient * scStwClient, QObject * parent = nullptr); + + enum SettingsMode { + LOCAL, + REMOTE + }; + +protected: + QVariant readSetting(QString key, int keyInt, int keyLevel); + bool writeSetting(QString key, QVariant value, int keyInt,int keyLevel = -1); + bool setDefaultSetting(QString key, QVariant defaultVariant, int keyInt,int keyLevel = -1); + +private: + ScStwClient * scStwClient; + +public slots: + ScStwClient *getScStwClient(); + void setScStwClient(ScStwClient *client); + +private slots: + void handleClientStateChange(); + void handleBaseStationSignal(ScStw::SignalKey key, QVariant data); + SettingsMode getMode(); + +signals: + void scStwClientChanged(); +}; + +#endif // SCSTWREMOTESETTINGS_H diff --git a/ScStwLibraries/headers/client/scstwremotetimer.h b/ScStwLibraries/headers/client/scstwremotetimer.h new file mode 100644 index 0000000..518140b --- /dev/null +++ b/ScStwLibraries/headers/client/scstwremotetimer.h @@ -0,0 +1,79 @@ +#ifndef SCSTWREMOTETIMER_H +#define SCSTWREMOTETIMER_H + +#include +#include + +class ScStwRemoteTimer : public ScStwTimer +{ + Q_OBJECT +public: + ScStwRemoteTimer(QObject *parent = nullptr); + + friend class ScStwRemoteRace; + +private: + ScStwTimer::ReadyState readyState; + +public slots: + /*! + * \brief Function to get the current ready status of a timer + * \return The current ready status + */ + virtual ScStwTimer::ReadyState getReadyState(); + +protected slots: + /*! + * \brief Function to dircetly change the start time + * + * Only works when directControlEnabled is set to true! + * + * \param startTime the time to change to + * \return false when directControlEnabled is set to false and the startTime was therefore not modified, true otherwise + */ + void setStartTime(double startTime); + + /*! + * \brief Function to dircetly change the stop time + * + * Only works when directControlEnabled is set to true! + * + * \param stopTime the time to change to + * \return false when directControlEnabled is set to false and the stopTime was therefore not modified, true otherwise + */ + void setStopTime(double stopTime); + + /*! + * \brief Function to dircetly change the rection time + * + * Only works when directControlEnabled is set to true! + * + * \param reactionTime the time to change to + * \return false when directControlEnabled is set to false and the reactionTime was therefore not modified, true otherwise + */ + void setReactionTime(double rectionTime); + + /*! + * \brief Function to dircetly change the letter + * + * Only works when directControlEnabled is set to true! + * + * \param newLetter the letter to change to + * \return false when directControlEnabled is set to false and the letter was therefore not modified, true otherwise + */ + void setLetter(QString newLetter); + + void setReadyState(ScStwTimer::ReadyState); + + /*! + * \brief Function to change the state of the timer + * + * Doing this will emit the ScStwTimer::stateChanged() signal (only if the new state differs from the current one) + * + * \param newState The new state + */ + void setState(TimerState newState); + +}; + +#endif // SCSTWREMOTETIMER_H diff --git a/ScStwLibraries/headers/scstwlibraries.h b/ScStwLibraries/headers/scstwlibraries.h index 47d5f84..5435edf 100644 --- a/ScStwLibraries/headers/scstwlibraries.h +++ b/ScStwLibraries/headers/scstwlibraries.h @@ -31,9 +31,12 @@ #endif #include "scstwtimer.h" #include "scstwrace.h" +#include "scstwsettings.h" #ifdef ScStwLibraries_ClientLibs -#include "scstwremotemonitorrace.h" +#include "scstwsetting.h" +#include "scstwremoterace.h" #include "scstwclient.h" +#include "scstwremotesettings.h" #endif class ScStwLibraries : public QObject diff --git a/ScStwLibraries/headers/scstwrace.h b/ScStwLibraries/headers/scstwrace.h index 206a457..70cc2b6 100644 --- a/ScStwLibraries/headers/scstwrace.h +++ b/ScStwLibraries/headers/scstwrace.h @@ -25,6 +25,9 @@ #include #include "scstwtimer.h" #include "scstwsoundplayer.h" +#include "scstwsettings.h" + +class ScStwRemoteRace; /*! * \brief The ScStwRace class can be used to measure timings of climbing races with multiple lanes at once. @@ -50,51 +53,49 @@ class ScStwRace : public QObject { Q_OBJECT Q_PROPERTY(RaceState state READ getState NOTIFY stateChanged) - Q_PROPERTY(QVariantList timers READ getTimerDetailList NOTIFY timersChanged) - Q_PROPERTY(QVariantList nextStartActionDetails READ getNextStartActionDetails NOTIFY nextStartActionDetailsChanged) + Q_PROPERTY(QVariantList timers READ getTimerDetailList NOTIFY timersChanged) + Q_PROPERTY(QVariantMap currentStartDelay READ getCurrentStartDelay NOTIFY currentStartDelayChanged) + Q_PROPERTY(bool isReadyForNextState READ getIsReadyForNextState NOTIFY isReadyForNextStateChanged) + Q_PROPERTY(bool competitionMode READ getCompetitionMode NOTIFY competitionModeChanged) + Q_PROPERTY(bool readySoundEnabled READ getReadySoundEnabled NOTIFY readySoundEnabledChanged) + Q_PROPERTY(QVariantMap details READ getDetails NOTIFY detailsChanged) + Q_PROPERTY(ScStwSettings* settings READ getSettings WRITE setSettings NOTIFY settingsChanged) + Q_PROPERTY(bool autoRefreshTimerText READ getAutoRefreshTimerText WRITE setAutoRefreshTimerText NOTIFY autoRefreshTimerTextChanged) public: explicit ScStwRace(QObject *parent = nullptr); + explicit ScStwRace(ScStwSettings *settings, QObject *parent = nullptr); - enum RaceState { IDLE, STARTING, WAITING, RUNNING, STOPPED }; + friend class ScStwRemoteRace; + + enum RaceState { IDLE, PREPAIRING, WAITING, STARTING, RUNNING, STOPPED, INCIDENT }; Q_ENUM(RaceState) - enum StartAction { None = -1, AtYourMarks = 0, Ready = 1, Start = 2 }; - Q_ENUM(StartAction) - - enum NextStartActionDetailAttributes { - NextStartAction = 0, - NextStartActionTotalDelay = 1, - NextStartActionDelayProgress = 2 - }; - Q_ENUM(NextStartActionDetailAttributes); - protected: - StartAction nextStartAction; QList timers; void setState(RaceState newState); private: RaceState state; - QList timerEnableQueque; - - QTimer *nextActionTimer; - QEventLoop *nextActionLoop; + QTimer *startDelayTimer; + QTimer *timerTextRefreshTimer; + QEventLoop *startWaitLoop; // sounds ScStwSoundPlayer * soundPlayer; + // settings + ScStwSettings *settings; + bool competitionMode; + bool autoRefreshTimerText; - // some settings - double soundVolume; - - /*! - * \brief stores the start action settings - * - * \details Stores the settings for the action ScStwRace::AtYourMarks and ScStwRace::Ready. The settings keys are: "Enabled" and "Delay" - */ - QMap startActionSettings; + enum LoopExitTypes { + LoopAutomaticExit = 0, + LoopReadyStateChangeExit = 1, + LoopManualExit = 2, + LoopCancelExit = 3 + }; public slots: @@ -105,57 +106,84 @@ public slots: * or if if should wait until the start sequence is over and quit after that (false) * \return 200: OK; 904: state not matching */ - int start(bool asyncronous = true); + virtual ScStw::StatusCode start(bool asyncronous = true); /*! * \brief Function to stop the currently running race * * \return 200: OK; 904: state not matching */ - virtual int stop(); + virtual ScStw::StatusCode stop(); /*! * \brief Function to reset a stopped race * \return */ - virtual int reset(); - virtual int cancel(); + virtual ScStw::StatusCode reset(); + virtual ScStw::StatusCode cancel(); - // setters - bool writeStartActionSetting(StartAction action, bool enabled, int delay); - bool setSoundVolume(double volume); - virtual bool addTimer(ScStwTimer *timer); + virtual ScStw::StatusCode setTimerDisabled(int id, bool disabled); + + Q_INVOKABLE virtual bool addTimer(ScStwTimer *timer); // getters RaceState getState(); - StartAction getNextStartAction(); - virtual QVariantList getNextStartActionDetails(); + virtual QVariantMap getCurrentStartDelay(); + QList getTimers(); QVariantList getTimerDetailList(); + QVariantMap getDetails(); + bool getCompetitionMode(); + virtual bool getReadySoundEnabled(); + + ScStwSettings* getSettings(); + void setSettings(ScStwSettings* settings); + + bool getAutoRefreshTimerText(); + void setAutoRefreshTimerText(bool autoRefresh); protected slots: - private slots: - void refreshTimerStates(); - void handleTimerEnable(ScStwTimer* timer); - int handleFalseStart(); - bool playSoundsAndStartTimers(StartAction thisAction); - + void handleTimerStateChange(ScStwTimer::TimerState newState); /** * \brief Function to declare the winner and looser timers after a timer has been stopped */ void handleTimerStop(); + void handleFalseStart(); + + + void handleTimerWantsToBeDisabledChange(ScStwTimer* timer, bool wantsToBeDisabled); + bool playSoundsAndStartTimers(); + ScStwSoundPlayer::PlayResult doDelayAndSoundOfCurrentStartState(double *timeOfSoundPlaybackStart = nullptr); + void technicalIncident(); + ScStw::StatusCode setTimerDisabled(ScStwTimer* timer, bool disabled); + + virtual void refreshCompetitionMode(); + + double getSoundVolume(); + ScStwSoundPlayer::StartSound getSoundForState(ScStwRace::RaceState state); + bool getSoundEnabledSetting(ScStwSoundPlayer::StartSound sound); + int getSoundDelaySetting(ScStwSoundPlayer::StartSound sound); + + + bool isStarting(); + virtual bool getIsReadyForNextState(); + void handleTimerReadyStateChange(ScStwTimer::ReadyState readyState); signals: void startTimers(); void stopTimers(int type); void resetTimers(); void stateChanged(RaceState state); - void nextStartActionChanged(); - void nextStartActionDetailsChanged(); + void currentStartDelayChanged(); void timersChanged(); - + void isReadyForNextStateChanged(); + void detailsChanged(); + void competitionModeChanged(); + void readySoundEnabledChanged(); + void settingsChanged(); + void autoRefreshTimerTextChanged(); }; diff --git a/ScStwLibraries/headers/client/scstwremotemonitorrace.h b/ScStwLibraries/headers/scstwsetting.h similarity index 56% rename from ScStwLibraries/headers/client/scstwremotemonitorrace.h rename to ScStwLibraries/headers/scstwsetting.h index c36db54..6a7b36c 100644 --- a/ScStwLibraries/headers/client/scstwremotemonitorrace.h +++ b/ScStwLibraries/headers/scstwsetting.h @@ -16,39 +16,43 @@ ** along with this program. If not, see . ****************************************************************************/ -#ifndef SCSTWREMOTEMONITORRACE_H -#define SCSTWREMOTEMONITORRACE_H +#ifndef SCSTWQMLSETTING_H +#define SCSTWQMLSETTING_H #include -#include "scstwrace.h" -#include "scstwclient.h" +#include -class ScStwRemoteMonitorRace : public ScStwRace +class ScStwSettings; + +class ScStwSetting : public QObject { Q_OBJECT -public: - ScStwRemoteMonitorRace(ScStwClient *monitorClient, QObject *parent = nullptr); + Q_PROPERTY(QVariant value READ getValue WRITE setValue NOTIFY valueChanged) + Q_PROPERTY(QVariant readonlyValue READ getValue NOTIFY valueChanged) protected: - double nextStartActionTotalDelay; - double nextStartActionDelayStartedAt; + explicit ScStwSetting(int key, int keyLevel, ScStwSettings*scStwSettings, QObject *parent); + + friend class ScStwSettings; + + int key; + int keyLevel; + bool hasToReload; private: - ScStwClient *scStwClient; + QVariant valueCache; + ScStwSettings *scStwSettings; public slots: - int start(); - int cancel(); - int stop(); - int reset(); - bool addTimer(ScStwTimer *timer); - QVariantList getNextStartActionDetails(); + QVariant getValue(); + void setValue(QVariant value); -private slots: - void handleClientStateChanged(); - void handleBaseStationSignal(ScStw::SignalKey key, QVariant data); - bool refreshRemoteTimers(QVariantList remoteTimers); +protected slots: + void handleSettingChange(int key, int keyLevel, QVariant value); + +signals: + void valueChanged(); }; -#endif // SCSTWREMOTEMONITORRACE_H +#endif // SCSTWQMLSETTING_H diff --git a/ScStwLibraries/headers/scstwsettings.h b/ScStwLibraries/headers/scstwsettings.h new file mode 100644 index 0000000..282a603 --- /dev/null +++ b/ScStwLibraries/headers/scstwsettings.h @@ -0,0 +1,124 @@ +/**************************************************************************** + ** ScStw Libraries + ** Copyright (C) 2020 Itsblue development + ** + ** This program is free software: you can redistribute it and/or modify + ** it under the terms of the GNU General Public License as published by + ** the Free Software Foundation, either version 3 of the License, or + ** (at your option) any later version. + ** + ** This program is distributed in the hope that it will be useful, + ** but WITHOUT ANY WARRANTY; without even the implied warranty of + ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + ** GNU General Public License for more details. + ** + ** You should have received a copy of the GNU General Public License + ** along with this program. If not, see . + ****************************************************************************/ + +#ifndef SCSTWSETTINGS_H +#define SCSTWSETTINGS_H + +#include +#include +#include +#include +#include +#include +#include +#include +#include +#include + +class ScStwSettings : public QObject +{ + Q_OBJECT +public: + explicit ScStwSettings(QObject *parent = nullptr); + + typedef QString(*keyToStringConverter)(int); + typedef QVariant::Type(*keyToTypeConverter)(int); + + /*! + * \brief The BaseStationSetting enum contains all settings of the base station that can be changed by a client + * + * \see ScStw::baseStationSettingFromInt() + * \see ScStw::baseStationSettingToString() + * \see ScStw::baseStationSettingFromString() + * \see ScStw::baseStationSettings + */ + enum BaseStationSetting { + InvalidSetting = -1, + ReadySoundEnableSetting, + ReadySoundDelaySetting, + AtYourMarksSoundEnableSetting, + AtYourMarksSoundDelaySetting, + SoundVolumeSetting, + CompetitionModeSetting + }; + Q_ENUM(BaseStationSetting) + + enum KeyLevelEnum { + KeyLevel = 0 + }; + Q_ENUM(KeyLevelEnum) + + virtual QVariant readSetting(BaseStationSetting key); + Q_INVOKABLE virtual QVariant readSetting(int key, int keyLevel); + virtual bool writeSetting(BaseStationSetting key, QVariant value); + Q_INVOKABLE virtual bool writeSetting(int key, int keyLevel, QVariant value); + virtual bool setDefaultSetting(BaseStationSetting key, QVariant defaultValue); + Q_INVOKABLE virtual bool setDefaultSetting(int key, int keyLevel, QVariant defaultValue); + + Q_INVOKABLE ScStwSetting * getSetting(int key, int keyLevel); + + static BaseStationSetting keyFromInt(int i) { + QMetaEnum enumeration = QMetaEnum::fromType(); + return static_cast(enumeration.keyToValue(enumeration.valueToKey(i))); + } + + static QString keyToString(int key) { + return QMetaEnum::fromType().valueToKey(key); + } + + static QVariant::Type keyToType(int key) { + QMap types = { + {ReadySoundEnableSetting, QVariant::Bool}, + {ReadySoundDelaySetting, QVariant::Double}, + {AtYourMarksSoundEnableSetting, QVariant::Bool}, + {AtYourMarksSoundDelaySetting, QVariant::Double}, + {SoundVolumeSetting, QVariant::Double}, + {CompetitionModeSetting, QVariant::Bool} + }; + + if(types.contains(BaseStationSetting(key))) + return types[BaseStationSetting(key)]; + + return QVariant::Invalid; + } + +protected: + virtual QVariant readSetting(QString key, int keyInt = -1, int keyLevel = -1); + virtual bool writeSetting(QString key, QVariant value, int keyInt = -1,int keyLevel = -1); + virtual bool setDefaultSetting(QString key, QVariant defaultValue, int keyInt,int keyLevel = -1); + bool registerKeyLevelConverters(int keyLevel, keyToStringConverter, keyToTypeConverter); + +private: + QFile * settingsFile; + QVariantMap settingsCache; + + bool loadSettingsFromFile(); + + QMap keyToStringConverters; + QMap keyToTypeConverters; + QMap> internalSettingHandlers; + +private slots: + bool writeSettingsToFile(); + +signals: + void settingChanged(int key, int keyLevel, QVariant value); + +}; + +#endif // SCSTWSETTINGS_H diff --git a/ScStwLibraries/headers/scstwsoundplayer.h b/ScStwLibraries/headers/scstwsoundplayer.h index d158725..47a678b 100644 --- a/ScStwLibraries/headers/scstwsoundplayer.h +++ b/ScStwLibraries/headers/scstwsoundplayer.h @@ -27,6 +27,11 @@ #include #include #include +#include + +#ifdef ScStwLibraries_Raspi +#include +#endif /*! * \brief The ScStwSoundPlayer class is used for ultra low latency sound playback of the speed clibing start tones and commands @@ -41,7 +46,26 @@ public: */ explicit ScStwSoundPlayer(QObject *parent = nullptr); + enum StartSound { + None = -1, + AtYourMarks = 0, + Ready = 1, + Start = 2, + FalseStart = 3 + }; + + enum PlayResult { + Success = 0, + Cancelled = -1, + Error = -2 + }; + private: + + bool _setSoundVolume(double volume); + + void _initializeSondEffect(); + /*! * \brief A map containing all sound files * 0: AtYourMarksSound @@ -49,13 +73,15 @@ private: * 2: StartSound * 3: FalseStartSound */ - QMap soundFiles; + QMap soundFiles; /*! * \brief The sound effect object */ QSoundEffect *soundEffect; + QAudioDeviceInfo *_audioOutputDevice; + /*! * \brief The QEventLoop used to wait for the sound to finish */ @@ -69,7 +95,7 @@ private: /*! * \brief The action that is currently played */ - int currentlyPlayingAction; + StartSound currentlyPlayingSound; /*! * \brief Holds the time the playback started at @@ -83,16 +109,16 @@ public slots: * \param action The action to play (0: AtYourMarks, 1:Ready, 2:Start) * \param volume The volume to play at * \param timeOfStop The time the playback actually started (msecs since epoch) - * \return true if the playback was successfully started, false otherwise + * \return TODO true if the playback was successfully started, false otherwise */ - bool play(int action, double volume, double *timeOfStart = nullptr); + ScStwSoundPlayer::PlayResult play(StartSound sound, double volume, double *timeOfStart = nullptr); /*! * \brief Function to wait for the playback to finish * \param timeOfStop the point in time when the plyback actually stopped (msecs since epoch) * \return false if there was any error (eg. there was no playback currently), true otherwise */ - bool waitForSoundFinish(double *timeOfStop = nullptr); + ScStwSoundPlayer::PlayResult waitForSoundFinish(double *timeOfStop = nullptr); /*! * \brief Function to cancel the current playback @@ -102,7 +128,9 @@ public slots: * \param volume the volume to play the false start sound at * \return true if the playback was successfully stopped, false otherwise */ - bool cancel(double volume = 0); + bool cancel(); + + bool isPlaying(); private slots: diff --git a/ScStwLibraries/headers/scstwtimer.h b/ScStwLibraries/headers/scstwtimer.h index e1aac72..ccd67b4 100644 --- a/ScStwLibraries/headers/scstwtimer.h +++ b/ScStwLibraries/headers/scstwtimer.h @@ -25,6 +25,9 @@ #include #include "ScStw.hpp" +class ScStwRace; +class ScStwRemoteRace; + /*! * \brief The ScStwTimer class is used for advanced time measurement. * @@ -62,13 +65,18 @@ class ScStwTimer : public QObject { Q_OBJECT public: + + explicit ScStwTimer(QObject *parent = nullptr); + /*! * \brief ScStwTimer constructor * \param parent the parent object * \param directControlEnabled Defines if protected properties (startTimer, stopTime, reactionTime and state) can be directly set from outside. * \param letter the letter of the timer (only first char will be used!) */ - explicit ScStwTimer(QObject *parent = nullptr, bool directControlEnabled = false, QString letter = "" ); + explicit ScStwTimer(QString letter, QObject *parent = nullptr); + + friend class ScStwRace; /*! * \brief The TimerState enum contains all state the timer can be in @@ -77,24 +85,30 @@ public: IDLE, /*!< Timer is waiting to be started */ STARTING, /*!< Timer is starting and will react with a false start if the climber starts */ RUNNING, /*!< Timer is running */ - WAITING, /*!< Timer was stopped and is waiting for the result */ + WAITING, /*!< Timer was stopped and is waiting to be set to either WON or LOST */ WON, /*!< Timer has won */ LOST, /*!< Timer has lost */ + FAILING, /*!< Timer encountered a false start and is waiting to be set to either FAILED or WILDCARD */ + WILDCARD, /*!< The opponent has done a false start */ FAILED, /*!< A false start occured */ CANCELLED, /*!< Timer was cancelled */ + INCIDENT, /*!< There was a technical incident (most likely a disconnected extension) */ DISABLED /*!< Timer is disabled */ }; Q_ENUM(TimerState); /*! - * \brief The StopReason enum contains all possible reasons for a stop + * \brief The ReadyStatus enum contains all possible reasons for a timer not to be ready (>0) and the case that it is ready (0) */ - enum StopReason { - ManualStop, /*!< Timer was stopped manually */ - CancelStop, /*!< Timer was cancelled */ - FailStop, /*!< A false start occured */ - TopPadStop /*!< Timer was stopped by a physical trigger (eg. a ScStwExtension) */ + enum ReadyState { + IsReady = 0, /*!< Timer is ready for start */ + NotInIdleState, /*!< Timer is not in IDLE state */ + IsDisabled, /*!< Timer is disabled */ + ExtensionIsNotConnected, /*!< Not all extension of the timer are conneted */ + ExtensionBatteryIsCritical, /*!< The battery level of one or more extension is critical or unknown */ + ClimberIsNotReady /*!< The startpad of the timer is not triggered */ }; + Q_ENUM(ReadyState) protected: /*! @@ -117,16 +131,16 @@ protected: */ double reactionTime; - /*! - * \brief Defines if protected properties (startTimer, stopTime, reactionTime and state) can be directly set from outside. - */ - bool directControlEnabled; - /*! * \brief The letter (eg. "A" or "B") of the Timer (only one char) */ QString letter; + /*! + * \brief Defines if the timer currently wants to be disabled or not + */ + bool wantsToBeDisabled; + public slots: /*! @@ -243,56 +257,20 @@ public slots: void setDisabled(bool disabled); /*! - * \brief Function to change the state of the timer - * - * Only works when directControlEnabled is set to true! - * Doing this will emit the ScStwTimer::stateChanged() signal (only if the new state differs from the current one) - * - * \param newState The new state - * \param force whether to force the state change (just to distinguish between protected and public function) - * \return false when directControlEnabled or force is set to false and the startTime was therefore not modified, true otherwise + * \brief Function to check if the timer currently wants to be disabled + * \return true or false */ - bool setState(TimerState newState, bool force); + bool getWantsToBeDisabled(); /*! - * \brief Function to dircetly change the start time - * - * Only works when directControlEnabled is set to true! - * - * \param startTime the time to change to - * \return false when directControlEnabled is set to false and the startTime was therefore not modified, true otherwise + * \brief Function to get the current ready status of a timer + * \return The current ready status */ - bool setStartTime(double startTime); + virtual ScStwTimer::ReadyState getReadyState(); - /*! - * \brief Function to dircetly change the stop time - * - * Only works when directControlEnabled is set to true! - * - * \param stopTime the time to change to - * \return false when directControlEnabled is set to false and the stopTime was therefore not modified, true otherwise - */ - bool setStopTime(double stopTime); + bool isRunning(); - /*! - * \brief Function to dircetly change the rection time - * - * Only works when directControlEnabled is set to true! - * - * \param reactionTime the time to change to - * \return false when directControlEnabled is set to false and the reactionTime was therefore not modified, true otherwise - */ - bool setReactionTime(double rectionTime); - - /*! - * \brief Function to dircetly change the letter - * - * Only works when directControlEnabled is set to true! - * - * \param newLetter the letter to change to - * \return false when directControlEnabled is set to false and the letter was therefore not modified, true otherwise - */ - bool setLetter(QString newLetter); + bool isDisabled(); protected slots: @@ -302,33 +280,6 @@ protected slots: */ void handleClimberStart(double timeOfStart); - /*! - * \brief Function to stop the timer at a given point in time (past or future) - * - * To do this, the timer has to be in ScStwTimer::RUNNING state! - * - * \param reason reason for the timer stop - * \param timeOfStop the point in time (msecs since epoch) when the timer is supposted to be stopped - * - * \return false if the timer was not in the required state and therefore not stopped, true otherwise - * - * \see ScStwTimer::StopReason - */ - virtual bool stop(StopReason reason, double timeOfStop); - - /*! - * \brief Function to stop the timer - * - * To do this, the timer has to be in ScStwTimer::RUNNING state! - * - * \param reason reason for the timer stop - * - * \return false if the timer was not in the required state and therefore not stopped, true otherwise - * - * \see ScStwTimer::StopReason - */ - bool stop(StopReason reason); - /*! * \brief Function to change the state of the timer * @@ -338,12 +289,36 @@ protected slots: */ void setState(TimerState newState); + /*! + * \brief Function to set whether the timer currently wants to be disabled + * \param wantsToBeDisabled true or false + */ + void setWantsToBeDisabled(bool wantsToBeDisabled); + + /*! + * \brief Function to set the timer into INCIDENT state immidieately + * + * The current state of the timer will be ignored! It can only get out of this state by calling ScStwTimer::reset + * + * \see reset + */ + void technicalIncident(); + + /*! + * \brief Function to set the timer into WILDCARD state + * + * Only works when the timer is in STARTING state. + * + * \return false if not in STARTING state + */ + bool wildcard(); + signals: /*! * \brief Emitted when the state of the timer changed */ - void stateChanged(); + void stateChanged(TimerState state); /*! * \brief Emitted when the reaction time changed @@ -354,7 +329,13 @@ signals: * \brief Emitted when the timer wants its state to be changed by the external handler * \param timer the timer object */ - void requestEnableChange(ScStwTimer* timer); + void wantsToBeDisabledChanged(ScStwTimer* timer, bool wantsToBeDisabled); + + /*! + * \brief Emitted when the ready state of the timer changes + * \param readyState the new ReadyState + */ + void readyStateChanged(ReadyState readyState); }; diff --git a/ScStwLibraries/headers/styling/scstwappthememanager.h b/ScStwLibraries/headers/styling/scstwappthememanager.h index bec5135..04b4ffb 100644 --- a/ScStwLibraries/headers/styling/scstwappthememanager.h +++ b/ScStwLibraries/headers/styling/scstwappthememanager.h @@ -27,6 +27,7 @@ class ScStwAppThemeManager : public QObject { Q_OBJECT Q_PROPERTY(ScStwAppTheme* theme READ getTheme NOTIFY themeChanged) + Q_PROPERTY(QString themeName READ getThemeName WRITE setTheme NOTIFY themeChanged) public: explicit ScStwAppThemeManager(QObject *parent = nullptr); @@ -45,6 +46,7 @@ signals: public slots: ScStwAppTheme* getTheme(); Q_INVOKABLE bool setTheme(QString themeName); + Q_INVOKABLE QString getThemeName(); }; diff --git a/ScStwLibraries/resources/qml/lib/de/itsblue/ScStw/Styling/Components/TimerColumn.qml b/ScStwLibraries/resources/qml/lib/de/itsblue/ScStw/Styling/Components/TimerColumn.qml index 5150516..4e252c7 100644 --- a/ScStwLibraries/resources/qml/lib/de/itsblue/ScStw/Styling/Components/TimerColumn.qml +++ b/ScStwLibraries/resources/qml/lib/de/itsblue/ScStw/Styling/Components/TimerColumn.qml @@ -87,7 +87,7 @@ Column { width: parent.width * 0.8 height: parent.height * 0.8 - color: ([ScStwTimer.WON].indexOf(control.timers[index]["state"]) >= 0 ? control.colors.success : + color: ([ScStwTimer.WON,ScStwTimer.WILDCARD].indexOf(control.timers[index]["state"]) >= 0 ? control.colors.success : [ScStwTimer.FAILED,ScStwTimer.LOST].indexOf(control.timers[index]["state"]) >= 0 ? control.colors.error: control.colors.text) diff --git a/ScStwLibraries/sources/ScStw.cpp b/ScStwLibraries/sources/ScStw.cpp index f01abb2..dc3cdc3 100644 --- a/ScStwLibraries/sources/ScStw.cpp +++ b/ScStwLibraries/sources/ScStw.cpp @@ -21,38 +21,6 @@ const char *ScStw::SOCKET_MESSAGE_START_KEY = ""; const char *ScStw::SOCKET_MESSAGE_END_KEY = ""; -const QMap ScStw::baseStationSettings = { - {"ReadySoundEnable", ScStw::ReadySoundEnableSetting}, - {"ReadySoundDelay", ScStw::ReadySoundDelaySetting}, - {"AtYourMarksSoundEnable", ScStw::AtYourMarksSoundEnableSetting}, - {"AtYourMarksSoundDelay", ScStw::AtYourMarksSoundDelaySetting}, - {"SoundVolume", ScStw::SoundVolumeSetting} -}; - -ScStw::BaseStationSetting ScStw::baseStationSettingFromString(QString s) { - if(!ScStw::baseStationSettings.contains(s)) - return ScStw::InvalidSetting; - - return ScStw::baseStationSettings[s]; -} - -QString ScStw::baseStationSettingToString(ScStw::BaseStationSetting s) { - for(QString key: ScStw::baseStationSettings.keys()) { - if(ScStw::baseStationSettings[key] == s) - return key; - } - return "Invalid"; -} - -ScStw::BaseStationSetting ScStw::baseStationSettingfromInt(int i) { - bool ok; - BaseStationSetting s = ScStw::toEnumValue(i, &ok); - if(!ok) - return InvalidSetting; - else - return s; -} - ScStw::SignalKey ScStw::signalKeyFromInt(int i) { bool ok; ScStw::SignalKey k = ScStw::toEnumValue(i, &ok); @@ -71,6 +39,17 @@ ScStw::SocketCommand ScStw::socketCommandFromInt(int i) { return c; } +QString ScStw::extensionTypeToString(ExtensionType t) { + switch (t) { + case StartPad: + return "StartPad"; + case TopPad: + return "TopPad"; + default: + return "Invalid"; + } +} + int ScStw::firmwareCompare(QString a, QString b) { /* * * \return -3: major of a is lower than b diff --git a/ScStwLibraries/sources/client/scstwclient.cpp b/ScStwLibraries/sources/client/scstwclient.cpp index e6eaaf8..c705b46 100644 --- a/ScStwLibraries/sources/client/scstwclient.cpp +++ b/ScStwLibraries/sources/client/scstwclient.cpp @@ -20,11 +20,12 @@ ScStwClient * pGlobalScStwClient = nullptr; -ScStwClient::ScStwClient(QObject * parent) : QObject(parent) +ScStwClient::ScStwClient(QObject * parent, QList signalSubscriptions) : QObject(parent) { this->state = DISCONNECTED; - this->nextConnectionId = 1; - this->extensions = QVariantList({}); + this->currentRequestId = 1; + this->extensions = QVariantMap({}); + this->signalSubscriptions = signalSubscriptions; this->socket = new QTcpSocket(this); @@ -49,6 +50,9 @@ ScStwClient::ScStwClient(QObject * parent) : QObject(parent) void ScStwClient::connectToHost() { + if(this->state != DISCONNECTED) + return; + setState(CONNECTING); //connect @@ -72,21 +76,24 @@ bool ScStwClient::init() { this->setState(INITIALISING); this->timeoutTimer->stop(); + QJsonArray signalSubs; + foreach (ScStw::SignalKey key, this->signalSubscriptions) { + signalSubs.append(key); + } + // init remote session - QJsonArray updateSubs = {ScStw::RaceStateChanged, ScStw::TimersChanged, ScStw::ExtensionsChanged, ScStw::NextStartActionChanged}; - QJsonObject sessionParams = {{"updateSubs", updateSubs}, {"init", true}, {"usingTerminationKeys", true}}; + QJsonObject sessionParams = {{"apiVersion", this->API_VERSION}, {"signalSubscriptions", signalSubs}, {"init", true}, {"usingTerminationKeys", true}}; QVariantMap initResponse = this->sendCommand(1, sessionParams, 3000, false); if(initResponse["status"] != 200) { - this->closeConnection(); return false; } this->apiVersion = initResponse["data"].toMap()["apiVersion"].toString(); qDebug() << "[INFO][CLIENT] base station api version is: " << this->apiVersion; int compareResult = ScStw::firmwareCompare(this->API_VERSION, this->apiVersion); - qDebug() << "compare result is: " << compareResult; + //qDebug() << "compare result is: " << compareResult; if( compareResult == -3 ){ // the client version is out of date!! this->closeConnection(); @@ -116,7 +123,17 @@ void ScStwClient::deInit() { if(this->state == DISCONNECTED) return; - this->setExtensions(QVariantList({})); + this->currentRequestId = 1; + this->firmwareVersion = ""; + this->apiVersion = ""; + this->setExtensions(QVariantMap({})); + + for(WaitingRequest waitingRequest : this->waitingRequests.values()) + if(waitingRequest.loop != nullptr) + waitingRequest.loop->exit(ScStw::NotConnectedError); + + this->waitingRequests.clear(); + this->setState(DISCONNECTED); } @@ -155,16 +172,22 @@ QVariantMap ScStwClient::sendCommand(int header, QJsonValue data, int timeout, b return {{"status", ScStw::NotConnectedError}, {"data", "not connected"}}; } - // generate id and witing requests entry - int thisId = nextConnectionId; + // pick a new request id that is not already active + do { + if(this->currentRequestId >= 99) + this->currentRequestId = 0; + else + this->currentRequestId ++; + } while(this->waitingRequests.contains(this->currentRequestId)); + + int thisId = currentRequestId; //qDebug() << "sending command: " << header << " with data: " << data << " and id: " << thisId; - nextConnectionId ++; QEventLoop *loop = new QEventLoop(this); QTimer *timer = new QTimer(this); QJsonObject reply; - this->waitingRequests.append({thisId, loop, reply}); + this->waitingRequests.insert(thisId, {loop, reply}); QJsonObject requestObj; requestObj.insert("id", thisId); @@ -175,7 +198,7 @@ QVariantMap ScStwClient::sendCommand(int header, QJsonValue data, int timeout, b timer->setSingleShot(true); // quit the loop when the timer times out - loop->connect(timer, SIGNAL(timeout()), loop, SLOT(quit())); + connect(timer, &QTimer::timeout, [=]{loop->exit(ScStw::TimeoutError);}); // start the timer before starting to connect timer->start(timeout); @@ -186,37 +209,38 @@ QVariantMap ScStwClient::sendCommand(int header, QJsonValue data, int timeout, b socket->write(jsonRequest.toUtf8()); //wait for an answer to finish (programm gets stuck in here) - loop->exec(); + ScStw::StatusCode statusCode = ScStw::StatusCode(loop->exec()); - - - bool replyFound = false; + // delete the timer + timer->deleteLater(); // find reply and delete the request from waiting list - for(int i = 0; iwaitingRequests.length(); i++){ - if(this->waitingRequests[i].id == thisId){ - // request was found - replyFound = true; - // delete event loop - if(this->waitingRequests[i].loop != nullptr) { - delete this->waitingRequests[i].loop; - } - // store reply - reply = this->waitingRequests[i].reply; - // remove reply from waiting list - this->waitingRequests.removeAt(i); + if(this->waitingRequests.contains(thisId)) { + // request was found + // delete event loop + if(this->waitingRequests[thisId].loop != nullptr) { + this->waitingRequests[thisId].loop->deleteLater(); } - } - if(!replyFound) { + // store reply + reply = this->waitingRequests[thisId].reply; + + // remove reply from waiting list + this->waitingRequests.remove(thisId); + } + else { // some internal error occured - return {{"status", ScStw::Error}, {"data", ""}}; + return {{"status", ScStw::InternalError}, {"data", ""}}; } - if(timer->remainingTime() == -1){ + if(statusCode == ScStw::TimeoutError){ //the time has been triggered -> timeout return {{"status", ScStw::TimeoutError}, {"data", ""}}; } + else if(statusCode == ScStw::NotConnectedError) { + // connection was closed during request + return {{"status", ScStw::NotConnectedError}, {"data", ""}}; + } delete timer; return {{"status", reply.value("header").toInt()}, {"data", reply.value("data").toVariant()}}; @@ -268,7 +292,6 @@ void ScStwClient::handleReadyRead() { //qDebug() << "ready to ready " << socket->bytesAvailable() << " bytes" ; QString reply = socket->readAll(); - //qWarning() << "socket read: " << reply; processSocketMessage(reply); @@ -330,7 +353,7 @@ void ScStwClient::handleSocketMessage(QString reply) { reply.replace(ScStw::SOCKET_MESSAGE_START_KEY, ""); reply.replace(ScStw::SOCKET_MESSAGE_END_KEY, ""); - //qDebug() << "got message: " << reply; + qDebug() << "got message: " << qPrintable(reply); int id = 0; @@ -347,14 +370,12 @@ void ScStwClient::handleSocketMessage(QString reply) { } // this message is the reply to a command! - for(int i = 0; i < this->waitingRequests.length(); i++){ - if(this->waitingRequests[i].id == id){ - this->waitingRequests[i].reply = replyObj; - if(this->waitingRequests[i].loop != nullptr){ - this->waitingRequests[i].loop->quit(); - } - return; + if(this->waitingRequests.contains(id)){ + this->waitingRequests[id].reply = replyObj; + if(this->waitingRequests[id].loop != nullptr){ + this->waitingRequests[id].loop->exit(ScStw::Success); } + return; } } @@ -375,7 +396,7 @@ void ScStwClient::handleSignal(QVariantMap data) { { // the extension connections have changed // -> handle locally - this->setExtensions(data["data"].toList()); + this->setExtensions(data["data"].toMap()); return; break; } @@ -392,19 +413,24 @@ void ScStwClient::handleSignal(QVariantMap data) { // --- helper functions --- // ------------------------ -ScStw::StatusCode ScStwClient::writeRemoteSetting(ScStw::BaseStationSetting key, QString value) { +ScStw::StatusCode ScStwClient::writeRemoteSetting(ScStwSettings::BaseStationSetting key, QVariant value) { QJsonArray requestData; - requestData.append(key); - requestData.append(value); + requestData.append(int(key)); + requestData.append(QJsonValue::fromVariant(value)); return ScStw::StatusCode(this->sendCommand(3000, requestData)["status"].toInt()); } -QString ScStwClient::readRemoteSetting(ScStw::BaseStationSetting key) { - QVariantMap reply = this->sendCommand(3001, key); +QVariant ScStwClient::readRemoteSetting(ScStwSettings::BaseStationSetting key, ScStw::StatusCode* status) { + QVariantMap reply = this->sendCommand(3001, int(key), 10000); + qDebug() << "Setting read status is: " << reply["status"]; + + if(status != nullptr) + *status = ScStw::StatusCode(reply["status"].toInt()); + if(reply["status"] != 200){ return "false"; } - return reply["data"].toString(); + return reply["data"]; } void ScStwClient::setIP(QString newIp){ @@ -429,7 +455,7 @@ void ScStwClient::setState(ScStwClient::State newState){ } } -QVariantList ScStwClient::getExtensions() { +QVariantMap ScStwClient::getExtensions() { return this->extensions; } @@ -445,10 +471,18 @@ QString ScStwClient::getApiVersion() { return this->apiVersion; } -void ScStwClient::setExtensions(QVariantList extensions) { +void ScStwClient::setExtensions(QVariantMap extensions) { + qDebug() << "[CLIENT][DEBUG] Extensions changed: " << extensions; + if(this->extensions != extensions){ this->extensions = extensions; emit this->gotSignal(ScStw::ExtensionsChanged, this->getExtensions()); emit this->extensionsChanged(); } } + + +void ScStwClient::addSignalSubscription(ScStw::SignalKey key) { + if(!this->signalSubscriptions.contains(key)) + this->signalSubscriptions.append(key); +} diff --git a/ScStwLibraries/sources/client/scstwremotemonitorrace.cpp b/ScStwLibraries/sources/client/scstwremotemonitorrace.cpp deleted file mode 100644 index 553598f..0000000 --- a/ScStwLibraries/sources/client/scstwremotemonitorrace.cpp +++ /dev/null @@ -1,241 +0,0 @@ -/**************************************************************************** - ** ScStw Libraries - ** Copyright (C) 2020 Itsblue development - ** - ** This program is free software: you can redistribute it and/or modify - ** it under the terms of the GNU General Public License as published by - ** the Free Software Foundation, either version 3 of the License, or - ** (at your option) any later version. - ** - ** This program is distributed in the hope that it will be useful, - ** but WITHOUT ANY WARRANTY; without even the implied warranty of - ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the - ** GNU General Public License for more details. - ** - ** You should have received a copy of the GNU General Public License - ** along with this program. If not, see . - ****************************************************************************/ - -#include "scstwremotemonitorrace.h" - -ScStwRemoteMonitorRace::ScStwRemoteMonitorRace(ScStwClient *monitorClient, QObject *parent) : ScStwRace(parent) -{ - this->scStwClient = monitorClient; - - connect(this->scStwClient, &ScStwClient::stateChanged, this, &ScStwRemoteMonitorRace::handleClientStateChanged); - connect(this->scStwClient, &ScStwClient::gotSignal, this, &ScStwRemoteMonitorRace::handleBaseStationSignal); -} - -// -------------------------- -// --- Main Functionality --- -// -------------------------- - -int ScStwRemoteMonitorRace::start() { - - if(this->getState() != ScStwRace::IDLE) { - return 904; - } - - qDebug() << "+ --- starting race"; - - int returnCode = 900; - - QVariantMap reply = this->scStwClient->sendCommand(1000); - - if(reply["status"] != 200){ - //handle Error!! - returnCode = reply["status"].toInt(); - } - else { - - returnCode = 200; - - } - - - return returnCode; -} - -int ScStwRemoteMonitorRace::cancel() { - return this->stop(); - -} - -int ScStwRemoteMonitorRace::stop() { - - if(this->getState() != ScStwRace::RUNNING && this->getState() != ScStwRace::STARTING) { - return 904; - } - - // type can be: - // 0: stopp - // 1: cancel - // 2: fail (fase start) - - qDebug() << "+ --- stopping race"; - - int returnCode = 900; - - QVariantMap reply = this->scStwClient->sendCommand(1001); - - if(reply["status"] != 200){ - returnCode = reply["status"].toInt(); - } - else { - returnCode = 200; - } - - - return returnCode; -} - -int ScStwRemoteMonitorRace::reset() { - - if(this->getState() != ScStwRace::STOPPED) { - return 904; - } - - qDebug() << "+ --- resetting race"; - - int returnCode = 900; - - - QVariantMap reply = this->scStwClient->sendCommand(1002); - - if(reply["status"] != 200){ - //handle Error!! - returnCode = reply["status"].toInt(); - } - else { - returnCode = 200; - } - - - - return returnCode; -} - - -// ------------------------- -// --- Base Station sync --- -// ------------------------- - -void ScStwRemoteMonitorRace::handleClientStateChanged() { - // TODO - switch (this->scStwClient->getState()) { - case ScStwClient::CONNECTED: - break; - default: - this->timers.clear(); - this->setState(IDLE); - break; - } -} - -/** - * @brief ScStwAppBackend::handleBaseStationUpdate - * - * Function to handle an update, sent by the base station, which indicates - * that some remote value (like a state) has changed - * - * @param data - */ -void ScStwRemoteMonitorRace::handleBaseStationSignal(ScStw::SignalKey key, QVariant data) { - //qDebug() << "got signal: " << data; - switch (key) { - case ScStw::RaceStateChanged: - { - // the remote race state has changed - this->setState( ScStwRace::RaceState( data.toInt() ) ); - break; - } - case ScStw::TimersChanged: - { - // the remote timers have changed - this->refreshRemoteTimers(data.toList()); - break; - } - case ScStw::NextStartActionChanged: - { - // the next start action has changed - this->nextStartActionTotalDelay = data.toList()[ScStwRace::NextStartActionTotalDelay].toDouble(); - this->nextStartActionDelayStartedAt = QDateTime::currentMSecsSinceEpoch() - (this->nextStartActionTotalDelay * data.toList()[ScStwRace::NextStartActionDelayProgress].toDouble()); - this->nextStartAction = ScStwRace::StartAction( data.toList()[ScStwRace::NextStartAction].toInt() ); - - emit this->nextStartActionChanged(); - break; - } - case ScStw::InvalidSignal: - return; - - default: - return; - } -} - -bool ScStwRemoteMonitorRace::refreshRemoteTimers(QVariantList remoteTimers) { - - if(remoteTimers.length() != this->timers.length()){ - // local timers are out of sync - - // delete all current timers - foreach(ScStwTimer * locTimer, this->timers){ - delete locTimer; - } - - this->timers.clear(); - - foreach(QVariant remoteTimer, remoteTimers){ - // create a local timer for each remote timer - ScStwTimer * timer = new ScStwTimer(this, true); - this->timers.append(timer); - connect(timer, &ScStwTimer::stateChanged, this, &ScStwRace::timersChanged); - connect(timer, &ScStwTimer::reactionTimeChanged, this, &ScStwRace::timersChanged); - } - } - - foreach(QVariant remoteTimer, remoteTimers){ - int currId = remoteTimer.toMap()["id"].toInt(); - ScStwTimer::TimerState newState = ScStwTimer::TimerState(remoteTimer.toMap()["state"].toInt()); - double currentMSecsSinceEpoch = QDateTime::currentMSecsSinceEpoch(); - - this->timers[currId]->setStartTime(currentMSecsSinceEpoch - remoteTimer.toMap()["currentTime"].toDouble()); - - if(newState >= ScStwTimer::WAITING) - this->timers[currId]->setStopTime(currentMSecsSinceEpoch); - - this->timers[currId]->setReactionTime(remoteTimer.toMap()["reactionTime"].toDouble()); - - this->timers[currId]->setState(newState, true); - - this->timers[currId]->setLetter(remoteTimer.toMap()["letter"].toString()); - } - - return true; -} - -bool ScStwRemoteMonitorRace::addTimer(ScStwTimer* timer) { - Q_UNUSED(timer) - return false; -} - -QVariantList ScStwRemoteMonitorRace::getNextStartActionDetails() { - int nextActionDelay = 0; - double nextActionDelayProg = -1; - - if(this->nextStartAction == AtYourMarks || this->nextStartAction == Ready) { - // get the total delay and the delay progress of the next action timer - double elapsed = QDateTime::currentMSecsSinceEpoch() - this->nextStartActionDelayStartedAt; - nextActionDelay = this->nextStartActionTotalDelay; - if(elapsed < 0 || elapsed > nextActionDelay) { - elapsed = nextActionDelay; - } - nextActionDelayProg = elapsed / nextActionDelay; - } - - return { - this->nextStartAction, - nextActionDelay, - nextActionDelayProg - }; -} diff --git a/ScStwLibraries/sources/client/scstwremoterace.cpp b/ScStwLibraries/sources/client/scstwremoterace.cpp new file mode 100644 index 0000000..53b3f54 --- /dev/null +++ b/ScStwLibraries/sources/client/scstwremoterace.cpp @@ -0,0 +1,370 @@ +/**************************************************************************** + ** ScStw Libraries + ** Copyright (C) 2020 Itsblue development + ** + ** This program is free software: you can redistribute it and/or modify + ** it under the terms of the GNU General Public License as published by + ** the Free Software Foundation, either version 3 of the License, or + ** (at your option) any later version. + ** + ** This program is distributed in the hope that it will be useful, + ** but WITHOUT ANY WARRANTY; without even the implied warranty of + ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + ** GNU General Public License for more details. + ** + ** You should have received a copy of the GNU General Public License + ** along with this program. If not, see . + ****************************************************************************/ + +#include "scstwremoterace.h" + +ScStwRemoteRace::ScStwRemoteRace(QObject* parent) : ScStwRemoteRace(nullptr, nullptr, parent) +{ +} + +ScStwRemoteRace::ScStwRemoteRace(ScStwClient *scStwClient, ScStwSettings *settings, QObject *parent) : ScStwRace(settings, parent) +{ + this->isReadyForNextState = true; + + this->remoteTimers = {}; + this->localTimers = {}; + + this->setScStwClient(scStwClient); +} + +// -------------------------- +// --- Main Functionality --- +// -------------------------- + +ScStw::StatusCode ScStwRemoteRace::start(bool asyncronous) { + if(this->local()) + return ScStwRace::start(asyncronous); + + if(this->getState() != ScStwRace::IDLE && this->getState() != ScStwRace::WAITING) + return ScStw::CurrentStateNotVaildForOperationError; + + qDebug() << "+ --- starting race"; + + QVariantMap reply = this->scStwClient->sendCommand(ScStw::StartRaceCommand); + + return ScStw::StatusCode(reply["status"].toInt()); +} + +ScStw::StatusCode ScStwRemoteRace::cancel() { + if(this->local()) + return ScStwRace::cancel(); + + if(this->getState() != PREPAIRING && this->getState() != WAITING && this->getState() != STARTING && this->getState() != RUNNING) + return ScStw::CurrentStateNotVaildForOperationError; + + qDebug() << "+ --- stopping race"; + + QVariantMap reply = this->scStwClient->sendCommand(ScStw::CancelRaceCommand); + + return ScStw::StatusCode(reply["status"].toInt()); +} + +ScStw::StatusCode ScStwRemoteRace::stop() { + if(this->local()) + return ScStwRace::stop(); + + if(this->getState() != ScStwRace::RUNNING && this->getState() != ScStwRace::STARTING) + return ScStw::CurrentStateNotVaildForOperationError; + + qDebug() << "+ --- stopping race"; + + QVariantMap reply = this->scStwClient->sendCommand(ScStw::StopRaceCommand); + + return ScStw::StatusCode(reply["status"].toInt()); +} + +ScStw::StatusCode ScStwRemoteRace::reset() { + if(this->local()) + return ScStwRace::reset(); + + if(this->getState() != ScStwRace::STOPPED && this->getState() != ScStwRace::INCIDENT) + return ScStw::CurrentStateNotVaildForOperationError; + + qDebug() << "+ --- resetting race"; + + QVariantMap reply = this->scStwClient->sendCommand(ScStw::ResetRaceCommand); + + return ScStw::StatusCode(reply["status"].toInt()); +} + +ScStw::StatusCode ScStwRemoteRace::setTimerDisabled(int timerId, bool disabled) { + if(this->state != IDLE && this->state != WAITING) + return ScStw::CurrentStateNotVaildForOperationError; + + if(timerId < 0 || timerId - 1 > this->timers.length()) + return ScStw::ItemNotFoundError; + + qDebug() << "+ --- setting timer " << timerId << " to disabled: " << disabled; + + QVariantMap requestData = { + {"timerId", timerId}, + {"disabled", disabled} + }; + + QVariantMap reply = this->scStwClient->sendCommand(ScStw::SetTimerDisabledCommand, QJsonValue::fromVariant(requestData)); + + qDebug() << "+ --- timer disabled command returned: " << reply; + + return ScStw::StatusCode(reply["status"].toInt()); +} + +// ------------------------- +// --- Base Station sync --- +// ------------------------- + +void ScStwRemoteRace::handleClientStateChange() { + switch (this->scStwClient->getState()) { + case ScStwClient::INITIALISING: + // disconnect all local timers from race + this->setTimers({}, false); + // delete all obsolete remote timers + for(ScStwRemoteTimer* oldRemoteTimer : this->remoteTimers) + oldRemoteTimer->deleteLater(); + + this->remoteTimers.clear(); + break; + case ScStwClient::DISCONNECTED: + this->remoteTimers.clear(); + this->setTimers(this->localTimers, false); + emit this->timersChanged(); + emit this->detailsChanged(); + emit this->currentStartDelayChanged(); + this->competitionMode = false; + this->setState(IDLE); + break; + default: + break; + } +} + +void ScStwRemoteRace::setTimers(QList timers, bool deleteOldTimers) { + + // disconnect all signals of all current timers + //qDebug() << "SETTING TIMERS"; + + foreach(ScStwTimer *existingTimer, this->timers) { + disconnect(existingTimer, &ScStwTimer::stateChanged, this, &ScStwRace::handleTimerStateChange); + disconnect(existingTimer, &ScStwTimer::stateChanged, this, &ScStwRace::timersChanged); + disconnect(existingTimer, &ScStwTimer::wantsToBeDisabledChanged, this, &ScStwRace::handleTimerWantsToBeDisabledChange); + disconnect(existingTimer, &ScStwTimer::reactionTimeChanged, this, &ScStwRace::timersChanged); + disconnect(existingTimer, &ScStwTimer::readyStateChanged, this, &ScStwRace::handleTimerReadyStateChange); + disconnect(existingTimer, &ScStwTimer::readyStateChanged, this, &ScStwRace::isReadyForNextStateChanged); + + if(deleteOldTimers) + existingTimer->deleteLater(); + } + + this->timers.clear(); + + for(ScStwTimer* timer : timers) { + this->addTimer(timer); + } +} + +ScStwRemoteRace::RaceMode ScStwRemoteRace::getMode() { + if(this->scStwClient == nullptr || this->scStwClient->getState() != ScStwClient::CONNECTED) + return LOCAL; + else + return REMOTE; +} + +bool ScStwRemoteRace::local() { + return this->getMode() == LOCAL; +} + +/** + * @brief ScStwAppBackend::handleBaseStationUpdate + * + * Function to handle an update, sent by the base station, which indicates + * that some remote value (like a state) has changed + * + * @param data + */ +void ScStwRemoteRace::handleBaseStationSignal(ScStw::SignalKey key, QVariant data) { + switch (key) { + case ScStw::RaceDetailsChanged: + { + this->refreshDetails(data.toMap()); + break; + } + default: + return; + } +} + +void ScStwRemoteRace::refreshDetails(QVariantMap details) { + // the details of the race have changed: + + qDebug() << "RACE DETAILS: " << details; + + // competition mode + if(this->competitionMode != details["competitionMode"].toBool()) { + this->competitionMode = details["competitionMode"].toBool(); + emit this->competitionModeChanged(); + } + + // ready sound enabled + if(this->readySoundEnabled != details["readySoundEnabled"].toBool()) { + this->readySoundEnabled = details["readySoundEnabled"].toBool(); + emit this->readySoundEnabledChanged(); + } + + // current start delay + this->currentStartTotalDelay = details["currentStartDelay"].toMap()["total"].toInt(); + this->latestStartDelayProgress = details["currentStartDelay"].toMap()["progress"].toDouble(); + this->currentStartDelayStartedAt = QDateTime::currentMSecsSinceEpoch() - (this->currentStartTotalDelay * this->latestStartDelayProgress); + + emit this->currentStartDelayChanged(); + + // timers + this->refreshRemoteTimers(details["timers"].toList()); + + // state + this->setState(ScStwRace::RaceState(details["state"].toInt())); + + // isReady + if(this->state == WAITING) { + this->isReadyForNextState = details["isReadyForNextState"].toBool(); + emit this->isReadyForNextStateChanged(); + } + + emit this->detailsChanged(); +} + +void ScStwRemoteRace::rebuildRemoteTimers(QVariantList remoteTimers) { + + qDebug() << "REBUILDING TIMERS"; + + // delete all current timers + foreach(ScStwTimer * oldTimer, this->timers){ + oldTimer->deleteLater(); + } + + this->remoteTimers.clear(); + this->timers.clear(); + + foreach(QVariant remoteTimer, remoteTimers){ + // create a local timer for each remote timer + ScStwRemoteTimer * timer = new ScStwRemoteTimer(this); + this->timers.append(timer); + this->remoteTimers.append(timer); + connect(timer, &ScStwTimer::stateChanged, this, &ScStwRace::timersChanged); + connect(timer, &ScStwTimer::reactionTimeChanged, this, &ScStwRace::timersChanged); + } +} + +bool ScStwRemoteRace::refreshRemoteTimers(QVariantList newRemoteTimers) { + + qDebug() << "REFRESHING TIMERS"; + + if(newRemoteTimers.length() != this->remoteTimers.length()){ + // local timers are out of sync + this->rebuildRemoteTimers(newRemoteTimers); + qDebug() << "rebuilding remote timers"; + } + + for(QVariant remoteTimer : newRemoteTimers){ + int currId = remoteTimer.toMap()["id"].toInt(); + + if(this->remoteTimers.length() <= currId) + this->rebuildRemoteTimers(newRemoteTimers); + + ScStwTimer::TimerState newState = ScStwTimer::TimerState(remoteTimer.toMap()["state"].toInt()); + + qDebug() << "refreshing timers: id: " << currId << " state: " << newState << " readyState: " << remoteTimer.toMap()["readyState"].toInt() << " currentTime: " << remoteTimer.toMap()["currentTime"].toDouble(); + + double currentMSecsSinceEpoch = QDateTime::currentMSecsSinceEpoch(); + + this->remoteTimers[currId]->setStartTime(currentMSecsSinceEpoch - remoteTimer.toMap()["currentTime"].toDouble()); + + if(newState >= ScStwTimer::WAITING) + this->remoteTimers[currId]->setStopTime(currentMSecsSinceEpoch); + + this->remoteTimers[currId]->setReactionTime(remoteTimer.toMap()["reactionTime"].toDouble()); + this->remoteTimers[currId]->setLetter(remoteTimer.toMap()["letter"].toString()); + this->remoteTimers[currId]->setReadyState(ScStwTimer::ReadyState(remoteTimer.toMap()["readyState"].toInt())); + + this->remoteTimers[currId]->setState(newState); + } + + emit this->timersChanged(); + + return true; +} + +bool ScStwRemoteRace::addTimer(ScStwTimer* timer) { + if(this->local()) { + this->localTimers.append(timer); + return ScStwRace::addTimer(timer); + } + + return false; +} + +QVariantMap ScStwRemoteRace::getCurrentStartDelay() { + if(this->local()) + return ScStwRace::getCurrentStartDelay(); + + QVariantMap currentStartDelay = { + {"total", -1.0}, + {"progress", -1.0} + }; + + if(this->currentStartTotalDelay == -1) { + currentStartDelay["progress"] = this->latestStartDelayProgress; + } + else if(this->getState() == PREPAIRING || this->getState() == WAITING) { + // get the total delay and the delay progress of the next action timer + double elapsed = QDateTime::currentMSecsSinceEpoch() - this->currentStartDelayStartedAt; + currentStartDelay["total"] = this->currentStartTotalDelay; + if(elapsed < 0 || elapsed > currentStartDelay["total"].toDouble()) { + elapsed = currentStartDelay["total"].toDouble(); + } + currentStartDelay["progress"] = elapsed / currentStartDelay["total"].toDouble(); + } + + return currentStartDelay; +} + +bool ScStwRemoteRace::getIsReadyForNextState() { + if(this->local()) + return ScStwRace::getIsReadyForNextState(); + + return this->isReadyForNextState; +} + +bool ScStwRemoteRace::getReadySoundEnabled() { + if(this->local()) + return ScStwRace::getReadySoundEnabled(); + + return this->readySoundEnabled; +} + +void ScStwRemoteRace::refreshCompetitionMode() { + if(this->local()) + return ScStwRace::refreshCompetitionMode(); +} + +ScStwClient* ScStwRemoteRace::getScStwClient() { + return this->scStwClient; +} + +void ScStwRemoteRace::setScStwClient(ScStwClient* client) { + if(client == this->scStwClient) + return; + + this->scStwClient = client; + + if(this->scStwClient != nullptr) { + this->scStwClient->addSignalSubscription(ScStw::RaceDetailsChanged); + + connect(this->scStwClient, &ScStwClient::stateChanged, this, &ScStwRemoteRace::handleClientStateChange); + connect(this->scStwClient, &ScStwClient::gotSignal, this, &ScStwRemoteRace::handleBaseStationSignal); + + } +} diff --git a/ScStwLibraries/sources/client/scstwremotesettings.cpp b/ScStwLibraries/sources/client/scstwremotesettings.cpp new file mode 100644 index 0000000..ca30b92 --- /dev/null +++ b/ScStwLibraries/sources/client/scstwremotesettings.cpp @@ -0,0 +1,104 @@ +/**************************************************************************** + ** ScStw Libraries + ** Copyright (C) 2020 Itsblue development + ** + ** This program is free software: you can redistribute it and/or modify + ** it under the terms of the GNU General Public License as published by + ** the Free Software Foundation, either version 3 of the License, or + ** (at your option) any later version. + ** + ** This program is distributed in the hope that it will be useful, + ** but WITHOUT ANY WARRANTY; without even the implied warranty of + ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + ** GNU General Public License for more details. + ** + ** You should have received a copy of the GNU General Public License + ** along with this program. If not, see . + ****************************************************************************/ + +#include "../../headers/client/scstwremotesettings.h" + +ScStwRemoteSettings::ScStwRemoteSettings(QObject* parent) : ScStwSettings(parent) +{ + this->scStwClient = nullptr; +} + +ScStwRemoteSettings::ScStwRemoteSettings(ScStwClient * scStwClient, QObject * parent) : ScStwSettings(parent) +{ + this->setScStwClient(scStwClient); +} + +ScStwRemoteSettings::SettingsMode ScStwRemoteSettings::getMode() { + if(this->scStwClient == nullptr || this->scStwClient->getState() != ScStwClient::CONNECTED) + return LOCAL; + else + return REMOTE; +} + +QVariant ScStwRemoteSettings::readSetting(QString key, int keyInt, int keyLevel) { + if(this->getMode() == LOCAL || keyLevel > ScStwSettings::KeyLevel) + return ScStwSettings::readSetting(key, keyInt, keyLevel); + + qDebug() << "Setting read: keyLevel: " << keyLevel << " key: " << key; + + ScStw::StatusCode status; + QVariant value = this->scStwClient->readRemoteSetting(ScStwSettings::BaseStationSetting(keyInt), &status); + + if(status == ScStw::Success) + return value; + else + return QVariant(); +} + +bool ScStwRemoteSettings::writeSetting(QString key, QVariant value, int keyInt, int keyLevel) { + if(this->getMode() == LOCAL || keyLevel > ScStwSettings::KeyLevel) + return ScStwSettings::writeSetting(key, value, keyInt, keyLevel); + + + qDebug() << "Setting write: keyLevel: " << keyLevel << " key: " << key << " value: " << value; + + ScStw::StatusCode res = this->scStwClient->writeRemoteSetting(ScStwSettings::BaseStationSetting(keyInt), value); + + return res == ScStw::Success; +} + +bool ScStwRemoteSettings::setDefaultSetting(QString key, QVariant defaultVariant, int keyInt, int keyLevel) { + if(this->getMode() == LOCAL || keyLevel > ScStwSettings::KeyLevel) + return ScStwSettings::setDefaultSetting(key, defaultVariant, keyInt, keyLevel); + + return false; +} + +void ScStwRemoteSettings::handleClientStateChange() { + if(this->scStwClient->getState() == ScStwClient::DISCONNECTED) + emit this->settingChanged(-1, ScStwSettings::KeyLevel, QVariant()); + // Dont't need to do that when changing to connected, + // as the basestation emits a wildcrd setting changed on connect anyway +} + +void ScStwRemoteSettings::handleBaseStationSignal(ScStw::SignalKey key, QVariant data) { + switch (key) { + case ScStw::SettingChanged: + emit this->settingChanged(data.toMap()["key"].toInt(), ScStwSettings::KeyLevel, data.toMap()["value"]); + default: + break; +} +} + +ScStwClient* ScStwRemoteSettings::getScStwClient() { + return this->scStwClient; +} + +void ScStwRemoteSettings::setScStwClient(ScStwClient* client) { + if(client == this->scStwClient) + return; + + this->scStwClient = client; + + if(this->scStwClient != nullptr) { + this->scStwClient->addSignalSubscription(ScStw::SettingChanged); + + connect(this->scStwClient, &ScStwClient::stateChanged, this, &ScStwRemoteSettings::handleClientStateChange); + connect(this->scStwClient, &ScStwClient::gotSignal, this, &ScStwRemoteSettings::handleBaseStationSignal); + } +} diff --git a/ScStwLibraries/sources/client/scstwremotetimer.cpp b/ScStwLibraries/sources/client/scstwremotetimer.cpp new file mode 100644 index 0000000..771d0c6 --- /dev/null +++ b/ScStwLibraries/sources/client/scstwremotetimer.cpp @@ -0,0 +1,42 @@ +#include "../../headers/client/scstwremotetimer.h" + +ScStwRemoteTimer::ScStwRemoteTimer(QObject *parent) : ScStwTimer(parent) +{ + this->readyState = ScStwTimer::getReadyState(); +} + +ScStwTimer::ReadyState ScStwRemoteTimer::getReadyState() { + return this->readyState; +} + +void ScStwRemoteTimer::setStartTime(double startTime) { + this->startTime = startTime; +} + +void ScStwRemoteTimer::setStopTime(double stopTime) { + this->stopTime = stopTime; +} + +void ScStwRemoteTimer::setReactionTime(double reactionTime) { + this->reactionTime = reactionTime; + emit this->reactionTimeChanged(); +} + +void ScStwRemoteTimer::setLetter(QString newLetter) { + this->letter = newLetter; +} + +void ScStwRemoteTimer::setReadyState(ScStwTimer::ReadyState readyState) { + if(this->readyState != readyState) { + this->readyState = readyState; + emit this->readyStateChanged(readyState); + } +} + +void ScStwRemoteTimer::setState(TimerState newState){ + if(this->state != newState) { + this->state = newState; + qDebug() << "+ [INFO][REMOTETIMER] timer state changed: " << newState; + emit this->stateChanged(this->state); + } +} diff --git a/ScStwLibraries/sources/scstwlibraries.cpp b/ScStwLibraries/sources/scstwlibraries.cpp index 14068d2..64696c8 100644 --- a/ScStwLibraries/sources/scstwlibraries.cpp +++ b/ScStwLibraries/sources/scstwlibraries.cpp @@ -26,16 +26,23 @@ ScStwLibraries::ScStwLibraries(QObject *parent) : QObject(parent) void ScStwLibraries::init() { #ifdef ScStwLibraries_QML qmlRegisterType("de.itsblue.ScStw", 2, 0, "ScStw"); - qRegisterMetaType("ScStw::BaseStationSetting"); + qRegisterMetaType("ScStwSettings::BaseStationSetting"); qRegisterMetaType("ScStw::SocketCommand"); + qRegisterMetaType("ScStw::StatusCode"); qmlRegisterType("de.itsblue.ScStw", 2, 0, "ScStwRace"); qmlRegisterType("de.itsblue.ScStw", 2, 0, "ScStwTimer"); #ifdef ScStwLibraries_ClientLibs qmlRegisterType("de.itsblue.ScStw", 2, 0, "ScStwClient"); + qmlRegisterType("de.itsblue.ScStw", 2, 0, "ScStwSettings"); + qmlRegisterUncreatableType("de.itsblue.ScStw", 2, 0, "ScStwSetting", "The ScStwSetting is manage by a ScStwSettings instance and therefore not creatable"); + qmlRegisterType("de.itsblue.ScStw", 2, 0, "ScStwRemoteRace"); + qmlRegisterType("de.itsblue.ScStw", 2, 0, "ScStwRemoteSettings"); #endif #endif + + qRegisterMetaType("ScStw::PadState"); } diff --git a/ScStwLibraries/sources/scstwrace.cpp b/ScStwLibraries/sources/scstwrace.cpp index 5998f13..b576d81 100644 --- a/ScStwLibraries/sources/scstwrace.cpp +++ b/ScStwLibraries/sources/scstwrace.cpp @@ -18,64 +18,86 @@ #include "../headers/scstwrace.h" -ScStwRace::ScStwRace(QObject *parent) : QObject(parent) +ScStwRace::ScStwRace(QObject* parent) : ScStwRace(nullptr, parent) +{ +} + +ScStwRace::ScStwRace(ScStwSettings *settings, QObject *parent) : QObject(parent) { this->state = IDLE; + this->competitionMode = false; + this->autoRefreshTimerText = false; // configure the loop that waits for the sound effect to finish this->soundPlayer = new ScStwSoundPlayer(); // configure timer that handles the delay between the start commands - this->nextActionTimer = new QTimer(this); - nextActionTimer->setSingleShot(true); - this->nextActionLoop = new QEventLoop(); - this->nextStartAction = None; + this->startDelayTimer = new QTimer(this); + startDelayTimer->setSingleShot(true); + this->startWaitLoop = new QEventLoop(this); - connect(this->nextActionTimer, &QTimer::timeout, this->nextActionLoop, &QEventLoop::quit); - connect(this, &ScStwRace::nextStartActionChanged, this, &ScStwRace::nextStartActionDetailsChanged); + connect(this->startDelayTimer, &QTimer::timeout, this->startWaitLoop, &QEventLoop::quit); - // write default settings - this->startActionSettings.insert(Start, {{"Enabled", true}, {"Delay", 1}}); - this->writeStartActionSetting(AtYourMarks, false, 0); - this->writeStartActionSetting(Ready, false, 0); - this->setSoundVolume(1.0); + connect(this, &ScStwRace::currentStartDelayChanged, this, &ScStwRace::detailsChanged); + connect(this, &ScStwRace::timersChanged, this, &ScStwRace::detailsChanged); + connect(this, &ScStwRace::stateChanged, this, &ScStwRace::detailsChanged); + + // init settings + this->setSettings(settings); } // -------------------------- // --- Main Functionality --- // -------------------------- -int ScStwRace::start(bool asyncronous) { - if(this->state != IDLE) { +ScStw::StatusCode ScStwRace::start(bool asyncronous) { + if(this->state == WAITING) { + if(this->getIsReadyForNextState()) { + this->startWaitLoop->exit(LoopManualExit); + return ScStw::Success; + } + else { + return ScStw::TimersNotReadyError; + } + } + else if(this->state != IDLE) { return ScStw::CurrentStateNotVaildForOperationError; } - qDebug() << "+ [INFO] starting race"; + this->refreshCompetitionMode(); - this->setState(STARTING); + if(!this->getIsReadyForNextState()) + return ScStw::TimersNotReadyError; + + this->setState(PREPAIRING); if(asyncronous) { - QTimer::singleShot(1, [=](){this->playSoundsAndStartTimers(None);}); + QTimer::singleShot(1, [=]() { + this->playSoundsAndStartTimers(); + }); } else - this->playSoundsAndStartTimers(None); + this->playSoundsAndStartTimers(); return ScStw::Success; } -int ScStwRace::stop() { +ScStw::StatusCode ScStwRace::stop() { if(this->state != RUNNING && this->state != STARTING) { return ScStw::CurrentStateNotVaildForOperationError; } - qDebug() << "+ [INFO] stopping race"; + if(this->competitionMode) + return this->cancel(); + + qDebug() << "[INFO][RACE] stopping race"; double timeOfStop = QDateTime::currentMSecsSinceEpoch(); - int returnCode = ScStw::Success; + ScStw::StatusCode returnCode = ScStw::Success; - foreach(ScStwTimer *speedTimer, this->timers){ - if(!speedTimer->stop(timeOfStop) && speedTimer->getState() != ScStwTimer::DISABLED){ + foreach(ScStwTimer *speedTimer, this->timers) { + if(!speedTimer->stop(timeOfStop) && speedTimer->getState() != ScStwTimer::DISABLED) { returnCode = ScStw::InternalErrorTimerOperationFailed; } } @@ -87,160 +109,270 @@ int ScStwRace::stop() { return returnCode; } -void ScStwRace::handleTimerStop() { - if(this->state == RUNNING) { - // find out which timer has won - double lowestStoppedTime = -1; - QList timersWhichHaveWonIds; - - // iterate through all timers and find the lowest time taht was stopped - foreach(ScStwTimer * timer, this->timers) { - if(timer->getCurrentTime() > 0 && (timer->getCurrentTime() <= lowestStoppedTime || lowestStoppedTime < 0)) { - // this is the timer with the lowest stopped time - lowestStoppedTime = timer->getCurrentTime(); - } - } - - // append the timer(s) with the lowest stopped time to the winner list - foreach(ScStwTimer * timer, this->timers) { - if(timer->getCurrentTime() > 0 - && (timer->getCurrentTime() <= lowestStoppedTime || lowestStoppedTime < 0) - && timer->getState() != ScStwTimer::RUNNING - ) { - // this is the timer with the lowest stopped time - timersWhichHaveWonIds.append(timer); - } - } - - // update the states of all timers - foreach(ScStwTimer * timer, this->timers) { - if(timer->getState() == ScStwTimer::RUNNING) - continue; - - if(timersWhichHaveWonIds.contains(timer)) { - timer->setResult(ScStwTimer::WON); - } - else { - timer->setResult(ScStwTimer::LOST); - } - } - } -} - -int ScStwRace::reset() { - if(this->state != STOPPED) { +ScStw::StatusCode ScStwRace::reset() { + if(this->state != STOPPED && this->state != INCIDENT) { return ScStw::CurrentStateNotVaildForOperationError; } - qDebug() << "+ [INFO] resetting race"; + qDebug() << "[INFO][RACE] resetting race"; - int returnCode = ScStw::Success; + if(this->competitionMode) { + for(int i = 0; i < this->timers.length(); i++) + this->setTimerDisabled(i, false); + } - foreach(ScStwTimer *speedTimer, this->timers){ - if(!speedTimer->reset() && speedTimer->getState() != ScStwTimer::DISABLED) { + ScStw::StatusCode returnCode = ScStw::Success; + + foreach(ScStwTimer *timer, this->timers) { + if(!timer->reset() && timer->getState() != ScStwTimer::DISABLED && timer->getState() != ScStwTimer::IDLE) { returnCode = ScStw::InternalErrorTimerOperationFailed; } } - if(returnCode == ScStw::Success){ + if(returnCode == ScStw::Success) this->setState(IDLE); - } + + this->soundPlayer->cancel(); return returnCode; } -int ScStwRace::cancel() { - if(this->state != STARTING && this->state != RUNNING) +ScStw::StatusCode ScStwRace::cancel() { + if(this->state != PREPAIRING && this->state != WAITING && this->state != STARTING && this->state != RUNNING) return ScStw::CurrentStateNotVaildForOperationError; qDebug() << "[INFO][RACE] cancelling race"; - this->soundPlayer->cancel(this->soundVolume); - this->nextActionTimer->stop(); - this->nextActionLoop->quit(); - this->nextStartAction = None; + ScStw::StatusCode returnCode = ScStw::Success; - int returnCode = ScStw::Success; - - foreach(ScStwTimer *timer, this->timers){ + foreach(ScStwTimer *timer, this->timers) { if(!timer->cancel() && timer->getState() != ScStwTimer::DISABLED) returnCode = ScStw::InternalErrorTimerOperationFailed; } - return returnCode; -} - -int ScStwRace::handleFalseStart() { - if(this->getState() != STARTING && this->getState() != RUNNING) - return ScStw::CurrentStateNotVaildForOperationError; - - int returnCode = ScStw::Success; - // cancel all running timers - foreach(ScStwTimer *timer, this->timers) { - if(!timer->cancel() && timer->getState() != ScStwTimer::DISABLED && timer->getState() != ScStwTimer::FAILED) - returnCode = ScStw::InternalErrorTimerOperationFailed; - } + if(returnCode != ScStw::Success) + return returnCode; this->setState(STOPPED); - this->soundPlayer->cancel(this->soundVolume); + + this->startWaitLoop->exit(LoopCancelExit); + this->soundPlayer->cancel(); + this->startDelayTimer->stop(); + + emit this->currentStartDelayChanged(); return returnCode; } -bool ScStwRace::playSoundsAndStartTimers(StartAction thisAction) { - if(this->state != STARTING) - return true; +ScStw::StatusCode ScStwRace::setTimerDisabled(int timerId, bool disabled) { + if(timerId < 0 || timerId - 1 > this->timers.length()) + return ScStw::ItemNotFoundError; - double timeOfSoundPlaybackStart; - if(this->startActionSettings.contains(thisAction) && this->startActionSettings[thisAction]["Enabled"].toBool()) { - if(!this->soundPlayer->play(thisAction, this->soundVolume, &timeOfSoundPlaybackStart)) - return false; - if(this->state != STARTING) - return true; + return this->setTimerDisabled(this->timers[timerId], disabled); +} + +ScStw::StatusCode ScStwRace::setTimerDisabled(ScStwTimer* timer, bool disabled) { + //qDebug() << "[INFO][RACE] Setting timer "<< timer->getLetter() << " to disabled: " << disabled << " this state: " << this->state; + if(this->state != IDLE && this->state != WAITING) + return ScStw::CurrentStateNotVaildForOperationError; + + if(!this->timers.contains(timer)) + return ScStw::ItemNotFoundError; + + int enabledTimerCount = 0; + + foreach(ScStwTimer *timer, this->timers) { + if(timer->getState() != ScStwTimer::DISABLED) + enabledTimerCount ++; } - if(thisAction < Start) { - this->nextStartAction = StartAction(thisAction + 1); - qDebug() << "next action is: " << nextStartAction; + if(disabled && enabledTimerCount <= 1) + return ScStw::LastTimerCannotBeDisabledError; - int nextActionDelay = -1; - if(this->startActionSettings.contains(this->nextStartAction) && this->startActionSettings[this->nextStartAction]["Enabled"].toBool()) { - nextActionDelay = this->startActionSettings[this->nextStartAction]["Delay"].toInt(); + //qDebug() << "[INFO][RACE] Setting timer "<< timer->getLetter() << " to disabled: " << disabled; + + timer->setDisabled(disabled); + + emit this->timersChanged(); + + return ScStw::Success; + +} + +void ScStwRace::handleTimerReadyStateChange(ScStwTimer::ReadyState readyState) { + + if(!this->competitionMode || this->state == IDLE || this->state == STOPPED || this->state == INCIDENT ) + return; + + qDebug() << "Some ready state changed: " << readyState; + + // cancel as a technical incident if extensions are disconnected or run low on battery + if(readyState == ScStwTimer::ExtensionIsNotConnected || readyState == ScStwTimer::ExtensionBatteryIsCritical) { + this->technicalIncident(); + return; + } + + // only continue if the current state is waiting + if(this->state == WAITING) { + this->startWaitLoop->exit(ScStwRace::LoopReadyStateChangeExit); + emit this->timersChanged(); + } +} + +bool ScStwRace::playSoundsAndStartTimers() { + if(this->state != PREPAIRING) + return true; + + // The check if all timers are ready has already happened at this point + ScStwSoundPlayer::PlayResult res = this->doDelayAndSoundOfCurrentStartState(); + + if(res == ScStwSoundPlayer::Error) { + qDebug() << "[ERROR][RACE] error playing at your marks sound"; + this->technicalIncident(); + return false; + } + else if(res == ScStwSoundPlayer::Cancelled) { + return false; + } + + // check if the start was cancelled + if(!this->isStarting()) + return false; + + this->setState(WAITING); + + // do climber readiness tests + + // if the automatic ready tone is enabled, wait for the climbers to become ready + if(this->competitionMode && this->getSoundEnabledSetting(ScStwSoundPlayer::Ready)) { + + qDebug() << "[RACE][INFO] Now waiting for climbers"; + + // get delay + int minimumReadyDelay = 1000; + if(this->getSoundDelaySetting(ScStwSoundPlayer::Ready) > 1000) + minimumReadyDelay = this->getSoundDelaySetting(ScStwSoundPlayer::Ready); + + this->startDelayTimer->setInterval(minimumReadyDelay); + + // wait for all climbers to be ready for the ReadyActionDelay, but at least one second continuosly + // the climber ready wait loop will also quit, if the climber steps of the pad + // -> wait for both climbers to stand on the pad for at least one second + + bool timerTriggered = true; + do { + + if(!this->getIsReadyForNextState()) { + this->startDelayTimer->stop(); + timerTriggered = false; + } + else if(this->startDelayTimer->isActive()) { + timerTriggered = true; + } + else { + this->startDelayTimer->stop(); + this->startDelayTimer->setInterval(minimumReadyDelay); + this->startDelayTimer->start(); + timerTriggered = true; + } + + emit this->currentStartDelayChanged(); + + int loopExitCode = this->startWaitLoop->exec(); + + switch (loopExitCode) { + case LoopAutomaticExit: + break; + case LoopManualExit: + // prevent manual stop + timerTriggered = false; + break; + case LoopCancelExit: + return false; + } + + //qDebug() << "At end of loop: remaining time: " << this->startDelayTimer->remainingTime() << " timer triggered: " << timerTriggered << " ready for next state: " << this->isReadyForNextState(); + + } while(this->startDelayTimer->remainingTime() > 0 || !timerTriggered || !this->getIsReadyForNextState()); + + qDebug() << "[DEBUG][RACE] Wait finished, starting now!"; + + // play ready tone + ScStwSoundPlayer::PlayResult res = this->soundPlayer->play(ScStwSoundPlayer::Ready, this->getSoundVolume()); + + if(res == ScStwSoundPlayer::Error) { + qDebug() << "[ERROR][RACE] error playing ready sound"; + this->technicalIncident(); + return false; } + else if(res == ScStwSoundPlayer::Cancelled) { + return false; + } + } + else if(this->competitionMode) { + // wait for climbers and manual start + int loopExitCode; + do { + loopExitCode = this->startWaitLoop->exec(); - // perform next action - this->nextActionTimer->start(nextActionDelay > 0 ? nextActionDelay:1); - if(nextActionDelay > 0) - emit this->nextStartActionChanged(); - this->nextActionLoop->exec(); - return this->playSoundsAndStartTimers(this->nextStartAction); + if(loopExitCode == LoopCancelExit) + return false; + + } while(loopExitCode != LoopManualExit || !this->getIsReadyForNextState()); + } + else { + ScStwSoundPlayer::PlayResult res = this->doDelayAndSoundOfCurrentStartState(); + + if(res == ScStwSoundPlayer::Error) { + qDebug() << "[ERROR][RACE] error playing ready sound"; + this->technicalIncident(); + return false; + } + else if(res == ScStwSoundPlayer::Cancelled) { + return false; + } + } + + // enter starting state + this->setState(STARTING); + + // play start tone + double timeOfSoundPlaybackStart; + res = this->doDelayAndSoundOfCurrentStartState(&timeOfSoundPlaybackStart); + + if(res == ScStwSoundPlayer::Error) { + qDebug() << "[ERROR][RACE] error playing at your marks sound"; + this->technicalIncident(); + return false; + } + else if(res == ScStwSoundPlayer::Cancelled) { + return false; } // perform start // start all timers bool startOk = true; - foreach(ScStwTimer *timer, this->timers){ - if(!timer->start(timeOfSoundPlaybackStart + 3100) && timer->getState() != ScStwTimer::DISABLED){ + foreach(ScStwTimer *timer, this->timers) { + if(!timer->start(timeOfSoundPlaybackStart + 3000) && timer->getState() != ScStwTimer::DISABLED) { startOk = false; } } if(!startOk) { - qDebug() << "[ERROR][START] error staring all timers"; + qDebug() << "[ERROR][RACE] error staring all timers"; + this->technicalIncident(); return false; } - if(!this->soundPlayer->waitForSoundFinish()) { - qDebug() << "[ERROR][START] start sound wait error"; + if(this->soundPlayer->waitForSoundFinish() == ScStwSoundPlayer::Error) { + qDebug() << "[ERROR][RACE] start sound wait error"; + this->technicalIncident(); return false; } // check if a false start occured - if(this->state != STARTING) - return true; - - nextStartAction = None; - emit this->nextStartActionChanged(); + if(!this->isStarting()) + return false; this->setState(RUNNING); @@ -249,6 +381,38 @@ bool ScStwRace::playSoundsAndStartTimers(StartAction thisAction) { } +ScStwSoundPlayer::PlayResult ScStwRace::doDelayAndSoundOfCurrentStartState(double *timeOfSoundPlaybackStart) { + ScStwSoundPlayer::StartSound sound = this->getSoundForState(this->state); + + if(this->getSoundEnabledSetting(sound)) { + if(sound != ScStwSoundPlayer::Start && this->getSoundDelaySetting(sound) > 0) { + // perform the delay before the start + + // get delay + int thisSoundDelay = this->getSoundDelaySetting(sound); + + // perform next action + if(thisSoundDelay > 0) { + this->startDelayTimer->setInterval(thisSoundDelay); + this->startDelayTimer->start(); + + emit this->currentStartDelayChanged(); + + if(this->startWaitLoop->exec() == LoopCancelExit) + return ScStwSoundPlayer::Cancelled; + } + } + + if(!this->isStarting()) + return ScStwSoundPlayer::Error; + + + return this->soundPlayer->play(sound, this->getSoundVolume(), timeOfSoundPlaybackStart); + } + + return ScStwSoundPlayer::Success; +} + void ScStwRace::setState(RaceState newState) { if(newState != this->state) { qDebug() << "[INFO][RACE] state changed: " << newState; @@ -256,106 +420,296 @@ void ScStwRace::setState(RaceState newState) { emit this->stateChanged(newState); if(this->state == IDLE) { - // if we changed to IDLE -> enable timers - foreach(ScStwTimer* timer, this->timerEnableQueque) { - this->handleTimerEnable(timer); - } - - this->timerEnableQueque.clear(); + this->refreshCompetitionMode(); } } } -void ScStwRace::refreshTimerStates() { +void ScStwRace::handleTimerStateChange(ScStwTimer::TimerState newState) { + if( + newState == ScStwTimer::WON || + newState == ScStwTimer::LOST || + newState == ScStwTimer::WILDCARD || + newState == ScStwTimer::FAILED || + this->state == INCIDENT + ) + return; - qDebug() << "[INFO][MAIN] refreshing timer states"; + qDebug() << "[INFO][MAIN] handling timer state change"; // check if the race is over bool raceIsOver = true; + int incidentCount = 0; - foreach(ScStwTimer * timer, this->timers){ - if(timer->getState() < ScStwTimer::WON && timer->getState() != ScStwTimer::WAITING){ + foreach(ScStwTimer * timer, this->timers) { + if(timer->getState() < ScStwTimer::WAITING) { // if the timer is not in stoped state raceIsOver = false; - break; } - else if(timer->getState() == ScStwTimer::WAITING) { + else if(timer->getState() == ScStwTimer::WAITING) this->handleTimerStop(); - } - else if (timer->getState() == ScStwTimer::FAILED) { + else if (timer->getState() == ScStwTimer::FAILING) this->handleFalseStart(); + else if (timer->getState() == ScStwTimer::INCIDENT) + incidentCount ++; + } + + if(incidentCount == this->timers.length()) { + this->setState(INCIDENT); + this->soundPlayer->play(ScStwSoundPlayer::FalseStart, this->getSoundVolume()); + } + else if(raceIsOver) + this->setState(STOPPED); +} + +void ScStwRace::handleTimerStop() { + if(this->state != RUNNING) + return; + + qDebug() << "[INFO][RACE] Got a TIMER STOP"; + + // find out which timer has won + double lowestStoppedTime = -1; + + // iterate through all timers and find the lowest time that was stopped + foreach(ScStwTimer * timer, this->timers) { + qDebug() << "Current stopped time is: " << timer->getCurrentTime(); + if(!timer->isRunning() && !timer->isDisabled() && (timer->getCurrentTime() <= lowestStoppedTime || lowestStoppedTime < 0)) { + // this is the timer with the lowest stopped time + lowestStoppedTime = timer->getCurrentTime(); } } - if(raceIsOver) + qDebug() << "LOWEST Stop time is: " << lowestStoppedTime; + + // append the timer(s) with the lowest stopped time to the winner list + foreach(ScStwTimer * timer, this->timers) { + if(timer->getCurrentTime() <= lowestStoppedTime) + // this is the timer with the lowest stopped time + timer->setResult(ScStwTimer::WON); + else + timer->setResult(ScStwTimer::LOST); + } +} + +void ScStwRace::handleFalseStart() { + if(this->state != STARTING && this->state != RUNNING && this->state != STOPPED) + return; + + qDebug() << "[INFO][RACE] Got a FALSE START"; + + // set lowest to a value that is impossible to reach (start tone is only 3100ms long) + double lowestReactionTime = -4000; + + // iterate through all timers and find the lowest reactiontime that was stopped + foreach(ScStwTimer *timer, this->timers) { + if( + timer->getState() >= ScStwTimer::FAILING && + timer->getState() <= ScStwTimer::FAILED && + !timer->isDisabled() && + ( + timer->getReactionTime() < lowestReactionTime || + lowestReactionTime == -4000 + ) + ) { + lowestReactionTime = timer->getReactionTime(); + } + } + + if(lowestReactionTime == -4000) + // no timer has failed + return; + + qDebug() << "LOWEST reaction time is: " << lowestReactionTime; + + // find out which timer(s) have lost and which get a wildcard + foreach(ScStwTimer * timer, this->timers) { + qDebug() << "THIS TIMERS reaction time is: " << timer->getReactionTime(); + if( + timer->getState() >= ScStwTimer::FAILING && + timer->getState() <= ScStwTimer::FAILED && + !timer->isDisabled() && + timer->getReactionTime() <= lowestReactionTime + ) { + // this is the timer with the lowest stopped time + timer->setResult(ScStwTimer::FAILED); + qDebug() << "FOUND BAD TIMER"; + } + else { + timer->setResult(ScStwTimer::WILDCARD); + } + } + + if(this->state != STOPPED) { this->setState(STOPPED); + this->soundPlayer->cancel(); + this->soundPlayer->play(ScStwSoundPlayer::FalseStart, this->getSoundVolume()); + } +} + +void ScStwRace::technicalIncident() { + + bool raceIsRunning = this->state == RUNNING; + + qDebug() << "[ERROR][RACE] Got a technical incident, state is: " << this->state; + + if(!raceIsRunning) { + this->startWaitLoop->exit(LoopCancelExit); + + if(this->state == STARTING) { + this->soundPlayer->cancel(); + this->soundPlayer->play(ScStwSoundPlayer::FalseStart, this->getSoundVolume()); + } + + this->setState(INCIDENT); + } + + foreach(ScStwTimer *timer, this->timers) { + if(raceIsRunning && timer->getReadyState() == ScStwTimer::ExtensionBatteryIsCritical) + continue; + else if(timer->getReadyState() == ScStwTimer::ExtensionIsNotConnected || timer->getReadyState() == ScStwTimer::ExtensionBatteryIsCritical) + timer->technicalIncident(); + else if(!raceIsRunning) + timer->setState(ScStwTimer::CANCELLED); + } } // ------------------------ // --- helper functions --- // ------------------------ +bool ScStwRace::getIsReadyForNextState() { + if(!this->competitionMode) { + return true; + } + + switch (this->state) { + case IDLE: { + foreach (ScStwTimer *timer, this->timers) { + if(timer->getState() == ScStwTimer::DISABLED) + continue; + + if(timer->getReadyState() == ScStwTimer::ExtensionIsNotConnected || timer->getReadyState() == ScStwTimer::ExtensionBatteryIsCritical) { + + this->technicalIncident(); + + qDebug() << "[ERROR][RACE] Could not start due to not-ready timers"; + + return false; + } + } + + break; + case WAITING: { + foreach (ScStwTimer *timer, this->timers) { + if(timer->getReadyState() != ScStwTimer::IsReady && timer->getReadyState() != ScStwTimer::IsDisabled) + return false; + } + break; + } + + default: + break;; + } + + } + + return true; + +} + /** * @brief ScStwRace::handleTimerEnable function to enable timers at the right moment to prevent them from bricking the state machine * @param {ScStwExtensionControlledTimer*} timer timer to be enabled */ -void ScStwRace::handleTimerEnable(ScStwTimer* timer) { +void ScStwRace::handleTimerWantsToBeDisabledChange(ScStwTimer* timer, bool wantsToBeDisabled) { + //qDebug() << "Handling timer wants to be disabled"; + if(this->competitionMode) + return; + if(this->state == IDLE) { - if(timer->getState() == ScStwTimer::DISABLED) { + //qDebug() << "Handling timer wants to be disabled: " << wantsToBeDisabled; + this->setTimerDisabled(timer, wantsToBeDisabled); + } +} + +void ScStwRace::refreshCompetitionMode() { + if(this->state != IDLE) + return; + + bool currentCompetitionMode = false; + + if(this->settings != nullptr) + currentCompetitionMode = this->settings->readSetting(ScStwSettings::CompetitionModeSetting).toBool(); + + if(this->competitionMode != currentCompetitionMode) { + + qDebug() << "[INFO][RACE] Setting competition mode to " << currentCompetitionMode; + + this->competitionMode = currentCompetitionMode; + } + + if(this->competitionMode) { + foreach (ScStwTimer *timer, this->timers) { timer->setDisabled(false); } - else { - timer->setDisabled(true); - } } else { - this->timerEnableQueque.append(timer); + foreach(ScStwTimer* timer, this->timers) { + if( + (timer->getWantsToBeDisabled() && timer->getState() != ScStwTimer::DISABLED) || + (!timer->getWantsToBeDisabled() && timer->getState() == ScStwTimer::DISABLED) + ) + this->handleTimerWantsToBeDisabledChange(timer, timer->getWantsToBeDisabled()); + } } } -QVariantList ScStwRace::getNextStartActionDetails() { - int nextActionDelay = 0; - double nextActionDelayProg = -1; +QVariantMap ScStwRace::getCurrentStartDelay() { - if(this->nextStartAction == AtYourMarks || this->nextStartAction == Ready) { - // get the total delay and the delay progress of the next action timer - double remaining = this->nextActionTimer->remainingTime(); - nextActionDelay = this->startActionSettings[this->nextStartAction]["Delay"].toInt(); - if(remaining < 0) { - remaining = nextActionDelay; - } - nextActionDelayProg = 1 - (remaining / nextActionDelay); - } - - return { - this->nextStartAction, - nextActionDelay, - nextActionDelayProg + QVariantMap currentStartDelay = { + {"total", -1.0}, + {"progress", -1.0} }; -} -bool ScStwRace::writeStartActionSetting(StartAction action, bool enabled, int delay) { - if(action != AtYourMarks && action != Ready) - return false; + switch (this->state) { - QVariantMap setting = {{"Enabled", enabled}, {"Delay", delay}}; + case WAITING: + if(!this->getSoundEnabledSetting(ScStwSoundPlayer::Ready)) + return currentStartDelay; + if(!this->getIsReadyForNextState()) { + // indicate that we are waiting for climbers and the progress shall be zero + currentStartDelay["progress"] = 0; + return currentStartDelay; + } + break; + case PREPAIRING: { + if(!this->getSoundEnabledSetting(ScStwSoundPlayer::AtYourMarks)) + return currentStartDelay; - if(!this->startActionSettings.contains(action)) - this->startActionSettings.insert(action, setting); - else - this->startActionSettings[action] = setting; - - return true; -} - -bool ScStwRace::setSoundVolume(double volume) { - if(volume >= 0 && volume <= 1) { - this->soundVolume = volume; - return true; + break; } - else { - return false; + + default: + return currentStartDelay; + break; + } + + // get the total delay and the delay progress of the next action timer + double remaining = this->startDelayTimer->remainingTime(); + + if(remaining < 0) + return currentStartDelay; + + currentStartDelay["total"] = this->startDelayTimer->interval(); + currentStartDelay["progress"] = 1 - (remaining / currentStartDelay["total"].toDouble()); + + if(currentStartDelay["progress"].toDouble() < 0) + currentStartDelay["progress"] = 0; + + + return currentStartDelay; } bool ScStwRace::addTimer(ScStwTimer *timer) { @@ -369,10 +723,19 @@ bool ScStwRace::addTimer(ScStwTimer *timer) { this->timers.append(timer); - connect(timer, &ScStwTimer::stateChanged, this, &ScStwRace::refreshTimerStates); - connect(timer, &ScStwTimer::requestEnableChange, this, &ScStwRace::handleTimerEnable); + connect(timer, &ScStwTimer::stateChanged, this, &ScStwRace::handleTimerStateChange); connect(timer, &ScStwTimer::stateChanged, this, &ScStwRace::timersChanged); + connect(timer, &ScStwTimer::wantsToBeDisabledChanged, this, &ScStwRace::handleTimerWantsToBeDisabledChange); connect(timer, &ScStwTimer::reactionTimeChanged, this, &ScStwRace::timersChanged); + connect(timer, &ScStwTimer::readyStateChanged, this, &ScStwRace::handleTimerReadyStateChange); + + connect(timer, &ScStwTimer::readyStateChanged, this, &ScStwRace::isReadyForNextStateChanged); + + if(this->competitionMode && timer->getState() == ScStwTimer::DISABLED) + timer->setDisabled(false); + else if(!this->competitionMode) + this->handleTimerWantsToBeDisabledChange(timer, timer->getWantsToBeDisabled()); + return true; @@ -382,24 +745,169 @@ ScStwRace::RaceState ScStwRace::getState() { return this->state; } -ScStwRace::StartAction ScStwRace::getNextStartAction() -{ - return this->nextStartAction; +QList ScStwRace::getTimers() { + return this->timers; +} + +double ScStwRace::getSoundVolume() { + if(this->settings == nullptr) + return 1; + + return this->settings->readSetting(ScStwSettings::SoundVolumeSetting).toDouble(); +} + +ScStwSoundPlayer::StartSound ScStwRace::getSoundForState(ScStwRace::RaceState state) { + switch (state) { + case PREPAIRING: + return ScStwSoundPlayer::AtYourMarks; + break; + case WAITING: + return ScStwSoundPlayer::Ready; + break; + case STARTING: + return ScStwSoundPlayer::Start; + break; + default: + return ScStwSoundPlayer::StartSound(-1); + } +} + +bool ScStwRace::getSoundEnabledSetting(ScStwSoundPlayer::StartSound sound) { + ScStwSettings::BaseStationSetting soundEnabledSetting; + switch (sound) { + case ScStwSoundPlayer::AtYourMarks: + soundEnabledSetting = ScStwSettings::AtYourMarksSoundEnableSetting; + break; + case ScStwSoundPlayer::Ready: + soundEnabledSetting = ScStwSettings::ReadySoundEnableSetting; + break; + case ScStwSoundPlayer::Start: + return true; + break; + default: + return false; + } + + if(this->settings == nullptr) + return false; + + return this->settings->readSetting(soundEnabledSetting).toBool(); +} + +int ScStwRace::getSoundDelaySetting(ScStwSoundPlayer::StartSound sound) { + ScStwSettings::BaseStationSetting soundDelaySetting; + switch (sound) { + case ScStwSoundPlayer::AtYourMarks: + soundDelaySetting = ScStwSettings::AtYourMarksSoundDelaySetting; + break; + case ScStwSoundPlayer::Ready: + soundDelaySetting = ScStwSettings::ReadySoundDelaySetting; + break; + case ScStwSoundPlayer::Start: + return 0; + break; + default: + return -1; + } + + if(this->settings == nullptr) + return -1; + + return this->settings->readSetting(soundDelaySetting).toInt(); } QVariantList ScStwRace::getTimerDetailList() { QVariantList tmpTimers; - foreach(ScStwTimer * timer, this->timers){ + foreach(ScStwTimer * timer, this->timers) { QVariantMap tmpTimer; tmpTimer.insert("id", this->timers.indexOf(timer)); tmpTimer.insert("state", timer->getState()); tmpTimer.insert("currentTime", timer->getCurrentTime()); tmpTimer.insert("reactionTime", timer->getReactionTime()); - tmpTimer.insert("text", timer->getText()); tmpTimer.insert("letter", timer->getLetter()); + tmpTimer.insert("readyState", timer->getReadyState()); + tmpTimer.insert("text", timer->getText()); tmpTimers.append(tmpTimer); } return tmpTimers; } + +QVariantMap ScStwRace::getDetails() { + QVariantMap tmpDetails; + + tmpDetails.insert("state", this->getState()); + tmpDetails.insert("competitionMode", this->competitionMode); + tmpDetails.insert("readySoundEnabled", this->getSoundEnabledSetting(ScStwSoundPlayer::Ready)); + tmpDetails.insert("currentStartDelay", this->getCurrentStartDelay()); + tmpDetails.insert("timers", this->getTimerDetailList()); + + if(this->state == WAITING) + tmpDetails.insert("isReadyForNextState", this->getIsReadyForNextState()); + else + tmpDetails.insert("isReadyForNextState", true); + + return tmpDetails; +} + +bool ScStwRace::isStarting() { + return this->state == PREPAIRING || this->state == WAITING || this->state == STARTING; +} +bool ScStwRace::getCompetitionMode() { + return this->competitionMode; +} + +bool ScStwRace::getReadySoundEnabled() { + return this->getSoundEnabledSetting(ScStwSoundPlayer::Ready); +} + +ScStwSettings* ScStwRace::getSettings() { + return this->settings; +} + +void ScStwRace::setSettings(ScStwSettings* settings) { + if(settings == this->settings) + return; + + this->settings = settings; + this->refreshCompetitionMode(); + emit this->settingsChanged(); +} + +bool ScStwRace::getAutoRefreshTimerText() { + return this->autoRefreshTimerText; +} + +void ScStwRace::setAutoRefreshTimerText(bool autoRefresh) { + if(autoRefresh == this->autoRefreshTimerText) + return; + + this->autoRefreshTimerText = autoRefresh; + + if(this->autoRefreshTimerText) { + this->timerTextRefreshTimer = new QTimer(this); + this->timerTextRefreshTimer->setInterval(1); + this->timerTextRefreshTimer->connect( + this->timerTextRefreshTimer, + &QTimer::timeout, + [=] { + // refresh timer text + if(this->getState() == ScStwRace::RUNNING) { + emit this->timersChanged(); + } + + // refresh next start action delay progress + if(this->getState() == ScStwRace::WAITING || this->getState() == ScStwRace::PREPAIRING) { + emit this->currentStartDelayChanged(); + } + } + ); + this->timerTextRefreshTimer->start(); + } + else if(this->timerTextRefreshTimer != nullptr) { + this->timerTextRefreshTimer->deleteLater(); + } + + emit this->autoRefreshTimerTextChanged(); +} diff --git a/ScStwLibraries/sources/scstwsetting.cpp b/ScStwLibraries/sources/scstwsetting.cpp new file mode 100644 index 0000000..83e3ed3 --- /dev/null +++ b/ScStwLibraries/sources/scstwsetting.cpp @@ -0,0 +1,59 @@ +/**************************************************************************** + ** ScStw Libraries + ** Copyright (C) 2020 Itsblue development + ** + ** This program is free software: you can redistribute it and/or modify + ** it under the terms of the GNU General Public License as published by + ** the Free Software Foundation, either version 3 of the License, or + ** (at your option) any later version. + ** + ** This program is distributed in the hope that it will be useful, + ** but WITHOUT ANY WARRANTY; without even the implied warranty of + ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + ** GNU General Public License for more details. + ** + ** You should have received a copy of the GNU General Public License + ** along with this program. If not, see . + ****************************************************************************/ + +#include "scstwsetting.h" +#include "scstwsettings.h" +#include + +ScStwSetting::ScStwSetting(int key, int keyLevel, ScStwSettings*scStwSettings, QObject *parent) : QObject(parent) +{ + this->scStwSettings = scStwSettings; + connect(this->scStwSettings, &ScStwSettings::settingChanged, this, &ScStwSetting::handleSettingChange); + this->key = key; + this->keyLevel = keyLevel; + this->handleSettingChange(-1, this->keyLevel, QVariant()); +} + +QVariant ScStwSetting::getValue() { + qDebug() << "Getting setting: " << this->key << " has to reload: " << this->hasToReload; + if(this->hasToReload) { + this->valueCache = this->scStwSettings->readSetting(this->key, this->keyLevel); + this->hasToReload = false; + } + + return this->valueCache; +} + +void ScStwSetting::setValue(QVariant value) { + if(value != this->valueCache) { + this->scStwSettings->writeSetting(this->key, this->keyLevel, value); + } +} + +void ScStwSetting::handleSettingChange(int key, int keyLevel, QVariant value) { + if(keyLevel == this->keyLevel && key == this->key) { + this->valueCache = value; + qDebug() << "value changed!!! key: " << key << " new value " << value; + emit this->valueChanged(); + } + else if(key == -1 && this->key != -1 && this->keyLevel != -1 && this->keyLevel == keyLevel) { + qDebug() << "value changed!!! key: " << key << " new value " << value; + this->hasToReload = true; + emit this->valueChanged(); + } +} diff --git a/ScStwLibraries/sources/scstwsettings.cpp b/ScStwLibraries/sources/scstwsettings.cpp new file mode 100644 index 0000000..d0b21b9 --- /dev/null +++ b/ScStwLibraries/sources/scstwsettings.cpp @@ -0,0 +1,165 @@ +/**************************************************************************** + ** ScStw Libraries + ** Copyright (C) 2020 Itsblue development + ** + ** This program is free software: you can redistribute it and/or modify + ** it under the terms of the GNU General Public License as published by + ** the Free Software Foundation, either version 3 of the License, or + ** (at your option) any later version. + ** + ** This program is distributed in the hope that it will be useful, + ** but WITHOUT ANY WARRANTY; without even the implied warranty of + ** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + ** GNU General Public License for more details. + ** + ** You should have received a copy of the GNU General Public License + ** along with this program. If not, see . + ****************************************************************************/ + +#include "../headers/scstwsettings.h" + +ScStwSettings::ScStwSettings(QObject *parent) : QObject(parent) +{ +#ifdef RASPI + QString path = "/root/.config/ScStwBasestation"; +#else + QString path = QStandardPaths::writableLocation(QStandardPaths::AppConfigLocation); +#endif + QDir dir(path); + if(!dir.mkpath(path)) + qFatal("[FATAL] Failed to create writable directory for settings at %s", qPrintable(path)); + + this->settingsFile = new QFile(path + "/settings.json"); + + if(!this->settingsFile->open(QFile::ReadWrite)) + qFatal("[FATAL] Couldn't open settings file %s", qPrintable(path + "/settings.json")); + + if(!this->loadSettingsFromFile() && this->settingsFile->size() != 0) { + this->writeSettingsToFile(); + qWarning("[WARNING] Settings file (%s) was of invalid format and therefore overwritten!", qPrintable(path + "/settings.json")); + } + + connect(this, &ScStwSettings::settingChanged, this, &ScStwSettings::writeSettingsToFile); + + this->registerKeyLevelConverters(ScStwSettings::KeyLevel, &ScStwSettings::keyToString, &ScStwSettings::keyToType); + + this->setDefaultSetting(ScStwSettings::BaseStationSetting::ReadySoundEnableSetting, false); + this->setDefaultSetting(ScStwSettings::BaseStationSetting::ReadySoundDelaySetting, 0); + this->setDefaultSetting(ScStwSettings::BaseStationSetting::AtYourMarksSoundEnableSetting, false); + this->setDefaultSetting(ScStwSettings::BaseStationSetting::AtYourMarksSoundDelaySetting, 0); + this->setDefaultSetting(ScStwSettings::SoundVolumeSetting, 1); + this->setDefaultSetting(ScStwSettings::CompetitionModeSetting, false); +} + +QVariant ScStwSettings::readSetting(BaseStationSetting key) { + return this->readSetting(key, 0); +} + +QVariant ScStwSettings::readSetting(int key, int keyLevel) { + if(this->keyToStringConverters.contains(keyLevel)) + return this->readSetting(this->keyToStringConverters[keyLevel](key), key, keyLevel); + else + return QVariant(); +} + +bool ScStwSettings::writeSetting(BaseStationSetting key, QVariant value) { + return this->writeSetting(key, 0, value); +} + +bool ScStwSettings::writeSetting(int key, int keyLevel, QVariant value) { + if(this->keyToStringConverters.contains(keyLevel)) + return this->writeSetting(this->keyToStringConverters[keyLevel](key), value, key, keyLevel); + else + return false; +} + +bool ScStwSettings::setDefaultSetting(BaseStationSetting key, QVariant defaultValue) { + return this->setDefaultSetting(key, 0, defaultValue); +} + +bool ScStwSettings::setDefaultSetting(int key, int keyLevel, QVariant defaultValue) { + if(this->keyToStringConverters.contains(keyLevel)) + return this->setDefaultSetting(this->keyToStringConverters[keyLevel](key), defaultValue, key, keyLevel); + else + return false; +} + +QVariant ScStwSettings::readSetting(QString key, int keyInt, int keyLevel) { + Q_UNUSED(keyInt) + Q_UNUSED(keyLevel) + + if(this->settingsCache.contains(key)) + return this->settingsCache[key]; + else + return QVariant(); +} + +bool ScStwSettings::writeSetting(QString key, QVariant value, int keyInt, int keyLevel) { + + // check if the value type is valid + if(!this->keyToTypeConverters.contains(keyLevel) || !value.convert(this->keyToTypeConverters[keyLevel](keyInt)) || value.type() == QVariant::Invalid) { + return false; + } + + // write the setting + if(!this->settingsCache.contains(key)) + this->settingsCache.insert(key, value); + else if (this->settingsCache[key] == value) + return true; + else + this->settingsCache[key] = value; + + emit this->settingChanged(keyInt, keyLevel, value); + + return true; +} + +bool ScStwSettings::setDefaultSetting(QString key, QVariant defaultValue, int keyInt, int keyLevel) { + if(!this->settingsCache.contains(key)) + return this->writeSetting(key, defaultValue, keyInt, keyLevel); + return true; +} + +ScStwSetting * ScStwSettings::getSetting(int key, int keyLevel) { + if(!this->internalSettingHandlers.contains(keyLevel)) + this->internalSettingHandlers.insert(keyLevel, {}); + + if(!this->internalSettingHandlers[keyLevel].contains(key)) + this->internalSettingHandlers[keyLevel].insert(key, new ScStwSetting(key, keyLevel, this, this)); + + return this->internalSettingHandlers[keyLevel][key]; +} + +bool ScStwSettings::registerKeyLevelConverters(int keyLevel, keyToStringConverter keyToStringConverterFunction, keyToTypeConverter keyToTypeConverterFunction) { + if(this->keyToStringConverters.contains(keyLevel)) + return false; + + this->keyToStringConverters.insert(keyLevel, keyToStringConverterFunction); + this->keyToTypeConverters.insert(keyLevel, keyToTypeConverterFunction); + return true; +} + +/* + * File handling + */ + +bool ScStwSettings::writeSettingsToFile() { + QJsonDocument doc = QJsonDocument::fromVariant(this->settingsCache); + + // overwrite file + this->settingsFile->reset(); + this->settingsFile->write(doc.toJson(QJsonDocument::Indented)); + this->settingsFile->flush(); + + return true; +} + +bool ScStwSettings::loadSettingsFromFile() { + QJsonDocument doc = QJsonDocument::fromJson(this->settingsFile->readAll()); + + if(doc.toVariant().type() != QVariant::Map) + return false; + + this->settingsCache = doc.toVariant().toMap(); + return true; +} diff --git a/ScStwLibraries/sources/scstwsoundplayer.cpp b/ScStwLibraries/sources/scstwsoundplayer.cpp index 0c576a5..3b50309 100644 --- a/ScStwLibraries/sources/scstwsoundplayer.cpp +++ b/ScStwLibraries/sources/scstwsoundplayer.cpp @@ -23,45 +23,46 @@ ScStwSoundPlayer::ScStwSoundPlayer(QObject *parent) : QObject(parent) this->waitLoop = new QEventLoop(this); this->waitTimer = new QTimer(this); - this->soundFiles.insert(0, {{"path","qrc:/sound/AtYourMarksSound.wav"}, {"duration", 1000}}); - this->soundFiles.insert(1, {{"path","qrc:/sound/ReadySound.wav"}, {"duration", 570}}); - this->soundFiles.insert(2, {{"path","qrc:/sound/StartsignalSoundExtended.wav"}, {"duration", 3200}}); - this->soundFiles.insert(3, {{"path","qrc:/sound/FalseStartSound.wav"}, {"duration", 2000}}); + this->soundFiles.insert(AtYourMarks, {{"path","qrc:/sound/AtYourMarksSound.wav"}, {"duration", 1000}}); + this->soundFiles.insert(Ready, {{"path","qrc:/sound/ReadySound.wav"}, {"duration", 570}}); + this->soundFiles.insert(Start, {{"path","qrc:/sound/StartsignalSoundExtended.wav"}, {"duration", 3200}}); + this->soundFiles.insert(FalseStart, {{"path","qrc:/sound/FalseStartSound.wav"}, {"duration", 2000}}); - this->soundEffect = new QSoundEffect(this); - this->soundEffect->setLoopCount(1); + this->_initializeSondEffect(); connect(this, &ScStwSoundPlayer::playbackStarted, this->waitLoop, &QEventLoop::quit); connect(this->soundEffect, &QSoundEffect::playingChanged, this->waitLoop, &QEventLoop::quit); } -bool ScStwSoundPlayer::play(int action, double volume, double *timeOfStart) { - if(!this->soundFiles.contains(action)) - return false; - - if(action > 2 || action < 0) - return false; +ScStwSoundPlayer::PlayResult ScStwSoundPlayer::play(ScStwSoundPlayer::StartSound sound, double volume, double *timeOfStart) { + if(!this->soundFiles.contains(sound)) { + qDebug() << "[ERROR][SoundPlayer] Sound file was not found for sound" << sound; + return ScStwSoundPlayer::Error; + } // stop playback if(this->soundEffect->isPlaying()) this->soundEffect->stop(); // update currently playing action - this->currentlyPlayingAction = action; + this->currentlyPlayingSound = sound; - // update volume - this->soundEffect->setVolume(volume); + if(!this->_setSoundVolume(volume)) + return Error; // load - this->soundEffect->setSource(this->soundFiles[action]["path"].toString()); + this->soundEffect->setSource(this->soundFiles[sound]["path"].toString()); // wait for the effect to load QEventLoop loop; while(this->soundEffect->status() != QSoundEffect::Ready) { + qDebug() << "[DEBUG][Sound] Sound is not ready yet, status is: " << this->soundEffect->status(); QObject::connect(this->soundEffect, &QSoundEffect::statusChanged, &loop, &QEventLoop::quit); loop.exec(); } + qDebug() << "[DEBUG][Sound] Playing sound now: " << sound; + // start this->soundEffect->play(); @@ -76,45 +77,34 @@ bool ScStwSoundPlayer::play(int action, double volume, double *timeOfStart) { *timeOfStart = this->playingStartedAt; } - if(action < 2) + if(sound < Start) return this->waitForSoundFinish(); - return true; + return ScStwSoundPlayer::Success; } -bool ScStwSoundPlayer::cancel(double volume) { +bool ScStwSoundPlayer::cancel() { if(!this->soundEffect->isPlaying() ) return false; // stop playback this->soundEffect->stop(); - this->waitLoop->quit(); - - if(this->currentlyPlayingAction != 2) - return true; - - // update volume - this->soundEffect->setVolume(volume); - - // load - this->soundEffect->setSource(this->soundFiles[3]["path"].toString()); - - // play - this->soundEffect->play(); + this->waitLoop->exit(-1); return true; } -bool ScStwSoundPlayer::waitForSoundFinish(double *timeOfStop) { +ScStwSoundPlayer::PlayResult ScStwSoundPlayer::waitForSoundFinish(double *timeOfStop) { if(!this->soundEffect->isPlaying()) - return false; + return ScStwSoundPlayer::Error; // wait until the audio output reports the sound is over - waitLoop->exec(); + if(waitLoop->exec() == -1) + return ScStwSoundPlayer::Cancelled; // wait until the sound is actually over // the timeOffset is the buffer time before the audio started! - int timeOffset = this->soundFiles[this->currentlyPlayingAction]["duration"].toDouble() - (QDateTime::currentMSecsSinceEpoch() - playingStartedAt); + int timeOffset = this->soundFiles[this->currentlyPlayingSound]["duration"].toDouble() - (QDateTime::currentMSecsSinceEpoch() - playingStartedAt); if(timeOffset > 0) { QTimer timer; @@ -124,9 +114,99 @@ bool ScStwSoundPlayer::waitForSoundFinish(double *timeOfStop) { // calculate the point in time where the sound playback actually ended if(timeOfStop != nullptr) { - int latency = this->playingStartedAt + this->soundFiles[this->currentlyPlayingAction]["duration"].toDouble(); + int latency = this->playingStartedAt + this->soundFiles[this->currentlyPlayingSound]["duration"].toDouble(); *timeOfStop = QDateTime::currentMSecsSinceEpoch() - latency; } + return ScStwSoundPlayer::Success; +} + +bool ScStwSoundPlayer::isPlaying() { + return this->soundEffect->isPlaying(); +} + +#ifdef ScStwLibraries_Raspi +void ScStwSoundPlayer::_initializeSondEffect() { + this->_audioOutputDevice = nullptr; + for(QAudioDeviceInfo info : QAudioDeviceInfo::availableDevices(QAudio::AudioOutput)) { + qDebug() << info.deviceName(); + if(info.deviceName().contains("Headphones")) + this->_audioOutputDevice = new QAudioDeviceInfo(info); + } + + if(this->_audioOutputDevice == nullptr) + this->_audioOutputDevice = new QAudioDeviceInfo(QAudioDeviceInfo::defaultOutputDevice()); + + this->soundEffect = new QSoundEffect(*this->_audioOutputDevice, this); + this->soundEffect->setLoopCount(1); +} +#else +void ScStwSoundPlayer::_initializeSondEffect() { + this->_audioOutputDevice = new QAudioDeviceInfo(QAudioDeviceInfo::defaultOutputDevice()); + this->soundEffect = new QSoundEffect(*this->_audioOutputDevice, this); + this->soundEffect->setLoopCount(1); +} +#endif + +#ifdef ScStwLibraries_Raspi +bool ScStwSoundPlayer::_setSoundVolume(double volume) { + + QString cardName = this->_audioOutputDevice->deviceName(); + QStringRef shortCardName = cardName.midRef(cardName.indexOf(QLatin1String("="), 0) + 1); + + int cardIndex = snd_card_get_index(shortCardName.toLocal8Bit().constData()); + QString soundDevice = QString(QLatin1String("hw:%1")).arg(cardIndex); + qDebug() << "[INFO][SoundPlayer] Using audio device: " << soundDevice; + + long min, max; + snd_mixer_t *handle; + snd_mixer_selem_id_t *sid; + snd_mixer_elem_t *elem; + + if(snd_mixer_open(&handle, 0) < 0) { + qDebug() << "[ERROR][SoundPlayer] Could not open mixer"; + return false; + } + if(snd_mixer_selem_register(handle, NULL, NULL) < 0) { + qDebug() << "[ERROR][SoundPlayer] Could not register selem"; + return false; + } + if(snd_mixer_attach(handle, soundDevice.toStdString().c_str()) < 0) { + qDebug() << "[ERROR][SoundPlayer] Could not attach mixer"; + return false; + } + if(snd_mixer_load(handle) < 0) { + qDebug() << "[ERROR][SoundPlayer] Could not load mixer"; + return false; + } + + // set volume for all channels + snd_mixer_selem_id_alloca(&sid); + bool success = false; + for (elem = snd_mixer_first_elem(handle); elem; elem = snd_mixer_elem_next(elem)) { + snd_mixer_selem_get_id(elem, sid); + if (!snd_mixer_selem_is_active(elem)) + continue; + + elem = snd_mixer_find_selem(handle, sid); + if(!elem) + continue; + + snd_mixer_selem_get_playback_volume_range(elem, &min, &max); + snd_mixer_selem_set_playback_volume_all(elem, min + (volume) * (max - min)); + + success = true; + } + + snd_mixer_close(handle); + + this->soundEffect->setVolume(1); + + return success; +} +#else +bool ScStwSoundPlayer::_setSoundVolume(double volume) { + this->soundEffect->setVolume(volume); return true; } +#endif diff --git a/ScStwLibraries/sources/scstwtimer.cpp b/ScStwLibraries/sources/scstwtimer.cpp index 11950e5..f426313 100644 --- a/ScStwLibraries/sources/scstwtimer.cpp +++ b/ScStwLibraries/sources/scstwtimer.cpp @@ -18,18 +18,23 @@ #include "../headers/scstwtimer.h" -ScStwTimer::ScStwTimer(QObject *parent, bool directControlEnabled, QString letter) : QObject(parent) +ScStwTimer::ScStwTimer(QObject* parent) : ScStwTimer("", parent) { - this->directControlEnabled = directControlEnabled; +} +ScStwTimer::ScStwTimer(QString letter, QObject *parent) : QObject(parent) +{ if(letter.length() > 1) this->letter = letter[0]; else this->letter = letter; + //qDebug() << "Timer created with letter: " << letter; + this->startTime = 0; this->stopTime = 0; this->reactionTime = 0; + this->wantsToBeDisabled = false; this->state = IDLE; } @@ -48,7 +53,7 @@ bool ScStwTimer::start(double timeOfStart) { if(timeOfStart - QDateTime::currentMSecsSinceEpoch() > 0) { this->setState(STARTING); - QTimer::singleShot(timeOfStart - QDateTime::currentMSecsSinceEpoch(), [=](){ + QTimer::singleShot(timeOfStart - QDateTime::currentMSecsSinceEpoch(), [=]() { if(this->state == STARTING) this->setState(RUNNING); }); @@ -64,13 +69,27 @@ bool ScStwTimer::start(double timeOfStart) { } } +void ScStwTimer::technicalIncident() { + qDebug() << "[INFO][TIMER] Timer got a technical incident"; + + this->setState(INCIDENT); +} + +bool ScStwTimer::wildcard() { + if(this->state != STARTING) + return false; + + this->setState(WILDCARD); + return true; +} + void ScStwTimer::handleClimberStart(double timeOfStart) { this->reactionTime = timeOfStart - this->startTime; qDebug() << "+ [INFO][TIMER] reaction time: " << this->reactionTime; - if(this->reactionTime <= 0){ - this->stop(FailStop); - return; + if(this->reactionTime <= 100 && this->reactionTime) { + qDebug() << "[INFO][TIMER] False Start detected: " << "start Time: " << startTime << " reactionTime: " << reactionTime; + this->setState(FAILING); } emit this->reactionTimeChanged(); @@ -80,6 +99,11 @@ bool ScStwTimer::cancel() { if(!(this->state == IDLE || this->state == STARTING || this->state == RUNNING)) return false; + qDebug() << "[INFO][TIMER] Timer was cancelled"; + this->reactionTime = 0; + this->startTime = 0; + this->stopTime = 0; + this->setState(CANCELLED); return true; } @@ -89,63 +113,66 @@ bool ScStwTimer::stop() { } bool ScStwTimer::stop(double timeOfStop) { - return this->stop(ManualStop, timeOfStop); -} - -bool ScStwTimer::stop(StopReason reason) { - return this->stop(reason, QDateTime::currentMSecsSinceEpoch()); -} - -bool ScStwTimer::stop(StopReason reason, double timeOfStop) { - if(this->state != STARTING && this->state != RUNNING && this->state != WAITING){ + if(this->state != RUNNING) return false; - } - switch (reason) { - case ManualStop: { - if(this->state == STARTING){ - this->setState(CANCELLED); - } - else { - this->stopTime = timeOfStop; + this->stopTime = timeOfStop; - // trigger an external state refresh to set the state to either WON or LOST depending on the other timers values (see ScStwRace::refreshTimerStates()) - this->setState(WAITING); - } - break; - } - case FailStop: { - qDebug() << "[INFO][TIMER] False Start detected: " << "start Time: " << startTime << " reactionTime: " << reactionTime; - this->setState(FAILED); - break; - } - default: { - return false; - } - } + // trigger an external state refresh to set the state to either WON or LOST depending on the other timers values (see ScStwRace::refreshTimerStates()) + this->setState(WAITING); qDebug() << "[INFO][TIMER] Stopped: " << "start Time: " << startTime << " stopTime: " << stopTime << " stoppedTime: " << this->getCurrentTime() << " reactionTime: " << reactionTime; return true; } bool ScStwTimer::setResult(TimerState result) { - if(this->state != WAITING) + + QList allowedStates = { + STARTING, + RUNNING, + WAITING, + WON, + FAILING, + FAILED + }; + + if(!allowedStates.contains(this->state)) return false; + /* The reasons why it has to accept all these states: + * + * STARTING: To set a timer to WILDCARD when its opponent has done a false start + * RUNNING: To set a timer to WILDCARD when its opponent has done a false start that was received with a delay + * WAITING: To set a timer to either WON or LOST + * WON: To set a timer to LOST when its opponent has won the race but their trigger was delayed + * FAILING: To set a timer to either FAILED or WILDCARD + * FAILED: To set a timer to WILDCARD when its opponent did an earlier false start but their tirgger was delayed + */ + switch (result) { - case WON: - this->setState(WON); - return true; case LOST: - this->setState(LOST); - return true; + case WON: + if(this->state == WAITING || this->state == WON) { + this->setState(result); + return true; + } + break; + case WILDCARD: + case FAILED: + if(this->state == STARTING || this->state == RUNNING || this->state == FAILED || this->state == FAILING) { + this->setState(result); + return true; + } + break; default: - return false; + break; } + + return false; } -bool ScStwTimer::reset(){ - if( this->state < WON || this->state == DISABLED ){ +bool ScStwTimer::reset() { + if( this->state < WON || this->state == DISABLED ) { return false; } @@ -157,61 +184,28 @@ bool ScStwTimer::reset(){ return true; } +ScStwTimer::ReadyState ScStwTimer::getReadyState() { + return this->state == IDLE ? ScStwTimer::IsReady : ScStwTimer::NotInIdleState; +} + // ------------------------ // --- helper functions --- // ------------------------ -bool ScStwTimer::setStartTime(double startTime) { - if(!this->directControlEnabled) - return false; - - this->startTime = startTime; - return true; -} - -bool ScStwTimer::setStopTime(double stopTime) { - if(!this->directControlEnabled) - return false; - - this->stopTime = stopTime; - return true; -} - -bool ScStwTimer::setReactionTime(double reactionTime) { - if(!this->directControlEnabled) - return false; - - this->reactionTime = reactionTime; - return true; -} - -bool ScStwTimer::setState(TimerState newState, bool force) { - if(!this->directControlEnabled || !force) - return false; - - this->setState(newState); - return true; -} - -void ScStwTimer::setState(TimerState newState){ +void ScStwTimer::setState(TimerState newState) { if(this->state == DISABLED && newState != IDLE) return; + if(this->state == INCIDENT && newState != IDLE) + return; + if(this->state != newState) { this->state = newState; - qDebug() << "+ [INFO][TIMER] timer state changed: " << newState; - emit this->stateChanged(); + qDebug() << "[INFO][TIMER] timer state changed: " << newState; + emit this->stateChanged(newState); } } -bool ScStwTimer::setLetter(QString newLetter) { - if(!this->directControlEnabled) - return false; - - this->letter = newLetter; - return true; -} - ScStwTimer::TimerState ScStwTimer::getState() { return this->state; } @@ -238,7 +232,6 @@ QString ScStwTimer::getLetter() { } QString ScStwTimer::getText() { - QString newText = ""; int newTime = 0; switch (this->state) { @@ -248,24 +241,33 @@ QString ScStwTimer::getText() { case ScStwTimer::STARTING: newTime = 0; break; - case ScStwTimer::WAITING: - newText = "please wait..."; - break; case ScStwTimer::RUNNING: newTime = this->getCurrentTime(); break; + case ScStwTimer::WAITING: + newText = "please wait..."; + break; case ScStwTimer::WON: newTime = this->getCurrentTime(); break; case ScStwTimer::LOST: newTime = this->getCurrentTime(); break; + case ScStwTimer::FAILING: + newText = "please wait..."; + break; case ScStwTimer::FAILED: newText = "false start"; break; + case ScStwTimer::WILDCARD: + newText = "wildcard"; + break; case ScStwTimer::CANCELLED: newText = "cancelled"; break; + case ScStwTimer::INCIDENT: + newText = "Technical incident!"; + break; case ScStwTimer::DISABLED: newText = "---"; break; @@ -278,8 +280,37 @@ QString ScStwTimer::getText() { } void ScStwTimer::setDisabled(bool disabled) { + if(this->isDisabled() == disabled) + return; + if(disabled) this->setState(DISABLED); else this->setState(IDLE); + + emit this->readyStateChanged(this->getReadyState()); + +} + +void ScStwTimer::setWantsToBeDisabled(bool wantsToBeDisabled) { + if(this->wantsToBeDisabled == wantsToBeDisabled) + return; + + qDebug() << "Wants to be disabled changed: " << wantsToBeDisabled; + + this->wantsToBeDisabled = wantsToBeDisabled; + + emit this->wantsToBeDisabledChanged(this, wantsToBeDisabled); +} + +bool ScStwTimer::getWantsToBeDisabled() { + return this->wantsToBeDisabled; +} + +bool ScStwTimer::isRunning() { + return this->state == RUNNING; +} + +bool ScStwTimer::isDisabled() { + return this->state == DISABLED; } diff --git a/ScStwLibraries/sources/styling/scstwappthememanager.cpp b/ScStwLibraries/sources/styling/scstwappthememanager.cpp index b8d3145..e2eca13 100644 --- a/ScStwLibraries/sources/styling/scstwappthememanager.cpp +++ b/ScStwLibraries/sources/styling/scstwappthememanager.cpp @@ -116,7 +116,7 @@ ScStwAppThemeManager::ScStwAppThemeManager(QObject *parent) : QObject(parent) {"success", "#6bd43b"}, {"error", "#e03b2f"}, - {"warning", "#e0b928"}, + {"warning", "#d6ae1e"}, {"line", "grey"}, @@ -174,3 +174,7 @@ QString ScStwAppThemeManager::lighter(QString color, double factor) { qcolor.setHsv(h,s,v * factor); return qcolor.name(); } + +QString ScStwAppThemeManager::getThemeName() { + return this->currentTheme->getName(); +} diff --git a/qt-openssl-encryption b/qt-openssl-encryption new file mode 160000 index 0000000..05fbf56 --- /dev/null +++ b/qt-openssl-encryption @@ -0,0 +1 @@ +Subproject commit 05fbf562c3435f3bbff10a68c031ff465e81d337