loadsave.cpp 25.3 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
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 )
    {
B
Brian Armstrong 已提交
73
        char* whence = eback();
74 75
        if (dir == std::ios_base::cur)
        {
B
Brian Armstrong 已提交
76
            whence = gptr();
77 78 79
        }
        else if (dir == std::ios_base::end)
        {
B
Brian Armstrong 已提交
80
            whence = egptr();
81
        }
B
Brian Armstrong 已提交
82
        char* to = whence + offset;
83 84

        // check limits
B
Brian Armstrong 已提交
85
        if (to >= eback() && to <= egptr())
86
        {
B
Brian Armstrong 已提交
87
            setg(eback(), to, egptr());
88 89 90 91 92 93 94 95 96
            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 560
    CV_TRACE_FUNCTION();

561
    /// create the basic container
562
    Mat img;
563 564

    /// load the data
565
    imread_( filename, flags, LOAD_MAT, &img );
566

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

573
    /// return a reference to the data
574 575 576
    return img;
}

577 578 579 580 581 582 583 584 585 586 587 588
/**
* 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)
{
589 590
    CV_TRACE_FUNCTION();

591 592 593
    return imreadmulti_(filename, flags, mats);
}

594
static bool imwrite_( const String& filename, const Mat& image,
595
                      const std::vector<int>& params, bool flipv )
596 597 598 599 600 601 602
{
    Mat temp;
    const Mat* pimage = &image;

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

    ImageEncoder encoder = findEncoder( filename );
R
Roman Donchenko 已提交
603
    if( !encoder )
604 605 606 607 608 609 610 611 612 613 614 615 616 617 618 619 620 621 622 623 624
        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;
}

625
bool imwrite( const String& filename, InputArray _img,
626
              const std::vector<int>& params )
627
{
628 629
    CV_TRACE_FUNCTION();

630
    Mat img = _img.getMat();
631 632 633 634 635 636
    return imwrite_(filename, img, params, false);
}

static void*
imdecode_( const Mat& buf, int flags, int hdrtype, Mat* mat=0 )
{
637
    CV_Assert(!buf.empty() && buf.isContinuous());
638 639 640
    IplImage* image = 0;
    CvMat *matrix = 0;
    Mat temp, *data = &temp;
641
    String filename;
642 643

    ImageDecoder decoder = findDecoder(buf);
R
Roman Donchenko 已提交
644
    if( !decoder )
645 646 647 648
        return 0;

    if( !decoder->setSource(buf) )
    {
R
Roy Reapor 已提交
649
        filename = tempfile();
650
        FILE* f = fopen( filename.c_str(), "wb" );
651 652 653
        if( !f )
            return 0;
        size_t bufSize = buf.cols*buf.rows*buf.elemSize();
654
        fwrite( buf.ptr(), 1, bufSize, f );
655 656 657 658 659 660
        fclose(f);
        decoder->setSource(filename);
    }

    if( !decoder->readHeader() )
    {
661 662 663 664 665 666 667 668
        decoder.release();
        if ( !filename.empty() )
        {
            if ( remove(filename.c_str()) != 0 )
            {
                CV_Error( CV_StsError, "unable to remove temporary file" );
            }
        }
669 670 671 672 673 674 675 676
        return 0;
    }

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

    int type = decoder->type();
677
    if( (flags & IMREAD_LOAD_GDAL) != IMREAD_LOAD_GDAL && flags != IMREAD_UNCHANGED )
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 703 704 705 706 707 708
    {
        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 );
709 710 711 712 713 714 715 716
    decoder.release();
    if ( !filename.empty() )
    {
        if ( remove(filename.c_str()) != 0 )
        {
            CV_Error( CV_StsError, "unable to remove temporary file" );
        }
    }
717 718 719 720 721 722 723 724 725 726 727 728 729 730 731

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


732
Mat imdecode( InputArray _buf, int flags )
733
{
734 735
    CV_TRACE_FUNCTION();

736
    Mat buf = _buf.getMat(), img;
737
    imdecode_( buf, flags, LOAD_MAT, &img );
738 739 740 741 742 743 744

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

745 746
    return img;
}
747

748 749
Mat imdecode( InputArray _buf, int flags, Mat* dst )
{
750 751
    CV_TRACE_FUNCTION();

752 753 754
    Mat buf = _buf.getMat(), img;
    dst = dst ? dst : &img;
    imdecode_( buf, flags, LOAD_MAT, dst );
755 756 757 758 759 760 761

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

762 763
    return *dst;
}
764

765
bool imencode( const String& ext, InputArray _image,
766
               std::vector<uchar>& buf, const std::vector<int>& params )
767
{
768 769
    CV_TRACE_FUNCTION();

770
    Mat image = _image.getMat();
771 772 773 774 775

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

    ImageEncoder encoder = findEncoder( ext );
R
Roman Donchenko 已提交
776
    if( !encoder )
777 778 779 780 781
        CV_Error( CV_StsError, "could not find encoder for the specified extension" );

    if( !encoder->isFormatSupported(image.depth()) )
    {
        CV_Assert( encoder->isFormatSupported(CV_8U) );
782
        Mat temp;
783
        image.convertTo(temp, CV_8U);
784
        image = temp;
785 786 787 788 789 790
    }

    bool code;
    if( encoder->setDestination(buf) )
    {
        code = encoder->write(image, params);
791
        encoder->throwOnEror();
792 793 794 795
        CV_Assert( code );
    }
    else
    {
796
        String filename = tempfile();
797 798
        code = encoder->setDestination(filename);
        CV_Assert( code );
799

800
        code = encoder->write(image, params);
801
        encoder->throwOnEror();
802
        CV_Assert( code );
803

804
        FILE* f = fopen( filename.c_str(), "rb" );
805 806 807 808 809 810 811
        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);
812
        remove(filename.c_str());
813 814 815 816 817 818 819
    }
    return code;
}

}

/****************************************************************************************\
820
*                         Imgcodecs loading & saving function implementation            *
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 846 847 848 849 850 851 852 853 854 855 856 857
\****************************************************************************************/

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),
858
        i > 0 ? std::vector<int>(_params, _params+i) : std::vector<int>(),
859 860 861 862 863 864 865 866 867 868 869 870 871 872 873 874 875 876 877 878 879 880 881 882 883 884 885 886 887 888 889 890 891 892 893 894
        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;
    }
895
    std::vector<uchar> buf;
896 897 898 899 900 901 902 903 904 905 906 907

    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. */