#ifndef STAN_SPRING_NETWORK_H
#define STAN_SPRING_NETWORK_H

#include "vec3n.h"

#define SPRING_STRUCT (0)
#define SPRING_SHEAR (1)
#define SPRING_BEND (2)

class SpringNetwork
{
public:
	class Spring
	{
	public:
		int type;  // index into coefficients spring_k[]
		float restlen;
		int a, b;      // spring endpoints vector indices
		int iab, iba;  // indices into off-diagonal blocks of sparse matrix
		Spring() {}
		Spring(int _type, int _a, int _b, float _restlen) : type(_type), a(_a), b(_b), restlen(_restlen) { iab = iba = -1; }
	};
	Array<Spring> springs;
	float3N X;        // positions of all points
	float3N V;        // velocities
	float3N F;        // force on each point
	float3N dV;       // change in velocity
	float3Nx3N A;     // big matrix we solve system with
	float3Nx3N dFdX;  // big matrix of derivative of force wrt position
	float3Nx3N dFdV;  // big matrix of derivative of force wrt velocity
	float3Nx3N S;     // used for our constraints - contains only some diagonal blocks as needed S[i,i]
	int awake;
	float3 bmin, bmax;
	union {
		struct
		{
			float spring_struct;
			float spring_shear;
			float spring_bend;
		};
		float spring_k[3];
	};
	float spring_damp;
	float spring_air;
	float cloth_step;  // delta time for cloth
	float3 cloth_gravity;
	float cloth_sleepthreshold;
	int cloth_sleepcount;

	SpringNetwork(int _n);
	Spring &AddBlocks(Spring &s);
	Spring &CreateSpring(int type, int a, int b, float restlen) { return AddBlocks(springs.Add(Spring(type, a, b, restlen))); }
	Spring &CreateSpring(int type, int a, int b) { return CreateSpring(type, a, b, magnitude(X[b] - X[a])); }
	void UpdateLimits() { BoxLimits(X.element, X.count, bmin, bmax); }
	void Wake() { awake = cloth_sleepcount; }
	void Simulate(float dt);
	void PreSolveSpring(const Spring &s);
	void CalcForces();
};

#endif  //STAN_SPRING_NETWORK_H