From patchwork Mon Feb 5 05:27:23 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Connor Worley X-Patchwork-Id: 46031 Delivered-To: ffmpegpatchwork2@gmail.com Received: by 2002:a05:6a20:7b08:b0:19e:8a94:b663 with SMTP id s8csp554821pzh; Sun, 4 Feb 2024 21:27:44 -0800 (PST) X-Google-Smtp-Source: AGHT+IHzdpwjqv4l0lJQBZX1HJvbf2jXdWFY/FssBl7jq6zTniz+hfdZYis0m438bvY4dEw249Mv X-Received: by 2002:a05:6402:5248:b0:560:1372:d950 with SMTP id t8-20020a056402524800b005601372d950mr4188148edd.22.1707110864021; Sun, 04 Feb 2024 21:27:44 -0800 (PST) ARC-Seal: i=1; a=rsa-sha256; t=1707110864; cv=none; d=google.com; s=arc-20160816; b=HRrJo3FXGMsJnC+rUkBWkFOemcuzggEPK71vbgMmj7zYqkGOjL6BpUFDrAoKJRcid6 5Do9TwR8H8E4GfZOTWtSCuBVCfEIJ8te/751PiuorNQK8GXzKzJTf3OI5QwQGi9XhwNF dwOCRppDBbVgwiqnrr3/Fa+eF57mmeDhGjJfXpbjDMEg3NC9c352ZgbpBWd/nPjjTJaX pM1AZVI0lKenfKxz3epa4W48ZALllnWcO4EnLz3mGmKXD0/pB991jPpilG3qcAg5BTD2 FEyUso1wHJME6reMxKXqP75Dgcb8JPoPw5YlHYMySu4rzV/huKbDlc00DIQmrrCq6gcO Q63A== ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=arc-20160816; h=sender:errors-to:content-transfer-encoding:cc:reply-to :list-subscribe:list-help:list-post:list-archive:list-unsubscribe :list-id:precedence:subject:mime-version:message-id:date:to:from :dkim-signature:delivered-to; bh=13H0R0lexjs3m5BP8NTnfYSPVSSuuU8QfpNb3VCbQsM=; fh=kYW042ngM5tVe9IKlnCKT4MHISkmaePOeqBv/FOOCB8=; b=jtUJKFS+DImG5969KD4bVksXoDGKmjJFVA3Z6l4tKtkLqmyNmxB5L7nd7+9Vp/fZNM Qw/TVK1et013/O76ViMq0gxfIcnFY9F8gE2nXREPuVVEa3cyn8Yp9HyUz13A+BFOXFmZ D+yfRQ4VZnIWd6MRLrGS0zJ3HRG7/uZlOw1FD1oBvIjGIEAuqS7pK4wspBX8mesRBNFl GNsH9vq25W7Vnqe+EowHoO9qx9QBuopWJWKcdBTn8+SApO3r+0CVE/VhHAzSaQVZCcJ1 GeaxPWObFW1s4CZX77Wh3LvgP7gXaauNIWsirnkaW6mJwNOUadd/fRC5I8GmdE6N3tnJ PyTg==; dara=google.com ARC-Authentication-Results: i=1; mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20230601 header.b=PwP6rbP4; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=gmail.com X-Forwarded-Encrypted: i=0; AJvYcCVDUXi2B/iz4RNisVpRK/KQA1iammYaeoUbWkNuRDjIAZmd5GvSfRtIOxKJTv17woXRIPVmsorsKSurPzGcsqsl4WmC119Zyb8Nte3bukqh0fOpSSiJSTR4bytS2l49w/GvBhDxg9fgx+XMuJR+RuR8juNN1T8/wJKjCO4QXyzN9qq07f7lwnJVIY3gffzwqqiX3Q9zgZllZmXhRHa3pS1zdVyJ1TiF0+eydiwjEMcZ0LA0HLNrx0ZlDrfldu3qiNfjWuRKFNQFsFGj/BXfii/Z0j+YzgKrgyho0v7fO+CPotSej1sG73+SCwPbczKziq+rIcJnxk5GqTYxE/LwvOZVLivMIP3krzwgv6RV1HNwm7qUiX+8j0LCiaMYhPssp8gCjUo1SXukNwZSy8ICGS+dVhz63LvLoaBdQs9QsWGik8soNnZJr6DAPwIl+E7M27szZilGOiBxaK1/BdKpkibtuJObAWKrXClzb5di3huONeiZIV9Qv6w5wWjlCNltAsvwJV8YJW8/3FWJ2/212whzdBdj/GVAO/MOJWuFp5dW10reIikezFrwwAvIYqGqkQXpDj4jtH6q+QgakB01Rl78JLQBjegplidrMqeHXH6IKq5Sp7E/ueoS715uScXhn0P9akgeG7IfiCvWBi0RP/chMK0TJNgpgCcXJKPxS0nvU+ENaQWmTKAduGuHLDIhoxjKfr0eA7hs2ymdrxUKDJ61c4SEyiwoGaxDs4nFIKA1gFwwjQcDZ15VHcMbnSn3wh+jlxykVPRfiIA14XX1AYjAIb14dabOFG2471m9I9CCpF1Jq050zwnmtJ2D35mkE4BiDZDenH4HPtVb6VJLMmW0gPVWkJvqEzh6Dvbw1KB9Zc9OEFrBEIxYDG2hRC3pc4TOkKSqeazT28B+zQdMvVuPlEmdc5CfQ+d6c+cAmP2rowJYf0Yg57Yzz1E8Bqh26+VpO9 D1uHTUWtZKTMXj6gSVYzGXApQCol54zbRQiaAaZ4aWY+7WKTM+7OSYrGNF3nVCbhdThNijTbyM+e4KGjMvpEXssYaMs1RGHqS1oqHx576caGnstiAX6KChSTEE/14jx14RvTinpJmMqiaUDXJ9EXjX8LXNfEoK6Nsm17SvygGlW28qjFGobItjE2/EtGYwQw1sYSTv+s7QWND5LLtDaEelACOSee/vMLQ425SUTA8aYmD+nE+MMEvbiDq2PlUIVo+D47H8b5zkj6qdqUjYDRkuDJptK+pslh1U+DCZhrzItLkffIiFhhyPpvKYFFirIsNnnNbPbJzNyy0dNif3ogvh/DHugYdZWs9JWsi/zoy7tKrYF9EjUa33mXAmh6m4TED6PnQU+EgIIY++xlHMLJOSYDImPjsUr277YcPGPQyDNv3NXswwmUIGAXieCkxlNICLFgqzdh+FpPdmSksXui1zTuZWY3X7yChC+Z6QrWvZJO3sYXq6FHKCFeKpd2vZoyttCPJj829i6TtXqxKoCsr2RzIMG5uOoHANNGgYEv1pgybZTJYoVhYfPacsIxpPBhU3e83eaANZ18IJ/bdPeKG8J9gX4ypTQRbrs+175/mAcHjd7CbEMaqq1ON5Ap8HczUomZcd8E15KWZ6ug== Return-Path: Received: from ffbox0-bg.mplayerhq.hu (ffbox0-bg.ffmpeg.org. [79.124.17.100]) by mx.google.com with ESMTP id x4-20020a05640226c400b005605adde36fsi1050117edd.616.2024.02.04.21.27.43; Sun, 04 Feb 2024 21:27:44 -0800 (PST) Received-SPF: pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) client-ip=79.124.17.100; Authentication-Results: mx.google.com; dkim=neutral (body hash did not verify) header.i=@gmail.com header.s=20230601 header.b=PwP6rbP4; spf=pass (google.com: domain of ffmpeg-devel-bounces@ffmpeg.org designates 79.124.17.100 as permitted sender) smtp.mailfrom=ffmpeg-devel-bounces@ffmpeg.org; dmarc=fail (p=NONE sp=QUARANTINE dis=NONE) header.from=gmail.com Received: from [127.0.1.1] (localhost [127.0.0.1]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTP id C35806800C1; Mon, 5 Feb 2024 07:27:39 +0200 (EET) X-Original-To: ffmpeg-devel@ffmpeg.org Delivered-To: ffmpeg-devel@ffmpeg.org Received: from mail-pl1-f174.google.com (mail-pl1-f174.google.com [209.85.214.174]) by ffbox0-bg.mplayerhq.hu (Postfix) with ESMTPS id 28F2D6800C1 for ; Mon, 5 Feb 2024 07:27:33 +0200 (EET) Received: by mail-pl1-f174.google.com with SMTP id d9443c01a7336-1d73066880eso35052625ad.3 for ; Sun, 04 Feb 2024 21:27:33 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=gmail.com; s=20230601; t=1707110850; x=1707715650; darn=ffmpeg.org; h=content-transfer-encoding:mime-version:message-id:date:subject:cc :to:from:from:to:cc:subject:date:message-id:reply-to; bh=pFjN63ul0J/TnPHc5knkelhOvb8xND9y5j1+UOvrsq0=; b=PwP6rbP4qsD6BXIv7GjgTq3pee+sFFb5Hx0mk5v5umKWXE8Us4/71y+eMucJseTtcA xdb128W4IBGLvanFyNGnpk+ptvw3tNYU+WFGi0n0Dk25WxR0V4tVtvrB/hcsm0gkSNBV VAwwJbYSYnYeeBPKap/0nbXd/huKCPJJO25HRvOgrvG4zBQeB7Z9f9CVlDInG9nZegLs oNtVPA5uW4uYNeXoySi5Uf0yo1UX9McfWxQ/oqkC2MRZlfSO6kIrmBcxTwEXpT23Ik0Q jrFZi092nvZAzP5WwASbwAjfiseprFHmN2TJUS2aJTr3wO0NUXZTgsPRYiLQpOR74+oc NbMg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1707110850; x=1707715650; h=content-transfer-encoding:mime-version:message-id:date:subject:cc :to:from:x-gm-message-state:from:to:cc:subject:date:message-id :reply-to; bh=pFjN63ul0J/TnPHc5knkelhOvb8xND9y5j1+UOvrsq0=; b=n276lOapFXr8P4YMvT292l+096Tgzznu/t/VKXxJoCNDBIlpVta1Ynk+hLYVCu3XSC dRVDVw0NlrsBJSZsgsHCj4Zq89g+TIxleX3PD7cbbXHxMGqWZQVYSmpGfjUZ79NG2IX9 URwB6HHdCVWiwmWakOsI4sDGBXE5ipRY3XMXEnK/G/1NS5OvOmUqxyLcjFVQr/vTZz5t xcDa5T+ggfWAnFV6ZfrZ54Lih4R/2iQPbdkcK/yrCAC74QZRI2Ivbfmiin3toe0e1lct tIfKlCRfFl2DpvWArrm1IVnmpRYAPRENkfbxRtk3uqMMFe7XTSJwvdk6AflbEncYzTDr tX0A== X-Gm-Message-State: AOJu0YzFcsco9wmqUVmNKOedHuKcC+ARN8qQorucXc8EgqxzpWkeCmZb 7O4U6R+x8ci3X4uIzh2Qluph4mKDI+1HOAasKIgqE9BEKlY0z3Pe0HLwBRPj X-Received: by 2002:a17:902:a513:b0:1d8:ee41:de89 with SMTP id s19-20020a170902a51300b001d8ee41de89mr12622491plq.69.1707110850248; Sun, 04 Feb 2024 21:27:30 -0800 (PST) Received: from localhost (c-24-4-236-173.hsd1.ca.comcast.net. [24.4.236.173]) by smtp.gmail.com with UTF8SMTPSA id z10-20020a170903018a00b001d958de13c7sm4837636plg.0.2024.02.04.21.27.29 (version=TLS1_3 cipher=TLS_AES_128_GCM_SHA256 bits=128/128); Sun, 04 Feb 2024 21:27:29 -0800 (PST) From: Connor Worley To: ffmpeg-devel@ffmpeg.org Date: Sun, 4 Feb 2024 21:27:23 -0800 Message-Id: <20240205052724.26862-1-connorbworley@gmail.com> X-Mailer: git-send-email 2.40.1 MIME-Version: 1.0 Subject: [FFmpeg-devel] [PATCH 1/2] lavu/hashtable: create generic robin hood hash table X-BeenThere: ffmpeg-devel@ffmpeg.org X-Mailman-Version: 2.1.29 Precedence: list List-Id: FFmpeg development discussions and patches List-Unsubscribe: , List-Archive: List-Post: List-Help: List-Subscribe: , Reply-To: FFmpeg development discussions and patches Cc: Connor Worley Errors-To: ffmpeg-devel-bounces@ffmpeg.org Sender: "ffmpeg-devel" X-TUID: CBVv2CI2QVZZ Signed-off-by: Connor Worley --- libavutil/Makefile | 2 + libavutil/hashtable.c | 172 ++++++++++++++++++++++++++++++++++++ libavutil/hashtable.h | 62 +++++++++++++ libavutil/tests/hashtable.c | 104 ++++++++++++++++++++++ 4 files changed, 340 insertions(+) create mode 100644 libavutil/hashtable.c create mode 100644 libavutil/hashtable.h create mode 100644 libavutil/tests/hashtable.c diff --git a/libavutil/Makefile b/libavutil/Makefile index e7709b97d0..be75d464fc 100644 --- a/libavutil/Makefile +++ b/libavutil/Makefile @@ -138,6 +138,7 @@ OBJS = adler32.o \ fixed_dsp.o \ frame.o \ hash.o \ + hashtable.o \ hdr_dynamic_metadata.o \ hdr_dynamic_vivid_metadata.o \ hmac.o \ @@ -251,6 +252,7 @@ TESTPROGS = adler32 \ file \ fifo \ hash \ + hashtable \ hmac \ hwdevice \ integer \ diff --git a/libavutil/hashtable.c b/libavutil/hashtable.c new file mode 100644 index 0000000000..fb0f0aae99 --- /dev/null +++ b/libavutil/hashtable.c @@ -0,0 +1,172 @@ +/* + * Generic hashtable + * Copyright (C) 2024 Connor Worley + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include + +#include "error.h" +#include "mem.h" +#include "hashtable.h" +#include "string.h" + +#define ENTRY_OCC(entry) (entry) +#define ENTRY_KEY(entry) (ENTRY_OCC(entry) + 1) +#define ENTRY_VAL(entry) (ENTRY_KEY(entry) + ctx->key_size) +#define ENTRY_PSL(entry) (size_t*) (ENTRY_VAL(entry) + ctx->val_size) + +#define KEYS_EQUAL(k1, k2) !memcmp(k1, k2, ctx->key_size) + +int av_hashtable_init(AVHashtableContext* ctx, size_t key_size, size_t val_size, size_t max_entries) +{ + ctx->key_size = key_size; + ctx->val_size = val_size; + ctx->entry_size = 1 + key_size + val_size + sizeof(size_t); + ctx->max_entries = max_entries; + ctx->utilization = 0; + ctx->crc = av_crc_get_table(AV_CRC_32_IEEE); + if (!ctx->crc) + return AVERROR_BUG; + ctx->table = av_mallocz(ctx->entry_size * max_entries); + if (!ctx->table) + return AVERROR(ENOMEM); + ctx->set_key = av_malloc(key_size); + if (!ctx->set_key) + return AVERROR(ENOMEM); + ctx->set_val = av_malloc(key_size); + if (!ctx->set_val) + return AVERROR(ENOMEM); + ctx->tmp_key = av_malloc(key_size); + if (!ctx->tmp_key) + return AVERROR(ENOMEM); + ctx->tmp_val = av_malloc(val_size); + if (!ctx->tmp_val) + return AVERROR(ENOMEM); + return 0; +} + +static size_t hash_key(const AVHashtableContext* ctx, const void* key) +{ + return av_crc(ctx->crc, 0, key, ctx->key_size) % ctx->max_entries; +} + +int av_hashtable_get(const AVHashtableContext* ctx, const void* key, void* val) +{ + size_t hash = hash_key(ctx, key); + + if (!ctx->utilization) + return 0; + + for (size_t psl = 0; psl < ctx->max_entries; psl++) { + size_t wrapped_index = (hash + psl) % ctx->max_entries; + uint8_t *entry = ctx->table + wrapped_index * ctx->entry_size; + if (!*ENTRY_OCC(entry) || *ENTRY_PSL(entry) < psl) + return 0; + if (KEYS_EQUAL(ENTRY_KEY(entry), key)) { + memcpy(val, ENTRY_VAL(entry), ctx->val_size); + return 1; + } + } + return 0; +} + +int av_hashtable_set(AVHashtableContext* ctx, const void* key, const void* val) +{ + int swapping = 0; + size_t psl = 0; + size_t hash = hash_key(ctx, key); + + memcpy(ctx->set_key, key, ctx->key_size); + memcpy(ctx->set_val, val, ctx->val_size); + + for (size_t i = 0; i < ctx->max_entries; i++) { + size_t wrapped_index = (hash + i) % ctx->max_entries; + uint8_t *entry = ctx->table + wrapped_index * ctx->entry_size; + if (!*ENTRY_OCC(entry) || (!swapping && KEYS_EQUAL(ENTRY_KEY(entry), ctx->set_key))) { + if (!*ENTRY_OCC(entry)) + ctx->utilization++; + *ENTRY_OCC(entry) = 1; + memcpy(ENTRY_KEY(entry), ctx->set_key, ctx->key_size); + memcpy(ENTRY_VAL(entry), ctx->set_val, ctx->val_size); + *ENTRY_PSL(entry) = psl; + return 1; + } + if (*ENTRY_PSL(entry) < psl) { + if (ctx->utilization == ctx->max_entries) + return 0; + swapping = 1; + memcpy(ctx->tmp_key, ctx->set_key, ctx->key_size); + memcpy(ctx->tmp_val, ctx->set_val, ctx->val_size); + memcpy(ctx->set_key, ENTRY_KEY(entry), ctx->key_size); + memcpy(ctx->set_val, ENTRY_VAL(entry), ctx->val_size); + memcpy(ENTRY_KEY(entry), ctx->tmp_key, ctx->key_size); + memcpy(ENTRY_VAL(entry), ctx->tmp_val, ctx->val_size); + FFSWAP(size_t, psl, *ENTRY_PSL(entry)); + } + psl++; + } + return 0; +} + +int av_hashtable_delete(AVHashtableContext* ctx, const void* key) +{ + uint8_t *next_entry; + size_t hash = hash_key(ctx, key); + + if (!ctx->utilization) + return 0; + + for (size_t psl = 0; psl < ctx->max_entries; psl++) { + size_t wrapped_index = (hash + psl) % ctx->max_entries; + uint8_t *entry = ctx->table + wrapped_index * ctx->entry_size; + if (!*ENTRY_OCC(entry) || *ENTRY_PSL(entry) < psl) + return 0; + if (KEYS_EQUAL(ENTRY_KEY(entry), key)) { + *ENTRY_OCC(entry) = 0; + for (psl++; psl < ctx->max_entries; psl++) { + wrapped_index = (hash + psl) % ctx->max_entries; + next_entry = ctx->table + wrapped_index * ctx->entry_size; + if (!*ENTRY_OCC(next_entry) || !*ENTRY_PSL(next_entry)) { + ctx->utilization--; + return 1; + } + FFSWAP(uint8_t, *ENTRY_OCC(entry), *ENTRY_OCC(next_entry)); + memcpy(ENTRY_KEY(entry), ENTRY_KEY(next_entry), ctx->key_size); + memcpy(ENTRY_VAL(entry), ENTRY_VAL(next_entry), ctx->val_size); + *ENTRY_PSL(entry) = *ENTRY_PSL(next_entry) - 1; + entry = next_entry; + } + } + }; + return 0; +} + +void av_hashtable_clear(AVHashtableContext* ctx) +{ + memset(ctx->table, 0, ctx->entry_size * ctx->max_entries); +} + +void av_hashtable_destroy(AVHashtableContext* ctx) +{ + av_freep(&ctx->table); + av_freep(&ctx->set_key); + av_freep(&ctx->set_val); + av_freep(&ctx->tmp_key); + av_freep(&ctx->tmp_val); +} \ No newline at end of file diff --git a/libavutil/hashtable.h b/libavutil/hashtable.h new file mode 100644 index 0000000000..aac31ee922 --- /dev/null +++ b/libavutil/hashtable.h @@ -0,0 +1,62 @@ +/* + * Generic hashtable + * Copyright (C) 2024 Connor Worley + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#ifndef AVUTIL_HASHTABLE_H +#define AVUTIL_HASHTABLE_H + +#include + +#include "crc.h" + +/* Implements a hash table using Robin Hood open addressing. + * See: https://cs.uwaterloo.ca/research/tr/1986/CS-86-14.pdf + * + * Entry slots are layed out in memory as follows: + * - [1 byte] occupancy flag that indicates whether a slot if occupied. + * - [key_size bytes] entry key. + * - [val_size bytes] entry value. + * - [sizeof(size_t) bytes] PSL (probe sequence length), used by the Robin Hood + * algorithm. + */ + +typedef struct AVHashtableContext { + size_t key_size; + size_t val_size; + size_t entry_size; + size_t max_entries; + size_t utilization; + const AVCRC *crc; + uint8_t *table; + /* used for swaps during insertion */ + uint8_t *set_key; + uint8_t *set_val; + uint8_t *tmp_key; + uint8_t *tmp_val; +} AVHashtableContext; + +int av_hashtable_init(AVHashtableContext* ctx, size_t key_size, size_t val_size, size_t max_entries); +int av_hashtable_get(const AVHashtableContext* ctx, const void* key, void* val); +int av_hashtable_set(AVHashtableContext* ctx, const void* key, const void* val); +int av_hashtable_delete(AVHashtableContext* ctx, const void* key); +void av_hashtable_clear(AVHashtableContext* ctx); +void av_hashtable_destroy(AVHashtableContext* ctx); + +#endif \ No newline at end of file diff --git a/libavutil/tests/hashtable.c b/libavutil/tests/hashtable.c new file mode 100644 index 0000000000..1cdbd7afd3 --- /dev/null +++ b/libavutil/tests/hashtable.c @@ -0,0 +1,104 @@ +/* + * Generic hashtable tests + * Copyright (C) 2024 Connor Worley + * + * This file is part of FFmpeg. + * + * FFmpeg is free software; you can redistribute it and/or + * modify it under the terms of the GNU Lesser General Public + * License as published by the Free Software Foundation; either + * version 2.1 of the License, or (at your option) any later version. + * + * FFmpeg is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU + * Lesser General Public License for more details. + * + * You should have received a copy of the GNU Lesser General Public + * License along with FFmpeg; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA + */ + +#include "libavutil/avassert.h" +#include "libavutil/hashtable.h" + +int main(void) +{ + AVHashtableContext ctx; + int k, v; + + // hashtable can store up to 3 int->int entries + av_assert0(!av_hashtable_init(&ctx, sizeof(int), sizeof(int), 3)); + + // unsuccessful deletes return 0 + k = 1; + av_assert0(!av_hashtable_delete(&ctx, &k)); + + // unsuccessful gets return 0 + k = 1; + av_assert0(!av_hashtable_get(&ctx, &k, &v)); + + // successful sets returns 1 + k = 1; + v = 1; + av_assert0(av_hashtable_set(&ctx, &k, &v)); + + // get should now contain 1 + k = 1; + v = 0; + av_assert0(av_hashtable_get(&ctx, &k, &v)); + av_assert0(v == 1); + + // updating sets should return 1 + k = 1; + v = 2; + av_assert0(av_hashtable_set(&ctx, &k, &v)); + + // get should now contain 2 + k = 1; + v = 0; + av_assert0(av_hashtable_get(&ctx, &k, &v)); + av_assert0(v == 2); + + // fill the table + k = 2; + v = 2; + av_assert0(av_hashtable_set(&ctx, &k, &v)); + k = 3; + v = 3; + av_assert0(av_hashtable_set(&ctx, &k, &v)); + + // inserting sets on a full table should return 0 + k = 4; + v = 4; + av_assert0(!av_hashtable_set(&ctx, &k, &v)); + + // updating sets on a full table should return 1 + k = 1; + v = 4; + av_assert0(av_hashtable_set(&ctx, &k, &v)); + v = 0; + av_assert0(av_hashtable_get(&ctx, &k, &v)); + av_assert0(v == 4); + + // successful deletes should return 1 + k = 1; + av_assert0(av_hashtable_delete(&ctx, &k)); + + // get should now return 0 + av_assert0(!av_hashtable_get(&ctx, &k, &v)); + + // sanity check remaining keys + k = 2; + v = 0; + av_assert0(av_hashtable_get(&ctx, &k, &v)); + av_assert0(v == 2); + k = 3; + v = 0; + av_assert0(av_hashtable_get(&ctx, &k, &v)); + av_assert0(v == 3); + + av_hashtable_destroy(&ctx); + + return 0; +} \ No newline at end of file