Skip to main content

A 3D mass-spring real world simulator with more types of forces(gravity, electricity, spring, collision, ...)

Project description

massspring

Author : Pooya Shams kolahi

Inspired by Saeed Sarkarati

Brief

using this library you can create great simulations of most of physical environments which can be calculated precisely. you can create many types of objects from which "mass" and "spring" are main ones.
more on project homepage on github and README

license

licensed under MIT License. you can freely use and modify this package, but please let me know if you are doing something interesting.

Installation

you can install the package via pip:

pip install massspring

for user-only installation use --user switch:

pip install --user massspring

if you have massspring installed and you are wishing to upgrade, you can upgrade via --upgrade switch:

pip install --upgrade massspring

all combined:

pip install --user --upgrade massspring

usage

you can import the library by using the following code (it is recommended to import as m or ms)

import massspring as m

mass

mass is the only type of object you can create right now.
you can create a mass object by using the mass class. you can learn more about the arguments and their usage in the mass class document

m1 = m.mass(x=0, y=0, z=0, vx=0, vy=0, vz=0, m=10, r=10, q=0, moveable=False, solid=True, bound=True, gravitational=False, resistible=False, electrical=False, conductive=False, color=(0, 255, 0), visible=True)

parameters meanings are explained here.
parameters / attributes:

  • x, y, z are the object's position.
  • vx, vy, vz are the object's velocity.
  • fx, fy, fz are the sum of forces applied to the object. they will be added by calling the set_force method in each force's object to the force's masses (m1, m2).
  • m is mass of object.
  • r is radius of object (each mass is consumed as a sphere).
  • q is electrical charge of the object.
  • moveable means if the object can move or not.
  • solid means if the object can hit other objects or not.
  • bound means if the object stays in the screen or not.
  • gravitational means if the object is affected by gravity force or not.
  • resistible means if the object is affected by air resistance force or not.
  • electrical means if the object is affected by other objects' electrical force and can affect them by electrical force or not.
  • conductive means if the object will share electrical charge with others or not.
  • color is the objects default color. visible means if the object is seen or not.

force

there are five types of forces (technically force subclass) you can create:
1.gravity
2.electricity
3.spring
4.collision
5.air resistance

gravity

the gravity class provides a gravity force between two masses according to the "Newton's law of universal gravitation" which you can see bellow:
gravity formula
more information is available at Newton's law of universal gravitation wikipedia page.
you can create a gravity force between two masses using the code bellow:

g1 = m.gravity(m1=m1, m2=m2)

(m1 and m2 are the two mass objects between which you want to set gravity force.)

electricity

the electricity class provides an electricity force between two masses according to the "Coulomb's law" (the vector version) which you can see bellow:
Coulomb's law
more information is available at Coulomb's law wikipedia page. you can create an electricity force between two masses using the code bellow:

e1 = m.electricity(m1=m1, m2=m2)

(m1 and m2 are the two mass objects between which you want to set electricity force.)

spring

spring can also be known as an object. However, spring class is a subclass of force class. So technically spring is a force. the force will be calculated using the "Hooke's law":
Hooke's law
more information is available on Hooke's law wikipedia page.
you can create a spring object by using the spring class. you can learn more about the arguments and their usage in the spring class document.

s1 = m.spring(k=1500000, l=0, m1=m1, m2=m2, color=(255, 0, 0), visible=True)

collision

uses the conservation of momentum and kinetic energy and the equations presented at the Elastic Collision wikipedia page.

c1 = m.collision(m1=m1, m2=m2)

air resistance

the air_resistance class provides an air_resistance force for an object according to drag equation shown bellow.
drag equation
where:

  • Fd is the drag force, which is by definition the force component in the direction of the flow velocity,
  • p is the mass density of the fluid,
  • u is the flow velocity relative to the object,
  • A is the reference area, and
  • Cd is the drag coefficient – a dimensionless coefficient related to the object's geometry and taking into account both skin friction and form drag. In general, Cd depends on the Reynolds number.

more information is available at Drag equation and Drag wikipedia page.
you can manually create an air resistance force for an object using the code bellow:

ar = m.air_resistance(m1=m1)

mainloop

