matrix.cpp 121.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.
//  If you do not agree to this license, do not download, install,
//  copy or use the software.
//
//
//                           License Agreement
//                For Open Source Computer Vision Library
//
// Copyright (C) 2000-2008, Intel Corporation, all rights reserved.
// Copyright (C) 2009, Willow Garage Inc., all rights reserved.
// Third party copyrights are property of their respective owners.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//   * Redistribution's of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//
//   * Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
//   * The name of the copyright holders may not be used to endorse or promote products
//     derived from this software without specific prior written permission.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the Intel Corporation or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
//M*/

#include "precomp.hpp"

/****************************************************************************************\
*                           [scaled] Identity matrix initialization                      *
\****************************************************************************************/

namespace cv {

V
Vadim Pisarevsky 已提交
51 52
void swap( Mat& a, Mat& b )
{
53 54 55 56 57 58 59 60 61 62
    std::swap(a.flags, b.flags);
    std::swap(a.dims, b.dims);
    std::swap(a.rows, b.rows);
    std::swap(a.cols, b.cols);
    std::swap(a.data, b.data);
    std::swap(a.refcount, b.refcount);
    std::swap(a.datastart, b.datastart);
    std::swap(a.dataend, b.dataend);
    std::swap(a.datalimit, b.datalimit);
    std::swap(a.allocator, b.allocator);
63

64 65 66 67
    std::swap(a.size.p, b.size.p);
    std::swap(a.step.p, b.step.p);
    std::swap(a.step.buf[0], b.step.buf[0]);
    std::swap(a.step.buf[1], b.step.buf[1]);
68

V
Vadim Pisarevsky 已提交
69 70 71 72 73
    if( a.step.p == b.step.buf )
    {
        a.step.p = a.step.buf;
        a.size.p = &a.rows;
    }
74

V
Vadim Pisarevsky 已提交
75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99
    if( b.step.p == a.step.buf )
    {
        b.step.p = b.step.buf;
        b.size.p = &b.rows;
    }
}


static inline void setSize( Mat& m, int _dims, const int* _sz,
                            const size_t* _steps, bool autoSteps=false )
{
    CV_Assert( 0 <= _dims && _dims <= CV_MAX_DIM );
    if( m.dims != _dims )
    {
        if( m.step.p != m.step.buf )
        {
            fastFree(m.step.p);
            m.step.p = m.step.buf;
            m.size.p = &m.rows;
        }
        if( _dims > 2 )
        {
            m.step.p = (size_t*)fastMalloc(_dims*sizeof(m.step.p[0]) + (_dims+1)*sizeof(m.size.p[0]));
            m.size.p = (int*)(m.step.p + _dims) + 1;
            m.size.p[-1] = _dims;
100
            m.rows = m.cols = -1;
V
Vadim Pisarevsky 已提交
101 102
        }
    }
103

V
Vadim Pisarevsky 已提交
104 105 106
    m.dims = _dims;
    if( !_sz )
        return;
107

108
    size_t esz = CV_ELEM_SIZE(m.flags), total = esz;
V
Vadim Pisarevsky 已提交
109 110 111 112
    int i;
    for( i = _dims-1; i >= 0; i-- )
    {
        int s = _sz[i];
113
        CV_Assert( s >= 0 );
V
Vadim Pisarevsky 已提交
114
        m.size.p[i] = s;
115

V
Vadim Pisarevsky 已提交
116 117 118 119 120 121 122 123 124 125 126
        if( _steps )
            m.step.p[i] = i < _dims-1 ? _steps[i] : esz;
        else if( autoSteps )
        {
            m.step.p[i] = total;
            int64 total1 = (int64)total*s;
            if( (uint64)total1 != (size_t)total1 )
                CV_Error( CV_StsOutOfRange, "The total matrix size does not fit to \"size_t\" type" );
            total = (size_t)total1;
        }
    }
127

V
Vadim Pisarevsky 已提交
128 129 130 131 132 133 134
    if( _dims == 1 )
    {
        m.dims = 2;
        m.cols = 1;
        m.step[1] = esz;
    }
}
135

136
static void updateContinuityFlag(Mat& m)
V
Vadim Pisarevsky 已提交
137 138 139 140 141 142 143
{
    int i, j;
    for( i = 0; i < m.dims; i++ )
    {
        if( m.size[i] > 1 )
            break;
    }
144

V
Vadim Pisarevsky 已提交
145 146 147 148 149
    for( j = m.dims-1; j > i; j-- )
    {
        if( m.step[j]*m.size[j] < m.step[j-1] )
            break;
    }
150

151
    uint64 t = (uint64)m.step[0]*m.size[0];
152
    if( j <= i && t == (size_t)t )
V
Vadim Pisarevsky 已提交
153
        m.flags |= Mat::CONTINUOUS_FLAG;
154 155 156
    else
        m.flags &= ~Mat::CONTINUOUS_FLAG;
}
157

158 159 160
static void finalizeHdr(Mat& m)
{
    updateContinuityFlag(m);
161 162
    int d = m.dims;
    if( d > 2 )
V
Vadim Pisarevsky 已提交
163 164 165
        m.rows = m.cols = -1;
    if( m.data )
    {
166 167 168
        m.datalimit = m.datastart + m.size[0]*m.step[0];
        if( m.size[0] > 0 )
        {
169 170
            m.dataend = m.data + m.size[d-1]*m.step[d-1];
            for( int i = 0; i < d-1; i++ )
171 172 173 174
                m.dataend += (m.size[i] - 1)*m.step[i];
        }
        else
            m.dataend = m.datalimit;
V
Vadim Pisarevsky 已提交
175
    }
176 177
    else
        m.dataend = m.datalimit = 0;
V
Vadim Pisarevsky 已提交
178
}
179 180


V
Vadim Pisarevsky 已提交
181 182 183
void Mat::create(int d, const int* _sizes, int _type)
{
    int i;
V
Vladislav Vinogradov 已提交
184
    CV_Assert(0 <= d && d <= CV_MAX_DIM && _sizes);
V
Vadim Pisarevsky 已提交
185
    _type = CV_MAT_TYPE(_type);
186

V
Vadim Pisarevsky 已提交
187 188 189 190 191 192 193 194 195 196
    if( data && (d == dims || (d == 1 && dims <= 2)) && _type == type() )
    {
        if( d == 2 && rows == _sizes[0] && cols == _sizes[1] )
            return;
        for( i = 0; i < d; i++ )
            if( size[i] != _sizes[i] )
                break;
        if( i == d && (d > 1 || size[1] == 1))
            return;
    }
197

V
Vadim Pisarevsky 已提交
198 199 200 201
    release();
    if( d == 0 )
        return;
    flags = (_type & CV_MAT_TYPE_MASK) | MAGIC_VAL;
202
    setSize(*this, d, _sizes, 0, true);
203

204
    if( total() > 0 )
V
Vadim Pisarevsky 已提交
205
    {
A
Andrey Pavlenko 已提交
206
#ifdef HAVE_TGPU
207
        if( !allocator || allocator == tegra::getAllocator() ) allocator = tegra::getAllocator(d, _sizes, _type);
A
Andrey Pavlenko 已提交
208
#endif
209 210
        if( !allocator )
        {
A
Andrey Kamaev 已提交
211 212 213
            size_t totalsize = alignSize(step.p[0]*size.p[0], (int)sizeof(*refcount));
            data = datastart = (uchar*)fastMalloc(totalsize + (int)sizeof(*refcount));
            refcount = (int*)(data + totalsize);
214 215 216 217
            *refcount = 1;
        }
        else
        {
218
#ifdef HAVE_TGPU
219
           try
220 221 222 223 224 225
            {
                allocator->allocate(dims, size, _type, refcount, datastart, data, step.p);
                CV_Assert( step[dims-1] == (size_t)CV_ELEM_SIZE(flags) );
            }catch(...)
            {
                allocator = 0;
226 227 228
                size_t totalSize = alignSize(step.p[0]*size.p[0], (int)sizeof(*refcount));
                data = datastart = (uchar*)fastMalloc(totalSize + (int)sizeof(*refcount));
                refcount = (int*)(data + totalSize);
229 230 231
                *refcount = 1;
            }
#else
232 233
            allocator->allocate(dims, size, _type, refcount, datastart, data, step.p);
            CV_Assert( step[dims-1] == (size_t)CV_ELEM_SIZE(flags) );
234
#endif
235
        }
V
Vadim Pisarevsky 已提交
236
    }
237

V
Vadim Pisarevsky 已提交
238 239 240 241 242 243 244 245 246 247 248 249
    finalizeHdr(*this);
}

void Mat::copySize(const Mat& m)
{
    setSize(*this, m.dims, 0, 0);
    for( int i = 0; i < dims; i++ )
    {
        size[i] = m.size[i];
        step[i] = m.step[i];
    }
}
250

V
Vadim Pisarevsky 已提交
251 252 253 254 255 256 257 258 259 260 261
void Mat::deallocate()
{
    if( allocator )
        allocator->deallocate(refcount, datastart, data);
    else
    {
        CV_DbgAssert(refcount != 0);
        fastFree(datastart);
    }
}

262

A
Andrey Kamaev 已提交
263 264 265
Mat::Mat(const Mat& m, const Range& _rowRange, const Range& _colRange)
    : flags(MAGIC_VAL), dims(0), rows(0), cols(0), data(0), refcount(0), datastart(0), dataend(0),
      datalimit(0), allocator(0), size(&rows)
V
Vadim Pisarevsky 已提交
266 267 268 269 270
{
    CV_Assert( m.dims >= 2 );
    if( m.dims > 2 )
    {
        AutoBuffer<Range> rs(m.dims);
A
Andrey Kamaev 已提交
271 272
        rs[0] = _rowRange;
        rs[1] = _colRange;
V
Vadim Pisarevsky 已提交
273 274 275 276 277
        for( int i = 2; i < m.dims; i++ )
            rs[i] = Range::all();
        *this = m(rs);
        return;
    }
278

V
Vadim Pisarevsky 已提交
279
    *this = m;
A
Andrey Kamaev 已提交
280
    if( _rowRange != Range::all() && _rowRange != Range(0,rows) )
V
Vadim Pisarevsky 已提交
281
    {
A
Andrey Kamaev 已提交
282 283 284
        CV_Assert( 0 <= _rowRange.start && _rowRange.start <= _rowRange.end && _rowRange.end <= m.rows );
        rows = _rowRange.size();
        data += step*_rowRange.start;
285
        flags |= SUBMATRIX_FLAG;
V
Vadim Pisarevsky 已提交
286
    }
287

A
Andrey Kamaev 已提交
288
    if( _colRange != Range::all() && _colRange != Range(0,cols) )
V
Vadim Pisarevsky 已提交
289
    {
A
Andrey Kamaev 已提交
290 291 292
        CV_Assert( 0 <= _colRange.start && _colRange.start <= _colRange.end && _colRange.end <= m.cols );
        cols = _colRange.size();
        data += _colRange.start*elemSize();
V
Vadim Pisarevsky 已提交
293
        flags &= cols < m.cols ? ~CONTINUOUS_FLAG : -1;
294
        flags |= SUBMATRIX_FLAG;
V
Vadim Pisarevsky 已提交
295
    }
296

V
Vadim Pisarevsky 已提交
297 298
    if( rows == 1 )
        flags |= CONTINUOUS_FLAG;
299

V
Vadim Pisarevsky 已提交
300 301 302 303 304 305
    if( rows <= 0 || cols <= 0 )
    {
        release();
        rows = cols = 0;
    }
}
306 307


V
Vadim Pisarevsky 已提交
308 309 310
Mat::Mat(const Mat& m, const Rect& roi)
    : flags(m.flags), dims(2), rows(roi.height), cols(roi.width),
    data(m.data + roi.y*m.step[0]), refcount(m.refcount),
311 312
    datastart(m.datastart), dataend(m.dataend), datalimit(m.datalimit),
    allocator(m.allocator), size(&rows)
V
Vadim Pisarevsky 已提交
313 314 315 316
{
    CV_Assert( m.dims <= 2 );
    flags &= roi.width < m.cols ? ~CONTINUOUS_FLAG : -1;
    flags |= roi.height == 1 ? CONTINUOUS_FLAG : 0;
317

318
    size_t esz = CV_ELEM_SIZE(flags);
V
Vadim Pisarevsky 已提交
319 320 321 322 323
    data += roi.x*esz;
    CV_Assert( 0 <= roi.x && 0 <= roi.width && roi.x + roi.width <= m.cols &&
              0 <= roi.y && 0 <= roi.height && roi.y + roi.height <= m.rows );
    if( refcount )
        CV_XADD(refcount, 1);
324 325
    if( roi.width < m.cols || roi.height < m.rows )
        flags |= SUBMATRIX_FLAG;
326

V
Vadim Pisarevsky 已提交
327
    step[0] = m.step[0]; step[1] = esz;
328

V
Vadim Pisarevsky 已提交
329 330 331 332 333 334 335
    if( rows <= 0 || cols <= 0 )
    {
        release();
        rows = cols = 0;
    }
}

336

A
Andrey Kamaev 已提交
337 338 339
Mat::Mat(int _dims, const int* _sizes, int _type, void* _data, const size_t* _steps)
    : flags(MAGIC_VAL), dims(0), rows(0), cols(0), data(0), refcount(0), datastart(0), dataend(0),
      datalimit(0), allocator(0), size(&rows)
V
Vadim Pisarevsky 已提交
340
{
341 342
    flags |= CV_MAT_TYPE(_type);
    data = datastart = (uchar*)_data;
V
Vadim Pisarevsky 已提交
343 344 345
    setSize(*this, _dims, _sizes, _steps, true);
    finalizeHdr(*this);
}
346 347


A
Andrey Kamaev 已提交
348 349 350
Mat::Mat(const Mat& m, const Range* ranges)
    : flags(MAGIC_VAL), dims(0), rows(0), cols(0), data(0), refcount(0), datastart(0), dataend(0),
      datalimit(0), allocator(0), size(&rows)
V
Vadim Pisarevsky 已提交
351 352
{
    int i, d = m.dims;
353

V
Vadim Pisarevsky 已提交
354 355 356 357 358 359 360 361 362 363
    CV_Assert(ranges);
    for( i = 0; i < d; i++ )
    {
        Range r = ranges[i];
        CV_Assert( r == Range::all() || (0 <= r.start && r.start < r.end && r.end <= m.size[i]) );
    }
    *this = m;
    for( i = 0; i < d; i++ )
    {
        Range r = ranges[i];
364
        if( r != Range::all() && r != Range(0, size.p[i]))
V
Vadim Pisarevsky 已提交
365
        {
366 367 368
            size.p[i] = r.end - r.start;
            data += r.start*step.p[i];
            flags |= SUBMATRIX_FLAG;
V
Vadim Pisarevsky 已提交
369 370
        }
    }
371
    updateContinuityFlag(*this);
V
Vadim Pisarevsky 已提交
372
}
373 374


A
Andrey Kamaev 已提交
375
static Mat cvMatNDToMat(const CvMatND* m, bool copyData)
V
Vadim Pisarevsky 已提交
376
{
A
Andrey Kamaev 已提交
377 378
    Mat thiz;

379
    if( !m )
A
Andrey Kamaev 已提交
380 381 382
        return thiz;
    thiz.data = thiz.datastart = m->data.ptr;
    thiz.flags |= CV_MAT_TYPE(m->type);
V
Vadim Pisarevsky 已提交
383 384
    int _sizes[CV_MAX_DIM];
    size_t _steps[CV_MAX_DIM];
385

V
Vadim Pisarevsky 已提交
386 387 388 389 390 391
    int i, d = m->dims;
    for( i = 0; i < d; i++ )
    {
        _sizes[i] = m->dim[i].size;
        _steps[i] = m->dim[i].step;
    }
392

A
Andrey Kamaev 已提交
393 394
    setSize(thiz, d, _sizes, _steps);
    finalizeHdr(thiz);
V
Vadim Pisarevsky 已提交
395 396 397

    if( copyData )
    {
A
Andrey Kamaev 已提交
398 399 400
        Mat temp(thiz);
        thiz.release();
        temp.copyTo(thiz);
V
Vadim Pisarevsky 已提交
401
    }
402

A
Andrey Kamaev 已提交
403
    return thiz;
V
Vadim Pisarevsky 已提交
404
}
405

A
Andrey Kamaev 已提交
406
static Mat cvMatToMat(const CvMat* m, bool copyData)
407
{
A
Andrey Kamaev 已提交
408
    Mat thiz;
409

410
    if( !m )
A
Andrey Kamaev 已提交
411
        return thiz;
412

413 414
    if( !copyData )
    {
A
Andrey Kamaev 已提交
415 416 417 418 419 420
        thiz.flags = Mat::MAGIC_VAL + (m->type & (CV_MAT_TYPE_MASK|CV_MAT_CONT_FLAG));
        thiz.dims = 2;
        thiz.rows = m->rows;
        thiz.cols = m->cols;
        thiz.data = thiz.datastart = m->data.ptr;
        size_t esz = CV_ELEM_SIZE(m->type), minstep = thiz.cols*esz, _step = m->step;
421 422
        if( _step == 0 )
            _step = minstep;
A
Andrey Kamaev 已提交
423 424 425
        thiz.datalimit = thiz.datastart + _step*thiz.rows;
        thiz.dataend = thiz.datalimit - _step + minstep;
        thiz.step[0] = _step; thiz.step[1] = esz;
426 427 428
    }
    else
    {
A
Andrey Kamaev 已提交
429 430
        thiz.data = thiz.datastart = thiz.dataend = 0;
        Mat(m->rows, m->cols, m->type, m->data.ptr, m->step).copyTo(thiz);
431
    }
A
Andrey Kamaev 已提交
432 433

    return thiz;
434 435
}

436

A
Andrey Kamaev 已提交
437
static Mat iplImageToMat(const IplImage* img, bool copyData)
438
{
A
Andrey Kamaev 已提交
439
    Mat m;
440

441
    if( !img )
A
Andrey Kamaev 已提交
442
        return m;
443

A
Andrey Kamaev 已提交
444
    m.dims = 2;
445
    CV_DbgAssert(CV_IS_IMAGE(img) && img->imageData != 0);
446

A
Andrey Kamaev 已提交
447
    int imgdepth = IPL2CV_DEPTH(img->depth);
448
    size_t esz;
A
Andrey Kamaev 已提交
449
    m.step[0] = img->widthStep;
450 451 452 453

    if(!img->roi)
    {
        CV_Assert(img->dataOrder == IPL_DATA_ORDER_PIXEL);
A
Andrey Kamaev 已提交
454 455 456 457 458
        m.flags = Mat::MAGIC_VAL + CV_MAKETYPE(imgdepth, img->nChannels);
        m.rows = img->height;
        m.cols = img->width;
        m.datastart = m.data = (uchar*)img->imageData;
        esz = CV_ELEM_SIZE(m.flags);
459 460 461 462 463
    }
    else
    {
        CV_Assert(img->dataOrder == IPL_DATA_ORDER_PIXEL || img->roi->coi != 0);
        bool selectedPlane = img->roi->coi && img->dataOrder == IPL_DATA_ORDER_PLANE;
A
Andrey Kamaev 已提交
464 465 466 467 468 469 470 471 472 473 474 475
        m.flags = Mat::MAGIC_VAL + CV_MAKETYPE(imgdepth, selectedPlane ? 1 : img->nChannels);
        m.rows = img->roi->height;
        m.cols = img->roi->width;
        esz = CV_ELEM_SIZE(m.flags);
        m.data = m.datastart = (uchar*)img->imageData +
            (selectedPlane ? (img->roi->coi - 1)*m.step*img->height : 0) +
            img->roi->yOffset*m.step[0] + img->roi->xOffset*esz;
    }
    m.datalimit = m.datastart + m.step.p[0]*m.rows;
    m.dataend = m.datastart + m.step.p[0]*(m.rows-1) + esz*m.cols;
    m.flags |= (m.cols*esz == m.step.p[0] || m.rows == 1 ? Mat::CONTINUOUS_FLAG : 0);
    m.step[1] = esz;
476 477 478

    if( copyData )
    {
A
Andrey Kamaev 已提交
479 480
        Mat m2 = m;
        m.release();
481 482
        if( !img->roi || !img->roi->coi ||
            img->dataOrder == IPL_DATA_ORDER_PLANE)
A
Andrey Kamaev 已提交
483
            m2.copyTo(m);
484 485 486
        else
        {
            int ch[] = {img->roi->coi - 1, 0};
A
Andrey Kamaev 已提交
487 488
            m.create(m2.rows, m2.cols, m2.type());
            mixChannels(&m2, 1, &m, 1, ch, 1);
489 490 491
        }
    }

A
Andrey Kamaev 已提交
492 493
    return m;
}
494

A
Andrey Kamaev 已提交
495
Mat Mat::diag(int d) const
496
{
V
Vadim Pisarevsky 已提交
497
    CV_Assert( dims <= 2 );
A
Andrey Kamaev 已提交
498 499 500
    Mat m = *this;
    size_t esz = elemSize();
    int len;
501

A
Andrey Kamaev 已提交
502 503 504 505 506 507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523 524 525 526 527
    if( d >= 0 )
    {
        len = std::min(cols - d, rows);
        m.data += esz*d;
    }
    else
    {
        len = std::min(rows + d, cols);
        m.data -= step[0]*d;
    }
    CV_DbgAssert( len > 0 );

    m.size[0] = m.rows = len;
    m.size[1] = m.cols = 1;
    m.step[0] += (len > 1 ? esz : 0);

    if( m.rows > 1 )
        m.flags &= ~CONTINUOUS_FLAG;
    else
        m.flags |= CONTINUOUS_FLAG;

    if( size() != Size(1,1) )
        m.flags |= SUBMATRIX_FLAG;

    return m;
}
528

529 530 531
void Mat::pop_back(size_t nelems)
{
    CV_Assert( nelems <= (size_t)size.p[0] );
532

533 534 535 536 537 538 539 540 541 542 543 544 545 546 547
    if( isSubmatrix() )
        *this = rowRange(0, size.p[0] - (int)nelems);
    else
    {
        size.p[0] -= (int)nelems;
        dataend -= nelems*step.p[0];
        /*if( size.p[0] <= 1 )
        {
            if( dims <= 2 )
                flags |= CONTINUOUS_FLAG;
            else
                updateContinuityFlag(*this);
        }*/
    }
}
548 549


550 551 552 553 554
void Mat::push_back_(const void* elem)
{
    int r = size.p[0];
    if( isSubmatrix() || dataend + step.p[0] > datalimit )
        reserve( std::max(r + 1, (r*3+1)/2) );
555

556 557 558 559 560 561 562 563 564 565 566
    size_t esz = elemSize();
    memcpy(data + r*step.p[0], elem, esz);
    size.p[0] = r + 1;
    dataend += step.p[0];
    if( esz < step.p[0] )
        flags &= ~CONTINUOUS_FLAG;
}

void Mat::reserve(size_t nelems)
{
    const size_t MIN_SIZE = 64;
567

568 569 570
    CV_Assert( (int)nelems >= 0 );
    if( !isSubmatrix() && data + step.p[0]*nelems <= datalimit )
        return;
571

572
    int r = size.p[0];
573

574 575
    if( (size_t)r >= nelems )
        return;
576

577 578
    size.p[0] = std::max((int)nelems, 1);
    size_t newsize = total()*elemSize();
579

580 581
    if( newsize < MIN_SIZE )
        size.p[0] = (int)((MIN_SIZE + newsize - 1)*nelems/newsize);
582

583 584 585 586 587 588 589
    Mat m(dims, size.p, type());
    size.p[0] = r;
    if( r > 0 )
    {
        Mat mpart = m.rowRange(0, r);
        copyTo(mpart);
    }
590

591 592 593 594 595
    *this = m;
    size.p[0] = r;
    dataend = data + step.p[0]*r;
}

596

597 598 599
void Mat::resize(size_t nelems)
{
    int saveRows = size.p[0];
600 601
    if( saveRows == (int)nelems )
        return;
602
    CV_Assert( (int)nelems >= 0 );
603

604 605
    if( isSubmatrix() || data + step.p[0]*nelems > datalimit )
        reserve(nelems);
606

607 608
    size.p[0] = (int)nelems;
    dataend += (size.p[0] - saveRows)*step.p[0];
609

610
    //updateContinuityFlag(*this);
611 612
}

613 614 615 616 617

void Mat::resize(size_t nelems, const Scalar& s)
{
    int saveRows = size.p[0];
    resize(nelems);
618

619 620 621 622 623
    if( size.p[0] > saveRows )
    {
        Mat part = rowRange(saveRows, size.p[0]);
        part = s;
    }
624 625
}

626 627 628 629 630
void Mat::push_back(const Mat& elems)
{
    int r = size.p[0], delta = elems.size.p[0];
    if( delta == 0 )
        return;
631 632 633 634 635
    if( this == &elems )
    {
        Mat tmp = elems;
        push_back(tmp);
        return;
636
    }
637 638 639 640 641
    if( !data )
    {
        *this = elems.clone();
        return;
    }
642 643 644 645 646 647 648 649

    size.p[0] = elems.size.p[0];
    bool eq = size == elems.size;
    size.p[0] = r;
    if( !eq )
        CV_Error(CV_StsUnmatchedSizes, "");
    if( type() != elems.type() )
        CV_Error(CV_StsUnmatchedFormats, "");
650

651 652
    if( isSubmatrix() || dataend + step.p[0]*delta > datalimit )
        reserve( std::max(r + delta, (r*3+1)/2) );
653

654 655
    size.p[0] += delta;
    dataend += step.p[0]*delta;
656

657
    //updateContinuityFlag(*this);
658

659 660 661 662 663 664 665 666 667
    if( isContinuous() && elems.isContinuous() )
        memcpy(data + r*step.p[0], elems.data, elems.total()*elems.elemSize());
    else
    {
        Mat part = rowRange(r, r + delta);
        elems.copyTo(part);
    }
}

668

669
Mat cvarrToMat(const CvArr* arr, bool copyData,
670
               bool /*allowND*/, int coiMode, AutoBuffer<double>* abuf )
671
{
V
Vadim Pisarevsky 已提交
672 673
    if( !arr )
        return Mat();
A
Andrey Kamaev 已提交
674 675
    if( CV_IS_MAT_HDR_Z(arr) )
        return cvMatToMat((const CvMat*)arr, copyData);
V
Vadim Pisarevsky 已提交
676
    if( CV_IS_MATND(arr) )
A
Andrey Kamaev 已提交
677
        return cvMatNDToMat((const CvMatND*)arr, copyData );
V
Vadim Pisarevsky 已提交
678
    if( CV_IS_IMAGE(arr) )
679 680 681 682
    {
        const IplImage* iplimg = (const IplImage*)arr;
        if( coiMode == 0 && iplimg->roi && iplimg->roi->coi > 0 )
            CV_Error(CV_BadCOI, "COI is not supported by the function");
A
Andrey Kamaev 已提交
683
        return iplImageToMat(iplimg, copyData);
684
    }
V
Vadim Pisarevsky 已提交
685
    if( CV_IS_SEQ(arr) )
686 687
    {
        CvSeq* seq = (CvSeq*)arr;
688 689 690 691
        int total = seq->total, type = CV_MAT_TYPE(seq->flags), esz = seq->elem_size;
        if( total == 0 )
            return Mat();
        CV_Assert(total > 0 && CV_ELEM_SIZE(seq->flags) == esz);
692
        if(!copyData && seq->first->next == seq->first)
693 694 695 696 697 698 699 700 701 702
            return Mat(total, 1, type, seq->first->data);
        if( abuf )
        {
            abuf->allocate(((size_t)total*esz + sizeof(double)-1)/sizeof(double));
            double* bufdata = *abuf;
            cvCvtSeqToArray(seq, bufdata, CV_WHOLE_SEQ);
            return Mat(total, 1, type, bufdata);
        }

        Mat buf(total, 1, type);
703 704 705
        cvCvtSeqToArray(seq, buf.data, CV_WHOLE_SEQ);
        return buf;
    }
V
Vadim Pisarevsky 已提交
706 707 708 709 710 711 712 713 714
    CV_Error(CV_StsBadArg, "Unknown array type");
    return Mat();
}

void Mat::locateROI( Size& wholeSize, Point& ofs ) const
{
    CV_Assert( dims <= 2 && step[0] > 0 );
    size_t esz = elemSize(), minstep;
    ptrdiff_t delta1 = data - datastart, delta2 = dataend - datastart;
715

V
Vadim Pisarevsky 已提交
716 717
    if( delta1 == 0 )
        ofs.x = ofs.y = 0;
718 719
    else
    {
V
Vadim Pisarevsky 已提交
720 721 722
        ofs.y = (int)(delta1/step[0]);
        ofs.x = (int)((delta1 - step[0]*ofs.y)/esz);
        CV_DbgAssert( data == datastart + ofs.y*step[0] + ofs.x*esz );
723
    }
V
Vadim Pisarevsky 已提交
724 725 726 727 728
    minstep = (ofs.x + cols)*esz;
    wholeSize.height = (int)((delta2 - minstep)/step[0] + 1);
    wholeSize.height = std::max(wholeSize.height, ofs.y + rows);
    wholeSize.width = (int)((delta2 - step*(wholeSize.height-1))/esz);
    wholeSize.width = std::max(wholeSize.width, ofs.x + cols);
729 730
}

V
Vadim Pisarevsky 已提交
731 732 733 734 735 736 737 738 739 740 741 742 743 744 745 746
Mat& Mat::adjustROI( int dtop, int dbottom, int dleft, int dright )
{
    CV_Assert( dims <= 2 && step[0] > 0 );
    Size wholeSize; Point ofs;
    size_t esz = elemSize();
    locateROI( wholeSize, ofs );
    int row1 = std::max(ofs.y - dtop, 0), row2 = std::min(ofs.y + rows + dbottom, wholeSize.height);
    int col1 = std::max(ofs.x - dleft, 0), col2 = std::min(ofs.x + cols + dright, wholeSize.width);
    data += (row1 - ofs.y)*step + (col1 - ofs.x)*esz;
    rows = row2 - row1; cols = col2 - col1;
    size.p[0] = rows; size.p[1] = cols;
    if( esz*cols == step[0] || rows == 1 )
        flags |= CONTINUOUS_FLAG;
    else
        flags &= ~CONTINUOUS_FLAG;
    return *this;
747
}
748 749

}
750

