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