loadsave.cpp 25.2 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47
/*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.
//
//
//                        Intel License Agreement
//                For Open Source Computer Vision Library
//
// Copyright (C) 2000, Intel Corporation, 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 Intel Corporation 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*/

//
//  Loading and saving IPL images.
//

#include "precomp.hpp"
#include "grfmts.hpp"
48 49
#include "utils.hpp"
#include "exif.hpp"
50 51
#undef min
#undef max
52
#include <iostream>
53
#include <fstream>
54 55 56 57

/****************************************************************************************\
*                                      Image Codecs                                      *
\****************************************************************************************/
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 83 84 85 86 87 88 89 90 91 92 93 94 95 96
namespace {

class ByteStreamBuffer: public std::streambuf
{
public:
    ByteStreamBuffer(char* base, size_t length)
    {
        setg(base, base, base + length);
    }

protected:
    virtual pos_type seekoff( off_type offset,
                              std::ios_base::seekdir dir,
                              std::ios_base::openmode )
    {
        // get absolute offset
        off_type off = offset;
        if (dir == std::ios_base::cur)
        {
            off += gptr() - eback();
        }
        else if (dir == std::ios_base::end)
        {
            off += egptr() - eback();
        }

        // check limits
        if (off >= (off_type)0 && off <= egptr() - eback())
        {
            setg(eback(), gptr() + off, egptr());
            return gptr() - eback();
        }

        return -1;
    }
};

}

