manifold_spine.cpp
8bce407ab5e2efficient_neuronet

manifold_spine.cpp
View Options

			#include "manifold_spine.h"

			#include <math.h>

			using namespace LAMMPS_NS;
			using namespace user_manifold;

			manifold_spine::manifold_spine( LAMMPS lmp, int argc, char *argv ) : manifold(lmp)
			{}


			/*
			* Equation for spine is:
			*
			* -x^2 - y^2 + (a^2 - (z/c)^2)( 1 + Asin(const(z) *z^2) )^4
			* const(z) = (z < 0) ? B2 : B
			* params[5] = { a, A, B, B2, c }
			*/


			double manifold_spine::g( const double *x )
			{
			double a = params[0];
			double A = params[1];
			double B = params[2];
			double B2 = params[3];
			double c = params[4];
			if (x[2]>0){
			return -1.0(x[0]x[0]+x[1]x[1])+(aa-x[2]x[2]/(cc))(1.0+pow(Asin(B x[2]x[2]),4));
			}else{
			return -1.0(x[0]x[0]+x[1]x[1])+(aa-x[2]x[2]) (1.0+pow(Asin(B2x[2]*x[2]),4));
			}
			}

			void manifold_spine::n( const double x, double nn )
			{
			double a = params[0];
			double A = params[1];
			double B = params[2];
			double B2 = params[3];
			double c = params[4];
			if (x[2]>0){
			nn[0] = -2.0*x[0];
			nn[1] = -2.0*x[1];
			nn[2] = 8.0AAAABx[2](aa-(x[2]x[2]/(cc)))cos(Bx[2]x[2])
			pow(sin(Bx[2]x[2]),3)-2x[2](1+pow(Asin(Bx[2]x[2]),4))/(cc);
			}else{
			nn[0] = -2.0*x[0];
			nn[1] = -2.0*x[1];
			nn[2] = 8.0AAAAB2x[2](aa-x[2]x[2])cos(B2x[2]x[2]) *
			pow(sin(B2x[2]x[2]),3)-2x[2](1+pow(Asin(B2x[2]*x[2]),4));
			}
			}