matrix.cpp 126.0 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 {

51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82
class StdMatAllocator : public MatAllocator
{
public:
    UMatData* allocate(int dims, const int* sizes, int type, size_t* step) const
    {
        size_t total = CV_ELEM_SIZE(type);
        for( int i = dims-1; i >= 0; i-- )
        {
            if( step )
                step[i] = total;
            total *= sizes[i];
        }
        uchar* data = (uchar*)fastMalloc(total);
        UMatData* u = new UMatData(this);
        u->data = u->origdata = data;
        u->size = total;
        u->refcount = 1;

        return u;
    }

    bool allocate(UMatData* u, int accessFlags) const
    {
        if(!u) return false;
        if(u->handle != 0)
            return true;
        return UMat::getStdAllocator()->allocate(u, accessFlags);
    }

    void deallocate(UMatData* u) const
    {
        if(u)
83
        {
84
            fastFree(u->origdata);
85 86
            delete u;
        }
87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200
    }

    void map(UMatData*, int) const
    {
    }

    void unmap(UMatData* u) const
    {
        if(u->urefcount == 0)
            deallocate(u);
    }

    void download(UMatData* u, void* dstptr,
                  int dims, const size_t sz[],
                  const size_t srcofs[], const size_t srcstep[],
                  const size_t dststep[]) const
    {
        if(!u)
            return;
        int isz[CV_MAX_DIM];
        uchar* srcptr = u->data;
        for( int i = 0; i < dims; i++ )
        {
            CV_Assert( sz[i] <= (size_t)INT_MAX );
            if( sz[i] == 0 )
                return;
            if( srcofs )
                srcptr += srcofs[i]*(i <= dims-2 ? srcstep[i] : 1);
            isz[i] = (int)sz[i];
        }

        Mat src(dims, isz, CV_8U, srcptr, srcstep);
        Mat dst(dims, isz, CV_8U, dstptr, dststep);

        const Mat* arrays[] = { &src, &dst };
        uchar* ptrs[2];
        NAryMatIterator it(arrays, ptrs, 2);
        size_t j, planesz = it.size;

        for( j = 0; j < it.nplanes; j++, ++it )
            memcpy(ptrs[1], ptrs[0], planesz);
    }

    void upload(UMatData* u, const void* srcptr, int dims, const size_t sz[],
                const size_t dstofs[], const size_t dststep[],
                const size_t srcstep[]) const
    {
        if(!u)
            return;
        int isz[CV_MAX_DIM];
        uchar* dstptr = u->data;
        for( int i = 0; i < dims; i++ )
        {
            CV_Assert( sz[i] <= (size_t)INT_MAX );
            if( sz[i] == 0 )
                return;
            if( dstofs )
                dstptr += dstofs[i]*(i <= dims-2 ? dststep[i] : 1);
            isz[i] = (int)sz[i];
        }

        Mat src(dims, isz, CV_8U, (void*)srcptr, srcstep);
        Mat dst(dims, isz, CV_8U, dstptr, dststep);

        const Mat* arrays[] = { &src, &dst };
        uchar* ptrs[2];
        NAryMatIterator it(arrays, ptrs, 2);
        size_t j, planesz = it.size;

        for( j = 0; j < it.nplanes; j++, ++it )
            memcpy(ptrs[1], ptrs[0], planesz);
    }

    void copy(UMatData* usrc, UMatData* udst, int dims, const size_t sz[],
              const size_t srcofs[], const size_t srcstep[],
              const size_t dstofs[], const size_t dststep[], bool) const
    {
        if(!usrc || !udst)
            return;
        int isz[CV_MAX_DIM];
        uchar* srcptr = usrc->data;
        uchar* dstptr = udst->data;
        for( int i = 0; i < dims; i++ )
        {
            CV_Assert( sz[i] <= (size_t)INT_MAX );
            if( sz[i] == 0 )
                return;
            if( srcofs )
                srcptr += srcofs[i]*(i <= dims-2 ? srcstep[i] : 1);
            if( dstofs )
                dstptr += dstofs[i]*(i <= dims-2 ? dststep[i] : 1);
            isz[i] = (int)sz[i];
        }

        Mat src(dims, isz, CV_8U, srcptr, srcstep);
        Mat dst(dims, isz, CV_8U, dstptr, dststep);

        const Mat* arrays[] = { &src, &dst };
        uchar* ptrs[2];
        NAryMatIterator it(arrays, ptrs, 2);
        size_t j, planesz = it.size;

        for( j = 0; j < it.nplanes; j++, ++it )
            memcpy(ptrs[1], ptrs[0], planesz);
    }
};


MatAllocator* Mat::getStdAllocator()
{
    static StdMatAllocator allocator;
    return &allocator;
}

V
Vadim Pisarevsky 已提交
201 202
void swap( Mat& a, Mat& b )
{
203 204 205 206 207 208 209 210 211 212
    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);
213
    std::swap(a.u, b.u);
214

215 216 217 218
    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]);
219

V
Vadim Pisarevsky 已提交
220 221 222 223 224
    if( a.step.p == b.step.buf )
    {
        a.step.p = a.step.buf;
        a.size.p = &a.rows;
    }
225

V
Vadim Pisarevsky 已提交
226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250
    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;
251
            m.rows = m.cols = -1;
V
Vadim Pisarevsky 已提交
252 253
        }
    }
254

V
Vadim Pisarevsky 已提交
255 256 257
    m.dims = _dims;
    if( !_sz )
        return;
258

259
    size_t esz = CV_ELEM_SIZE(m.flags), total = esz;
V
Vadim Pisarevsky 已提交
260 261 262 263
    int i;
    for( i = _dims-1; i >= 0; i-- )
    {
        int s = _sz[i];
264
        CV_Assert( s >= 0 );
V
Vadim Pisarevsky 已提交
265
        m.size.p[i] = s;
266

V
Vadim Pisarevsky 已提交
267 268 269 270 271 272 273 274 275 276 277
        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;
        }
    }
278

V
Vadim Pisarevsky 已提交
279 280 281 282 283 284 285
    if( _dims == 1 )
    {
        m.dims = 2;
        m.cols = 1;
        m.step[1] = esz;
    }
}
286

287
static void updateContinuityFlag(Mat& m)
V
Vadim Pisarevsky 已提交
288 289 290 291 292 293 294
{
    int i, j;
    for( i = 0; i < m.dims; i++ )
    {
        if( m.size[i] > 1 )
            break;
    }
295

V
Vadim Pisarevsky 已提交
296 297 298 299 300
    for( j = m.dims-1; j > i; j-- )
    {
        if( m.step[j]*m.size[j] < m.step[j-1] )
            break;
    }
301

302
    uint64 t = (uint64)m.step[0]*m.size[0];
303
    if( j <= i && t == (size_t)t )
V
Vadim Pisarevsky 已提交
304
        m.flags |= Mat::CONTINUOUS_FLAG;
305 306 307
    else
        m.flags &= ~Mat::CONTINUOUS_FLAG;
}
308

309 310 311
static void finalizeHdr(Mat& m)
{
    updateContinuityFlag(m);
312 313
    int d = m.dims;
    if( d > 2 )
V
Vadim Pisarevsky 已提交
314
        m.rows = m.cols = -1;
315 316
    if(m.u)
        m.data = m.datastart = m.u->data;
V
Vadim Pisarevsky 已提交
317 318
    if( m.data )
    {
319 320 321
        m.datalimit = m.datastart + m.size[0]*m.step[0];
        if( m.size[0] > 0 )
        {
322 323
            m.dataend = m.data + m.size[d-1]*m.step[d-1];
            for( int i = 0; i < d-1; i++ )
324 325 326 327
                m.dataend += (m.size[i] - 1)*m.step[i];
        }
        else
            m.dataend = m.datalimit;
V
Vadim Pisarevsky 已提交
328
    }
329 330
    else
        m.dataend = m.datalimit = 0;
V
Vadim Pisarevsky 已提交
331
}
332 333


V
Vadim Pisarevsky 已提交
334 335 336
void Mat::create(int d, const int* _sizes, int _type)
{
    int i;
V
Vladislav Vinogradov 已提交
337
    CV_Assert(0 <= d && d <= CV_MAX_DIM && _sizes);
V
Vadim Pisarevsky 已提交
338
    _type = CV_MAT_TYPE(_type);
339

V
Vadim Pisarevsky 已提交
340 341 342 343 344 345 346 347 348 349
    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;
    }
350

V
Vadim Pisarevsky 已提交
351 352 353 354
    release();
    if( d == 0 )
        return;
    flags = (_type & CV_MAT_TYPE_MASK) | MAGIC_VAL;
355
    setSize(*this, d, _sizes, 0, true);
356

357
    if( total() > 0 )
V
Vadim Pisarevsky 已提交
358
    {
359
        MatAllocator *a = allocator, *a0 = getStdAllocator();
A
Andrey Pavlenko 已提交
360
#ifdef HAVE_TGPU
361 362
        if( !a || a == tegra::getAllocator() )
            a = tegra::getAllocator(d, _sizes, _type);
A
Andrey Pavlenko 已提交
363
#endif
364 365 366
        if(!a)
            a = a0;
        try
367
        {
368 369
            u = a->allocate(dims, size, _type, step.p);
            CV_Assert(u != 0);
370
        }
371
        catch(...)
372
        {
373 374 375
            if(a != a0)
                u = a0->allocate(dims, size, _type, step.p);
            CV_Assert(u != 0);
376
        }
377
        CV_Assert( step[dims-1] == (size_t)CV_ELEM_SIZE(flags) );
V
Vadim Pisarevsky 已提交
378
    }
379

V
Vadim Pisarevsky 已提交
380 381 382 383 384 385 386 387 388 389 390 391
    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];
    }
}
392

V
Vadim Pisarevsky 已提交
393 394
void Mat::deallocate()
{
395 396
    if(u)
        (u->currAllocator ? u->currAllocator : allocator ? allocator : getStdAllocator())->unmap(u);
V
Vadim Pisarevsky 已提交
397 398
}

A
Andrey Kamaev 已提交
399 400 401
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 已提交
402 403 404 405 406
{
    CV_Assert( m.dims >= 2 );
    if( m.dims > 2 )
    {
        AutoBuffer<Range> rs(m.dims);
A
Andrey Kamaev 已提交
407 408
        rs[0] = _rowRange;
        rs[1] = _colRange;
V
Vadim Pisarevsky 已提交
409 410 411 412 413
        for( int i = 2; i < m.dims; i++ )
            rs[i] = Range::all();
        *this = m(rs);
        return;
    }
414

V
Vadim Pisarevsky 已提交
415
    *this = m;
A
Andrey Kamaev 已提交
416
    if( _rowRange != Range::all() && _rowRange != Range(0,rows) )
V
Vadim Pisarevsky 已提交
417
    {
A
Andrey Kamaev 已提交
418 419 420
        CV_Assert( 0 <= _rowRange.start && _rowRange.start <= _rowRange.end && _rowRange.end <= m.rows );
        rows = _rowRange.size();
        data += step*_rowRange.start;
421
        flags |= SUBMATRIX_FLAG;
V
Vadim Pisarevsky 已提交
422
    }
423

A
Andrey Kamaev 已提交
424
    if( _colRange != Range::all() && _colRange != Range(0,cols) )
V
Vadim Pisarevsky 已提交
425
    {
A
Andrey Kamaev 已提交
426 427 428
        CV_Assert( 0 <= _colRange.start && _colRange.start <= _colRange.end && _colRange.end <= m.cols );
        cols = _colRange.size();
        data += _colRange.start*elemSize();
V
Vadim Pisarevsky 已提交
429
        flags &= cols < m.cols ? ~CONTINUOUS_FLAG : -1;
430
        flags |= SUBMATRIX_FLAG;
V
Vadim Pisarevsky 已提交
431
    }
432

V
Vadim Pisarevsky 已提交
433 434
    if( rows == 1 )
        flags |= CONTINUOUS_FLAG;
435

V
Vadim Pisarevsky 已提交
436 437 438 439 440 441
    if( rows <= 0 || cols <= 0 )
    {
        release();
        rows = cols = 0;
    }
}
442 443


V
Vadim Pisarevsky 已提交
444 445 446
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),
447 448
    datastart(m.datastart), dataend(m.dataend), datalimit(m.datalimit),
    allocator(m.allocator), size(&rows)
V
Vadim Pisarevsky 已提交
449 450 451 452
{
    CV_Assert( m.dims <= 2 );
    flags &= roi.width < m.cols ? ~CONTINUOUS_FLAG : -1;
    flags |= roi.height == 1 ? CONTINUOUS_FLAG : 0;
453

454
    size_t esz = CV_ELEM_SIZE(flags);
V
Vadim Pisarevsky 已提交
455 456 457 458 459
    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);
460 461
    if( roi.width < m.cols || roi.height < m.rows )
        flags |= SUBMATRIX_FLAG;
462

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

V
Vadim Pisarevsky 已提交
465 466 467 468 469 470 471
    if( rows <= 0 || cols <= 0 )
    {
        release();
        rows = cols = 0;
    }
}

472

A
Andrey Kamaev 已提交
473 474 475
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 已提交
476
{
477 478
    flags |= CV_MAT_TYPE(_type);
    data = datastart = (uchar*)_data;
V
Vadim Pisarevsky 已提交
479 480 481
    setSize(*this, _dims, _sizes, _steps, true);
    finalizeHdr(*this);
}
482 483


