Loki/test/SafeFormat/main.cpp
rich_sposato 1fcba1aa3b Added 4 void declaration lines to remove compiler warning.
git-svn-id: svn://svn.code.sf.net/p/loki-lib/code/trunk@1141 7ec92016-0320-0410-acc4-a06ded1c099a
2011-10-07 07:06:01 +00:00

622 lines
16 KiB
C++

////////////////////////////////////////////////////////////////////////////////
// Copyright (c) 2005 by Andrei Alexandrescu
// Permission to use, copy, modify, distribute, and sell this software for any
// purpose is hereby granted without fee, provided that the above copyright
// notice appear in all copies and that both that copyright notice and this
// permission notice appear in supporting documentation.
// The author makes no representations about the suitability of this software
// for any purpose. It is provided "as is" without express or implied
// warranty.
////////////////////////////////////////////////////////////////////////////////
// $Id$
#include <loki/SafeFormat.h>
#include <iostream>
#include <cassert>
#include <utility>
#include <cctype>
#include <cstdlib>
#include "../SmallObj/timer.h"
#include "ThreadPool.hpp"
#if defined(_MSC_VER)
#if _MSC_VER >= 1400
#define sprintf sprintf_s
#define _snprintf _snprintf_s
#else
#define sprintf sprintf
#define _snprintf _snprintf
#endif
#endif
using namespace std;
using namespace Loki;
template <class Integral1, class Integral2>
Integral2 RandomInt(Integral1 low, Integral2 up)
{
// From ``Accelerated C++'', page 135:
// random integer in the range [0, n)
// We adjust to generate in the range [0, n]
const Integral2
low2 = low,
n = up - low;
assert(n > 0);
const size_t bucket_size = RAND_MAX / n;
assert(bucket_size > 0);
Integral2 r;
do
r = Integral2(rand() / bucket_size);
while (r > n);
r = r + low2;
assert(r >= low2 && r <= up);
return r;
}
string RandomString(unsigned int maxSize)
{
string result(RandomInt(0, maxSize), '\0');
unsigned int i = 0;
for (; i != result.size(); ++i)
{
result[i] = RandomInt('a', 'z');
}
return result;
}
template <class T>
void TestCase(const string& fmt, T value)
{
char buf[4096];
std::string s;
const int i1 = SPrintf(s, fmt.c_str())(value);
#ifdef _MSC_VER
const int i2 = _snprintf(buf, sizeof(buf), fmt.c_str(), value);
#else
const int i2 = snprintf(buf, sizeof(buf), fmt.c_str(), value);
#endif
if (i1 != i2 || s != buf)
{
cout << endl
<< "Reference: " << i2 << "; Actual: " << i1 << ", Difference = " << i2-i1 << endl
<< "V: [" << value << "]" << endl
<< "F: [" << fmt << "]" << endl
<< "R: [" << buf << "]" << endl
<< "A: [" << s.c_str() << "]" << endl;
assert(false);
}
}
template <class T, class U>
void TestCase2(const string& fmt, T value, U value2)
{
char buf[4096];
std::string s;
const int i1 = SPrintf(s, fmt.c_str())(value)(value2);
const int i2 = snprintf(buf, sizeof(buf), fmt.c_str(), value, value2);
assert(i1 == i2);
assert(s == buf);
}
// ----------------------------------------------------------------------------
void test_dword()
{
typedef signed int Int;
typedef unsigned int UInt;
typedef signed long Long;
typedef unsigned long ULong;
Int i(0);
UInt ui(0);
Long l(0);
ULong ul(0);
Printf("%d")(i);
Printf("%d")(ui);
Printf("%d")(l);
Printf("%d")(ul);
}
// ----------------------------------------------------------------------------
void SpeedTest( unsigned int loop )
{
cout << "Starting SpeedTest" << endl;
// test speed
Timer t;
if(loop < 100)
loop = 100;
t.start();
for (int i=loop; i > 0; --i)
printf("Hey, %u frobnicators and %u twiddlicators\n",i, i);
t.stop();
t.t100 = t.t();
int t_printf = t.t();
t.start();
for (int i=loop; i > 0; --i)
::Loki::Printf("Hey, %u frobnicators and %u twiddlicators\n")(i)(i);
t.stop();
int t_Printf = t.t();
t.start();
for (int i=loop; i > 0; --i)
cout << "Hey, " << i << " frobnicators and " << i <<" twiddlicators\n";
t.stop();
int t_cout = t.t();
Printf("\n\nElapsed time for %i outputs\n\n")(loop);
t.print(t_printf,"printf : ");
t.print(t_Printf,"Printf : ");
t.print(t_cout, "std::cout: ");
cout << "Finished SpeedTest" << endl;
}
// ----------------------------------------------------------------------------
void FormatTest( void )
{
cout << "Starting FormatTest" << endl;
string result;
unsigned int i = 1;
signed short ss = 2;
float f = 3.4;
const char * message = " over here! ";
const char * format = "";
::Loki::SPrintf( result, format );
assert( result == format );
result.clear();
format = "abc";
::Loki::SPrintf( result, format );
assert( result == format );
result.clear();
format = "%%";
::Loki::SPrintf( result, format );
assert( result == "%" );
result.clear();
format = "%d";
::Loki::SPrintf( result, format )( i );
assert( result == "1" );
result.clear();
format = "%s";
::Loki::SPrintf( result, format )( message );
assert( result == message );
result.clear();
format = "%d";
try
{
::Loki::SPrintf( result, format )( i )( message );
assert( false );
}
catch ( const ::std::logic_error & ex )
{
(void)ex;
assert( true );
}
result.clear();
format = "%d";
try
{
::Loki::SPrintf( result, format )( i )( message )( ss );
assert( false );
}
catch ( const ::std::logic_error & ex )
{
(void)ex;
assert( true );
}
result.clear();
format = "%d";
try
{
::Loki::SPrintf( result, format )( i )( ss );
assert( false );
}
catch ( const ::std::logic_error & ex )
{
(void)ex;
assert( true );
}
result.clear();
format = "%d";
try
{
::Loki::SPrintf( result, format )( i )( f );
assert( false );
}
catch ( const ::std::logic_error & ex )
{
(void)ex;
assert( true );
}
result.clear();
cout << "Finished FormatTest" << endl;
}
// ----------------------------------------------------------------------------
void RandomTest( unsigned int loopCount )
{
cout << "Starting RandomTest" << endl;
if ( loopCount < 100 )
loopCount = 100;
//srand(time(0));
srand(0);
printf("\nNumber of tests:\n");
for ( unsigned int i = 0; i < loopCount; ++i )
{
printf("%u\r", i);
// Generate a random string for the head
string lead = RandomString(100);
// This string will hold a random format specification
string formatSpec(lead + "|%");
// Generate a random set of flags
static const string flags("-+0 #");
size_t maxFlags = RandomInt(0u, flags.length() - 1);
for (size_t i = 0; i != maxFlags; ++i)
{
formatSpec += flags[RandomInt(0u, flags.length() - 1)];
}
// Generate an optional random width
if (RandomInt(0, 1))
{
const unsigned int width = RandomInt(0, 100);
char buf[4];
sprintf(buf, "%u", width);
formatSpec += buf;
}
// Generate an optional random precision
if (RandomInt(0, 1))
{
const unsigned int prec = RandomInt(0, 100);
char buf[4];
sprintf(buf, "%u", prec);
formatSpec += '.';
formatSpec += buf;
}
// Generate a random type character
static const string type("cdeEfgGinopsuxX");
const char typeSpec = type[RandomInt(0u, type.size() - 1)];
// Generate an optional type prefix
static const string prefix("hl");
if (typeSpec != 's' && RandomInt(0, 1))
{
formatSpec += prefix[RandomInt(0u, prefix.size() - 1)];
}
formatSpec += typeSpec;
formatSpec += '|';
formatSpec += RandomString(100);
switch (typeSpec)
{
case 'c':
TestCase(formatSpec, RandomInt(1, 127));
break;
case 'd':
case 'i':
case 'o':
case 'u':
case 'x':
case 'X':
//TestCase(formatSpec, RandomInt(-10000, 10000));
// don't test negative values on 64bit systems, because
// snprintf does not support 64 Bit values
TestCase(formatSpec, RandomInt( -10000 * (sizeof(size_t)>4 ? 0 : 1) , 10000));
break;
case 'e':
case 'E':
case 'f':
case 'g':
case 'G':
TestCase(formatSpec,
RandomInt(-10000, 10000) / double(RandomInt(1, 100)));
break;
case 'n':
break;
case 'p':
{
void * p = malloc(RandomInt(1, 1000));
TestCase(formatSpec, p);
free(p);
}
break;
case 's':
TestCase(formatSpec, RandomString(100).c_str());
break;
default:
assert(false);
break;
}
}
cout << "Finished RandomTest" << endl;
}
// ----------------------------------------------------------------------------
void * DoLokiPrintfLoop( void * p )
{
const unsigned int threadIndex = reinterpret_cast< unsigned int >( p );
for ( unsigned int loop = 0; loop < 10; ++loop )
{
::Loki::Printf( "Loop: [%u] Thread: [%u]\n" )( loop )( threadIndex );
}
return 0;
}
// ----------------------------------------------------------------------------
void * DoLokiFPrintfLoop( void * p )
{
const unsigned int threadIndex = reinterpret_cast< unsigned int >( p );
for ( unsigned int loop = 0; loop < 10; ++loop )
{
::Loki::FPrintf( cout, "Loop: [%u] Thread: [%u]\n" )( loop )( threadIndex );
}
return 0;
}
// ----------------------------------------------------------------------------
void * DoCoutLoop( void * p )
{
const unsigned int threadIndex = reinterpret_cast< unsigned int >( p );
for ( unsigned int loop = 0; loop < 10; ++loop )
{
cout << "Loop: [" << loop << "] Thread: [" << threadIndex << "]\n";
}
return 0;
}
// ----------------------------------------------------------------------------
void * DoStdOutLoop( void * p )
{
const unsigned int threadIndex = reinterpret_cast< unsigned int >( p );
for ( unsigned int loop = 0; loop < 10; ++loop )
{
printf( "Loop: [%d] Thread: [%d]\n", loop, threadIndex );
}
return 0;
}
// ----------------------------------------------------------------------------
void AtomicTest( void )
{
char ender;
cout << "Starting Loki::Printf AtomicTest" << endl;
{
ThreadPool pool;
pool.Create( 20, &DoLokiPrintfLoop );
pool.Start();
pool.Join();
}
cout << "Finished Loki::Printf AtomicTest." << endl;
cout << "If the output lines up in neat columns, the test passed." << endl;
cout << "If the output is not in columns, then the test failed." << endl;
cout << "Press <Enter> key to continue. ";
cin.get( ender );
cout << "Starting Loki::FPrintf AtomicTest" << endl;
{
ThreadPool pool;
pool.Create( 20, &DoLokiFPrintfLoop );
pool.Start();
pool.Join();
}
cout << "Finished Loki::FPrintf AtomicTest." << endl;
cout << "If the output lines up in neat columns, the test passed." << endl;
cout << "If the output is not in columns, then the test failed." << endl;
cout << "Press <Enter> key to continue. ";
cin.get( ender );
cout << "Starting stdout AtomicTest" << endl;
{
ThreadPool pool;
pool.Create( 20, &DoStdOutLoop );
pool.Start();
pool.Join();
}
cout << "Finished stdout AtomicTest." << endl;
cout << "If the output lines up in neat columns, your compiler implements printf correctly." << endl;
cout << "If the output is not in columns, then your compiler implements printf incorrectly." << endl;
cout << "Press <Enter> key to continue. ";
cin.get( ender );
cout << "Starting cout AtomicTest" << endl;
{
ThreadPool pool;
pool.Create( 20, &DoCoutLoop );
pool.Start();
pool.Join();
}
cout << "Finished cout AtomicTest." << endl;
cout << "If the output lines up in neat columns, your compiler implements cout correctly." << endl;
cout << "If the output is not in columns, then your compiler implements cout incorrectly." << endl;
cout << "Press <Enter> key to continue. ";
cin.get( ender );
}
// ----------------------------------------------------------------------------
class CommandLineArgs
{
public:
CommandLineArgs( unsigned int argc, const char * argv[] );
inline unsigned int GetSpeedLoopCount( void ) const { return m_speedLoopCount; }
inline unsigned int GetRandomLoopCount( void ) const { return m_randomLoopCount; }
inline bool DoSpeedTest( void ) const { return ( 0 < m_speedLoopCount ); }
inline bool DoRandomTest( void ) const { return ( 0 < m_randomLoopCount ); }
inline bool DoFormatTest( void ) const { return m_doFormatTest; }
inline bool DoAtomicTest( void ) const { return m_doAtomicTest; }
inline bool DoShowHelp( void ) const { return m_showHelp; }
void ShowHelp( void ) const;
private:
unsigned int m_speedLoopCount;
unsigned int m_randomLoopCount;
bool m_doFormatTest;
bool m_doAtomicTest;
bool m_showHelp;
const char * m_exeName;
};
// ----------------------------------------------------------------------------
CommandLineArgs::CommandLineArgs( unsigned int argc, const char * argv[] ) :
m_speedLoopCount( 0 ),
m_randomLoopCount( 0 ),
m_doFormatTest( false ),
m_doAtomicTest( false ),
m_showHelp( false ),
m_exeName( argv[0] )
{
if ( argc < 2 )
{
m_showHelp = true;
return;
}
bool isValid = true;
for ( unsigned int ii = 1; ( isValid ) && ( ii < argc ); ++ii )
{
const char * arg = argv[ii];
const unsigned int length = ::strlen( arg );
if ( ( '-' != *arg ) || ( length < 2 ) )
{
isValid = false;
break;
}
switch ( arg[1] )
{
default: isValid = false; break;
case 'h': m_showHelp = true; break;
case 'f': m_doFormatTest = true; break;
case 'a': m_doAtomicTest = true; break;
case 'r':
{
if ( ( length < 3 ) || ( ':' != arg[2] ) || ( 0 == isdigit( arg[3] ) ) )
{
isValid = false;
break;
}
m_randomLoopCount = ::atoi( arg + 3 );
break;
}
case 's':
{
if ( ( length < 3 ) || ( ':' != arg[2] ) || ( 0 == isdigit( arg[3] ) ) )
{
isValid = false;
break;
}
m_speedLoopCount = ::atoi( arg + 3 );
break;
}
}
}
if ( !isValid )
{
m_showHelp = true;
m_speedLoopCount = 0;
m_randomLoopCount = 0;
m_doFormatTest = false;
}
}
// ----------------------------------------------------------------------------
void CommandLineArgs::ShowHelp( void ) const
{
cout << "Usage: " << m_exeName << " [-h] [-f] [-r:#] [-s:#]" << endl;
cout << " -h Show this help info and exit. Overrides all other options." << endl;
cout << " -f Run formatting tests." << endl;
cout << " -a Run atomic tests." << endl;
cout << " -r:# Run random tests for # of loops. # is a positive decimal value greater than 100." << endl;
cout << " -s:# Run speed tests for # of loops. # is a positive decimal value greater than 100." << endl;
}
// ----------------------------------------------------------------------------
int main( int argc, const char * argv[] )
{
const CommandLineArgs args( argc, argv );
if ( args.DoShowHelp() )
{
args.ShowHelp();
return 0;
}
if ( args.DoSpeedTest() )
{
SpeedTest( args.GetSpeedLoopCount() );
}
if ( args.DoRandomTest() )
{
RandomTest( args.GetRandomLoopCount() );
}
if ( args.DoFormatTest() )
{
FormatTest();
}
if ( args.DoAtomicTest() )
{
AtomicTest();
}
}
// ----------------------------------------------------------------------------