Adding basic NIF template

This commit is contained in:
Dave Smith 2010-03-25 22:32:53 -06:00
parent 84f5108e17
commit b2cfdcb34a
3 changed files with 90 additions and 0 deletions

51
priv/templates/basicnif.c Normal file
View file

@ -0,0 +1,51 @@
#include "erl_nif.h"
static ErlNifResourceType* {{module}}_RESOURCE;
typedef struct
{
} {{module}}_handle;
// Prototypes
ERL_NIF_TERM {{module}}_new(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]);
ERL_NIF_TERM {{module}}_myfunction(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[]);
static ErlNifFunc nif_funcs[] =
{
{"new", 0, {{module}}_new},
{"myfunction", 1, {{module}}_myfunction}
};
ERL_NIF_TERM {{module}}_new(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
{{module}}_handle* handle = enif_alloc_resource(env,
{{module}}_RESOURCE,
sizeof({{module}}_handle));
ERL_NIF_TERM result = enif_make_resource(env, handle);
enif_release_resource(env, handle);
return enif_make_tuple2(env, enif_make_atom(env, "ok"), result);
}
ERL_NIF_TERM {{module}}_myfunction(ErlNifEnv* env, int argc, const ERL_NIF_TERM argv[])
{
return enif_make_atom(env, "ok");
}
static void {{module}}_resource_cleanup(ErlNifEnv* env, void* arg)
{
// Delete any dynamically allocated memory stored in {{module}}_handle
// {{module}}_handle* handle = ({{module}}_handle*)arg;
}
static int on_load(ErlNifEnv* env, void** priv_data, ERL_NIF_TERM load_info)
{
{{module}}_RESOURCE = enif_open_resource_type(env, "{{module}}_resource",
&{{module}}_resource_cleanup,
ERL_NIF_RT_CREATE | ERL_NIF_RT_TAKEOVER,
0);
return 0;
}
ERL_NIF_INIT({{module}}, nif_funcs, &on_load, NULL, NULL, NULL);

View file

@ -0,0 +1,36 @@
-module({{module}}).
-export([new/0,
myfunction/1]).
-on_load(init/0).
-ifdef(TEST).
-include_lib("eunit/include/eunit.hrl").
-endif.
init() ->
case code:priv_dir({{module}}) of
{error, bad_name} ->
SoName = filename:join("../priv", {{module}});
Dir ->
SoName = filename:join(Dir, {{module}})
end,
erlang:load_nif(SoName, 0).
new() ->
"NIF library not loaded".
myfunction(Ref) ->
"NIF library not loaded".
%% ===================================================================
%% EUnit tests
%% ===================================================================
-ifdef(TEST).
basic_test() ->
{ok, Ref} = new(),
ok = myfunction(Ref).
-endif.

View file

@ -0,0 +1,3 @@
{variables, [{module, "mymodule"}]}.
{template, "basicnif.erl", "src/{{module}}.erl"}.
{template, "basicnif.c", "c_src/{{module}}.c"}.