code refactorisation, and few new fonctionalities copied from pmpd3d

svn path=/trunk/externals/pmpd/; revision=17073

1 files changed, 473 insertions, 0 deletions

diff --git a/pmpd2d_various.c b/pmpd2d_various.c
new file mode 100644
index 0000000..26d01d1
--- /dev/null
+++ b/pmpd2d_various.c
@@ -0,0 +1,473 @@
+void pmpd2d_infosL(t_pmpd2d *x)
+{
+    int i;
+    post("list of mass");
+    post("number, Id, mobile, mass, damping, positionX Y, speedX Y, forcesX Y");
+    for(i=0; i < x->nb_mass; i++)
+    {
+        post("masse %i: %s, %d, %f, %f, %f, %f, %f, %f, %f, %f",i, x->mass[i].Id->s_name, \
+            x->mass[i].mobile, 1/x->mass[i].invM, x->mass[i].D2, x->mass[i].posX, x->mass[i].posY, \
+            x->mass[i].speedX, x->mass[i].speedY, x->mass[i].forceX, x->mass[i].forceY );
+    }
+
+    post("list of link");
+    post("number, Id, mass1, mass2, K, D, Pow, L, Lmin, Lmax");
+    for(i=0; i < x->nb_link; i++)
+    {
+        switch(x->link[i].lType)
+        {
+        case 0 :
+            post("link %i: %s, %i, %i, %f, %f, %f, %f, %f, %f", i, x->link[i].Id->s_name, \
+                x->link[i].mass1->num, x->link[i].mass2->num, x->link[i].K, x->link[i].D, \
+                x->link[i].Pow, x->link[i].L, x->link[i].Lmin, x->link[i].Lmax);
+            break;
+        case 1 :
+            post("tLink %i: %s, %i, %i, %f, %f, %f, %f, %f, %f, %f, %f", i, x->link[i].Id->s_name, \
+                x->link[i].mass1->num, x->link[i].mass2->num, x->link[i].K, x->link[i].D, \
+                x->link[i].Pow, x->link[i].L, x->link[i].Lmin, x->link[i].Lmax, x->link[i].VX, x->link[i].VY);
+            break;
+        case 2 :
+            post("tabLink %i: %s, %i, %i, %f, %f, %s, %f, %s, %f", i, x->link[i].Id->s_name, \
+                x->link[i].mass1->num, x->link[i].mass2->num, x->link[i].K, x->link[i].D, \
+                x->link[i].arrayK->s_name, x->link[i].K_L, x->link[i].arrayD->s_name, x->link[i].D_L);
+            break;
+        }
+    }
+}
+
+void pmpd2d_force(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+// add a force to a specific mass
+    int tmp, i;
+
+    if ( ( argv[0].a_type == A_FLOAT ) && ( argv[1].a_type == A_FLOAT ) && ( argv[2].a_type == A_FLOAT ) )
+    {
+        tmp = atom_getfloatarg(0, argc, argv);
+        tmp = max(0, min( x->nb_mass-1, tmp));
+        x->mass[tmp].forceX += atom_getfloatarg(1, argc, argv);
+        x->mass[tmp].forceY += atom_getfloatarg(2, argc, argv);
+    }
+    if ( ( argv[0].a_type == A_SYMBOL ) && ( argv[1].a_type == A_FLOAT ) && ( argv[2].a_type == A_FLOAT ) )
+    {
+        for (i=0; i< x->nb_mass; i++)
+        {
+            if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+            {
+                x->mass[i].forceX += atom_getfloatarg(1, argc, argv);
+                x->mass[i].forceY += atom_getfloatarg(2, argc, argv);
+            }
+        }
+    }
+}
+
+void pmpd2d_forceX(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+// add a force to a specific mass
+    int tmp, i;
+
+    if ( ( argv[0].a_type == A_FLOAT ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        tmp = atom_getfloatarg(0, argc, argv);
+        tmp = max(0, min( x->nb_mass-1, tmp));
+        x->mass[tmp].forceX += atom_getfloatarg(1, argc, argv);
+    }
+    if ( ( argv[0].a_type == A_SYMBOL ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        for (i=0; i< x->nb_mass; i++)
+        {
+            if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+            {
+                x->mass[i].forceX += atom_getfloatarg(1, argc, argv);
+            }
+        }
+    }
+}
+
+void pmpd2d_forceY(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+// add a force to a specific mass
+    int tmp, i;
+
+    if ( ( argv[0].a_type == A_FLOAT ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        tmp = atom_getfloatarg(0, argc, argv);
+        tmp = max(0, min( x->nb_mass-1, tmp));
+        x->mass[tmp].forceY += atom_getfloatarg(1, argc, argv);
+    }
+    if ( ( argv[0].a_type == A_SYMBOL ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        for (i=0; i< x->nb_mass; i++)
+        {
+            if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+            {
+                x->mass[i].forceY += atom_getfloatarg(1, argc, argv);
+            }
+        }
+    }
+}
+
+void pmpd2d_min(t_pmpd2d *x, t_float minX, t_float minY)
+{
+    x->minX = minX;
+    x->minY = minY;
+}
+
+void pmpd2d_max(t_pmpd2d *x, t_float maxX, t_float maxY)
+{
+    x->maxX = maxX;
+    x->maxY = maxY;
+}
+
+void pmpd2d_minX(t_pmpd2d *x, t_float min)
+{
+    x->minX = min;
+}
+
+void pmpd2d_maxX(t_pmpd2d *x, t_float max)
+{
+    x->maxX = max;
+}
+
+void pmpd2d_minY(t_pmpd2d *x, t_float min)
+{
+    x->minY = min;
+}
+
+void pmpd2d_maxY(t_pmpd2d *x, t_float max)
+{
+    x->maxY = max;
+}
+
+void pmpd2d_addPos(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+    int tmp, i;
+
+    if ( ( argv[0].a_type == A_FLOAT ) && ( argv[1].a_type == A_FLOAT ) && ( argv[2].a_type == A_FLOAT ) )
+    {
+        tmp = atom_getfloatarg(0, argc, argv);
+        tmp = max(0, min( x->nb_mass-1, tmp));
+        x->mass[tmp].posX += atom_getfloatarg(1, argc, argv);
+        x->mass[tmp].posY += atom_getfloatarg(2, argc, argv);
+    }
+    if ( ( argv[0].a_type == A_SYMBOL ) && ( argv[1].a_type == A_FLOAT ) && ( argv[2].a_type == A_FLOAT ) )
+    {
+        for (i=0; i< x->nb_mass; i++)
+        {
+            if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+            {
+                x->mass[i].posX += atom_getfloatarg(1, argc, argv);
+                x->mass[i].posY += atom_getfloatarg(2, argc, argv);
+            }
+        }
+    }
+}
+
+void pmpd2d_addPosX(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+    int tmp, i;
+
+    if ( ( argv[0].a_type == A_FLOAT ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        tmp = atom_getfloatarg(0, argc, argv);
+        tmp = max(0, min( x->nb_mass-1, tmp));
+        x->mass[tmp].posX += atom_getfloatarg(1, argc, argv);
+    }
+    if ( ( argv[0].a_type == A_SYMBOL ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        for (i=0; i< x->nb_mass; i++)
+        {
+            if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+            {
+                x->mass[i].posX += atom_getfloatarg(1, argc, argv);
+            }
+        }
+    }
+}
+
+void pmpd2d_addPosY(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+    int tmp, i;
+
+    if ( ( argv[0].a_type == A_FLOAT ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        tmp = atom_getfloatarg(0, argc, argv);
+        tmp = max(0, min( x->nb_mass-1, tmp));
+        x->mass[tmp].posY += atom_getfloatarg(1, argc, argv);
+    }
+    if ( ( argv[0].a_type == A_SYMBOL ) && ( argv[1].a_type == A_FLOAT ) )
+    {
+        for (i=0; i< x->nb_mass; i++)
+        {
+            if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+            {
+                x->mass[i].posY += atom_getfloatarg(1, argc, argv);
+            }
+        }
+    }
+}
+
+void pmpd2d_grabMass(t_pmpd2d *x, t_float posX, t_float posY, t_float grab)
+{
+    t_float dist, tmp;
+    t_int i;
+    
+    if (grab == 0)
+        x->grab=0;
+    if ((x->grab == 0)&(grab == 1)&(x->nb_mass > 0))
+    {
+        x->grab=1;
+        x->grab_nb= 0;
+        dist = sqr(x->mass[0].posX - posX) + sqr(x->mass[0].posY - posY);
+        for (i=1; i<x->nb_mass; i++)
+        {
+            tmp = sqr(x->mass[i].posX - posX) + sqr(x->mass[i].posY - posY);
+            if (tmp < dist)
+            {
+                dist = tmp;
+                x->grab_nb= i;
+            }
+        }
+    }
+    if (x->grab == 1)
+    {
+        x->mass[x->grab_nb].posX = posX;
+        x->mass[x->grab_nb].posY = posY;        
+    }
+}
+
+void pmpd2d_closestMass(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)// t_float posX, t_float posY)
+{
+    t_float dist, tmp;
+    t_int i;
+    t_atom std_out[3];
+    
+    t_float posX, posY;
+    posX = atom_getfloatarg(0, argc, argv);
+    posY = atom_getfloatarg(1, argc, argv);    
+    if ( (argc == 3)  && (argv[2].a_type == A_SYMBOL) )
+    {
+        //t_symbol *mass_name = atom_getsymbolarg(2, argc, argv);
+        if ((x->nb_mass > 0))
+        {
+            dist = 1000000000;//sqr(x->mass[0].posX - posX) + sqr(x->mass[0].posY - posY);
+            for (i=0; i<x->nb_mass; i++)
+            {
+                if (atom_getsymbolarg(2,argc,argv) == x->mass[i].Id)
+                {
+                    tmp = sqr(x->mass[i].posX - posX) + sqr(x->mass[i].posY - posY);
+                    if (tmp < dist)
+                    {
+                        dist = tmp;
+                        x->grab_nb= i;
+                    }
+                }
+            }
+        }
+    }
+    else {
+        if ((x->nb_mass > 0))
+        {
+            dist = sqr(x->mass[0].posX - posX) + sqr(x->mass[0].posY - posY);
+            for (i=1; i<x->nb_mass; i++)
+            {
+                tmp = sqr(x->mass[i].posX - posX) + sqr(x->mass[i].posY - posY);
+                if (tmp < dist)
+                {
+                    dist = tmp;
+                    x->grab_nb= i;
+                }
+            }
+        }
+    }
+    
+        
+    SETFLOAT(&(std_out[0]),x->grab_nb);
+    SETFLOAT(&(std_out[1]), x->mass[x->grab_nb].posX);
+    SETFLOAT(&(std_out[2]), x->mass[x->grab_nb].posY);
+    outlet_anything(x->main_outlet, gensym("closestMass"),3,std_out);
+}
+
+void pmpd2d_massDistances_f_f(t_pmpd2d *x, t_int i, t_int j)
+{
+	t_float dist, tmp;
+	t_atom to_out[3];
+
+	tmp = x->mass[i].posX - x->mass[j].posX;
+	dist = sqr(tmp);
+	tmp = x->mass[i].posY - x->mass[j].posY;
+	dist += sqr(tmp);
+	dist = sqrt(dist);
+
+	SETFLOAT(&(to_out[0]), i);
+	SETFLOAT(&(to_out[1]), j);
+	SETFLOAT(&(to_out[2]), dist);
+	outlet_anything(x->main_outlet, gensym("distance"), 3, to_out);
+}
+
+void pmpd2d_massDistances(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+	t_int i,j;
+
+	if ( (argc == 2) && (argv[0].a_type == A_FLOAT) && (argv[1].a_type == A_FLOAT) )
+	{
+		pmpd2d_massesDistances_f_f(x, atom_getfloatarg(0, argc, argv), atom_getfloatarg(1, argc, argv));
+	}
+	else if ( (argc == 2) && (argv[0].a_type == A_FLOAT) && (argv[1].a_type == A_SYMBOL) )
+	{
+		for (i=0; i < x->nb_mass; i++)
+		{
+			if ( atom_getsymbolarg(1,argc,argv) == x->mass[i].Id)
+			{
+				pmpd2d_massesDistances_f_f(x, atom_getfloatarg(0, argc, argv), i);
+			}
+		}
+	}
+	else if ( (argc == 2) && (argv[0].a_type == A_SYMBOL) && (argv[1].a_type == A_FLOAT) )
+	{
+		for (i=0; i < x->nb_mass; i++)
+		{
+			if ( (atom_getsymbolarg(0,argc,argv) == x->mass[i].Id) )
+			{
+				pmpd2d_massesDistances_f_f(x, atom_getfloatarg(1, argc, argv), i);
+			}
+		}
+	}
+	else if ( (argc == 2) && (argv[0].a_type == A_SYMBOL) && (argv[1].a_type == A_SYMBOL) )
+	{
+		for (i=0; i < x->nb_mass; i++)
+		{
+			if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+			{
+				for (j=i+1; j < x->nb_mass; j++)
+				{
+					if ( atom_getsymbolarg(1,argc,argv) == x->mass[j].Id)
+					{
+						pmpd2d_massesDistances_f_f(x,i, j);
+					}
+				}
+			}
+		}
+	}
+	else if (argc == 0)
+	{
+		for (i=0; i < x->nb_mass; i++)
+		{
+			for (j=i+1; j < x->nb_mass; j++)
+			{
+				pmpd2d_massesDistances_f_f(x,i, j);
+			}
+		}
+	}
+	else if ((argc == 1) && (argv[0].a_type == A_SYMBOL) )
+	{
+		for (i=0; i < x->nb_mass; i++)
+		{
+			if ( atom_getsymbolarg(0,argc,argv) == x->mass[i].Id)
+			{
+				for (j=i+1; j < x->nb_mass; j++)
+				{
+					pmpd2d_massesDistances_f_f(x,i, j);
+				}
+			}
+		}
+	}
+	else if ( (argc == 1) && (argv[0].a_type == A_FLOAT) )
+	{
+		for (i=0; i < x->nb_mass; i++)
+		{
+			pmpd2d_massesDistances_f_f(x, atom_getfloatarg(0, argc, argv), i);
+		}
+	}
+}
+
+void pmpd2d_forcesXT(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+// add forces to masses. forces comes from a table, masse can be filter on ther Id or not
+
+	t_int i, j;
+	t_garray *a;
+	int npoints;
+	t_word *vec;
+	t_symbol *array;
+
+	if ( argv[0].a_type == A_SYMBOL )
+	{
+		array = atom_getsymbolarg(0,argc,argv);
+		if (!(a = (t_garray *)pd_findbyclass(array, garray_class)))
+			pd_error(x, "%s: no such array", array->s_name);
+		else if (!garray_getfloatwords(a, &npoints, &vec))
+			pd_error(x, "%s: bad template for tabLink", array->s_name);
+		else
+		{
+			if ( argc == 1 )  // apply forces on all masses
+			{
+				j = min(x->nb_mass, npoints);
+				for (i=0; i< j; i++)
+				{
+					x->mass[i].forceX += vec[i].w_float;
+				}
+			}
+			else if (( argc > 1 ) && ( argv[1].a_type == A_SYMBOL ))
+			{
+				i=0;
+				j=0;
+
+				while ( (i < x->nb_mass) && (j < npoints) )
+				{
+					if ( atom_getsymbolarg(1,argc,argv) == x->mass[i].Id)
+					{
+						x->mass[i].forceX += vec[j].w_float;
+						j++;
+					}
+					i++;
+				}
+			}
+		}
+	}
+}
+
+void pmpd2d_forcesYT(t_pmpd2d *x, t_symbol *s, int argc, t_atom *argv)
+{
+// add forces to masses. forces comes from a table, masse can be filter on ther Id or not
+
+	t_int i, j;
+	t_garray *a;
+	int npoints;
+	t_word *vec;
+	t_symbol *array;
+
+	if ( argv[0].a_type == A_SYMBOL )
+	{
+		array = atom_getsymbolarg(0,argc,argv);
+		if (!(a = (t_garray *)pd_findbyclass(array, garray_class)))
+			pd_error(x, "%s: no such array", array->s_name);
+		else if (!garray_getfloatwords(a, &npoints, &vec))
+			pd_error(x, "%s: bad template for tabLink", array->s_name);
+		else
+		{
+			if ( argc == 1 )  // apply forces on all masses
+			{
+				j = min(x->nb_mass, npoints);
+				for (i=0; i< j; i++)
+				{
+					x->mass[i].forceY += vec[i].w_float;
+				}
+			}
+			else if (( argc > 1 ) && (argv[1].a_type == A_SYMBOL ))
+			{
+				i=0;
+				j=0;
+
+				while ( (i < x->nb_mass) && (j < npoints) )
+				{
+					if ( atom_getsymbolarg(1,argc,argv) == x->mass[i].Id)
+					{
+						x->mass[i].forceY += vec[j].w_float;
+						j++;
+					}
+					i++;
+				}
+			}
+		}
+	}
+}