A
Andrey Kamaev 已提交
484 485 486
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 已提交
487 488
{
    int i, d = m.dims;
489

V
Vadim Pisarevsky 已提交
490 491 492 493 494 495 496 497 498 499
    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];
500
        if( r != Range::all() && r != Range(0, size.p[i]))
V
Vadim Pisarevsky 已提交
501
        {
502 503 504
            size.p[i] = r.end - r.start;
            data += r.start*step.p[i];
            flags |= SUBMATRIX_FLAG;
V
Vadim Pisarevsky 已提交
505 506
        }
    }
507
    updateContinuityFlag(*this);
V
Vadim Pisarevsky 已提交
508
}
509 510


A
Andrey Kamaev 已提交
511
static Mat cvMatNDToMat(const CvMatND* m, bool copyData)
V
Vadim Pisarevsky 已提交
512
{
A
Andrey Kamaev 已提交
513 514
    Mat thiz;

515
    if( !m )
A
Andrey Kamaev 已提交
516 517 518
        return thiz;
    thiz.data = thiz.datastart = m->data.ptr;
    thiz.flags |= CV_MAT_TYPE(m->type);
V
Vadim Pisarevsky 已提交
519 520
    int _sizes[CV_MAX_DIM];
    size_t _steps[CV_MAX_DIM];
521

V
Vadim Pisarevsky 已提交
522 523 524 525 526 527
    int i, d = m->dims;
    for( i = 0; i < d; i++ )
    {
        _sizes[i] = m->dim[i].size;
        _steps[i] = m->dim[i].step;
    }
528

A
Andrey Kamaev 已提交
529 530
    setSize(thiz, d, _sizes, _steps);
    finalizeHdr(thiz);
V
Vadim Pisarevsky 已提交
531 532 533

    if( copyData )
    {
A
Andrey Kamaev 已提交
534 535 536
        Mat temp(thiz);
        thiz.release();
        temp.copyTo(thiz);
V
Vadim Pisarevsky 已提交
537
    }
538

A
Andrey Kamaev 已提交
539
    return thiz;
V
Vadim Pisarevsky 已提交
540
}
541

A
Andrey Kamaev 已提交
542
static Mat cvMatToMat(const CvMat* m, bool copyData)
543
{
A
Andrey Kamaev 已提交
544
    Mat thiz;
545

546
    if( !m )
A
Andrey Kamaev 已提交
547
        return thiz;
548

549 550
    if( !copyData )
    {
A
Andrey Kamaev 已提交
551 552 553 554 555 556
        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;
557 558
        if( _step == 0 )
            _step = minstep;
A
Andrey Kamaev 已提交
559 560 561
        thiz.datalimit = thiz.datastart + _step*thiz.rows;
        thiz.dataend = thiz.datalimit - _step + minstep;
        thiz.step[0] = _step; thiz.step[1] = esz;
562 563 564
    }
    else
    {
A
Andrey Kamaev 已提交
565 566
        thiz.data = thiz.datastart = thiz.dataend = 0;
        Mat(m->rows, m->cols, m->type, m->data.ptr, m->step).copyTo(thiz);
567
    }
A
Andrey Kamaev 已提交
568 569

    return thiz;
570 571
}

572

A
Andrey Kamaev 已提交
573
static Mat iplImageToMat(const IplImage* img, bool copyData)
574
{
A
Andrey Kamaev 已提交
575
    Mat m;
576

577
    if( !img )
A
Andrey Kamaev 已提交
578
        return m;
579

A
Andrey Kamaev 已提交
580
    m.dims = 2;
581
    CV_DbgAssert(CV_IS_IMAGE(img) && img->imageData != 0);
582

A
Andrey Kamaev 已提交
583
    int imgdepth = IPL2CV_DEPTH(img->depth);
584
    size_t esz;
A
Andrey Kamaev 已提交
585
    m.step[0] = img->widthStep;
586 587 588 589

    if(!img->roi)
    {
        CV_Assert(img->dataOrder == IPL_DATA_ORDER_PIXEL);
A
Andrey Kamaev 已提交
590 591 592 593 594
        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);
595 596 597 598 599
    }
    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 已提交
600 601 602 603 604 605 606 607 608 609 610 611
        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;
612 613 614

    if( copyData )
    {
A
Andrey Kamaev 已提交
615 616
        Mat m2 = m;
        m.release();
617 618
        if( !img->roi || !img->roi->coi ||
            img->dataOrder == IPL_DATA_ORDER_PLANE)
A
Andrey Kamaev 已提交
619
            m2.copyTo(m);
620 621 622
        else
        {
            int ch[] = {img->roi->coi - 1, 0};
A
Andrey Kamaev 已提交
623 624
            m.create(m2.rows, m2.cols, m2.type());
            mixChannels(&m2, 1, &m, 1, ch, 1);
625 626 627
        }
    }

A
Andrey Kamaev 已提交
628 629
    return m;
}
630

A
Andrey Kamaev 已提交
631
Mat Mat::diag(int d) const
632
{
V
Vadim Pisarevsky 已提交
633
    CV_Assert( dims <= 2 );
A
Andrey Kamaev 已提交
634 635 636
    Mat m = *this;
    size_t esz = elemSize();
    int len;
637

A
Andrey Kamaev 已提交
638 639 640 641 642 643 644 645 646 647 648 649 650 651 652 653 654 655 656 657 658 659 660 661 662 663
    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;
}
664

665 666 667
void Mat::pop_back(size_t nelems)
{
    CV_Assert( nelems <= (size_t)size.p[0] );
668

669 670 671 672 673 674 675 676 677 678 679 680 681 682 683
    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);
        }*/
    }
}
684 685


686 687 688 689 690
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) );
691

692 693 694 695 696 697 698 699 700 701 702
    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;
703

704 705 706
    CV_Assert( (int)nelems >= 0 );
    if( !isSubmatrix() && data + step.p[0]*nelems <= datalimit )
        return;
707

708
    int r = size.p[0];
709

710 711
    if( (size_t)r >= nelems )
        return;
712

713 714
    size.p[0] = std::max((int)nelems, 1);
    size_t newsize = total()*elemSize();
715

716 717
    if( newsize < MIN_SIZE )
        size.p[0] = (int)((MIN_SIZE + newsize - 1)*nelems/newsize);
718

719 720 721 722 723 724 725
    Mat m(dims, size.p, type());
    size.p[0] = r;
    if( r > 0 )
    {
        Mat mpart = m.rowRange(0, r);
        copyTo(mpart);
    }
726

727 728 729 730 731
    *this = m;
    size.p[0] = r;
    dataend = data + step.p[0]*r;
}

732

733 734 735
void Mat::resize(size_t nelems)
{
    int saveRows = size.p[0];
736 737
    if( saveRows == (int)nelems )
        return;
738
    CV_Assert( (int)nelems >= 0 );
739

740 741
    if( isSubmatrix() || data + step.p[0]*nelems > datalimit )
        reserve(nelems);
742

743 744
    size.p[0] = (int)nelems;
    dataend += (size.p[0] - saveRows)*step.p[0];
745

746
    //updateContinuityFlag(*this);
747 748
}

749 750 751 752 753

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

755 756 757 758 759
    if( size.p[0] > saveRows )
    {
        Mat part = rowRange(saveRows, size.p[0]);
        part = s;
    }
760 761
}

762 763 764 765 766
void Mat::push_back(const Mat& elems)
{
    int r = size.p[0], delta = elems.size.p[0];
    if( delta == 0 )
        return;
767 768 769 770 771
    if( this == &elems )
    {
        Mat tmp = elems;
        push_back(tmp);
        return;
772
    }
773 774 775 776 777
    if( !data )
    {
        *this = elems.clone();
        return;
    }
778 779 780 781 782 783 784 785

    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, "");
786

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

790 791
    size.p[0] += delta;
    dataend += step.p[0]*delta;
792

793
    //updateContinuityFlag(*this);
794

795 796 797 798 799 800 801 802 803
    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);
    }
}

804

805
Mat cvarrToMat(const CvArr* arr, bool copyData,
806
               bool /*allowND*/, int coiMode, AutoBuffer<double>* abuf )
807
{
V
Vadim Pisarevsky 已提交
808 809
    if( !arr )
        return Mat();
A
Andrey Kamaev 已提交
810 811
    if( CV_IS_MAT_HDR_Z(arr) )
        return cvMatToMat((const CvMat*)arr, copyData);
V
Vadim Pisarevsky 已提交
812
    if( CV_IS_MATND(arr) )
A
Andrey Kamaev 已提交
813
        return cvMatNDToMat((const CvMatND*)arr, copyData );
V
Vadim Pisarevsky 已提交
814
    if( CV_IS_IMAGE(arr) )
815 816 817 818
    {
        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 已提交
819
        return iplImageToMat(iplimg, copyData);
820
    }
V
Vadim Pisarevsky 已提交
821
    if( CV_IS_SEQ(arr) )
822 823
    {
        CvSeq* seq = (CvSeq*)arr;
824 825 826 827
        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);
828
        if(!copyData && seq->first->next == seq->first)
829 830 831 832 833 834 835 836 837 838
            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);
839 840 841
        cvCvtSeqToArray(seq, buf.data, CV_WHOLE_SEQ);
        return buf;
    }
V
Vadim Pisarevsky 已提交
842 843 844 845 846 847 848 849 850
    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;
851

V
Vadim Pisarevsky 已提交
852 853
    if( delta1 == 0 )
        ofs.x = ofs.y = 0;
854 855
    else
    {
V
Vadim Pisarevsky 已提交
856 857 858
        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 );
859
    }
V
Vadim Pisarevsky 已提交
860 861 862 863 864
    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);
865 866
}

V
Vadim Pisarevsky 已提交
867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882
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;
883
}
884 885

}
886

887
void cv::extractImageCOI(const CvArr* arr, OutputArray _ch, int coi)
888 889
{
    Mat mat = cvarrToMat(arr, false, true, 1);
890 891
    _ch.create(mat.dims, mat.size, mat.depth());
    Mat ch = _ch.getMat();
V
Vadim Pisarevsky 已提交
892
    if(coi < 0)
893
    {
V
Vadim Pisarevsky 已提交
894 895 896
        CV_Assert( CV_IS_IMAGE(arr) );
        coi = cvGetImageCOI((const IplImage*)arr)-1;
    }
897 898 899 900
    CV_Assert(0 <= coi && coi < mat.channels());
    int _pairs[] = { coi, 0 };
    mixChannels( &mat, 1, &ch, 1, _pairs, 1 );
}
901

902
void cv::insertImageCOI(InputArray _ch, CvArr* arr, int coi)
903
{
904
    Mat ch = _ch.getMat(), mat = cvarrToMat(arr, false, true, 1);
V
Vadim Pisarevsky 已提交
905
    if(coi < 0)
906
    {
V
Vadim Pisarevsky 已提交
907 908 909
        CV_Assert( CV_IS_IMAGE(arr) );
        coi = cvGetImageCOI((const IplImage*)arr)-1;
    }
V
Vadim Pisarevsky 已提交
910
    CV_Assert(ch.size == mat.size && ch.depth() == mat.depth() && 0 <= coi && coi < mat.channels());
911 912 913
    int _pairs[] = { 0, coi };
    mixChannels( &ch, 1, &mat, 1, _pairs, 1 );
}
914

915 916
namespace cv
{
917 918 919 920

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

923 924 925 926 927 928 929
    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;
    }
930

931
    CV_Assert( dims <= 2 );
932

933 934 935 936 937 938 939 940 941 942 943 944 945 946 947 948 949 950 951 952 953 954 955 956 957
    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 已提交
958
        hdr.step[0] = total_width * elemSize1();
959 960 961 962 963 964 965 966 967 968
    }

    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);
969
    hdr.step[1] = CV_ELEM_SIZE(hdr.flags);
970 971 972
    return hdr;
}

A
Andrey Kamaev 已提交
973 974 975 976 977 978 979 980 981 982 983 984
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;
}
985

986 987 988 989
int Mat::checkVector(int _elemChannels, int _depth, bool _requireContinuous) const
{
    return (depth() == _depth || _depth <= 0) &&
        (isContinuous() || !_requireContinuous) &&
990 991
        ((dims == 2 && (((rows == 1 || cols == 1) && channels() == _elemChannels) ||
                        (cols == _elemChannels && channels() == 1))) ||
992 993 994 995
        (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;
}
996 997 998 999 1000 1001 1002 1003 1004 1005 1006 1007 1008 1009 1010 1011 1012 1013 1014 1015 1016 1017 1018 1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059 1060 1061 1062 1063 1064 1065 1066 1067 1068 1069 1070


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,"");
    }
}
1071

1072

1073 1074 1075 1076
/*************************************************************************************************\
                                        Input/Output Array
\*************************************************************************************************/

