tests/slice-color.c

   1 /* GLIB sliced memory - fast threaded memory chunk allocator
   2  * Copyright (C) 2005 Tim Janik
   3  *
   4  * This library is free software; you can redistribute it and/or
   5  * modify it under the terms of the GNU Lesser General Public
   6  * License as published by the Free Software Foundation; either
   7  * version 2 of the License, or (at your option) any later version.
   8  *
   9  * This library is distributed in the hope that it will be useful,
  10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12  * Lesser General Public License for more details.
  13  *
  14  * You should have received a copy of the GNU Lesser General Public
  15  * License along with this library; if not, see <http://www.gnu.org/licenses/>.
  16  */
  17 #include <glib.h>
  18 #include <string.h>
  19
  20 #define ALIGN(size, base)       ((base) * (gsize) (((size) + (base) - 1) / (base)))
  21
  22 static gdouble parse_memsize (const gchar *cstring);
  23 static void    usage         (void);
  24
  25 static void
  26 fill_memory (guint **mem,
  27              guint   n,
  28              guint   val)
  29 {
  30   guint j, o = 0;
  31   for (j = 0; j < n; j++)
  32     mem[j][o] = val;
  33 }
  34
  35 static guint64
  36 access_memory3 (guint  **mema,
  37                 guint  **memb,
  38                 guint  **memd,
  39                 guint    n,
  40                 guint64  repeats)
  41 {
  42   guint64 accu = 0, i, j;
  43   const guint o = 0;
  44   for (i = 0; i < repeats; i++)
  45     {
  46       for (j = 1; j < n; j += 2)
  47         memd[j][o] = mema[j][o] + memb[j][o];
  48     }
  49   for (i = 0; i < repeats; i++)
  50     for (j = 0; j < n; j++)
  51       accu += memd[j][o];
  52   return accu;
  53 }
  54
  55 static void
  56 touch_mem (guint64 block_size,
  57            guint64 n_blocks,
  58            guint64 repeats)
  59 {
  60   guint64 j, accu, n = n_blocks;
  61   GTimer *timer;
  62   guint **memc;
  63   guint **memb;
  64   guint **mema = g_new (guint*, n);
  65   for (j = 0; j < n; j++)
  66     mema[j] = g_slice_alloc (block_size);
  67   memb = g_new (guint*, n);
  68   for (j = 0; j < n; j++)
  69     memb[j] = g_slice_alloc (block_size);
  70   memc = g_new (guint*, n);
  71   for (j = 0; j < n; j++)
  72     memc[j] = g_slice_alloc (block_size);
  73
  74   timer = g_timer_new();
  75   fill_memory (mema, n, 2);
  76   fill_memory (memb, n, 3);
  77   fill_memory (memc, n, 4);
  78   access_memory3 (mema, memb, memc, n, 3);
  79   g_timer_start (timer);
  80   accu = access_memory3 (mema, memb, memc, n, repeats);
  81   g_timer_stop (timer);
  82
  83   g_print ("Access-time = %fs\n", g_timer_elapsed (timer, NULL));
  84   g_assert (accu / repeats == (2 + 3) * n / 2 + 4 * n / 2);
  85
  86   for (j = 0; j < n; j++)
  87     {
  88       g_slice_free1 (block_size, mema[j]);
  89       g_slice_free1 (block_size, memb[j]);
  90       g_slice_free1 (block_size, memc[j]);
  91     }
  92   g_timer_destroy (timer);
  93   g_free (mema);
  94   g_free (memb);
  95   g_free (memc);
  96 }
  97
  98 static void
  99 usage (void)
 100 {
 101   g_print ("Usage: slice-color <block-size> [memory-size] [repeats] [colorization]\n");
 102 }
 103
 104 int
 105 main (int   argc,
 106       char *argv[])
 107 {
 108   guint64 block_size = 512, area_size = 1024 * 1024, n_blocks, repeats = 1000000;
 109
 110   if (argc > 1)
 111     block_size = parse_memsize (argv[1]);
 112   else
 113     {
 114       usage();
 115       block_size = 512;
 116     }
 117   if (argc > 2)
 118     area_size = parse_memsize (argv[2]);
 119   if (argc > 3)
 120     repeats = parse_memsize (argv[3]);
 121   if (argc > 4)
 122     g_slice_set_config (G_SLICE_CONFIG_COLOR_INCREMENT, parse_memsize (argv[4]));
 123
 124   /* figure number of blocks from block and area size.
 125    * divide area by 3 because touch_mem() allocates 3 areas
 126    */
 127   n_blocks = area_size / 3 / ALIGN (block_size, sizeof (gsize) * 2);
 128
 129   /* basic sanity checks */
 130   if (!block_size || !n_blocks || block_size >= area_size)
 131     {
 132       g_printerr ("Invalid arguments: block-size=%" G_GUINT64_FORMAT " memory-size=%" G_GUINT64_FORMAT "\n", block_size, area_size);
 133       usage();
 134       return 1;
 135     }
 136
 137   g_printerr ("Will allocate and touch %" G_GUINT64_FORMAT " blocks of %" G_GUINT64_FORMAT " bytes (= %" G_GUINT64_FORMAT " bytes) %" G_GUINT64_FORMAT " times with color increment: 0x%08" G_GINT64_MODIFIER "x\n",
 138               n_blocks, block_size, n_blocks * block_size, repeats,
 139               (guint64)g_slice_get_config (G_SLICE_CONFIG_COLOR_INCREMENT));
 140
 141   touch_mem (block_size, n_blocks, repeats);
 142
 143   return 0;
 144 }
 145
 146 static gdouble
 147 parse_memsize (const gchar *cstring)
 148 {
 149   gchar *mem = g_strdup (cstring);
 150   gchar *string = g_strstrip (mem);
 151   guint l = strlen (string);
 152   gdouble f = 0;
 153   gchar *derr = NULL;
 154   gdouble msize;
 155
 156   switch (l ? string[l - 1] : 0)
 157     {
 158     case 'k':   f = 1000;               break;
 159     case 'K':   f = 1024;               break;
 160     case 'm':   f = 1000000;            break;
 161     case 'M':   f = 1024 * 1024;        break;
 162     case 'g':   f = 1000000000;         break;
 163     case 'G':   f = 1024 * 1024 * 1024; break;
 164     }
 165   if (f)
 166     string[l - 1] = 0;
 167   msize = g_ascii_strtod (string, &derr);
 168   g_free (mem);
 169   if (derr && *derr)
 170     {
 171       g_printerr ("failed to parse number at: %s\n", derr);
 172       msize = 0;
 173     }
 174   if (f)
 175     msize *= f;
 176   return msize;
 177 }