97 98 99
namespace cv
{

100 101 102 103 104
/**
 * @struct ImageCodecInitializer
 *
 * Container which stores the registered codecs to be used by OpenCV
*/
105 106
struct ImageCodecInitializer
{
107 108 109
    /**
     * Default Constructor for the ImageCodeInitializer
    */
110 111
    ImageCodecInitializer()
    {
112
        /// BMP Support
R
Roman Donchenko 已提交
113 114
        decoders.push_back( makePtr<BmpDecoder>() );
        encoders.push_back( makePtr<BmpEncoder>() );
115

116 117
        decoders.push_back( makePtr<HdrDecoder>() );
        encoders.push_back( makePtr<HdrEncoder>() );
118
    #ifdef HAVE_JPEG
R
Roman Donchenko 已提交
119 120
        decoders.push_back( makePtr<JpegDecoder>() );
        encoders.push_back( makePtr<JpegEncoder>() );
121 122
    #endif
    #ifdef HAVE_WEBP
R
Roman Donchenko 已提交
123 124
        decoders.push_back( makePtr<WebPDecoder>() );
        encoders.push_back( makePtr<WebPEncoder>() );
125
    #endif
R
Roman Donchenko 已提交
126 127 128 129
        decoders.push_back( makePtr<SunRasterDecoder>() );
        encoders.push_back( makePtr<SunRasterEncoder>() );
        decoders.push_back( makePtr<PxMDecoder>() );
        encoders.push_back( makePtr<PxMEncoder>() );
130
    #ifdef HAVE_TIFF
R
Roman Donchenko 已提交
131
        decoders.push_back( makePtr<TiffDecoder>() );
132
    #endif
R
Roman Donchenko 已提交
133
        encoders.push_back( makePtr<TiffEncoder>() );
134
    #ifdef HAVE_PNG
R
Roman Donchenko 已提交
135 136
        decoders.push_back( makePtr<PngDecoder>() );
        encoders.push_back( makePtr<PngEncoder>() );
137
    #endif
138 139 140
    #ifdef HAVE_GDCM
        decoders.push_back( makePtr<DICOMDecoder>() );
    #endif
141
    #ifdef HAVE_JASPER
R
Roman Donchenko 已提交
142 143
        decoders.push_back( makePtr<Jpeg2KDecoder>() );
        encoders.push_back( makePtr<Jpeg2KEncoder>() );
144 145
    #endif
    #ifdef HAVE_OPENEXR
R
Roman Donchenko 已提交
146 147
        decoders.push_back( makePtr<ExrDecoder>() );
        encoders.push_back( makePtr<ExrEncoder>() );
148
    #endif
149 150 151 152 153

    #ifdef HAVE_GDAL
        /// Attach the GDAL Decoder
        decoders.push_back( makePtr<GdalDecoder>() );
    #endif/*HAVE_GDAL*/
154 155
        decoders.push_back( makePtr<PAMDecoder>() );
        encoders.push_back( makePtr<PAMEncoder>() );
156 157
    }

158 159
    std::vector<ImageDecoder> decoders;
    std::vector<ImageEncoder> encoders;
160 161 162
};

static ImageCodecInitializer codecs;
163

164 165 166 167 168 169 170 171 172
/**
 * Find the decoders
 *
 * @param[in] filename File to search
 *
 * @return Image decoder to parse image file.
*/
static ImageDecoder findDecoder( const String& filename ) {

173
    size_t i, maxlen = 0;
174 175

    /// iterate through list of registered codecs
176
    for( i = 0; i < codecs.decoders.size(); i++ )
177
    {
178
        size_t len = codecs.decoders[i]->signatureLength();
179 180 181
        maxlen = std::max(maxlen, len);
    }

182
    /// Open the file
183
    FILE* f= fopen( filename.c_str(), "rb" );
184 185

    /// in the event of a failure, return an empty image decoder
186 187
    if( !f )
        return ImageDecoder();
188 189

    // read the file signature
190
    String signature(maxlen, ' ');
191
    maxlen = fread( (void*)signature.c_str(), 1, maxlen, f );
192 193 194
    fclose(f);
    signature = signature.substr(0, maxlen);

195
    /// compare signature against all decoders
196
    for( i = 0; i < codecs.decoders.size(); i++ )
197
    {
198 199
        if( codecs.decoders[i]->checkSignature(signature) )
            return codecs.decoders[i]->newDecoder();
200 201
    }

202
    /// If no decoder was found, return base type
203 204 205
    return ImageDecoder();
}

206
static ImageDecoder findDecoder( const Mat& buf )
207 208 209 210 211 212
{
    size_t i, maxlen = 0;

    if( buf.rows*buf.cols < 1 || !buf.isContinuous() )
        return ImageDecoder();

213
    for( i = 0; i < codecs.decoders.size(); i++ )
214
    {
215
        size_t len = codecs.decoders[i]->signatureLength();
216 217 218
        maxlen = std::max(maxlen, len);
    }

219
    String signature(maxlen, ' ');
220 221
    size_t bufSize = buf.rows*buf.cols*buf.elemSize();
    maxlen = std::min(maxlen, bufSize);
222
    memcpy( (void*)signature.c_str(), buf.data, maxlen );
223

224
    for( i = 0; i < codecs.decoders.size(); i++ )
225
    {
226 227
        if( codecs.decoders[i]->checkSignature(signature) )
            return codecs.decoders[i]->newDecoder();
228 229 230 231 232
    }

    return ImageDecoder();
}

233
static ImageEncoder findEncoder( const String& _ext )
234 235 236 237 238 239 240 241
{
    if( _ext.size() <= 1 )
        return ImageEncoder();

    const char* ext = strrchr( _ext.c_str(), '.' );
    if( !ext )
        return ImageEncoder();
    int len = 0;
J
Julien Nabet 已提交
242
    for( ext++; len < 128 && isalnum(ext[len]); len++ )
243 244
        ;

245
    for( size_t i = 0; i < codecs.encoders.size(); i++ )
246
    {
247
        String description = codecs.encoders[i]->getDescription();
248 249 250 251 252 253 254 255
        const char* descr = strchr( description.c_str(), '(' );

        while( descr )
        {
            descr = strchr( descr + 1, '.' );
            if( !descr )
                break;
            int j = 0;
J
Julien Nabet 已提交
256
            for( descr++; j < len && isalnum(descr[j]) ; j++ )
257 258 259 260 261 262 263
            {
                int c1 = tolower(ext[j]);
                int c2 = tolower(descr[j]);
                if( c1 != c2 )
                    break;
            }
            if( j == len && !isalnum(descr[j]))
264
                return codecs.encoders[i]->newEncoder();
265 266 267 268 269 270 271
            descr += j;
        }
    }

    return ImageEncoder();
}

272

273 274
enum { LOAD_CVMAT=0, LOAD_IMAGE=1, LOAD_MAT=2 };

275
static void ExifTransform(int orientation, Mat& img)
A
Arkadiusz Raj 已提交
276 277 278 279 280 281 282 283 284 285 286 287 288 289 290 291 292 293 294 295 296 297 298 299 300 301 302 303 304 305 306 307 308 309 310
{
    switch( orientation )
    {
        case    IMAGE_ORIENTATION_TL: //0th row == visual top, 0th column == visual left-hand side
            //do nothing, the image already has proper orientation
            break;
        case    IMAGE_ORIENTATION_TR: //0th row == visual top, 0th column == visual right-hand side
            flip(img, img, 1); //flip horizontally
            break;
        case    IMAGE_ORIENTATION_BR: //0th row == visual bottom, 0th column == visual right-hand side
            flip(img, img, -1);//flip both horizontally and vertically
            break;
        case    IMAGE_ORIENTATION_BL: //0th row == visual bottom, 0th column == visual left-hand side
            flip(img, img, 0); //flip vertically
            break;
        case    IMAGE_ORIENTATION_LT: //0th row == visual left-hand side, 0th column == visual top
            transpose(img, img);
            break;
        case    IMAGE_ORIENTATION_RT: //0th row == visual right-hand side, 0th column == visual top
            transpose(img, img);
            flip(img, img, 1); //flip horizontally
            break;
        case    IMAGE_ORIENTATION_RB: //0th row == visual right-hand side, 0th column == visual bottom
            transpose(img, img);
            flip(img, img, -1); //flip both horizontally and vertically
            break;
        case    IMAGE_ORIENTATION_LB: //0th row == visual left-hand side, 0th column == visual bottom
            transpose(img, img);
            flip(img, img, 0); //flip vertically
            break;
        default:
            //by default the image read has normal (JPEG_ORIENTATION_TL) orientation
            break;
    }
}
311

312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354 355
static void ApplyExifOrientation(const String& filename, Mat& img)
{
    int orientation = IMAGE_ORIENTATION_TL;

    if (filename.size() > 0)
    {
        std::ifstream stream( filename.c_str(), std::ios_base::in | std::ios_base::binary );
        ExifReader reader( stream );
        if( reader.parse() )
        {
            ExifEntry_t entry = reader.getTag( ORIENTATION );
            if (entry.tag != INVALID_TAG)
            {
                orientation = entry.field_u16; //orientation is unsigned short, so check field_u16
            }
        }
        stream.close();
    }

    ExifTransform(orientation, img);
}

static void ApplyExifOrientation(const Mat& buf, Mat& img)
{
    int orientation = IMAGE_ORIENTATION_TL;

    if( buf.isContinuous() )
    {
        ByteStreamBuffer bsb( reinterpret_cast<char*>(buf.data), buf.total() * buf.elemSize() );
        std::istream stream( &bsb );
        ExifReader reader( stream );
        if( reader.parse() )
        {
            ExifEntry_t entry = reader.getTag( ORIENTATION );
            if (entry.tag != INVALID_TAG)
            {
                orientation = entry.field_u16; //orientation is unsigned short, so check field_u16
            }
        }
    }

    ExifTransform(orientation, img);
}

356 357 358 359 360 361 362 363 364 365
/**
 * Read an image into memory and return the information
 *
 * @param[in] filename File to load
 * @param[in] flags Flags
 * @param[in] hdrtype { LOAD_CVMAT=0,
 *                      LOAD_IMAGE=1,
 *                      LOAD_MAT=2
 *                    }
 * @param[in] mat Reference to C++ Mat object (If LOAD_MAT)
S
Suleyman TURKMEN 已提交
366
 * @param[in] scale_denom Scale value
367 368
 *
*/
369
static void*
370
imread_( const String& filename, int flags, int hdrtype, Mat* mat=0 )
371 372 373 374 375
{
    IplImage* image = 0;
    CvMat *matrix = 0;
    Mat temp, *data = &temp;

376 377 378 379
    /// Search for the relevant decoder to handle the imagery
    ImageDecoder decoder;

#ifdef HAVE_GDAL
M
Maksim Shabunin 已提交
380
    if(flags != IMREAD_UNCHANGED && (flags & IMREAD_LOAD_GDAL) == IMREAD_LOAD_GDAL ){
381 382 383
        decoder = GdalDecoder().newDecoder();
    }else{
#endif
384
        decoder = findDecoder( filename );
385 386 387 388 389 390
#ifdef HAVE_GDAL
    }
#endif

    /// if no decoder was found, return nothing.
    if( !decoder ){
391
        return 0;
392 393
    }

394 395 396
    int scale_denom = 1;
    if( flags > IMREAD_LOAD_GDAL )
    {
S
Suleyman TURKMEN 已提交
397
    if( flags & IMREAD_REDUCED_GRAYSCALE_2 )
398
        scale_denom = 2;
S
Suleyman TURKMEN 已提交
399
    else if( flags & IMREAD_REDUCED_GRAYSCALE_4 )
400
        scale_denom = 4;
S
Suleyman TURKMEN 已提交
401
    else if( flags & IMREAD_REDUCED_GRAYSCALE_8 )
402 403 404
        scale_denom = 8;
    }

S
Suleyman TURKMEN 已提交
405 406 407
    /// set the scale_denom in the driver
    decoder->setScale( scale_denom );

408
    /// set the filename in the driver
409
    decoder->setSource( filename );
410 411 412

   // read the header to make sure it succeeds
   if( !decoder->readHeader() )
413
        return 0;
414 415

    // established the required input image size
416 417 418 419
    CvSize size;
    size.width = decoder->width();
    size.height = decoder->height();

420
    // grab the decoded type
421
    int type = decoder->type();
422
    if( (flags & IMREAD_LOAD_GDAL) != IMREAD_LOAD_GDAL && flags != IMREAD_UNCHANGED )
423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438
    {
        if( (flags & CV_LOAD_IMAGE_ANYDEPTH) == 0 )
            type = CV_MAKETYPE(CV_8U, CV_MAT_CN(type));

        if( (flags & CV_LOAD_IMAGE_COLOR) != 0 ||
           ((flags & CV_LOAD_IMAGE_ANYCOLOR) != 0 && CV_MAT_CN(type) > 1) )
            type = CV_MAKETYPE(CV_MAT_DEPTH(type), 3);
        else
            type = CV_MAKETYPE(CV_MAT_DEPTH(type), 1);
    }

    if( hdrtype == LOAD_CVMAT || hdrtype == LOAD_MAT )
    {
        if( hdrtype == LOAD_CVMAT )
        {
            matrix = cvCreateMat( size.height, size.width, type );
439
            temp = cvarrToMat( matrix );
440 441 442 443 444 445 446 447 448 449
        }
        else
        {
            mat->create( size.height, size.width, type );
            data = mat;
        }
    }
    else
    {
        image = cvCreateImage( size, cvIplDepth(type), CV_MAT_CN(type) );
450
        temp = cvarrToMat( image );
451 452
    }

453
    // read the image data
454 455 456 457 458 459 460 461 462
    if( !decoder->readData( *data ))
    {
        cvReleaseImage( &image );
        cvReleaseMat( &matrix );
        if( mat )
            mat->release();
        return 0;
    }

463
    if( decoder->setScale( scale_denom ) > 1 ) // if decoder is JpegDecoder then decoder->setScale always returns 1
S
Suleyman TURKMEN 已提交
464
    {
465
        resize( *mat, *mat, Size( size.width / scale_denom, size.height / scale_denom ) );
S
Suleyman TURKMEN 已提交
466 467
    }

468 469 470 471
    return hdrtype == LOAD_CVMAT ? (void*)matrix :
        hdrtype == LOAD_IMAGE ? (void*)image : (void*)mat;
}

472 473 474 475 476 477 478 479 480 481 482 483 484 485 486 487

/**
* Read an image into memory and return the information
*
* @param[in] filename File to load
* @param[in] flags Flags
* @param[in] mats Reference to C++ vector<Mat> object to hold the images
*
*/
static bool
imreadmulti_(const String& filename, int flags, std::vector<Mat>& mats)
{
    /// Search for the relevant decoder to handle the imagery
    ImageDecoder decoder;

#ifdef HAVE_GDAL
M
Maksim Shabunin 已提交
488
    if (flags != IMREAD_UNCHANGED && (flags & IMREAD_LOAD_GDAL) == IMREAD_LOAD_GDAL){
489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513
        decoder = GdalDecoder().newDecoder();
    }
    else{
#endif
        decoder = findDecoder(filename);
#ifdef HAVE_GDAL
    }
#endif

    /// if no decoder was found, return nothing.
    if (!decoder){
        return 0;
    }

    /// set the filename in the driver
    decoder->setSource(filename);

    // read the header to make sure it succeeds
    if (!decoder->readHeader())
        return 0;

    for (;;)
    {
        // grab the decoded type
        int type = decoder->type();
514
        if( (flags & IMREAD_LOAD_GDAL) != IMREAD_LOAD_GDAL && flags != IMREAD_UNCHANGED )
515 516 517 518 519 520 521 522 523 524 525 526
        {
            if ((flags & CV_LOAD_IMAGE_ANYDEPTH) == 0)
                type = CV_MAKETYPE(CV_8U, CV_MAT_CN(type));

            if ((flags & CV_LOAD_IMAGE_COLOR) != 0 ||
                ((flags & CV_LOAD_IMAGE_ANYCOLOR) != 0 && CV_MAT_CN(type) > 1))
                type = CV_MAKETYPE(CV_MAT_DEPTH(type), 3);
            else
                type = CV_MAKETYPE(CV_MAT_DEPTH(type), 1);
        }

        // read the image data
527
        Mat mat(decoder->height(), decoder->width(), type);
528 529
        if (!decoder->readData(mat))
        {
530
            // optionally rotate the data if EXIF' orientation flag says so
A
Arkadiusz Raj 已提交
531
            if( (flags & IMREAD_IGNORE_ORIENTATION) == 0 && flags != IMREAD_UNCHANGED )
532
            {
A
Arkadiusz Raj 已提交
533
                ApplyExifOrientation(filename, mat);
534 535
            }

536 537 538 539 540 541 542 543 544 545 546 547 548
            break;
        }

        mats.push_back(mat);
        if (!decoder->nextPage())
        {
            break;
        }
    }

    return !mats.empty();
}

549 550 551 552 553 554 555 556
/**
 * Read an image
 *
 *  This function merely calls the actual implementation above and returns itself.
 *
 * @param[in] filename File to load
 * @param[in] flags Flags you wish to set.
*/
557
Mat imread( const String& filename, int flags )
558
{
559
    /// create the basic container
560
    Mat img;
561 562

    /// load the data
563
    imread_( filename, flags, LOAD_MAT, &img );
564

565
    /// optionally rotate the data if EXIF' orientation flag says so
566
    if( !img.empty() && (flags & IMREAD_IGNORE_ORIENTATION) == 0 && flags != IMREAD_UNCHANGED )
567
    {
A
Arkadiusz Raj 已提交
568
        ApplyExifOrientation(filename, img);
569 570
    }

571
    /// return a reference to the data
572 573 574
    return img;
}

575 576 577 578 579 580 581 582 583 584 585 586 587 588 589
/**
* Read a multi-page image
*
*  This function merely calls the actual implementation above and returns itself.
*
* @param[in] filename File to load
* @param[in] mats Reference to C++ vector<Mat> object to hold the images
* @param[in] flags Flags you wish to set.
*
*/
bool imreadmulti(const String& filename, std::vector<Mat>& mats, int flags)
{
    return imreadmulti_(filename, flags, mats);
}

590
static bool imwrite_( const String& filename, const Mat& image,
591
                      const std::vector<int>& params, bool flipv )
592 593 594 595 596 597 598
{
    Mat temp;
    const Mat* pimage = &image;

    CV_Assert( image.channels() == 1 || image.channels() == 3 || image.channels() == 4 );

    ImageEncoder encoder = findEncoder( filename );
R
Roman Donchenko 已提交
599
    if( !encoder )
600 601 602 603 604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620
        CV_Error( CV_StsError, "could not find a writer for the specified extension" );
    if( !encoder->isFormatSupported(image.depth()) )
    {
        CV_Assert( encoder->isFormatSupported(CV_8U) );
        image.convertTo( temp, CV_8U );
        pimage = &temp;
    }

    if( flipv )
    {
        flip(*pimage, temp, 0);
        pimage = &temp;
    }

    encoder->setDestination( filename );
    bool code = encoder->write( *pimage, params );

    //    CV_Assert( code );
    return code;
}

621
bool imwrite( const String& filename, InputArray _img,
622
              const std::vector<int>& params )
623
{
624
    Mat img = _img.getMat();
625 626 627 628 629 630
    return imwrite_(filename, img, params, false);
}

static void*
imdecode_( const Mat& buf, int flags, int hdrtype, Mat* mat=0 )
{
631
    CV_Assert(!buf.empty() && buf.isContinuous());
632 633 634
    IplImage* image = 0;
    CvMat *matrix = 0;
    Mat temp, *data = &temp;
635
    String filename;
636 637

    ImageDecoder decoder = findDecoder(buf);
R
Roman Donchenko 已提交
638
    if( !decoder )
639 640 641 642
        return 0;

    if( !decoder->setSource(buf) )
    {
R
Roy Reapor 已提交
643
        filename = tempfile();
644
        FILE* f = fopen( filename.c_str(), "wb" );
645 646 647
        if( !f )
            return 0;
        size_t bufSize = buf.cols*buf.rows*buf.elemSize();
648
        fwrite( buf.ptr(), 1, bufSize, f );
649 650 651 652 653 654
        fclose(f);
        decoder->setSource(filename);
    }

    if( !decoder->readHeader() )
    {
655 656 657 658 659 660 661 662
        decoder.release();
        if ( !filename.empty() )
        {
            if ( remove(filename.c_str()) != 0 )
            {
                CV_Error( CV_StsError, "unable to remove temporary file" );
            }
        }
663 664 665 666 667 668 669 670
        return 0;
    }

    CvSize size;
    size.width = decoder->width();
    size.height = decoder->height();

    int type = decoder->type();
671
    if( (flags & IMREAD_LOAD_GDAL) != IMREAD_LOAD_GDAL && flags != IMREAD_UNCHANGED )
672 673 674 675 676 677 678 679 680 681 682 683 684 685 686 687 688 689 690 691 692 693 694 695 696 697 698 699 700 701 702
    {
        if( (flags & CV_LOAD_IMAGE_ANYDEPTH) == 0 )
            type = CV_MAKETYPE(CV_8U, CV_MAT_CN(type));

        if( (flags & CV_LOAD_IMAGE_COLOR) != 0 ||
           ((flags & CV_LOAD_IMAGE_ANYCOLOR) != 0 && CV_MAT_CN(type) > 1) )
            type = CV_MAKETYPE(CV_MAT_DEPTH(type), 3);
        else
            type = CV_MAKETYPE(CV_MAT_DEPTH(type), 1);
    }

    if( hdrtype == LOAD_CVMAT || hdrtype == LOAD_MAT )
    {
        if( hdrtype == LOAD_CVMAT )
        {
            matrix = cvCreateMat( size.height, size.width, type );
            temp = cvarrToMat(matrix);
        }
        else
        {
            mat->create( size.height, size.width, type );
            data = mat;
        }
    }
    else
    {
        image = cvCreateImage( size, cvIplDepth(type), CV_MAT_CN(type) );
        temp = cvarrToMat(image);
    }

    bool code = decoder->readData( *data );
703 704 705 706 707 708 709 710
    decoder.release();
    if ( !filename.empty() )
    {
        if ( remove(filename.c_str()) != 0 )
        {
            CV_Error( CV_StsError, "unable to remove temporary file" );
        }
    }
711 712 713 714 715 716 717 718 719 720 721 722 723 724 725

    if( !code )
    {
        cvReleaseImage( &image );
        cvReleaseMat( &matrix );
        if( mat )
            mat->release();
        return 0;
    }

    return hdrtype == LOAD_CVMAT ? (void*)matrix :
        hdrtype == LOAD_IMAGE ? (void*)image : (void*)mat;
}


726
Mat imdecode( InputArray _buf, int flags )
727
{
728
    Mat buf = _buf.getMat(), img;
729
    imdecode_( buf, flags, LOAD_MAT, &img );
730 731 732 733 734 735 736

    /// optionally rotate the data if EXIF' orientation flag says so
    if( !img.empty() && (flags & IMREAD_IGNORE_ORIENTATION) == 0 && flags != IMREAD_UNCHANGED )
    {
        ApplyExifOrientation(buf, img);
    }

737 738
    return img;
}
739

740 741 742 743 744
Mat imdecode( InputArray _buf, int flags, Mat* dst )
{
    Mat buf = _buf.getMat(), img;
    dst = dst ? dst : &img;
    imdecode_( buf, flags, LOAD_MAT, dst );
745 746 747 748 749 750 751

    /// optionally rotate the data if EXIF' orientation flag says so
    if( !dst->empty() && (flags & IMREAD_IGNORE_ORIENTATION) == 0 && flags != IMREAD_UNCHANGED )
    {
        ApplyExifOrientation(buf, *dst);
    }

752 753
    return *dst;
}
754

755
bool imencode( const String& ext, InputArray _image,
756
               std::vector<uchar>& buf, const std::vector<int>& params )
757
{
758
    Mat image = _image.getMat();
759 760 761 762 763

    int channels = image.channels();
    CV_Assert( channels == 1 || channels == 3 || channels == 4 );

    ImageEncoder encoder = findEncoder( ext );
R
Roman Donchenko 已提交
764
    if( !encoder )
765 766 767 768 769
        CV_Error( CV_StsError, "could not find encoder for the specified extension" );

    if( !encoder->isFormatSupported(image.depth()) )
    {
        CV_Assert( encoder->isFormatSupported(CV_8U) );
770
        Mat temp;
771
        image.convertTo(temp, CV_8U);
772
        image = temp;
773 774 775 776 777 778
    }

    bool code;
    if( encoder->setDestination(buf) )
    {
        code = encoder->write(image, params);
779
        encoder->throwOnEror();
780 781 782 783
        CV_Assert( code );
    }
    else
    {
784
        String filename = tempfile();
785 786
        code = encoder->setDestination(filename);
        CV_Assert( code );
787

788
        code = encoder->write(image, params);
789
        encoder->throwOnEror();
790
        CV_Assert( code );
791

792
        FILE* f = fopen( filename.c_str(), "rb" );
793 794 795 796 797 798 799
        CV_Assert(f != 0);
        fseek( f, 0, SEEK_END );
        long pos = ftell(f);
        buf.resize((size_t)pos);
        fseek( f, 0, SEEK_SET );
        buf.resize(fread( &buf[0], 1, buf.size(), f ));
        fclose(f);
800
        remove(filename.c_str());
801 802 803 804 805 806 807
    }
    return code;
}

}