1077
Mat _InputArray::getMat(int i) const
1078 1079
{
    int k = kind();
1080
    int accessFlags = flags & ACCESS_MASK;
1081

1082 1083
    if( k == MAT )
    {
V
Vadim Pisarevsky 已提交
1084 1085 1086 1087
        const Mat* m = (const Mat*)obj;
        if( i < 0 )
            return *m;
        return m->row(i);
1088
    }
1089

1090 1091 1092 1093 1094 1095 1096 1097
    if( k == UMAT )
    {
        const UMat* m = (const UMat*)obj;
        if( i < 0 )
            return m->getMat(accessFlags);
        return m->getMat(accessFlags).row(i);
    }

1098 1099 1100 1101 1102
    if( k == EXPR )
    {
        CV_Assert( i < 0 );
        return (Mat)*((const MatExpr*)obj);
    }
1103

1104 1105 1106 1107 1108
    if( k == MATX )
    {
        CV_Assert( i < 0 );
        return Mat(sz, flags, obj);
    }
1109

1110 1111 1112 1113
    if( k == STD_VECTOR )
    {
        CV_Assert( i < 0 );
        int t = CV_MAT_TYPE(flags);
1114
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
1115

1116 1117
        return !v.empty() ? Mat(size(), t, (void*)&v[0]) : Mat();
    }
1118

1119 1120
    if( k == NONE )
        return Mat();
1121

1122 1123 1124
    if( k == STD_VECTOR_VECTOR )
    {
        int t = type(i);
1125
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
1126
        CV_Assert( 0 <= i && i < (int)vv.size() );
1127
        const std::vector<uchar>& v = vv[i];
1128

1129 1130
        return !v.empty() ? Mat(size(i), t, (void*)&v[0]) : Mat();
    }
1131

1132
    if( k == STD_VECTOR_MAT )
1133
    {
1134
        const std::vector<Mat>& v = *(const std::vector<Mat>*)obj;
1135
        CV_Assert( 0 <= i && i < (int)v.size() );
1136

1137
        return v[i];
1138
    }
1139

1140 1141 1142 1143 1144 1145 1146 1147
    if( k == STD_VECTOR_UMAT )
    {
        const std::vector<UMat>& v = *(const std::vector<UMat>*)obj;
        CV_Assert( 0 <= i && i < (int)v.size() );

        return v[i].getMat(accessFlags);
    }

1148 1149 1150 1151 1152 1153 1154 1155 1156 1157
    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 );
1158
        CV_Error(cv::Error::StsNotImplemented, "You should explicitly call download method for cuda::GpuMat object");
1159 1160 1161
        return Mat();
    }

1162
    if( k == CUDA_MEM )
1163 1164 1165
    {
        CV_Assert( i < 0 );

1166
        const cuda::CudaMem* cuda_mem = (const cuda::CudaMem*)obj;
1167 1168 1169

        return cuda_mem->createMatHeader();
    }
1170 1171 1172 1173 1174 1175 1176 1177 1178 1179 1180 1181 1182 1183 1184 1185 1186 1187 1188 1189 1190 1191 1192 1193 1194 1195 1196 1197 1198 1199 1200 1201 1202 1203 1204 1205

    CV_Error(Error::StsNotImplemented, "Unknown/unsupported array type");
    return Mat();
}


UMat _InputArray::getUMat(int i) const
{
    int k = kind();
    int accessFlags = flags & ACCESS_MASK;

    if( k == UMAT )
    {
        const UMat* m = (const UMat*)obj;
        if( i < 0 )
            return *m;
        return m->row(i);
    }

    if( k == STD_VECTOR_UMAT )
    {
        const std::vector<UMat>& v = *(const std::vector<UMat>*)obj;
        CV_Assert( 0 <= i && i < (int)v.size() );

        return v[i];
    }

    if( k == MAT )
    {
        const Mat* m = (const Mat*)obj;
        if( i < 0 )
            return m->getUMat(accessFlags);
        return m->row(i).getUMat(accessFlags);
    }

    return getMat(i).getUMat(accessFlags);
1206
}
1207 1208


1209
void _InputArray::getMatVector(std::vector<Mat>& mv) const
1210 1211
{
    int k = kind();
1212
    int accessFlags = flags & ACCESS_MASK;
1213

1214 1215 1216
    if( k == MAT )
    {
        const Mat& m = *(const Mat*)obj;
1217
        int i, n = (int)m.size[0];
1218
        mv.resize(n);
1219

1220 1221 1222 1223 1224
        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;
    }
1225

1226 1227 1228
    if( k == EXPR )
    {
        Mat m = *(const MatExpr*)obj;
1229
        int i, n = m.size[0];
1230
        mv.resize(n);
1231

1232 1233 1234 1235
        for( i = 0; i < n; i++ )
            mv[i] = m.row(i);
        return;
    }
1236

1237 1238 1239 1240
    if( k == MATX )
    {
        size_t i, n = sz.height, esz = CV_ELEM_SIZE(flags);
        mv.resize(n);
1241

1242 1243 1244 1245
        for( i = 0; i < n; i++ )
            mv[i] = Mat(1, sz.width, CV_MAT_TYPE(flags), (uchar*)obj + esz*sz.width*i);
        return;
    }
1246

1247 1248
    if( k == STD_VECTOR )
    {
1249
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
1250

1251 1252 1253
        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);
1254

1255 1256 1257 1258
        for( i = 0; i < n; i++ )
            mv[i] = Mat(1, cn, t, (void*)(&v[0] + esz*i));
        return;
    }
1259

1260 1261 1262 1263 1264
    if( k == NONE )
    {
        mv.clear();
        return;
    }
1265

1266 1267
    if( k == STD_VECTOR_VECTOR )
    {
1268
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
1269
        int i, n = (int)vv.size();
1270 1271
        int t = CV_MAT_TYPE(flags);
        mv.resize(n);
1272

1273 1274
        for( i = 0; i < n; i++ )
        {
1275
            const std::vector<uchar>& v = vv[i];
1276 1277 1278 1279
            mv[i] = Mat(size(i), t, (void*)&v[0]);
        }
        return;
    }
1280

1281 1282 1283 1284 1285 1286 1287 1288 1289 1290 1291
    if( k == STD_VECTOR_MAT )
    {
        const std::vector<Mat>& v = *(const std::vector<Mat>*)obj;
        size_t i, n = v.size();
        mv.resize(n);

        for( i = 0; i < n; i++ )
            mv[i] = v[i];
        return;
    }

1292
    if( k == STD_VECTOR_UMAT )
1293
    {
1294 1295 1296
        const std::vector<UMat>& v = *(const std::vector<UMat>*)obj;
        size_t i, n = v.size();
        mv.resize(n);
1297

1298 1299
        for( i = 0; i < n; i++ )
            mv[i] = v[i].getMat(accessFlags);
1300 1301
        return;
    }
1302 1303

    CV_Error(Error::StsNotImplemented, "Unknown/unsupported array type");
1304
}
1305

1306
cuda::GpuMat _InputArray::getGpuMat() const
1307 1308 1309
{
    int k = kind();

1310 1311
    if (k == GPU_MAT)
    {
1312
        const cuda::GpuMat* d_mat = (const cuda::GpuMat*)obj;
1313 1314
        return *d_mat;
    }
1315

1316 1317
    if (k == CUDA_MEM)
    {
1318
        const cuda::CudaMem* cuda_mem = (const cuda::CudaMem*)obj;
1319 1320 1321 1322 1323 1324
        return cuda_mem->createGpuMatHeader();
    }

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

V
Vladislav Vinogradov 已提交
1328
    if (k == NONE)
1329
        return cuda::GpuMat();
V
Vladislav Vinogradov 已提交
1330

1331 1332
    CV_Error(cv::Error::StsNotImplemented, "getGpuMat is available only for cuda::GpuMat and cuda::CudaMem");
    return cuda::GpuMat();
1333 1334
}

1335
ogl::Buffer _InputArray::getOGlBuffer() const
1336 1337 1338
{
    int k = kind();

1339 1340 1341 1342
    CV_Assert(k == OPENGL_BUFFER);

    const ogl::Buffer* gl_buf = (const ogl::Buffer*)obj;
    return *gl_buf;
1343 1344
}

1345
int _InputArray::kind() const
1346
{
1347
    return flags & KIND_MASK;
1348
}
1349

1350
Size _InputArray::size(int i) const
1351 1352
{
    int k = kind();
1353

1354 1355 1356 1357 1358
    if( k == MAT )
    {
        CV_Assert( i < 0 );
        return ((const Mat*)obj)->size();
    }
1359

1360 1361 1362 1363 1364
    if( k == EXPR )
    {
        CV_Assert( i < 0 );
        return ((const MatExpr*)obj)->size();
    }
1365

1366 1367 1368 1369 1370 1371
    if( k == UMAT )
    {
        CV_Assert( i < 0 );
        return ((const UMat*)obj)->size();
    }

1372 1373 1374 1375 1376
    if( k == MATX )
    {
        CV_Assert( i < 0 );
        return sz;
    }
1377

1378 1379 1380
    if( k == STD_VECTOR )
    {
        CV_Assert( i < 0 );
1381 1382
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
        const std::vector<int>& iv = *(const std::vector<int>*)obj;
1383 1384 1385
        size_t szb = v.size(), szi = iv.size();
        return szb == szi ? Size((int)szb, 1) : Size((int)(szb/CV_ELEM_SIZE(flags)), 1);
    }
1386

1387 1388
    if( k == NONE )
        return Size();
1389

1390 1391
    if( k == STD_VECTOR_VECTOR )
    {
1392
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
1393 1394 1395
        if( i < 0 )
            return vv.empty() ? Size() : Size((int)vv.size(), 1);
        CV_Assert( i < (int)vv.size() );
1396
        const std::vector<std::vector<int> >& ivv = *(const std::vector<std::vector<int> >*)obj;
1397

1398 1399 1400
        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);
    }
1401

1402
    if( k == STD_VECTOR_MAT )
1403
    {
1404
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1405 1406 1407
        if( i < 0 )
            return vv.empty() ? Size() : Size((int)vv.size(), 1);
        CV_Assert( i < (int)vv.size() );
1408

1409 1410
        return vv[i].size();
    }
1411 1412 1413 1414

    if( k == OPENGL_BUFFER )
    {
        CV_Assert( i < 0 );
1415
        const ogl::Buffer* buf = (const ogl::Buffer*)obj;
1416 1417 1418
        return buf->size();
    }

1419
    if( k == GPU_MAT )
1420 1421
    {
        CV_Assert( i < 0 );
1422
        const cuda::GpuMat* d_mat = (const cuda::GpuMat*)obj;
1423 1424
        return d_mat->size();
    }
1425

1426 1427
    if( k == OCL_MAT )
    {
1428
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1429 1430
    }

1431 1432 1433 1434
    CV_Assert( k == CUDA_MEM );
    //if( k == CUDA_MEM )
    {
        CV_Assert( i < 0 );
1435
        const cuda::CudaMem* cuda_mem = (const cuda::CudaMem*)obj;
1436 1437
        return cuda_mem->size();
    }
1438 1439
}

1440
size_t _InputArray::total(int i) const
1441
{
1442 1443 1444 1445 1446 1447 1448 1449
    int k = kind();

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

1450 1451 1452 1453 1454 1455
    if( k == UMAT )
    {
        CV_Assert( i < 0 );
        return ((const UMat*)obj)->total();
    }

1456 1457
    if( k == STD_VECTOR_MAT )
    {
1458
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1459 1460 1461 1462 1463 1464 1465
        if( i < 0 )
            return vv.size();

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

1466 1467
    return size(i).area();
}
1468

1469
int _InputArray::type(int i) const
1470 1471
{
    int k = kind();
1472

1473 1474
    if( k == MAT )
        return ((const Mat*)obj)->type();
1475

1476 1477 1478
    if( k == UMAT )
        return ((const UMat*)obj)->type();

1479 1480
    if( k == EXPR )
        return ((const MatExpr*)obj)->type();
1481

1482 1483
    if( k == MATX || k == STD_VECTOR || k == STD_VECTOR_VECTOR )
        return CV_MAT_TYPE(flags);
1484

1485 1486
    if( k == NONE )
        return -1;
1487

1488
    if( k == STD_VECTOR_MAT )
1489
    {
1490
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1491
        CV_Assert( i < (int)vv.size() );
1492

1493 1494
        return vv[i >= 0 ? i : 0].type();
    }
1495

1496
    if( k == OPENGL_BUFFER )
1497
        return ((const ogl::Buffer*)obj)->type();
1498

1499
    if( k == GPU_MAT )
1500
        return ((const cuda::GpuMat*)obj)->type();
1501 1502 1503

    CV_Assert( k == CUDA_MEM );
    //if( k == CUDA_MEM )
1504
        return ((const cuda::CudaMem*)obj)->type();
1505
}
1506

1507
int _InputArray::depth(int i) const
1508 1509 1510
{
    return CV_MAT_DEPTH(type(i));
}
1511

1512
int _InputArray::channels(int i) const
1513 1514 1515
{
    return CV_MAT_CN(type(i));
}
1516

1517
bool _InputArray::empty() const
1518 1519
{
    int k = kind();
1520

1521 1522
    if( k == MAT )
        return ((const Mat*)obj)->empty();
1523

1524 1525 1526
    if( k == UMAT )
        return ((const UMat*)obj)->empty();

1527 1528
    if( k == EXPR )
        return false;
1529

1530 1531
    if( k == MATX )
        return false;
1532

1533 1534
    if( k == STD_VECTOR )
    {
1535
        const std::vector<uchar>& v = *(const std::vector<uchar>*)obj;
1536 1537
        return v.empty();
    }
1538

1539 1540
    if( k == NONE )
        return true;
1541

1542 1543
    if( k == STD_VECTOR_VECTOR )
    {
1544
        const std::vector<std::vector<uchar> >& vv = *(const std::vector<std::vector<uchar> >*)obj;
1545 1546
        return vv.empty();
    }
1547

1548
    if( k == STD_VECTOR_MAT )
1549
    {
1550
        const std::vector<Mat>& vv = *(const std::vector<Mat>*)obj;
1551 1552
        return vv.empty();
    }
1553

1554
    if( k == OPENGL_BUFFER )
1555
        return ((const ogl::Buffer*)obj)->empty();
1556

1557 1558
    if( k == OCL_MAT )
    {
1559
        CV_Error(CV_StsNotImplemented, "This method is not implemented for oclMat yet");
1560 1561
    }

1562
    if( k == GPU_MAT )
1563
        return ((const cuda::GpuMat*)obj)->empty();
1564 1565 1566

    CV_Assert( k == CUDA_MEM );
    //if( k == CUDA_MEM )
1567
        return ((const cuda::CudaMem*)obj)->empty();
1568
}
1569 1570


1571
bool _OutputArray::fixedSize() const
1572
{
1573
    return (flags & FIXED_SIZE) == FIXED_SIZE;
1574 1575
}

1576
bool _OutputArray::fixedType() const
1577
{
1578
    return (flags & FIXED_TYPE) == FIXED_TYPE;
1579
}
1580

A
Andrey Kamaev 已提交
1581
void _OutputArray::create(Size _sz, int mtype, int i, bool allowTransposed, int fixedDepthMask) const
1582 1583 1584 1585
{
    int k = kind();
    if( k == MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1586
        CV_Assert(!fixedSize() || ((Mat*)obj)->size.operator()() == _sz);
A
Andrey Kamaev 已提交
1587 1588
        CV_Assert(!fixedType() || ((Mat*)obj)->type() == mtype);
        ((Mat*)obj)->create(_sz, mtype);
1589 1590
        return;
    }
1591 1592 1593 1594 1595 1596 1597
    if( k == UMAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
        CV_Assert(!fixedSize() || ((UMat*)obj)->size.operator()() == _sz);
        CV_Assert(!fixedType() || ((UMat*)obj)->type() == mtype);
        ((UMat*)obj)->create(_sz, mtype);
        return;
    }
1598 1599
    if( k == GPU_MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1600 1601 1602
        CV_Assert(!fixedSize() || ((cuda::GpuMat*)obj)->size() == _sz);
        CV_Assert(!fixedType() || ((cuda::GpuMat*)obj)->type() == mtype);
        ((cuda::GpuMat*)obj)->create(_sz, mtype);
1603 1604
        return;
    }
1605 1606
    if( k == OPENGL_BUFFER && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1607 1608 1609
        CV_Assert(!fixedSize() || ((ogl::Buffer*)obj)->size() == _sz);
        CV_Assert(!fixedType() || ((ogl::Buffer*)obj)->type() == mtype);
        ((ogl::Buffer*)obj)->create(_sz, mtype);
1610 1611
        return;
    }
1612 1613
    if( k == CUDA_MEM && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1614 1615 1616
        CV_Assert(!fixedSize() || ((cuda::CudaMem*)obj)->size() == _sz);
        CV_Assert(!fixedType() || ((cuda::CudaMem*)obj)->type() == mtype);
        ((cuda::CudaMem*)obj)->create(_sz, mtype);
1617 1618
        return;
    }
A
Andrey Kamaev 已提交
1619 1620
    int sizes[] = {_sz.height, _sz.width};
    create(2, sizes, mtype, i, allowTransposed, fixedDepthMask);
1621 1622
}

A
Andrey Kamaev 已提交
1623
void _OutputArray::create(int rows, int cols, int mtype, int i, bool allowTransposed, int fixedDepthMask) const
1624 1625 1626 1627
{
    int k = kind();
    if( k == MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1628
        CV_Assert(!fixedSize() || ((Mat*)obj)->size.operator()() == Size(cols, rows));
A
Andrey Kamaev 已提交
1629 1630
        CV_Assert(!fixedType() || ((Mat*)obj)->type() == mtype);
        ((Mat*)obj)->create(rows, cols, mtype);
1631 1632
        return;
    }
1633 1634 1635 1636 1637 1638 1639
    if( k == UMAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
        CV_Assert(!fixedSize() || ((UMat*)obj)->size.operator()() == Size(cols, rows));
        CV_Assert(!fixedType() || ((UMat*)obj)->type() == mtype);
        ((UMat*)obj)->create(rows, cols, mtype);
        return;
    }
1640 1641
    if( k == GPU_MAT && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1642 1643 1644
        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);
1645 1646
        return;
    }
1647 1648
    if( k == OPENGL_BUFFER && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1649 1650 1651
        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);
1652 1653
        return;
    }
1654 1655
    if( k == CUDA_MEM && i < 0 && !allowTransposed && fixedDepthMask == 0 )
    {
1656 1657 1658
        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);
1659 1660
        return;
    }
A
Andrey Kamaev 已提交
1661 1662
    int sizes[] = {rows, cols};
    create(2, sizes, mtype, i, allowTransposed, fixedDepthMask);
1663
}
1664

1665 1666
void _OutputArray::create(int dims, const int* sizes, int mtype, int i,
                          bool allowTransposed, int fixedDepthMask) const
1667 1668
{
    int k = kind();
A
Andrey Kamaev 已提交
1669
    mtype = CV_MAT_TYPE(mtype);
1670

1671 1672 1673 1674
    if( k == MAT )
    {
        CV_Assert( i < 0 );
        Mat& m = *(Mat*)obj;
1675
        if( allowTransposed )
1676 1677
        {
            if( !m.isContinuous() )
1678 1679
            {
                CV_Assert(!fixedType() && !fixedSize());
1680
                m.release();
1681
            }
1682

1683
            if( dims == 2 && m.dims == 2 && m.data &&
A
Andrey Kamaev 已提交
1684
                m.type() == mtype && m.rows == sizes[1] && m.cols == sizes[0] )
1685 1686
                return;
        }
1687 1688 1689

        if(fixedType())
        {
A
Andrey Kamaev 已提交
1690 1691
            if(CV_MAT_CN(mtype) == m.channels() && ((1 << CV_MAT_TYPE(flags)) & fixedDepthMask) != 0 )
                mtype = m.type();
1692
            else
A
Andrey Kamaev 已提交
1693
                CV_Assert(CV_MAT_TYPE(mtype) == m.type());
1694 1695 1696 1697 1698
        }
        if(fixedSize())
        {
            CV_Assert(m.dims == dims);
            for(int j = 0; j < dims; ++j)
A
Andrey Kamaev 已提交
1699
                CV_Assert(m.size[j] == sizes[j]);
1700
        }
A
Andrey Kamaev 已提交
1701
        m.create(dims, sizes, mtype);
1702 1703
        return;
    }
1704

1705 1706 1707 1708 1709 1710 1711 1712 1713 1714 1715 1716 1717 1718 1719 1720 1721 1722 1723 1724 1725 1726 1727 1728 1729 1730 1731 1732 1733 1734 1735 1736 1737 1738
    if( k == UMAT )
    {
        CV_Assert( i < 0 );
        UMat& m = *(UMat*)obj;
        if( allowTransposed )
        {
            if( !m.isContinuous() )
            {
                CV_Assert(!fixedType() && !fixedSize());
                m.release();
            }

            if( dims == 2 && m.dims == 2 && !m.empty() &&
                m.type() == mtype && m.rows == sizes[1] && m.cols == sizes[0] )
                return;
        }

        if(fixedType())
        {
            if(CV_MAT_CN(mtype) == m.channels() && ((1 << CV_MAT_TYPE(flags)) & fixedDepthMask) != 0 )
                mtype = m.type();
            else
                CV_Assert(CV_MAT_TYPE(mtype) == m.type());
        }
        if(fixedSize())
        {
            CV_Assert(m.dims == dims);
            for(int j = 0; j < dims; ++j)
                CV_Assert(m.size[j] == sizes[j]);
        }
        m.create(dims, sizes, mtype);
        return;
    }

1739 1740 1741 1742
    if( k == MATX )
    {
        CV_Assert( i < 0 );
        int type0 = CV_MAT_TYPE(flags);
A
Andrey Kamaev 已提交
1743 1744 1745
        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)));
1746 1747
        return;
    }
1748

1749 1750
    if( k == STD_VECTOR || k == STD_VECTOR_VECTOR )
    {
A
Andrey Kamaev 已提交
1751 1752
        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;
1753
        std::vector<uchar>* v = (std::vector<uchar>*)obj;
1754

1755 1756
        if( k == STD_VECTOR_VECTOR )
        {
1757
            std::vector<std::vector<uchar> >& vv = *(std::vector<std::vector<uchar> >*)obj;
1758 1759
            if( i < 0 )
            {
1760
                CV_Assert(!fixedSize() || len == vv.size());
1761 1762 1763 1764 1765 1766 1767 1768
                vv.resize(len);
                return;
            }
            CV_Assert( i < (int)vv.size() );
            v = &vv[i];
        }
        else
            CV_Assert( i < 0 );
1769

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

1773
        int esz = CV_ELEM_SIZE(type0);
1774
        CV_Assert(!fixedSize() || len == ((std::vector<uchar>*)v)->size() / esz);
1775 1776 1777
        switch( esz )
        {
        case 1:
1778
            ((std::vector<uchar>*)v)->resize(len);
1779 1780
            break;
        case 2:
1781
            ((std::vector<Vec2b>*)v)->resize(len);
1782 1783
            break;
        case 3:
1784
            ((std::vector<Vec3b>*)v)->resize(len);
1785 1786
            break;
        case 4:
1787
            ((std::vector<int>*)v)->resize(len);
1788 1789
            break;
        case 6:
1790
            ((std::vector<Vec3s>*)v)->resize(len);
1791 1792
            break;
        case 8:
1793
            ((std::vector<Vec2i>*)v)->resize(len);
1794 1795
            break;
        case 12:
1796
            ((std::vector<Vec3i>*)v)->resize(len);
1797 1798
            break;
        case 16:
1799
            ((std::vector<Vec4i>*)v)->resize(len);
1800 1801
            break;
        case 24:
1802
            ((std::vector<Vec6i>*)v)->resize(len);
1803 1804
            break;
        case 32:
1805
            ((std::vector<Vec8i>*)v)->resize(len);
1806 1807
            break;
        case 36:
1808
            ((std::vector<Vec<int, 9> >*)v)->resize(len);
1809 1810
            break;
        case 48:
1811
            ((std::vector<Vec<int, 12> >*)v)->resize(len);
1812 1813
            break;
        case 64:
1814
            ((std::vector<Vec<int, 16> >*)v)->resize(len);
1815 1816
            break;
        case 128:
1817
            ((std::vector<Vec<int, 32> >*)v)->resize(len);
1818 1819
            break;
        case 256:
1820
            ((std::vector<Vec<int, 64> >*)v)->resize(len);
1821 1822
            break;
        case 512:
1823
            ((std::vector<Vec<int, 128> >*)v)->resize(len);
1824 1825 1826 1827 1828 1829
            break;
        default:
            CV_Error_(CV_StsBadArg, ("Vectors with element size %d are not supported. Please, modify OutputArray::create()\n", esz));
        }
        return;
    }
1830

1831 1832
    if( k == NONE )
    {
1833
        CV_Error(CV_StsNullPtr, "create() called for the missing output array" );
1834 1835
        return;
    }
1836

1837
    if( k == STD_VECTOR_MAT )
1838
    {
1839
        std::vector<Mat>& v = *(std::vector<Mat>*)obj;
1840

1841 1842
        if( i < 0 )
        {
A
Andrey Kamaev 已提交
1843 1844
            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();
1845

1846
            CV_Assert(!fixedSize() || len == len0);
1847
            v.resize(len);
1848 1849
            if( fixedType() )
            {
A
Andrey Kamaev 已提交
1850
                int _type = CV_MAT_TYPE(flags);
1851 1852
                for( size_t j = len0; j < len; j++ )
                {
1853
                    if( v[j].type() == _type )
1854
                        continue;
1855 1856
                    CV_Assert( v[j].empty() );
                    v[j].flags = (v[j].flags & ~CV_MAT_TYPE_MASK) | _type;
1857 1858
                }
            }
1859 1860
            return;
        }
1861

1862 1863
        CV_Assert( i < (int)v.size() );
        Mat& m = v[i];
1864

1865
        if( allowTransposed )
1866 1867
        {
            if( !m.isContinuous() )
1868 1869
            {
                CV_Assert(!fixedType() && !fixedSize());
1870
                m.release();
1871
            }
1872

1873
            if( dims == 2 && m.dims == 2 && m.data &&
A
Andrey Kamaev 已提交
1874
                m.type() == mtype && m.rows == sizes[1] && m.cols == sizes[0] )
1875 1876
                return;
        }
1877 1878 1879

        if(fixedType())
        {
A
Andrey Kamaev 已提交
1880 1881
            if(CV_MAT_CN(mtype) == m.channels() && ((1 << CV_MAT_TYPE(flags)) & fixedDepthMask) != 0 )
                mtype = m.type();
1882
            else
A
Andrey Kamaev 已提交
1883
                CV_Assert(!fixedType() || (CV_MAT_CN(mtype) == m.channels() && ((1 << CV_MAT_TYPE(flags)) & fixedDepthMask) != 0));
1884 1885 1886 1887 1888
        }
        if(fixedSize())
        {
            CV_Assert(m.dims == dims);
            for(int j = 0; j < dims; ++j)
A
Andrey Kamaev 已提交
1889
                CV_Assert(m.size[j] == sizes[j]);
1890 1891
        }

A
Andrey Kamaev 已提交
1892
        m.create(dims, sizes, mtype);
1893
    }
1894 1895

    CV_Error(Error::StsNotImplemented, "Unknown/unsupported array type");
1896
}
1897

1898
void _OutputArray::release() const
1899
{
1900 1901
    CV_Assert(!fixedSize());

1902
    int k = kind();
1903

1904 1905 1906 1907 1908
    if( k == MAT )
    {
        ((Mat*)obj)->release();
        return;
    }
1909

1910 1911
    if( k == GPU_MAT )
    {
1912
        ((cuda::GpuMat*)obj)->release();
1913 1914 1915
        return;
    }

1916 1917
    if( k == CUDA_MEM )
    {
1918
        ((cuda::CudaMem*)obj)->release();
1919 1920 1921
        return;
    }

1922 1923
    if( k == OPENGL_BUFFER )
    {
1924
        ((ogl::Buffer*)obj)->release();
1925 1926 1927
        return;
    }

1928 1929
    if( k == NONE )
        return;
1930

1931 1932 1933 1934 1935
    if( k == STD_VECTOR )
    {
        create(Size(), CV_MAT_TYPE(flags));
        return;
    }
1936

1937 1938
    if( k == STD_VECTOR_VECTOR )
    {
1939
        ((std::vector<std::vector<uchar> >*)obj)->clear();
1940 1941
        return;
    }
1942

1943
    if( k == STD_VECTOR_MAT )
1944
    {
1945
        ((std::vector<Mat>*)obj)->clear();
1946
    }
1947 1948

    CV_Error(Error::StsNotImplemented, "Unknown/unsupported array type");
1949 1950
}

1951
void _OutputArray::clear() const
1952 1953
{
    int k = kind();
1954

1955 1956
    if( k == MAT )
    {
1957
        CV_Assert(!fixedSize());
1958 1959 1960
        ((Mat*)obj)->resize(0);
        return;
    }
1961

1962 1963
    release();
}
1964

1965
bool _OutputArray::needed() const
1966 1967 1968 1969
{
    return kind() != NONE;
}

1970
Mat& _OutputArray::getMatRef(int i) const
1971 1972 1973 1974 1975 1976 1977 1978 1979 1980
{
    int k = kind();
    if( i < 0 )
    {
        CV_Assert( k == MAT );
        return *(Mat*)obj;
    }
    else
    {
        CV_Assert( k == STD_VECTOR_MAT );
1981
        std::vector<Mat>& v = *(std::vector<Mat>*)obj;
1982 1983 1984 1985
        CV_Assert( i < (int)v.size() );
        return v[i];
    }
}
1986

1987
cuda::GpuMat& _OutputArray::getGpuMatRef() const
1988 1989 1990
{
    int k = kind();
    CV_Assert( k == GPU_MAT );
1991
    return *(cuda::GpuMat*)obj;
1992 1993
}

1994
ogl::Buffer& _OutputArray::getOGlBufferRef() const
1995 1996 1997
{
    int k = kind();
    CV_Assert( k == OPENGL_BUFFER );
1998
    return *(ogl::Buffer*)obj;
1999 2000
}

2001
cuda::CudaMem& _OutputArray::getCudaMemRef() const
2002 2003 2004
{
    int k = kind();
    CV_Assert( k == CUDA_MEM );
2005
    return *(cuda::CudaMem*)obj;
2006 2007
}

2008 2009
static _InputOutputArray _none;
InputOutputArray noArray() { return _none; }
2010

2011 2012
}

2013 2014 2015
/*************************************************************************************************\
                                        Matrix Operations
\*************************************************************************************************/
2016

2017
void cv::hconcat(const Mat* src, size_t nsrc, OutputArray _dst)
2018 2019 2020
{
    if( nsrc == 0 || !src )
    {
2021
        _dst.release();
2022 2023
        return;
    }
2024

2025 2026 2027 2028 2029 2030 2031 2032 2033
    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;
    }
2034 2035
    _dst.create( src[0].rows, totalCols, src[0].type());
    Mat dst = _dst.getMat();
2036 2037
    for( i = 0; i < nsrc; i++ )
    {
2038
        Mat dpart = dst(Rect(cols, 0, src[i].cols, src[i].rows));
2039 2040 2041 2042
        src[i].copyTo(dpart);
        cols += src[i].cols;
    }
}
2043

2044
void cv::hconcat(InputArray src1, InputArray src2, OutputArray dst)
2045
{
2046
    Mat src[] = {src1.getMat(), src2.getMat()};
2047 2048
    hconcat(src, 2, dst);
}
2049

2050
void cv::hconcat(InputArray _src, OutputArray dst)
2051
{
2052
    std::vector<Mat> src;
2053
    _src.getMatVector(src);
2054 2055 2056
    hconcat(!src.empty() ? &src[0] : 0, src.size(), dst);
}

2057
void cv::vconcat(const Mat* src, size_t nsrc, OutputArray _dst)
2058 2059 2060
{
    if( nsrc == 0 || !src )
    {
2061
        _dst.release();
2062 2063
        return;
    }
2064

2065 2066 2067 2068 2069 2070 2071 2072 2073
    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;
    }
2074 2075
    _dst.create( totalRows, src[0].cols, src[0].type());
    Mat dst = _dst.getMat();
2076 2077 2078 2079 2080 2081 2082
    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;
    }
}
2083

2084
void cv::vconcat(InputArray src1, InputArray src2, OutputArray dst)
2085
{
2086
    Mat src[] = {src1.getMat(), src2.getMat()};
2087
    vconcat(src, 2, dst);
2088
}
2089

2090
void cv::vconcat(InputArray _src, OutputArray dst)
2091
{
2092
    std::vector<Mat> src;
2093
    _src.getMatVector(src);
2094 2095
    vconcat(!src.empty() ? &src[0] : 0, src.size(), dst);
}
2096

2097
//////////////////////////////////////// set identity ////////////////////////////////////////////
2098
void cv::setIdentity( InputOutputArray _m, const Scalar& s )
2099
{
2100
    Mat m = _m.getMat();
V
Vadim Pisarevsky 已提交
2101
    CV_Assert( m.dims <= 2 );
2102
    int i, j, rows = m.rows, cols = m.cols, type = m.type();
2103

2104 2105 2106 2107 2108 2109 2110 2111 2112 2113 2114 2115 2116 2117 2118 2119 2120 2121 2122 2123 2124 2125 2126 2127 2128 2129 2130 2131 2132 2133 2134 2135 2136
    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;
    }
}

2137 2138
//////////////////////////////////////////// trace ///////////////////////////////////////////

2139
cv::Scalar cv::trace( InputArray _m )
2140
{
2141
    Mat m = _m.getMat();
V
Vadim Pisarevsky 已提交
2142
    CV_Assert( m.dims <= 2 );
2143 2144
    int i, type = m.type();
    int nm = std::min(m.rows, m.cols);
2145

2146 2147 2148 2149 2150 2151 2152 2153 2154
    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;
    }
2155

2156 2157 2158 2159 2160 2161 2162 2163 2164
    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;
    }
2165

2166 2167 2168
    return cv::sum(m.diag());
}

2169
////////////////////////////////////// transpose /////////////////////////////////////////
2170 2171 2172 2173

namespace cv
{

2174
template<typename T> static void
2175
transpose_( const uchar* src, size_t sstep, uchar* dst, size_t dstep, Size sz )
2176
{
V
Victoria Zhislina 已提交
2177 2178
    int i=0, j, m = sz.width, n = sz.height;

2179
    #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
2180
    for(; i <= m - 4; i += 4 )
2181 2182 2183 2184 2185
    {
        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));
2186

2187 2188 2189 2190 2191 2192
        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));
2193

2194 2195 2196 2197 2198
            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];
        }
2199

2200 2201 2202 2203 2204 2205
        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 已提交
2206
    #endif
2207 2208 2209
    for( ; i < m; i++ )
    {
        T* d0 = (T*)(dst + dstep*i);
V
Victoria Zhislina 已提交
2210
        j = 0;
2211
        #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
2212
        for(; j <= n - 4; j += 4 )
2213 2214 2215 2216 2217
        {
            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));
2218

2219 2220
            d0[j] = s0[0]; d0[j+1] = s1[0]; d0[j+2] = s2[0]; d0[j+3] = s3[0];
        }
V
Victoria Zhislina 已提交
2221
        #endif
2222 2223 2224 2225 2226 2227 2228
        for( ; j < n; j++ )
        {
            const T* s0 = (const T*)(src + i*sizeof(T) + j*sstep);
            d0[j] = s0[0];
        }
    }
}
2229

2230 2231 2232 2233 2234
template<typename T> static void
transposeI_( uchar* data, size_t step, int n )
{
    int i, j;
    for( i = 0; i < n; i++ )
2235 2236 2237
    {
        T* row = (T*)(data + step*i);
        uchar* data1 = data + i*sizeof(T);
2238
        for( j = i+1; j < n; j++ )
2239 2240 2241
            std::swap( row[j], *(T*)(data1 + step*j) );
    }
}
2242

2243 2244
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 );
2245

2246 2247 2248 2249 2250 2251
#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); }
2252

2253 2254 2255 2256 2257 2258 2259 2260 2261 2262
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)
2263

2264 2265 2266 2267 2268 2269
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
};
2270

2271 2272 2273 2274 2275 2276
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
};
2277

2278
}
2279

2280
void cv::transpose( InputArray _src, OutputArray _dst )
2281 2282
{
    Mat src = _src.getMat();
2283 2284 2285 2286 2287
    if( src.empty() )
    {
        _dst.release();
        return;
    }
2288
    size_t esz = src.elemSize();
V
Vadim Pisarevsky 已提交
2289
    CV_Assert( src.dims <= 2 && esz <= (size_t)32 );
2290

2291 2292
    _dst.create(src.cols, src.rows, src.type());
    Mat dst = _dst.getMat();
2293

2294 2295 2296
    // handle the case of single-column/single-row matrices, stored in STL vectors.
    if( src.rows != dst.cols || src.cols != dst.rows )
    {
2297
        CV_Assert( src.size() == dst.size() && (src.cols == 1 || src.rows == 1) );
2298 2299 2300 2301
        src.copyTo(dst);
        return;
    }

2302
    if( dst.data == src.data )
2303
    {
2304
        TransposeInplaceFunc func = transposeInplaceTab[esz];
2305
        CV_Assert( func != 0 );
2306
        func( dst.data, dst.step, dst.rows );
2307 2308 2309
    }
    else
    {
2310
        TransposeFunc func = transposeTab[esz];
2311
        CV_Assert( func != 0 );
2312
        func( src.data, src.step, dst.data, dst.step, src.size() );
2313 2314 2315 2316
    }
}


2317
void cv::completeSymm( InputOutputArray _m, bool LtoR )
2318
{
2319
    Mat m = _m.getMat();
V
Vadim Pisarevsky 已提交
2320
    CV_Assert( m.dims <= 2 );
2321

V
Vadim Pisarevsky 已提交
2322
    int i, j, nrows = m.rows, type = m.type();
2323
    int j0 = 0, j1 = nrows;
V
Vadim Pisarevsky 已提交
2324
    CV_Assert( m.rows == m.cols );
2325 2326 2327

    if( type == CV_32FC1 || type == CV_32SC1 )
    {
V
Vadim Pisarevsky 已提交
2328 2329
        int* data = (int*)m.data;
        size_t step = m.step/sizeof(data[0]);
2330 2331 2332 2333 2334 2335 2336 2337 2338
        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 已提交
2339 2340
        double* data = (double*)m.data;
        size_t step = m.step/sizeof(data[0]);
2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351
        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, "" );
}

2352

2353
cv::Mat cv::Mat::cross(InputArray _m) const
2354
{
2355
    Mat m = _m.getMat();
A
Andrey Kamaev 已提交
2356 2357
    int tp = type(), d = CV_MAT_DEPTH(tp);
    CV_Assert( dims <= 2 && m.dims <= 2 && size() == m.size() && tp == m.type() &&
2358
        ((rows == 3 && cols == 1) || (cols*channels() == 3 && rows == 1)));
A
Andrey Kamaev 已提交
2359
    Mat result(rows, cols, tp);
2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387

    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;
}


2388
////////////////////////////////////////// reduce ////////////////////////////////////////////
2389

2390 2391 2392
namespace cv
{

2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411 2412
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 已提交
2413
        i = 0;
2414
        #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
2415
        for(; i <= size.width - 4; i += 4 )
2416 2417 2418 2419 2420 2421 2422 2423 2424 2425
        {
            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 已提交
2426
        #endif
2427 2428 2429 2430 2431 2432 2433 2434 2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452 2453 2454 2455 2456 2457 2458 2459 2460 2461 2462 2463 2464 2465 2466
        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 )
                {
2467
                    a0 = op(a0, (WT)src[i+k]);
2468 2469
                }
                a0 = op(a0, a1);
2470
              dst[k] = (ST)a0;
2471 2472
            }
        }
2473
    }
2474 2475 2476 2477
}

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

2478
}
2479 2480 2481 2482 2483 2484 2485 2486 2487 2488 2489 2490 2491 2492 2493 2494 2495 2496 2497 2498 2499 2500 2501 2502 2503 2504 2505 2506 2507 2508 2509 2510 2511 2512 2513 2514 2515 2516 2517 2518 2519 2520 2521 2522 2523 2524 2525

#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> >

2526
void cv::reduce(InputArray _src, OutputArray _dst, int dim, int op, int dtype)
2527
{
2528
    Mat src = _src.getMat();
V
Vadim Pisarevsky 已提交
2529
    CV_Assert( src.dims <= 2 );
2530
    int op0 = op;
2531
    int stype = src.type(), sdepth = src.depth(), cn = src.channels();
2532
    if( dtype < 0 )
2533
        dtype = _dst.fixedType() ? _dst.type() : stype;
2534 2535
    int ddepth = CV_MAT_DEPTH(dtype);

2536 2537 2538
    _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;
2539

2540
    CV_Assert( op == CV_REDUCE_SUM || op == CV_REDUCE_MAX ||
2541
               op == CV_REDUCE_MIN || op == CV_REDUCE_AVG );
2542 2543 2544 2545 2546 2547
    CV_Assert( src.channels() == dst.channels() );

    if( op == CV_REDUCE_AVG )
    {
        op = CV_REDUCE_SUM;
        if( sdepth < CV_32S && ddepth < CV_32S )
2548
        {
2549
            temp.create(dst.rows, dst.cols, CV_32SC(cn));
2550 2551
            ddepth = CV_32S;
        }
2552 2553 2554 2555 2556 2557 2558
    }

    ReduceFunc func = 0;
    if( dim == 0 )
    {
        if( op == CV_REDUCE_SUM )
        {
2559
            if(sdepth == CV_8U && ddepth == CV_32S)
2560
                func = GET_OPTIMIZED(reduceSumR8u32s);
2561
            else if(sdepth == CV_8U && ddepth == CV_32F)
2562
                func = reduceSumR8u32f;
2563
            else if(sdepth == CV_8U && ddepth == CV_64F)
2564
                func = reduceSumR8u64f;
2565
            else if(sdepth == CV_16U && ddepth == CV_32F)
2566
                func = reduceSumR16u32f;
2567
            else if(sdepth == CV_16U && ddepth == CV_64F)
2568
                func = reduceSumR16u64f;
2569
            else if(sdepth == CV_16S && ddepth == CV_32F)
2570
                func = reduceSumR16s32f;
2571
            else if(sdepth == CV_16S && ddepth == CV_64F)
2572 2573 2574
                func = reduceSumR16s64f;
            else if(sdepth == CV_32F && ddepth == CV_32F)
                func = GET_OPTIMIZED(reduceSumR32f32f);
2575
            else if(sdepth == CV_32F && ddepth == CV_64F)
2576
                func = reduceSumR32f64f;
2577
            else if(sdepth == CV_64F && ddepth == CV_64F)
2578
                func = reduceSumR64f64f;
2579 2580 2581
        }
        else if(op == CV_REDUCE_MAX)
        {
2582
            if(sdepth == CV_8U && ddepth == CV_8U)
2583 2584 2585
                func = GET_OPTIMIZED(reduceMaxR8u);
            else if(sdepth == CV_16U && ddepth == CV_16U)
                func = reduceMaxR16u;
2586
            else if(sdepth == CV_16S && ddepth == CV_16S)
2587
                func = reduceMaxR16s;
2588
            else if(sdepth == CV_32F && ddepth == CV_32F)
2589 2590 2591
                func = GET_OPTIMIZED(reduceMaxR32f);
            else if(sdepth == CV_64F && ddepth == CV_64F)
                func = reduceMaxR64f;
2592 2593 2594
        }
        else if(op == CV_REDUCE_MIN)
        {
2595
            if(sdepth == CV_8U && ddepth == CV_8U)
2596
                func = GET_OPTIMIZED(reduceMinR8u);
2597
            else if(sdepth == CV_16U && ddepth == CV_16U)
2598
                func = reduceMinR16u;
2599
            else if(sdepth == CV_16S && ddepth == CV_16S)
2600
                func = reduceMinR16s;
2601
            else if(sdepth == CV_32F && ddepth == CV_32F)
2602
                func = GET_OPTIMIZED(reduceMinR32f);
2603
            else if(sdepth == CV_64F && ddepth == CV_64F)
2604
                func = reduceMinR64f;
2605 2606 2607 2608 2609 2610
        }
    }
    else
    {
        if(op == CV_REDUCE_SUM)
        {
2611
            if(sdepth == CV_8U && ddepth == CV_32S)
2612
                func = GET_OPTIMIZED(reduceSumC8u32s);
2613
            else if(sdepth == CV_8U && ddepth == CV_32F)
2614
                func = reduceSumC8u32f;
2615
            else if(sdepth == CV_8U && ddepth == CV_64F)
2616
                func = reduceSumC8u64f;
2617
            else if(sdepth == CV_16U && ddepth == CV_32F)
2618
                func = reduceSumC16u32f;
2619
            else if(sdepth == CV_16U && ddepth == CV_64F)
2620
                func = reduceSumC16u64f;
2621
            else if(sdepth == CV_16S && ddepth == CV_32F)
2622
                func = reduceSumC16s32f;
2623
            else if(sdepth == CV_16S && ddepth == CV_64F)
2624 2625 2626
                func = reduceSumC16s64f;
            else if(sdepth == CV_32F && ddepth == CV_32F)
                func = GET_OPTIMIZED(reduceSumC32f32f);
2627
            else if(sdepth == CV_32F && ddepth == CV_64F)
2628
                func = reduceSumC32f64f;
2629
            else if(sdepth == CV_64F && ddepth == CV_64F)
2630
                func = reduceSumC64f64f;
2631 2632 2633
        }
        else if(op == CV_REDUCE_MAX)
        {
2634
            if(sdepth == CV_8U && ddepth == CV_8U)
2635 2636 2637
                func = GET_OPTIMIZED(reduceMaxC8u);
            else if(sdepth == CV_16U && ddepth == CV_16U)
                func = reduceMaxC16u;
2638
            else if(sdepth == CV_16S && ddepth == CV_16S)
2639
                func = reduceMaxC16s;
2640
            else if(sdepth == CV_32F && ddepth == CV_32F)
2641
                func = GET_OPTIMIZED(reduceMaxC32f);
2642
            else if(sdepth == CV_64F && ddepth == CV_64F)
2643
                func = reduceMaxC64f;
2644 2645 2646
        }
        else if(op == CV_REDUCE_MIN)
        {
2647
            if(sdepth == CV_8U && ddepth == CV_8U)
2648
                func = GET_OPTIMIZED(reduceMinC8u);
2649
            else if(sdepth == CV_16U && ddepth == CV_16U)
2650
                func = reduceMinC16u;
2651
            else if(sdepth == CV_16S && ddepth == CV_16S)
2652
                func = reduceMinC16s;
2653
            else if(sdepth == CV_32F && ddepth == CV_32F)
2654
                func = GET_OPTIMIZED(reduceMinC32f);
2655
            else if(sdepth == CV_64F && ddepth == CV_64F)
2656
                func = reduceMinC64f;
2657 2658 2659 2660 2661
        }
    }

    if( !func )
        CV_Error( CV_StsUnsupportedFormat,
2662
                  "Unsupported combination of input and output array formats" );
2663 2664 2665

    func( src, temp );

2666
    if( op0 == CV_REDUCE_AVG )
2667
        temp.convertTo(dst, dst.type(), 1./(dim == 0 ? src.rows : src.cols));
2668
}
2669 2670


2671
//////////////////////////////////////// sort ///////////////////////////////////////////
2672

2673 2674 2675
namespace cv
{

2676 2677 2678 2679 2680 2681 2682 2683
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;
2684

2685 2686 2687 2688 2689 2690 2691 2692 2693 2694 2695 2696 2697 2698 2699 2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712
    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];
        }
2713
        std::sort( ptr, ptr + len );
2714 2715 2716 2717 2718 2719 2720 2721 2722
        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];
    }
}

2723 2724 2725 2726 2727 2728 2729 2730 2731
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;
};


2732 2733 2734 2735 2736 2737 2738 2739 2740 2741 2742 2743

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 );
2744

2745 2746 2747 2748 2749 2750 2751 2752 2753 2754 2755 2756 2757 2758 2759 2760 2761 2762 2763 2764 2765 2766 2767 2768 2769 2770 2771 2772 2773 2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784
    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);

2785
}
2786

2787
void cv::sort( InputArray _src, OutputArray _dst, int flags )
2788 2789 2790 2791 2792 2793
{
    static SortFunc tab[] =
    {
        sort_<uchar>, sort_<schar>, sort_<ushort>, sort_<short>,
        sort_<int>, sort_<float>, sort_<double>, 0
    };
2794
    Mat src = _src.getMat();
2795
    SortFunc func = tab[src.depth()];
V
Vadim Pisarevsky 已提交
2796
    CV_Assert( src.dims <= 2 && src.channels() == 1 && func != 0 );
2797 2798
    _dst.create( src.size(), src.type() );
    Mat dst = _dst.getMat();
2799 2800 2801
    func( src, dst, flags );
}

2802
void cv::sortIdx( InputArray _src, OutputArray _dst, int flags )
2803 2804 2805 2806 2807 2808
{
    static SortFunc tab[] =
    {
        sortIdx_<uchar>, sortIdx_<schar>, sortIdx_<ushort>, sortIdx_<short>,
        sortIdx_<int>, sortIdx_<float>, sortIdx_<double>, 0
    };
2809
    Mat src = _src.getMat();
2810
    SortFunc func = tab[src.depth()];
V
Vadim Pisarevsky 已提交
2811
    CV_Assert( src.dims <= 2 && src.channels() == 1 && func != 0 );
2812

2813
    Mat dst = _dst.getMat();
2814
    if( dst.data == src.data )
2815 2816 2817
        _dst.release();
    _dst.create( src.size(), CV_32S );
    dst = _dst.getMat();
2818 2819
    func( src, dst, flags );
}
2820 2821


2822
////////////////////////////////////////// kmeans ////////////////////////////////////////////
2823 2824 2825 2826

namespace cv
{

2827
static void generateRandomCenter(const std::vector<Vec2f>& box, float* center, RNG& rng)
2828 2829 2830 2831 2832 2833 2834
{
    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];
}

2835
class KMeansPPDistanceComputer : public ParallelLoopBody
2836 2837 2838 2839 2840 2841 2842 2843 2844 2845 2846 2847 2848 2849 2850
{
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) { }

2851
    void operator()( const cv::Range& range ) const
2852
    {
2853 2854
        const int begin = range.start;
        const int end = range.end;
2855 2856 2857 2858 2859 2860 2861 2862

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

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

2865 2866 2867 2868 2869 2870 2871
    float *tdist2;
    const float *data;
    const float *dist;
    const int dims;
    const size_t step;
    const size_t stepci;
};
2872 2873 2874 2875 2876 2877 2878 2879 2880 2881

/*
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);
2882
    size_t step = _data.step/sizeof(data[0]);
2883
    std::vector<int> _centers(K);
2884
    int* centers = &_centers[0];
2885
    std::vector<float> _dist(N*3);
2886 2887 2888 2889 2890 2891 2892
    float* dist = &_dist[0], *tdist = dist + N, *tdist2 = tdist + N;
    double sum0 = 0;

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

    for( i = 0; i < N; i++ )
    {
2893
        dist[i] = normL2Sqr_(data + step*i, data + step*centers[0], dims);
2894 2895
        sum0 += dist[i];
    }
2896

2897 2898 2899 2900 2901 2902 2903 2904 2905 2906 2907 2908
    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;
2909

2910
            parallel_for_(Range(0, N),
2911
                         KMeansPPDistanceComputer(tdist2, data, dist, dims, step, step*ci));
2912 2913 2914 2915
            for( i = 0; i < N; i++ )
            {
                s += tdist2[i];
            }
2916

2917 2918 2919 2920 2921 2922 2923 2924 2925 2926 2927 2928 2929 2930 2931 2932 2933 2934 2935 2936 2937
            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];
    }
}

2938
class KMeansDistanceComputer : public ParallelLoopBody
2939 2940 2941 2942 2943 2944 2945 2946 2947 2948 2949 2950 2951
{
public:
    KMeansDistanceComputer( double *_distances,
                            int *_labels,
                            const Mat& _data,
                            const Mat& _centers )
        : distances(_distances),
          labels(_labels),
          data(_data),
          centers(_centers)
    {
    }

2952
    void operator()( const Range& range ) const
2953
    {
2954 2955
        const int begin = range.start;
        const int end = range.end;
2956 2957 2958 2959 2960 2961 2962 2963 2964 2965 2966 2967 2968 2969 2970 2971 2972 2973 2974 2975 2976 2977 2978 2979 2980 2981 2982 2983
        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:
2984 2985
    KMeansDistanceComputer& operator=(const KMeansDistanceComputer&); // to quiet MSVC

2986 2987 2988 2989 2990 2991
    double *distances;
    int *labels;
    const Mat& data;
    const Mat& centers;
};

2992
}
2993

2994
double cv::kmeans( InputArray _data, int K,
2995 2996 2997
                   InputOutputArray _bestLabels,
                   TermCriteria criteria, int attempts,
                   int flags, OutputArray _centers )
2998 2999
{
    const int SPP_TRIALS = 3;
3000
    Mat data = _data.getMat();
3001 3002 3003
    bool isrow = data.rows == 1 && data.channels() > 1;
    int N = !isrow ? data.rows : data.cols;
    int dims = (!isrow ? data.cols : 1)*data.channels();
3004 3005 3006
    int type = data.depth();

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

3010
    _bestLabels.create(N, 1, CV_32S, -1, true);
3011

3012
    Mat _labels, best_labels = _bestLabels.getMat();
3013 3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030 3031
    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>();

3032
    Mat centers(K, dims, type), old_centers(K, dims, type), temp(1, dims, type);
3033 3034
    std::vector<int> counters(K);
    std::vector<Vec2f> _box(dims);
3035 3036 3037 3038 3039 3040 3041 3042 3043 3044 3045 3046 3047 3048 3049 3050 3051 3052 3053 3054 3055 3056 3057 3058 3059 3060 3061 3062 3063 3064 3065 3066 3067 3068 3069 3070 3071 3072 3073 3074
    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;
3075
        for( iter = 0;; )
3076 3077 3078 3079 3080 3081 3082 3083 3084 3085 3086 3087 3088 3089 3090 3091 3092 3093 3094 3095
        {
            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 );
                }
3096

3097 3098 3099 3100 3101 3102 3103 3104 3105 3106
                // 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);
3107 3108
                    j=0;
                    #if CV_ENABLE_UNROLLED
V
Victoria Zhislina 已提交
3109
                    for(; j <= dims - 4; j += 4 )
3110 3111 3112 3113 3114 3115 3116 3117 3118 3119 3120 3121 3122
                    {
                        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 已提交
3123
                    #endif
3124 3125 3126 3127 3128 3129 3130
                    for( ; j < dims; j++ )
                        center[j] += sample[j];
                    counters[k]++;
                }

                if( iter > 0 )
                    max_center_shift = 0;
3131

3132 3133 3134
                for( k = 0; k < K; k++ )
                {
                    if( counters[k] != 0 )
3135 3136 3137 3138 3139 3140 3141
                        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 已提交
3142
                    for( int k1 = 1; k1 < K; k1++ )
3143 3144 3145 3146
                    {
                        if( counters[max_k] < counters[k1] )
                            max_k = k1;
                    }
3147 3148

                    double max_dist = 0;
3149 3150 3151
                    int farthest_i = -1;
                    float* new_center = centers.ptr<float>(k);
                    float* old_center = centers.ptr<float>(max_k);
3152 3153 3154 3155
                    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;
3156

3157 3158 3159 3160 3161
                    for( i = 0; i < N; i++ )
                    {
                        if( labels[i] != max_k )
                            continue;
                        sample = data.ptr<float>(i);
3162
                        double dist = normL2Sqr_(sample, _old_center, dims);
3163

3164 3165 3166 3167 3168 3169
                        if( max_dist <= dist )
                        {
                            max_dist = dist;
                            farthest_i = i;
                        }
                    }
3170

3171 3172
                    counters[max_k]--;
                    counters[k]++;
3173
                    labels[farthest_i] = k;
3174
                    sample = data.ptr<float>(farthest_i);
3175

3176
                    for( j = 0; j < dims; j++ )
3177
                    {
3178 3179
                        old_center[j] -= sample[j];
                        new_center[j] += sample[j];
3180
                    }
3181 3182 3183 3184 3185 3186 3187 3188 3189 3190
                }

                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;
3191

3192 3193 3194 3195 3196 3197 3198 3199 3200 3201 3202 3203 3204
                    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);
                    }
                }
            }
3205

3206 3207
            if( ++iter == MAX(criteria.maxCount, 2) || max_center_shift <= criteria.epsilon )
                break;
3208 3209

            // assign labels
3210 3211
            Mat dists(1, N, CV_64F);
            double* dist = dists.ptr<double>(0);
3212
            parallel_for_(Range(0, N),
3213
                         KMeansDistanceComputer(dist, labels, data, centers));
3214 3215 3216
            compactness = 0;
            for( i = 0; i < N; i++ )
            {
3217
                compactness += dist[i];
3218 3219 3220 3221 3222 3223
            }
        }

        if( compactness < best_compactness )
        {
            best_compactness = compactness;
3224 3225
            if( _centers.needed() )
                centers.copyTo(_centers);
3226 3227 3228 3229 3230 3231 3232 3233 3234 3235 3236 3237 3238 3239 3240 3241 3242 3243 3244 3245 3246 3247 3248 3249 3250 3251 3252 3253 3254 3255 3256 3257
            _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 已提交
3258
    cv::Mat m = cv::cvarrToMat(matrix);
3259 3260 3261 3262 3263 3264 3265 3266 3267 3268 3269 3270 3271 3272 3273 3274 3275
    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);
3276

3277 3278 3279 3280 3281 3282 3283 3284 3285
    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" );
3286

3287 3288 3289 3290 3291 3292 3293 3294 3295 3296 3297
    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;
3298

3299 3300 3301 3302 3303 3304
    CvMat stub, *mat = (CvMat*)arr;
    double delta;
    int type, step;
    double val = start;
    int i, j;
    int rows, cols;
3305

3306 3307 3308 3309 3310 3311 3312 3313 3314 3315 3316 3317 3318 3319 3320 3321 3322 3323 3324 3325 3326 3327 3328 3329 3330 3331 3332 3333 3334 3335 3336 3337 3338 3339 3340 3341 3342 3343 3344 3345 3346 3347 3348 3349 3350 3351 3352 3353 3354 3355 3356 3357 3358 3359
    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 已提交
3360
    cv::Mat src = cv::cvarrToMat(_src);
3361

3362 3363 3364 3365 3366 3367 3368 3369 3370 3371 3372 3373 3374 3375 3376 3377 3378 3379 3380 3381 3382 3383 3384 3385 3386
    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 )
3387
    {
3388
        centers = cv::cvarrToMat(_centers);
M
Maria Dimashova 已提交
3389

3390
        centers = centers.reshape(1);
M
Maria Dimashova 已提交
3391 3392 3393 3394 3395 3396
        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() );
3397
    }
3398 3399 3400
    CV_Assert( labels.isContinuous() && labels.type() == CV_32S &&
        (labels.cols == 1 || labels.rows == 1) &&
        labels.cols + labels.rows - 1 == data.rows );
3401

3402
    double compactness = cv::kmeans(data, cluster_count, labels, termcrit, attempts,
3403
                                    flags, _centers ? cv::_OutputArray(centers) : cv::_OutputArray() );
3404 3405 3406 3407 3408 3409 3410 3411 3412 3413
    if( _compactness )
        *_compactness = compactness;
    return 1;
}

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

namespace cv
{

3414
Mat Mat::reshape(int _cn, int _newndims, const int* _newsz) const
3415
{
3416 3417 3418 3419 3420 3421 3422 3423
    if(_newndims == dims)
    {
        if(_newsz == 0)
            return reshape(_cn);
        if(_newndims == 2)
            return reshape(_cn, _newsz[0]);
    }

V
Vadim Pisarevsky 已提交
3424 3425 3426 3427
    CV_Error(CV_StsNotImplemented, "");
    // TBD
    return Mat();
}
3428

V
Vadim Pisarevsky 已提交
3429
NAryMatIterator::NAryMatIterator()
3430
    : arrays(0), planes(0), ptrs(0), narrays(0), nplanes(0), size(0), iterdepth(0), idx(0)
V
Vadim Pisarevsky 已提交
3431 3432
{
}
3433

V
Vadim Pisarevsky 已提交
3434
NAryMatIterator::NAryMatIterator(const Mat** _arrays, Mat* _planes, int _narrays)
3435 3436 3437
: arrays(0), planes(0), ptrs(0), narrays(0), nplanes(0), size(0), iterdepth(0), idx(0)
{
    init(_arrays, _planes, 0, _narrays);
3438 3439
}

3440 3441
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 已提交
3442
{
3443
    init(_arrays, 0, _ptrs, _narrays);
V
Vadim Pisarevsky 已提交
3444
}
3445

3446
void NAryMatIterator::init(const Mat** _arrays, Mat* _planes, uchar** _ptrs, int _narrays)
V
Vadim Pisarevsky 已提交
3447
{
3448 3449
    CV_Assert( _arrays && (_ptrs || _planes) );
    int i, j, d1=0, i0 = -1, d = -1;
3450

V
Vadim Pisarevsky 已提交
3451
    arrays = _arrays;
3452
    ptrs = _ptrs;
V
Vadim Pisarevsky 已提交
3453 3454 3455
    planes = _planes;
    narrays = _narrays;
    nplanes = 0;
3456
    size = 0;
3457

V
Vadim Pisarevsky 已提交
3458
    if( narrays < 0 )
3459
    {
V
Vadim Pisarevsky 已提交
3460 3461 3462 3463
        for( i = 0; _arrays[i] != 0; i++ )
            ;
        narrays = i;
        CV_Assert(narrays <= 1000);
3464
    }
V
Vadim Pisarevsky 已提交
3465 3466 3467 3468

    iterdepth = 0;

    for( i = 0; i < narrays; i++ )
3469
    {
V
Vadim Pisarevsky 已提交
3470 3471
        CV_Assert(arrays[i] != 0);
        const Mat& A = *arrays[i];
3472 3473
        if( ptrs )
            ptrs[i] = A.data;
3474

3475 3476
        if( !A.data )
            continue;
3477

V
Vadim Pisarevsky 已提交
3478
        if( i0 < 0 )
3479
        {
V
Vadim Pisarevsky 已提交
3480 3481
            i0 = i;
            d = A.dims;
3482

V
Vadim Pisarevsky 已提交
3483 3484 3485 3486 3487 3488 3489 3490 3491 3492 3493 3494 3495 3496 3497 3498
            // 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);
3499 3500
        }
    }
V
Vadim Pisarevsky 已提交
3501 3502

    if( i0 >= 0 )
3503
    {
3504
        size = arrays[i0]->size[d-1];
V
Vadim Pisarevsky 已提交
3505 3506
        for( j = d-1; j > iterdepth; j-- )
        {
3507
            int64 total1 = (int64)size*arrays[i0]->size[j-1];
V
Vadim Pisarevsky 已提交
3508 3509
            if( total1 != (int)total1 )
                break;
3510
            size = (int)total1;
V
Vadim Pisarevsky 已提交
3511 3512 3513 3514 3515
        }

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

V
Vadim Pisarevsky 已提交
3517 3518 3519
        nplanes = 1;
        for( j = iterdepth-1; j >= 0; j-- )
            nplanes *= arrays[i0]->size[j];
3520
    }
V
Vadim Pisarevsky 已提交
3521
    else
3522
        iterdepth = 0;
3523

3524
    idx = 0;
3525

3526 3527
    if( !planes )
        return;
3528

V
Vadim Pisarevsky 已提交
3529
    for( i = 0; i < narrays; i++ )
3530
    {
3531 3532
        CV_Assert(arrays[i] != 0);
        const Mat& A = *arrays[i];
3533

3534
        if( !A.data )
V
Vadim Pisarevsky 已提交
3535 3536 3537 3538
        {
            planes[i] = Mat();
            continue;
        }
3539 3540

        planes[i] = Mat(1, (int)size, A.type(), A.data);
3541 3542 3543
    }
}

V
Vadim Pisarevsky 已提交
3544 3545

NAryMatIterator& NAryMatIterator::operator ++()
3546 3547 3548 3549
{
    if( idx >= nplanes-1 )
        return *this;
    ++idx;
3550

3551
    if( iterdepth == 1 )
3552
    {
3553 3554 3555 3556 3557 3558 3559 3560 3561 3562
        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 )
3563
        {
3564 3565 3566 3567 3568 3569 3570 3571 3572 3573 3574 3575 3576 3577 3578
            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;
3579
            int _idx = (int)idx;
3580 3581 3582 3583 3584 3585 3586 3587 3588 3589 3590
            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;
3591 3592
        }
    }
3593

3594 3595 3596
    return *this;
}

V
Vadim Pisarevsky 已提交
3597
NAryMatIterator NAryMatIterator::operator ++(int)
3598
{
V
Vadim Pisarevsky 已提交
3599
    NAryMatIterator it = *this;
3600 3601 3602 3603
    ++*this;
    return it;
}

V
Vadim Pisarevsky 已提交
3604 3605 3606
///////////////////////////////////////////////////////////////////////////
//                              MatConstIterator                         //
///////////////////////////////////////////////////////////////////////////
3607

V
Vadim Pisarevsky 已提交
3608
Point MatConstIterator::pos() const
3609
{
V
Vadim Pisarevsky 已提交
3610 3611 3612
    if( !m )
        return Point();
    CV_DbgAssert(m->dims <= 2);
3613

V
Vadim Pisarevsky 已提交
3614 3615 3616
    ptrdiff_t ofs = ptr - m->data;
    int y = (int)(ofs/m->step[0]);
    return Point((int)((ofs - y*m->step[0])/elemSize), y);
3617 3618
}

V
Vadim Pisarevsky 已提交
3619
void MatConstIterator::pos(int* _idx) const
3620
{
V
Vadim Pisarevsky 已提交
3621 3622 3623
    CV_Assert(m != 0 && _idx);
    ptrdiff_t ofs = ptr - m->data;
    for( int i = 0; i < m->dims; i++ )
3624
    {
V
Vadim Pisarevsky 已提交
3625 3626 3627
        size_t s = m->step[i], v = ofs/s;
        ofs -= v*s;
        _idx[i] = (int)v;
3628 3629 3630
    }
}

V
Vadim Pisarevsky 已提交
3631
ptrdiff_t MatConstIterator::lpos() const
3632
{
V
Vadim Pisarevsky 已提交
3633 3634 3635 3636 3637 3638 3639
    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 )
3640
    {
V
Vadim Pisarevsky 已提交
3641 3642
        ptrdiff_t y = ofs/m->step[0];
        return y*m->cols + (ofs - y*m->step[0])/elemSize;
3643
    }
V
Vadim Pisarevsky 已提交
3644 3645
    ptrdiff_t result = 0;
    for( i = 0; i < d; i++ )
3646
    {
V
Vadim Pisarevsky 已提交
3647 3648 3649
        size_t s = m->step[i], v = ofs/s;
        ofs -= v*s;
        result = result*m->size[i] + v;
3650
    }
V
Vadim Pisarevsky 已提交
3651
    return result;
3652
}
3653

V
Vadim Pisarevsky 已提交
3654
void MatConstIterator::seek(ptrdiff_t ofs, bool relative)
3655
{
V
Vadim Pisarevsky 已提交
3656
    if( m->isContinuous() )
3657
    {
V
Vadim Pisarevsky 已提交
3658 3659 3660 3661 3662 3663
        ptr = (relative ? ptr : sliceStart) + ofs*elemSize;
        if( ptr < sliceStart )
            ptr = sliceStart;
        else if( ptr > sliceEnd )
            ptr = sliceEnd;
        return;
3664
    }
3665

V
Vadim Pisarevsky 已提交
3666 3667
    int d = m->dims;
    if( d == 2 )
3668
    {
V
Vadim Pisarevsky 已提交
3669 3670
        ptrdiff_t ofs0, y;
        if( relative )
3671
        {
V
Vadim Pisarevsky 已提交
3672 3673 3674
            ofs0 = ptr - m->data;
            y = ofs0/m->step[0];
            ofs += y*m->cols + (ofs0 - y*m->step[0])/elemSize;
3675
        }
V
Vadim Pisarevsky 已提交
3676 3677 3678
        y = ofs/m->cols;
        int y1 = std::min(std::max((int)y, 0), m->rows-1);
        sliceStart = m->data + y1*m->step[0];
3679
        sliceEnd = sliceStart + m->cols*elemSize;
V
Vadim Pisarevsky 已提交
3680 3681 3682
        ptr = y < 0 ? sliceStart : y >= m->rows ? sliceEnd :
            sliceStart + (ofs - y*m->cols)*elemSize;
        return;
3683
    }
3684

V
Vadim Pisarevsky 已提交
3685 3686
    if( relative )
        ofs += lpos();
3687

V
Vadim Pisarevsky 已提交
3688 3689
    if( ofs < 0 )
        ofs = 0;
3690

V
Vadim Pisarevsky 已提交
3691 3692 3693 3694 3695 3696
    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;
3697

V
Vadim Pisarevsky 已提交
3698
    for( int i = d-2; i >= 0; i-- )
3699
    {
V
Vadim Pisarevsky 已提交
3700 3701 3702 3703 3704
        szi = m->size[i];
        t = ofs/szi;
        v = (int)(ofs - t*szi);
        ofs = t;
        sliceStart += v*m->step[i];
3705
    }
3706

V
Vadim Pisarevsky 已提交
3707 3708 3709 3710 3711
    sliceEnd = sliceStart + m->size[d-1]*elemSize;
    if( ofs > 0 )
        ptr = sliceEnd;
    else
        ptr = sliceStart + (ptr - m->data);
3712
}
3713

V
Vadim Pisarevsky 已提交
3714 3715 3716 3717 3718 3719 3720 3721 3722
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
3723
    {
V
Vadim Pisarevsky 已提交
3724 3725
        for( i = 0; i < d; i++ )
            ofs = ofs*m->size[i] + _idx[i];
3726
    }
V
Vadim Pisarevsky 已提交
3727
    seek(ofs, relative);
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
}

//////////////////////////////// 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 已提交
3756 3757 3758 3759
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)
3760 3761 3762 3763 3764 3765 3766 3767 3768 3769 3770 3771 3772 3773 3774 3775 3776 3777 3778 3779 3780 3781 3782 3783 3784 3785 3786 3787 3788 3789 3790 3791 3792 3793 3794 3795 3796 3797 3798 3799 3800 3801 3802 3803
{
    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 已提交
3804
static ConvertScaleData getConvertScaleElem(int fromType, int toType)
3805 3806 3807 3808 3809 3810 3811 3812 3813 3814 3815 3816 3817 3818 3819 3820 3821 3822 3823 3824 3825 3826 3827 3828 3829 3830 3831 3832 3833 3834 3835 3836 3837 3838 3839 3840 3841 3842 3843 3844 3845 3846 3847 3848 3849 3850 3851 3852 3853
{
    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;
3854
    for( i = 0; i + sizeof(int) <= elemSize; i += sizeof(int) )
3855 3856 3857 3858 3859 3860 3861 3862
        *(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;
3863
    for( i = 0; i + sizeof(int) <= elemSize; i += sizeof(int) )
3864 3865 3866 3867 3868 3869 3870 3871 3872 3873 3874 3875 3876 3877
        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) +
3878
        sizeof(int)*std::max(dims - CV_MAX_DIM, 0), CV_ELEM_SIZE1(_type));
3879 3880
    nodeSize = alignSize(valueOffset +
        CV_ELEM_SIZE(_type), (int)sizeof(size_t));
3881

3882 3883 3884 3885 3886 3887 3888 3889 3890 3891 3892 3893 3894 3895 3896 3897 3898 3899
    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 已提交
3900
SparseMat::SparseMat(const Mat& m)
3901 3902 3903 3904 3905 3906
: 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 已提交
3907
    uchar* dptr = m.data;
3908 3909 3910

    for(;;)
    {
A
Andrey Kamaev 已提交
3911
        for( i = 0; i < lastSize; i++, dptr += esz )
3912
        {
A
Andrey Kamaev 已提交
3913
            if( isZeroElem(dptr, esz) )
3914 3915 3916
                continue;
            idx[d-1] = i;
            uchar* to = newNode(idx, hash(idx));
A
Andrey Kamaev 已提交
3917
            copyElem( dptr, to, esz );
3918
        }
3919

3920 3921
        for( i = d - 2; i >= 0; i-- )
        {
A
Andrey Kamaev 已提交
3922
            dptr += m.step[i] - m.size[i+1]*m.step[i+1];
3923 3924 3925 3926 3927 3928 3929 3930
            if( ++idx[i] < m.size[i] )
                break;
            idx[i] = 0;
        }
        if( i < 0 )
            break;
    }
}
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
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;
    }
4006

4007 4008 4009
    CV_Assert(hdr != 0);
    if( hdr != m.hdr )
        m.create( hdr->dims, hdr->size, rtype );
4010

4011 4012 4013 4014 4015
    SparseMatConstIterator from = begin();
    size_t i, N = nzcount();

    if( alpha == 1 )
    {
4016
        ConvertData cvtfunc = getConvertElem(type(), rtype);
4017 4018 4019 4020
        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);
4021
            cvtfunc( from.ptr, to, cn );
4022 4023 4024 4025
        }
    }
    else
    {
4026
        ConvertScaleData cvtfunc = getConvertScaleElem(type(), rtype);
4027 4028 4029 4030
        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);
4031
            cvtfunc( from.ptr, to, cn, alpha, 0 );
4032 4033 4034 4035 4036 4037 4038 4039 4040 4041 4042
        }
    }
}


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);
4043

4044 4045 4046 4047 4048 4049 4050 4051 4052
    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 )
    {
4053
        ConvertData cvtfunc = getConvertElem(type(), rtype);
4054 4055 4056 4057 4058 4059 4060 4061 4062
        for( i = 0; i < N; i++, ++from )
        {
            const Node* n = from.node();
            uchar* to = m.ptr(n->idx);
            cvtfunc( from.ptr, to, cn );
        }
    }
    else
    {
4063
        ConvertScaleData cvtfunc = getConvertScaleElem(type(), rtype);
4064 4065 4066 4067 4068 4069 4070 4071 4072 4073 4074 4075 4076 4077 4078
        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();
}

4079 4080 4081 4082 4083 4084 4085 4086 4087 4088 4089 4090 4091
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;
    }
4092

4093 4094 4095 4096 4097 4098 4099
    if( createMissing )
    {
        int idx[] = { i0 };
        return newNode( idx, h );
    }
    return 0;
}
4100

4101 4102 4103 4104 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 4149 4150 4151 4152 4153 4154 4155 4156 4157 4158 4159 4160 4161 4162 4163 4164 4165 4166 4167 4168 4169 4170 4171 4172 4173 4174 4175 4176 4177 4178 4179 4180 4181 4182 4183 4184 4185 4186 4187 4188 4189 4190 4191 4192 4193 4194 4195 4196 4197 4198 4199 4200 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
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)
4239
        newsize = (size_t)1 << cvCeil(std::log((double)newsize)/CV_LOG2);
4240 4241

    size_t i, hsize = hdr->hashtab.size();
4242
    std::vector<size_t> _newh(newsize);
4243 4244 4245 4246 4247 4248 4249 4250 4251 4252 4253 4254 4255 4256 4257 4258 4259 4260 4261 4262 4263 4264 4265 4266 4267 4268 4269 4270 4271 4272
    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();
    }
4273

4274 4275 4276 4277 4278 4279 4280 4281 4282 4283 4284 4285 4286 4287 4288 4289 4290 4291 4292 4293 4294 4295
    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];
4296
    size_t esz = elemSize();
4297
    uchar* p = &value<uchar>(elem);
4298
    if( esz == sizeof(float) )
4299
        *((float*)p) = 0.f;
4300
    else if( esz == sizeof(double) )
4301 4302
        *((double*)p) = 0.;
    else
4303
        memset(p, 0, esz);
4304

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
    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;
4331
    const std::vector<size_t>& htab = hdr.hashtab;
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 4359 4360 4361 4362 4363 4364 4365 4366 4367 4368 4369 4370 4371 4372 4373 4374 4375
    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();
4376

4377 4378 4379 4380
    size_t i, N = src.nzcount();
    normType &= NORM_TYPE_MASK;
    int type = src.type();
    double result = 0;
4381

4382
    CV_Assert( normType == NORM_INF || normType == NORM_L1 || normType == NORM_L2 );
4383

4384 4385 4386 4387 4388 4389 4390 4391 4392 4393 4394
    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 )
            {
4395
                double v = *(const float*)it.ptr;
4396 4397 4398 4399 4400 4401 4402 4403 4404 4405 4406 4407 4408 4409
                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 )
            {
4410
                double v = *(const double*)it.ptr;
4411 4412 4413 4414 4415
                result += v*v;
            }
    }
    else
        CV_Error( CV_StsUnsupportedFormat, "Only 32f and 64f are supported" );
4416

4417 4418 4419 4420
    if( normType == NORM_L2 )
        result = std::sqrt(result);
    return result;
}
4421

4422 4423 4424 4425 4426 4427
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;
4428

4429 4430 4431 4432 4433 4434 4435 4436 4437 4438 4439 4440 4441 4442 4443 4444 4445 4446 4447 4448 4449 4450 4451 4452 4453 4454 4455 4456 4457 4458 4459 4460 4461 4462 4463 4464 4465 4466 4467 4468 4469 4470 4471 4472 4473 4474
    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" );
4475

4476 4477 4478 4479 4480 4481 4482 4483
    if( _minidx )
        for( i = 0; i < d; i++ )
            _minidx[i] = minidx[i];
    if( _maxidx )
        for( i = 0; i < d; i++ )
            _maxidx[i] = maxidx[i];
}

4484

4485 4486 4487 4488 4489 4490 4491 4492 4493 4494
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" );
4495

4496 4497
    src.convertTo( dst, -1, scale );
}
4498 4499

////////////////////// RotatedRect //////////////////////
4500

4501 4502 4503 4504 4505
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;
4506

4507 4508 4509 4510 4511 4512 4513 4514 4515 4516
    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;
}

4517
Rect RotatedRect::boundingRect() const
4518 4519 4520
{
    Point2f pt[4];
    points(pt);
4521 4522 4523 4524
    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)));
4525 4526 4527
    r.width -= r.x - 1;
    r.height -= r.y - 1;
    return r;
4528
}
4529 4530 4531

}

A
Andrey Kamaev 已提交
4532 4533 4534 4535 4536 4537 4538 4539 4540 4541 4542 4543 4544 4545 4546 4547 4548 4549 4550 4551 4552 4553 4554 4555 4556 4557 4558 4559 4560 4561 4562 4563 4564 4565 4566 4567 4568 4569 4570 4571 4572 4573 4574 4575 4576 4577 4578 4579 4580 4581 4582 4583 4584 4585
// 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);
    }
}


4586
/* End of file. */