- Move existing OndselSolver, GNN ML layer, and tooling into GNN/ directory for integration in later phases - Add Create addon scaffold: package.xml, Init.py - Add expression DAG with eval, symbolic diff, simplification - Add parameter table with fixed/free variable tracking - Add quaternion rotation as polynomial Expr trees - Add RigidBody entity (7 DOF: position + unit quaternion) - Add constraint classes: Coincident, DistancePointPoint, Fixed - Add Newton-Raphson solver with symbolic Jacobian + numpy lstsq - Add pre-solve passes: substitution + single-equation - Add DOF counting via Jacobian SVD rank - Add KindredSolver IKCSolver bridge for kcsolve integration - Add 82 unit tests covering all modules Registers as 'kindred' solver via kcsolve.register_solver() when loaded by Create's addon_loader.
101 lines
3.2 KiB
C++
101 lines
3.2 KiB
C++
/***************************************************************************
|
|
* Copyright (c) 2023 Ondsel, Inc. *
|
|
* *
|
|
* This file is part of OndselSolver. *
|
|
* *
|
|
* See LICENSE file for details about copyright. *
|
|
***************************************************************************/
|
|
|
|
#include "ASMTPrincipalMassMarker.h"
|
|
#include <cassert>
|
|
#include "FullMatrix.h"
|
|
|
|
using namespace MbD;
|
|
|
|
MbD::ASMTPrincipalMassMarker::ASMTPrincipalMassMarker()
|
|
{
|
|
name = "MassMarker";
|
|
}
|
|
|
|
std::shared_ptr<ASMTPrincipalMassMarker> MbD::ASMTPrincipalMassMarker::With()
|
|
{
|
|
auto asmt = std::make_shared<ASMTPrincipalMassMarker>();
|
|
asmt->initialize();
|
|
return asmt;
|
|
}
|
|
|
|
void MbD::ASMTPrincipalMassMarker::parseASMT(std::vector<std::string>& lines)
|
|
{
|
|
auto pos = lines[0].find_first_not_of("\t");
|
|
auto leadingTabs = lines[0].substr(0, pos);
|
|
assert(lines[0] == (leadingTabs + "Name"));
|
|
lines.erase(lines.begin());
|
|
name = readString(lines[0]);
|
|
lines.erase(lines.begin());
|
|
assert(lines[0] == (leadingTabs + "Position3D"));
|
|
lines.erase(lines.begin());
|
|
position3D = readColumnOfDoubles(lines[0]);
|
|
lines.erase(lines.begin());
|
|
assert(lines[0] == (leadingTabs + "RotationMatrix"));
|
|
lines.erase(lines.begin());
|
|
rotationMatrix = std::make_shared<FullMatrix<double>>(3);
|
|
for (size_t i = 0; i < 3; i++)
|
|
{
|
|
auto row = readRowOfDoubles(lines[0]);
|
|
rotationMatrix->atiput(i, row);
|
|
lines.erase(lines.begin());
|
|
}
|
|
assert(lines[0] == (leadingTabs + "Mass"));
|
|
lines.erase(lines.begin());
|
|
mass = readDouble(lines[0]);
|
|
lines.erase(lines.begin());
|
|
assert(lines[0] == (leadingTabs + "MomentOfInertias"));
|
|
lines.erase(lines.begin());
|
|
momentOfInertias = std::make_shared<DiagonalMatrix<double>>(3);
|
|
auto row = readRowOfDoubles(lines[0]);
|
|
lines.erase(lines.begin());
|
|
for (size_t i = 0; i < 3; i++)
|
|
{
|
|
momentOfInertias->atiput(i, row->at(i));
|
|
}
|
|
assert(lines[0] == (leadingTabs + "Density"));
|
|
lines.erase(lines.begin());
|
|
density = readDouble(lines[0]);
|
|
lines.erase(lines.begin());
|
|
}
|
|
|
|
void MbD::ASMTPrincipalMassMarker::setMass(double m)
|
|
{
|
|
mass = m;
|
|
}
|
|
|
|
void MbD::ASMTPrincipalMassMarker::setDensity(double rho)
|
|
{
|
|
density = rho;
|
|
}
|
|
|
|
void MbD::ASMTPrincipalMassMarker::setMomentOfInertias(DiagMatDsptr mat)
|
|
{
|
|
momentOfInertias = mat;
|
|
}
|
|
|
|
// Overloads to simplify syntax.
|
|
void MbD::ASMTPrincipalMassMarker::setMomentOfInertias(double a, double b, double c)
|
|
{
|
|
momentOfInertias = std::make_shared<DiagonalMatrix<double>>(ListD{ a, b, c });
|
|
}
|
|
|
|
void MbD::ASMTPrincipalMassMarker::storeOnLevel(std::ofstream& os, size_t level)
|
|
{
|
|
storeOnLevelString(os, level, "PrincipalMassMarker");
|
|
storeOnLevelString(os, level + 1, "Name");
|
|
storeOnLevelString(os, level + 2, name);
|
|
ASMTSpatialItem::storeOnLevel(os, level);
|
|
storeOnLevelString(os, level + 1, "Mass");
|
|
storeOnLevelDouble(os, level + 2, mass);
|
|
storeOnLevelString(os, level + 1, "MomentOfInertias");
|
|
storeOnLevelArray(os, level + 2, *momentOfInertias);
|
|
storeOnLevelString(os, level + 1, "Density");
|
|
storeOnLevelDouble(os, level + 2, density);
|
|
}
|