shithub: pdffs

ref: 1d93500ddcda77cd265d492c6c9094c0f5c7488f
dir: pdffs/dict.c

#include <u.h>
#include <libc.h>
#include <bio.h>
#include "pdf.h"

/* 7.3.7 Dictionary Objects */

Object *
pdfdict(Pdf *pdf, Biobuf *b)
{
	Object *o, *k, *v;
	KeyValue *kv;
	int c, nokey;

	/* skip '<<' */
	Bseek(b, 2, 1);

	k = v = nil;
	o = calloc(1, sizeof(*o));
	o->type = Odict;
	o->pdf = pdf;
	for(nokey = 0;;){
		if((c = Bgetc(b)) < 0)
			goto err;
		if(c == '>'){
			if(Bgetc(b) == '>')
				break;
			werrstr("no '>>'");
			goto err;
		}
		if(nokey){
			werrstr("no '>>'");
			goto err;
		}

		Bungetc(b);
		if((k = pdfobj(pdf, b)) == nil){
			nokey = 1;
			continue;
		}
		if(k->type != Oname){
			werrstr("expected name as a key");
			goto err;
		}
		if((v = pdfobj(pdf, b)) == nil)
			goto err;

		if((kv = realloc(o->dict.kv, (o->dict.nkv+1)*sizeof(KeyValue))) == nil)
			goto err;

		o->dict.kv = kv;
		kv[o->dict.nkv].key = strdup(k->name);
		pdfobjfree(k);
		kv[o->dict.nkv++].value = v;
		k = v = nil;
	}

	return o;
err:
	pdfobjfree(o);
	pdfobjfree(k);
	pdfobjfree(v);
	werrstr("dict: %r");

	return nil;
}

Object *
dictget(Object *o, char *name)
{
	int i;

	o = pdfeval(o);
	if((o->type != Ostream && o->type != Odict) || name == nil)
		return &null;
	for(i = 0; i < o->dict.nkv && strcmp(name, o->dict.kv[i].key) != 0; i++);

	return pdfeval(i < o->dict.nkv ? o->dict.kv[i].value : &null);
}

vlong
dictint(Object *o, char *name)
{
	o = dictget(o, name);
	return o->type == Onum ? o->num : 0;
}

char *
dictstring(Object *o, char *name)
{
	o = dictget(o, name);
	return o->type == Ostr ? o->str : "";
}

Object *
dictdict(Object *o, char *name)
{
	o = dictget(o, name);
	return o->type == Odict ? o : &null;
}

int
dictints(Object *o, char *name, int *el, int nel)
{
	int n;
	Object *v;

	o = dictget(o, name);
	if(o->type != Oarray){
		werrstr("not an array");
		return -1;
	}

	for(n = 0; n < o->array.ne && n < nel; n++){
		if((v = o->array.e[n])->type != Onum){
			werrstr("not an integer array");
			return -1;
		}
		el[n] = v->num;
	}

	return n;
}