]> rtime.felk.cvut.cz Git - opencv.git/blob - opencv/tests/cxts/cxts.cpp
added help information
[opencv.git] / opencv / tests / cxts / cxts.cpp
1 /*M///////////////////////////////////////////////////////////////////////////////////////
2 //
3 //  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
4 //
5 //  By downloading, copying, installing or using the software you agree to this license.
6 //  If you do not agree to this license, do not download, install,
7 //  copy or use the software.
8 //
9 //
10 //                        Intel License Agreement
11 //                For Open Source Computer Vision Library
12 //
13 // Copyright (C) 2000, Intel Corporation, all rights reserved.
14 // Third party copyrights are property of their respective owners.
15 //
16 // Redistribution and use in source and binary forms, with or without modification,
17 // are permitted provided that the following conditions are met:
18 //
19 //   * Redistribution's of source code must retain the above copyright notice,
20 //     this list of conditions and the following disclaimer.
21 //
22 //   * Redistribution's in binary form must reproduce the above copyright notice,
23 //     this list of conditions and the following disclaimer in the documentation
24 //     and/or other materials provided with the distribution.
25 //
26 //   * The name of Intel Corporation may not be used to endorse or promote products
27 //     derived from this software without specific prior written permission.
28 //
29 // This software is provided by the copyright holders and contributors "as is" and
30 // any express or implied warranties, including, but not limited to, the implied
31 // warranties of merchantability and fitness for a particular purpose are disclaimed.
32 // In no event shall the Intel Corporation or contributors be liable for any direct,
33 // indirect, incidental, special, exemplary, or consequential damages
34 // (including, but not limited to, procurement of substitute goods or services;
35 // loss of use, data, or profits; or business interruption) however caused
36 // and on any theory of liability, whether in contract, strict liability,
37 // or tort (including negligence or otherwise) arising in any way out of
38 // the use of this software, even if advised of the possibility of such damage.
39 //
40 //M*/
41
42 #include "_cxts.h"
43 #include <ctype.h>
44 #include <stdarg.h>
45 #include <fcntl.h>
46 #include <time.h>
47 #if defined WIN32 || defined WIN64
48 #include <io.h>
49 #else
50 #include <unistd.h>
51 #endif
52
53 CvTest* CvTest::first = 0;
54 CvTest* CvTest::last = 0;
55 int CvTest::test_count = 0;
56
57 /*****************************************************************************************\
58 *                                Exception and memory handlers                            *
59 \*****************************************************************************************/
60
61 // a few platform-dependent declarations
62
63 #define CV_TS_NORMAL 0
64 #define CV_TS_BLUE   1
65 #define CV_TS_GREEN  2
66 #define CV_TS_RED    4
67
68 #if defined WIN32 || defined WIN64
69 #include <windows.h>
70
71 #ifdef _MSC_VER
72 #include <eh.h>
73 #endif
74
75 #ifdef _MSC_VER
76 static void cv_seh_translator( unsigned int /*u*/, EXCEPTION_POINTERS* pExp )
77 {
78     int code = CvTS::FAIL_EXCEPTION;
79     switch( pExp->ExceptionRecord->ExceptionCode )
80     {
81     case EXCEPTION_ACCESS_VIOLATION:
82     case EXCEPTION_ARRAY_BOUNDS_EXCEEDED:
83     case EXCEPTION_DATATYPE_MISALIGNMENT:
84     case EXCEPTION_FLT_STACK_CHECK:
85     case EXCEPTION_STACK_OVERFLOW:
86     case EXCEPTION_IN_PAGE_ERROR:
87         code = CvTS::FAIL_MEMORY_EXCEPTION;
88         break;
89     case EXCEPTION_FLT_DENORMAL_OPERAND:
90     case EXCEPTION_FLT_DIVIDE_BY_ZERO:
91     case EXCEPTION_FLT_INEXACT_RESULT:
92     case EXCEPTION_FLT_INVALID_OPERATION:
93     case EXCEPTION_FLT_OVERFLOW:
94     case EXCEPTION_FLT_UNDERFLOW:
95     case EXCEPTION_INT_DIVIDE_BY_ZERO:
96     case EXCEPTION_INT_OVERFLOW:
97         code = CvTS::FAIL_ARITHM_EXCEPTION;
98         break;
99     case EXCEPTION_BREAKPOINT:
100     case EXCEPTION_ILLEGAL_INSTRUCTION:
101     case EXCEPTION_INVALID_DISPOSITION:
102     case EXCEPTION_NONCONTINUABLE_EXCEPTION:
103     case EXCEPTION_PRIV_INSTRUCTION:
104     case EXCEPTION_SINGLE_STEP:
105         code = CvTS::FAIL_EXCEPTION;
106     }
107     throw code;
108 }
109 #endif
110
111 #define CV_TS_TRY_BLOCK_BEGIN                   \
112     try {
113
114 #define CV_TS_TRY_BLOCK_END                     \
115     } catch( int _code ) {                      \
116         ts->set_failed_test_info( _code );      \
117     }
118
119 static void change_color( int color )
120 {
121     static int normal_attributes = -1;
122     HANDLE hstdout = GetStdHandle(STD_OUTPUT_HANDLE);
123     fflush(stdout);
124
125     if( normal_attributes < 0 )
126     {
127         CONSOLE_SCREEN_BUFFER_INFO info;
128         GetConsoleScreenBufferInfo( hstdout, &info );
129         normal_attributes = info.wAttributes;
130     }
131
132     SetConsoleTextAttribute( hstdout,
133         (WORD)(color == CV_TS_NORMAL ? normal_attributes :
134         ((color & CV_TS_BLUE ? FOREGROUND_BLUE : 0)|
135         (color & CV_TS_GREEN ? FOREGROUND_GREEN : 0)|
136         (color & CV_TS_RED ? FOREGROUND_RED : 0)|FOREGROUND_INTENSITY)) );
137 }
138
139 #else
140
141 #include <signal.h>
142
143 static const int cv_ts_sig_id[] = { SIGSEGV, SIGBUS, SIGFPE, SIGILL, -1 };
144
145 static jmp_buf cv_ts_jmp_mark;
146
147 void cv_signal_handler( int sig_code )
148 {
149     int code = CvTS::FAIL_EXCEPTION;
150     switch( sig_code )
151     {
152     case SIGFPE:
153         code = CvTS::FAIL_ARITHM_EXCEPTION;
154         break;
155     case SIGSEGV:
156     case SIGBUS:
157         code = CvTS::FAIL_ARITHM_EXCEPTION;
158         break;
159     case SIGILL:
160         code = CvTS::FAIL_EXCEPTION;
161     }
162
163     longjmp( cv_ts_jmp_mark, code );
164 }
165
166 #define CV_TS_TRY_BLOCK_BEGIN                   \
167     {                                           \
168         int _code = setjmp( cv_ts_jmp_mark );   \
169         if( !_code ) {
170
171 #define CV_TS_TRY_BLOCK_END                     \
172         }                                       \
173         else  {                                 \
174             ts->set_failed_test_info( _code );  \
175         }                                       \
176     }
177
178 static void change_color( int color )
179 {
180     static const uchar ansi_tab[] = { 30, 34, 32, 36, 31, 35, 33, 37 };
181     char buf[16];
182     int code = 0;
183     fflush( stdout );
184     if( color != CV_TS_NORMAL )
185         code = ansi_tab[color & (CV_TS_BLUE|CV_TS_GREEN|CV_TS_RED)];
186     sprintf( buf, "\x1b[%dm", code );
187     fputs( buf, stdout );
188 }
189
190 #endif
191
192
193 /***************************** memory manager *****************************/
194
195 typedef struct CvTestAllocBlock
196 {
197     struct CvTestAllocBlock* prev;
198     struct CvTestAllocBlock* next;
199     char* origin;
200     char* data;
201     size_t size;
202     int index;
203 }
204 CvTestAllocBlock;
205
206
207 class CvTestMemoryManager
208 {
209 public:
210     CvTestMemoryManager( CvTS* ts );
211     virtual ~CvTestMemoryManager();
212
213     virtual void clear_and_check( int min_index = -1 );
214     virtual void start_tracking( int index_to_stop_at=-1 );
215     virtual void stop_tracking_and_check();
216     int get_alloc_index() { return index; }
217
218     static void* alloc_proxy( size_t size, void* userdata );
219     static int free_proxy( void* ptr, void* userdata );
220
221 protected:
222     virtual void* alloc( size_t size );
223     virtual int free( void* ptr );
224     virtual int free_block( CvTestAllocBlock* block );
225
226     int index;
227     int track_blocks;
228     int show_msg_box;
229     int index_to_stop_at;
230     const char* guard_pattern;
231     int guard_size;
232     int block_align;
233     enum { MAX_MARKS = 1024 };
234     int marks[MAX_MARKS];
235     int marks_top;
236     CvTS* ts;
237     CvTestAllocBlock* first;
238     CvTestAllocBlock* last;
239 };
240
241
242 void* CvTestMemoryManager::alloc_proxy( size_t size, void* userdata )
243 {
244     return ((CvTestMemoryManager*)userdata)->alloc( size );
245 }
246
247
248 int CvTestMemoryManager::free_proxy( void* ptr, void* userdata )
249 {
250     return ((CvTestMemoryManager*)userdata)->free( ptr );
251 }
252
253
254 CvTestMemoryManager::CvTestMemoryManager( CvTS* _test_system )
255 {
256     ts = _test_system;
257     guard_pattern = "THIS IS A GUARD PATTERN!";
258     guard_size = (int)strlen(guard_pattern);
259     block_align = CV_MALLOC_ALIGN;
260     track_blocks = 0;
261     marks_top = 0;
262     first = last = 0;
263     index = 0;
264     index_to_stop_at = -1;
265     show_msg_box = 1;
266 }
267
268
269 CvTestMemoryManager::~CvTestMemoryManager()
270 {
271     clear_and_check();
272 }
273
274
275 void CvTestMemoryManager::clear_and_check( int min_index )
276 {
277     int alloc_index = -1;
278     CvTestAllocBlock* block;
279     int leak_size = 0, leak_block_count = 0, mem_size = 0;
280     void* mem_addr = 0;
281
282     while( marks_top > 0 && marks[marks_top - 1] >= min_index )
283         marks_top--;
284
285     for( block = last; block != 0; )
286     {
287         CvTestAllocBlock* prev = block->prev;
288         if( block->index < min_index )
289             break;
290         leak_size += (int)block->size;
291         leak_block_count++;
292         alloc_index = block->index;
293         mem_addr = block->data;
294         mem_size = (int)block->size;
295         free_block( block );
296         block = prev;
297     }
298     track_blocks--;
299     if( leak_block_count > 0 )
300     {
301         ts->set_failed_test_info( CvTS::FAIL_MEMORY_LEAK, alloc_index );
302         ts->printf( CvTS::LOG, "Memory leaks: %u blocks, %u bytes total\n"
303                     "%s leaked block: %p, %u bytes\n",
304                     leak_block_count, leak_size, leak_block_count > 1 ? "The first" : "The",
305                     mem_addr, mem_size );
306     }
307
308     index = block ? block->index + 1 : 0;
309 }
310
311
312 void CvTestMemoryManager::start_tracking( int _index_to_stop_at )
313 {
314     track_blocks--;
315     marks[marks_top++] = index;
316     assert( marks_top <= MAX_MARKS );
317     track_blocks+=2;
318     index_to_stop_at = _index_to_stop_at >= index ? _index_to_stop_at : -1;
319 }
320
321
322 void CvTestMemoryManager::stop_tracking_and_check()
323 {
324     if( marks_top > 0 )
325     {
326         int min_index = marks[--marks_top];
327         clear_and_check( min_index );
328     }
329 }
330
331
332 int CvTestMemoryManager::free_block( CvTestAllocBlock* block )
333 {
334     int code = 0;
335     char* data = block->data;
336
337     if( block->origin == 0 || ((size_t)block->origin & (sizeof(double)-1)) != 0 )
338         code = CvTS::FAIL_MEMORY_CORRUPTION_BEGIN;
339
340     if( memcmp( data - guard_size, guard_pattern, guard_size ) != 0 )
341         code = CvTS::FAIL_MEMORY_CORRUPTION_BEGIN;
342     else if( memcmp( data + block->size, guard_pattern, guard_size ) != 0 )
343         code = CvTS::FAIL_MEMORY_CORRUPTION_END;
344
345     if( code >= 0 )
346     {
347         if( block->prev )
348             block->prev->next = block->next;
349         else if( first == block )
350             first = block->next;
351
352         if( block->next )
353             block->next->prev = block->prev;
354         else if( last == block )
355             last = block->prev;
356
357         free( block->origin );
358     }
359     else
360     {
361         ts->set_failed_test_info( code, block->index );
362         ts->printf( CvTS::LOG, "Corrupted block (%s): %p, %u bytes\n",
363                     code == CvTS::FAIL_MEMORY_CORRUPTION_BEGIN ? "beginning" : "end",
364                     block->data, block->size );
365     }
366
367     return code;
368 }
369
370
371 void* CvTestMemoryManager::alloc( size_t size )
372 {
373     char* data;
374     CvTestAllocBlock* block;
375     size_t new_size = sizeof(*block) + size + guard_size*2 + block_align + sizeof(size_t)*2;
376     char* ptr = (char*)malloc( new_size );
377
378     if( !ptr )
379         return 0;
380
381     data = (char*)cvAlignPtr( ptr + sizeof(size_t) + sizeof(*block) + guard_size, block_align );
382     block = (CvTestAllocBlock*)cvAlignPtr( data - guard_size -
383             sizeof(size_t) - sizeof(*block), sizeof(size_t) );
384     block->origin = ptr;
385     block->data = data;
386     block->size = 0;
387     block->index = -1;
388     block->next = block->prev = 0;
389     memcpy( data - guard_size, guard_pattern, guard_size );
390     memcpy( data + size, guard_pattern, guard_size );
391
392     if( track_blocks > 0 )
393     {
394         track_blocks--;
395         block->size = size;
396
397         if( index == index_to_stop_at )
398         {
399             if( show_msg_box )
400             {
401         #ifdef WIN32
402                 MessageBox( NULL, "The block that is corrupted and/or not deallocated has been just allocated\n"
403                             "Press Ok to start debugging", "Memory Manager", MB_ICONERROR|MB_OK|MB_SYSTEMMODAL );
404         #endif
405             }
406             CV_DBG_BREAK();
407         }
408
409         block->index = index++;
410
411         block->prev = last;
412         block->next = 0;
413         if( last )
414             last = last->next = block;
415         else
416             first = last = block;
417
418         track_blocks++;
419     }
420
421     return data;
422 }
423
424
425 int CvTestMemoryManager::free( void* ptr )
426 {
427     char* data = (char*)ptr;
428     CvTestAllocBlock* block = (CvTestAllocBlock*)
429         cvAlignPtr( data - guard_size - sizeof(size_t) - sizeof(*block), sizeof(size_t) );
430
431     int code = free_block( block );
432     if( code < 0 && ts->is_debug_mode() )
433         CV_DBG_BREAK();
434     return 0;
435 }
436
437
438 /***************************** error handler *****************************/
439
440 #if 0
441 static int cvTestErrorCallback( int status, const char* func_name, const char* err_msg,
442                          const char* file_name, int line, void* userdata )
443 {
444     if( status < 0 && status != CV_StsBackTrace && status != CV_StsAutoTrace )
445         ((CvTS*)userdata)->set_failed_test_info( CvTS::FAIL_ERROR_IN_CALLED_FUNC );
446
447     // print error message
448     return cvStdErrReport( status, func_name, err_msg, file_name, line, 0 );
449 }
450 #endif
451
452 /*****************************************************************************************\
453 *                                    Base Class for Tests                                 *
454 \*****************************************************************************************/
455
456 CvTest::CvTest( const char* _test_name, const char* _test_funcs, const char* _test_descr ) :
457     name(_test_name ? _test_name : ""), tested_functions(_test_funcs ? _test_funcs : ""),
458     description(_test_descr ? _test_descr : ""), ts(0)
459 {
460     if( last )
461         last->next = this;
462     else
463         first = this;
464     last = this;
465     test_count++;
466     ts = 0;
467     hdr_state = 0;
468
469     timing_param_names = 0;
470     timing_param_current = 0;
471     timing_param_seqs = 0;
472     timing_param_idxs = 0;
473     timing_param_count = -1;
474
475     test_case_count = -1;
476     support_testing_modes = CvTS::CORRECTNESS_CHECK_MODE;
477 }
478
479 CvTest::~CvTest()
480 {
481     clear();
482 }
483
484 CvTest* CvTest::get_first_test()
485 {
486     return first;
487 }
488
489 void CvTest::clear()
490 {
491     if( timing_param_current )
492         free( timing_param_current );
493     if( timing_param_seqs )
494         free( timing_param_seqs );
495     if( timing_param_idxs )
496         free( timing_param_idxs );
497
498     timing_param_current = 0;
499     timing_param_seqs = 0;
500     timing_param_idxs = 0;
501     timing_param_count = -1;
502 }
503
504
505 int CvTest::init( CvTS* _test_system )
506 {
507     clear();
508     ts = _test_system;
509     return read_params( ts->get_file_storage() );
510 }
511
512
513 const char* CvTest::get_parent_name( const char* name, char* buffer )
514 {
515     const char* dash_pos = strrchr( name ? name : "", '-' );
516     if( !dash_pos )
517         return 0;
518
519     if( name != (const char*)buffer )
520         strncpy( buffer, name, dash_pos - name );
521     buffer[dash_pos - name] = '\0';
522     return buffer;
523 }
524
525
526 const CvFileNode* CvTest::find_param( CvFileStorage* fs, const char* param_name )
527 {
528     char buffer[256];
529     const char* name = get_name();
530     CvFileNode* node = 0;
531
532     for(;;)
533     {
534         if( !name )
535             break;
536         node = cvGetFileNodeByName( fs, 0, name );
537         if( node )
538         {
539             node = cvGetFileNodeByName( fs, node, param_name );
540             if( node )
541                 break;
542         }
543         name = get_parent_name( name, buffer );
544     }
545
546     return node;
547 }
548
549
550 void CvTest::start_write_param( CvFileStorage* fs )
551 {
552     if( hdr_state == 0 )
553     {
554         cvStartWriteStruct( fs, get_name(), CV_NODE_MAP );
555         hdr_state = 1;
556     }
557 }
558
559
560 void CvTest::write_param( CvFileStorage* fs, const char* paramname, int val )
561 {
562     if( !ts->find_written_param( this, paramname, CV_NODE_INT, &val) )
563     {
564         start_write_param( fs );
565         cvWriteInt( fs, paramname, val );
566     }
567 }
568
569
570 void CvTest::write_param( CvFileStorage* fs, const char* paramname, double val )
571 {
572     if( !ts->find_written_param( this, paramname, CV_NODE_REAL, &val) )
573     {
574         start_write_param( fs );
575         cvWriteReal( fs, paramname, val );
576     }
577 }
578
579
580 void CvTest::write_param( CvFileStorage* fs, const char* paramname, const char* val )
581 {
582     if( !ts->find_written_param( this, paramname, CV_NODE_STRING, &val) )
583     {
584         start_write_param( fs );
585         cvWriteString( fs, paramname, val );
586     }
587 }
588
589
590 void CvTest::write_string_list( CvFileStorage* fs, const char* paramname, const char** val, int count )
591 {
592     if( val )
593     {
594         start_write_param( fs );
595         int i;
596         if( count < 0 )
597             count = INT_MAX;
598
599         cvStartWriteStruct( fs, paramname, CV_NODE_SEQ + CV_NODE_FLOW );
600         for( i = 0; i < count && val[i] != 0; i++ )
601             cvWriteString( fs, 0, val[i] );
602         cvEndWriteStruct( fs );
603     }
604 }
605
606
607 void CvTest::write_int_list( CvFileStorage* fs, const char* paramname,
608                              const int* val, int count, int stop_value )
609 {
610     if( val )
611     {
612         start_write_param( fs );
613         int i;
614         if( count < 0 )
615             count = INT_MAX;
616
617         cvStartWriteStruct( fs, paramname, CV_NODE_SEQ + CV_NODE_FLOW );
618         for( i = 0; i < count && val[i] != stop_value; i++ )
619             cvWriteInt( fs, 0, val[i] );
620         cvEndWriteStruct( fs );
621     }
622 }
623
624
625 void CvTest::write_real_list( CvFileStorage* fs, const char* paramname,
626                               const double* val, int count, double stop_value )
627 {
628     if( val )
629     {
630         start_write_param( fs );
631         int i;
632         if( count < 0 )
633             count = INT_MAX;
634
635         cvStartWriteStruct( fs, paramname, CV_NODE_SEQ + CV_NODE_FLOW );
636         for( i = 0; i < count && val[i] != stop_value; i++ )
637             cvWriteReal( fs, 0, val[i] );
638         cvEndWriteStruct( fs );
639     }
640 }
641
642
643 int CvTest::read_params( CvFileStorage* fs )
644 {
645     int code = 0;
646
647     if( ts->get_testing_mode() == CvTS::TIMING_MODE )
648     {
649         timing_param_names = find_param( fs, "timing_params" );
650         if( CV_NODE_IS_SEQ(timing_param_names->tag) )
651         {
652             CvSeq* seq = timing_param_names->data.seq;
653             CvSeqReader reader;
654             cvStartReadSeq( seq, &reader );
655             int i;
656
657             timing_param_count = seq->total;
658             timing_param_seqs = (const CvFileNode**)malloc( timing_param_count*sizeof(timing_param_seqs[0]));
659             timing_param_idxs = (int*)malloc( timing_param_count*sizeof(timing_param_idxs[0]));
660             timing_param_current = (const CvFileNode**)malloc( timing_param_count*sizeof(timing_param_current[0]));
661             test_case_count = 1;
662
663             for( i = 0; i < timing_param_count; i++ )
664             {
665                 CvFileNode* param_name = (CvFileNode*)(reader.ptr);
666
667                 if( !CV_NODE_IS_STRING(param_name->tag) )
668                 {
669                     ts->printf( CvTS::LOG, "ERROR: name of timing parameter #%d is not a string\n", i );
670                     code = -1;
671                     break;
672                 }
673
674                 timing_param_idxs[i] = 0;
675                 timing_param_current[i] = 0;
676                 timing_param_seqs[i] = find_param( fs, param_name->data.str.ptr );
677                 if( !timing_param_seqs[i] )
678                 {
679                     ts->printf( CvTS::LOG, "ERROR: timing parameter %s is not found\n", param_name->data.str.ptr );
680                     code = -1;
681                     break;
682                 }
683
684                 if( CV_NODE_IS_SEQ(timing_param_seqs[i]->tag) )
685                     test_case_count *= timing_param_seqs[i]->data.seq->total;
686
687                 CV_NEXT_SEQ_ELEM( seq->elem_size, reader );
688             }
689
690             if( i < timing_param_count )
691                 timing_param_count = 0;
692         }
693         else
694         {
695             ts->printf( CvTS::LOG, "ERROR: \"timing_params\" is not found" );
696             code = -1;
697         }
698     }
699
700     return code;
701 }
702
703
704 int CvTest::get_next_timing_param_tuple()
705 {
706     bool increment;
707     int i;
708
709     if( timing_param_count <= 0 || !timing_param_names || !timing_param_seqs )
710         return -1;
711
712     increment = timing_param_current[0] != 0; // if already have some valid test tuple, move to the next
713     for( i = 0; i < timing_param_count; i++ )
714     {
715         const CvFileNode* node = timing_param_seqs[i];
716         int total = CV_NODE_IS_SEQ(node->tag) ? node->data.seq->total : 1;
717         int new_idx = timing_param_idxs[i];
718
719         if( !timing_param_current[i] )
720             timing_param_idxs[i] = new_idx = 0;
721         else if( increment )
722         {
723             new_idx++;
724             if( new_idx >= total )
725                 new_idx = 0;
726             else if( total > 1 )
727                 increment = false;
728         }
729
730         if( !timing_param_current[i] || new_idx != timing_param_idxs[i] )
731         {
732             if( CV_NODE_IS_SEQ(node->tag) )
733                 timing_param_current[i] = (CvFileNode*)cvGetSeqElem( node->data.seq, new_idx );
734             else
735                 timing_param_current[i] = node;
736             timing_param_idxs[i] = new_idx;
737         }
738     }
739
740     return !increment; // return 0 in case of overflow (i.e. if there is no more test cases)
741 }
742
743
744 const CvFileNode* CvTest::find_timing_param( const char* paramname )
745 {
746     if( timing_param_names )
747     {
748         int i;
749         CvSeqReader reader;
750         cvStartReadSeq( timing_param_names->data.seq, &reader, 0 );
751
752         for( i = 0; i < timing_param_count; i++ )
753         {
754             const char* ptr = ((const CvFileNode*)(reader.ptr))->data.str.ptr;
755             if( ptr[0] == paramname[0] && strcmp(ptr, paramname) == 0 )
756                 return timing_param_current[i];
757             CV_NEXT_SEQ_ELEM( reader.seq->elem_size, reader );
758         }
759     }
760     return 0;
761 }
762
763
764 int CvTest::write_defaults(CvTS* _ts)
765 {
766     ts = _ts;
767     hdr_state = 0;
768     write_default_params( ts->get_file_storage() );
769     if( hdr_state )
770         cvEndWriteStruct( ts->get_file_storage() );
771     return 0;
772 }
773
774
775 int CvTest::write_default_params( CvFileStorage* fs )
776 {
777     if( ts->get_testing_mode() == CvTS::TIMING_MODE )
778         write_string_list( fs, "timing_params", default_timing_param_names, timing_param_count );
779     return 0;
780 }
781
782
783 bool CvTest::can_do_fast_forward()
784 {
785     return true;
786 }
787
788
789 int CvTest::get_support_testing_modes()
790 {
791     return support_testing_modes;
792 }
793
794 void CvTest::safe_run( int start_from )
795 {
796     CV_TS_TRY_BLOCK_BEGIN;
797
798     run( start_from );
799
800     CV_TS_TRY_BLOCK_END;
801 }
802
803
804 void CvTest::run( int start_from )
805 {
806     int i, test_case_idx, count = get_test_case_count();
807     int64 t_start = cvGetTickCount();
808     double freq = cvGetTickFrequency();
809     bool ff = can_do_fast_forward();
810     int progress = 0, code;
811
812     for( test_case_idx = ff && start_from >= 0 ? start_from : 0;
813          count < 0 || test_case_idx < count; test_case_idx++ )
814     {
815         ts->update_context( this, test_case_idx, ff );
816         int64 t00 = 0, t0, t1 = 0;
817         double t_acc = 0;
818
819         if( ts->get_testing_mode() == CvTS::TIMING_MODE )
820         {
821             const int iterations = 15;
822             code = prepare_test_case( test_case_idx );
823
824             if( code < 0 || ts->get_err_code() < 0 )
825                 return;
826
827             if( code == 0 )
828                 continue;
829
830             for( i = 0; i < iterations; i++ )
831             {
832                 t0 = cvGetTickCount();
833                 run_func();
834                 t1 = cvGetTickCount();
835                 if( ts->get_err_code() < 0 )
836                     return;
837
838                 if( i == 0 )
839                 {
840                     t_acc = (double)(t1 - t0);
841                     t00 = t0;
842                 }
843                 else
844                 {
845                     t0 = t1 - t0;
846
847                     if( ts->get_timing_mode() == CvTS::MIN_TIME )
848                     {
849                         if( (double)t0 < t_acc )
850                             t_acc = (double)t0;
851                     }
852                     else
853                     {
854                         assert( ts->get_timing_mode() == CvTS::AVG_TIME );
855                         t_acc += (double)t0;
856                     }
857
858                     if( t1 - t00 > freq*2000000 )
859                         break;
860                 }
861             }
862
863             if( ts->get_timing_mode() == CvTS::AVG_TIME )
864                 t_acc /= i;
865             print_time( test_case_idx, t_acc );
866         }
867         else
868         {
869             code = prepare_test_case( test_case_idx );
870             if( code < 0 || ts->get_err_code() < 0 )
871                 return;
872
873             if( code == 0 )
874                 continue;
875
876             run_func();
877             if( ts->get_err_code() < 0 )
878                 return;
879
880             if( validate_test_results( test_case_idx ) < 0 || ts->get_err_code() < 0 )
881                 return;
882         }
883
884         progress = update_progress( progress, test_case_idx, count, (double)(t1 - t_start)/(freq*1000) );
885     }
886 }
887
888
889 void CvTest::run_func()
890 {
891     assert(0);
892 }
893
894
895 int CvTest::get_test_case_count()
896 {
897     return test_case_count;
898 }
899
900
901 int CvTest::prepare_test_case( int )
902 {
903     return 0;
904 }
905
906
907 int CvTest::validate_test_results( int )
908 {
909     return 0;
910 }
911
912
913 void CvTest::print_time( int /*test_case_idx*/, double /*time_usecs*/ )
914 {
915 }
916
917
918 int CvTest::update_progress( int progress, int test_case_idx, int count, double dt )
919 {
920     int width = 60 - (int)strlen(get_name());
921     if( count > 0 )
922     {
923         int t = cvRound( ((double)test_case_idx * width)/count );
924         if( t > progress )
925         {
926             ts->printf( CvTS::CONSOLE, "." );
927             progress = t;
928         }
929     }
930     else if( cvRound(dt*0.001) > progress )
931     {
932         ts->printf( CvTS::CONSOLE, "." );
933         progress = cvRound(dt*0.001);
934     }
935
936     return progress;
937 }
938
939 /*****************************************************************************************\
940 *                                 Base Class for Test System                              *
941 \*****************************************************************************************/
942
943 /******************************** Constructors/Destructors ******************************/
944
945 CvTS::CvTS()
946 {
947     start_time = 0;
948     version = CV_TS_VERSION;
949     memory_manager = 0;
950     /*
951     memory_manager = new CvTestMemoryManager(this);
952     cvSetMemoryManager( CvTestMemoryManager::alloc_proxy,
953                         CvTestMemoryManager::free_proxy,
954                         memory_manager );*/
955     ostrm_suffixes[SUMMARY_IDX] = ".sum";
956     ostrm_suffixes[LOG_IDX] = ".log";
957     ostrm_suffixes[CSV_IDX] = ".csv";
958     ostrm_suffixes[CONSOLE_IDX] = 0;
959     ostrm_base_name = 0;
960     memset( output_streams, 0, sizeof(output_streams) );
961     memset( &params, 0, sizeof(params) );
962     selected_tests = new CvTestPtrVec();
963     failed_tests = new CvTestInfoVec();
964     written_params = new CvTestPtrVec();
965
966     clear();
967 }
968
969
970 void CvTS::clear()
971 {
972     int i;
973     CvTest* test;
974
975     for( test = get_first_test(); test != 0; test = test->get_next() )
976         test->clear();
977
978     for( i = 0; i <= CONSOLE_IDX; i++ )
979     {
980         if( i == LOG_IDX )
981             fflush( stderr );
982         else if( i == CONSOLE_IDX )
983             fflush( stdout );
984
985         if( i < CONSOLE_IDX && output_streams[i].f )
986         {
987             fclose( output_streams[i].f );
988             output_streams[i].f = 0;
989         }
990
991         if( i == LOG_IDX && output_streams[i].default_handle > 0 )
992         {
993             dup2( output_streams[i].default_handle, 2 );
994             output_streams[i].default_handle = 0;
995         }
996         output_streams[i].enable = 1;
997     }
998     cvReleaseFileStorage( &fs );
999     selected_tests->clear();
1000     failed_tests->clear();
1001     if( ostrm_base_name )
1002     {
1003         free( ostrm_base_name );
1004         ostrm_base_name = 0;
1005     }
1006     params.rng_seed = (uint64)-1;
1007     params.debug_mode = 1;
1008     params.print_only_failed = 0;
1009     params.skip_header = 0;
1010     params.test_mode = CORRECTNESS_CHECK_MODE;
1011     params.timing_mode = MIN_TIME;
1012     params.use_optimized = -1;
1013
1014     if( memory_manager )
1015         memory_manager->clear_and_check();
1016 }
1017
1018
1019 CvTS::~CvTS()
1020 {
1021     clear();
1022     set_data_path(0);
1023
1024     if( written_params )
1025     {
1026         for( int i = 0; i < written_params->size(); i++ )
1027             free( written_params->at(i) );
1028         delete written_params;
1029     }
1030
1031     delete selected_tests;
1032     delete failed_tests;
1033     cvSetMemoryManager( 0, 0 );
1034 }
1035
1036
1037 const char* CvTS::str_from_code( int code )
1038 {
1039     switch( code )
1040     {
1041     case OK: return "Ok";
1042     case FAIL_GENERIC: return "Generic/Unknown";
1043     case FAIL_MISSING_TEST_DATA: return "No test data";
1044     case FAIL_INVALID_TEST_DATA: return "Invalid test data";
1045     case FAIL_ERROR_IN_CALLED_FUNC: return "cvError invoked";
1046     case FAIL_EXCEPTION: return "Hardware/OS exception";
1047     case FAIL_MEMORY_EXCEPTION: return "Invalid memory access";
1048     case FAIL_ARITHM_EXCEPTION: return "Arithmetic exception";
1049     case FAIL_MEMORY_CORRUPTION_BEGIN: return "Corrupted memblock (beginning)";
1050     case FAIL_MEMORY_CORRUPTION_END: return "Corrupted memblock (end)";
1051     case FAIL_MEMORY_LEAK: return "Memory leak";
1052     case FAIL_INVALID_OUTPUT: return "Invalid function output";
1053     case FAIL_MISMATCH: return "Unexpected output";
1054     case FAIL_BAD_ACCURACY: return "Bad accuracy";
1055     case FAIL_HANG: return "Infinite loop(?)";
1056     case FAIL_BAD_ARG_CHECK: return "Incorrect handling of bad arguments";
1057     default: return "Generic/Unknown";
1058     }
1059 }
1060
1061 /************************************** Running tests **********************************/
1062
1063 void CvTS::make_output_stream_base_name( const char* config_name )
1064 {
1065     int k, len = (int)strlen( config_name );
1066
1067     if( ostrm_base_name )
1068         free( ostrm_base_name );
1069
1070     for( k = len-1; k >= 0; k-- )
1071     {
1072         char c = config_name[k];
1073         if( c == '.' || c == '/' || c == '\\' || c == ':' )
1074             break;
1075     }
1076
1077     if( k > 0 && config_name[k] == '.' )
1078         len = k;
1079
1080     ostrm_base_name = (char*)malloc( len + 1 );
1081     memcpy( ostrm_base_name, config_name, len );
1082     ostrm_base_name[len] = '\0';
1083 }
1084
1085
1086 void CvTS::set_handlers( bool on )
1087 {
1088     if( on )
1089     {
1090         cvSetErrMode( CV_ErrModeParent );
1091         cvRedirectError( cvStdErrReport );
1092     #ifdef WIN32
1093         #ifdef _MSC_VER
1094         _set_se_translator( cv_seh_translator );
1095         #endif
1096     #else
1097         for( int i = 0; cv_ts_sig_id[i] >= 0; i++ )
1098             signal( cv_ts_sig_id[i], cv_signal_handler );
1099     #endif
1100     }
1101     else
1102     {
1103         cvSetErrMode( CV_ErrModeLeaf );
1104         cvRedirectError( cvGuiBoxReport );
1105     #ifdef WIN32
1106         #ifdef _MSC_VER
1107         _set_se_translator( 0 );
1108         #endif
1109     #else
1110         for( int i = 0; cv_ts_sig_id[i] >= 0; i++ )
1111             signal( cv_ts_sig_id[i], SIG_DFL );
1112     #endif
1113     }
1114 }
1115
1116
1117 void CvTS::set_data_path( const char* data_path )
1118 {
1119     if( data_path == params.data_path )
1120         return;
1121
1122     if( params.data_path )
1123         delete[] params.data_path;
1124     if( data_path )
1125     {
1126         int size = (int)strlen(data_path)+1;
1127         bool append_slash = data_path[size-1] != '/' && data_path[size-1] != '\\';
1128         params.data_path = new char[size+1];
1129         memcpy( params.data_path, data_path, size );
1130         if( append_slash )
1131             strcat( params.data_path, "/" );
1132     }
1133 }
1134
1135
1136 typedef struct CvTsParamVal
1137 {
1138     const char* fullname;
1139     const void* val;
1140 }
1141 CvTsParamVal;
1142
1143 int CvTS::find_written_param( CvTest* test, const char* paramname, int valtype, const void* val )
1144 {
1145     const char* testname = test->get_name();
1146     bool add_to_list = test->get_func_list()[0] == '\0';
1147     char buffer[256];
1148     int paramname_len = (int)strlen(paramname);
1149     int paramval_len = valtype == CV_NODE_INT ? (int)sizeof(int) :
1150         valtype == CV_NODE_REAL ? (int)sizeof(double) : -1;
1151     const char* name = CvTest::get_parent_name( testname, buffer );
1152
1153     if( !fs )
1154         return -1;
1155
1156     if( paramval_len < 0 )
1157     {
1158         assert(0); // unsupported parameter type
1159         return -1;
1160     }
1161
1162     while( name )
1163     {
1164         int i, len = (int)strlen(buffer);
1165         buffer[len] = '.';
1166         memcpy( buffer + len + 1, paramname, paramname_len + 1 );
1167         for( i = 0; i < written_params->size(); i++ )
1168         {
1169             CvTsParamVal* param = (CvTsParamVal*)written_params->at(i);
1170             if( strcmp( param->fullname, buffer ) == 0 )
1171             {
1172                 if( (paramval_len > 0 && memcmp( param->val, val, paramval_len ) == 0) ||
1173                     (paramval_len < 0 && strcmp( (const char*)param->val, (const char*)val ) == 0) )
1174                     return 1;
1175                 break;
1176             }
1177         }
1178         if( i < written_params->size() )
1179             break;
1180         buffer[len] = '\0';
1181         name = CvTest::get_parent_name( buffer, buffer );
1182     }
1183
1184     if( add_to_list )
1185     {
1186         int bufsize, fullname_len = (int)strlen(testname) + paramname_len + 2;
1187         CvTsParamVal* param;
1188         if( paramval_len < 0 )
1189             paramval_len = (int)strlen((const char*)val) + 1;
1190         bufsize = sizeof(*param) + fullname_len + paramval_len;
1191         param = (CvTsParamVal*)malloc(bufsize);
1192         param->fullname = (const char*)(param + 1);
1193         param->val = param->fullname + fullname_len;
1194         sprintf( (char*)param->fullname, "%s.%s", testname, paramname );
1195         memcpy( (void*)param->val, val, paramval_len );
1196         written_params->push( param );
1197     }
1198
1199     return 0;
1200 }
1201
1202
1203 #ifndef MAX_PATH
1204 #define MAX_PATH 1024
1205 #endif
1206
1207 static int CV_CDECL cmp_test_names( const void* a, const void* b )
1208 {
1209     return strcmp( (*(const CvTest**)a)->get_name(), (*(const CvTest**)b)->get_name() );
1210 }
1211
1212 int CvTS::run( int argc, char** argv )
1213 {
1214     time( &start_time );
1215
1216     int i, write_params = 0;
1217     int list_tests = 0;
1218     CvTestPtrVec all_tests;
1219     CvTest* test;
1220
1221     // 0. reset all the parameters, reorder tests
1222     clear();
1223
1224     for( test = get_first_test(), i = 0; test != 0; test = test->get_next(), i++ )
1225         all_tests.push(test);
1226
1227     if( all_tests.size() > 0 && all_tests.data() )
1228         qsort( all_tests.data(), all_tests.size(), sizeof(CvTest*), cmp_test_names );
1229
1230     // 1. parse command line options
1231     for( i = 1; i < argc; i++ )
1232     {
1233         if( strcmp( argv[i], "-h" ) == 0 || strcmp( argv[i], "--help" ) == 0 )
1234         {
1235             print_help();
1236             return 0;
1237         }
1238         else if( strcmp( argv[i], "-f" ) == 0 )
1239             config_name = argv[++i];
1240         else if( strcmp( argv[i], "-w" ) == 0 )
1241             write_params = 1;
1242         else if( strcmp( argv[i], "-t" ) == 0 )
1243             params.test_mode = TIMING_MODE;
1244         else if( strcmp( argv[i], "-O0" ) == 0 || strcmp( argv[i], "-O1" ) == 0 )
1245             params.use_optimized = argv[i][2] - '0';
1246         else if( strcmp( argv[i], "-l" ) == 0 )
1247             list_tests = 1;
1248         else if( strcmp( argv[i], "-d" ) == 0 )
1249             set_data_path(argv[++i]);
1250     }
1251
1252     if( write_params )
1253     {
1254         if( !config_name )
1255         {
1256             printf( LOG, "ERROR: output config name is not specified\n" );
1257             return -1;
1258         }
1259         fs = cvOpenFileStorage( config_name, 0, CV_STORAGE_WRITE );
1260         if( !fs )
1261         {
1262             printf( LOG, "ERROR: could not open config file %s\n", config_name );
1263             return -1;
1264         }
1265         cvWriteComment( fs, CV_TS_VERSION " config file", 0 );
1266         cvStartWriteStruct( fs, "common", CV_NODE_MAP );
1267         write_default_params( fs );
1268         cvEndWriteStruct( fs );
1269
1270         for( i = 0; i < all_tests.size(); i++ )
1271         {
1272             test = (CvTest*)all_tests[i];
1273             if( !(test->get_support_testing_modes() & get_testing_mode()) )
1274                 continue;
1275             test->write_defaults( this );
1276             test->clear();
1277         }
1278         cvReleaseFileStorage( &fs );
1279         return 0;
1280     }
1281
1282     if( !config_name )
1283         printf( LOG, "WARNING: config name is not specified, using default parameters\n" );
1284     else
1285     {
1286         // 2. read common parameters of test system
1287         fs = cvOpenFileStorage( config_name, 0, CV_STORAGE_READ );
1288         if( !fs )
1289         {
1290             printf( LOG, "ERROR: could not open config file %s", config_name );
1291             return -1;
1292         }
1293     }
1294
1295     if( read_params(fs) < 0 )
1296         return -1;
1297
1298     if( !ostrm_base_name )
1299         make_output_stream_base_name( config_name ? config_name : argv[0] );
1300
1301     ostream_testname_mask = -1; // disable printing test names at initial stage
1302
1303     // 3. open file streams
1304     for( i = 0; i < CONSOLE_IDX; i++ )
1305     {
1306         char filename[MAX_PATH];
1307         sprintf( filename, "%s%s", ostrm_base_name, ostrm_suffixes[i] );
1308         output_streams[i].f = fopen( filename, "wt" );
1309         if( !output_streams[i].f )
1310         {
1311             printf( LOG, "ERROR: could not open %s\n", filename );
1312             return -1;
1313         }
1314
1315         if( i == LOG_IDX )
1316         {
1317             // redirect stderr to log file
1318             fflush( stderr );
1319             output_streams[i].default_handle = dup(2);
1320             dup2( fileno(output_streams[i].f), 2 );
1321         }
1322     }
1323
1324     // 4. traverse through the list of all registered tests.
1325     // Initialize the selected tests and put them into the separate sequence
1326     for( i = 0; i < all_tests.size(); i++ )
1327     {
1328         test = (CvTest*)all_tests[i];
1329         if( !(test->get_support_testing_modes() & get_testing_mode()) )
1330             continue;
1331
1332         if( strcmp( test->get_func_list(), "" ) != 0 && filter(test) )
1333         {
1334             if( test->init(this) >= 0 )
1335             {
1336                 selected_tests->push( test );
1337                 if( list_tests )
1338                     ::printf( "%s\n", test->get_name() );
1339             }
1340             else
1341                 printf( LOG, "WARNING: an error occured during test %s initialization\n", test->get_name() );
1342         }
1343     }
1344
1345     if( list_tests )
1346         goto _exit_;
1347
1348     // 5. setup all the neccessary handlers and print header
1349     set_handlers( !params.debug_mode );
1350
1351     if( params.use_optimized == 0 )
1352         cvUseOptimized(0);
1353
1354     if( !params.skip_header )
1355         print_summary_header( SUMMARY + LOG + CONSOLE + CSV );
1356     rng = params.rng_seed;
1357     update_context( 0, -1, true );
1358
1359     // 6. run all the tests
1360     for( i = 0; i < selected_tests->size(); i++ )
1361     {
1362         CvTest* test = (CvTest*)selected_tests->at(i);
1363         int code;
1364         CvTestInfo temp;
1365
1366         if( memory_manager )
1367             memory_manager->start_tracking();
1368         update_context( test, -1, true );
1369         ostream_testname_mask = 0; // reset "test name was printed" flags
1370         if( output_streams[LOG_IDX].f )
1371             fflush( output_streams[LOG_IDX].f );
1372
1373         temp = current_test_info;
1374         test->safe_run(0);
1375         if( get_err_code() >= 0 )
1376         {
1377             update_context( test, -1, false );
1378             current_test_info.rng_seed = temp.rng_seed;
1379             current_test_info.base_alloc_index = temp.base_alloc_index;
1380         }
1381         test->clear();
1382         if( memory_manager )
1383             memory_manager->stop_tracking_and_check();
1384
1385         code = get_err_code();
1386         if( code >= 0 )
1387         {
1388             if( !params.print_only_failed )
1389             {
1390                 printf( SUMMARY + CONSOLE, "\t" );
1391                 change_color( CV_TS_GREEN );
1392                 printf( SUMMARY + CONSOLE, "Ok\n" );
1393                 change_color( CV_TS_NORMAL );
1394             }
1395         }
1396         else
1397         {
1398             printf( SUMMARY + CONSOLE, "\t" );
1399             change_color( CV_TS_RED );
1400             printf( SUMMARY + CONSOLE, "FAIL(%s)\n", str_from_code(code) );
1401             change_color( CV_TS_NORMAL );
1402             printf( LOG, "context: test case = %d, seed = %08x%08x\n",
1403                     current_test_info.test_case_idx,
1404                     (unsigned)(current_test_info.rng_seed>>32),
1405                     (unsigned)(current_test_info.rng_seed));
1406             failed_tests->push(current_test_info);
1407             if( params.rerun_immediately )
1408                 break;
1409         }
1410     }
1411
1412     ostream_testname_mask = -1;
1413     print_summary_tailer( SUMMARY + CONSOLE + LOG );
1414
1415     if( !params.debug_mode && (params.rerun_failed || params.rerun_immediately) )
1416     {
1417         set_handlers(0);
1418         update_context( 0, -1, true );
1419         for( i = 0; i < failed_tests->size(); i++ )
1420         {
1421             CvTestInfo info = failed_tests->at(i);
1422             if( (info.code == FAIL_MEMORY_CORRUPTION_BEGIN ||
1423                 info.code == FAIL_MEMORY_CORRUPTION_END ||
1424                 info.code == FAIL_MEMORY_LEAK) && memory_manager )
1425                 memory_manager->start_tracking( info.alloc_index - info.base_alloc_index
1426                                                 + memory_manager->get_alloc_index() );
1427             rng = info.rng_seed;
1428             test->safe_run( info.test_case_idx );
1429         }
1430     }
1431 _exit_:
1432     clear();
1433
1434     return 0;
1435 }
1436
1437
1438 void CvTS::print_help()
1439 {
1440     ::printf(
1441         "Usage: <test_executable> [{-h|--help}][-l] [-w] [-t] [-f <config_name>] [-d <data_path>] [-O{0|1}]\n\n"
1442         "-d - specify the test data path\n\n"
1443         "-f - use parameters from the provided config XML/YAML file instead of the default parameters\n\n"
1444         "-h or --help - print this help information\n\n"
1445         "-l - list all the registered tests or subset of the tests, selected in the config file, and exit\n\n"
1446         "-O{0|1} - disable/enable on-fly detection of IPP and other supported optimized libs. It's enabled by default\n\n"
1447         "-t - switch to the performance testing mode instead of the default algorithmic/correctness testing mode\n\n"
1448         "-w - write default parameters of the algorithmic or performance (when -t is passed) tests to the specifed config file (see -f) and exit\n\n"
1449         );
1450 }
1451
1452
1453 #ifdef WIN32
1454 const char* default_data_path = "../tests/cv/testdata/";
1455 #else
1456 const char* default_data_path = "../../../../tests/cv/testdata/";
1457 #endif
1458
1459
1460 int CvTS::read_params( CvFileStorage* fs )
1461 {
1462     CvFileNode* node = fs ? cvGetFileNodeByName( fs, 0, "common" ) : 0;
1463     params.debug_mode = cvReadIntByName( fs, node, "debug_mode", 1 ) != 0;
1464     params.skip_header = cvReadIntByName( fs, node, "skip_header", 0 ) != 0;
1465     params.print_only_failed = cvReadIntByName( fs, node, "print_only_failed", 0 ) != 0;
1466     params.rerun_failed = cvReadIntByName( fs, node, "rerun_failed", 0 ) != 0;
1467     params.rerun_immediately = cvReadIntByName( fs, node, "rerun_immediately", 0 ) != 0;
1468     const char* str = cvReadStringByName( fs, node, "filter_mode", "tests" );
1469     params.test_filter_mode = strcmp( str, "functions" ) == 0 ? CHOOSE_FUNCTIONS : CHOOSE_TESTS;
1470     str = cvReadStringByName( fs, node, "test_mode", params.test_mode == TIMING_MODE ? "timing" : "correctness" );
1471     params.test_mode = strcmp( str, "timing" ) == 0 || strcmp( str, "performance" ) == 0 ?
1472                         TIMING_MODE : CORRECTNESS_CHECK_MODE;
1473     str = cvReadStringByName( fs, node, "timing_mode", params.timing_mode == AVG_TIME ? "avg" : "min" );
1474     params.timing_mode = strcmp( str, "average" ) == 0 || strcmp( str, "avg" ) == 0 ? AVG_TIME : MIN_TIME;
1475     params.test_filter_pattern = cvReadStringByName( fs, node, params.test_filter_mode == CHOOSE_FUNCTIONS ?
1476                                                      "functions" : "tests", "" );
1477     params.resource_path = cvReadStringByName( fs, node, "." );
1478     if( params.use_optimized < 0 )
1479         params.use_optimized = cvReadIntByName( fs, node, "use_optimized", -1 );
1480 #ifndef WIN32
1481     if( !params.data_path || !params.data_path[0] )
1482     {
1483         char* srcdir = getenv("srcdir");
1484         char buf[1024];
1485         if( srcdir )
1486         {
1487             sprintf( buf, "%s/../testdata/", srcdir );
1488             set_data_path(buf);
1489         }
1490     }
1491 #endif
1492     if( !params.data_path || !params.data_path[0] )
1493     {
1494         const char* data_path =
1495             cvReadStringByName( fs, node, "data_path", default_data_path );
1496         set_data_path(data_path);
1497     }
1498     params.test_case_count_scale = cvReadRealByName( fs, node, "test_case_count_scale", 1. );
1499     if( params.test_case_count_scale <= 0 )
1500         params.test_case_count_scale = 1.;
1501     str = cvReadStringByName( fs, node, "seed", 0 );
1502     params.rng_seed = 0;
1503     if( str && strlen(str) == 16 )
1504     {
1505         params.rng_seed = 0;
1506         for( int i = 0; i < 16; i++ )
1507         {
1508             int c = tolower(str[i]);
1509             if( !isxdigit(c) )
1510             {
1511                 params.rng_seed = 0;
1512                 break;
1513             }
1514             params.rng_seed = params.rng_seed * 16 +
1515                 (str[i] < 'a' ? str[i] - '0' : str[i] - 'a' + 10);
1516         }
1517     }
1518
1519     if( params.rng_seed == 0 )
1520         params.rng_seed = cvGetTickCount();
1521
1522     str = cvReadStringByName( fs, node, "output_file_base_name", 0 );
1523     if( str )
1524         make_output_stream_base_name( str );
1525
1526     return 0;
1527 }
1528
1529
1530 void CvTS::write_default_params( CvFileStorage* fs )
1531 {
1532     read_params(0); // fill parameters with default values
1533
1534     cvWriteInt( fs, "debug_mode", params.debug_mode );
1535     cvWriteInt( fs, "skip_header", params.skip_header );
1536     cvWriteInt( fs, "print_only_failed", params.print_only_failed );
1537     cvWriteInt( fs, "rerun_failed", params.rerun_failed );
1538     cvWriteInt( fs, "rerun_immediately", params.rerun_immediately );
1539     cvWriteString( fs, "filter_mode", params.test_filter_mode == CHOOSE_FUNCTIONS ? "functions" : "tests" );
1540     cvWriteString( fs, "test_mode", params.test_mode == TIMING_MODE ? "timing" : "correctness" );
1541     cvWriteString( fs, "data_path", params.data_path ? params.data_path : default_data_path, 1 );
1542     if( params.test_mode == TIMING_MODE )
1543         cvWriteString( fs, "timing_mode", params.timing_mode == AVG_TIME ? "avg" : "min" );
1544     // test_filter, seed & output_file_base_name are not written
1545 }
1546
1547
1548 void CvTS::enable_output_streams( int stream_mask, int value )
1549 {
1550     for( int i = 0; i < MAX_IDX; i++ )
1551         if( stream_mask & (1 << i) )
1552             output_streams[i].enable = value != 0;
1553 }
1554
1555
1556 void CvTS::update_context( CvTest* test, int test_case_idx, bool update_ts_context )
1557 {
1558     current_test_info.test = test;
1559     current_test_info.test_case_idx = test_case_idx;
1560     current_test_info.alloc_index = 0;
1561     current_test_info.code = 0;
1562     cvSetErrStatus( CV_StsOk );
1563     if( update_ts_context )
1564     {
1565         current_test_info.rng_seed = rng;
1566         current_test_info.base_alloc_index = memory_manager ?
1567             memory_manager->get_alloc_index() : 0;
1568     }
1569 }
1570
1571
1572 void CvTS::set_failed_test_info( int fail_code, int alloc_index )
1573 {
1574     if( fail_code == FAIL_MEMORY_CORRUPTION_BEGIN ||
1575         fail_code == FAIL_MEMORY_CORRUPTION_END ||
1576         current_test_info.code >= 0 )
1577     {
1578         current_test_info.code = fail_code;
1579         current_test_info.alloc_index = alloc_index;
1580     }
1581 }
1582
1583
1584 const char* CvTS::get_libs_info( const char** addon_modules )
1585 {
1586     const char* all_info = 0;
1587     cvGetModuleInfo( 0, &all_info, addon_modules );
1588     return all_info;
1589 }
1590
1591
1592 void CvTS::print_summary_header( int streams )
1593 {
1594     char csv_header[256], *ptr = csv_header;
1595     int i;
1596
1597     printf( streams, "Engine: %s\n", version );
1598     time_t t1;
1599     time( &t1 );
1600     struct tm *t2 = localtime( &t1 );
1601     char buf[1024];
1602     strftime( buf, sizeof(buf)-1, "%c", t2 );
1603     printf( streams, "Execution Date & Time: %s\n", buf );
1604     printf( streams, "Config File: %s\n", config_name );
1605     const char* plugins = 0;
1606     const char* lib_verinfo = get_libs_info( &plugins );
1607     printf( streams, "Tested Libraries: %s\n", lib_verinfo );
1608     printf( streams, "Optimized Low-level Plugin\'s: %s\n", plugins );
1609     printf( streams, "=================================================\n");
1610
1611     sprintf( ptr, "funcName,dataType,channels,size," );
1612     ptr += strlen(ptr);
1613
1614     for( i = 0; i < CvTest::TIMING_EXTRA_PARAMS; i++ )
1615     {
1616         sprintf( ptr, "param%d,", i );
1617         ptr += strlen(ptr);
1618     }
1619
1620     sprintf( ptr, "CPE,Time(uSecs)" );
1621     printf( CSV, "%s\n", csv_header );
1622 }
1623
1624
1625 void CvTS::print_summary_tailer( int streams )
1626 {
1627     printf( streams, "=================================================\n");
1628     if( selected_tests && failed_tests )
1629     {
1630         time_t end_time;
1631         time( &end_time );
1632         double total_time = difftime( end_time, start_time );
1633         printf( streams, "Summary: %d out of %d tests failed\n",
1634             failed_tests->size(), selected_tests->size() );
1635         int minutes = cvFloor(total_time/60.);
1636         int seconds = cvRound(total_time - minutes*60);
1637         int hours = minutes / 60;
1638         minutes %= 60;
1639         printf( streams, "Running time: %02d:%02d:%02d\n", hours, minutes, seconds );
1640     }
1641 }
1642
1643
1644 void CvTS::vprintf( int streams, const char* fmt, va_list l )
1645 {
1646     if( streams )
1647     {
1648         char str[1 << 14];
1649         vsprintf( str, fmt, l );
1650
1651         for( int i = 0; i < MAX_IDX; i++ )
1652         {
1653             if( (streams & (1 << i)) && output_streams[i].enable )
1654             {
1655                 FILE* f = i == CONSOLE_IDX ? stdout :
1656                           i == LOG_IDX ? stderr : output_streams[i].f;
1657                 if( f )
1658                 {
1659                     if( i != CSV_IDX && !(ostream_testname_mask & (1 << i)) && current_test_info.test )
1660                     {
1661                         fprintf( f, "-------------------------------------------------\n" );
1662                         fprintf( f, "%s: ", current_test_info.test->get_name() );
1663                         fflush( f );
1664                         ostream_testname_mask |= 1 << i;
1665                     }
1666                     fputs( str, f );
1667                     if( i == CONSOLE_IDX )
1668                         fflush(f);
1669                 }
1670             }
1671         }
1672     }
1673 }
1674
1675
1676 void CvTS::printf( int streams, const char* fmt, ... )
1677 {
1678     if( streams )
1679     {
1680         va_list l;
1681         va_start( l, fmt );
1682         vprintf( streams, fmt, l );
1683         va_end( l );
1684     }
1685 }
1686
1687
1688 static char* cv_strnstr( const char* str, int len,
1689                          const char* pattern,
1690                          int pattern_len = -1,
1691                          int whole_word = 1 )
1692 {
1693     int i;
1694
1695     if( len < 0 && pattern_len < 0 )
1696         return (char*)strstr( str, pattern );
1697
1698     if( len < 0 )
1699         len = (int)strlen( str );
1700
1701     if( pattern_len < 0 )
1702         pattern_len = (int)strlen( pattern );
1703
1704     for( i = 0; i < len - pattern_len + 1; i++ )
1705     {
1706         int j = i + pattern_len;
1707         if( str[i] == pattern[0] &&
1708             memcmp( str + i, pattern, pattern_len ) == 0 &&
1709             (!whole_word ||
1710             ((i == 0 || (!isalnum(str[i-1]) && str[i-1] != '_')) &&
1711              (j == len || (!isalnum(str[j]) && str[j] != '_')))))
1712             return (char*)(str + i);
1713     }
1714
1715     return 0;
1716 }
1717
1718
1719 int CvTS::filter( CvTest* test )
1720 {
1721     const char* pattern = params.test_filter_pattern;
1722     int inverse = 0;
1723
1724     if( pattern && pattern[0] == '!' )
1725     {
1726         inverse = 1;
1727         pattern++;
1728     }
1729
1730     if( !pattern || strcmp( pattern, "" ) == 0 || strcmp( pattern, "*" ) == 0 )
1731         return 1 ^ inverse;
1732
1733     if( params.test_filter_mode == CHOOSE_TESTS )
1734     {
1735         int found = 0;
1736
1737         while( pattern && *pattern )
1738         {
1739             char *ptr, *endptr = (char*)strchr( pattern, ',' );
1740             int len, have_wildcard;
1741             int t_name_len;
1742
1743             if( endptr )
1744                 *endptr = '\0';
1745
1746             ptr = (char*)strchr( pattern, '*' );
1747             if( ptr )
1748             {
1749                 len = (int)(ptr - pattern);
1750                 have_wildcard = 1;
1751             }
1752             else
1753             {
1754                 len = (int)strlen( pattern );
1755                 have_wildcard = 0;
1756             }
1757
1758             t_name_len = (int)strlen( test->get_name() );
1759             found = (t_name_len == len || (have_wildcard && t_name_len > len)) &&
1760                     (len == 0 || memcmp( test->get_name(), pattern, len ) == 0);
1761             if( endptr )
1762             {
1763                 *endptr = ',';
1764                 pattern = endptr + 1;
1765                 while( isspace(*pattern) )
1766                     pattern++;
1767             }
1768
1769             if( found || !endptr )
1770                 break;
1771         }
1772
1773         return found ^ inverse;
1774     }
1775     else
1776     {
1777         assert( params.test_filter_mode == CHOOSE_FUNCTIONS );
1778         int glob_len = (int)strlen( pattern );
1779         const char* ptr = test->get_func_list();
1780         const char *tmp_ptr;
1781
1782         while( ptr && *ptr )
1783         {
1784             const char* endptr = ptr - 1;
1785             const char* name_ptr;
1786             const char* name_first_match;
1787             int name_len;
1788             char c;
1789
1790             do c = *++endptr;
1791             while( isspace(c) );
1792
1793             if( !c )
1794                 break;
1795
1796             assert( isalpha(c) );
1797             name_ptr = endptr;
1798
1799             do c = *++endptr;
1800             while( isalnum(c) || c == '_' );
1801
1802             if( c == ':' ) // class
1803             {
1804                 assert( endptr[1] == ':' );
1805                 endptr = endptr + 2;
1806                 name_len = (int)(endptr - name_ptr);
1807
1808                 // find the first occurence of the class name
1809                 // in pattern
1810                 name_first_match = cv_strnstr( pattern,
1811                                       glob_len, name_ptr, name_len, 1 );
1812
1813                 if( *endptr == '*' )
1814                 {
1815                     if( name_first_match )
1816                         return 1 ^ inverse;
1817                 }
1818                 else
1819                 {
1820                     assert( *endptr == '{' ); // a list of methods
1821
1822                     if( !name_first_match )
1823                     {
1824                         // skip all the methods, if there is no such a class name
1825                         // in pattern
1826                         endptr = strchr( endptr, '}' );
1827                         assert( endptr != 0 );
1828                         endptr--;
1829                     }
1830
1831                     for( ;; )
1832                     {
1833                         const char* method_name_ptr;
1834                         int method_name_len;
1835
1836                         do c = *++endptr;
1837                         while( isspace(c) );
1838
1839                         if( c == '}' )
1840                             break;
1841                         assert( isalpha(c) );
1842
1843                         method_name_ptr = endptr;
1844
1845                         do c = *++endptr;
1846                         while( isalnum(c) || c == '_' );
1847
1848                         method_name_len = (int)(endptr - method_name_ptr);
1849
1850                         // search for class_name::* or
1851                         // class_name::{...method_name...}
1852                         tmp_ptr = name_first_match;
1853                         do
1854                         {
1855                             const char* tmp_ptr2;
1856                             tmp_ptr += name_len;
1857                             if( *tmp_ptr == '*' )
1858                                 return 1;
1859                             assert( *tmp_ptr == '{' );
1860                             tmp_ptr2 = strchr( tmp_ptr, '}' );
1861                             assert( tmp_ptr2 );
1862
1863                             if( cv_strnstr( tmp_ptr, (int)(tmp_ptr2 - tmp_ptr) + 1,
1864                                              method_name_ptr, method_name_len, 1 ))
1865                                 return 1 ^ inverse;
1866
1867                             tmp_ptr = cv_strnstr( tmp_ptr2, glob_len -
1868                                                    (int)(tmp_ptr2 - pattern),
1869                                                    name_ptr, name_len, 1 );
1870                         }
1871                         while( tmp_ptr );
1872
1873                         endptr--;
1874                         do c = *++endptr;
1875                         while( isspace(c) );
1876
1877                         if( c != ',' )
1878                             endptr--;
1879                     }
1880                 }
1881             }
1882             else
1883             {
1884                 assert( !c || isspace(c) || c == ',' );
1885                 name_len = (int)(endptr - name_ptr);
1886                 tmp_ptr = pattern;
1887
1888                 for(;;)
1889                 {
1890                     const char *tmp_ptr2, *tmp_ptr3;
1891
1892                     tmp_ptr = cv_strnstr( tmp_ptr, glob_len -
1893                         (int)(tmp_ptr - pattern), name_ptr, name_len, 1 );
1894
1895                     if( !tmp_ptr )
1896                         break;
1897
1898                     // make sure it is not a method
1899                     tmp_ptr2 = strchr( tmp_ptr, '}' );
1900                     if( !tmp_ptr2 )
1901                         return 1 ^ inverse;
1902
1903                     tmp_ptr3 = strchr( tmp_ptr, '{' );
1904                     if( tmp_ptr3 < tmp_ptr2 )
1905                         return 1 ^ inverse;
1906
1907                     tmp_ptr = tmp_ptr2 + 1;
1908                 }
1909
1910                 endptr--;
1911             }
1912
1913             do c = *++endptr;
1914             while( isspace(c) );
1915
1916             if( c == ',' )
1917                 endptr++;
1918             ptr = endptr;
1919         }
1920
1921         return 0 ^ inverse;
1922     }
1923 }
1924
1925 /* End of file. */