751
void cv::extractImageCOI(const CvArr* arr, OutputArray _ch, int coi)
752 753
{
    Mat mat = cvarrToMat(arr, false, true, 1);
754 755
    _ch.create(mat.dims, mat.size, mat.depth());
    Mat ch = _ch.getMat();
V
Vadim Pisarevsky 已提交
756
    if(coi < 0)
757
    {
V
Vadim Pisarevsky 已提交
758 759 760
        CV_Assert( CV_IS_IMAGE(arr) );
        coi = cvGetImageCOI((const IplImage*)arr)-1;
    }
761 762 763 764
    CV_Assert(0 <= coi && coi < mat.channels());
    int _pairs[] = { coi, 0 };
    mixChannels( &mat, 1, &ch, 1, _pairs, 1 );
}
765

766
void cv::insertImageCOI(InputArray _ch, CvArr* arr, int coi)
767
{
768
    Mat ch = _ch.getMat(), mat = cvarrToMat(arr, false, true, 1);
V
Vadim Pisarevsky 已提交
769
    if(coi < 0)
770
    {
V
Vadim Pisarevsky 已提交
771 772 773
        CV_Assert( CV_IS_IMAGE(arr) );
        coi = cvGetImageCOI((const IplImage*)arr)-1;
    }
V
Vadim Pisarevsky 已提交
774
    CV_Assert(ch.size == mat.size && ch.depth() == mat.depth() && 0 <= coi && coi < mat.channels());
775 776 777
    int _pairs[] = { 0, coi };
    mixChannels( &ch, 1, &mat, 1, _pairs, 1 );
}
778