/****************************************************************************************\
808
*                         Imgcodecs loading & saving function implementation            *
809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838 839 840 841 842 843 844 845
\****************************************************************************************/

CV_IMPL int
cvHaveImageReader( const char* filename )
{
    cv::ImageDecoder decoder = cv::findDecoder(filename);
    return !decoder.empty();
}

CV_IMPL int cvHaveImageWriter( const char* filename )
{
    cv::ImageEncoder encoder = cv::findEncoder(filename);
    return !encoder.empty();
}

CV_IMPL IplImage*
cvLoadImage( const char* filename, int iscolor )
{
    return (IplImage*)cv::imread_(filename, iscolor, cv::LOAD_IMAGE );
}

CV_IMPL CvMat*
cvLoadImageM( const char* filename, int iscolor )
{
    return (CvMat*)cv::imread_( filename, iscolor, cv::LOAD_CVMAT );
}

CV_IMPL int
cvSaveImage( const char* filename, const CvArr* arr, const int* _params )
{
    int i = 0;
    if( _params )
    {
        for( ; _params[i] > 0; i += 2 )
            ;
    }
    return cv::imwrite_(filename, cv::cvarrToMat(arr),
846
        i > 0 ? std::vector<int>(_params, _params+i) : std::vector<int>(),
847 848 849 850 851 852 853 854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882
        CV_IS_IMAGE(arr) && ((const IplImage*)arr)->origin == IPL_ORIGIN_BL );
}

