Src/Loaders/LdrFormatsLoader/LdrFormatsLoader.cpp

//      LdrFormatsLoader.cpp
//      
//      Copyright 2010 Jérémy Laumon <jeremy.laumon@gmail.com>
//      
//      This program is free software; you can redistribute it and/or modify
//      it under the terms of the GNU General Public License as published by
//      the Free Software Foundation; either version 2 of the License, or
//      (at your option) any later version.
//      
//      This program is distributed in the hope that it will be useful,
//      but WITHOUT ANY WARRANTY; without even the implied warranty of
//      MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
//      GNU General Public License for more details.
//      
//      You should have received a copy of the GNU General Public License
//      along with this program; if not, write to the Free Software
//      Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
//      MA 02110-1301, USA.


#include <LdrFormatsLoader.h>

#define tr(s) QObject::tr(s)

Q_EXPORT_PLUGIN2(TT_OpenExrLoader, LdrFormatsLoaderFactory)


QString LdrFormatsLoader::name() const
{
    return tr("LDR Formats Loader");
}

void LdrFormatsLoader::setFileName(const QString& f)
{
    fileName = f;
}

bool LdrFormatsLoader::openFile()
{
    return image.load(fileName);
}

QSize LdrFormatsLoader::getSize()
{
    return image.size();
}

Color* LdrFormatsLoader::getData()
{
    QSize size = image.size();
    int width = size.width();
    int height = size.height();
    
    float rgbToXyzMatrix  [3][3] = {{0.5141364, 0.3238786,  0.16036376},
                                    {0.265068,  0.67023428, 0.06409157},
                                    {0.0241188, 0.1228178,  0.84442666}};
    
    Color* data = new Color[width*height];
    float w;
    int ii, jj;
    for (int i=0; i<height; ++i)
        for (int j=0; j<width; ++j)
        {
            QRgb pixel = image.pixel(j,i);
            float rgb[3] = {float(qRed(pixel))/255.0,
                            float(qGreen(pixel))/255.0,
                            float(qBlue(pixel))/255.0};
            float xyz[3] = {0.0, 0.0, 0.0};
            
            for (ii = 0; ii < 3; ++ii)
                for (jj = 0; jj < 3; ++jj)
                    xyz[ii] += rgbToXyzMatrix[ii][jj] * rgb[jj];
            
            if((w = xyz[0] + xyz[1] + xyz[2]) > 0.0)
                data[i*width + j] = Color(xyz[1],     // Y
                                          xyz[0]/w,   // x
                                          xyz[1]/w);  // y
            else
                data[i*width + j] = Color(0.0,0.0,0.0);

        }

    return data;
}

HdrImage::ColorSpace LdrFormatsLoader::getColorSpace()
{
    return HdrImage::Yxy;
}


//----------------------------------------------------------------------


QStringList LdrFormatsLoaderFactory::extensions() const 
{
    return QStringList() << "bmp" << "gif" << "jpg" << "jpeg" << "png" << "tiff";
}

ImageLoaderPtr LdrFormatsLoaderFactory::createLoader(const QString& fileName) const
{
    ImageLoaderPtr loader(new LdrFormatsLoader);
    loader->setFileName(fileName);
    return loader;
}