1 files changed, 92 insertions, 0 deletions
diff --git a/runtime/tests/pmaps/map_format.c b/runtime/tests/pmaps/map_format.c
new file mode 100644
index 00000000..5b99b64e
--- /dev/null
+++ b/runtime/tests/pmaps/map_format.c
@@ -0,0 +1,92 @@
+#include "runtime.h"
+
+/* map formatting test. Same as the non-pmap version. Output should be identical */
+
+/* torture test of map formatting */
+#define VALUE_TYPE STRING
+#define KEY1_TYPE INT64
+#define KEY2_TYPE INT64
+#define KEY3_TYPE STRING
+#include "pmap-gen.c"
+
+#define VALUE_TYPE INT64
+#define KEY1_TYPE STRING
+#define KEY2_TYPE STRING
+#include "pmap-gen.c"
+
+#define VALUE_TYPE STAT
+#define KEY1_TYPE STRING
+#define KEY2_TYPE STRING
+#include "pmap-gen.c"
+
+#include "map.c"
+
+void inc_cpu(void)
+{
+  _processor_number++;
+  if (_processor_number == NR_CPUS)
+    _processor_number = 0;
+}
+
+int main ()
+{
+  MAP mapiis = _stp_pmap_new_iiss(4);
+  _processor_number = 0;
+  _stp_pmap_set_iiss (mapiis, 1,2,"Ohio", "Columbus" );
+  _stp_pmap_set_iiss (mapiis, 3,4,"California", "Sacramento" );
+  _stp_pmap_set_iiss (mapiis, 5,6,"Washington", "Olympia" );
+  _stp_pmap_set_iiss (mapiis, 7,8,"Oregon", "Salem" );
+  _stp_pmap_print (mapiis, "%s -> mapiis %1d %2d %3s");
+
+  /* test printing of '%' */
+  _stp_pmap_print (mapiis, "%s %% %3s");
+
+  /* very bad string.  don't crash */
+  _stp_pmap_print (mapiis, "%s -> mapiis %1s %2s %3d %4d");
+
+  MAP mapss = _stp_pmap_new_ssi(4);
+  _stp_pmap_set_ssi (mapss, "Riga", "Latvia", 0x0000c0dedbad0000LL);
+  _stp_pmap_set_ssi (mapss, "Sofia", "Bulgaria", 0xdeadf00d12345678LL);
+  _stp_pmap_set_ssi (mapss, "Valletta", "Malta", 1);
+  _stp_pmap_set_ssi (mapss, "Nicosia", "Cyprus", -1);
+  _stp_pmap_print (mapss, "The capitol of %1s is %2s and the nerd population is %d");
+  _stp_pmap_print (mapss, "The capitol of %1s is %2s and the nerd population is %x");
+  _stp_pmap_print (mapss, "The capitol of %1s is %2s and the nerd population is %X");
+
+  MAP mapssx = _stp_pmap_new_ssx (4, HIST_LINEAR, 0, 100, 10 );
+  int i,j;
+
+  for (i = 0; i < 100; i++)
+    for (j = 0; j <= i*10 ; j++ ) {
+      inc_cpu();
+      _stp_pmap_add_ssx (mapssx, "Riga", "Latvia", i);
+    }
+  
+  for (i = 0; i < 10; i++)
+    for (j = 0; j < 10 ; j++ ) {
+      inc_cpu();
+      _stp_pmap_add_ssx (mapssx, "Sofia", "Bulgaria", j * i );
+    }
+
+  for (i = 0; i < 100; i += 10)
+    for (j = 0; j < i/10 ; j++ ) {
+      inc_cpu();
+      _stp_pmap_add_ssx (mapssx, "Valletta", "Malta", i);
+    }
+
+    _stp_pmap_print (mapssx, "Bogons per packet for %1s\ncount:%C  sum:%S  avg:%A  min:%m  max:%M\n%H");
+
+    _stp_pmap_print (mapssx, "%C was the count for %1s, %2s");
+
+    /* here's how to print a map without using _stp_pmap_print(). */
+    _stp_pmap_agg (mapssx);
+    struct map_node *ptr;
+    foreach (AGG_PMAP(mapssx), ptr)
+      _stp_printf ("mapssx[%09s,%09s] = %llX\n", key1str(ptr), key2str(ptr), _stp_get_stat(ptr)->sum);
+    _stp_print_flush();
+    
+    _stp_pmap_del (mapssx);
+    _stp_pmap_del (mapiis);
+    _stp_pmap_del (mapss);
+    return 0;
+}