oidset.con commit general improvements (43abf13)
   1#include "cache.h"
   2#include "oidset.h"
   3
   4void oidset_init(struct oidset *set, size_t initial_size)
   5{
   6        memset(&set->set, 0, sizeof(set->set));
   7        if (initial_size)
   8                kh_resize_oid_set(&set->set, initial_size);
   9}
  10
  11int oidset_contains(const struct oidset *set, const struct object_id *oid)
  12{
  13        khiter_t pos = kh_get_oid_set(&set->set, *oid);
  14        return pos != kh_end(&set->set);
  15}
  16
  17int oidset_insert(struct oidset *set, const struct object_id *oid)
  18{
  19        int added;
  20        kh_put_oid_set(&set->set, *oid, &added);
  21        return !added;
  22}
  23
  24int oidset_remove(struct oidset *set, const struct object_id *oid)
  25{
  26        khiter_t pos = kh_get_oid_set(&set->set, *oid);
  27        if (pos == kh_end(&set->set))
  28                return 0;
  29        kh_del_oid_set(&set->set, pos);
  30        return 1;
  31}
  32
  33void oidset_clear(struct oidset *set)
  34{
  35        kh_release_oid_set(&set->set);
  36        oidset_init(set, 0);
  37}
  38
  39void oidset_parse_file(struct oidset *set, const char *path)
  40{
  41        FILE *fp;
  42        struct strbuf sb = STRBUF_INIT;
  43        struct object_id oid;
  44
  45        fp = fopen(path, "r");
  46        if (!fp)
  47                die("could not open object name list: %s", path);
  48        while (!strbuf_getline(&sb, fp)) {
  49                const char *p;
  50                const char *name;
  51
  52                /*
  53                 * Allow trailing comments, leading whitespace
  54                 * (including before commits), and empty or whitespace
  55                 * only lines.
  56                 */
  57                name = strchr(sb.buf, '#');
  58                if (name)
  59                        strbuf_setlen(&sb, name - sb.buf);
  60                strbuf_trim(&sb);
  61                if (!sb.len)
  62                        continue;
  63
  64                if (parse_oid_hex(sb.buf, &oid, &p) || *p != '\0')
  65                        die("invalid object name: %s", sb.buf);
  66                oidset_insert(set, &oid);
  67        }
  68        if (ferror(fp))
  69                die_errno("Could not read '%s'", path);
  70        fclose(fp);
  71        strbuf_release(&sb);
  72}