/* decode image stored in the buffer */
CV_IMPL IplImage*
cvDecodeImage( const CvMat* _buf, int iscolor )
{
    CV_Assert( _buf && CV_IS_MAT_CONT(_buf->type) );
    cv::Mat buf(1, _buf->rows*_buf->cols*CV_ELEM_SIZE(_buf->type), CV_8U, _buf->data.ptr);
    return (IplImage*)cv::imdecode_(buf, iscolor, cv::LOAD_IMAGE );
}

CV_IMPL CvMat*
cvDecodeImageM( const CvMat* _buf, int iscolor )
{
    CV_Assert( _buf && CV_IS_MAT_CONT(_buf->type) );
    cv::Mat buf(1, _buf->rows*_buf->cols*CV_ELEM_SIZE(_buf->type), CV_8U, _buf->data.ptr);
    return (CvMat*)cv::imdecode_(buf, iscolor, cv::LOAD_CVMAT );
}

CV_IMPL CvMat*
cvEncodeImage( const char* ext, const CvArr* arr, const int* _params )
{
    int i = 0;
    if( _params )
    {
        for( ; _params[i] > 0; i += 2 )
            ;
    }
    cv::Mat img = cv::cvarrToMat(arr);
    if( CV_IS_IMAGE(arr) && ((const IplImage*)arr)->origin == IPL_ORIGIN_BL )
    {
        cv::Mat temp;
        cv::flip(img, temp, 0);
        img = temp;
    }
883
    std::vector<uchar> buf;
884 885 886 887 888 889 890 891 892 893 894 895

    bool code = cv::imencode(ext, img, buf,
        i > 0 ? std::vector<int>(_params, _params+i) : std::vector<int>() );
    if( !code )
        return 0;
    CvMat* _buf = cvCreateMat(1, (int)buf.size(), CV_8U);
    memcpy( _buf->data.ptr, &buf[0], buf.size() );

    return _buf;
}

/* End of file. */