Implement Power function

This commit is contained in:
Aik-Siong Koh
2025-08-07 20:37:33 -06:00
committed by Chris Hennes
parent 09d6175a2b
commit 1a8fdc32d3
27 changed files with 220 additions and 44 deletions

View File

@@ -8,6 +8,7 @@
#include "FunctionXY.h"
#include "Sum.h"
#include "Constant.h"
using namespace MbD;
@@ -19,6 +20,11 @@ MbD::FunctionXY::FunctionXY(Symsptr base, Symsptr exp) : x(base), y(exp)
{
}
Symsptr MbD::FunctionXY::copyWith(Symsptr argx, Symsptr argy)
{
return Symsptr();
}
void MbD::FunctionXY::arguments(Symsptr args)
{
//args is a Sum with "terms" containing the actual arguments
@@ -28,7 +34,44 @@ void MbD::FunctionXY::arguments(Symsptr args)
y = sum->terms->at(1);
}
Symsptr MbD::FunctionXY::expandUntil(Symsptr sptr, std::shared_ptr<std::unordered_set<Symsptr>> set)
{
auto itr = std::find_if(set->begin(), set->end(), [sptr](Symsptr sym) {return sptr.get() == sym.get(); });
if (itr != set->end()) return sptr;
auto newx = x->expandUntil(x, set);
auto newy = y->expandUntil(y, set);
auto copy = copyWith(newx, newy);
return copy;
}
Symsptr MbD::FunctionXY::simplifyUntil(Symsptr sptr, std::shared_ptr<std::unordered_set<Symsptr>> set)
{
auto itr = std::find_if(set->begin(), set->end(), [sptr](Symsptr sym) {return sptr.get() == sym.get(); });
if (itr != set->end()) return sptr;
auto newx = x->simplifyUntil(x, set);
auto newy = y->simplifyUntil(y, set);
auto copy = copyWith(newx, newy);
return copy;
}
void MbD::FunctionXY::createMbD(std::shared_ptr<System> mbdSys, std::shared_ptr<Units> mbdUnits)
{
x->createMbD(mbdSys, mbdUnits);
y->createMbD(mbdSys, mbdUnits);
}
bool MbD::FunctionXY::isConstant()
{
return x->isConstant() && y->isConstant();
}
Symsptr MbD::FunctionXY::differentiateWRT(Symsptr var)
{
if (this == var.get()) return sptrConstant(1.0);
auto dfdx = differentiateWRTx();
auto dfdy = differentiateWRTy();
auto dxdvar = x->differentiateWRT(var);
auto dydvar = y->differentiateWRT(var);
return Symbolic::sum(Symbolic::times(dfdx, dxdvar), Symbolic::times(dfdy, dydvar));
}