779 780
namespace cv
{
781 782 783 784

Mat Mat::reshape(int new_cn, int new_rows) const
{
    int cn = channels();
785
    Mat hdr = *this;
786

787 788 789 790 791 792 793
    if( dims > 2 && new_rows == 0 && new_cn != 0 && size[dims-1]*cn % new_cn == 0 )
    {
        hdr.flags = (hdr.flags & ~CV_MAT_CN_MASK) | ((new_cn-1) << CV_CN_SHIFT);
        hdr.step[dims-1] = CV_ELEM_SIZE(hdr.flags);
        hdr.size[dims-1] = hdr.size[dims-1]*cn / new_cn;
        return hdr;
    }
794

795
    CV_Assert( dims <= 2 );
796

797 798 799 800 801 802 803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821
    if( new_cn == 0 )
        new_cn = cn;

    int total_width = cols * cn;

    if( (new_cn > total_width || total_width % new_cn != 0) && new_rows == 0 )
        new_rows = rows * total_width / new_cn;

    if( new_rows != 0 && new_rows != rows )
    {
        int total_size = total_width * rows;
        if( !isContinuous() )
            CV_Error( CV_BadStep,
            "The matrix is not continuous, thus its number of rows can not be changed" );

        if( (unsigned)new_rows > (unsigned)total_size )
            CV_Error( CV_StsOutOfRange, "Bad new number of rows" );

        total_width = total_size / new_rows;

        if( total_width * new_rows != total_size )
            CV_Error( CV_StsBadArg, "The total number of matrix elements "
                                    "is not divisible by the new number of rows" );

        hdr.rows = new_rows;
V
Vadim Pisarevsky 已提交
822
        hdr.step[0] = total_width * elemSize1();
823 824 825 826 827 828 829 830 831 832
    }

    int new_width = total_width / new_cn;

    if( new_width * new_cn != total_width )
        CV_Error( CV_BadNumChannels,
        "The total width is not divisible by the new number of channels" );

    hdr.cols = new_width;
    hdr.flags = (hdr.flags & ~CV_MAT_CN_MASK) | ((new_cn-1) << CV_CN_SHIFT);
833
    hdr.step[1] = CV_ELEM_SIZE(hdr.flags);
834 835 836
    return hdr;
}

A
Andrey Kamaev 已提交
837 838 839 840 841 842 843 844 845 846 847 848
Mat Mat::diag(const Mat& d)
{
    CV_Assert( d.cols == 1 || d.rows == 1 );
    int len = d.rows + d.cols - 1;
    Mat m(len, len, d.type(), Scalar(0));
    Mat md = m.diag();
    if( d.cols == 1 )
        d.copyTo(md);
    else
        transpose(d, md);
    return m;
}
849

850 851 852 853
int Mat::checkVector(int _elemChannels, int _depth, bool _requireContinuous) const
{
    return (depth() == _depth || _depth <= 0) &&
        (isContinuous() || !_requireContinuous) &&
854 855
        ((dims == 2 && (((rows == 1 || cols == 1) && channels() == _elemChannels) ||
                        (cols == _elemChannels && channels() == 1))) ||
856 857 858 859
        (dims == 3 && channels() == 1 && size.p[2] == _elemChannels && (size.p[0] == 1 || size.p[1] == 1) &&
         (isContinuous() || step.p[1] == step.p[2]*size.p[2])))
    ? (int)(total()*channels()/_elemChannels) : -1;
}
860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894 895 896 897 898 899 900 901 902 903 904 905 906 907 908 909 910 911 912 913 914 915 916 917 918 919 920 921 922 923 924 925 926 927 928 929 930 931 932 933 934


void scalarToRawData(const Scalar& s, void* _buf, int type, int unroll_to)
{
    int i, depth = CV_MAT_DEPTH(type), cn = CV_MAT_CN(type);
    CV_Assert(cn <= 4);
    switch(depth)
    {
    case CV_8U:
        {
        uchar* buf = (uchar*)_buf;
        for(i = 0; i < cn; i++)
            buf[i] = saturate_cast<uchar>(s.val[i]);
        for(; i < unroll_to; i++)
            buf[i] = buf[i-cn];
        }
        break;
    case CV_8S:
        {
        schar* buf = (schar*)_buf;
        for(i = 0; i < cn; i++)
            buf[i] = saturate_cast<schar>(s.val[i]);
        for(; i < unroll_to; i++)
            buf[i] = buf[i-cn];
        }
        break;
    case CV_16U:
        {
        ushort* buf = (ushort*)_buf;
        for(i = 0; i < cn; i++)
            buf[i] = saturate_cast<ushort>(s.val[i]);
        for(; i < unroll_to; i++)
            buf[i] = buf[i-cn];
        }
        break;
    case CV_16S:
        {
        short* buf = (short*)_buf;
        for(i = 0; i < cn; i++)
            buf[i] = saturate_cast<short>(s.val[i]);
        for(; i < unroll_to; i++)
            buf[i] = buf[i-cn];
        }
        break;
    case CV_32S:
        {
        int* buf = (int*)_buf;
        for(i = 0; i < cn; i++)
            buf[i] = saturate_cast<int>(s.val[i]);
        for(; i < unroll_to; i++)
            buf[i] = buf[i-cn];
        }
        break;
    case CV_32F:
        {
        float* buf = (float*)_buf;
        for(i = 0; i < cn; i++)
            buf[i] = saturate_cast<float>(s.val[i]);
        for(; i < unroll_to; i++)
            buf[i] = buf[i-cn];
        }
        break;
    case CV_64F:
        {
        double* buf = (double*)_buf;
        for(i = 0; i < cn; i++)
            buf[i] = saturate_cast<double>(s.val[i]);
        for(; i < unroll_to; i++)
            buf[i] = buf[i-cn];
        break;
        }
    default:
        CV_Error(CV_StsUnsupportedFormat,"");
    }
}
935

936

937 938 939 940
/*************************************************************************************************\
                                        Input/Output Array
\*************************************************************************************************/

941 942
_InputArray::_InputArray() : flags(0), obj(0) {}
_InputArray::_InputArray(const Mat& m) : flags(MAT), obj((void*)&m) {}
943
_InputArray::_InputArray(const std::vector<Mat>& vec) : flags(STD_VECTOR_MAT), obj((void*)&vec) {}
944 945
_InputArray::_InputArray(const double& val) : flags(FIXED_TYPE + FIXED_SIZE + MATX + CV_64F), obj((void*)&val), sz(Size(1,1)) {}
_InputArray::_InputArray(const MatExpr& expr) : flags(FIXED_TYPE + FIXED_SIZE + EXPR), obj((void*)&expr) {}
946
_InputArray::_InputArray(const cuda::GpuMat& d_mat) : flags(GPU_MAT), obj((void*)&d_mat) {}
947
_InputArray::_InputArray(const ogl::Buffer& buf) : flags(OPENGL_BUFFER), obj((void*)&buf) {}
948
_InputArray::_InputArray(const cuda::CudaMem& cuda_mem) : flags(CUDA_MEM), obj((void*)&cuda_mem) {}
949 950

_InputArray::~_InputArray() {}
951

952
Mat _InputArray::getMat(int i) const
953 954
{
    int k = kind();
955

956 957
    if( k == MAT )
    {
V
Vadim Pisarevsky 已提交
958 959 960 961
        const Mat* m = (const Mat*)obj;
        if( i < 0 )
            return *m;
        return m->row(i);
962
    }
963

964 965 966 967 968
    if( k == EXPR )
    {
        CV_Assert( i < 0 );
        return (Mat)*((const MatExpr*)obj);
    }
969

970 971 972 973 974
    if( k == MATX )
    {
        CV_Assert( i < 0 );
        return Mat(sz, flags, obj);
    }
975

976 977 978 979
    if( k == STD_VECTOR )
    {
        CV_Assert( i < 0 );
        int t = CV_MAT_TYPE(flags);
980
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
981

982 983
        return !v.empty() ? Mat(size(), t, (void*)&v[0]) : Mat();
    }
984

985 986
    if( k == NONE )
        return Mat();
987

988 989 990
    if( k == STD_VECTOR_VECTOR )
    {
        int t = type(i);
991
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
992
        CV_Assert( 0 <= i && i < (int)vv.size() );
993
        const std::vector<uchar>& v = vv[i];
994

995 996
        return !v.empty() ? Mat(size(i), t, (void*)&v[0]) : Mat();
    }
997

998 999
    if( k == OCL_MAT )
    {
1000
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1001 1002
    }

1003
    if( k == STD_VECTOR_MAT )
1004
    {
1005
        const std::vector<Mat>& v = *(const std::vector<Mat>*)obj;
1006
        CV_Assert( 0 <= i && i < (int)v.size() );
1007

1008
        return v[i];
1009
    }
1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020

    if( k == OPENGL_BUFFER )
    {
        CV_Assert( i < 0 );
        CV_Error(cv::Error::StsNotImplemented, "You should explicitly call mapHost/unmapHost methods for ogl::Buffer object");
        return Mat();
    }

    if( k == GPU_MAT )
    {
        CV_Assert( i < 0 );
1021
        CV_Error(cv::Error::StsNotImplemented, "You should explicitly call download method for cuda::GpuMat object");
1022 1023 1024 1025 1026 1027 1028 1029
        return Mat();
    }

    CV_Assert( k == CUDA_MEM );
    //if( k == CUDA_MEM )
    {
        CV_Assert( i < 0 );

1030
        const cuda::CudaMem* cuda_mem = (const cuda::CudaMem*)obj;
1031 1032 1033

        return cuda_mem->createMatHeader();
    }
1034
}
1035 1036


1037
void _InputArray::getMatVector(std::vector<Mat>& mv) const
1038 1039
{
    int k = kind();
1040

1041 1042 1043
    if( k == MAT )
    {
        const Mat& m = *(const Mat*)obj;
1044
        int i, n = (int)m.size[0];
1045
        mv.resize(n);
1046

1047 1048 1049 1050 1051
        for( i = 0; i < n; i++ )
            mv[i] = m.dims == 2 ? Mat(1, m.cols, m.type(), (void*)m.ptr(i)) :
                Mat(m.dims-1, &m.size[1], m.type(), (void*)m.ptr(i), &m.step[1]);
        return;
    }
1052

1053 1054 1055
    if( k == EXPR )
    {
        Mat m = *(const MatExpr*)obj;
1056
        int i, n = m.size[0];
1057
        mv.resize(n);
1058

1059 1060 1061 1062
        for( i = 0; i < n; i++ )
            mv[i] = m.row(i);
        return;
    }
1063

1064 1065 1066 1067
    if( k == MATX )
    {
        size_t i, n = sz.height, esz = CV_ELEM_SIZE(flags);
        mv.resize(n);
1068

1069 1070 1071 1072
        for( i = 0; i < n; i++ )
            mv[i] = Mat(1, sz.width, CV_MAT_TYPE(flags), (uchar*)obj + esz*sz.width*i);
        return;
    }
1073

1074 1075
    if( k == STD_VECTOR )
    {
1076
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
1077

1078 1079 1080
        size_t i, n = v.size(), esz = CV_ELEM_SIZE(flags);
        int t = CV_MAT_DEPTH(flags), cn = CV_MAT_CN(flags);
        mv.resize(n);
1081

1082 1083 1084 1085
        for( i = 0; i < n; i++ )
            mv[i] = Mat(1, cn, t, (void*)(&v[0] + esz*i));
        return;
    }
1086

1087 1088 1089 1090 1091
    if( k == NONE )
    {
        mv.clear();
        return;
    }
1092

1093 1094
    if( k == STD_VECTOR_VECTOR )
    {
1095
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
1096
        int i, n = (int)vv.size();
1097 1098
        int t = CV_MAT_TYPE(flags);
        mv.resize(n);
1099

1100 1101
        for( i = 0; i < n; i++ )
        {
1102
            const std::vector<uchar>& v = vv[i];
1103 1104 1105 1106
            mv[i] = Mat(size(i), t, (void*)&v[0]);
        }
        return;
    }
1107

1108 1109
    if( k == OCL_MAT )
    {
1110
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1111 1112
    }

1113 1114 1115
    CV_Assert( k == STD_VECTOR_MAT );
    //if( k == STD_VECTOR_MAT )
    {
1116
        const std::vector<Mat>& v = *(const std::vector<Mat>*)obj;
1117 1118 1119 1120 1121
        mv.resize(v.size());
        std::copy(v.begin(), v.end(), mv.begin());
        return;
    }
}
1122

1123
cuda::GpuMat _InputArray::getGpuMat() const
1124 1125 1126
{
    int k = kind();

1127 1128
    if (k == GPU_MAT)
    {
1129
        const cuda::GpuMat* d_mat = (const cuda::GpuMat*)obj;
1130 1131
        return *d_mat;
    }
1132

1133 1134
    if (k == CUDA_MEM)
    {
1135
        const cuda::CudaMem* cuda_mem = (const cuda::CudaMem*)obj;
1136 1137 1138 1139 1140 1141
        return cuda_mem->createGpuMatHeader();
    }

    if (k == OPENGL_BUFFER)
    {
        CV_Error(cv::Error::StsNotImplemented, "You should explicitly call mapDevice/unmapDevice methods for ogl::Buffer object");
1142
        return cuda::GpuMat();
1143 1144
    }

V
Vladislav Vinogradov 已提交
1145
    if (k == NONE)
1146
        return cuda::GpuMat();
V
Vladislav Vinogradov 已提交
1147

1148 1149
    CV_Error(cv::Error::StsNotImplemented, "getGpuMat is available only for cuda::GpuMat and cuda::CudaMem");
    return cuda::GpuMat();
1150 1151
}

1152
ogl::Buffer _InputArray::getOGlBuffer() const
1153 1154 1155
{
    int k = kind();

1156 1157 1158 1159
    CV_Assert(k == OPENGL_BUFFER);

    const ogl::Buffer* gl_buf = (const ogl::Buffer*)obj;
    return *gl_buf;
1160 1161
}

1162
int _InputArray::kind() const
1163
{
1164
    return flags & KIND_MASK;
1165
}
1166

1167
Size _InputArray::size(int i) const
1168 1169
{
    int k = kind();
1170

1171 1172 1173 1174 1175
    if( k == MAT )
    {
        CV_Assert( i < 0 );
        return ((const Mat*)obj)->size();
    }
1176

1177 1178 1179 1180 1181
    if( k == EXPR )
    {
        CV_Assert( i < 0 );
        return ((const MatExpr*)obj)->size();
    }
1182

1183 1184 1185 1186 1187
    if( k == MATX )
    {
        CV_Assert( i < 0 );
        return sz;
    }
1188

1189 1190 1191
    if( k == STD_VECTOR )
    {
        CV_Assert( i < 0 );
1192 1193
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
        const std::vector<int>& iv = *(const std::vector<int>*)obj;
1194 1195 1196
        size_t szb = v.size(), szi = iv.size();
        return szb == szi ? Size((int)szb, 1) : Size((int)(szb/CV_ELEM_SIZE(flags)), 1);
    }
1197

1198 1199
    if( k == NONE )
        return Size();
1200

1201 1202
    if( k == STD_VECTOR_VECTOR )
    {
1203
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
1204 1205 1206
        if( i < 0 )
            return vv.empty() ? Size() : Size((int)vv.size(), 1);
        CV_Assert( i < (int)vv.size() );
1207
        const std::vector<std::vector<int> >& ivv = *(const std::vector<std::vector<int> >*)obj;
1208

1209 1210 1211
        size_t szb = vv[i].size(), szi = ivv[i].size();
        return szb == szi ? Size((int)szb, 1) : Size((int)(szb/CV_ELEM_SIZE(flags)), 1);
    }
1212

1213
    if( k == STD_VECTOR_MAT )
1214
    {
1215
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1216 1217 1218
        if( i < 0 )
            return vv.empty() ? Size() : Size((int)vv.size(), 1);
        CV_Assert( i < (int)vv.size() );
1219

1220 1221
        return vv[i].size();
    }
1222 1223 1224 1225

    if( k == OPENGL_BUFFER )
    {
        CV_Assert( i < 0 );
1226
        const ogl::Buffer* buf = (const ogl::Buffer*)obj;
1227 1228 1229
        return buf->size();
    }

1230
    if( k == GPU_MAT )
1231 1232
    {
        CV_Assert( i < 0 );
1233
        const cuda::GpuMat* d_mat = (const cuda::GpuMat*)obj;
1234 1235
        return d_mat->size();
    }
1236

1237 1238
    if( k == OCL_MAT )
    {
1239
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1240 1241
    }

1242 1243 1244 1245
    CV_Assert( k == CUDA_MEM );
    //if( k == CUDA_MEM )
    {
        CV_Assert( i < 0 );
1246
        const cuda::CudaMem* cuda_mem = (const cuda::CudaMem*)obj;
1247 1248
        return cuda_mem->size();
    }
1249 1250
}

1251
size_t _InputArray::total(int i) const
1252
{
1253 1254 1255 1256 1257 1258 1259 1260 1261 1262
    int k = kind();

    if( k == MAT )
    {
        CV_Assert( i < 0 );
        return ((const Mat*)obj)->total();
    }

    if( k == STD_VECTOR_MAT )
    {
1263
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1264 1265 1266 1267 1268 1269 1270
        if( i < 0 )
            return vv.size();

        CV_Assert( i < (int)vv.size() );
        return vv[i].total();
    }

1271 1272
    return size(i).area();
}
1273

1274
int _InputArray::type(int i) const
1275 1276
{
    int k = kind();
1277

1278 1279
    if( k == MAT )
        return ((const Mat*)obj)->type();
1280

1281 1282
    if( k == EXPR )
        return ((const MatExpr*)obj)->type();
1283

1284 1285
    if( k == MATX || k == STD_VECTOR || k == STD_VECTOR_VECTOR )
        return CV_MAT_TYPE(flags);
1286

1287 1288
    if( k == NONE )
        return -1;
1289

1290
    if( k == STD_VECTOR_MAT )
1291
    {
1292
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1293
        CV_Assert( i < (int)vv.size() );
1294

1295 1296
        return vv[i >= 0 ? i : 0].type();
    }
1297

1298
    if( k == OPENGL_BUFFER )
1299
        return ((const ogl::Buffer*)obj)->type();
1300

1301
    if( k == GPU_MAT )
1302
        return ((const cuda::GpuMat*)obj)->type();
1303 1304 1305

    CV_Assert( k == CUDA_MEM );
    //if( k == CUDA_MEM )
1306
        return ((const cuda::CudaMem*)obj)->type();
1307
}
1308

1309
int _InputArray::depth(int i) const
1310 1311 1312
{
    return CV_MAT_DEPTH(type(i));
}
1313

1314
int _InputArray::channels(int i) const
1315 1316 1317
{
    return CV_MAT_CN(type(i));
}
1318

1319
bool _InputArray::empty() const
1320 1321
{
    int k = kind();
1322

1323 1324
    if( k == MAT )
        return ((const Mat*)obj)->empty();
1325

1326 1327
    if( k == EXPR )
        return false;
1328

1329 1330
    if( k == MATX )
        return false;
1331

1332 1333
    if( k == STD_VECTOR )
    {
1334
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
1335 1336
        return v.empty();
    }
1337

1338 1339
    if( k == NONE )
        return true;
1340

1341 1342
    if( k == STD_VECTOR_VECTOR )
    {
1343
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
1344 1345
        return vv.empty();
    }
1346

1347
    if( k == STD_VECTOR_MAT )
1348
    {
1349
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1350 1351
        return vv.empty();
    }
1352

1353
    if( k == OPENGL_BUFFER )
1354
        return ((const ogl::Buffer*)obj)->empty();
1355

1356 1357
    if( k == OCL_MAT )
    {
1358
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1359 1360
    }

1361
    if( k == GPU_MAT )
1362
        return ((const cuda::GpuMat*)obj)->empty();
1363 1364 1365

    CV_Assert( k == CUDA_MEM );
    //if( k == CUDA_MEM )
1366
        return ((const cuda::CudaMem*)obj)->empty();
1367
}
1368 1369


1370 1371
_OutputArray::_OutputArray() {}
_OutputArray::_OutputArray(Mat& m) : _InputArray(m) {}
1372
_OutputArray::_OutputArray(std::vector<Mat>& vec) : _InputArray(vec) {}
1373
_OutputArray::_OutputArray(cuda::GpuMat& d_mat) : _InputArray(d_mat) {}
1374
_OutputArray::_OutputArray(ogl::Buffer& buf) : _InputArray(buf) {}
1375
_OutputArray::_OutputArray(cuda::CudaMem& cuda_mem) : _InputArray(cuda_mem) {}
1376 1377

_OutputArray::_OutputArray(const Mat& m) : _InputArray(m) {flags |= FIXED_SIZE|FIXED_TYPE;}
1378
_OutputArray::_OutputArray(const std::vector<Mat>& vec) : _InputArray(vec) {flags |= FIXED_SIZE;}
1379
_OutputArray::_OutputArray(const cuda::GpuMat& d_mat) : _InputArray(d_mat) {flags |= FIXED_SIZE|FIXED_TYPE;}
1380
_OutputArray::_OutputArray(const ogl::Buffer& buf) : _InputArray(buf) {flags |= FIXED_SIZE|FIXED_TYPE;}
1381
_OutputArray::_OutputArray(const cuda::CudaMem& cuda_mem) : _InputArray(cuda_mem) {flags |= FIXED_SIZE|FIXED_TYPE;}
1382

1383
_OutputArray::~_OutputArray() {}
1384

1385
bool _OutputArray::fixedSize() const
1386
{
1387
    return (flags & FIXED_SIZE) == FIXED_SIZE;
1388 1389
}

1390
bool _OutputArray::fixedType() const
1391
{
1392
    return (flags & FIXED_TYPE) == FIXED_TYPE;
1393
}
1394

A
Andrey Kamaev 已提交
1395
void _OutputArray::create(Size _sz, int mtype, int i, bool allowTransposed, int fixedDepthMask) const
1396 1397 1398 1399
{
    int k = kind();
    if( k == MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1400
        CV_Assert(!fixedSize() || ((Mat*)obj)->size.operator()() == _sz);
A
Andrey Kamaev 已提交
1401 1402
        CV_Assert(!fixedType() || ((Mat*)obj)->type() == mtype);
        ((Mat*)obj)->create(_sz, mtype);
1403 1404
        return;
    }
1405 1406
    if( k == GPU_MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1407 1408 1409
        CV_Assert(!fixedSize() || ((cuda::GpuMat*)obj)->size() == _sz);
        CV_Assert(!fixedType() || ((cuda::GpuMat*)obj)->type() == mtype);
        ((cuda::GpuMat*)obj)->create(_sz, mtype);
1410 1411
        return;
    }
1412 1413
    if( k == OPENGL_BUFFER && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1414 1415 1416
        CV_Assert(!fixedSize() || ((ogl::Buffer*)obj)->size() == _sz);
        CV_Assert(!fixedType() || ((ogl::Buffer*)obj)->type() == mtype);
        ((ogl::Buffer*)obj)->create(_sz, mtype);
1417 1418
        return;
    }
1419 1420
    if( k == CUDA_MEM && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1421 1422 1423
        CV_Assert(!fixedSize() || ((cuda::CudaMem*)obj)->size() == _sz);
        CV_Assert(!fixedType() || ((cuda::CudaMem*)obj)->type() == mtype);
        ((cuda::CudaMem*)obj)->create(_sz, mtype);
1424 1425
        return;
    }
A
Andrey Kamaev 已提交
1426 1427
    int sizes[] = {_sz.height, _sz.width};
    create(2, sizes, mtype, i, allowTransposed, fixedDepthMask);
1428 1429
}

A
Andrey Kamaev 已提交
1430
void _OutputArray::create(int rows, int cols, int mtype, int i, bool allowTransposed, int fixedDepthMask) const
1431 1432 1433 1434
{
    int k = kind();
    if( k == MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1435
        CV_Assert(!fixedSize() || ((Mat*)obj)->size.operator()() == Size(cols, rows));
A
Andrey Kamaev 已提交
1436 1437
        CV_Assert(!fixedType() || ((Mat*)obj)->type() == mtype);
        ((Mat*)obj)->create(rows, cols, mtype);
1438 1439
        return;
    }
1440 1441
    if( k == GPU_MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1442 1443 1444
        CV_Assert(!fixedSize() || ((cuda::GpuMat*)obj)->size() == Size(cols, rows));
        CV_Assert(!fixedType() || ((cuda::GpuMat*)obj)->type() == mtype);
        ((cuda::GpuMat*)obj)->create(rows, cols, mtype);
1445 1446
        return;
    }
1447 1448
    if( k == OPENGL_BUFFER && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1449 1450 1451
        CV_Assert(!fixedSize() || ((ogl::Buffer*)obj)->size() == Size(cols, rows));
        CV_Assert(!fixedType() || ((ogl::Buffer*)obj)->type() == mtype);
        ((ogl::Buffer*)obj)->create(rows, cols, mtype);
1452 1453
        return;
    }
1454 1455
    if( k == CUDA_MEM && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1456 1457 1458
        CV_Assert(!fixedSize() || ((cuda::CudaMem*)obj)->size() == Size(cols, rows));
        CV_Assert(!fixedType() || ((cuda::CudaMem*)obj)->type() == mtype);
        ((cuda::CudaMem*)obj)->create(rows, cols, mtype);
1459 1460
        return;
    }
A
Andrey Kamaev 已提交
1461 1462
    int sizes[] = {rows, cols};
    create(2, sizes, mtype, i, allowTransposed, fixedDepthMask);
1463
}
1464

A
Andrey Kamaev 已提交
1465
void _OutputArray::create(int dims, const int* sizes, int mtype, int i, bool allowTransposed, int fixedDepthMask) const
1466 1467
{
    int k = kind();
A
Andrey Kamaev 已提交
1468
    mtype = CV_MAT_TYPE(mtype);
1469

1470 1471 1472 1473
    if( k == MAT )
    {
        CV_Assert( i < 0 );
        Mat& m = *(Mat*)obj;
1474
        if( allowTransposed )
1475 1476
        {
            if( !m.isContinuous() )
1477 1478
            {
                CV_Assert(!fixedType() && !fixedSize());
1479
                m.release();
1480
            }
1481

1482
            if( dims == 2 && m.dims == 2 && m.data &&
A
Andrey Kamaev 已提交
1483
                m.type() == mtype && m.rows == sizes[1] && m.cols == sizes[0] )
1484 1485
                return;
        }
1486 1487 1488

        if(fixedType())
        {
A
Andrey Kamaev 已提交
1489 1490
            if(CV_MAT_CN(mtype) == m.channels() && ((1 << CV_MAT_TYPE(flags)) & fixedDepthMask) != 0 )
                mtype = m.type();
1491
            else
A
Andrey Kamaev 已提交
1492
                CV_Assert(CV_MAT_TYPE(mtype) == m.type());
1493 1494 1495 1496 1497
        }
        if(fixedSize())
        {
            CV_Assert(m.dims == dims);
            for(int j = 0; j < dims; ++j)
A
Andrey Kamaev 已提交
1498
                CV_Assert(m.size[j] == sizes[j]);
1499
        }
A
Andrey Kamaev 已提交
1500
        m.create(dims, sizes, mtype);
1501 1502
        return;
    }
1503

1504 1505 1506 1507
    if( k == MATX )
    {
        CV_Assert( i < 0 );
        int type0 = CV_MAT_TYPE(flags);
A
Andrey Kamaev 已提交
1508 1509 1510
        CV_Assert( mtype == type0 || (CV_MAT_CN(mtype) == 1 && ((1 << type0) & fixedDepthMask) != 0) );
        CV_Assert( dims == 2 && ((sizes[0] == sz.height && sizes[1] == sz.width) ||
                                 (allowTransposed && sizes[0] == sz.width && sizes[1] == sz.height)));
1511 1512
        return;
    }
1513

1514 1515
    if( k == STD_VECTOR || k == STD_VECTOR_VECTOR )
    {
A
Andrey Kamaev 已提交
1516 1517
        CV_Assert( dims == 2 && (sizes[0] == 1 || sizes[1] == 1 || sizes[0]*sizes[1] == 0) );
        size_t len = sizes[0]*sizes[1] > 0 ? sizes[0] + sizes[1] - 1 : 0;
1518
        std::vector<uchar>* v = (std::vector<uchar>*)obj;
1519

1520 1521
        if( k == STD_VECTOR_VECTOR )
        {
1522
            std::vector<std::vector<uchar> >& vv = *(std::vector<std::vector<uchar> >*)obj;
1523 1524
            if( i < 0 )
            {
1525
                CV_Assert(!fixedSize() || len == vv.size());
1526 1527 1528 1529 1530 1531 1532 1533
                vv.resize(len);
                return;
            }
            CV_Assert( i < (int)vv.size() );
            v = &vv[i];
        }
        else
            CV_Assert( i < 0 );
1534

1535
        int type0 = CV_MAT_TYPE(flags);
A
Andrey Kamaev 已提交
1536
        CV_Assert( mtype == type0 || (CV_MAT_CN(mtype) == CV_MAT_CN(type0) && ((1 << type0) & fixedDepthMask) != 0) );
1537

1538
        int esz = CV_ELEM_SIZE(type0);
1539
        CV_Assert(!fixedSize() || len == ((std::vector<uchar>*)v)->size() / esz);
1540 1541 1542
        switch( esz )
        {
        case 1:
1543
            ((std::vector<uchar>*)v)->resize(len);
1544 1545
            break;
        case 2:
1546
            ((std::vector<Vec2b>*)v)->resize(len);
1547 1548
            break;
        case 3:
1549
            ((std::vector<Vec3b>*)v)->resize(len);
1550 1551
            break;
        case 4:
1552
            ((std::vector<int>*)v)->resize(len);
1553 1554
            break;
        case 6:
1555
            ((std::vector<Vec3s>*)v)->resize(len);
1556 1557
            break;
        case 8:
1558
            ((std::vector<Vec2i>*)v)->resize(len);
1559 1560
            break;
        case 12:
1561
            ((std::vector<Vec3i>*)v)->resize(len);
1562 1563
            break;
        case 16:
1564
            ((std::vector<Vec4i>*)v)->resize(len);
1565 1566
            break;
        case 24:
1567
            ((std::vector<Vec6i>*)v)->resize(len);
1568 1569
            break;
        case 32:
1570
            ((std::vector<Vec8i>*)v)->resize(len);
1571 1572
            break;
        case 36:
1573
            ((std::vector<Vec<int, 9> >*)v)->resize(len);
1574 1575
            break;
        case 48:
1576
            ((std::vector<Vec<int, 12> >*)v)->resize(len);
1577 1578
            break;
        case 64:
1579
            ((std::vector<Vec<int, 16> >*)v)->resize(len);
1580 1581
            break;
        case 128:
1582
            ((std::vector<Vec<int, 32> >*)v)->resize(len);
1583 1584
            break;
        case 256:
1585
            ((std::vector<Vec<int, 64> >*)v)->resize(len);
1586 1587
            break;
        case 512:
1588
            ((std::vector<Vec<int, 128> >*)v)->resize(len);
1589 1590 1591 1592 1593 1594
            break;
        default:
            CV_Error_(CV_StsBadArg, ("Vectors with element size %d are not supported. Please, modify OutputArray::create()\n", esz));
        }
        return;
    }
1595

1596 1597
    if( k == OCL_MAT )
    {
1598
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1599 1600
    }

1601 1602
    if( k == NONE )
    {
1603
        CV_Error(CV_StsNullPtr, "create() called for the missing output array" );
1604 1605
        return;
    }
1606

1607 1608 1609
    CV_Assert( k == STD_VECTOR_MAT );
    //if( k == STD_VECTOR_MAT )
    {
1610
        std::vector<Mat>& v = *(std::vector<Mat>*)obj;
1611

1612 1613
        if( i < 0 )
        {
A
Andrey Kamaev 已提交
1614 1615
            CV_Assert( dims == 2 && (sizes[0] == 1 || sizes[1] == 1 || sizes[0]*sizes[1] == 0) );
            size_t len = sizes[0]*sizes[1] > 0 ? sizes[0] + sizes[1] - 1 : 0, len0 = v.size();
1616

1617
            CV_Assert(!fixedSize() || len == len0);
1618
            v.resize(len);
1619 1620
            if( fixedType() )
            {
A
Andrey Kamaev 已提交
1621
                int _type = CV_MAT_TYPE(flags);
1622 1623
                for( size_t j = len0; j < len; j++ )
                {
1624
                    if( v[j].type() == _type )
1625
                        continue;
1626 1627
                    CV_Assert( v[j].empty() );
                    v[j].flags = (v[j].flags & ~CV_MAT_TYPE_MASK) | _type;
1628 1629
                }
            }
1630 1631
            return;
        }
1632

1633 1634
        CV_Assert( i < (int)v.size() );
        Mat& m = v[i];
1635

1636
        if( allowTransposed )
1637 1638
        {
            if( !m.isContinuous() )
1639 1640
            {
                CV_Assert(!fixedType() && !fixedSize());
1641
                m.release();
1642
            }
1643

1644
            if( dims == 2 && m.dims == 2 && m.data &&
A
Andrey Kamaev 已提交
1645
                m.type() == mtype && m.rows == sizes[1] && m.cols == sizes[0] )
1646 1647
                return;
        }
1648 1649 1650

        if(fixedType())
        {
A
Andrey Kamaev 已提交
1651 1652
            if(CV_MAT_CN(mtype) == m.channels() && ((1 << CV_MAT_TYPE(flags)) & fixedDepthMask) != 0 )
                mtype = m.type();
1653
            else
A
Andrey Kamaev 已提交
1654
                CV_Assert(!fixedType() || (CV_MAT_CN(mtype) == m.channels() && ((1 << CV_MAT_TYPE(flags)) & fixedDepthMask) != 0));
1655 1656 1657 1658 1659
        }
        if(fixedSize())
        {
            CV_Assert(m.dims == dims);
            for(int j = 0; j < dims; ++j)
A
Andrey Kamaev 已提交
1660
                CV_Assert(m.size[j] == sizes[j]);
1661 1662
        }

A
Andrey Kamaev 已提交
1663
        m.create(dims, sizes, mtype);
1664 1665
    }
}
1666

1667
void _OutputArray::release() const
1668
{
1669 1670
    CV_Assert(!fixedSize());

1671
    int k = kind();
1672

1673 1674 1675 1676 1677
    if( k == MAT )
    {
        ((Mat*)obj)->release();
        return;
    }
1678

1679 1680
    if( k == GPU_MAT )
    {
1681
        ((cuda::GpuMat*)obj)->release();
1682 1683 1684
        return;
    }

1685 1686
    if( k == CUDA_MEM )
    {
1687
        ((cuda::CudaMem*)obj)->release();
1688 1689 1690
        return;
    }

1691 1692
    if( k == OPENGL_BUFFER )
    {
1693
        ((ogl::Buffer*)obj)->release();
1694 1695 1696
        return;
    }

1697 1698
    if( k == NONE )
        return;
1699

1700 1701 1702 1703 1704
    if( k == STD_VECTOR )
    {
        create(Size(), CV_MAT_TYPE(flags));
        return;
    }
1705

1706 1707
    if( k == STD_VECTOR_VECTOR )
    {
1708
        ((std::vector<std::vector<uchar> >*)obj)->clear();
1709 1710
        return;
    }
1711

1712 1713
    if( k == OCL_MAT )
    {
1714
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1715 1716
    }

1717 1718 1719
    CV_Assert( k == STD_VECTOR_MAT );
    //if( k == STD_VECTOR_MAT )
    {
1720
        ((std::vector<Mat>*)obj)->clear();
1721
    }
1722 1723
}

1724
void _OutputArray::clear() const
1725 1726
{
    int k = kind();
1727

1728 1729
    if( k == MAT )
    {
1730
        CV_Assert(!fixedSize());
1731 1732 1733
        ((Mat*)obj)->resize(0);
        return;
    }
1734

1735 1736
    release();
}
1737

1738
bool _OutputArray::needed() const
1739 1740 1741 1742
{
    return kind() != NONE;
}

1743
Mat& _OutputArray::getMatRef(int i) const
1744 1745 1746 1747 1748 1749 1750 1751 1752 1753
{
    int k = kind();
    if( i < 0 )
    {
        CV_Assert( k == MAT );
        return *(Mat*)obj;
    }
    else
    {
        CV_Assert( k == STD_VECTOR_MAT );
1754
        std::vector<Mat>& v = *(std::vector<Mat>*)obj;
1755 1756 1757 1758
        CV_Assert( i < (int)v.size() );
        return v[i];
    }
}
1759

1760
cuda::GpuMat& _OutputArray::getGpuMatRef() const
1761 1762 1763
{
    int k = kind();
    CV_Assert( k == GPU_MAT );
1764
    return *(cuda::GpuMat*)obj;
1765 1766
}

1767
ogl::Buffer& _OutputArray::getOGlBufferRef() const
1768 1769 1770
{
    int k = kind();
    CV_Assert( k == OPENGL_BUFFER );
1771
    return *(ogl::Buffer*)obj;
1772 1773
}

1774
cuda::CudaMem& _OutputArray::getCudaMemRef() const
1775 1776 1777
{
    int k = kind();
    CV_Assert( k == CUDA_MEM );
1778
    return *(cuda::CudaMem*)obj;
1779 1780
}

1781
static _OutputArray _none;
1782
OutputArray noArray() { return _none; }
1783

1784 1785
}

