基于无质量分段的2D绳索Python仿真：鼠标控制失效求助

你的代码里鼠标无法固定第一个节点的核心原因是：约束迭代过程中，第一个节点的位置被后续的修正逻辑覆盖了。

在约束处理的循环里，当处理第一个分段（i=1，连接节点0和1）时，代码会同时调整节点0和节点1的位置，但节点0应该是被鼠标固定的，不能被修改。这就导致你设置的control_position在约束修正后被重置，最终整个绳索还是会下落。

修复方案 #

修改约束迭代的逻辑，让第一个节点（索引0）在约束修正时保持固定，不参与位置调整：

在约束处理的循环中，判断当前处理的是第一个分段（i=1）时，只调整当前节点（i）的位置，不修改节点0；对于其他分段，保持原来的双向调整逻辑。

修改后的完整代码 #

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.animation import FuncAnimation

# Simulation parameters
num_nodes = 10
length = 1.0
dt = 0.01
total_time = 10.0
gravity = 9.81
segment_length = length / (num_nodes - 1)
constraint_iterations = 10

# Initialize positions and velocities
node_positions = np.zeros((num_nodes, 2))
node_velocities = np.zeros((num_nodes, 2))

# Set initial positions (straight line)
for i in range(num_nodes):
    node_positions[i, 0] = i * segment_length

# Function to update node positions
def update_positions(positions, velocities, dt, control_position):
    new_positions = positions.copy()
    new_velocities = velocities.copy()

    # Apply gravity to velocities
    new_velocities[1:, 1] -= gravity * dt  # Skip the controlled endpoint

    # Update positions with velocities
    new_positions[1:] += new_velocities[1:] * dt  # Skip the controlled endpoint

    # Set the controlled endpoint to the control_position
    new_positions[0] = control_position

    # Constraint iterations to maintain constant segment lengths
    for _ in range(constraint_iterations):
        for i in range(1, num_nodes):
            segment_vector = new_positions[i] - new_positions[i - 1]
            segment_length_current = np.linalg.norm(segment_vector)
            if segment_length_current != 0:
                correction_vector = (segment_length_current - segment_length) * (segment_vector / segment_length_current)
                if i == 1:
                    # 第一个分段，只调整当前节点，固定节点0
                    new_positions[i] -= correction_vector
                elif i < num_nodes - 1:  # Adjust both nodes if not the last segment
                    new_positions[i] -= correction_vector / 2
                    new_positions[i - 1] += correction_vector / 2
                else:  # Adjust only the previous node if the last segment
                    new_positions[i - 1] += correction_vector

    return new_positions, new_velocities

# Function to handle mouse motion
def on_mouse_move(event):
    if event.inaxes:
        global control_position
        control_position = np.array([event.xdata, event.ydata])

# Initialize control position
control_position = node_positions[0]

# Set up plot
fig, ax = plt.subplots()
ax.set_xlim(-0.1, length + 0.1)
ax.set_ylim(-length, length)
line, = ax.plot(node_positions[:, 0], node_positions[:, 1], 'o-', color='blue')

# Update function for animation
def animate(frame):
    global node_positions, node_velocities
    node_positions, node_velocities = update_positions(node_positions, node_velocities, dt, control_position)
    line.set_data(node_positions[:, 0], node_positions[:, 1])
    return line,

# Connect the mouse motion event
fig.canvas.mpl_connect('motion_notify_event', on_mouse_move)

# Create animation
ani = FuncAnimation(fig, animate, frames=int(total_time / dt), interval=dt*1000, blit=True)

plt.xlabel('X Position')
plt.ylabel('Y Position')
plt.title('2D Cable Simulation with Cursor Control')
plt.show()

额外优化建议 #

• 可以给control_position添加激活判断，比如只有鼠标按下时才控制节点，避免误触
• 增加约束迭代次数可以提升绳索的刚性表现，但会增加计算量

内容的提问来源于stack exchange，提问作者galtor