after creating all of the objects and forces you want, you should call the mainloop function which is an infinite while loop and ends when the user closes the window or presses the ESCape button.
arguments are:
1.speed: the more this number is the faster your program runs. it will decrease the number of frames in which the screen will be updated(increases the space between them).
2.FPS: sets the fps parameter for the pygame.time.Clock().tick(FPS) function. the less this number is the less your cpu will be under pressure.
3.frame: it's a function which will be called every frame of the program.
4.*args: the arguments you want to give to the frame function.

m.mainloop(speed=2, FPS=0, frame=None, *args)

network

in version 1.2.0 there is a new functionality available instead of mainloop. in the new networklib.py file you can find a bunch of functions that can help in running your massspring-based module through network. some of them are explained bellow.

start_server_mainloop

this function is the main function that you run instead of massspring.mainloop and runs the mainloop just like massspring.mainloop does but with a little difference that now it is available through the network. you can pass it a couple of arguments that control how it works.
the first three are related to massspring: the massspring module that you imported and used yourself, the args to the mainloop under that massspring module which is called inside this function, and also kwargs to the mainloop.
the fourth argument is a function called client_handler it lets you define your own function that will be used to communicate over the net. However you can use the predefined functions in networklib too.
the last two arguments are the host and port which are going to be used for the server socket to bind to. if set to None, the default host name and port are gonna be used which are "localhost:7783".

handle_client

this function is the default client handler suggested by networklib.
handle_client takes two arguments itself: the client socket which is going to be used to communicate to the clients connecting to the server, and a request analyser function that takes the requests sent from clients and returns the desired answer.
However, you might have noticed that according to the type hints in the start_server_mainloop taking two arguments is not allowed, so there is also a wrapper function called handle_client_wrapper that returns a function that works just like handle_client but takes just one argument which is the client socket and uses the request analyser passed into the wrapper as the second argument for handle_client.

analyse_request

analyse_request is the default request analyser suggested by networklib.
it takes three arguments (which are not allowed in the handle_client function but there is also a wrapper for that too): the request that came from client socket, the mass_lis and the spring_lis from massspring.
the handle client just accepts functions that take one argument and can not use this analyse_request function so there is a wrapper function called analyse_request_wrapper that takes two arguments, mass_lis and spring_lis, and returns a function that takes just one argument, the request, and uses analyse_request with the mass_lis and spring_lis passed to the wrapper function.

encode/decode mass/spring positions

these functions are the default functions used by analyse_request to encode/decode all the mass/spring positions into a bytes object/ list of positions which will be sent to clients and decoded by them.

examples

You can see a simple example of using massspring library to create a pendulum. It has an acceleration of -g in the Y ordinate. "pendulum"

available at example 1

Here is another example of using this library.
As you can see there is a double pendulum shown in the image. However, it is show in both x-y and z-y coordinates (x-y at left and z-y at right). you might feel they are completely different objects but they are just showing one object from two different points of view.

double pendulum

available at example 2.

3.example 3
4.example 4
5.example 5

Requirements

python >= 3.6
pygame >= 1.9.2
Note: if you want to hide the pygame support message, place this code before importing massspring.

import os
os.environ["PYGAME_HIDE_SUPPORT_PROMPT"] = '1'

TODO

(feel free to complete these tasks)

  • Add position checker to avoid integer too big error of pygame and python itself.
  • Update the Elastic collision.
  • divide massspring.py to separate modules. (won't be possible because inheritance levels forces each class to import all previous classes and it results in extreme usage of memory because for example mass class and force class will be imported in each file for each force class (spring, gravity, electricity, collision) and it's inefficient)
  • Divide the program to a server and a client, the server will do physical calculations and the client will recieve them and show them on screen (will be usefull when migrating to c/c++).
  • Divide the collision to Elastic collision and Inelastic collision.
  • Create non-sphere/non-mass objects.
  • Collision between mass and spring (mass: sphere with mass, spring: line without mass).
  • Collision between non-sphere objects, sphere objects and line objects(spring).
  • Add statistic screen.
  • Create the c/c++ project.

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

massspring-1.2.0.tar.gz (20.4 kB view hashes)

Uploaded Source

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page