diff options
Diffstat (limited to 'cppawk-include/array.h')
-rw-r--r-- | cppawk-include/array.h | 98 |
1 files changed, 98 insertions, 0 deletions
diff --git a/cppawk-include/array.h b/cppawk-include/array.h new file mode 100644 index 0000000..f3b7a22 --- /dev/null +++ b/cppawk-include/array.h @@ -0,0 +1,98 @@ +// cppawk: C preprocessor wrapper around awk +// Copyright 2022 Kaz Kylheku <kaz@kylheku.com> +// +// BSD-2 License +// +// Redistribution and use in source and binary forms, with or without +// modification, are permitted provided that the following conditions are met: +// +// 1. Redistributions of source code must retain the above copyright notice, +// this list of conditions and the following disclaimer. +// +// 2. Redistributions in binary form must reproduce the above copyright notice, +// this list of conditions and the following disclaimer in the documentation +// and/or other materials provided with the distribution. +// +// THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" +// AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE +// IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE +// ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE +// LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR +// CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF +// SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS +// INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN +// CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) +// ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE +// POSSIBILITY OF SUCH DAMAGE. + +#ifndef __CPPAWK_ARRAY_H +#define __CPPAWK_ARRAY_H + +#ifndef __CPPAWK_BASE_H +#include "base.h" +#endif + +#ifndef __CPPAWK_ITER_PRIV_H +#include "iter-priv.h" +#endif + +#ifndef __CPPAWK_CONS_PRIV_H +#include "cons-priv.h" +#endif + +#if __have_indirect_functions +#ifndef __CPPAWK_FUN_PRIV_H +#include "fun-priv.h" +#endif +#endif + +function fields_to_array(__to_array, + __i, __val) +{ + delete __to_array + + __to_array[0] = $0 + + __dofields (__i, __val) + __to_array[__i] = __val +} + +function array_to_fields(__from_array, + __i, __val) +{ + for (__i = 0; ; __i++) + if (__i in __from_array) + $__i = __from_array[__i] + else + break +} + +#if __have_indirect_functions + +function maparray(__fun, __from_array, __to_array) +{ + delete __to_array + __doarray (key, value, __from_array) + __to_array[key] = __call(__fun, value) +} + +function group_by(__fun, __list, __to_array, + __item, __key) +{ + delete __to_array + + __dolist (__item, __list) { + __key = __call(__fun, __item) + if (__key in __to_array) + __to_array[__key] = __list_add(__to_array[__key], __item) + else + __to_array[__key] = __list_add(__list_begin(), __item) + } + + for (__key in __to_array) + __to_array[__key] = __list_end(__to_array[__key]) +} + +#endif + +#endif |