Check in the pristine copy of ICU 4.6...

icu46/source/test/perf/unisetperf/unisetperf.cpp

+/*  
+**********************************************************************
+*   Copyright (C) 2007, International Business Machines
+*   Corporation and others.  All Rights Reserved.
+**********************************************************************
+*   file name:  unisetperf.cpp
+*   encoding:   US-ASCII
+*   tab size:   8 (not used)
+*   indentation:4
+*
+*   created on: 2007jan31
+*   created by: Markus Scherer
+*/
+
+#include <stdio.h>
+#include <stdlib.h>
+#include <string.h>
+#include "unicode/uperf.h"
+#include "unicode/uniset.h"
+#include "unicode/unistr.h"
+#include "uoptions.h"
+
+#define LENGTHOF(array) (int32_t)(sizeof(array)/sizeof((array)[0]))
+
+// Command-line options specific to unisetperf.
+// Options do not have abbreviations: Force readable command lines.
+// (Using U+0001 for abbreviation characters.)
+enum {
+    SET_PATTERN,
+    FAST_TYPE,
+    UNISETPERF_OPTIONS_COUNT
+};
+
+static UOption options[UNISETPERF_OPTIONS_COUNT]={
+    UOPTION_DEF("pattern", '\x01', UOPT_REQUIRES_ARG),
+    UOPTION_DEF("type",    '\x01', UOPT_REQUIRES_ARG)
+};
+
+static const char *const unisetperf_usage =
+    "\t--pattern   UnicodeSet pattern for instantiation.\n"
+    "\t            Default: [:ID_Continue:]\n"
+    "\t--type      Type of UnicodeSet: slow fast\n"
+    "\t            Default: slow\n";
+
+// Test object with setup data.
+class UnicodeSetPerformanceTest : public UPerfTest {
+public:
+    UnicodeSetPerformanceTest(int32_t argc, const char *argv[], UErrorCode &status)
+            : UPerfTest(argc, argv, options, LENGTHOF(options), unisetperf_usage, status),
+              utf8(NULL), utf8Length(0), countInputCodePoints(0), spanCount(0) {
+        if (U_SUCCESS(status)) {
+            UnicodeString pattern=UnicodeString(options[SET_PATTERN].value, -1, US_INV).unescape();
+            set.applyPattern(pattern, status);
+            prefrozen=set;
+            if(0==strcmp(options[FAST_TYPE].value, "fast")) {
+                set.freeze();
+            }
+
+            int32_t inputLength;
+            UPerfTest::getBuffer(inputLength, status);
+            if(U_SUCCESS(status) && inputLength>0) {
+                countInputCodePoints = u_countChar32(buffer, bufferLen);
+
+                countSpans();
+
+                // Preflight the UTF-8 length and allocate utf8.
+                u_strToUTF8(NULL, 0, &utf8Length, buffer, bufferLen, &status);
+                if(status==U_BUFFER_OVERFLOW_ERROR) {
+                    utf8=(char *)malloc(utf8Length);
+                    if(utf8!=NULL) {
+                        status=U_ZERO_ERROR;
+                        u_strToUTF8(utf8, utf8Length, NULL, buffer, bufferLen, &status);
+                    } else {
+                        status=U_MEMORY_ALLOCATION_ERROR;
+                    }
+                }
+
+                if(verbose) {
+                    printf("code points:%ld  len16:%ld  len8:%ld  spans:%ld  "
+                           "cp/span:%.3g  UChar/span:%.3g  B/span:%.3g  B/cp:%.3g\n",
+                           (long)countInputCodePoints, (long)bufferLen, (long)utf8Length, (long)spanCount,
+                           (double)countInputCodePoints/spanCount, (double)bufferLen/spanCount, (double)utf8Length/spanCount,
+                           (double)utf8Length/countInputCodePoints);
+                }
+            }
+        }
+    }
+
+    virtual UPerfFunction* runIndexedTest(int32_t index, UBool exec, const char* &name, char* par = NULL);
+
+    // Count spans of characters that are in the set,
+    // and spans of characters that are not in the set.
+    // If the very first character is in the set, then one additional
+    // not-span is counted.
+    void countSpans() {
+        const UChar *s=getBuffer();
+        int32_t length=getBufferLen();
+        int32_t i=0;
+        UBool tf=FALSE;
+        while(i<length) {
+            i=span(s, length, i, tf);
+            tf=(UBool)(!tf);
+            ++spanCount;
+        }
+    }
+    int32_t span(const UChar *s, int32_t length, int32_t start, UBool tf) const {
+        UChar32 c;
+        int32_t prev;
+        while((prev=start)<length) {
+            U16_NEXT(s, start, length, c);
+            if(tf!=set.contains(c)) {
+                break;
+            }
+        }
+        return prev;
+    }
+
+    const UChar *getBuffer() const { return buffer; }
+    int32_t getBufferLen() const { return bufferLen; }
+
+    char *utf8;
+    int32_t utf8Length;
+
+    // Number of code points in the input text.
+    int32_t countInputCodePoints;
+    int32_t spanCount;
+
+    UnicodeSet set;
+    UnicodeSet prefrozen;
+};
+
+// Performance test function object.
+class Command : public UPerfFunction {
+protected:
+    Command(const UnicodeSetPerformanceTest &testcase) : testcase(testcase) {}
+
+public:
+    virtual ~Command() {}
+
+    // virtual void call(UErrorCode* pErrorCode) { ... }
+
+    virtual long getOperationsPerIteration() {
+        // Number of code points tested:
+        // Input code points, plus one for the end of each span except the last span.
+        return testcase.countInputCodePoints+testcase.spanCount-1;
+    }
+
+    virtual long getEventsPerIteration() {
+        return testcase.spanCount;
+    }
+
+    const UnicodeSetPerformanceTest &testcase;
+};
+
+class Contains : public Command {
+protected:
+    Contains(const UnicodeSetPerformanceTest &testcase) : Command(testcase) {
+        // Verify that the frozen set is equal to the unfrozen one.
+        UnicodeSet set;
+        UChar32 c;
+
+        for(c=0; c<=0x10ffff; ++c) {
+            if(testcase.set.contains(c)) {
+                set.add(c);
+            }
+        }
+        if(set!=testcase.set) {
+            fprintf(stderr, "error: frozen set != original!\n");
+        }
+    }
+public:
+    static UPerfFunction* get(const UnicodeSetPerformanceTest &testcase) {
+        return new Contains(testcase);
+    }
+    virtual void call(UErrorCode* pErrorCode) {
+        const UnicodeSet &set=testcase.set;
+        const UChar *s=testcase.getBuffer();
+        int32_t length=testcase.getBufferLen();
+        int32_t count=0;
+        int32_t i=0;
+        UBool tf=FALSE;
+        while(i<length) {
+            i+=span(set, s+i, length-i, tf);
+            tf=(UBool)(!tf);
+            ++count;
+        }
+        if(count!=testcase.spanCount) {
+            fprintf(stderr, "error: Contains() count=%ld != %ld=UnicodeSetPerformanceTest.spanCount\n",
+                    (long)count, (long)testcase.spanCount);
+        }
+    }
+    static int32_t span(const UnicodeSet &set, const UChar *s, int32_t length, UBool tf) {
+        UChar32 c;
+        int32_t start=0, prev;
+        while((prev=start)<length) {
+            U16_NEXT(s, start, length, c);
+            if(tf!=set.contains(c)) {
+                break;
+            }
+        }
+        return prev;
+    }
+};
+
+class SpanUTF16 : public Command {
+protected:
+    SpanUTF16(const UnicodeSetPerformanceTest &testcase) : Command(testcase) {
+        // Verify that the frozen set is equal to the unfrozen one.
+        UnicodeSet set;
+        UChar utf16[2];
+        UChar32 c, c2;
+
+        for(c=0; c<=0xffff; ++c) {
+            utf16[0]=(UChar)c;
+            if(testcase.set.span(utf16, 1, USET_SPAN_CONTAINED)>0) {
+                set.add(c);
+            }
+        }
+        for(c=0xd800; c<=0xdbff; ++c) {
+            utf16[0]=(UChar)c;
+            for(c2=0xdc00; c2<=0xdfff; ++c2) {
+                utf16[1]=(UChar)c2;
+                if(testcase.set.span(utf16, 2, USET_SPAN_CONTAINED)>0) {
+                    set.add(U16_GET_SUPPLEMENTARY(c, c2));
+                }
+            }
+        }
+
+        if(set!=testcase.set) {
+            fprintf(stderr, "error: frozen set != original!\n");
+        }
+    }
+public:
+    static UPerfFunction* get(const UnicodeSetPerformanceTest &testcase) {
+        return new SpanUTF16(testcase);
+    }
+    virtual void call(UErrorCode* pErrorCode) {
+        const UnicodeSet &set=testcase.set;
+        const UChar *s=testcase.getBuffer();
+        int32_t length=testcase.getBufferLen();
+        int32_t count=0;
+        int32_t i=0;
+        UBool tf=FALSE;
+        while(i<length) {
+            i+=set.span(s+i, length-i, (USetSpanCondition)tf);
+            tf=(UBool)(!tf);
+            ++count;
+        }
+        if(count!=testcase.spanCount) {
+            fprintf(stderr, "error: SpanUTF16() count=%ld != %ld=UnicodeSetPerformanceTest.spanCount\n",
+                    (long)count, (long)testcase.spanCount);
+        }
+    }
+};
+
+class SpanBackUTF16 : public Command {
+protected:
+    SpanBackUTF16(const UnicodeSetPerformanceTest &testcase) : Command(testcase) {
+        // Verify that the frozen set is equal to the unfrozen one.
+        UnicodeSet set;
+        UChar utf16[2];
+        UChar32 c, c2;
+
+        for(c=0; c<=0xffff; ++c) {
+            utf16[0]=(UChar)c;
+            if(testcase.set.spanBack(utf16, 1, USET_SPAN_CONTAINED)==0) {
+                set.add(c);
+            }
+        }
+        for(c=0xd800; c<=0xdbff; ++c) {
+            utf16[0]=(UChar)c;
+            for(c2=0xdc00; c2<=0xdfff; ++c2) {
+                utf16[1]=(UChar)c2;
+                if(testcase.set.spanBack(utf16, 2, USET_SPAN_CONTAINED)==0) {
+                    set.add(U16_GET_SUPPLEMENTARY(c, c2));
+                }
+            }
+        }
+
+        if(set!=testcase.set) {
+            fprintf(stderr, "error: frozen set != original!\n");
+        }
+    }
+public:
+    static UPerfFunction* get(const UnicodeSetPerformanceTest &testcase) {
+        return new SpanBackUTF16(testcase);
+    }
+    virtual void call(UErrorCode* pErrorCode) {
+        const UnicodeSet &set=testcase.set;
+        const UChar *s=testcase.getBuffer();
+        int32_t length=testcase.getBufferLen();
+        int32_t count=0;
+        /*
+         * Get the same spans as with span() where we always start with a not-contained span.
+         * If testcase.spanCount is an odd number, then the last span() was not-contained.
+         * The last spanBack() must be not-contained to match the first span().
+         */
+        UBool tf=(UBool)((testcase.spanCount&1)==0);
+        while(length>0 || !tf) {
+            length=set.spanBack(s, length, (USetSpanCondition)tf);
+            tf=(UBool)(!tf);
+            ++count;
+        }
+        if(count!=testcase.spanCount) {
+            fprintf(stderr, "error: SpanBackUTF16() count=%ld != %ld=UnicodeSetPerformanceTest.spanCount\n",
+                    (long)count, (long)testcase.spanCount);
+        }
+    }
+};
+
+class SpanUTF8 : public Command {
+protected:
+    SpanUTF8(const UnicodeSetPerformanceTest &testcase) : Command(testcase) {
+        // Verify that the frozen set is equal to the unfrozen one.
+        UnicodeSet set;
+        char utf8[4];
+        UChar32 c;
+        int32_t length;
+
+        for(c=0; c<=0x10ffff; ++c) {
+            if(c==0xd800) {
+                c=0xe000;
+            }
+            length=0;
+            U8_APPEND_UNSAFE(utf8, length, c);
+            if(testcase.set.spanUTF8(utf8, length, USET_SPAN_CONTAINED)>0) {
+                set.add(c);
+            }
+        }
+        if(set!=testcase.set) {
+            fprintf(stderr, "error: frozen set != original!\n");
+        }
+    }
+public:
+    static UPerfFunction* get(const UnicodeSetPerformanceTest &testcase) {
+        return new SpanUTF8(testcase);
+    }
+    virtual void call(UErrorCode* pErrorCode) {
+        const UnicodeSet &set=testcase.set;
+        const char *s=testcase.utf8;
+        int32_t length=testcase.utf8Length;
+        int32_t count=0;
+        int32_t i=0;
+        UBool tf=FALSE;
+        while(i<length) {
+            i+=set.spanUTF8(s+i, length-i, (USetSpanCondition)tf);
+            tf=(UBool)(!tf);
+            ++count;
+        }
+        if(count!=testcase.spanCount) {
+            fprintf(stderr, "error: SpanUTF8() count=%ld != %ld=UnicodeSetPerformanceTest.spanCount\n",
+                    (long)count, (long)testcase.spanCount);
+        }
+    }
+};
+
+class SpanBackUTF8 : public Command {
+protected:
+    SpanBackUTF8(const UnicodeSetPerformanceTest &testcase) : Command(testcase) {
+        // Verify that the frozen set is equal to the unfrozen one.
+        UnicodeSet set;
+        char utf8[4];
+        UChar32 c;
+        int32_t length;
+
+        for(c=0; c<=0x10ffff; ++c) {
+            if(c==0xd800) {
+                c=0xe000;
+            }
+            length=0;
+            U8_APPEND_UNSAFE(utf8, length, c);
+            if(testcase.set.spanBackUTF8(utf8, length, USET_SPAN_CONTAINED)==0) {
+                set.add(c);
+            }
+        }
+        if(set!=testcase.set) {
+            fprintf(stderr, "error: frozen set != original!\n");
+        }
+    }
+public:
+    static UPerfFunction* get(const UnicodeSetPerformanceTest &testcase) {
+        return new SpanBackUTF8(testcase);
+    }
+    virtual void call(UErrorCode* pErrorCode) {
+        const UnicodeSet &set=testcase.set;
+        const char *s=testcase.utf8;
+        int32_t length=testcase.utf8Length;
+        int32_t count=0;
+        /*
+         * Get the same spans as with span() where we always start with a not-contained span.
+         * If testcase.spanCount is an odd number, then the last span() was not-contained.
+         * The last spanBack() must be not-contained to match the first span().
+         */
+        UBool tf=(UBool)((testcase.spanCount&1)==0);
+        while(length>0 || !tf) {
+            length=set.spanBackUTF8(s, length, (USetSpanCondition)tf);
+            tf=(UBool)(!tf);
+            ++count;
+        }
+        if(count!=testcase.spanCount) {
+            fprintf(stderr, "error: SpanBackUTF8() count=%ld != %ld=UnicodeSetPerformanceTest.spanCount\n",
+                    (long)count, (long)testcase.spanCount);
+        }
+    }
+};
+
+UPerfFunction* UnicodeSetPerformanceTest::runIndexedTest(int32_t index, UBool exec, const char* &name, char* par) {
+    switch (index) {
+        case 0: name = "Contains";     if (exec) return Contains::get(*this); break;
+        case 1: name = "SpanUTF16";    if (exec) return SpanUTF16::get(*this); break;
+        case 2: name = "SpanBackUTF16";if (exec) return SpanBackUTF16::get(*this); break;
+        case 3: name = "SpanUTF8";     if (exec) return SpanUTF8::get(*this); break;
+        case 4: name = "SpanBackUTF8"; if (exec) return SpanBackUTF8::get(*this); break;
+        default: name = ""; break;
+    }
+    return NULL;
+}
+
+int main(int argc, const char *argv[])
+{
+    // Default values for command-line options.
+    options[SET_PATTERN].value = "[:ID_Continue:]";
+    options[FAST_TYPE].value = "slow";
+
+    UErrorCode status = U_ZERO_ERROR;
+    UnicodeSetPerformanceTest test(argc, argv, status);
+
+        if (U_FAILURE(status)){
+        printf("The error is %s\n", u_errorName(status));
+        test.usage();
+        return status;
+    }
+        
+    if (test.run() == FALSE){
+        fprintf(stderr, "FAILED: Tests could not be run, please check the "
+                                    "arguments.\n");
+        return 1;
+    }
+
+    return 0;
+}