Files
solver/MbDCode/AtPointConstraintIqcJc.cpp
2023-07-08 23:20:27 -06:00

101 lines
3.0 KiB
C++

#include "AtPointConstraintIqcJc.h"
#include "DispCompIeqcJecO.h"
#include "CREATE.h"
#include "EndFrameqc.h"
using namespace MbD;
AtPointConstraintIqcJc::AtPointConstraintIqcJc(EndFrmcptr frmi, EndFrmcptr frmj, int axisi) :
AtPointConstraintIJ(frmi, frmj, axisi)
{
}
void AtPointConstraintIqcJc::initializeGlobally()
{
AtPointConstraintIJ::initializeGlobally();
ppGpEIpEI = (std::static_pointer_cast<DispCompIeqcJecO>(riIeJeO))->ppriIeJeOpEIpEI;
}
void AtPointConstraintIqcJc::initriIeJeO()
{
riIeJeO = CREATE<DispCompIeqcJecO>::With(frmI, frmJ, axis);
}
void AtPointConstraintIqcJc::calcPostDynCorrectorIteration()
{
AtPointConstraintIJ::calcPostDynCorrectorIteration();
pGpEI = std::static_pointer_cast<DispCompIeqcJecO>(riIeJeO)->priIeJeOpEI;
}
void AtPointConstraintIqcJc::useEquationNumbers()
{
auto frmIeqc = std::static_pointer_cast<EndFrameqc>(frmI);
iqXIminusOnePlusAxis = frmIeqc->iqX() + axis;
iqEI = frmIeqc->iqE();
}
void AtPointConstraintIqcJc::fillPosICError(FColDsptr col)
{
Constraint::fillPosICError(col);
col->at(iqXIminusOnePlusAxis) -= lam;
col->atiplusFullVectortimes(iqEI, pGpEI, lam);
}
void AtPointConstraintIqcJc::fillPosICJacob(SpMatDsptr mat)
{
mat->atijplusNumber(iG, iqXIminusOnePlusAxis, -1.0);
mat->atijplusNumber(iqXIminusOnePlusAxis, iG, -1.0);
mat->atijplusFullRow(iG, iqEI, pGpEI);
mat->atijplusFullColumn(iqEI, iG, pGpEI->transpose());
mat->atijplusFullMatrixtimes(iqEI, iqEI, ppGpEIpEI, lam);
}
void AtPointConstraintIqcJc::fillPosKineJacob(SpMatDsptr mat)
{
mat->atijplusNumber(iG, iqXIminusOnePlusAxis, -1.0);
mat->atijplusFullRow(iG, iqEI, pGpEI);
}
void AtPointConstraintIqcJc::fillVelICJacob(SpMatDsptr mat)
{
mat->atijplusNumber(iG, iqXIminusOnePlusAxis, -1.0);
mat->atijplusNumber(iqXIminusOnePlusAxis, iG, -1.0);
mat->atijplusFullRow(iG, iqEI, pGpEI);
mat->atijplusFullColumn(iqEI, iG, pGpEI->transpose());
}
void AtPointConstraintIqcJc::fillAccICIterError(FColDsptr col)
{
col->atiminusNumber(iqXIminusOnePlusAxis, lam);
col->atiplusFullVectortimes(iqEI, pGpEI, lam);
auto efrmIqc = std::static_pointer_cast<EndFrameqc>(frmI);
auto qEdotI = efrmIqc->qEdot();
auto sum = -efrmIqc->qXddot()->at(axis);
sum += pGpEI->timesFullColumn(efrmIqc->qEddot());
sum += qEdotI->transposeTimesFullColumn(ppGpEIpEI->timesFullColumn(qEdotI));
col->atiplusNumber(iG, sum);
}
void AtPointConstraintIqcJc::addToJointForceI(FColDsptr col)
{
col->atiminusNumber(axis, lam);
}
void AtPointConstraintIqcJc::addToJointTorqueI(FColDsptr jointTorque)
{
auto cForceT = std::make_shared<FullRow<double>>(3, 0.0);
cForceT->atiput(axis, -lam);
auto rIpIeIp = frmI->rpep();
auto pAOIppEI = frmI->pAOppE();
auto aBOIp = frmI->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);
}