1 #include <petsc.h> 2 #include <lua.h> 3 4 int lua_VecCreate(lua_State *L) 5 { 6 PetscErrorCode ierr; 7 Vec vec; 8 9 ierr = VecCreate(PETSC_COMM_SELF,&vec); 10 lua_pushlightuserdata (L,vec); 11 return 1; 12 } 13 14 int lua_VecSetSize(lua_State *L) 15 { 16 PetscErrorCode ierr; 17 Vec vec; 18 PetscInt n; 19 int isnum; 20 21 vec = (Vec) lua_touserdata(L,1); 22 n = (PetscInt) lua_tointegerx(L,2,&isnum); 23 ierr = VecSetSizes(vec,n,n); 24 return 0; 25 } 26 27 int luaopen_libpetsc(lua_State *L) 28 { 29 PetscInitializeNoArguments(); 30 lua_register(L,"VecCreate",(lua_CFunction)lua_VecCreate); 31 lua_register(L,"VecSetSize",(lua_CFunction)lua_VecSetSize); 32 return(0); 33 } 34