1786 1787 1788
/*************************************************************************************************\
                                        Matrix Operations
\*************************************************************************************************/
1789

1790
void cv::hconcat(const Mat* src, size_t nsrc, OutputArray _dst)
1791 1792 1793
{
    if( nsrc == 0 || !src )
    {
1794
        _dst.release();
1795 1796
        return;
    }
1797

1798 1799 1800 1801 1802 1803 1804 1805 1806
    int totalCols = 0, cols = 0;
    size_t i;
    for( i = 0; i < nsrc; i++ )
    {
        CV_Assert( !src[i].empty() && src[i].dims <= 2 &&
                   src[i].rows == src[0].rows &&
                   src[i].type() == src[0].type());
        totalCols += src[i].cols;
    }
1807 1808
    _dst.create( src[0].rows, totalCols, src[0].type());
    Mat dst = _dst.getMat();
1809 1810
    for( i = 0; i < nsrc; i++ )
    {
1811
        Mat dpart = dst(Rect(cols, 0, src[i].cols, src[i].rows));
1812 1813 1814 1815
        src[i].copyTo(dpart);
        cols += src[i].cols;
    }
}
1816

1817
void cv::hconcat(InputArray src1, InputArray src2, OutputArray dst)
1818
{
1819
    Mat src[] = {src1.getMat(), src2.getMat()};
1820 1821
    hconcat(src, 2, dst);
}
1822

1823
void cv::hconcat(InputArray _src, OutputArray dst)
1824
{
1825
    std::vector<Mat> src;
1826
    _src.getMatVector(src);
1827 1828 1829
    hconcat(!src.empty() ? &src[0] : 0, src.size(), dst);
}

1830
void cv::vconcat(const Mat* src, size_t nsrc, OutputArray _dst)
1831 1832 1833
{
    if( nsrc == 0 || !src )
    {
1834
        _dst.release();
1835 1836
        return;
    }
1837

1838 1839 1840 1841 1842 1843 1844 1845 1846
    int totalRows = 0, rows = 0;
    size_t i;
    for( i = 0; i < nsrc; i++ )
    {
        CV_Assert( !src[i].empty() && src[i].dims <= 2 &&
                  src[i].cols == src[0].cols &&
                  src[i].type() == src[0].type());
        totalRows += src[i].rows;
    }
1847 1848
    _dst.create( totalRows, src[0].cols, src[0].type());
    Mat dst = _dst.getMat();
1849 1850 1851 1852 1853 1854 1855
    for( i = 0; i < nsrc; i++ )
    {
        Mat dpart(dst, Rect(0, rows, src[i].cols, src[i].rows));
        src[i].copyTo(dpart);
        rows += src[i].rows;
    }
}
1856

1857
void cv::vconcat(InputArray src1, InputArray src2, OutputArray dst)
1858
{
1859
    Mat src[] = {src1.getMat(), src2.getMat()};
1860
    vconcat(src, 2, dst);
1861
}
1862

1863
void cv::vconcat(InputArray _src, OutputArray dst)
1864
{
1865
    std::vector<Mat> src;
1866
    _src.getMatVector(src);
1867 1868
    vconcat(!src.empty() ? &src[0] : 0, src.size(), dst);
}
1869

1870
//////////////////////////////////////// set identity ////////////////////////////////////////////
1871
void cv::setIdentity( InputOutputArray _m, const Scalar& s )
1872
{
1873
    Mat m = _m.getMat();
V
Vadim Pisarevsky 已提交
1874
    CV_Assert( m.dims <= 2 );
1875
    int i, j, rows = m.rows, cols = m.cols, type = m.type();
1876

1877 1878 1879 1880 1881 1882 1883 1884 1885 1886 1887 1888 1889 1890 1891 1892 1893 1894 1895 1896 1897 1898 1899 1900 1901 1902 1903 1904 1905 1906 1907 1908 1909
    if( type == CV_32FC1 )
    {
        float* data = (float*)m.data;
        float val = (float)s[0];
        size_t step = m.step/sizeof(data[0]);

        for( i = 0; i < rows; i++, data += step )
        {
            for( j = 0; j < cols; j++ )
                data[j] = 0;
            if( i < cols )
                data[i] = val;
        }
    }
    else if( type == CV_64FC1 )
    {
        double* data = (double*)m.data;
        double val = s[0];
        size_t step = m.step/sizeof(data[0]);

        for( i = 0; i < rows; i++, data += step )
        {
            for( j = 0; j < cols; j++ )
                data[j] = j == i ? val : 0;
        }
    }
    else
    {
        m = Scalar(0);
        m.diag() = s;
    }
}

1910 1911
//////////////////////////////////////////// trace ///////////////////////////////////////////

1912
cv::Scalar cv::trace( InputArray _m )
1913
{
1914
    Mat m = _m.getMat();
V
Vadim Pisarevsky 已提交
1915
    CV_Assert( m.dims <= 2 );
1916 1917
    int i, type = m.type();
    int nm = std::min(m.rows, m.cols);
1918

1919 1920 1921 1922 1923 1924 1925 1926 1927
    if( type == CV_32FC1 )
    {
        const float* ptr = (const float*)m.data;
        size_t step = m.step/sizeof(ptr[0]) + 1;
        double _s = 0;
        for( i = 0; i < nm; i++ )
            _s += ptr[i*step];
        return _s;
    }
1928

1929 1930 1931 1932 1933 1934 1935 1936 1937
    if( type == CV_64FC1 )
    {
        const double* ptr = (const double*)m.data;
        size_t step = m.step/sizeof(ptr[0]) + 1;
        double _s = 0;
        for( i = 0; i < nm; i++ )
            _s += ptr[i*step];
        return _s;
    }
1938

1939 1940 1941
    return cv::sum(m.diag());
}

1942
////////////////////////////////////// transpose /////////////////////////////////////////
1943 1944 1945 1946

namespace cv
{

1947
template<typename T> static void
1948
transpose_( const uchar* src, size_t sstep, uchar* dst, size_t dstep, Size sz )
1949
{
V
Victoria Zhislina 已提交
1950 1951
    int i=0, j, m = sz.width, n = sz.height;

1952
    #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
1953
    for(; i <= m - 4; i += 4 )
1954 1955 1956 1957 1958
    {
        T* d0 = (T*)(dst + dstep*i);
        T* d1 = (T*)(dst + dstep*(i+1));
        T* d2 = (T*)(dst + dstep*(i+2));
        T* d3 = (T*)(dst + dstep*(i+3));
1959

1960 1961 1962 1963 1964 1965
        for( j = 0; j <= n - 4; j += 4 )
        {
            const T* s0 = (const T*)(src + i*sizeof(T) + sstep*j);
            const T* s1 = (const T*)(src + i*sizeof(T) + sstep*(j+1));
            const T* s2 = (const T*)(src + i*sizeof(T) + sstep*(j+2));
            const T* s3 = (const T*)(src + i*sizeof(T) + sstep*(j+3));
1966

1967 1968 1969 1970 1971
            d0[j] = s0[0]; d0[j+1] = s1[0]; d0[j+2] = s2[0]; d0[j+3] = s3[0];
            d1[j] = s0[1]; d1[j+1] = s1[1]; d1[j+2] = s2[1]; d1[j+3] = s3[1];
            d2[j] = s0[2]; d2[j+1] = s1[2]; d2[j+2] = s2[2]; d2[j+3] = s3[2];
            d3[j] = s0[3]; d3[j+1] = s1[3]; d3[j+2] = s2[3]; d3[j+3] = s3[3];
        }
1972

1973 1974 1975 1976 1977 1978
        for( ; j < n; j++ )
        {
            const T* s0 = (const T*)(src + i*sizeof(T) + j*sstep);
            d0[j] = s0[0]; d1[j] = s0[1]; d2[j] = s0[2]; d3[j] = s0[3];
        }
    }
V
Victoria Zhislina 已提交
1979
    #endif
1980 1981 1982
    for( ; i < m; i++ )
    {
        T* d0 = (T*)(dst + dstep*i);
V
Victoria Zhislina 已提交
1983
        j = 0;
1984
        #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
1985
        for(; j <= n - 4; j += 4 )
1986 1987 1988 1989 1990
        {
            const T* s0 = (const T*)(src + i*sizeof(T) + sstep*j);
            const T* s1 = (const T*)(src + i*sizeof(T) + sstep*(j+1));
            const T* s2 = (const T*)(src + i*sizeof(T) + sstep*(j+2));
            const T* s3 = (const T*)(src + i*sizeof(T) + sstep*(j+3));
1991

1992 1993
            d0[j] = s0[0]; d0[j+1] = s1[0]; d0[j+2] = s2[0]; d0[j+3] = s3[0];
        }
V
Victoria Zhislina 已提交
1994
        #endif
1995 1996 1997 1998 1999 2000 2001
        for( ; j < n; j++ )
        {
            const T* s0 = (const T*)(src + i*sizeof(T) + j*sstep);
            d0[j] = s0[0];
        }
    }
}
2002

2003 2004 2005 2006 2007
template<typename T> static void
transposeI_( uchar* data, size_t step, int n )
{
    int i, j;
    for( i = 0; i < n; i++ )
2008 2009 2010
    {
        T* row = (T*)(data + step*i);
        uchar* data1 = data + i*sizeof(T);
2011
        for( j = i+1; j < n; j++ )
2012 2013 2014
            std::swap( row[j], *(T*)(data1 + step*j) );
    }
}
2015

2016 2017
typedef void (*TransposeFunc)( const uchar* src, size_t sstep, uchar* dst, size_t dstep, Size sz );
typedef void (*TransposeInplaceFunc)( uchar* data, size_t step, int n );
2018

2019 2020 2021 2022 2023 2024
#define DEF_TRANSPOSE_FUNC(suffix, type) \
static void transpose_##suffix( const uchar* src, size_t sstep, uchar* dst, size_t dstep, Size sz ) \
{ transpose_<type>(src, sstep, dst, dstep, sz); } \
\
static void transposeI_##suffix( uchar* data, size_t step, int n ) \
{ transposeI_<type>(data, step, n); }
2025

2026 2027 2028 2029 2030 2031 2032 2033 2034 2035
DEF_TRANSPOSE_FUNC(8u, uchar)
DEF_TRANSPOSE_FUNC(16u, ushort)
DEF_TRANSPOSE_FUNC(8uC3, Vec3b)
DEF_TRANSPOSE_FUNC(32s, int)
DEF_TRANSPOSE_FUNC(16uC3, Vec3s)
DEF_TRANSPOSE_FUNC(32sC2, Vec2i)
DEF_TRANSPOSE_FUNC(32sC3, Vec3i)
DEF_TRANSPOSE_FUNC(32sC4, Vec4i)
DEF_TRANSPOSE_FUNC(32sC6, Vec6i)
DEF_TRANSPOSE_FUNC(32sC8, Vec8i)
2036

2037 2038 2039 2040 2041 2042
static TransposeFunc transposeTab[] =
{
    0, transpose_8u, transpose_16u, transpose_8uC3, transpose_32s, 0, transpose_16uC3, 0,
    transpose_32sC2, 0, 0, 0, transpose_32sC3, 0, 0, 0, transpose_32sC4,
    0, 0, 0, 0, 0, 0, 0, transpose_32sC6, 0, 0, 0, 0, 0, 0, 0, transpose_32sC8
};
2043

2044 2045 2046 2047 2048 2049
static TransposeInplaceFunc transposeInplaceTab[] =
{
    0, transposeI_8u, transposeI_16u, transposeI_8uC3, transposeI_32s, 0, transposeI_16uC3, 0,
    transposeI_32sC2, 0, 0, 0, transposeI_32sC3, 0, 0, 0, transposeI_32sC4,
    0, 0, 0, 0, 0, 0, 0, transposeI_32sC6, 0, 0, 0, 0, 0, 0, 0, transposeI_32sC8
};
2050

2051
}
2052

2053
void cv::transpose( InputArray _src, OutputArray _dst )
2054 2055
{
    Mat src = _src.getMat();
2056 2057 2058 2059 2060
    if( src.empty() )
    {
        _dst.release();
        return;
    }
2061
    size_t esz = src.elemSize();
V
Vadim Pisarevsky 已提交
2062
    CV_Assert( src.dims <= 2 && esz <= (size_t)32 );
2063

2064 2065
    _dst.create(src.cols, src.rows, src.type());
    Mat dst = _dst.getMat();
2066

2067 2068 2069
    // handle the case of single-column/single-row matrices, stored in STL vectors.
    if( src.rows != dst.cols || src.cols != dst.rows )
    {
2070
        CV_Assert( src.size() == dst.size() && (src.cols == 1 || src.rows == 1) );
2071 2072 2073 2074
        src.copyTo(dst);
        return;
    }

2075
    if( dst.data == src.data )
2076
    {
2077
        TransposeInplaceFunc func = transposeInplaceTab[esz];
2078
        CV_Assert( func != 0 );
2079
        func( dst.data, dst.step, dst.rows );
2080 2081 2082
    }
    else
    {
2083
        TransposeFunc func = transposeTab[esz];
2084
        CV_Assert( func != 0 );
2085
        func( src.data, src.step, dst.data, dst.step, src.size() );
2086 2087 2088 2089
    }
}


2090
void cv::completeSymm( InputOutputArray _m, bool LtoR )
2091
{
2092
    Mat m = _m.getMat();
V
Vadim Pisarevsky 已提交
2093
    CV_Assert( m.dims <= 2 );
2094

V
Vadim Pisarevsky 已提交
2095
    int i, j, nrows = m.rows, type = m.type();
2096
    int j0 = 0, j1 = nrows;
V
Vadim Pisarevsky 已提交
2097
    CV_Assert( m.rows == m.cols );
2098 2099 2100

    if( type == CV_32FC1 || type == CV_32SC1 )
    {
V
Vadim Pisarevsky 已提交
2101 2102
        int* data = (int*)m.data;
        size_t step = m.step/sizeof(data[0]);
2103 2104 2105 2106 2107 2108 2109 2110 2111
        for( i = 0; i < nrows; i++ )
        {
            if( !LtoR ) j1 = i; else j0 = i+1;
            for( j = j0; j < j1; j++ )
                data[i*step + j] = data[j*step + i];
        }
    }
    else if( type == CV_64FC1 )
    {
V
Vadim Pisarevsky 已提交
2112 2113
        double* data = (double*)m.data;
        size_t step = m.step/sizeof(data[0]);
2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124
        for( i = 0; i < nrows; i++ )
        {
            if( !LtoR ) j1 = i; else j0 = i+1;
            for( j = j0; j < j1; j++ )
                data[i*step + j] = data[j*step + i];
        }
    }
    else
        CV_Error( CV_StsUnsupportedFormat, "" );
}

2125

2126
cv::Mat cv::Mat::cross(InputArray _m) const
2127
{
2128
    Mat m = _m.getMat();
A
Andrey Kamaev 已提交
2129 2130
    int tp = type(), d = CV_MAT_DEPTH(tp);
    CV_Assert( dims <= 2 && m.dims <= 2 && size() == m.size() && tp == m.type() &&
2131
        ((rows == 3 && cols == 1) || (cols*channels() == 3 && rows == 1)));
A
Andrey Kamaev 已提交
2132
    Mat result(rows, cols, tp);
2133 2134 2135 2136 2137 2138 2139 2140 2141 2142 2143 2144 2145 2146 2147 2148 2149 2150 2151 2152 2153 2154 2155 2156 2157 2158 2159 2160

    if( d == CV_32F )
    {
        const float *a = (const float*)data, *b = (const float*)m.data;
        float* c = (float*)result.data;
        size_t lda = rows > 1 ? step/sizeof(a[0]) : 1;
        size_t ldb = rows > 1 ? m.step/sizeof(b[0]) : 1;

        c[0] = a[lda] * b[ldb*2] - a[lda*2] * b[ldb];
        c[1] = a[lda*2] * b[0] - a[0] * b[ldb*2];
        c[2] = a[0] * b[ldb] - a[lda] * b[0];
    }
    else if( d == CV_64F )
    {
        const double *a = (const double*)data, *b = (const double*)m.data;
        double* c = (double*)result.data;
        size_t lda = rows > 1 ? step/sizeof(a[0]) : 1;
        size_t ldb = rows > 1 ? m.step/sizeof(b[0]) : 1;

        c[0] = a[lda] * b[ldb*2] - a[lda*2] * b[ldb];
        c[1] = a[lda*2] * b[0] - a[0] * b[ldb*2];
        c[2] = a[0] * b[ldb] - a[lda] * b[0];
    }

    return result;
}


2161
////////////////////////////////////////// reduce ////////////////////////////////////////////
2162

