-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
moved deriv vector stuff in own file
- Loading branch information
1 parent
d1a48af
commit 4b83a61
Showing
3 changed files
with
127 additions
and
222 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,126 @@ | ||
| #ifndef DERIV_ETR_H | ||
| #define DERIV_ETR_H | ||
|
|
||
| #include "VectorClass.hpp" | ||
| #include <type_traits> | ||
|
|
||
| template <typename LT, typename RT, typename LDeriv, typename RDeriv, | ||
| typename TraitTD, typename OpTrait> | ||
| static constexpr QuarternyType<LT, RT, LDeriv, RDeriv, TraitTD, OpTrait> | ||
| produceQuarternyType() { | ||
| return QuarternyType<LT, RT, LDeriv, RDeriv, TraitTD, OpTrait>(); | ||
| } | ||
|
|
||
| template <typename LDeriv, typename RDeriv, typename TraitTD, typename OpTrait> | ||
| static constexpr BinaryType<LDeriv, RDeriv, TraitTD, OpTrait> | ||
| produceBinaryType() { | ||
| return BinaryType<LDeriv, RDeriv, TraitTD, OpTrait>(); | ||
| } | ||
|
|
||
| template <typename I, typename TraitTD, typename OpTrait> | ||
| static constexpr UnaryType<I, TraitTD, OpTrait> produceUnaryType() { | ||
| return UnaryType<I, TraitTD, OpTrait>(); // issue: wrong wrong wrong | ||
| } | ||
|
|
||
| template <typename TRaw> static constexpr auto produceVariableType() { | ||
| using TD = std::remove_reference_t<TRaw>; | ||
| if constexpr (IsVec<TD>) { | ||
| using tD = ExtractedTypeD<TD>; | ||
| return VariableType<tD>(); | ||
| } else { | ||
| return VariableType<TRaw>(); | ||
| } | ||
| } | ||
|
|
||
| template <typename TD> | ||
| requires IsVarPointer<TD> | ||
| static constexpr auto walkTD() -> VariableType<TD> { | ||
| static_assert(!IsVec<TD>, "Cannot create derivative expression tree"); | ||
| return VariableType<TD>(); | ||
| } | ||
|
|
||
| template <typename TD> | ||
| requires IsConstant<TD> | ||
| static constexpr auto walkTD() { | ||
| return Constants<typename TD::Type, TD::I, ConstantTypeTrait>{}; | ||
| } | ||
|
|
||
| template <typename TRaw> | ||
| requires IsMultiplication<TRaw> | ||
| static constexpr auto walkTD() { | ||
| using TDMultiplication = std::remove_const_t<std::remove_reference_t<TRaw>>; | ||
| constexpr auto LT = | ||
| produceVariableType<typename TDMultiplication::typeTraitL>(); | ||
| constexpr auto RT = | ||
| produceVariableType<typename TDMultiplication::typeTraitR>(); | ||
| constexpr auto LDeriv = walkTD<typename TDMultiplication::typeTraitL>(); | ||
| constexpr auto RDeriv = walkTD<typename TDMultiplication::typeTraitR>(); | ||
| return produceQuarternyType<decltype(LT), decltype(RT), decltype(LDeriv), | ||
| decltype(RDeriv), QuarternaryTrait, | ||
| TimesDerivTrait>(); | ||
| } | ||
|
|
||
| template <typename TD> | ||
| requires IsAddition<TD> | ||
| static constexpr auto walkTD() { | ||
| constexpr auto LDeriv = walkTD<typename TD::typeTraitL>(); | ||
| constexpr auto RDeriv = walkTD<typename TD::typeTraitR>(); | ||
| return produceBinaryType<decltype(LDeriv), decltype(RDeriv), BinaryTrait, | ||
| PlusDerivTrait>(); | ||
| } | ||
|
|
||
| template <typename TD> | ||
| requires IsSinus<TD> | ||
| static constexpr auto walkTD() { | ||
| constexpr auto obj = walkTD<typename TD::typeTraitObj>(); | ||
| return produceUnaryType<decltype(obj), UnaryTrait, SinusDerivTrait>(); | ||
| } | ||
|
|
||
| template <typename TD> | ||
| requires(IsVec<TD> && !IsVariable<TD>) | ||
| static constexpr auto walkTD() { | ||
| using TDVec = ExtractedTypeD<TD>; | ||
| constexpr auto res = walkTD<TDVec>(); | ||
| return res; | ||
| } | ||
|
|
||
| // TODO: finish this | ||
| template <typename T2, typename R2, typename Trait2> | ||
| Vec &operator<<(const Vec<T2, R2, Trait2> &otherVec) { | ||
| using tD = decltype(otherVec.d); | ||
| if constexpr (IsVarPointer<tD> && IsVarPointer<DType>) { | ||
| d.resize(otherVec.size()); | ||
| for (std::size_t i = 0; i < d.size(); i++) { | ||
| d.setVal(otherVec.d.AllVarsRef, i, tD::getVal(otherVec.d.AllVarsRef, i)); | ||
| } | ||
| } else if constexpr (IsVarPointer<tD> && !IsVarPointer<DType>) { | ||
| d.resize(otherVec.size()); | ||
| for (std::size_t i = 0; i < d.size(); i++) { | ||
| d[i] = otherVec[i]; | ||
| } | ||
| } else if constexpr (!IsVarPointer<tD> && IsVarPointer<DType>) { | ||
| using tDRaw = std::remove_reference<decltype(otherVec)>::type; | ||
| using typeExpr = std::remove_reference<ExtractedTypeD<tDRaw>>::type; | ||
| /* printTAST<typeExpr>(); */ | ||
| constexpr auto res = walkTD<typeExpr>(); | ||
| /* printTAST<decltype(res)>(); */ | ||
| d.resize(otherVec.size()); | ||
| for (std::size_t i = 0; i < otherVec.size(); i++) { | ||
| d.AllVarsRef.varBuffer[d.I][i] = otherVec[i]; | ||
| } | ||
|
|
||
| d.AllVarsRef.resizeDerivs(R::I, R::TIdx, otherVec.size()); | ||
| for (std::size_t i = 0; i < res.getSize(d.AllVarsRef); i++) { | ||
| d.setDeriv(d.AllVarsRef, i, res.getDeriv(d.AllVarsRef, i)); | ||
| } | ||
|
|
||
| } else if constexpr (!IsVarPointer<tD> && !IsVarPointer<DType>) { | ||
| d.resize(otherVec.size()); | ||
| for (std::size_t i = 0; i < otherVec.size(); i++) { | ||
| d[i] = otherVec[i]; | ||
| } | ||
| } | ||
| return *this; | ||
| } | ||
|
|
||
| #endif |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters