Files
solver/MbDCode/ScrewConstraintIqcJc.cpp
2023-07-16 22:00:07 -06:00

126 lines
3.9 KiB
C++

#include <corecrt_math_defines.h>
#include "ScrewConstraintIqcJc.h"
#include "EndFrameqc.h"
#include "CREATE.h"
#include "DispCompIeqcJecIe.h"
#include "AngleZIeqcJec.h"
using namespace MbD;
MbD::ScrewConstraintIqcJc::ScrewConstraintIqcJc(EndFrmcptr frmi, EndFrmcptr frmj) : ScrewConstraintIJ(frmi, frmj)
{
}
void MbD::ScrewConstraintIqcJc::addToJointForceI(FColDsptr col)
{
col->equalSelfPlusFullVectortimes(pGpXI, lam);
}
void MbD::ScrewConstraintIqcJc::addToJointTorqueI(FColDsptr jointTorque)
{
auto cForceT = pGpXI->times(lam);
auto frmIeqc = std::static_pointer_cast<EndFrameqc>(frmI);
auto rIpIeIp = frmIeqc->rpep();
auto pAOIppEI = frmIeqc->pAOppE();
auto aBOIp = frmIeqc->aBOp();
auto fpAOIppEIrIpIeIp = std::make_shared<FullColumn<double>>(4, 0.0);
for (int i = 0; i < 4; i++)
{
auto dum = cForceT->timesFullColumn(pAOIppEI->at(i)->timesFullColumn(rIpIeIp));
fpAOIppEIrIpIeIp->atiput(i, dum);
}
auto lampGpE = pGpEI->transpose()->times(lam);
auto c2Torque = aBOIp->timesFullColumn(lampGpE->minusFullColumn(fpAOIppEIrIpIeIp));
jointTorque->equalSelfPlusFullColumntimes(c2Torque, 0.5);
}
void MbD::ScrewConstraintIqcJc::calc_pGpEI()
{
pGpEI = zIeJeIe->pvaluepEI()->times(2.0 * M_PI)->minusFullRow(thezIeJe->pvaluepEI()->times(pitch));
}
void MbD::ScrewConstraintIqcJc::calc_pGpXI()
{
pGpXI = zIeJeIe->pvaluepXI()->times(2.0 * M_PI)->minusFullRow(thezIeJe->pvaluepXI()->times(pitch));
}
void MbD::ScrewConstraintIqcJc::calc_ppGpEIpEI()
{
ppGpEIpEI = zIeJeIe->ppvaluepEIpEI()->times(2.0 * M_PI)->minusFullMatrix(thezIeJe->ppvaluepEIpEI()->times(pitch));
}
void MbD::ScrewConstraintIqcJc::calc_ppGpXIpEI()
{
ppGpXIpEI = zIeJeIe->ppvaluepXIpEI()->times(2.0 * M_PI)->minusFullMatrix(thezIeJe->ppvaluepXIpEI()->times(pitch));
}
void MbD::ScrewConstraintIqcJc::calcPostDynCorrectorIteration()
{
ScrewConstraintIJ::calcPostDynCorrectorIteration();
this->calc_pGpXI();
this->calc_pGpEI();
this->calc_ppGpXIpEI();
this->calc_ppGpEIpEI();
}
void MbD::ScrewConstraintIqcJc::fillAccICIterError(FColDsptr col)
{
col->atiplusFullVectortimes(iqXI, pGpXI, lam);
col->atiplusFullVectortimes(iqEI, pGpEI, lam);
auto efrmIqc = std::static_pointer_cast<EndFrameqc>(frmI);
auto qXdotI = efrmIqc->qXdot();
auto qEdotI = efrmIqc->qEdot();
auto sum = pGpXI->timesFullColumn(efrmIqc->qXddot());
sum += pGpEI->timesFullColumn(efrmIqc->qEddot());
sum += 2.0 * (qXdotI->transposeTimesFullColumn(ppGpXIpEI->timesFullColumn(qEdotI)));
sum += qEdotI->transposeTimesFullColumn(ppGpEIpEI->timesFullColumn(qEdotI));
col->atiplusNumber(iG, sum);
}
void MbD::ScrewConstraintIqcJc::fillPosICError(FColDsptr col)
{
ScrewConstraintIJ::fillPosICError(col);
col->atiplusFullVectortimes(iqXI, pGpXI, lam);
col->atiplusFullVectortimes(iqEI, pGpEI, lam);
}
void MbD::ScrewConstraintIqcJc::fillPosICJacob(SpMatDsptr mat)
{
mat->atijplusFullRow(iG, iqXI, pGpXI);
mat->atijplusFullColumn(iqXI, iG, pGpXI->transpose());
mat->atijplusFullRow(iG, iqEI, pGpEI);
mat->atijplusFullColumn(iqEI, iG, pGpEI->transpose());
auto ppGpXIpEIlam = ppGpXIpEI->times(lam);
mat->atijplusFullMatrix(iqXI, iqEI, ppGpXIpEIlam);
mat->atijplusTransposeFullMatrix(iqEI, iqXI, ppGpXIpEIlam);
mat->atijplusFullMatrixtimes(iqEI, iqEI, ppGpEIpEI, lam);
}
void MbD::ScrewConstraintIqcJc::fillPosKineJacob(SpMatDsptr mat)
{
mat->atijplusFullRow(iG, iqXI, pGpXI);
mat->atijplusFullRow(iG, iqEI, pGpEI);
}
void MbD::ScrewConstraintIqcJc::fillVelICJacob(SpMatDsptr mat)
{
mat->atijplusFullRow(iG, iqXI, pGpXI);
mat->atijplusFullColumn(iqXI, iG, pGpXI->transpose());
mat->atijplusFullRow(iG, iqEI, pGpEI);
mat->atijplusFullColumn(iqEI, iG, pGpEI->transpose());
}
void MbD::ScrewConstraintIqcJc::init_zthez()
{
zIeJeIe = CREATE<DispCompIeqcJecIe>::With(frmI, frmJ, 2);
thezIeJe = CREATE<AngleZIeqcJec>::With(frmJ, frmI);
}
void MbD::ScrewConstraintIqcJc::useEquationNumbers()
{
auto frmIeqc = std::static_pointer_cast<EndFrameqc>(frmI);
iqXI = frmIeqc->iqX();
iqEI = frmIeqc->iqE();
}