2163 2164 2165
namespace cv
{

2166 2167 2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179 2180 2181 2182 2183 2184 2185
template<typename T, typename ST, class Op> static void
reduceR_( const Mat& srcmat, Mat& dstmat )
{
    typedef typename Op::rtype WT;
    Size size = srcmat.size();
    size.width *= srcmat.channels();
    AutoBuffer<WT> buffer(size.width);
    WT* buf = buffer;
    ST* dst = (ST*)dstmat.data;
    const T* src = (const T*)srcmat.data;
    size_t srcstep = srcmat.step/sizeof(src[0]);
    int i;
    Op op;

    for( i = 0; i < size.width; i++ )
        buf[i] = src[i];

    for( ; --size.height; )
    {
        src += srcstep;
V
Victoria Zhislina 已提交
2186
        i = 0;
2187
        #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
2188
        for(; i <= size.width - 4; i += 4 )
2189 2190 2191 2192 2193 2194 2195 2196 2197 2198
        {
            WT s0, s1;
            s0 = op(buf[i], (WT)src[i]);
            s1 = op(buf[i+1], (WT)src[i+1]);
            buf[i] = s0; buf[i+1] = s1;

            s0 = op(buf[i+2], (WT)src[i+2]);
            s1 = op(buf[i+3], (WT)src[i+3]);
            buf[i+2] = s0; buf[i+3] = s1;
        }
V
Victoria Zhislina 已提交
2199
        #endif
2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217 2218 2219 2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239
        for( ; i < size.width; i++ )
            buf[i] = op(buf[i], (WT)src[i]);
    }

    for( i = 0; i < size.width; i++ )
        dst[i] = (ST)buf[i];
}


template<typename T, typename ST, class Op> static void
reduceC_( const Mat& srcmat, Mat& dstmat )
{
    typedef typename Op::rtype WT;
    Size size = srcmat.size();
    int i, k, cn = srcmat.channels();
    size.width *= cn;
    Op op;

    for( int y = 0; y < size.height; y++ )
    {
        const T* src = (const T*)(srcmat.data + srcmat.step*y);
        ST* dst = (ST*)(dstmat.data + dstmat.step*y);
        if( size.width == cn )
            for( k = 0; k < cn; k++ )
                dst[k] = src[k];
        else
        {
            for( k = 0; k < cn; k++ )
            {
                WT a0 = src[k], a1 = src[k+cn];
                for( i = 2*cn; i <= size.width - 4*cn; i += 4*cn )
                {
                    a0 = op(a0, (WT)src[i+k]);
                    a1 = op(a1, (WT)src[i+k+cn]);
                    a0 = op(a0, (WT)src[i+k+cn*2]);
                    a1 = op(a1, (WT)src[i+k+cn*3]);
                }

                for( ; i < size.width; i += cn )
                {
2240
                    a0 = op(a0, (WT)src[i+k]);
2241 2242
                }
                a0 = op(a0, a1);
2243
              dst[k] = (ST)a0;
2244 2245
            }
        }
2246
    }
2247 2248 2249 2250
}

typedef void (*ReduceFunc)( const Mat& src, Mat& dst );

2251
}
2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298

#define reduceSumR8u32s  reduceR_<uchar, int,   OpAdd<int> >
#define reduceSumR8u32f  reduceR_<uchar, float, OpAdd<int> >
#define reduceSumR8u64f  reduceR_<uchar, double,OpAdd<int> >
#define reduceSumR16u32f reduceR_<ushort,float, OpAdd<float> >
#define reduceSumR16u64f reduceR_<ushort,double,OpAdd<double> >
#define reduceSumR16s32f reduceR_<short, float, OpAdd<float> >
#define reduceSumR16s64f reduceR_<short, double,OpAdd<double> >
#define reduceSumR32f32f reduceR_<float, float, OpAdd<float> >
#define reduceSumR32f64f reduceR_<float, double,OpAdd<double> >
#define reduceSumR64f64f reduceR_<double,double,OpAdd<double> >

#define reduceMaxR8u  reduceR_<uchar, uchar, OpMax<uchar> >
#define reduceMaxR16u reduceR_<ushort,ushort,OpMax<ushort> >
#define reduceMaxR16s reduceR_<short, short, OpMax<short> >
#define reduceMaxR32f reduceR_<float, float, OpMax<float> >
#define reduceMaxR64f reduceR_<double,double,OpMax<double> >

#define reduceMinR8u  reduceR_<uchar, uchar, OpMin<uchar> >
#define reduceMinR16u reduceR_<ushort,ushort,OpMin<ushort> >
#define reduceMinR16s reduceR_<short, short, OpMin<short> >
#define reduceMinR32f reduceR_<float, float, OpMin<float> >
#define reduceMinR64f reduceR_<double,double,OpMin<double> >

#define reduceSumC8u32s  reduceC_<uchar, int,   OpAdd<int> >
#define reduceSumC8u32f  reduceC_<uchar, float, OpAdd<int> >
#define reduceSumC8u64f  reduceC_<uchar, double,OpAdd<int> >
#define reduceSumC16u32f reduceC_<ushort,float, OpAdd<float> >
#define reduceSumC16u64f reduceC_<ushort,double,OpAdd<double> >
#define reduceSumC16s32f reduceC_<short, float, OpAdd<float> >
#define reduceSumC16s64f reduceC_<short, double,OpAdd<double> >
#define reduceSumC32f32f reduceC_<float, float, OpAdd<float> >
#define reduceSumC32f64f reduceC_<float, double,OpAdd<double> >
#define reduceSumC64f64f reduceC_<double,double,OpAdd<double> >

#define reduceMaxC8u  reduceC_<uchar, uchar, OpMax<uchar> >
#define reduceMaxC16u reduceC_<ushort,ushort,OpMax<ushort> >
#define reduceMaxC16s reduceC_<short, short, OpMax<short> >
#define reduceMaxC32f reduceC_<float, float, OpMax<float> >
#define reduceMaxC64f reduceC_<double,double,OpMax<double> >

#define reduceMinC8u  reduceC_<uchar, uchar, OpMin<uchar> >
#define reduceMinC16u reduceC_<ushort,ushort,OpMin<ushort> >
#define reduceMinC16s reduceC_<short, short, OpMin<short> >
#define reduceMinC32f reduceC_<float, float, OpMin<float> >
#define reduceMinC64f reduceC_<double,double,OpMin<double> >

2299
void cv::reduce(InputArray _src, OutputArray _dst, int dim, int op, int dtype)
2300
{
2301
    Mat src = _src.getMat();
V
Vadim Pisarevsky 已提交
2302
    CV_Assert( src.dims <= 2 );
2303
    int op0 = op;
2304
    int stype = src.type(), sdepth = src.depth(), cn = src.channels();
2305
    if( dtype < 0 )
2306
        dtype = _dst.fixedType() ? _dst.type() : stype;
2307 2308
    int ddepth = CV_MAT_DEPTH(dtype);

2309 2310 2311
    _dst.create(dim == 0 ? 1 : src.rows, dim == 0 ? src.cols : 1,
                CV_MAKETYPE(dtype >= 0 ? dtype : stype, cn));
    Mat dst = _dst.getMat(), temp = dst;
2312

2313
    CV_Assert( op == CV_REDUCE_SUM || op == CV_REDUCE_MAX ||
2314
               op == CV_REDUCE_MIN || op == CV_REDUCE_AVG );
2315 2316 2317 2318 2319 2320
    CV_Assert( src.channels() == dst.channels() );

    if( op == CV_REDUCE_AVG )
    {
        op = CV_REDUCE_SUM;
        if( sdepth < CV_32S && ddepth < CV_32S )
2321
        {
2322
            temp.create(dst.rows, dst.cols, CV_32SC(cn));
2323 2324
            ddepth = CV_32S;
        }
2325 2326 2327 2328 2329 2330 2331
    }

    ReduceFunc func = 0;
    if( dim == 0 )
    {
        if( op == CV_REDUCE_SUM )
        {
2332
            if(sdepth == CV_8U && ddepth == CV_32S)
2333
                func = GET_OPTIMIZED(reduceSumR8u32s);
2334
            else if(sdepth == CV_8U && ddepth == CV_32F)
2335
                func = reduceSumR8u32f;
2336
            else if(sdepth == CV_8U && ddepth == CV_64F)
2337
                func = reduceSumR8u64f;
2338
            else if(sdepth == CV_16U && ddepth == CV_32F)
2339
                func = reduceSumR16u32f;
2340
            else if(sdepth == CV_16U && ddepth == CV_64F)
2341
                func = reduceSumR16u64f;
2342
            else if(sdepth == CV_16S && ddepth == CV_32F)
2343
                func = reduceSumR16s32f;
2344
            else if(sdepth == CV_16S && ddepth == CV_64F)
2345 2346 2347
                func = reduceSumR16s64f;
            else if(sdepth == CV_32F && ddepth == CV_32F)
                func = GET_OPTIMIZED(reduceSumR32f32f);
2348
            else if(sdepth == CV_32F && ddepth == CV_64F)
2349
                func = reduceSumR32f64f;
2350
            else if(sdepth == CV_64F && ddepth == CV_64F)
2351
                func = reduceSumR64f64f;
2352 2353 2354
        }
        else if(op == CV_REDUCE_MAX)
        {
2355
            if(sdepth == CV_8U && ddepth == CV_8U)
2356 2357 2358
                func = GET_OPTIMIZED(reduceMaxR8u);
            else if(sdepth == CV_16U && ddepth == CV_16U)
                func = reduceMaxR16u;
2359
            else if(sdepth == CV_16S && ddepth == CV_16S)
2360
                func = reduceMaxR16s;
2361
            else if(sdepth == CV_32F && ddepth == CV_32F)
2362 2363 2364
                func = GET_OPTIMIZED(reduceMaxR32f);
            else if(sdepth == CV_64F && ddepth == CV_64F)
                func = reduceMaxR64f;
2365 2366 2367
        }
        else if(op == CV_REDUCE_MIN)
        {
2368
            if(sdepth == CV_8U && ddepth == CV_8U)
2369
                func = GET_OPTIMIZED(reduceMinR8u);
2370
            else if(sdepth == CV_16U && ddepth == CV_16U)
2371
                func = reduceMinR16u;
2372
            else if(sdepth == CV_16S && ddepth == CV_16S)
2373
                func = reduceMinR16s;
2374
            else if(sdepth == CV_32F && ddepth == CV_32F)
2375
                func = GET_OPTIMIZED(reduceMinR32f);
2376
            else if(sdepth == CV_64F && ddepth == CV_64F)
2377
                func = reduceMinR64f;
2378 2379 2380 2381 2382 2383
        }
    }
    else
    {
        if(op == CV_REDUCE_SUM)
        {
2384
            if(sdepth == CV_8U && ddepth == CV_32S)
2385
                func = GET_OPTIMIZED(reduceSumC8u32s);
2386
            else if(sdepth == CV_8U && ddepth == CV_32F)
2387
                func = reduceSumC8u32f;
2388
            else if(sdepth == CV_8U && ddepth == CV_64F)
2389
                func = reduceSumC8u64f;
2390
            else if(sdepth == CV_16U && ddepth == CV_32F)
2391
                func = reduceSumC16u32f;
2392
            else if(sdepth == CV_16U && ddepth == CV_64F)
2393
                func = reduceSumC16u64f;
2394
            else if(sdepth == CV_16S && ddepth == CV_32F)
2395
                func = reduceSumC16s32f;
2396
            else if(sdepth == CV_16S && ddepth == CV_64F)
2397 2398 2399
                func = reduceSumC16s64f;
            else if(sdepth == CV_32F && ddepth == CV_32F)
                func = GET_OPTIMIZED(reduceSumC32f32f);
2400
            else if(sdepth == CV_32F && ddepth == CV_64F)
2401
                func = reduceSumC32f64f;
2402
            else if(sdepth == CV_64F && ddepth == CV_64F)
2403
                func = reduceSumC64f64f;
2404 2405 2406
        }
        else if(op == CV_REDUCE_MAX)
        {
2407
            if(sdepth == CV_8U && ddepth == CV_8U)
2408 2409 2410
                func = GET_OPTIMIZED(reduceMaxC8u);
            else if(sdepth == CV_16U && ddepth == CV_16U)
                func = reduceMaxC16u;
2411
            else if(sdepth == CV_16S && ddepth == CV_16S)
2412
                func = reduceMaxC16s;
2413
            else if(sdepth == CV_32F && ddepth == CV_32F)
2414
                func = GET_OPTIMIZED(reduceMaxC32f);
2415
            else if(sdepth == CV_64F && ddepth == CV_64F)
2416
                func = reduceMaxC64f;
2417 2418 2419
        }
        else if(op == CV_REDUCE_MIN)
        {
2420
            if(sdepth == CV_8U && ddepth == CV_8U)
2421
                func = GET_OPTIMIZED(reduceMinC8u);
2422
            else if(sdepth == CV_16U && ddepth == CV_16U)
2423
                func = reduceMinC16u;
2424
            else if(sdepth == CV_16S && ddepth == CV_16S)
2425
                func = reduceMinC16s;
2426
            else if(sdepth == CV_32F && ddepth == CV_32F)
2427
                func = GET_OPTIMIZED(reduceMinC32f);
2428
            else if(sdepth == CV_64F && ddepth == CV_64F)
2429
                func = reduceMinC64f;
2430 2431 2432 2433 2434
        }
    }

    if( !func )
        CV_Error( CV_StsUnsupportedFormat,
2435
                  "Unsupported combination of input and output array formats" );
2436 2437 2438

    func( src, temp );

2439
    if( op0 == CV_REDUCE_AVG )
2440
        temp.convertTo(dst, dst.type(), 1./(dim == 0 ? src.rows : src.cols));
2441
}
2442 2443


2444
//////////////////////////////////////// sort ///////////////////////////////////////////
2445

2446 2447 2448
namespace cv
{

2449 2450 2451 2452 2453 2454 2455 2456
template<typename T> static void sort_( const Mat& src, Mat& dst, int flags )
{
    AutoBuffer<T> buf;
    T* bptr;
    int i, j, n, len;
    bool sortRows = (flags & 1) == CV_SORT_EVERY_ROW;
    bool inplace = src.data == dst.data;
    bool sortDescending = (flags & CV_SORT_DESCENDING) != 0;
2457

2458 2459 2460 2461 2462 2463 2464 2465 2466 2467 2468 2469 2470 2471 2472 2473 2474 2475 2476 2477 2478 2479 2480 2481 2482 2483 2484 2485
    if( sortRows )
        n = src.rows, len = src.cols;
    else
    {
        n = src.cols, len = src.rows;
        buf.allocate(len);
    }
    bptr = (T*)buf;

    for( i = 0; i < n; i++ )
    {
        T* ptr = bptr;
        if( sortRows )
        {
            T* dptr = (T*)(dst.data + dst.step*i);
            if( !inplace )
            {
                const T* sptr = (const T*)(src.data + src.step*i);
                for( j = 0; j < len; j++ )
                    dptr[j] = sptr[j];
            }
            ptr = dptr;
        }
        else
        {
            for( j = 0; j < len; j++ )
                ptr[j] = ((const T*)(src.data + src.step*j))[i];
        }
2486
        std::sort( ptr, ptr + len );
2487 2488 2489 2490 2491 2492 2493 2494 2495
        if( sortDescending )
            for( j = 0; j < len/2; j++ )
                std::swap(ptr[j], ptr[len-1-j]);
        if( !sortRows )
            for( j = 0; j < len; j++ )
                ((T*)(dst.data + dst.step*j))[i] = ptr[j];
    }
}

2496 2497 2498 2499 2500 2501 2502 2503 2504
template<typename _Tp> class LessThanIdx
{
public:
    LessThanIdx( const _Tp* _arr ) : arr(_arr) {}
    bool operator()(int a, int b) const { return arr[a] < arr[b]; }
    const _Tp* arr;
};


2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516

template<typename T> static void sortIdx_( const Mat& src, Mat& dst, int flags )
{
    AutoBuffer<T> buf;
    AutoBuffer<int> ibuf;
    T* bptr;
    int* _iptr;
    int i, j, n, len;
    bool sortRows = (flags & 1) == CV_SORT_EVERY_ROW;
    bool sortDescending = (flags & CV_SORT_DESCENDING) != 0;

    CV_Assert( src.data != dst.data );
2517

2518 2519 2520 2521 2522 2523 2524 2525 2526 2527 2528 2529 2530 2531 2532 2533 2534 2535 2536 2537 2538 2539 2540 2541 2542 2543 2544 2545 2546 2547 2548 2549 2550 2551 2552 2553 2554 2555 2556 2557
    if( sortRows )
        n = src.rows, len = src.cols;
    else
    {
        n = src.cols, len = src.rows;
        buf.allocate(len);
        ibuf.allocate(len);
    }
    bptr = (T*)buf;
    _iptr = (int*)ibuf;

    for( i = 0; i < n; i++ )
    {
        T* ptr = bptr;
        int* iptr = _iptr;

        if( sortRows )
        {
            ptr = (T*)(src.data + src.step*i);
            iptr = (int*)(dst.data + dst.step*i);
        }
        else
        {
            for( j = 0; j < len; j++ )
                ptr[j] = ((const T*)(src.data + src.step*j))[i];
        }
        for( j = 0; j < len; j++ )
            iptr[j] = j;
        std::sort( iptr, iptr + len, LessThanIdx<T>(ptr) );
        if( sortDescending )
            for( j = 0; j < len/2; j++ )
                std::swap(iptr[j], iptr[len-1-j]);
        if( !sortRows )
            for( j = 0; j < len; j++ )
                ((int*)(dst.data + dst.step*j))[i] = iptr[j];
    }
}

typedef void (*SortFunc)(const Mat& src, Mat& dst, int flags);

2558
}
2559

2560
void cv::sort( InputArray _src, OutputArray _dst, int flags )
2561 2562 2563 2564 2565 2566
{
    static SortFunc tab[] =
    {
        sort_<uchar>, sort_<schar>, sort_<ushort>, sort_<short>,
        sort_<int>, sort_<float>, sort_<double>, 0
    };
2567
    Mat src = _src.getMat();
2568
    SortFunc func = tab[src.depth()];
V
Vadim Pisarevsky 已提交
2569
    CV_Assert( src.dims <= 2 && src.channels() == 1 && func != 0 );
2570 2571
    _dst.create( src.size(), src.type() );
    Mat dst = _dst.getMat();
2572 2573 2574
    func( src, dst, flags );
}

2575
void cv::sortIdx( InputArray _src, OutputArray _dst, int flags )
2576 2577 2578 2579 2580 2581
{
    static SortFunc tab[] =
    {
        sortIdx_<uchar>, sortIdx_<schar>, sortIdx_<ushort>, sortIdx_<short>,
        sortIdx_<int>, sortIdx_<float>, sortIdx_<double>, 0
    };
2582
    Mat src = _src.getMat();
2583
    SortFunc func = tab[src.depth()];
V
Vadim Pisarevsky 已提交
2584
    CV_Assert( src.dims <= 2 && src.channels() == 1 && func != 0 );
2585

2586
    Mat dst = _dst.getMat();
2587
    if( dst.data == src.data )
2588 2589 2590
        _dst.release();
    _dst.create( src.size(), CV_32S );
    dst = _dst.getMat();
2591 2592
    func( src, dst, flags );
}
2593 2594


2595
////////////////////////////////////////// kmeans ////////////////////////////////////////////
2596 2597 2598 2599

namespace cv
{

2600
static void generateRandomCenter(const std::vector<Vec2f>& box, float* center, RNG& rng)
2601 2602 2603 2604 2605 2606 2607
{
    size_t j, dims = box.size();
    float margin = 1.f/dims;
    for( j = 0; j < dims; j++ )
        center[j] = ((float)rng*(1.f+margin*2.f)-margin)*(box[j][1] - box[j][0]) + box[j][0];
}

2608
class KMeansPPDistanceComputer : public ParallelLoopBody
2609 2610 2611 2612 2613 2614 2615 2616 2617 2618 2619 2620 2621 2622 2623
{
public:
    KMeansPPDistanceComputer( float *_tdist2,
                              const float *_data,
                              const float *_dist,
                              int _dims,
                              size_t _step,
                              size_t _stepci )
        : tdist2(_tdist2),
          data(_data),
          dist(_dist),
          dims(_dims),
          step(_step),
          stepci(_stepci) { }

2624
    void operator()( const cv::Range& range ) const
2625
    {
2626 2627
        const int begin = range.start;
        const int end = range.end;
2628 2629 2630 2631 2632 2633 2634 2635

        for ( int i = begin; i<end; i++ )
        {
            tdist2[i] = std::min(normL2Sqr_(data + step*i, data + stepci, dims), dist[i]);
        }
    }

private:
D
Daniil Osokin 已提交
2636 2637
    KMeansPPDistanceComputer& operator=(const KMeansPPDistanceComputer&); // to quiet MSVC

2638 2639 2640 2641 2642 2643 2644
    float *tdist2;
    const float *data;
    const float *dist;
    const int dims;
    const size_t step;
    const size_t stepci;
};
2645 2646 2647 2648 2649 2650 2651 2652 2653 2654

/*
k-means center initialization using the following algorithm:
Arthur & Vassilvitskii (2007) k-means++: The Advantages of Careful Seeding
*/
static void generateCentersPP(const Mat& _data, Mat& _out_centers,
                              int K, RNG& rng, int trials)
{
    int i, j, k, dims = _data.cols, N = _data.rows;
    const float* data = _data.ptr<float>(0);
2655
    size_t step = _data.step/sizeof(data[0]);
2656
    std::vector<int> _centers(K);
2657
    int* centers = &_centers[0];
2658
    std::vector<float> _dist(N*3);
2659 2660 2661 2662 2663 2664 2665
    float* dist = &_dist[0], *tdist = dist + N, *tdist2 = tdist + N;
    double sum0 = 0;

    centers[0] = (unsigned)rng % N;

    for( i = 0; i < N; i++ )
    {
2666
        dist[i] = normL2Sqr_(data + step*i, data + step*centers[0], dims);
2667 2668
        sum0 += dist[i];
    }
2669

2670 2671 2672 2673 2674 2675 2676 2677 2678 2679 2680 2681
    for( k = 1; k < K; k++ )
    {
        double bestSum = DBL_MAX;
        int bestCenter = -1;

        for( j = 0; j < trials; j++ )
        {
            double p = (double)rng*sum0, s = 0;
            for( i = 0; i < N-1; i++ )
                if( (p -= dist[i]) <= 0 )
                    break;
            int ci = i;
2682

2683
            parallel_for_(Range(0, N),
2684
                         KMeansPPDistanceComputer(tdist2, data, dist, dims, step, step*ci));
2685 2686 2687 2688
            for( i = 0; i < N; i++ )
            {
                s += tdist2[i];
            }
2689

2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710
            if( s < bestSum )
            {
                bestSum = s;
                bestCenter = ci;
                std::swap(tdist, tdist2);
            }
        }
        centers[k] = bestCenter;
        sum0 = bestSum;
        std::swap(dist, tdist);
    }

    for( k = 0; k < K; k++ )
    {
        const float* src = data + step*centers[k];
        float* dst = _out_centers.ptr<float>(k);
        for( j = 0; j < dims; j++ )
            dst[j] = src[j];
    }
}

2711
class KMeansDistanceComputer : public ParallelLoopBody
2712 2713 2714 2715 2716 2717 2718 2719 2720 2721 2722 2723 2724
{
public:
    KMeansDistanceComputer( double *_distances,
                            int *_labels,
                            const Mat& _data,
                            const Mat& _centers )
        : distances(_distances),
          labels(_labels),
          data(_data),
          centers(_centers)
    {
    }

2725
    void operator()( const Range& range ) const
2726
    {
2727 2728
        const int begin = range.start;
        const int end = range.end;
2729 2730 2731 2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743 2744 2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756
        const int K = centers.rows;
        const int dims = centers.cols;

        const float *sample;
        for( int i = begin; i<end; ++i)
        {
            sample = data.ptr<float>(i);
            int k_best = 0;
            double min_dist = DBL_MAX;

            for( int k = 0; k < K; k++ )
            {
                const float* center = centers.ptr<float>(k);
                const double dist = normL2Sqr_(sample, center, dims);

                if( min_dist > dist )
                {
                    min_dist = dist;
                    k_best = k;
                }
            }

            distances[i] = min_dist;
            labels[i] = k_best;
        }
    }

private:
2757 2758
    KMeansDistanceComputer& operator=(const KMeansDistanceComputer&); // to quiet MSVC

2759 2760 2761 2762 2763 2764
    double *distances;
    int *labels;
    const Mat& data;
    const Mat& centers;
};

2765
}
2766

