#include #include #include #include #include #include #include #include #include #include #include #include "parsing/parser.h" namespace Poincare { bool Expression::sSymbolReplacementsCountLock = false; static bool sApproximationEncounteredComplex = false; /* Constructor & Destructor */ Expression Expression::clone() const { TreeHandle c = TreeHandle::clone(); return static_cast(c); } Expression Expression::Parse(char const * string) { if (string[0] == 0) { return Expression(); } Parser p(string); Expression expression = p.parse(); if (p.getStatus() != Parser::Status::Success) { expression = Expression(); } return expression; } Expression Expression::ExpressionFromAddress(const void * address, size_t size) { if (address == nullptr || size == 0) { return Expression(); } // Build the Expression in the Tree Pool return Expression(static_cast(TreePool::sharedPool()->copyTreeFromAddress(address, size))); } /* Circuit breaker */ static Expression::CircuitBreaker sCircuitBreaker = nullptr; static bool sSimplificationHasBeenInterrupted = false; void Expression::SetCircuitBreaker(CircuitBreaker cb) { sCircuitBreaker = cb; } bool Expression::ShouldStopProcessing() { if (sCircuitBreaker == nullptr) { return false; } if (sCircuitBreaker()) { sSimplificationHasBeenInterrupted = true; return true; } return false; } void Expression::SetInterruption(bool interrupt) { sSimplificationHasBeenInterrupted = interrupt; } /* Hierarchy */ Expression Expression::childAtIndex(int i) const { TreeHandle c = TreeHandle::childAtIndex(i); return static_cast(c); } /* Properties */ bool Expression::isRationalZero() const { return type() == ExpressionNode::Type::Rational && convert().isZero(); } bool Expression::isRationalOne() const { return type() == ExpressionNode::Type::Rational && convert().isOne(); } bool Expression::recursivelyMatches(ExpressionTest test, Context & context, bool replaceSymbols) const { if (test(*this, context)) { return true; } ExpressionNode::Type t = type(); if (replaceSymbols && (t == ExpressionNode::Type::Symbol || t == ExpressionNode::Type::Function)) { return SymbolAbstract::matches(convert(), test, context); } for (int i = 0; i < this->numberOfChildren(); i++) { if (childAtIndex(i).recursivelyMatches(test, context, replaceSymbols)) { return true; } } return false; } bool Expression::IsApproximate(const Expression e, Context & context) { return e.type() == ExpressionNode::Type::Decimal || e.type() == ExpressionNode::Type::Float; } bool Expression::IsRandom(const Expression e, Context & context) { return e.isRandom(); } bool Expression::IsMatrix(const Expression e, Context & context) { return e.type() == ExpressionNode::Type::Matrix || e.type() == ExpressionNode::Type::ConfidenceInterval || e.type() == ExpressionNode::Type::MatrixDimension || e.type() == ExpressionNode::Type::PredictionInterval || e.type() == ExpressionNode::Type::MatrixInverse || e.type() == ExpressionNode::Type::MatrixIdentity || e.type() == ExpressionNode::Type::MatrixTranspose; } bool Expression::IsInfinity(const Expression e, Context & context) { return e.type() == ExpressionNode::Type::Infinity; } bool containsVariables(const Expression e, char * variables, int maxVariableSize) { if (e.type() == ExpressionNode::Type::Symbol) { int index = 0; while (variables[index*maxVariableSize] != 0) { if (strcmp(static_cast(e).name(), &variables[index*maxVariableSize]) == 0) { return true; } index++; } } for (int i = 0; i < e.numberOfChildren(); i++) { if (containsVariables(e.childAtIndex(i), variables, maxVariableSize)) { return true; } } return false; } bool Expression::getLinearCoefficients(char * variables, int maxVariableSize, Expression coefficients[], Expression constant[], Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { assert(!recursivelyMatches(IsMatrix, context, true)); // variables is in fact of type char[k_maxNumberOfVariables][maxVariableSize] int index = 0; while (variables[index*maxVariableSize] != 0) { int degree = polynomialDegree(context, &variables[index*maxVariableSize]); if (degree > 1 || degree < 0) { return false; } index++; } Expression equation = *this; index = 0; Expression polynomialCoefficients[k_maxNumberOfPolynomialCoefficients]; while (variables[index*maxVariableSize] != 0) { int degree = equation.getPolynomialReducedCoefficients(&variables[index*maxVariableSize], polynomialCoefficients, context, complexFormat, angleUnit); switch (degree) { case 0: coefficients[index] = Rational::Builder(0); break; case 1: coefficients[index] = polynomialCoefficients[1]; break; default: /* degree is supposed to be 0 or 1. Otherwise, it means that equation * is 'undefined' due to the reduction of 0*inf for example. * (ie, x*y*inf = 0) */ assert(!recursivelyMatches([](const Expression e, Context & context) { return e.isUndefined(); }, context, true)); return false; } /* The equation is can be written: a_1*x+a_0 with a_1 and a_0 x-independent. * The equation supposed to be linear in all variables, so we can look for * the coefficients linked to the other variables in a_0. */ equation = polynomialCoefficients[0]; index++; } constant[0] = Opposite::Builder(equation.clone()).reduce(context, complexFormat, angleUnit); /* The expression can be linear on all coefficients taken one by one but * non-linear (ex: xy = 2). We delete the results and return false if one of * the coefficients contains a variable. */ bool isMultivariablePolynomial = containsVariables(constant[0], variables, maxVariableSize); for (int i = 0; i < index; i++) { if (isMultivariablePolynomial) { break; } isMultivariablePolynomial |= containsVariables(coefficients[i], variables, maxVariableSize); } return !isMultivariablePolynomial; } // Private void Expression::defaultDeepReduceChildren(Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, ExpressionNode::ReductionTarget target, bool symbolicComputation) { for (int i = 0; i < numberOfChildren(); i++) { childAtIndex(i).deepReduce(context, complexFormat, angleUnit, target, symbolicComputation); } } Expression Expression::defaultShallowReduce() { Expression result; for (int i = 0; i < numberOfChildren(); i++) { // The reduction is shortcutted if one child is unreal or undefined: // - the result is unreal is at least one child is unreal // - the result is undefined is at least one child is undefined but no child is unreal if (childAtIndex(i).type() == ExpressionNode::Type::Unreal) { result = Unreal::Builder(); break; } else if (childAtIndex(i).type() == ExpressionNode::Type::Undefined) { result = Undefined::Builder(); } } if (!result.isUninitialized()) { replaceWithInPlace(result); return result; } return *this; } bool Expression::SimplificationHasBeenInterrupted() { return sSimplificationHasBeenInterrupted; } Expression Expression::parent() const { TreeHandle p = TreeHandle::parent(); return static_cast(p); } void Expression::defaultSetChildrenInPlace(Expression other) { assert(numberOfChildren() == other.numberOfChildren()); for (int i = 0; i < numberOfChildren(); i++) { replaceChildAtIndexInPlace(i, other.childAtIndex(i)); } } bool Expression::hasReplaceableSymbols(Context & context) const { return recursivelyMatches([](const Expression e, Context & context) { return (e.type() == ExpressionNode::Type::Symbol && !static_cast(e).isSystemSymbol() && !context.expressionForSymbol(static_cast(e), false).isUninitialized()) || (e.type() == ExpressionNode::Type::Function && !context.expressionForSymbol(static_cast(e), false).isUninitialized()); }, context, false); } Expression Expression::defaultReplaceReplaceableSymbols(Context & context) { int nbChildren = numberOfChildren(); for (int i = 0; i < nbChildren; i++) { childAtIndex(i).shallowReplaceReplaceableSymbols(context); } return *this; } Expression Expression::makePositiveAnyNegativeNumeralFactor(Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, ExpressionNode::ReductionTarget target) { // The expression is a negative number if (isNumber() && sign(&context) == ExpressionNode::Sign::Negative) { return setSign(ExpressionNode::Sign::Positive, &context, complexFormat, angleUnit, target); } // The expression is a multiplication whose numeral factor is negative if (type() == ExpressionNode::Type::Multiplication && numberOfChildren() > 0 && childAtIndex(0).isNumber() && childAtIndex(0).sign(&context) == ExpressionNode::Sign::Negative) { Multiplication m = convert(); if (m.childAtIndex(0).type() == ExpressionNode::Type::Rational && m.childAtIndex(0).convert().isMinusOne()) { // The negative numeral factor is -1, we just remove it m.removeChildAtIndexInPlace(0); // The multiplication can have only one child after removing -1 return m.squashUnaryHierarchyInPlace(); } else { // Otherwise, we make it positive m.childAtIndex(0).setSign(ExpressionNode::Sign::Positive, &context, complexFormat, angleUnit, target); } return m; } return Expression(); } template Evaluation Expression::approximateToEvaluation(Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { sApproximationEncounteredComplex = false; // Reset interrupting flag because some evaluation methods use it sSimplificationHasBeenInterrupted = false; Evaluation e = node()->approximate(U(), context, complexFormat, angleUnit); if (complexFormat == Preferences::ComplexFormat::Real && sApproximationEncounteredComplex) { e = Complex::Undefined(); } return e; } Expression Expression::defaultReplaceSymbolWithExpression(const SymbolAbstract & symbol, const Expression expression) { for (int i = 0; i < numberOfChildren(); i++) { childAtIndex(i).replaceSymbolWithExpression(symbol, expression); } return *this; } int Expression::defaultGetPolynomialCoefficients(Context & context, const char * symbol, Expression coefficients[]) const { int deg = polynomialDegree(context, symbol); if (deg == 0) { coefficients[0] = clone(); return 0; } return -1; } int Expression::getPolynomialReducedCoefficients(const char * symbolName, Expression coefficients[], Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { // Reset interrupting flag because we use deepReduce int degree = getPolynomialCoefficients(context, symbolName, coefficients); for (int i = 0; i <= degree; i++) { coefficients[i] = coefficients[i].reduce(context, complexFormat, angleUnit); } return degree; } Expression Expression::replaceUnknown(const Symbol & symbol, const Symbol & unknownSymbol) { return node()->replaceUnknown(symbol, unknownSymbol); } Expression Expression::defaultReplaceUnknown(const Symbol & symbol, const Symbol & unknownSymbol) { int nbChildren = numberOfChildren(); for (int i = 0; i < nbChildren; i++) { childAtIndex(i).replaceUnknown(symbol, unknownSymbol); } return *this; } /* Complex */ bool Expression::EncounteredComplex() { return sApproximationEncounteredComplex; } void Expression::SetEncounteredComplex(bool encounterComplex) { sApproximationEncounteredComplex = encounterComplex; } Preferences::ComplexFormat Expression::UpdatedComplexFormatWithTextInput(Preferences::ComplexFormat complexFormat, const char * textInput) { if (complexFormat == Preferences::ComplexFormat::Real && *(UTF8Helper::CodePointSearch(textInput, UCodePointMathematicalBoldSmallI)) != 0) { return Preferences::ComplexFormat::Cartesian; } return complexFormat; } Preferences::ComplexFormat Expression::UpdatedComplexFormatWithExpressionInput(Preferences::ComplexFormat complexFormat, const Expression & exp, Context & context) { if (complexFormat == Preferences::ComplexFormat::Real && exp.recursivelyMatches([](const Expression e, Context & context) { return e.type() == ExpressionNode::Type::Constant && static_cast(e).isIComplex(); }, context, true)) { return Preferences::ComplexFormat::Cartesian; } return complexFormat; } /* Comparison */ bool Expression::isIdenticalTo(const Expression e) const { /* We use the simplification order only because it is a already-coded total * order on expresssions. */ return ExpressionNode::SimplificationOrder(node(), e.node(), true, true) == 0; } bool Expression::isEqualToItsApproximationLayout(Expression approximation, char * buffer, int bufferSize, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, Preferences::PrintFloatMode floatDisplayMode, int numberOfSignificantDigits, Context & context) { approximation.serialize(buffer, bufferSize, floatDisplayMode, numberOfSignificantDigits); /* Warning: we cannot use directly the the approximate expression but we have * to re-serialize it because the number of stored significative * numbers and the number of displayed significative numbers might not be * identical. (For example, 0.000025 might be displayed "0.00003" and stored * as Decimal(0.000025) and isEqualToItsApproximationLayout should return * false) */ Expression approximateOutput = Expression::ParseAndSimplify(buffer, context, complexFormat, angleUnit, true); bool equal = isIdenticalTo(approximateOutput); return equal; } /* Layout Helper */ Layout Expression::createLayout(Preferences::PrintFloatMode floatDisplayMode, int numberOfSignificantDigits) const { return isUninitialized() ? Layout() : node()->createLayout(floatDisplayMode, numberOfSignificantDigits); } int Expression::serialize(char * buffer, int bufferSize, Preferences::PrintFloatMode floatDisplayMode, int numberOfSignificantDigits) const { return isUninitialized() ? 0 : node()->serialize(buffer, bufferSize, floatDisplayMode, numberOfSignificantDigits); } /* Simplification */ Expression Expression::ParseAndSimplify(const char * text, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, bool symbolicSimplification) { Expression exp = Parse(text); if (exp.isUninitialized()) { return Undefined::Builder(); } exp = exp.simplify(context, complexFormat, angleUnit, symbolicSimplification); /* simplify might have been interrupted, in which case the resulting * expression is uninitialized, so we need to check that. */ if (exp.isUninitialized()) { return Parse(text); } return exp; } void Expression::ParseAndSimplifyAndApproximate(const char * text, Expression * simplifiedExpression, Expression * approximateExpression, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, bool symbolicComputation) { assert(simplifiedExpression); Expression exp = Parse(text); if (exp.isUninitialized()) { *simplifiedExpression = Undefined::Builder(); *approximateExpression = Undefined::Builder(); return; } exp.simplifyAndApproximate(simplifiedExpression, approximateExpression, context, complexFormat, angleUnit, symbolicComputation); /* simplify might have been interrupted, in which case the resulting * expression is uninitialized, so we need to check that. */ if (simplifiedExpression->isUninitialized()) { *simplifiedExpression = Parse(text); if (approximateExpression) { *approximateExpression = simplifiedExpression->approximate(context, complexFormat, angleUnit); } } } Expression Expression::simplify(Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, bool symbolicComputation) { sSimplificationHasBeenInterrupted = false; Expression e = deepReduce(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::System, symbolicComputation); if (!sSimplificationHasBeenInterrupted) { e = e.deepBeautify(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::System); } return sSimplificationHasBeenInterrupted ? Expression() : e; } void makePositive(Expression * e, bool * isNegative) { if (e->type() == ExpressionNode::Type::Opposite) { *isNegative = true; *e = e->childAtIndex(0); } } void Expression::simplifyAndApproximate(Expression * simplifiedExpression, Expression * approximateExpression, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, bool symbolicComputation) { assert(simplifiedExpression); sSimplificationHasBeenInterrupted = false; // Step 1: we reduce the expression Expression e = clone().deepReduce(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User, symbolicComputation); if (sSimplificationHasBeenInterrupted) { sSimplificationHasBeenInterrupted = false; e = deepReduce(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::System, symbolicComputation); } *simplifiedExpression = Expression(); if (sSimplificationHasBeenInterrupted) { return; } /* Case 1: the reduced expression is ComplexCartesian or pure real, we can * take into account the complex format to display a+i*b or r*e^(i*th) */ if (e.type() == ExpressionNode::Type::ComplexCartesian || e.isReal(context)) { ComplexCartesian ecomplex = e.type() == ExpressionNode::Type::ComplexCartesian ? static_cast(e) : ComplexCartesian::Builder(e, Rational::Builder(0)); if (approximateExpression) { /* Step 2: Approximation * We compute the approximate expression from the Cartesian form to avoid * unprecision. For example, if the result is the ComplexCartesian(a,b), * the final expression is goind to be sqrt(a^2+b^2)*exp(i*arctan(b/a)... * in Polar ComplexFormat. If we approximate this expression instead of * ComplexCartesian(a,b), we are going to loose precision on the resulting * complex.*/ // Clone the ComplexCartesian to use it to compute the approximation ComplexCartesian ecomplexClone = ecomplex.clone().convert(); // To minimize the error on the approximation, we reduce the number of nodes in the expression by beautifying ecomplexClone.real().deepBeautify(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User); ecomplexClone.imag().deepBeautify(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User); *approximateExpression = ecomplexClone.approximate(context, complexFormat, angleUnit); } // Step 3: create the simplied expression with the required complex format Expression ra = complexFormat == Preferences::ComplexFormat::Polar ? ecomplex.clone().convert().norm(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User).shallowReduce(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User, symbolicComputation) : ecomplex.real(); Expression tb = complexFormat == Preferences::ComplexFormat::Polar ? ecomplex.argument(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User).shallowReduce(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User, symbolicComputation) : ecomplex.imag(); ra = ra.deepBeautify(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User); tb = tb.deepBeautify(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User); bool raIsNegative = false; bool tbIsNegative = false; makePositive(&ra, &raIsNegative); makePositive(&tb, &tbIsNegative); *simplifiedExpression = CreateComplexExpression(ra, tb, complexFormat, ra.isUndefined() || tb.isUndefined(), IsZero(ra), IsOne(ra), IsZero(tb), IsOne(tb), raIsNegative, tbIsNegative); } else { /* Case 2: The reduced expression has a complex component that could not * be bubbled up. */ *simplifiedExpression = e.deepBeautify(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::User); if (approximateExpression) { *approximateExpression = simplifiedExpression->approximate(context, complexFormat, angleUnit); } } } Expression Expression::ExpressionWithoutSymbols(Expression e, Context & context) { if (e.isUninitialized()) { return e; } /* Replace all the symbols iteratively. If we make too many replacements, the * symbols are likely to be defined circularly, in which case we return an * uninitialized expression. * We need a static "replacement count" to aggregate all calls to * ExpressionWithoutSymbols, as this method might be called from * hasReplaceableSymbols. */ static int replacementCount = 0; bool unlock = false; if (!sSymbolReplacementsCountLock) { replacementCount = 0; sSymbolReplacementsCountLock = true; unlock = true; } while (e.hasReplaceableSymbols(context)) { replacementCount++; if (replacementCount > k_maxSymbolReplacementsCount) { e = Expression(); break; } e = e.shallowReplaceReplaceableSymbols(context); } if (unlock) { sSymbolReplacementsCountLock = false; } return e; } Expression Expression::radianToDegree() { // e*180/Pi return Multiplication::Builder(*this, Rational::Builder(180), Power::Builder(Constant::Builder(UCodePointGreekSmallLetterPi), Rational::Builder(-1))); } Expression Expression::degreeToRadian() { // e*Pi/180 return Multiplication::Builder(*this, Rational::Builder(1, 180), Constant::Builder(UCodePointGreekSmallLetterPi)); } Expression Expression::reduce(Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) { sSimplificationHasBeenInterrupted = false; return deepReduce(context, complexFormat, angleUnit, ExpressionNode::ReductionTarget::System, true); } Expression Expression::deepReduce(Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, ExpressionNode::ReductionTarget target, bool symbolicComputation) { #if MATRIX_EXACT_REDUCING #else #if 0 if (IsMatrix(*this, context)) { sSimplificationHasBeenInterrupted = true; return *this; } #endif #endif deepReduceChildren(context, complexFormat, angleUnit, target, symbolicComputation); if (sSimplificationHasBeenInterrupted) { return *this; } return shallowReduce(context, complexFormat, angleUnit, target, symbolicComputation); } Expression Expression::deepBeautify(Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, ExpressionNode::ReductionTarget target) { Expression e = shallowBeautify(context, complexFormat, angleUnit, target); int nbChildren = e.numberOfChildren(); for (int i = 0; i < nbChildren; i++) { e.childAtIndex(i).deepBeautify(context, complexFormat, angleUnit, target); } return e; } Expression Expression::setSign(ExpressionNode::Sign s, Context * context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, ExpressionNode::ReductionTarget target) { assert(s == ExpressionNode::Sign::Positive || s == ExpressionNode::Sign::Negative); return node()->setSign(s, context, complexFormat, angleUnit, target); } /* Evaluation */ template Expression Expression::approximate(Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { return isUninitialized() ? Undefined::Builder() : approximateToEvaluation(context, complexFormat, angleUnit).complexToExpression(complexFormat); } template U Expression::approximateToScalar(Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { return approximateToEvaluation(context, complexFormat, angleUnit).toScalar(); } template U Expression::ApproximateToScalar(const char * text, Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, bool symbolicSimplification) { Expression exp = ParseAndSimplify(text, context, complexFormat, angleUnit, symbolicSimplification); assert(!exp.isUninitialized()); return exp.approximateToScalar(context, complexFormat, angleUnit); } template U Expression::approximateWithValueForSymbol(const char * symbol, U x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { VariableContext variableContext = VariableContext(symbol, &context); variableContext.setApproximationForVariable(x); return approximateToScalar(variableContext, complexFormat, angleUnit); } template U Expression::Epsilon() { static U epsilon = sizeof(U) == sizeof(double) ? 1E-15 : 1E-7f; return epsilon; } /* Builder */ bool Expression::IsZero(const Expression e) { return e.type() == ExpressionNode::Type::Rational && static_cast(e).isZero(); } bool Expression::IsOne(const Expression e) { return e.type() == ExpressionNode::Type::Rational && static_cast(e).isOne(); } bool Expression::IsMinusOne(const Expression e) { return e.type() == ExpressionNode::Type::Rational && static_cast(e).isMinusOne(); } Expression Expression::CreateComplexExpression(Expression ra, Expression tb, Preferences::ComplexFormat complexFormat, bool undefined, bool isZeroRa, bool isOneRa, bool isZeroTb, bool isOneTb, bool isNegativeRa, bool isNegativeTb) { if (undefined) { return Undefined::Builder(); } switch (complexFormat) { case Preferences::ComplexFormat::Real: case Preferences::ComplexFormat::Cartesian: { Expression real; Expression imag; if (!isZeroRa || isZeroTb) { if (isNegativeRa) { real = Opposite::Builder(ra); } else { real = ra; } } if (!isZeroTb) { if (isOneTb) { imag = Constant::Builder(UCodePointMathematicalBoldSmallI); } else { imag = Multiplication::Builder(tb , Constant::Builder(UCodePointMathematicalBoldSmallI)); } } if (imag.isUninitialized()) { return real; } else if (real.isUninitialized()) { if (isNegativeTb) { return Opposite::Builder(imag); } else { return imag; } } else if (isNegativeTb) { return Subtraction::Builder(real, imag); } else { return Addition::Builder(real, imag); } } default: { assert(complexFormat == Preferences::ComplexFormat::Polar); Expression norm; Expression exp; if (!isOneRa || isZeroTb) { assert(!isNegativeRa); // norm cannot be negative norm = ra; } if (!isZeroRa && !isZeroTb) { Expression arg; if (isOneTb) { arg = Constant::Builder(UCodePointMathematicalBoldSmallI); } else { arg = Multiplication::Builder(tb, Constant::Builder(UCodePointMathematicalBoldSmallI)); } if (isNegativeTb) { arg = Opposite::Builder(arg); } exp = Power::Builder(Constant::Builder(UCodePointScriptSmallE), arg); } if (exp.isUninitialized()) { return norm; } else if (norm.isUninitialized()) { return exp; } else { return Multiplication::Builder(norm, exp); } } } } /* Expression roots/extrema solver*/ typename Expression::Coordinate2D Expression::nextMinimum(const char * symbol, double start, double step, double max, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { return nextMinimumOfExpression(symbol, start, step, max, [](const char * symbol, double x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression0, const Expression expression1 = Expression()) { return expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); }, context, complexFormat, angleUnit); } typename Expression::Coordinate2D Expression::nextMaximum(const char * symbol, double start, double step, double max, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { Coordinate2D minimumOfOpposite = nextMinimumOfExpression(symbol, start, step, max, [](const char * symbol, double x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression0, const Expression expression1 = Expression()) { return -expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); }, context, complexFormat, angleUnit); return {.abscissa = minimumOfOpposite.abscissa, .value = -minimumOfOpposite.value}; } double Expression::nextRoot(const char * symbol, double start, double step, double max, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const { return nextIntersectionWithExpression(symbol, start, step, max, [](const char * symbol, double x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression0, const Expression expression1 = Expression()) { return expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); }, context, complexFormat, angleUnit, nullptr); } typename Expression::Coordinate2D Expression::nextIntersection(const char * symbol, double start, double step, double max, Poincare::Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression) const { double resultAbscissa = nextIntersectionWithExpression(symbol, start, step, max, [](const char * symbol, double x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression0, const Expression expression1) { return expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit)-expression1.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); }, context, complexFormat, angleUnit, expression); typename Expression::Coordinate2D result = {.abscissa = resultAbscissa, .value = approximateWithValueForSymbol(symbol, resultAbscissa, context, complexFormat, angleUnit)}; if (std::fabs(result.value) < step*k_solverPrecision) { result.value = 0.0; } return result; } typename Expression::Coordinate2D Expression::nextMinimumOfExpression(const char * symbol, double start, double step, double max, EvaluationAtAbscissa evaluate, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression, bool lookForRootMinimum) const { Coordinate2D result = {.abscissa = NAN, .value = NAN}; if (start == max || step == 0.0) { return result; } double bracket[3]; double x = start; bool endCondition = false; do { bracketMinimum(symbol, x, step, max, bracket, evaluate, context, complexFormat, angleUnit, expression); result = brentMinimum(symbol, bracket[0], bracket[2], evaluate, context, complexFormat, angleUnit, expression); x = bracket[1]; // Because of float approximation, exact zero is never reached if (std::fabs(result.abscissa) < std::fabs(step)*k_solverPrecision) { result.abscissa = 0; result.value = evaluate(symbol, 0, context, complexFormat, angleUnit, *this, expression); } /* Ignore extremum whose value is undefined or too big because they are * really unlikely to be local extremum. */ if (std::isnan(result.value) || std::fabs(result.value) > k_maxFloat) { result.abscissa = NAN; } // Idem, exact 0 never reached if (std::fabs(result.value) < std::fabs(step)*k_solverPrecision) { result.value = 0; } endCondition = std::isnan(result.abscissa) && (step > 0.0 ? x <= max : x >= max); if (lookForRootMinimum) { endCondition |= std::fabs(result.value) > 0 && (step > 0.0 ? x <= max : x >= max); } } while (endCondition); if (lookForRootMinimum) { result.abscissa = std::fabs(result.value) > 0 ? NAN : result.abscissa; } return result; } void Expression::bracketMinimum(const char * symbol, double start, double step, double max, double result[3], EvaluationAtAbscissa evaluate, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression) const { Coordinate2D p[3]; p[0] = {.abscissa = start, .value = evaluate(symbol, start, context, complexFormat, angleUnit, *this, expression)}; p[1] = {.abscissa = start+step, .value = evaluate(symbol, start+step, context, complexFormat, angleUnit, *this, expression)}; double x = start+2.0*step; while (step > 0.0 ? x <= max : x >= max) { p[2] = {.abscissa = x, .value = evaluate(symbol, x, context, complexFormat, angleUnit, *this, expression)}; if ((p[0].value > p[1].value || std::isnan(p[0].value)) && (p[2].value > p[1].value || std::isnan(p[2].value)) && (!std::isnan(p[0].value) || !std::isnan(p[2].value))) { result[0] = p[0].abscissa; result[1] = p[1].abscissa; result[2] = p[2].abscissa; return; } if (p[0].value > p[1].value && p[1].value == p[2].value) { } else { p[0] = p[1]; p[1] = p[2]; } x += step; } result[0] = NAN; result[1] = NAN; result[2] = NAN; } typename Expression::Coordinate2D Expression::brentMinimum(const char * symbol, double ax, double bx, EvaluationAtAbscissa evaluate, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression) const { /* Bibliography: R. P. Brent, Algorithms for finding zeros and extrema of * functions without calculating derivatives */ if (ax > bx) { return brentMinimum(symbol, bx, ax, evaluate, context, complexFormat, angleUnit, expression); } double e = 0.0; double a = ax; double b = bx; double x = a+k_goldenRatio*(b-a); double v = x; double w = x; double fx = evaluate(symbol, x, context, complexFormat, angleUnit, *this, expression); double fw = fx; double fv = fw; double d = NAN; double u, fu; for (int i = 0; i < 100; i++) { double m = 0.5*(a+b); double tol1 = k_sqrtEps*std::fabs(x)+1E-10; double tol2 = 2.0*tol1; if (std::fabs(x-m) <= tol2-0.5*(b-a)) { double middleFax = evaluate(symbol, (x+a)/2.0, context, complexFormat, angleUnit, *this, expression); double middleFbx = evaluate(symbol, (x+b)/2.0, context, complexFormat, angleUnit, *this, expression); double fa = evaluate(symbol, a, context, complexFormat, angleUnit, *this, expression); double fb = evaluate(symbol, b, context, complexFormat, angleUnit, *this, expression); if (middleFax-fa <= k_sqrtEps && fx-middleFax <= k_sqrtEps && fx-middleFbx <= k_sqrtEps && middleFbx-fb <= k_sqrtEps) { Coordinate2D result = {.abscissa = x, .value = fx}; return result; } } double p = 0; double q = 0; double r = 0; if (std::fabs(e) > tol1) { r = (x-w)*(fx-fv); q = (x-v)*(fx-fw); p = (x-v)*q -(x-w)*r; q = 2.0*(q-r); if (q>0.0) { p = -p; } else { q = -q; } r = e; e = d; } if (std::fabs(p) < std::fabs(0.5*q*r) && p= tol1 ? d : (d>0 ? tol1 : -tol1)); fu = evaluate(symbol, u, context, complexFormat, angleUnit, *this, expression); if (fu <= fx) { if (u 0.0 ? x <= max : x >= max)); double extremumMax = std::isnan(result) ? max : result; Coordinate2D resultExtremum[2] = { nextMinimumOfExpression(symbol, start, step, extremumMax, [](const char * symbol, double x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression0, const Expression expression1) { if (expression1.isUninitialized()) { return expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); } else { return expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit)-expression1.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); } }, context, complexFormat, angleUnit, expression, true), nextMinimumOfExpression(symbol, start, step, extremumMax, [](const char * symbol, double x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression0, const Expression expression1) { if (expression1.isUninitialized()) { return -expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); } else { return expression1.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit)-expression0.approximateWithValueForSymbol(symbol, x, context, complexFormat, angleUnit); } }, context, complexFormat, angleUnit, expression, true)}; for (int i = 0; i < 2; i++) { if (!std::isnan(resultExtremum[i].abscissa) && (std::isnan(result) || std::fabs(result - start) > std::fabs(resultExtremum[i].abscissa - start))) { result = resultExtremum[i].abscissa; } } if (std::fabs(result) < std::fabs(step)*k_solverPrecision) { result = 0; } return result; } void Expression::bracketRoot(const char * symbol, double start, double step, double max, double result[2], EvaluationAtAbscissa evaluation, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression) const { double a = start; double b = start+step; while (step > 0.0 ? b <= max : b >= max) { double fa = evaluation(symbol, a, context, complexFormat, angleUnit, *this, expression); double fb = evaluation(symbol, b, context, complexFormat, angleUnit,* this, expression); if (fa*fb <= 0) { result[0] = a; result[1] = b; return; } a = b; b = b+step; } result[0] = NAN; result[1] = NAN; } double Expression::brentRoot(const char * symbol, double ax, double bx, double precision, EvaluationAtAbscissa evaluation, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, const Expression expression) const { if (ax > bx) { return brentRoot(symbol, bx, ax, precision, evaluation, context, complexFormat, angleUnit, expression); } double a = ax; double b = bx; double c = bx; double d = b-a; double e = b-a; double fa = evaluation(symbol, a, context, complexFormat, angleUnit, *this, expression); double fb = evaluation(symbol, b, context, complexFormat, angleUnit, *this, expression); double fc = fb; for (int i = 0; i < 100; i++) { if ((fb > 0.0 && fc > 0.0) || (fb < 0.0 && fc < 0.0)) { c = a; fc = fa; e = b-a; d = b-a; } if (std::fabs(fc) < std::fabs(fb)) { a = b; b = c; c = a; fa = fb; fb = fc; fc = fa; } double tol1 = 2.0*DBL_EPSILON*std::fabs(b)+0.5*precision; double xm = 0.5*(c-b); if (std::fabs(xm) <= tol1 || fb == 0.0) { double fbcMiddle = evaluation(symbol, 0.5*(b+c), context, complexFormat, angleUnit, *this, expression); double isContinuous = (fb <= fbcMiddle && fbcMiddle <= fc) || (fc <= fbcMiddle && fbcMiddle <= fb); if (isContinuous) { return b; } } if (std::fabs(e) >= tol1 && std::fabs(fa) > std::fabs(b)) { double s = fb/fa; double p = 2.0*xm*s; double q = 1.0-s; if (a != c) { q = fa/fc; double r = fb/fc; p = s*(2.0*xm*q*(q-r)-(b-a)*(r-1.0)); q = (q-1.0)*(r-1.0)*(s-1.0); } q = p > 0.0 ? -q : q; p = std::fabs(p); double min1 = 3.0*xm*q-std::fabs(tol1*q); double min2 = std::fabs(e*q); if (2.0*p < (min1 < min2 ? min1 : min2)) { e = d; d = p/q; } else { d = xm; e =d; } } else { d = xm; e = d; } a = b; fa = fb; if (std::fabs(d) > tol1) { b += d; } else { b += xm > 0.0 ? tol1 : tol1; } fb = evaluation(symbol, b, context, complexFormat, angleUnit, *this, expression); } return NAN; } template float Expression::Epsilon(); template double Expression::Epsilon(); template Expression Expression::approximate(Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const; template Expression Expression::approximate(Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const; template float Expression::approximateToScalar(Context& context, Preferences::ComplexFormat, Preferences::AngleUnit angleUnit) const; template double Expression::approximateToScalar(Context& context, Preferences::ComplexFormat, Preferences::AngleUnit angleUnit) const; template float Expression::ApproximateToScalar(const char * text, Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, bool symbolicComputation); template double Expression::ApproximateToScalar(const char * text, Context& context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit, bool symbolicComputation); template Evaluation Expression::approximateToEvaluation(Context& context, Preferences::ComplexFormat, Preferences::AngleUnit angleUnit) const; template Evaluation Expression::approximateToEvaluation(Context& context, Preferences::ComplexFormat, Preferences::AngleUnit angleUnit) const; template float Expression::approximateWithValueForSymbol(const char * symbol, float x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const; template double Expression::approximateWithValueForSymbol(const char * symbol, double x, Context & context, Preferences::ComplexFormat complexFormat, Preferences::AngleUnit angleUnit) const; }