axmol/tests/cpp-tests/Classes/Box2DTestBed/tests/rope.cpp

287 lines
7.9 KiB
C++

// MIT License
// Copyright (c) 2019 Erin Catto
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.
#include "settings.h"
#include "test.h"
#include "box2d/b2_rope.h"
#include "ImGuiEXT/CCImGuiEXT.h"
///
class Rope : public Test
{
public:
Rope()
{
const int32 N = 20;
const float L = 0.5f;
b2Vec2 vertices[N];
float masses[N];
for (int32 i = 0; i < N; ++i)
{
vertices[i].Set(0.0f, L * (N - i));
masses[i] = 1.0f;
}
masses[0] = 0.0f;
masses[1] = 0.0f;
m_tuning1.bendHertz = 30.0f;
m_tuning1.bendDamping = 4.0f;
m_tuning1.bendStiffness = 1.0f;
m_tuning1.bendingModel = b2_pbdTriangleBendingModel;
m_tuning1.isometric = true;
m_tuning1.stretchHertz = 30.0f;
m_tuning1.stretchDamping = 4.0f;
m_tuning1.stretchStiffness = 1.0f;
m_tuning1.stretchingModel = b2_pbdStretchingModel;
m_tuning2.bendHertz = 30.0f;
m_tuning2.bendDamping = 0.7f;
m_tuning2.bendStiffness = 1.0f;
m_tuning2.bendingModel = b2_pbdHeightBendingModel;
m_tuning2.isometric = true;
m_tuning2.stretchHertz = 30.0f;
m_tuning2.stretchDamping = 1.0f;
m_tuning2.stretchStiffness = 1.0f;
m_tuning2.stretchingModel = b2_pbdStretchingModel;
m_position1.Set(-5.0f, 15.0f);
m_position2.Set(5.0f, 15.0f);
b2RopeDef def;
def.vertices = vertices;
def.count = N;
def.gravity.Set(0.0f, -10.0f);
def.masses = masses;
def.position = m_position1;
def.tuning = m_tuning1;
m_rope1.Create(def);
def.position = m_position2;
def.tuning = m_tuning2;
m_rope2.Create(def);
m_iterations1 = 8;
m_iterations2 = 8;
m_speed = 10.0f;
}
void UpdateUI() override
{
//ImGui::SetNextWindowPos(ImVec2(10.0f, 100.0f));
ImGui::SetNextWindowSize(ImVec2(200.0f, 700.0f));
ImGui::Begin("Tuning", nullptr, ImGuiWindowFlags_NoResize);
ImGui::Separator();
ImGui::PushItemWidth(ImGui::GetWindowWidth() * 0.5f);
const ImGuiComboFlags comboFlags = 0;
const char* bendModels[] = { "Spring", "PBD Ang", "XPBD Ang", "PBD Dist", "PBD Height", "PBD Triangle" };
const char* stretchModels[] = { "PBD", "XPBD" };
ImGui::Text("Rope 1");
static int bendModel1 = m_tuning1.bendingModel;
if (ImGui::BeginCombo("Bend Model##1", bendModels[bendModel1], comboFlags))
{
for (int i = 0; i < IM_ARRAYSIZE(bendModels); ++i)
{
bool isSelected = (bendModel1 == i);
if (ImGui::Selectable(bendModels[i], isSelected))
{
bendModel1 = i;
m_tuning1.bendingModel = b2BendingModel(i);
}
if (isSelected)
{
ImGui::SetItemDefaultFocus();
}
}
ImGui::EndCombo();
}
ImGui::SliderFloat("Damping##B1", &m_tuning1.bendDamping, 0.0f, 4.0f, "%.1f");
ImGui::SliderFloat("Hertz##B1", &m_tuning1.bendHertz, 0.0f, 60.0f, "%.0f");
ImGui::SliderFloat("Stiffness##B1", &m_tuning1.bendStiffness, 0.0f, 1.0f, "%.1f");
ImGui::Checkbox("Isometric##1", &m_tuning1.isometric);
ImGui::Checkbox("Fixed Mass##1", &m_tuning1.fixedEffectiveMass);
ImGui::Checkbox("Warm Start##1", &m_tuning1.warmStart);
static int stretchModel1 = m_tuning1.stretchingModel;
if (ImGui::BeginCombo("Stretch Model##1", stretchModels[stretchModel1], comboFlags))
{
for (int i = 0; i < IM_ARRAYSIZE(stretchModels); ++i)
{
bool isSelected = (stretchModel1 == i);
if (ImGui::Selectable(stretchModels[i], isSelected))
{
stretchModel1 = i;
m_tuning1.stretchingModel = b2StretchingModel(i);
}
if (isSelected)
{
ImGui::SetItemDefaultFocus();
}
}
ImGui::EndCombo();
}
ImGui::SliderFloat("Damping##S1", &m_tuning1.stretchDamping, 0.0f, 4.0f, "%.1f");
ImGui::SliderFloat("Hertz##S1", &m_tuning1.stretchHertz, 0.0f, 60.0f, "%.0f");
ImGui::SliderFloat("Stiffness##S1", &m_tuning1.stretchStiffness, 0.0f, 1.0f, "%.1f");
ImGui::SliderInt("Iterations##1", &m_iterations1, 1, 100, "%d");
ImGui::Separator();
ImGui::Text("Rope 2");
static int bendModel2 = m_tuning2.bendingModel;
if (ImGui::BeginCombo("Bend Model##2", bendModels[bendModel2], comboFlags))
{
for (int i = 0; i < IM_ARRAYSIZE(bendModels); ++i)
{
bool isSelected = (bendModel2 == i);
if (ImGui::Selectable(bendModels[i], isSelected))
{
bendModel2 = i;
m_tuning2.bendingModel = b2BendingModel(i);
}
if (isSelected)
{
ImGui::SetItemDefaultFocus();
}
}
ImGui::EndCombo();
}
ImGui::SliderFloat("Damping##B2", &m_tuning2.bendDamping, 0.0f, 4.0f, "%.1f");
ImGui::SliderFloat("Hertz##B2", &m_tuning2.bendHertz, 0.0f, 60.0f, "%.0f");
ImGui::SliderFloat("Stiffness##B2", &m_tuning2.bendStiffness, 0.0f, 1.0f, "%.1f");
ImGui::Checkbox("Isometric##2", &m_tuning2.isometric);
ImGui::Checkbox("Fixed Mass##2", &m_tuning2.fixedEffectiveMass);
ImGui::Checkbox("Warm Start##2", &m_tuning2.warmStart);
static int stretchModel2 = m_tuning2.stretchingModel;
if (ImGui::BeginCombo("Stretch Model##2", stretchModels[stretchModel2], comboFlags))
{
for (int i = 0; i < IM_ARRAYSIZE(stretchModels); ++i)
{
bool isSelected = (stretchModel2 == i);
if (ImGui::Selectable(stretchModels[i], isSelected))
{
stretchModel2 = i;
m_tuning2.stretchingModel = b2StretchingModel(i);
}
if (isSelected)
{
ImGui::SetItemDefaultFocus();
}
}
ImGui::EndCombo();
}
ImGui::SliderFloat("Damping##S2", &m_tuning2.stretchDamping, 0.0f, 4.0f, "%.1f");
ImGui::SliderFloat("Hertz##S2", &m_tuning2.stretchHertz, 0.0f, 60.0f, "%.0f");
ImGui::SliderFloat("Stiffness##S2", &m_tuning2.stretchStiffness, 0.0f, 1.0f, "%.1f");
ImGui::SliderInt("Iterations##2", &m_iterations2, 1, 100, "%d");
ImGui::Separator();
ImGui::SliderFloat("Speed", &m_speed, 10.0f, 100.0f, "%.0f");
if (ImGui::Button("Reset"))
{
m_position1.Set(-5.0f, 15.0f);
m_position2.Set(5.0f, 15.0f);
m_rope1.Reset(m_position1);
m_rope2.Reset(m_position2);
}
ImGui::PopItemWidth();
ImGui::End();
}
void Step(Settings& settings) override
{
float dt = settings.m_hertz > 0.0f ? 1.0f / settings.m_hertz : 0.0f;
if (settings.m_pause == 1 && settings.m_singleStep == 0)
{
dt = 0.0f;
}
//if (glfwGetKey(g_mainWindow, GLFW_KEY_COMMA) == GLFW_PRESS)
//{
// m_position1.x -= m_speed * dt;
// m_position2.x -= m_speed * dt;
//}
//if (glfwGetKey(g_mainWindow, GLFW_KEY_PERIOD) == GLFW_PRESS)
//{
// m_position1.x += m_speed * dt;
// m_position2.x += m_speed * dt;
//}
m_rope1.SetTuning(m_tuning1);
m_rope2.SetTuning(m_tuning2);
m_rope1.Step(dt, m_iterations1, m_position1);
m_rope2.Step(dt, m_iterations2, m_position2);
Test::Step(settings);
m_rope1.Draw(&g_debugDraw);
m_rope2.Draw(&g_debugDraw);
DrawString(5, m_textLine, "Press comma and period to move left and right");
}
static Test* Create()
{
return new Rope;
}
b2Rope m_rope1;
b2Rope m_rope2;
b2RopeTuning m_tuning1;
b2RopeTuning m_tuning2;
int32 m_iterations1;
int32 m_iterations2;
b2Vec2 m_position1;
b2Vec2 m_position2;
float m_speed;
};
static int testIndex = RegisterTest("Rope", "Bending", Rope::Create);