2767
double cv::kmeans( InputArray _data, int K,
2768 2769 2770
                   InputOutputArray _bestLabels,
                   TermCriteria criteria, int attempts,
                   int flags, OutputArray _centers )
2771 2772
{
    const int SPP_TRIALS = 3;
2773
    Mat data = _data.getMat();
2774 2775 2776
    bool isrow = data.rows == 1 && data.channels() > 1;
    int N = !isrow ? data.rows : data.cols;
    int dims = (!isrow ? data.cols : 1)*data.channels();
2777 2778 2779
    int type = data.depth();

    attempts = std::max(attempts, 1);
V
Vadim Pisarevsky 已提交
2780
    CV_Assert( data.dims <= 2 && type == CV_32F && K > 0 );
2781
    CV_Assert( N >= K );
2782

2783
    _bestLabels.create(N, 1, CV_32S, -1, true);
2784

2785
    Mat _labels, best_labels = _bestLabels.getMat();
2786 2787 2788 2789 2790 2791 2792 2793 2794 2795 2796 2797 2798 2799 2800 2801 2802 2803 2804
    if( flags & CV_KMEANS_USE_INITIAL_LABELS )
    {
        CV_Assert( (best_labels.cols == 1 || best_labels.rows == 1) &&
                  best_labels.cols*best_labels.rows == N &&
                  best_labels.type() == CV_32S &&
                  best_labels.isContinuous());
        best_labels.copyTo(_labels);
    }
    else
    {
        if( !((best_labels.cols == 1 || best_labels.rows == 1) &&
             best_labels.cols*best_labels.rows == N &&
            best_labels.type() == CV_32S &&
            best_labels.isContinuous()))
            best_labels.create(N, 1, CV_32S);
        _labels.create(best_labels.size(), best_labels.type());
    }
    int* labels = _labels.ptr<int>();

2805
    Mat centers(K, dims, type), old_centers(K, dims, type), temp(1, dims, type);
2806 2807
    std::vector<int> counters(K);
    std::vector<Vec2f> _box(dims);
2808 2809 2810 2811 2812 2813 2814 2815 2816 2817 2818 2819 2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847
    Vec2f* box = &_box[0];
    double best_compactness = DBL_MAX, compactness = 0;
    RNG& rng = theRNG();
    int a, iter, i, j, k;

    if( criteria.type & TermCriteria::EPS )
        criteria.epsilon = std::max(criteria.epsilon, 0.);
    else
        criteria.epsilon = FLT_EPSILON;
    criteria.epsilon *= criteria.epsilon;

    if( criteria.type & TermCriteria::COUNT )
        criteria.maxCount = std::min(std::max(criteria.maxCount, 2), 100);
    else
        criteria.maxCount = 100;

    if( K == 1 )
    {
        attempts = 1;
        criteria.maxCount = 2;
    }

    const float* sample = data.ptr<float>(0);
    for( j = 0; j < dims; j++ )
        box[j] = Vec2f(sample[j], sample[j]);

    for( i = 1; i < N; i++ )
    {
        sample = data.ptr<float>(i);
        for( j = 0; j < dims; j++ )
        {
            float v = sample[j];
            box[j][0] = std::min(box[j][0], v);
            box[j][1] = std::max(box[j][1], v);
        }
    }

    for( a = 0; a < attempts; a++ )
    {
        double max_center_shift = DBL_MAX;
2848
        for( iter = 0;; )
2849 2850 2851 2852 2853 2854 2855 2856 2857 2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868
        {
            swap(centers, old_centers);

            if( iter == 0 && (a > 0 || !(flags & KMEANS_USE_INITIAL_LABELS)) )
            {
                if( flags & KMEANS_PP_CENTERS )
                    generateCentersPP(data, centers, K, rng, SPP_TRIALS);
                else
                {
                    for( k = 0; k < K; k++ )
                        generateRandomCenter(_box, centers.ptr<float>(k), rng);
                }
            }
            else
            {
                if( iter == 0 && a == 0 && (flags & KMEANS_USE_INITIAL_LABELS) )
                {
                    for( i = 0; i < N; i++ )
                        CV_Assert( (unsigned)labels[i] < (unsigned)K );
                }
2869

2870 2871 2872 2873 2874 2875 2876 2877 2878 2879
                // compute centers
                centers = Scalar(0);
                for( k = 0; k < K; k++ )
                    counters[k] = 0;

                for( i = 0; i < N; i++ )
                {
                    sample = data.ptr<float>(i);
                    k = labels[i];
                    float* center = centers.ptr<float>(k);
2880 2881
                    j=0;
                    #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
2882
                    for(; j <= dims - 4; j += 4 )
2883 2884 2885 2886 2887 2888 2889 2890 2891 2892 2893 2894 2895
                    {
                        float t0 = center[j] + sample[j];
                        float t1 = center[j+1] + sample[j+1];

                        center[j] = t0;
                        center[j+1] = t1;

                        t0 = center[j+2] + sample[j+2];
                        t1 = center[j+3] + sample[j+3];

                        center[j+2] = t0;
                        center[j+3] = t1;
                    }
V
Victoria Zhislina 已提交
2896
                    #endif
2897 2898 2899 2900 2901 2902 2903
                    for( ; j < dims; j++ )
                        center[j] += sample[j];
                    counters[k]++;
                }

                if( iter > 0 )
                    max_center_shift = 0;
2904

2905 2906 2907
                for( k = 0; k < K; k++ )
                {
                    if( counters[k] != 0 )
2908 2909 2910 2911 2912 2913 2914
                        continue;

                    // if some cluster appeared to be empty then:
                    //   1. find the biggest cluster
                    //   2. find the farthest from the center point in the biggest cluster
                    //   3. exclude the farthest point from the biggest cluster and form a new 1-point cluster.
                    int max_k = 0;
M
Maria Dimashova 已提交
2915
                    for( int k1 = 1; k1 < K; k1++ )
2916 2917 2918 2919
                    {
                        if( counters[max_k] < counters[k1] )
                            max_k = k1;
                    }
2920 2921

                    double max_dist = 0;
2922 2923 2924
                    int farthest_i = -1;
                    float* new_center = centers.ptr<float>(k);
                    float* old_center = centers.ptr<float>(max_k);
2925 2926 2927 2928
                    float* _old_center = temp.ptr<float>(); // normalized
                    float scale = 1.f/counters[max_k];
                    for( j = 0; j < dims; j++ )
                        _old_center[j] = old_center[j]*scale;
2929

2930 2931 2932 2933 2934
                    for( i = 0; i < N; i++ )
                    {
                        if( labels[i] != max_k )
                            continue;
                        sample = data.ptr<float>(i);
2935
                        double dist = normL2Sqr_(sample, _old_center, dims);
2936

2937 2938 2939 2940 2941 2942
                        if( max_dist <= dist )
                        {
                            max_dist = dist;
                            farthest_i = i;
                        }
                    }
2943

2944 2945
                    counters[max_k]--;
                    counters[k]++;
2946
                    labels[farthest_i] = k;
2947
                    sample = data.ptr<float>(farthest_i);
2948

2949
                    for( j = 0; j < dims; j++ )
2950
                    {
2951 2952
                        old_center[j] -= sample[j];
                        new_center[j] += sample[j];
2953
                    }
2954 2955 2956 2957 2958 2959 2960 2961 2962 2963
                }

                for( k = 0; k < K; k++ )
                {
                    float* center = centers.ptr<float>(k);
                    CV_Assert( counters[k] != 0 );

                    float scale = 1.f/counters[k];
                    for( j = 0; j < dims; j++ )
                        center[j] *= scale;
2964

2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977
                    if( iter > 0 )
                    {
                        double dist = 0;
                        const float* old_center = old_centers.ptr<float>(k);
                        for( j = 0; j < dims; j++ )
                        {
                            double t = center[j] - old_center[j];
                            dist += t*t;
                        }
                        max_center_shift = std::max(max_center_shift, dist);
                    }
                }
            }
2978

2979 2980
            if( ++iter == MAX(criteria.maxCount, 2) || max_center_shift <= criteria.epsilon )
                break;
2981 2982

            // assign labels
2983 2984
            Mat dists(1, N, CV_64F);
            double* dist = dists.ptr<double>(0);
2985
            parallel_for_(Range(0, N),
2986
                         KMeansDistanceComputer(dist, labels, data, centers));
2987 2988 2989
            compactness = 0;
            for( i = 0; i < N; i++ )
            {
2990
                compactness += dist[i];
2991 2992 2993 2994 2995 2996
            }
        }

        if( compactness < best_compactness )
        {
            best_compactness = compactness;
2997 2998
            if( _centers.needed() )
                centers.copyTo(_centers);
2999 3000 3001 3002 3003 3004 3005 3006 3007 3008 3009 3010 3011 3012 3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030
            _labels.copyTo(best_labels);
        }
    }

    return best_compactness;
}


CV_IMPL void cvSetIdentity( CvArr* arr, CvScalar value )
{
    cv::Mat m = cv::cvarrToMat(arr);
    cv::setIdentity(m, value);
}


CV_IMPL CvScalar cvTrace( const CvArr* arr )
{
    return cv::trace(cv::cvarrToMat(arr));
}


CV_IMPL void cvTranspose( const CvArr* srcarr, CvArr* dstarr )
{
    cv::Mat src = cv::cvarrToMat(srcarr), dst = cv::cvarrToMat(dstarr);

    CV_Assert( src.rows == dst.cols && src.cols == dst.rows && src.type() == dst.type() );
    transpose( src, dst );
}


CV_IMPL void cvCompleteSymm( CvMat* matrix, int LtoR )
{
A
Andrey Kamaev 已提交
3031
    cv::Mat m = cv::cvarrToMat(matrix);
3032 3033 3034 3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048
    cv::completeSymm( m, LtoR != 0 );
}


CV_IMPL void cvCrossProduct( const CvArr* srcAarr, const CvArr* srcBarr, CvArr* dstarr )
{
    cv::Mat srcA = cv::cvarrToMat(srcAarr), dst = cv::cvarrToMat(dstarr);

    CV_Assert( srcA.size() == dst.size() && srcA.type() == dst.type() );
    srcA.cross(cv::cvarrToMat(srcBarr)).copyTo(dst);
}


CV_IMPL void
cvReduce( const CvArr* srcarr, CvArr* dstarr, int dim, int op )
{
    cv::Mat src = cv::cvarrToMat(srcarr), dst = cv::cvarrToMat(dstarr);
3049

3050 3051 3052 3053 3054 3055 3056 3057 3058
    if( dim < 0 )
        dim = src.rows > dst.rows ? 0 : src.cols > dst.cols ? 1 : dst.cols == 1;

    if( dim > 1 )
        CV_Error( CV_StsOutOfRange, "The reduced dimensionality index is out of range" );

    if( (dim == 0 && (dst.cols != src.cols || dst.rows != 1)) ||
        (dim == 1 && (dst.rows != src.rows || dst.cols != 1)) )
        CV_Error( CV_StsBadSize, "The output array size is incorrect" );
3059

3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070
    if( src.channels() != dst.channels() )
        CV_Error( CV_StsUnmatchedFormats, "Input and output arrays must have the same number of channels" );

    cv::reduce(src, dst, dim, op, dst.type());
}


CV_IMPL CvArr*
cvRange( CvArr* arr, double start, double end )
{
    int ok = 0;
3071

3072 3073 3074 3075 3076 3077
    CvMat stub, *mat = (CvMat*)arr;
    double delta;
    int type, step;
    double val = start;
    int i, j;
    int rows, cols;
3078

3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095 3096 3097 3098 3099 3100 3101 3102 3103 3104 3105 3106 3107 3108 3109 3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122 3123 3124 3125 3126 3127 3128 3129 3130 3131 3132
    if( !CV_IS_MAT(mat) )
        mat = cvGetMat( mat, &stub);

    rows = mat->rows;
    cols = mat->cols;
    type = CV_MAT_TYPE(mat->type);
    delta = (end-start)/(rows*cols);

    if( CV_IS_MAT_CONT(mat->type) )
    {
        cols *= rows;
        rows = 1;
        step = 1;
    }
    else
        step = mat->step / CV_ELEM_SIZE(type);

    if( type == CV_32SC1 )
    {
        int* idata = mat->data.i;
        int ival = cvRound(val), idelta = cvRound(delta);

        if( fabs(val - ival) < DBL_EPSILON &&
            fabs(delta - idelta) < DBL_EPSILON )
        {
            for( i = 0; i < rows; i++, idata += step )
                for( j = 0; j < cols; j++, ival += idelta )
                    idata[j] = ival;
        }
        else
        {
            for( i = 0; i < rows; i++, idata += step )
                for( j = 0; j < cols; j++, val += delta )
                    idata[j] = cvRound(val);
        }
    }
    else if( type == CV_32FC1 )
    {
        float* fdata = mat->data.fl;
        for( i = 0; i < rows; i++, fdata += step )
            for( j = 0; j < cols; j++, val += delta )
                fdata[j] = (float)val;
    }
    else
        CV_Error( CV_StsUnsupportedFormat, "The function only supports 32sC1 and 32fC1 datatypes" );

    ok = 1;
    return ok ? arr : 0;
}


CV_IMPL void
cvSort( const CvArr* _src, CvArr* _dst, CvArr* _idx, int flags )
{
A
Andrey Kamaev 已提交
3133
    cv::Mat src = cv::cvarrToMat(_src);
3134

3135 3136 3137 3138 3139 3140 3141 3142 3143 3144 3145 3146 3147 3148 3149 3150 3151 3152 3153 3154 3155 3156 3157 3158 3159
    if( _idx )
    {
        cv::Mat idx0 = cv::cvarrToMat(_idx), idx = idx0;
        CV_Assert( src.size() == idx.size() && idx.type() == CV_32S && src.data != idx.data );
        cv::sortIdx( src, idx, flags );
        CV_Assert( idx0.data == idx.data );
    }

    if( _dst )
    {
        cv::Mat dst0 = cv::cvarrToMat(_dst), dst = dst0;
        CV_Assert( src.size() == dst.size() && src.type() == dst.type() );
        cv::sort( src, dst, flags );
        CV_Assert( dst0.data == dst.data );
    }
}


CV_IMPL int
cvKMeans2( const CvArr* _samples, int cluster_count, CvArr* _labels,
           CvTermCriteria termcrit, int attempts, CvRNG*,
           int flags, CvArr* _centers, double* _compactness )
{
    cv::Mat data = cv::cvarrToMat(_samples), labels = cv::cvarrToMat(_labels), centers;
    if( _centers )
3160
    {
3161
        centers = cv::cvarrToMat(_centers);
M
Maria Dimashova 已提交
3162

3163
        centers = centers.reshape(1);
M
Maria Dimashova 已提交
3164 3165 3166 3167 3168 3169
        data = data.reshape(1);

        CV_Assert( !centers.empty() );
        CV_Assert( centers.rows == cluster_count );
        CV_Assert( centers.cols == data.cols );
        CV_Assert( centers.depth() == data.depth() );
3170
    }
3171 3172 3173
    CV_Assert( labels.isContinuous() && labels.type() == CV_32S &&
        (labels.cols == 1 || labels.rows == 1) &&
        labels.cols + labels.rows - 1 == data.rows );
3174

3175
    double compactness = cv::kmeans(data, cluster_count, labels, termcrit, attempts,
3176
                                    flags, _centers ? cv::_OutputArray(centers) : cv::_OutputArray() );
3177 3178 3179 3180 3181 3182 3183 3184 3185 3186
    if( _compactness )
        *_compactness = compactness;
    return 1;
}

///////////////////////////// n-dimensional matrices ////////////////////////////

