picrin/src/dict.c

/**
 * See Copyright Notice in picrin.h
 */

#include "picrin.h"
#include "picrin/dict.h"

static pic_value
pic_dict_dict(pic_state *pic)
{
  struct pic_dict *dict;

  pic_get_args(pic, "");

  dict = (struct pic_dict *)pic_obj_alloc(pic, sizeof(struct pic_dict), PIC_TT_DICT);

  xh_init_int(&dict->hash, sizeof(pic_value));

  return pic_obj_value(dict);
}

static pic_value
pic_dict_dict_p(pic_state *pic)
{
  pic_value obj;

  pic_get_args(pic, "o", &obj);

  return pic_bool_value(pic_dict_p(obj));
}

static pic_value
pic_dict_dict_ref(pic_state *pic)
{
  struct pic_dict *dict;
  pic_sym key;
  xh_entry *e;

  pic_get_args(pic, "dm", &dict, &key);

  e = xh_get_int(&dict->hash, key);
  if (! e) {
    return pic_none_value();
  }
  return xh_val(e, pic_value);
}

static pic_value
pic_dict_dict_set(pic_state *pic)
{
  struct pic_dict *dict;
  pic_sym key;
  pic_value val;

  pic_get_args(pic, "dmo", &dict, &key, &val);

  xh_put_int(&dict->hash, key, &val);

  return pic_none_value();
}

static pic_value
pic_dict_dict_del(pic_state *pic)
{
  struct pic_dict *dict;
  pic_sym key;

  pic_get_args(pic, "dm", &dict, &key);

  xh_del_int(&dict->hash, key);

  return pic_none_value();
}

void
pic_init_dict(pic_state *pic)
{
  pic_deflibrary ("(picrin dictionary)") {
    pic_defun(pic, "dictionary", pic_dict_dict);
    pic_defun(pic, "dictionary?", pic_dict_dict_p);
    pic_defun(pic, "dictionary-ref", pic_dict_dict_ref);
    pic_defun(pic, "dictionary-set!", pic_dict_dict_set);
    pic_defun(pic, "dictionary-delete", pic_dict_dict_del);
  }
}