mt.c

Go to the documentation of this file.

/*----------------------------------------------------------------------------
 * Copyright (c) 2006, Novell, Inc.
 * All rights reserved.
 * 
 * Redistribution and use in source and binary forms, with or without 
 * modification, are permitted provided that the following conditions are 
 * met:
 * 
 *     * Redistributions of source code must retain the above copyright 
 *       notice, this list of conditions and the following disclaimer.
 *     * 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.
 *     * Neither the name of the Novell nor the names of its contributors 
 *       may be used to endorse or promote products derived from this 
 *       software without specific prior written permission.
 * 
 * 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 OWNER 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.
 *--------------------------------------------------------------------------*/

#include "mt.h"

#include <string.h>
#include <stdlib.h>

#ifdef _WIN32
# include <windows.h>
#else
# include <time.h>
#endif

#ifdef _WIN32
# define mt_sint64 __int64
# define mt_uint64 unsigned mt_sint64
#else
# define mt_sint64 long long
# define mt_uint64 unsigned mt_sint64
#endif

typedef struct mt_tag
{
   int sz;              
   int cp;              
   mt_uint64 freq;      
   mt_uint64 times[1];  
} MTimer;

/*---------------------------------------------------------------------------
 */
MultiTimer MtCreate(int count)
{
   MTimer * t;

   t = (MTimer *)malloc(sizeof(*t) + count * sizeof(t->times));
   if (t != 0)
   {
      memset(t, 0, sizeof(*t) + count * sizeof(t->times));
#ifdef _WIN32
      {
         LARGE_INTEGER fr, st;
         QueryPerformanceFrequency(&fr);
         t->freq = ((mt_uint64)fr.HighPart << 32) + fr.LowPart;
         QueryPerformanceCounter(&st);
         t->times[0] = ((mt_uint64)st.HighPart << 32) + st.LowPart;
      }
#else
      t->freq = CLOCKS_PER_SEC;
      t->times[0] = clock();
#endif
      t->sz = count;
   }
   return (MultiTimer *)t;
}

/*---------------------------------------------------------------------------
 */
void MtRecord(MultiTimer mt)
{
   MTimer * t = (MTimer *)mt;

   if (t->cp < t->sz)
   {
#ifdef _WIN32
      LARGE_INTEGER tm;
      QueryPerformanceCounter(&tm);
      t->times[++t->cp] = ((mt_uint64)tm.HighPart << 32) + tm.LowPart;
#else
      t->times[++t->cp] = clock();
#endif
   }
}

/*---------------------------------------------------------------------------
 */
unsigned long MtGetRelativeNS(MultiTimer mt, int idx)
{
#define NS_PER_SEC 1000000000

   MTimer * t = (MTimer *)mt;

   if (idx > t->cp)
      return 0;
   return (unsigned long)((((t->times[idx + 1] - t->times[idx]) 
         * NS_PER_SEC) / t->freq) & 0xFFFFFFFF);
}

/*---------------------------------------------------------------------------
 */
void MtDestroy(MultiTimer mt) 
{
   free(mt); 
}