namespace cv
{

3187
Mat Mat::reshape(int _cn, int _newndims, const int* _newsz) const
3188
{
3189 3190 3191 3192 3193 3194 3195 3196
    if(_newndims == dims)
    {
        if(_newsz == 0)
            return reshape(_cn);
        if(_newndims == 2)
            return reshape(_cn, _newsz[0]);
    }

V
Vadim Pisarevsky 已提交
3197 3198 3199 3200
    CV_Error(CV_StsNotImplemented, "");
    // TBD
    return Mat();
}
3201

V
Vadim Pisarevsky 已提交
3202
NAryMatIterator::NAryMatIterator()
3203
    : arrays(0), planes(0), ptrs(0), narrays(0), nplanes(0), size(0), iterdepth(0), idx(0)
V
Vadim Pisarevsky 已提交
3204 3205
{
}
3206

V
Vadim Pisarevsky 已提交
3207
NAryMatIterator::NAryMatIterator(const Mat** _arrays, Mat* _planes, int _narrays)
3208 3209 3210
: arrays(0), planes(0), ptrs(0), narrays(0), nplanes(0), size(0), iterdepth(0), idx(0)
{
    init(_arrays, _planes, 0, _narrays);
3211 3212
}

3213 3214
NAryMatIterator::NAryMatIterator(const Mat** _arrays, uchar** _ptrs, int _narrays)
    : arrays(0), planes(0), ptrs(0), narrays(0), nplanes(0), size(0), iterdepth(0), idx(0)
V
Vadim Pisarevsky 已提交
3215
{
3216
    init(_arrays, 0, _ptrs, _narrays);
V
Vadim Pisarevsky 已提交
3217
}
3218

3219
void NAryMatIterator::init(const Mat** _arrays, Mat* _planes, uchar** _ptrs, int _narrays)
V
Vadim Pisarevsky 已提交
3220
{
3221 3222
    CV_Assert( _arrays && (_ptrs || _planes) );
    int i, j, d1=0, i0 = -1, d = -1;
3223

V
Vadim Pisarevsky 已提交
3224
    arrays = _arrays;
3225
    ptrs = _ptrs;
V
Vadim Pisarevsky 已提交
3226 3227 3228
    planes = _planes;
    narrays = _narrays;
    nplanes = 0;
3229
    size = 0;
3230

V
Vadim Pisarevsky 已提交
3231
    if( narrays < 0 )
3232
    {
V
Vadim Pisarevsky 已提交
3233 3234 3235 3236
        for( i = 0; _arrays[i] != 0; i++ )
            ;
        narrays = i;
        CV_Assert(narrays <= 1000);
3237
    }
V
Vadim Pisarevsky 已提交
3238 3239 3240 3241

    iterdepth = 0;

    for( i = 0; i < narrays; i++ )
3242
    {
V
Vadim Pisarevsky 已提交
3243 3244
        CV_Assert(arrays[i] != 0);
        const Mat& A = *arrays[i];
3245 3246
        if( ptrs )
            ptrs[i] = A.data;
3247

3248 3249
        if( !A.data )
            continue;
3250

V
Vadim Pisarevsky 已提交
3251
        if( i0 < 0 )
3252
        {
V
Vadim Pisarevsky 已提交
3253 3254
            i0 = i;
            d = A.dims;
3255

V
Vadim Pisarevsky 已提交
3256 3257 3258 3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271
            // find the first dimensionality which is different from 1;
            // in any of the arrays the first "d1" step do not affect the continuity
            for( d1 = 0; d1 < d; d1++ )
                if( A.size[d1] > 1 )
                    break;
        }
        else
            CV_Assert( A.size == arrays[i0]->size );

        if( !A.isContinuous() )
        {
            CV_Assert( A.step[d-1] == A.elemSize() );
            for( j = d-1; j > d1; j-- )
                if( A.step[j]*A.size[j] < A.step[j-1] )
                    break;
            iterdepth = std::max(iterdepth, j);
3272 3273
        }
    }
V
Vadim Pisarevsky 已提交
3274 3275

    if( i0 >= 0 )
3276
    {
3277
        size = arrays[i0]->size[d-1];
V
Vadim Pisarevsky 已提交
3278 3279
        for( j = d-1; j > iterdepth; j-- )
        {
3280
            int64 total1 = (int64)size*arrays[i0]->size[j-1];
V
Vadim Pisarevsky 已提交
3281 3282
            if( total1 != (int)total1 )
                break;
3283
            size = (int)total1;
V
Vadim Pisarevsky 已提交
3284 3285 3286 3287 3288
        }

        iterdepth = j;
        if( iterdepth == d1 )
            iterdepth = 0;
3289

V
Vadim Pisarevsky 已提交
3290 3291 3292
        nplanes = 1;
        for( j = iterdepth-1; j >= 0; j-- )
            nplanes *= arrays[i0]->size[j];
3293
    }
V
Vadim Pisarevsky 已提交
3294
    else
3295
        iterdepth = 0;
3296

3297
    idx = 0;
3298

3299 3300
    if( !planes )
        return;
3301

V
Vadim Pisarevsky 已提交
3302
    for( i = 0; i < narrays; i++ )
3303
    {
3304 3305
        CV_Assert(arrays[i] != 0);
        const Mat& A = *arrays[i];
3306

3307
        if( !A.data )
V
Vadim Pisarevsky 已提交
3308 3309 3310 3311
        {
            planes[i] = Mat();
            continue;
        }
3312 3313

        planes[i] = Mat(1, (int)size, A.type(), A.data);
3314 3315 3316
    }
}

V
Vadim Pisarevsky 已提交
3317 3318

NAryMatIterator& NAryMatIterator::operator ++()
3319 3320 3321 3322
{
    if( idx >= nplanes-1 )
        return *this;
    ++idx;
3323

3324
    if( iterdepth == 1 )
3325
    {
3326 3327 3328 3329 3330 3331 3332 3333 3334 3335
        if( ptrs )
        {
            for( int i = 0; i < narrays; i++ )
            {
                if( !ptrs[i] )
                    continue;
                ptrs[i] = arrays[i]->data + arrays[i]->step[0]*idx;
            }
        }
        if( planes )
3336
        {
3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351
            for( int i = 0; i < narrays; i++ )
            {
                if( !planes[i].data )
                    continue;
                planes[i].data = arrays[i]->data + arrays[i]->step[0]*idx;
            }
        }
    }
    else
    {
        for( int i = 0; i < narrays; i++ )
        {
            const Mat& A = *arrays[i];
            if( !A.data )
                continue;
3352
            int _idx = (int)idx;
3353 3354 3355 3356 3357 3358 3359 3360 3361 3362 3363
            uchar* data = A.data;
            for( int j = iterdepth-1; j >= 0 && _idx > 0; j-- )
            {
                int szi = A.size[j], t = _idx/szi;
                data += (_idx - t * szi)*A.step[j];
                _idx = t;
            }
            if( ptrs )
                ptrs[i] = data;
            if( planes )
                planes[i].data = data;
3364 3365
        }
    }
3366

3367 3368 3369
    return *this;
}

V
Vadim Pisarevsky 已提交
3370
NAryMatIterator NAryMatIterator::operator ++(int)
3371
{
V
Vadim Pisarevsky 已提交
3372
    NAryMatIterator it = *this;
3373 3374 3375 3376
    ++*this;
    return it;
}

V
Vadim Pisarevsky 已提交
3377 3378 3379
///////////////////////////////////////////////////////////////////////////
//                              MatConstIterator                         //
///////////////////////////////////////////////////////////////////////////
3380

V
Vadim Pisarevsky 已提交
3381
Point MatConstIterator::pos() const
3382
{
V
Vadim Pisarevsky 已提交
3383 3384 3385
    if( !m )
        return Point();
    CV_DbgAssert(m->dims <= 2);
3386

V
Vadim Pisarevsky 已提交
3387 3388 3389
    ptrdiff_t ofs = ptr - m->data;
    int y = (int)(ofs/m->step[0]);
    return Point((int)((ofs - y*m->step[0])/elemSize), y);
3390 3391
}

V
Vadim Pisarevsky 已提交
3392
void MatConstIterator::pos(int* _idx) const
3393
{
V
Vadim Pisarevsky 已提交
3394 3395 3396
    CV_Assert(m != 0 && _idx);
    ptrdiff_t ofs = ptr - m->data;
    for( int i = 0; i < m->dims; i++ )
3397
    {
V
Vadim Pisarevsky 已提交
3398 3399 3400
        size_t s = m->step[i], v = ofs/s;
        ofs -= v*s;
        _idx[i] = (int)v;
3401 3402 3403
    }
}

V
Vadim Pisarevsky 已提交
3404
ptrdiff_t MatConstIterator::lpos() const
3405
{
V
Vadim Pisarevsky 已提交
3406 3407 3408 3409 3410 3411 3412
    if(!m)
        return 0;
    if( m->isContinuous() )
        return (ptr - sliceStart)/elemSize;
    ptrdiff_t ofs = ptr - m->data;
    int i, d = m->dims;
    if( d == 2 )
3413
    {
V
Vadim Pisarevsky 已提交
3414 3415
        ptrdiff_t y = ofs/m->step[0];
        return y*m->cols + (ofs - y*m->step[0])/elemSize;
3416
    }
V
Vadim Pisarevsky 已提交
3417 3418
    ptrdiff_t result = 0;
    for( i = 0; i < d; i++ )
3419
    {
V
Vadim Pisarevsky 已提交
3420 3421 3422
        size_t s = m->step[i], v = ofs/s;
        ofs -= v*s;
        result = result*m->size[i] + v;
3423
    }
V
Vadim Pisarevsky 已提交
3424
    return result;
3425
}
3426

V
Vadim Pisarevsky 已提交
3427
void MatConstIterator::seek(ptrdiff_t ofs, bool relative)
3428
{
V
Vadim Pisarevsky 已提交
3429
    if( m->isContinuous() )
3430
    {
V
Vadim Pisarevsky 已提交
3431 3432 3433 3434 3435 3436
        ptr = (relative ? ptr : sliceStart) + ofs*elemSize;
        if( ptr < sliceStart )
            ptr = sliceStart;
        else if( ptr > sliceEnd )
            ptr = sliceEnd;
        return;
3437
    }
3438

V
Vadim Pisarevsky 已提交
3439 3440
    int d = m->dims;
    if( d == 2 )
3441
    {
V
Vadim Pisarevsky 已提交
3442 3443
        ptrdiff_t ofs0, y;
        if( relative )
3444
        {
V
Vadim Pisarevsky 已提交
3445 3446 3447
            ofs0 = ptr - m->data;
            y = ofs0/m->step[0];
            ofs += y*m->cols + (ofs0 - y*m->step[0])/elemSize;
3448
        }
V
Vadim Pisarevsky 已提交
3449 3450 3451
        y = ofs/m->cols;
        int y1 = std::min(std::max((int)y, 0), m->rows-1);
        sliceStart = m->data + y1*m->step[0];
3452
        sliceEnd = sliceStart + m->cols*elemSize;
V
Vadim Pisarevsky 已提交
3453 3454 3455
        ptr = y < 0 ? sliceStart : y >= m->rows ? sliceEnd :
            sliceStart + (ofs - y*m->cols)*elemSize;
        return;
3456
    }
3457

V
Vadim Pisarevsky 已提交
3458 3459
    if( relative )
        ofs += lpos();
3460

V
Vadim Pisarevsky 已提交
3461 3462
    if( ofs < 0 )
        ofs = 0;
3463

V
Vadim Pisarevsky 已提交
3464 3465 3466 3467 3468 3469
    int szi = m->size[d-1];
    ptrdiff_t t = ofs/szi;
    int v = (int)(ofs - t*szi);
    ofs = t;
    ptr = m->data + v*elemSize;
    sliceStart = m->data;
3470

V
Vadim Pisarevsky 已提交
3471
    for( int i = d-2; i >= 0; i-- )
3472
    {
V
Vadim Pisarevsky 已提交
3473 3474 3475 3476 3477
        szi = m->size[i];
        t = ofs/szi;
        v = (int)(ofs - t*szi);
        ofs = t;
        sliceStart += v*m->step[i];
3478
    }
3479

V
Vadim Pisarevsky 已提交
3480 3481 3482 3483 3484
    sliceEnd = sliceStart + m->size[d-1]*elemSize;
    if( ofs > 0 )
        ptr = sliceEnd;
    else
        ptr = sliceStart + (ptr - m->data);
3485
}
3486

V
Vadim Pisarevsky 已提交
3487 3488 3489 3490 3491 3492 3493 3494 3495
void MatConstIterator::seek(const int* _idx, bool relative)
{
    int i, d = m->dims;
    ptrdiff_t ofs = 0;
    if( !_idx )
        ;
    else if( d == 2 )
        ofs = _idx[0]*m->size[1] + _idx[1];
    else
3496
    {
V
Vadim Pisarevsky 已提交
3497 3498
        for( i = 0; i < d; i++ )
            ofs = ofs*m->size[i] + _idx[i];
3499
    }
V
Vadim Pisarevsky 已提交
3500
    seek(ofs, relative);
3501 3502 3503 3504 3505 3506 3507 3508 3509 3510 3511 3512 3513 3514 3515 3516 3517 3518 3519 3520 3521 3522 3523 3524 3525 3526 3527 3528
}

//////////////////////////////// SparseMat ////////////////////////////////

template<typename T1, typename T2> void
convertData_(const void* _from, void* _to, int cn)
{
    const T1* from = (const T1*)_from;
    T2* to = (T2*)_to;
    if( cn == 1 )
        *to = saturate_cast<T2>(*from);
    else
        for( int i = 0; i < cn; i++ )
            to[i] = saturate_cast<T2>(from[i]);
}

template<typename T1, typename T2> void
convertScaleData_(const void* _from, void* _to, int cn, double alpha, double beta)
{
    const T1* from = (const T1*)_from;
    T2* to = (T2*)_to;
    if( cn == 1 )
        *to = saturate_cast<T2>(*from*alpha + beta);
    else
        for( int i = 0; i < cn; i++ )
            to[i] = saturate_cast<T2>(from[i]*alpha + beta);
}

A
Andrey Kamaev 已提交
3529 3530 3531 3532
typedef void (*ConvertData)(const void* from, void* to, int cn);
typedef void (*ConvertScaleData)(const void* from, void* to, int cn, double alpha, double beta);

static ConvertData getConvertElem(int fromType, int toType)
3533 3534 3535 3536 3537 3538 3539 3540 3541 3542 3543 3544 3545 3546 3547 3548 3549 3550 3551 3552 3553 3554 3555 3556 3557 3558 3559 3560 3561 3562 3563 3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576
{
    static ConvertData tab[][8] =
    {{ convertData_<uchar, uchar>, convertData_<uchar, schar>,
      convertData_<uchar, ushort>, convertData_<uchar, short>,
      convertData_<uchar, int>, convertData_<uchar, float>,
      convertData_<uchar, double>, 0 },

    { convertData_<schar, uchar>, convertData_<schar, schar>,
      convertData_<schar, ushort>, convertData_<schar, short>,
      convertData_<schar, int>, convertData_<schar, float>,
      convertData_<schar, double>, 0 },

    { convertData_<ushort, uchar>, convertData_<ushort, schar>,
      convertData_<ushort, ushort>, convertData_<ushort, short>,
      convertData_<ushort, int>, convertData_<ushort, float>,
      convertData_<ushort, double>, 0 },

    { convertData_<short, uchar>, convertData_<short, schar>,
      convertData_<short, ushort>, convertData_<short, short>,
      convertData_<short, int>, convertData_<short, float>,
      convertData_<short, double>, 0 },

    { convertData_<int, uchar>, convertData_<int, schar>,
      convertData_<int, ushort>, convertData_<int, short>,
      convertData_<int, int>, convertData_<int, float>,
      convertData_<int, double>, 0 },

    { convertData_<float, uchar>, convertData_<float, schar>,
      convertData_<float, ushort>, convertData_<float, short>,
      convertData_<float, int>, convertData_<float, float>,
      convertData_<float, double>, 0 },

    { convertData_<double, uchar>, convertData_<double, schar>,
      convertData_<double, ushort>, convertData_<double, short>,
      convertData_<double, int>, convertData_<double, float>,
      convertData_<double, double>, 0 },

    { 0, 0, 0, 0, 0, 0, 0, 0 }};

    ConvertData func = tab[CV_MAT_DEPTH(fromType)][CV_MAT_DEPTH(toType)];
    CV_Assert( func != 0 );
    return func;
}

A
Andrey Kamaev 已提交
3577
static ConvertScaleData getConvertScaleElem(int fromType, int toType)
3578 3579 3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590 3591 3592 3593 3594 3595 3596 3597 3598 3599 3600 3601 3602 3603 3604 3605 3606 3607 3608 3609 3610 3611 3612 3613 3614 3615 3616 3617 3618 3619 3620 3621 3622 3623 3624 3625 3626
{
    static ConvertScaleData tab[][8] =
    {{ convertScaleData_<uchar, uchar>, convertScaleData_<uchar, schar>,
      convertScaleData_<uchar, ushort>, convertScaleData_<uchar, short>,
      convertScaleData_<uchar, int>, convertScaleData_<uchar, float>,
      convertScaleData_<uchar, double>, 0 },

    { convertScaleData_<schar, uchar>, convertScaleData_<schar, schar>,
      convertScaleData_<schar, ushort>, convertScaleData_<schar, short>,
      convertScaleData_<schar, int>, convertScaleData_<schar, float>,
      convertScaleData_<schar, double>, 0 },

    { convertScaleData_<ushort, uchar>, convertScaleData_<ushort, schar>,
      convertScaleData_<ushort, ushort>, convertScaleData_<ushort, short>,
      convertScaleData_<ushort, int>, convertScaleData_<ushort, float>,
      convertScaleData_<ushort, double>, 0 },

    { convertScaleData_<short, uchar>, convertScaleData_<short, schar>,
      convertScaleData_<short, ushort>, convertScaleData_<short, short>,
      convertScaleData_<short, int>, convertScaleData_<short, float>,
      convertScaleData_<short, double>, 0 },

    { convertScaleData_<int, uchar>, convertScaleData_<int, schar>,
      convertScaleData_<int, ushort>, convertScaleData_<int, short>,
      convertScaleData_<int, int>, convertScaleData_<int, float>,
      convertScaleData_<int, double>, 0 },

    { convertScaleData_<float, uchar>, convertScaleData_<float, schar>,
      convertScaleData_<float, ushort>, convertScaleData_<float, short>,
      convertScaleData_<float, int>, convertScaleData_<float, float>,
      convertScaleData_<float, double>, 0 },

    { convertScaleData_<double, uchar>, convertScaleData_<double, schar>,
      convertScaleData_<double, ushort>, convertScaleData_<double, short>,
      convertScaleData_<double, int>, convertScaleData_<double, float>,
      convertScaleData_<double, double>, 0 },

    { 0, 0, 0, 0, 0, 0, 0, 0 }};

    ConvertScaleData func = tab[CV_MAT_DEPTH(fromType)][CV_MAT_DEPTH(toType)];
    CV_Assert( func != 0 );
    return func;
}

enum { HASH_SIZE0 = 8 };

static inline void copyElem(const uchar* from, uchar* to, size_t elemSize)
{
    size_t i;
3627
    for( i = 0; i + sizeof(int) <= elemSize; i += sizeof(int) )
3628 3629 3630 3631 3632 3633 3634 3635
        *(int*)(to + i) = *(const int*)(from + i);
    for( ; i < elemSize; i++ )
        to[i] = from[i];
}

static inline bool isZeroElem(const uchar* data, size_t elemSize)
{
    size_t i;
3636
    for( i = 0; i + sizeof(int) <= elemSize; i += sizeof(int) )
3637 3638 3639 3640 3641 3642 3643 3644 3645 3646 3647 3648 3649 3650
        if( *(int*)(data + i) != 0 )
            return false;
    for( ; i < elemSize; i++ )
        if( data[i] != 0 )
            return false;
    return true;
}

SparseMat::Hdr::Hdr( int _dims, const int* _sizes, int _type )
{
    refcount = 1;

    dims = _dims;
    valueOffset = (int)alignSize(sizeof(SparseMat::Node) +
3651
        sizeof(int)*std::max(dims - CV_MAX_DIM, 0), CV_ELEM_SIZE1(_type));
3652 3653
    nodeSize = alignSize(valueOffset +
        CV_ELEM_SIZE(_type), (int)sizeof(size_t));
3654

3655 3656 3657 3658 3659 3660 3661 3662 3663 3664 3665 3666 3667 3668 3669 3670 3671 3672
    int i;
    for( i = 0; i < dims; i++ )
        size[i] = _sizes[i];
    for( ; i < CV_MAX_DIM; i++ )
        size[i] = 0;
    clear();
}

void SparseMat::Hdr::clear()
{
    hashtab.clear();
    hashtab.resize(HASH_SIZE0);
    pool.clear();
    pool.resize(nodeSize);
    nodeCount = freeList = 0;
}


V
Vadim Pisarevsky 已提交
3673
SparseMat::SparseMat(const Mat& m)
3674 3675 3676 3677 3678 3679
: flags(MAGIC_VAL), hdr(0)
{
    create( m.dims, m.size, m.type() );

    int i, idx[CV_MAX_DIM] = {0}, d = m.dims, lastSize = m.size[d - 1];
    size_t esz = m.elemSize();
A
Andrey Kamaev 已提交
3680
    uchar* dptr = m.data;
3681 3682 3683

    for(;;)
    {
A
Andrey Kamaev 已提交
3684
        for( i = 0; i < lastSize; i++, dptr += esz )
3685
        {
A
Andrey Kamaev 已提交
3686
            if( isZeroElem(dptr, esz) )
3687 3688 3689
                continue;
            idx[d-1] = i;
            uchar* to = newNode(idx, hash(idx));
A
Andrey Kamaev 已提交
3690
            copyElem( dptr, to, esz );
3691
        }
3692

3693 3694
        for( i = d - 2; i >= 0; i-- )
        {
A
Andrey Kamaev 已提交
3695
            dptr += m.step[i] - m.size[i+1]*m.step[i+1];
3696 3697 3698 3699 3700 3701 3702 3703
            if( ++idx[i] < m.size[i] )
                break;
            idx[i] = 0;
        }
        if( i < 0 )
            break;
    }
}
3704

3705 3706 3707 3708 3709 3710 3711 3712 3713 3714 3715 3716 3717 3718 3719 3720 3721 3722 3723 3724 3725 3726 3727 3728 3729 3730 3731 3732 3733 3734 3735 3736 3737 3738 3739 3740 3741 3742 3743 3744 3745 3746 3747 3748 3749 3750 3751 3752 3753 3754 3755 3756 3757 3758 3759 3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778
void SparseMat::create(int d, const int* _sizes, int _type)
{
    int i;
    CV_Assert( _sizes && 0 < d && d <= CV_MAX_DIM );
    for( i = 0; i < d; i++ )
        CV_Assert( _sizes[i] > 0 );
    _type = CV_MAT_TYPE(_type);
    if( hdr && _type == type() && hdr->dims == d && hdr->refcount == 1 )
    {
        for( i = 0; i < d; i++ )
            if( _sizes[i] != hdr->size[i] )
                break;
        if( i == d )
        {
            clear();
            return;
        }
    }
    release();
    flags = MAGIC_VAL | _type;
    hdr = new Hdr(d, _sizes, _type);
}

void SparseMat::copyTo( SparseMat& m ) const
{
    if( hdr == m.hdr )
        return;
    if( !hdr )
    {
        m.release();
        return;
    }
    m.create( hdr->dims, hdr->size, type() );
    SparseMatConstIterator from = begin();
    size_t i, N = nzcount(), esz = elemSize();

    for( i = 0; i < N; i++, ++from )
    {
        const Node* n = from.node();
        uchar* to = m.newNode(n->idx, n->hashval);
        copyElem( from.ptr, to, esz );
    }
}

void SparseMat::copyTo( Mat& m ) const
{
    CV_Assert( hdr );
    m.create( dims(), hdr->size, type() );
    m = Scalar(0);

    SparseMatConstIterator from = begin();
    size_t i, N = nzcount(), esz = elemSize();

    for( i = 0; i < N; i++, ++from )
    {
        const Node* n = from.node();
        copyElem( from.ptr, m.ptr(n->idx), esz);
    }
}


void SparseMat::convertTo( SparseMat& m, int rtype, double alpha ) const
{
    int cn = channels();
    if( rtype < 0 )
        rtype = type();
    rtype = CV_MAKETYPE(rtype, cn);
    if( hdr == m.hdr && rtype != type()  )
    {
        SparseMat temp;
        convertTo(temp, rtype, alpha);
        m = temp;
        return;
    }
3779

3780 3781 3782
    CV_Assert(hdr != 0);
    if( hdr != m.hdr )
        m.create( hdr->dims, hdr->size, rtype );
3783

3784 3785 3786 3787 3788
    SparseMatConstIterator from = begin();
    size_t i, N = nzcount();

    if( alpha == 1 )
    {
3789
        ConvertData cvtfunc = getConvertElem(type(), rtype);
3790 3791 3792 3793
        for( i = 0; i < N; i++, ++from )
        {
            const Node* n = from.node();
            uchar* to = hdr == m.hdr ? from.ptr : m.newNode(n->idx, n->hashval);
3794
            cvtfunc( from.ptr, to, cn );
3795 3796 3797 3798
        }
    }
    else
    {
3799
        ConvertScaleData cvtfunc = getConvertScaleElem(type(), rtype);
3800 3801 3802 3803
        for( i = 0; i < N; i++, ++from )
        {
            const Node* n = from.node();
            uchar* to = hdr == m.hdr ? from.ptr : m.newNode(n->idx, n->hashval);
3804
            cvtfunc( from.ptr, to, cn, alpha, 0 );
3805 3806 3807 3808 3809 3810 3811 3812 3813 3814 3815
        }
    }
}


void SparseMat::convertTo( Mat& m, int rtype, double alpha, double beta ) const
{
    int cn = channels();
    if( rtype < 0 )
        rtype = type();
    rtype = CV_MAKETYPE(rtype, cn);
3816

3817 3818 3819 3820 3821 3822 3823 3824 3825
    CV_Assert( hdr );
    m.create( dims(), hdr->size, rtype );
    m = Scalar(beta);

    SparseMatConstIterator from = begin();
    size_t i, N = nzcount();

    if( alpha == 1 && beta == 0 )
    {
3826
        ConvertData cvtfunc = getConvertElem(type(), rtype);
3827 3828 3829 3830 3831 3832 3833 3834 3835
        for( i = 0; i < N; i++, ++from )
        {
            const Node* n = from.node();
            uchar* to = m.ptr(n->idx);
            cvtfunc( from.ptr, to, cn );
        }
    }
    else
    {
3836
        ConvertScaleData cvtfunc = getConvertScaleElem(type(), rtype);
3837 3838 3839 3840 3841 3842 3843 3844 3845 3846 3847 3848 3849 3850 3851
        for( i = 0; i < N; i++, ++from )
        {
            const Node* n = from.node();
            uchar* to = m.ptr(n->idx);
            cvtfunc( from.ptr, to, cn, alpha, beta );
        }
    }
}

void SparseMat::clear()
{
    if( hdr )
        hdr->clear();
}

3852 3853 3854 3855 3856 3857 3858 3859 3860 3861 3862 3863 3864
uchar* SparseMat::ptr(int i0, bool createMissing, size_t* hashval)
{
    CV_Assert( hdr && hdr->dims == 1 );
    size_t h = hashval ? *hashval : hash(i0);
    size_t hidx = h & (hdr->hashtab.size() - 1), nidx = hdr->hashtab[hidx];
    uchar* pool = &hdr->pool[0];
    while( nidx != 0 )
    {
        Node* elem = (Node*)(pool + nidx);
        if( elem->hashval == h && elem->idx[0] == i0 )
            return &value<uchar>(elem);
        nidx = elem->next;
    }
3865

3866 3867 3868 3869 3870 3871 3872
    if( createMissing )
    {
        int idx[] = { i0 };
        return newNode( idx, h );
    }
    return 0;
}
3873

3874 3875 3876 3877 3878 3879 3880 3881 3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899 3900 3901 3902 3903 3904 3905 3906 3907 3908 3909 3910 3911 3912 3913 3914 3915 3916 3917 3918 3919 3920 3921 3922 3923 3924 3925 3926 3927 3928 3929 3930 3931 3932 3933 3934 3935 3936 3937 3938 3939 3940 3941 3942 3943 3944 3945 3946 3947 3948 3949 3950 3951 3952 3953 3954 3955 3956 3957 3958 3959 3960 3961 3962 3963 3964 3965 3966 3967 3968 3969 3970 3971 3972 3973 3974 3975 3976 3977 3978 3979 3980 3981 3982 3983 3984 3985 3986 3987 3988 3989 3990 3991 3992 3993 3994 3995 3996 3997 3998 3999 4000 4001 4002 4003 4004 4005 4006 4007 4008 4009 4010 4011
uchar* SparseMat::ptr(int i0, int i1, bool createMissing, size_t* hashval)
{
    CV_Assert( hdr && hdr->dims == 2 );
    size_t h = hashval ? *hashval : hash(i0, i1);
    size_t hidx = h & (hdr->hashtab.size() - 1), nidx = hdr->hashtab[hidx];
    uchar* pool = &hdr->pool[0];
    while( nidx != 0 )
    {
        Node* elem = (Node*)(pool + nidx);
        if( elem->hashval == h && elem->idx[0] == i0 && elem->idx[1] == i1 )
            return &value<uchar>(elem);
        nidx = elem->next;
    }

    if( createMissing )
    {
        int idx[] = { i0, i1 };
        return newNode( idx, h );
    }
    return 0;
}

uchar* SparseMat::ptr(int i0, int i1, int i2, bool createMissing, size_t* hashval)
{
    CV_Assert( hdr && hdr->dims == 3 );
    size_t h = hashval ? *hashval : hash(i0, i1, i2);
    size_t hidx = h & (hdr->hashtab.size() - 1), nidx = hdr->hashtab[hidx];
    uchar* pool = &hdr->pool[0];
    while( nidx != 0 )
    {
        Node* elem = (Node*)(pool + nidx);
        if( elem->hashval == h && elem->idx[0] == i0 &&
            elem->idx[1] == i1 && elem->idx[2] == i2 )
            return &value<uchar>(elem);
        nidx = elem->next;
    }

    if( createMissing )
    {
        int idx[] = { i0, i1, i2 };
        return newNode( idx, h );
    }
    return 0;
}

uchar* SparseMat::ptr(const int* idx, bool createMissing, size_t* hashval)
{
    CV_Assert( hdr );
    int i, d = hdr->dims;
    size_t h = hashval ? *hashval : hash(idx);
    size_t hidx = h & (hdr->hashtab.size() - 1), nidx = hdr->hashtab[hidx];
    uchar* pool = &hdr->pool[0];
    while( nidx != 0 )
    {
        Node* elem = (Node*)(pool + nidx);
        if( elem->hashval == h )
        {
            for( i = 0; i < d; i++ )
                if( elem->idx[i] != idx[i] )
                    break;
            if( i == d )
                return &value<uchar>(elem);
        }
        nidx = elem->next;
    }

    return createMissing ? newNode(idx, h) : 0;
}

void SparseMat::erase(int i0, int i1, size_t* hashval)
{
    CV_Assert( hdr && hdr->dims == 2 );
    size_t h = hashval ? *hashval : hash(i0, i1);
    size_t hidx = h & (hdr->hashtab.size() - 1), nidx = hdr->hashtab[hidx], previdx=0;
    uchar* pool = &hdr->pool[0];
    while( nidx != 0 )
    {
        Node* elem = (Node*)(pool + nidx);
        if( elem->hashval == h && elem->idx[0] == i0 && elem->idx[1] == i1 )
            break;
        previdx = nidx;
        nidx = elem->next;
    }

    if( nidx )
        removeNode(hidx, nidx, previdx);
}

void SparseMat::erase(int i0, int i1, int i2, size_t* hashval)
{
    CV_Assert( hdr && hdr->dims == 3 );
    size_t h = hashval ? *hashval : hash(i0, i1, i2);
    size_t hidx = h & (hdr->hashtab.size() - 1), nidx = hdr->hashtab[hidx], previdx=0;
    uchar* pool = &hdr->pool[0];
    while( nidx != 0 )
    {
        Node* elem = (Node*)(pool + nidx);
        if( elem->hashval == h && elem->idx[0] == i0 &&
            elem->idx[1] == i1 && elem->idx[2] == i2 )
            break;
        previdx = nidx;
        nidx = elem->next;
    }

    if( nidx )
        removeNode(hidx, nidx, previdx);
}

void SparseMat::erase(const int* idx, size_t* hashval)
{
    CV_Assert( hdr );
    int i, d = hdr->dims;
    size_t h = hashval ? *hashval : hash(idx);
    size_t hidx = h & (hdr->hashtab.size() - 1), nidx = hdr->hashtab[hidx], previdx=0;
    uchar* pool = &hdr->pool[0];
    while( nidx != 0 )
    {
        Node* elem = (Node*)(pool + nidx);
        if( elem->hashval == h )
        {
            for( i = 0; i < d; i++ )
                if( elem->idx[i] != idx[i] )
                    break;
            if( i == d )
                break;
        }
        previdx = nidx;
        nidx = elem->next;
    }

    if( nidx )
        removeNode(hidx, nidx, previdx);
}

void SparseMat::resizeHashTab(size_t newsize)
{
    newsize = std::max(newsize, (size_t)8);
    if((newsize & (newsize-1)) != 0)
4012
        newsize = (size_t)1 << cvCeil(std::log((double)newsize)/CV_LOG2);
4013 4014

    size_t i, hsize = hdr->hashtab.size();
4015
    std::vector<size_t> _newh(newsize);
4016 4017 4018 4019 4020 4021 4022 4023 4024 4025 4026 4027 4028 4029 4030 4031 4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042 4043 4044 4045
    size_t* newh = &_newh[0];
    for( i = 0; i < newsize; i++ )
        newh[i] = 0;
    uchar* pool = &hdr->pool[0];
    for( i = 0; i < hsize; i++ )
    {
        size_t nidx = hdr->hashtab[i];
        while( nidx )
        {
            Node* elem = (Node*)(pool + nidx);
            size_t next = elem->next;
            size_t newhidx = elem->hashval & (newsize - 1);
            elem->next = newh[newhidx];
            newh[newhidx] = nidx;
            nidx = next;
        }
    }
    hdr->hashtab = _newh;
}

uchar* SparseMat::newNode(const int* idx, size_t hashval)
{
    const int HASH_MAX_FILL_FACTOR=3;
    assert(hdr);
    size_t hsize = hdr->hashtab.size();
    if( ++hdr->nodeCount > hsize*HASH_MAX_FILL_FACTOR )
    {
        resizeHashTab(std::max(hsize*2, (size_t)8));
        hsize = hdr->hashtab.size();
    }
4046

4047 4048 4049 4050 4051 4052 4053 4054 4055 4056 4057 4058 4059 4060 4061 4062 4063 4064 4065 4066 4067 4068
    if( !hdr->freeList )
    {
        size_t i, nsz = hdr->nodeSize, psize = hdr->pool.size(),
            newpsize = std::max(psize*2, 8*nsz);
        hdr->pool.resize(newpsize);
        uchar* pool = &hdr->pool[0];
        hdr->freeList = std::max(psize, nsz);
        for( i = hdr->freeList; i < newpsize - nsz; i += nsz )
            ((Node*)(pool + i))->next = i + nsz;
        ((Node*)(pool + i))->next = 0;
    }
    size_t nidx = hdr->freeList;
    Node* elem = (Node*)&hdr->pool[nidx];
    hdr->freeList = elem->next;
    elem->hashval = hashval;
    size_t hidx = hashval & (hsize - 1);
    elem->next = hdr->hashtab[hidx];
    hdr->hashtab[hidx] = nidx;

    int i, d = hdr->dims;
    for( i = 0; i < d; i++ )
        elem->idx[i] = idx[i];
4069
    size_t esz = elemSize();
4070
    uchar* p = &value<uchar>(elem);
4071
    if( esz == sizeof(float) )
4072
        *((float*)p) = 0.f;
4073
    else if( esz == sizeof(double) )
4074 4075
        *((double*)p) = 0.;
    else
4076
        memset(p, 0, esz);
4077

4078 4079 4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091 4092 4093 4094 4095 4096 4097 4098 4099 4100 4101 4102 4103
    return p;
}


void SparseMat::removeNode(size_t hidx, size_t nidx, size_t previdx)
{
    Node* n = node(nidx);
    if( previdx )
    {
        Node* prev = node(previdx);
        prev->next = n->next;
    }
    else
        hdr->hashtab[hidx] = n->next;
    n->next = hdr->freeList;
    hdr->freeList = nidx;
    --hdr->nodeCount;
}


SparseMatConstIterator::SparseMatConstIterator(const SparseMat* _m)
: m((SparseMat*)_m), hashidx(0), ptr(0)
{
    if(!_m || !_m->hdr)
        return;
    SparseMat::Hdr& hdr = *m->hdr;
4104
    const std::vector<size_t>& htab = hdr.hashtab;
4105 4106 4107 4108 4109 4110 4111 4112 4113 4114 4115 4116 4117 4118 4119 4120 4121 4122 4123 4124 4125 4126 4127 4128 4129 4130 4131 4132 4133 4134 4135 4136 4137 4138 4139 4140 4141 4142 4143 4144 4145 4146 4147 4148
    size_t i, hsize = htab.size();
    for( i = 0; i < hsize; i++ )
    {
        size_t nidx = htab[i];
        if( nidx )
        {
            hashidx = i;
            ptr = &hdr.pool[nidx] + hdr.valueOffset;
            return;
        }
    }
}

SparseMatConstIterator& SparseMatConstIterator::operator ++()
{
    if( !ptr || !m || !m->hdr )
        return *this;
    SparseMat::Hdr& hdr = *m->hdr;
    size_t next = ((const SparseMat::Node*)(ptr - hdr.valueOffset))->next;
    if( next )
    {
        ptr = &hdr.pool[next] + hdr.valueOffset;
        return *this;
    }
    size_t i = hashidx + 1, sz = hdr.hashtab.size();
    for( ; i < sz; i++ )
    {
        size_t nidx = hdr.hashtab[i];
        if( nidx )
        {
            hashidx = i;
            ptr = &hdr.pool[nidx] + hdr.valueOffset;
            return *this;
        }
    }
    hashidx = sz;
    ptr = 0;
    return *this;
}


double norm( const SparseMat& src, int normType )
{
    SparseMatConstIterator it = src.begin();
4149

4150 4151 4152 4153
    size_t i, N = src.nzcount();
    normType &= NORM_TYPE_MASK;
    int type = src.type();
    double result = 0;
4154

4155
    CV_Assert( normType == NORM_INF || normType == NORM_L1 || normType == NORM_L2 );
4156

4157 4158 4159 4160 4161 4162 4163 4164 4165 4166 4167
    if( type == CV_32F )
    {
        if( normType == NORM_INF )
            for( i = 0; i < N; i++, ++it )
                result = std::max(result, std::abs((double)*(const float*)it.ptr));
        else if( normType == NORM_L1 )
            for( i = 0; i < N; i++, ++it )
                result += std::abs(*(const float*)it.ptr);
        else
            for( i = 0; i < N; i++, ++it )
            {
4168
                double v = *(const float*)it.ptr;
4169 4170 4171 4172 4173 4174 4175 4176 4177 4178 4179 4180 4181 4182
                result += v*v;
            }
    }
    else if( type == CV_64F )
    {
        if( normType == NORM_INF )
            for( i = 0; i < N; i++, ++it )
                result = std::max(result, std::abs(*(const double*)it.ptr));
        else if( normType == NORM_L1 )
            for( i = 0; i < N; i++, ++it )
                result += std::abs(*(const double*)it.ptr);
        else
            for( i = 0; i < N; i++, ++it )
            {
4183
                double v = *(const double*)it.ptr;
4184 4185 4186 4187 4188
                result += v*v;
            }
    }
    else
        CV_Error( CV_StsUnsupportedFormat, "Only 32f and 64f are supported" );
4189

4190 4191 4192 4193
    if( normType == NORM_L2 )
        result = std::sqrt(result);
    return result;
}
4194

4195 4196 4197 4198 4199 4200
void minMaxLoc( const SparseMat& src, double* _minval, double* _maxval, int* _minidx, int* _maxidx )
{
    SparseMatConstIterator it = src.begin();
    size_t i, N = src.nzcount(), d = src.hdr ? src.hdr->dims : 0;
    int type = src.type();
    const int *minidx = 0, *maxidx = 0;
4201

4202 4203 4204 4205 4206 4207 4208 4209 4210 4211 4212 4213 4214 4215 4216 4217 4218 4219 4220 4221 4222 4223 4224 4225 4226 4227 4228 4229 4230 4231 4232 4233 4234 4235 4236 4237 4238 4239 4240 4241 4242 4243 4244 4245 4246 4247
    if( type == CV_32F )
    {
        float minval = FLT_MAX, maxval = -FLT_MAX;
        for( i = 0; i < N; i++, ++it )
        {
            float v = *(const float*)it.ptr;
            if( v < minval )
            {
                minval = v;
                minidx = it.node()->idx;
            }
            if( v > maxval )
            {
                maxval = v;
                maxidx = it.node()->idx;
            }
        }
        if( _minval )
            *_minval = minval;
        if( _maxval )
            *_maxval = maxval;
    }
    else if( type == CV_64F )
    {
        double minval = DBL_MAX, maxval = -DBL_MAX;
        for( i = 0; i < N; i++, ++it )
        {
            double v = *(const double*)it.ptr;
            if( v < minval )
            {
                minval = v;
                minidx = it.node()->idx;
            }
            if( v > maxval )
            {
                maxval = v;
                maxidx = it.node()->idx;
            }
        }
        if( _minval )
            *_minval = minval;
        if( _maxval )
            *_maxval = maxval;
    }
    else
        CV_Error( CV_StsUnsupportedFormat, "Only 32f and 64f are supported" );
4248

4249 4250 4251 4252 4253 4254 4255 4256
    if( _minidx )
        for( i = 0; i < d; i++ )
            _minidx[i] = minidx[i];
    if( _maxidx )
        for( i = 0; i < d; i++ )
            _maxidx[i] = maxidx[i];
}

4257

4258 4259 4260 4261 4262 4263 4264 4265 4266 4267
void normalize( const SparseMat& src, SparseMat& dst, double a, int norm_type )
{
    double scale = 1;
    if( norm_type == CV_L2 || norm_type == CV_L1 || norm_type == CV_C )
    {
        scale = norm( src, norm_type );
        scale = scale > DBL_EPSILON ? a/scale : 0.;
    }
    else
        CV_Error( CV_StsBadArg, "Unknown/unsupported norm type" );
4268

4269 4270
    src.convertTo( dst, -1, scale );
}
4271 4272

////////////////////// RotatedRect //////////////////////
4273

4274 4275 4276 4277 4278
void RotatedRect::points(Point2f pt[]) const
{
    double _angle = angle*CV_PI/180.;
    float b = (float)cos(_angle)*0.5f;
    float a = (float)sin(_angle)*0.5f;
4279

4280 4281 4282 4283 4284 4285 4286 4287 4288 4289
    pt[0].x = center.x - a*size.height - b*size.width;
    pt[0].y = center.y + b*size.height - a*size.width;
    pt[1].x = center.x + a*size.height - b*size.width;
    pt[1].y = center.y - b*size.height - a*size.width;
    pt[2].x = 2*center.x - pt[0].x;
    pt[2].y = 2*center.y - pt[0].y;
    pt[3].x = 2*center.x - pt[1].x;
    pt[3].y = 2*center.y - pt[1].y;
}

4290
Rect RotatedRect::boundingRect() const
4291 4292 4293
{
    Point2f pt[4];
    points(pt);
4294 4295 4296 4297
    Rect r(cvFloor(std::min(std::min(std::min(pt[0].x, pt[1].x), pt[2].x), pt[3].x)),
           cvFloor(std::min(std::min(std::min(pt[0].y, pt[1].y), pt[2].y), pt[3].y)),
           cvCeil(std::max(std::max(std::max(pt[0].x, pt[1].x), pt[2].x), pt[3].x)),
           cvCeil(std::max(std::max(std::max(pt[0].y, pt[1].y), pt[2].y), pt[3].y)));
4298 4299 4300
    r.width -= r.x - 1;
    r.height -= r.y - 1;
    return r;
4301
}
4302 4303 4304

}

