mirror of
https://github.com/UpsilonNumworks/Upsilon.git
synced 2026-01-19 16:57:31 +01:00
94 lines
3.2 KiB
C++
94 lines
3.2 KiB
C++
#include "trigonometric_model.h"
|
|
#include "../../shared/poincare_helpers.h"
|
|
#include "../../poincare/include/poincare_layouts.h"
|
|
#include <math.h>
|
|
#include <poincare/preferences.h>
|
|
#include <assert.h>
|
|
|
|
using namespace Poincare;
|
|
using namespace Shared;
|
|
|
|
namespace Regression {
|
|
|
|
ExpressionLayout * TrigonometricModel::layout() {
|
|
static ExpressionLayout * layout = nullptr;
|
|
if (layout == nullptr) {
|
|
const ExpressionLayout * layoutChildren[] = {
|
|
new CharLayout('a', KDText::FontSize::Small),
|
|
new CharLayout(Ion::Charset::MiddleDot, KDText::FontSize::Small),
|
|
new CharLayout('s', KDText::FontSize::Small),
|
|
new CharLayout('i', KDText::FontSize::Small),
|
|
new CharLayout('n', KDText::FontSize::Small),
|
|
new CharLayout('(', KDText::FontSize::Small),
|
|
new CharLayout('b', KDText::FontSize::Small),
|
|
new CharLayout(Ion::Charset::MiddleDot, KDText::FontSize::Small),
|
|
new CharLayout('X', KDText::FontSize::Small),
|
|
new CharLayout('+', KDText::FontSize::Small),
|
|
new CharLayout('c', KDText::FontSize::Small),
|
|
new CharLayout(')', KDText::FontSize::Small),
|
|
new CharLayout('+', KDText::FontSize::Small),
|
|
new CharLayout('d', KDText::FontSize::Small)
|
|
};
|
|
layout = new HorizontalLayout(layoutChildren, 14, false);
|
|
}
|
|
return layout;
|
|
}
|
|
|
|
Expression * TrigonometricModel::simplifiedExpression(double * modelCoefficients, Poincare::Context * context) {
|
|
double a = modelCoefficients[0];
|
|
double b = modelCoefficients[1];
|
|
double c = modelCoefficients[2];
|
|
double d = modelCoefficients[3];
|
|
Expression * aExpression = new Decimal(a);
|
|
Expression * sinExpression = new Sine(
|
|
new Addition(
|
|
new Multiplication(
|
|
new Decimal(b),
|
|
new Symbol('x'),
|
|
false),
|
|
new Decimal(c),
|
|
false),
|
|
false);
|
|
Expression * asinExpression = new Multiplication(aExpression, sinExpression, false);
|
|
Expression * dExpression = new Decimal(d);
|
|
Expression * result = new Addition(asinExpression, dExpression, false);
|
|
PoincareHelpers::Simplify(&result, *context);
|
|
return result;
|
|
}
|
|
|
|
double TrigonometricModel::evaluate(double * modelCoefficients, double x) const {
|
|
double a = modelCoefficients[0];
|
|
double b = modelCoefficients[1];
|
|
double c = modelCoefficients[2];
|
|
double d = modelCoefficients[3];
|
|
double radianX = Poincare::Preferences::sharedPreferences()->angleUnit() == Poincare::Preferences::AngleUnit::Radian ? x : x * M_PI/180.0;
|
|
return a*sin(b*radianX+c)+d;
|
|
}
|
|
|
|
double TrigonometricModel::partialDerivate(double * modelCoefficients, int derivateCoefficientIndex, double x) const {
|
|
double a = modelCoefficients[0];
|
|
double b = modelCoefficients[1];
|
|
double c = modelCoefficients[2];
|
|
double radianX = Poincare::Preferences::sharedPreferences()->angleUnit() == Poincare::Preferences::AngleUnit::Radian ? x : x * M_PI/180.0;
|
|
if (derivateCoefficientIndex == 0) {
|
|
// Derivate: sin(b*x+c)
|
|
return sin(b*radianX+c);
|
|
}
|
|
if (derivateCoefficientIndex == 1) {
|
|
// Derivate: x*a*cos(b*x+c);
|
|
return radianX*a*cos(b*radianX+c);
|
|
}
|
|
if (derivateCoefficientIndex == 2) {
|
|
// Derivate: a*cos(b*x+c)
|
|
return a*cos(b*radianX+c);
|
|
}
|
|
if (derivateCoefficientIndex == 3) {
|
|
// Derivate: 1
|
|
return 1.0;
|
|
}
|
|
assert(false);
|
|
return 0.0;
|
|
}
|
|
|
|
}
|