1/*
2**********************************************************************
3* Copyright (c) 2002-2005, International Business Machines
4* Corporation and others.  All Rights Reserved.
5**********************************************************************
6* 2005Nov22 Raymond Yang
7*
8* migrate old test created by aliu to perf test framework.
9*/
10
11#include <stdio.h>
12
13#include "unicode/utypes.h"
14#include "unicode/uniset.h"
15#include "unicode/uchar.h"
16#include "unicode/usetiter.h"
17#include "bitset.h"
18#include "unicode/uperf.h"
19
20static const char* PAT[] = {
21    "['A-Za-z\\u00C0-\\u00C5\\u00C7-\\u00CF\\u00D1-\\u00D6\\u00D9-\\u00DD\\u00E0-\\u00E5\\u00E7-\\u00EF\\u00F1-\\u00F6\\u00F9-\\u00FD\\u00FF-\\u010F\\u0112-\\u0125\\u0128-\\u0130\\u0134-\\u0137\\u0139-\\u013E\\u0143-\\u0148\\u014C-\\u0151\\u0154-\\u0165\\u0168-\\u017E\\u01A0-\\u01A1\\u01AF-\\u01B0\\u01CD-\\u01DC\\u01DE-\\u01E1\\u01E6-\\u01ED\\u01F0\\u01F4-\\u01F5\\u01F8-\\u01FB\\u0200-\\u021B\\u021E-\\u021F\\u0226-\\u0233\\u1E00-\\u1E99\\u1EA0-\\u1EF9\\u212A-\\u212B]",
22
23    "['.0-9A-Za-z~\\u00C0-\\u00C5\\u00C7-\\u00CF\\u00D1-\\u00D6\\u00D9-\\u00DD\\u00E0-\\u00E5\\u00E7-\\u00EF\\u00F1-\\u00F6\\u00F9-\\u00FD\\u00FF-\\u010F\\u0112-\\u0125\\u0128-\\u0130\\u0134-\\u0137\\u0139-\\u013E\\u0143-\\u0148\\u014C-\\u0151\\u0154-\\u0165\\u0168-\\u017E\\u01A0-\\u01A1\\u01AF-\\u01B0\\u01CD-\\u01DC\\u01DE-\\u01E3\\u01E6-\\u01ED\\u01F0\\u01F4-\\u01F5\\u01F8-\\u021B\\u021E-\\u021F\\u0226-\\u0233\\u0301\\u0303-\\u0304\\u0306-\\u0307\\u0310\\u0314-\\u0315\\u0323\\u0325\\u0331\\u0341\\u0344\\u0385-\\u0386\\u0388-\\u038A\\u038C\\u038E-\\u0390\\u03AC-\\u03B0\\u03CC-\\u03CE\\u03D3\\u0403\\u040C\\u040E\\u0419\\u0439\\u0453\\u045C\\u045E\\u04C1-\\u04C2\\u04D0-\\u04D1\\u04D6-\\u04D7\\u04E2-\\u04E3\\u04EE-\\u04EF\\u1E00-\\u1E99\\u1EA0-\\u1EF9\\u1F01\\u1F03-\\u1F05\\u1F07\\u1F09\\u1F0B-\\u1F0D\\u1F0F\\u1F11\\u1F13-\\u1F15\\u1F19\\u1F1B-\\u1F1D\\u1F21\\u1F23-\\u1F25\\u1F27\\u1F29\\u1F2B-\\u1F2D\\u1F2F\\u1F31\\u1F33-\\u1F35\\u1F37\\u1F39\\u1F3B-\\u1F3D\\u1F3F\\u1F41\\u1F43-\\u1F45\\u1F49\\u1F4B-\\u1F4D\\u1F51\\u1F53-\\u1F55\\u1F57\\u1F59\\u1F5B\\u1F5D\\u1F5F\\u1F61\\u1F63-\\u1F65\\u1F67\\u1F69\\u1F6B-\\u1F6D\\u1F6F\\u1F71\\u1F73\\u1F75\\u1F77\\u1F79\\u1F7B\\u1F7D\\u1F81\\u1F83-\\u1F85\\u1F87\\u1F89\\u1F8B-\\u1F8D\\u1F8F\\u1F91\\u1F93-\\u1F95\\u1F97\\u1F99\\u1F9B-\\u1F9D\\u1F9F\\u1FA1\\u1FA3-\\u1FA5\\u1FA7\\u1FA9\\u1FAB-\\u1FAD\\u1FAF-\\u1FB1\\u1FB4\\u1FB8-\\u1FB9\\u1FBB\\u1FC4\\u1FC9\\u1FCB\\u1FCE\\u1FD0-\\u1FD1\\u1FD3\\u1FD8-\\u1FD9\\u1FDB\\u1FDE\\u1FE0-\\u1FE1\\u1FE3\\u1FE5\\u1FE8-\\u1FE9\\u1FEB-\\u1FEC\\u1FEE\\u1FF4\\u1FF9\\u1FFB\\u212A-\\u212B\\uE04D\\uE064]",
24
25    "[\\u0901-\\u0903\\u0905-\\u0939\\u093C-\\u094D\\u0950-\\u0954\\u0958-\\u096F]",
26};
27
28class CmdPattern : public UPerfFunction {
29private:
30    UnicodeString pat;
31    UnicodeSet set;
32public:
33    CmdPattern(const char * pattern):pat(pattern,""){
34    }
35    virtual long getOperationsPerIteration(){
36        return 1;
37    }
38    virtual void call(UErrorCode* pErrorCode){
39        set.applyPattern(pat, *pErrorCode);
40    }
41};
42
43class CmdOp : public UPerfFunction {
44private:
45    UnicodeSet us;
46    BitSet bs;
47    int32_t total;
48    void (CmdOp::*op) ();
49public:
50    CmdOp(UCharCategory prop, void (CmdOp::*op)()):op(op){
51        total = 0;
52        bs.clearAll();
53        for (UChar32 cp=0; cp<0x110000; ++cp) {
54            if (u_charType(cp) == prop) {
55                bs.set((int32_t) cp);
56                ++total;
57            }
58        }
59    }
60    virtual long getOperationsPerIteration(){
61        return total;
62    }
63
64    virtual void call(UErrorCode* pErrorCode){
65        (this->*op)();
66    }
67    void add (void){
68        us.clear();
69        for (UChar32 cp=0; cp<0x110000; ++cp) {
70            if (bs.get((int32_t) cp)) {
71                us.add(cp);
72            }
73        }
74    }
75
76    void contains(void){
77        int32_t temp = 0;
78        us.clear();
79        for (UChar32 cp=0; cp<0x110000; ++cp) {
80            if (us.contains(cp)) {
81                temp += cp;
82            }
83        }
84    }
85
86    void iterator(void){
87        int32_t temp = 0;
88        UnicodeSetIterator uit(us);
89        while (uit.next()) {
90            temp += uit.getCodepoint();
91        }
92    }
93};
94
95class  UsetPerformanceTest : public UPerfTest{
96public:
97    UsetPerformanceTest(int32_t argc, const char *argv[], UErrorCode &status) :UPerfTest(argc,argv,status){
98    }
99
100    virtual UPerfFunction* runIndexedTest( int32_t index, UBool exec, const char* &name, char* par = NULL ){
101        switch (index) {
102            case 0: name = "titlecase_letter_add";
103                if (exec) return new CmdOp(U_TITLECASE_LETTER, &CmdOp::add) ; break;
104            case 1: name = "titlecase_letter_contains";
105                if (exec) return new CmdOp(U_TITLECASE_LETTER, &CmdOp::contains)  ; break;
106            case 2: name = "titlecase_letter_iterator";
107                if (exec) return new CmdOp(U_TITLECASE_LETTER, &CmdOp::iterator)  ; break;
108            case 3: name = "unassigned_add";
109                if (exec) return new CmdOp(U_UNASSIGNED, &CmdOp::add)  ; break;
110            case 4: name = "unassigned_contains";
111                if (exec) return new CmdOp(U_UNASSIGNED, &CmdOp::contains)  ; break;
112            case 5: name = "unassigned_iterator";
113                if (exec) return new CmdOp(U_UNASSIGNED, &CmdOp::iterator)  ; break;
114            case 6: name = "pattern1";
115                if (exec) return new CmdPattern(PAT[0])  ; break;
116            case 7: name = "pattern2";
117                if (exec) return new CmdPattern(PAT[1])  ; break;
118            case 8: name = "pattern3";
119                if (exec) return new CmdPattern(PAT[2])  ; break;
120            default: name = ""; break;
121        }
122        return NULL;
123    }
124};
125
126
127int main(int argc, const char *argv[])
128{
129    UErrorCode status = U_ZERO_ERROR;
130    UsetPerformanceTest test(argc, argv, status);
131
132	if (U_FAILURE(status)){
133        printf("The error is %s\n", u_errorName(status));
134        return status;
135    }
136
137    if (test.run() == FALSE){
138        fprintf(stderr, "FAILED: Tests could not be run please check the "
139			            "arguments.\n");
140        return -1;
141    }
142    return 0;
143}
144