A
Andrey Kamaev 已提交
4305 4306 4307 4308 4309 4310 4311 4312 4313 4314 4315 4316 4317 4318 4319 4320 4321 4322 4323 4324 4325 4326 4327 4328 4329 4330 4331 4332 4333 4334 4335 4336 4337 4338 4339 4340 4341 4342 4343 4344 4345 4346 4347 4348 4349 4350 4351 4352 4353 4354 4355 4356 4357 4358
// glue

CvMatND::CvMatND(const cv::Mat& m)
{
    cvInitMatNDHeader(this, m.dims, m.size, m.type(), m.data );
    int i, d = m.dims;
    for( i = 0; i < d; i++ )
        dim[i].step = (int)m.step[i];
    type |= m.flags & cv::Mat::CONTINUOUS_FLAG;
}

_IplImage::_IplImage(const cv::Mat& m)
{
    CV_Assert( m.dims <= 2 );
    cvInitImageHeader(this, m.size(), cvIplDepth(m.flags), m.channels());
    cvSetData(this, m.data, (int)m.step[0]);
}

CvSparseMat* cvCreateSparseMat(const cv::SparseMat& sm)
{
    if( !sm.hdr )
        return 0;

    CvSparseMat* m = cvCreateSparseMat(sm.hdr->dims, sm.hdr->size, sm.type());

    cv::SparseMatConstIterator from = sm.begin();
    size_t i, N = sm.nzcount(), esz = sm.elemSize();

    for( i = 0; i < N; i++, ++from )
    {
        const cv::SparseMat::Node* n = from.node();
        uchar* to = cvPtrND(m, n->idx, 0, -2, 0);
        cv::copyElem(from.ptr, to, esz);
    }
    return m;
}

void CvSparseMat::copyToSparseMat(cv::SparseMat& m) const
{
    m.create( dims, &size[0], type );

    CvSparseMatIterator it;
    CvSparseNode* n = cvInitSparseMatIterator(this, &it);
    size_t esz = m.elemSize();

    for( ; n != 0; n = cvGetNextSparseNode(&it) )
    {
        const int* idx = CV_NODE_IDX(this, n);
        uchar* to = m.newNode(idx, m.hash(idx));
        cv::copyElem((const uchar*)CV_NODE_VAL(this, n), to, esz);
    }
}


4359
/* End of file. */