New header: <array.h>

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