comparison mercurial/cext/revlog.c @ 39072:34eb999e29bf

index: make capacity argument to nt_init be measured in revisions

The nodetree's internal capacity field is measures in nodetree nodes,
which is not something the caller should have to know about.

Differential Revision: https://phab.mercurial-scm.org/D4166

39071:06ff7ea4f440

}

static int nt_init(nodetree *self, indexObject *index, unsigned capacity)
{
	self->index = index;
	self->capacity = capacity;
	self->depth = 0;
	self->splits = 0;
	if ((size_t)self->capacity > INT_MAX / sizeof(nodetreenode)) {
		PyErr_SetString(PyExc_ValueError, "overflow in init_nt");
		return -1;

		self->nt = PyMem_Malloc(sizeof(nodetree));
		if (self->nt == NULL) {
			PyErr_NoMemory();
			return -1;
		}
		unsigned capacity = (self->raw_length < 4 ? 4 : (int)self->raw_length / 2);
		if (nt_init(self->nt, self, capacity) == -1) {
			PyMem_Free(self->nt);
			self->nt = NULL;
			return -1;
		}
		if (nt_insert(self->nt, nullid, -1) == -1) {

39072:34eb999e29bf

}

static int nt_init(nodetree *self, indexObject *index, unsigned capacity)
{
	self->index = index;
	/* The input capacity is in terms of revisions, while the field is in
	 * terms of nodetree nodes. */
	self->capacity = (capacity < 4 ? 4 : capacity / 2);
	self->depth = 0;
	self->splits = 0;
	if ((size_t)self->capacity > INT_MAX / sizeof(nodetreenode)) {
		PyErr_SetString(PyExc_ValueError, "overflow in init_nt");
		return -1;

		self->nt = PyMem_Malloc(sizeof(nodetree));
		if (self->nt == NULL) {
			PyErr_NoMemory();
			return -1;
		}
		if (nt_init(self->nt, self, self->raw_length) == -1) {
			PyMem_Free(self->nt);
			self->nt = NULL;
			return -1;
		}
		if (nt_insert(self->nt, nullid, -1) == -1) {