404 lines
10 KiB
C++
404 lines
10 KiB
C++
#include <string>
|
|
#include <cassert>
|
|
#include <fstream>
|
|
#include <algorithm>
|
|
#include <numeric>
|
|
|
|
#include "MBDynSystem.h"
|
|
#include "CREATE.h"
|
|
#include "FullColumn.h"
|
|
#include "MBDynInitialValue.h"
|
|
#include "MBDynData.h"
|
|
#include "MBDynControlData.h"
|
|
#include "ASMTAssembly.h"
|
|
#include "ASMTConstantGravity.h"
|
|
#include "ASMTTime.h"
|
|
#include "MBDynBody.h"
|
|
#include "MBDynStructural.h"
|
|
#include "SymbolicParser.h"
|
|
#include "BasicUserFunction.h"
|
|
#include "MBDynJoint.h"
|
|
#include "MBDynReference.h"
|
|
#include <iomanip>
|
|
|
|
using namespace MbD;
|
|
|
|
void MbD::MBDynSystem::runFile(const char* filename)
|
|
{
|
|
std::ifstream stream(filename);
|
|
std::string line;
|
|
std::vector<std::string> lines;
|
|
while (std::getline(stream, line)) {
|
|
lines.push_back(line);
|
|
}
|
|
eraseComments(lines);
|
|
auto statements = collectStatements(lines);
|
|
|
|
auto system = std::make_shared<MBDynSystem>();
|
|
system->setFilename(filename);
|
|
system->parseMBDyn(statements);
|
|
system->runKINEMATIC();
|
|
}
|
|
|
|
void MbD::MBDynSystem::parseMBDyn(std::vector<std::string>& lines)
|
|
{
|
|
readDataBlock(lines);
|
|
readInitialValueBlock(lines);
|
|
readControlDataBlock(lines);
|
|
readLabels(lines);
|
|
readVariables(lines);
|
|
readReferences(lines);
|
|
readNodesBlock(lines);
|
|
readElementsBlock(lines);
|
|
|
|
}
|
|
|
|
std::shared_ptr<std::vector<std::shared_ptr<MBDynNode>>> MbD::MBDynSystem::mbdynNodes()
|
|
{
|
|
return nodes;
|
|
}
|
|
|
|
std::shared_ptr<std::map<std::string, Symsptr>> MbD::MBDynSystem::mbdynVariables()
|
|
{
|
|
return variables;
|
|
}
|
|
|
|
std::shared_ptr<std::map<std::string, std::shared_ptr<MBDynReference>>> MbD::MBDynSystem::mbdynReferences()
|
|
{
|
|
return references;
|
|
}
|
|
|
|
void MbD::MBDynSystem::createASMT()
|
|
{
|
|
auto asmtAsm = std::make_shared<ASMTAssembly>();
|
|
asmtAsm->mbdynItem = this;
|
|
asmtItem = asmtAsm;
|
|
asmtItem->setName("Assembly");
|
|
initialValue->createASMT();
|
|
for (auto& node : *nodes) node->createASMT();
|
|
for (auto& element : *elements) element->createASMT();
|
|
}
|
|
|
|
std::shared_ptr<MBDynNode> MbD::MBDynSystem::nodeAt(std::string nodeName)
|
|
{
|
|
for (auto& node : *nodes) {
|
|
if (node->name == nodeName) return node;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
int MbD::MBDynSystem::nodeidAt(std::string nodeName)
|
|
{
|
|
return labels->at(nodeName);
|
|
}
|
|
|
|
std::shared_ptr<MBDynBody> MbD::MBDynSystem::bodyWithNode(std::string nodeName)
|
|
{
|
|
for (auto& element : *elements) {
|
|
std::string str = typeid(element).name();
|
|
if (str.find("MBDynBody") != std::string::npos) {
|
|
auto body = std::static_pointer_cast<MBDynBody>(element);
|
|
if (body->nodeName == nodeName) return body;
|
|
}
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
std::shared_ptr<ASMTAssembly> MbD::MBDynSystem::asmtAssembly()
|
|
{
|
|
return std::static_pointer_cast<ASMTAssembly>(asmtItem);
|
|
}
|
|
|
|
std::vector<std::string> MbD::MBDynSystem::nodeNames()
|
|
{
|
|
auto nodeNames = std::vector<std::string>();
|
|
for (auto& node : *nodes) {
|
|
nodeNames.push_back(node->name);
|
|
}
|
|
return nodeNames;
|
|
}
|
|
|
|
void MbD::MBDynSystem::runKINEMATIC()
|
|
{
|
|
createASMT();
|
|
std::static_pointer_cast<ASMTAssembly>(asmtItem)->runKINEMATIC();
|
|
outputFiles();
|
|
}
|
|
|
|
void MbD::MBDynSystem::outputFiles()
|
|
{
|
|
auto movFile = filename.substr(0, filename.find_last_of('.')) + ".mov";
|
|
auto asmtAsm = asmtAssembly();
|
|
auto asmtTimes = asmtAsm->times;
|
|
auto asmtParts = asmtAsm->parts;
|
|
auto asmtJoints = asmtAsm->joints;
|
|
auto asmtMotions = asmtAsm->motions;
|
|
std::ofstream os(movFile);
|
|
os << std::setprecision(std::numeric_limits<double>::digits10 + 1);
|
|
for (int i = 1; i < asmtTimes->size(); i++)
|
|
{
|
|
for (auto& node : *nodes) {
|
|
node->outputLine(i, os);
|
|
}
|
|
}
|
|
os.close();
|
|
}
|
|
|
|
void MbD::MBDynSystem::setFilename(std::string str)
|
|
{
|
|
filename = str;
|
|
}
|
|
|
|
void MbD::MBDynSystem::readDataBlock(std::vector<std::string>& lines)
|
|
{
|
|
std::vector<std::string> tokens{ "begin:", "data" };
|
|
auto beginit = findLineWith(lines, tokens);
|
|
std::vector<std::string> tokens1{ "end:", "data" };
|
|
auto endit = findLineWith(lines, tokens1);
|
|
std::vector<std::string> blocklines = { beginit, endit + 1 };
|
|
parseMBDynData(blocklines);
|
|
lines.erase(beginit, endit + 1);
|
|
}
|
|
|
|
void MbD::MBDynSystem::readInitialValueBlock(std::vector<std::string>& lines)
|
|
{
|
|
std::vector<std::string> tokens{ "begin:", "initial", "value" };
|
|
auto beginit = findLineWith(lines, tokens);
|
|
std::vector<std::string> tokens1{ "end:", "initial", "value" };
|
|
auto endit = findLineWith(lines, tokens1);
|
|
std::vector<std::string> blocklines = { beginit, endit + 1 };
|
|
initialValue = std::make_shared<MBDynInitialValue>();
|
|
initialValue->owner = this;
|
|
initialValue->parseMBDyn(blocklines);
|
|
lines.erase(beginit, endit + 1);
|
|
}
|
|
|
|
void MbD::MBDynSystem::readControlDataBlock(std::vector<std::string>& lines)
|
|
{
|
|
std::vector<std::string> tokens{ "begin:", "control", "data" };
|
|
auto beginit = findLineWith(lines, tokens);
|
|
std::vector<std::string> tokens1{ "end:", "control", "data" };
|
|
auto endit = findLineWith(lines, tokens1);
|
|
std::vector<std::string> blocklines = { beginit, endit + 1 };
|
|
controlData = std::make_shared<MBDynControlData>();
|
|
controlData->owner = this;
|
|
controlData->parseMBDyn(blocklines);
|
|
lines.erase(beginit, endit + 1);
|
|
}
|
|
|
|
void MbD::MBDynSystem::readLabels(std::vector<std::string>& lines)
|
|
{
|
|
parseMBDynLabels(lines);
|
|
}
|
|
|
|
void MbD::MBDynSystem::readVariables(std::vector<std::string>& lines)
|
|
{
|
|
parseMBDynVariables(lines);
|
|
}
|
|
|
|
void MbD::MBDynSystem::readReferences(std::vector<std::string>& lines)
|
|
{
|
|
parseMBDynReferences(lines);
|
|
}
|
|
|
|
void MbD::MBDynSystem::readNodesBlock(std::vector<std::string>& lines)
|
|
{
|
|
std::vector<std::string> tokens{ "begin:", "nodes" };
|
|
auto beginit = findLineWith(lines, tokens);
|
|
std::vector<std::string> tokens1{ "end:", "nodes" };
|
|
auto endit = findLineWith(lines, tokens1);
|
|
std::vector<std::string> blocklines = { beginit, endit + 1 };
|
|
parseMBDynNodes(blocklines);
|
|
lines.erase(beginit, endit + 1);
|
|
}
|
|
|
|
void MbD::MBDynSystem::readElementsBlock(std::vector<std::string>& lines)
|
|
{
|
|
std::vector<std::string> tokens{ "begin:", "elements" };
|
|
auto beginit = findLineWith(lines, tokens);
|
|
std::vector<std::string> tokens1{ "end:", "elements" };
|
|
auto endit = findLineWith(lines, tokens1);
|
|
std::vector<std::string> blocklines = { beginit, endit + 1 };
|
|
parseMBDynElements(blocklines);
|
|
lines.erase(beginit, endit + 1);
|
|
}
|
|
|
|
void MbD::MBDynSystem::eraseComments(std::vector<std::string>& lines)
|
|
{
|
|
for (int i = 0; i < lines.size(); i++)
|
|
{
|
|
auto line = lines[i];
|
|
auto it = line.find('#');
|
|
if (it != std::string::npos) {
|
|
lines[i] = line.substr(0, it);
|
|
}
|
|
}
|
|
for (int i = lines.size() - 1; i >= 0; i--) {
|
|
auto& line = lines[i];
|
|
auto it = std::find_if(line.begin(), line.end(), [](unsigned char ch) { return !std::isspace(ch); });
|
|
if (it == line.end()) lines.erase(lines.begin() + i);
|
|
}
|
|
}
|
|
|
|
std::vector<std::string> MbD::MBDynSystem::collectStatements(std::vector<std::string>& lines)
|
|
{
|
|
auto statements = std::vector<std::string>();
|
|
while (!lines.empty()) {
|
|
std::stringstream ss;
|
|
while (!lines.empty()) {
|
|
auto line = lines[0];
|
|
lines.erase(lines.begin());
|
|
auto i = line.find(';');
|
|
if (i != std::string::npos) {
|
|
ss << line.substr(0, i);
|
|
if (line.size() > i + 1) {
|
|
auto remainder = line.substr(i + 1);
|
|
auto it = std::find_if(remainder.begin(), remainder.end(), [](unsigned char ch) { return !std::isspace(ch); });
|
|
if (it != remainder.end()) lines.insert(lines.begin(), remainder);
|
|
}
|
|
break;
|
|
}
|
|
else {
|
|
ss << line;
|
|
}
|
|
}
|
|
statements.push_back(ss.str());
|
|
}
|
|
return statements;
|
|
}
|
|
|
|
void MbD::MBDynSystem::initialize()
|
|
{
|
|
}
|
|
|
|
void MbD::MBDynSystem::parseMBDynData(std::vector<std::string>& lines)
|
|
{
|
|
assert(lines.size() == 3);
|
|
std::vector<std::string> tokens{ "problem:", "initial", "value" };
|
|
auto problemit = findLineWith(lines, tokens);
|
|
assert(problemit != lines.end());
|
|
data = *problemit;
|
|
}
|
|
|
|
void MbD::MBDynSystem::parseMBDynNodes(std::vector<std::string>& lines)
|
|
{
|
|
nodes = std::make_shared<std::vector<std::shared_ptr<MBDynNode>>>();
|
|
std::vector<std::string> tokens{ "structural:" };
|
|
while (true) {
|
|
auto it = findLineWith(lines, tokens);
|
|
if (it != lines.end()) {
|
|
auto structural = std::make_shared<MBDynStructural>();
|
|
structural->owner = this;
|
|
structural->parseMBDyn(*it);
|
|
nodes->push_back(structural);
|
|
lines.erase(it);
|
|
}
|
|
else {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void MbD::MBDynSystem::parseMBDynElements(std::vector<std::string>& lines)
|
|
{
|
|
elements = std::make_shared<std::vector<std::shared_ptr<MBDynElement>>>();
|
|
std::vector<std::string> bodyToken{ "body:" };
|
|
std::vector<std::string> jointToken{ "joint:" };
|
|
std::vector<std::string>::iterator it;
|
|
while (true) {
|
|
it = findLineWith(lines, bodyToken);
|
|
if (it != lines.end()) {
|
|
auto body = std::make_shared<MBDynBody>();
|
|
body->owner = this;
|
|
body->parseMBDyn(*it);
|
|
elements->push_back(body);
|
|
lines.erase(it);
|
|
continue;
|
|
}
|
|
it = findLineWith(lines, jointToken);
|
|
if (it != lines.end()) {
|
|
auto joint = std::make_shared<MBDynJoint>();
|
|
joint->owner = this;
|
|
joint->parseMBDyn(*it);
|
|
elements->push_back(joint);
|
|
lines.erase(it);
|
|
continue;
|
|
}
|
|
break;
|
|
}
|
|
}
|
|
|
|
void MbD::MBDynSystem::parseMBDynVariables(std::vector<std::string>& lines)
|
|
{
|
|
variables = std::make_shared<std::map<std::string, Symsptr>>();
|
|
std::string str, variable;
|
|
double doubleValue;
|
|
std::vector<std::string> tokens{ "set:", "real" };
|
|
while (true) {
|
|
auto it = findLineWith(lines, tokens);
|
|
if (it != lines.end()) {
|
|
std::istringstream iss(*it);
|
|
iss >> str;
|
|
iss >> str;
|
|
iss >> variable;
|
|
iss >> str;
|
|
iss >> str;
|
|
auto parser = std::make_shared<SymbolicParser>();
|
|
parser->variables = variables;
|
|
auto userFunc = std::make_shared<BasicUserFunction>(str, 1.0);
|
|
parser->parseUserFunction(userFunc);
|
|
auto sym = parser->stack->top();
|
|
variables->insert(std::make_pair(variable, sym));
|
|
lines.erase(it);
|
|
}
|
|
else {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void MbD::MBDynSystem::parseMBDynLabels(std::vector<std::string>& lines)
|
|
{
|
|
labels = std::make_shared<std::map<std::string, int>>();
|
|
std::string str, label;
|
|
int intValue;
|
|
std::vector<std::string> tokens{ "set:", "integer" };
|
|
while (true) {
|
|
auto it = findLineWith(lines, tokens);
|
|
if (it != lines.end()) {
|
|
std::istringstream iss(*it);
|
|
iss >> str;
|
|
iss >> str;
|
|
iss >> label;
|
|
iss >> str;
|
|
iss >> intValue;
|
|
labels->insert(std::make_pair(label, intValue));
|
|
lines.erase(it);
|
|
}
|
|
else {
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
|
|
void MbD::MBDynSystem::parseMBDynReferences(std::vector<std::string>& lines)
|
|
{
|
|
references = std::make_shared<std::map<std::string, std::shared_ptr<MBDynReference>>>();
|
|
std::string str, refName;
|
|
double doubleValue;
|
|
std::vector<std::string> tokens{ "reference:" };
|
|
while (true) {
|
|
auto it = findLineWith(lines, tokens);
|
|
if (it != lines.end()) {
|
|
auto reference = std::make_shared<MBDynReference>();
|
|
reference->owner = this;
|
|
reference->parseMBDyn(*it);
|
|
references->insert(std::make_pair(reference->name, reference));
|
|
lines.erase(it);
|
|
}
|
|
else {
|
|
break;
|
|
}
|
|
}
|
|
}
|