/* ============================================================
 *
 * This file is a part of kipi-plugins project
 * http://www.digikam.org
 *
 * Date        : 2007-02-11
 * Description : a kipi plugin to show image using an OpenGL interface.
 *
 * Copyright (C) 2007-2008 by Markus Leuthold <kusi at forum dot titlis dot org>
 * Copyright (C) 2008-2015 by Gilles Caulier <caulier dot gilles at gmail dot 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, 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.
 *
 * ============================================================ */

#include "texture.h"

// Qt includes

#include <QMatrix>
#include <QFileInfo>

// KDE includes

#include <kdebug.h>

// LibKDcraw includes

#include <libkdcraw/version.h>
#include <libkdcraw/kdcraw.h>

// Local includes

#include "kpimageinfo.h"
#include "timer.h"

using namespace KDcrawIface;

namespace KIPIViewerPlugin
{

class Texture::Private
{
public:

    Private()
    {
        rdx            = 0.0;
        rdy            = 0.0;
        z              = 0.0;
        ux             = 0.0;
        uy             = 0.0;
        rtx            = 0.0;
        rty            = 0.0;
        vtop           = 0.0;
        vbottom        = 0.0;
        vleft          = 0.0;
        vright         = 0.0;
        display_x      = 0;
        display_y      = 0;
        texnr          = 0;
        rotate_idx     = 0;
        rotate_list[0] = KPMetadata::ORIENTATION_ROT_90;
        rotate_list[1] = KPMetadata::ORIENTATION_ROT_180;
        rotate_list[2] = KPMetadata::ORIENTATION_ROT_270;
        rotate_list[3] = KPMetadata::ORIENTATION_ROT_180;

    }

    float                        rdx, rdy, z, ux, uy, rtx, rty;
    float                        vtop, vbottom, vleft, vright;
    int                          display_x, display_y;
    GLuint                       texnr;
    QString                      filename;
    QImage                       qimage, glimage;
    QSize                        initial_size;
    KPMetadata::ImageOrientation rotate_list[4];
    int                          rotate_idx;
};

Texture::Texture()
    : d(new Private)
{
    reset();
}

Texture::~Texture()
{
    delete d;
}

/*!
    \fn Texture::height()
 */
int Texture::height() const
{
    return d->glimage.height();
}

/*!
    \fn Texture::width()
 */
int Texture::width() const
{
    return d->glimage.width();
}

/*!
    \fn Texture::load(QString fn, QSize size, GLuint tn)
    \brief load file from disc and save it in texture
    \param fn filename to load
    \param size size of image which is downloaded to texture mem
    \param tn texture id generated by glGenTexture
    if "size" is set to image size, scaling is only performed by the GPU but not
    by the CPU, however the AGP usage to texture memory is increased (20MB for a 5mp image)
 */
bool Texture::load(const QString& fn, const QSize& size, GLuint tn)
{
    d->filename     = fn;
    d->initial_size = size;
    d->texnr        = tn;

    // check if its a RAW file.
    if (KPMetadata::isRawFile(d->filename))
    {
        // it's a RAW file, use the libkdcraw loader
        KDcraw::loadRawPreview(d->qimage, d->filename);
    }
    else
    {
        // use the standard loader
        d->qimage = QImage(d->filename);
    }

    //handle rotation
    KPImageInfo info(d->filename);

    if ( info.orientation() != KPMetadata::ORIENTATION_UNSPECIFIED )
    {
        QMatrix matrix = RotationMatrix::toMatrix(info.orientation());
        d->qimage      = d->qimage.transformed(matrix);
    }

    if (d->qimage.isNull())
    {
        return false;
    }

    loadInternal();
    reset();
    d->rotate_idx = 0;
    return true;
}

/*!
    \fn Texture::load(QImage im, QSize size, GLuint tn)
    \brief copy file from QImage to texture
    \param im Qimage to be copied from
    \param size size of image which is downloaded to texture mem
    \param tn texture id generated by glGenTexture
    if "size" is set to image size, scaling is only performed by the GPU but not
    by the CPU, however the AGP usage to texture memory is increased (20MB for a 5mp image)
 */
bool Texture::load(const QImage& im, const QSize& size, GLuint tn)
{
    d->qimage       = im;
    d->initial_size = size;
    d->texnr        = tn;
    loadInternal();
    reset();
    d->rotate_idx   = 0;
    return true;
}

/*!
    \fn Texture::load()
    internal load function
    rt[xy] <= 1
 */
bool Texture::loadInternal()
{
    int w = d->initial_size.width();
    int h = d->initial_size.height();

    if (w == 0 || w > d->qimage.width() || h > d->qimage.height())
    {
        d->glimage = QGLWidget::convertToGLFormat(d->qimage);
    }
    else
    {
        d->glimage = QGLWidget::convertToGLFormat(d->qimage.scaled(w,h,Qt::KeepAspectRatio,Qt::FastTransformation));
    }

    w = d->glimage.width();
    h = d->glimage.height();

    if (h < w)
    {
        d->rtx = 1;
        d->rty = float(h)/float(w);
    }
    else
    {
        d->rtx = float(w)/float(h);
        d->rty = 1;
    }

    return true;
}

/*!
    \fn Texture::data() const
 */
GLvoid* Texture::data() const
{
    return d->glimage.bits();
}

/*!
    \fn Texture::texnr() const
 */
GLuint Texture::texnr() const
{
    return d->texnr;
}

/*!
    \fn Texture::zoom(float delta, QPoint mousepos)
    \brief calculate new tex coords on zooming
    \param delta delta between previous zoom and current zoom
    \param mousepos mouse position returned by QT
    \TODO rename mousepos to something more generic
*/
void Texture::zoom(float delta, const QPoint& mousepos)
//u: start in texture, u=[0..1], u=0 is begin, u=1 is end of texture
//z=[0..1], z=1 -> no zoom
//l: length of tex in glFrustum coordinate system
//rt: ratio of tex, rt<=1, see loadInternal() for definition
//rd: ratio of display, rd>=1
//m: mouse pos normalized, cd=[0..rd]
//c:  mouse pos normalized to zoom*l, c=[0..1]
{
    d->z    *= delta;
    delta    =  d->z*(1.0/delta-1.0); //convert to real delta=z_old-z_new

    float mx = mousepos.x()/(float)d->display_x*d->rdx;
    float cx = (mx-d->rdx/2.0+d->rtx/2.0)/d->rtx;
    float vx = d->ux+cx*d->z;
    d->ux    = d->ux+(vx-d->ux)*delta/d->z;

    float my = mousepos.y()/(float)d->display_y*d->rdy;
    float cy = (my-d->rdy/2.0+d->rty/2.0)/d->rty;
    cy       = 1-cy;
    float vy = d->uy+cy*d->z;
    d->uy    = d->uy+(vy-d->uy)*delta/d->z;

    calcVertex();
}

/*!
    \fn Texture::calcVertex()
    Calculate vertices according internal state variables
    z, ux, uy are calculated in Texture::zoom()
 */
void Texture::calcVertex()
// rt: ratio of tex, rt<=1, see loadInternal() for definition
// u: start in texture, u=[0..1], u=0 is begin, u=1 is end of texture
// l: length of tex in glFrustum coordinate system
// halftexel: the color of a texel is determined by a corner of the texel and not its center point
//                  this seems to introduce a visible jump on changing the tex-size.
//
// the glFrustum coord-sys is visible in [-rdx..rdx] ([-1..1] for square screen) for z=1 (no zoom)
// the tex coord-sys goes from [-rtx..rtx] ([-1..1] for square texture)
{
    // x part
    float lx          = 2*d->rtx/d->z;  //length of tex
    float tsx         = lx/(float)d->glimage.width(); //texelsize in glFrustum coordinates
    float halftexel_x = tsx/2.0;
    float wx          = lx*(1-d->ux-d->z);
    d->vleft          = -d->rtx-d->ux*lx - halftexel_x;  //left
    d->vright         = d->rtx+wx - halftexel_x;      //right

    // y part
    float ly          = 2*d->rty/d->z;
    float tsy         = ly/(float)d->glimage.height(); //texelsize in glFrustum coordinates
    float halftexel_y = tsy/2.0;
    float wy          = ly*(1-d->uy-d->z);
    d->vbottom        = -d->rty - d->uy*ly + halftexel_y; //bottom
    d->vtop           = d->rty + wy + halftexel_y;     //top
}

/*!
    \fn Texture::vertex_bottom() const
 */
GLfloat Texture::vertex_bottom() const
{
    return (GLfloat) d->vbottom;
}

/*!
    \fn Texture::vertex_top() const
 */
GLfloat Texture::vertex_top() const
{
    return (GLfloat) d->vtop;
}

/*!
    \fn Texture::vertex_left() const
 */
GLfloat Texture::vertex_left() const
{
    return (GLfloat) d->vleft;
}

/*!
    \fn Texture::vertex_right() const
 */
GLfloat Texture::vertex_right() const
{
    return (GLfloat) d->vright;
}

/*!
    \fn Texture::setViewport(int w, int h)
    \param w width of window
    \param h height of window
    Set widget's viewport. Ensures that rdx & rdy are always > 1
 */
void Texture::setViewport(int w, int h)
{
    if (h > w)
    {
        d->rdx = 1.0;
        d->rdy = h/float(w);
    }
    else
    {
        d->rdx = w/float(h);
        d->rdy = 1.0;
    }

    d->display_x = w;
    d->display_y = h;
}

/*!
    \fn Texture::move(QPoint diff)
    new tex coordinates have to be calculated if the view is panned
 */
void Texture::move(const QPoint& diff)
{
    d->ux = d->ux - diff.x()/float(d->display_x)*d->z*d->rdx/d->rtx;
    d->uy = d->uy + diff.y()/float(d->display_y)*d->z*d->rdy/d->rty;
    calcVertex();
}

/*!
    \fn Texture::reset()
 */
void Texture::reset()
{
    d->ux           = 0;
    d->uy           = 0;
    d->z            = 1.0;
    float zoomdelta = 0;

    if ((d->rtx < d->rty) && (d->rdx < d->rdy) && (d->rtx/d->rty < d->rdx/d->rdy))
    {
        zoomdelta = d->z-d->rdx/d->rdy;
    }

    if ((d->rtx < d->rty) && (d->rtx/d->rty > d->rdx/d->rdy))
    {
        zoomdelta = d->z-d->rtx;
    }

    if ((d->rtx >= d->rty) && (d->rdy < d->rdx) && (d->rty/d->rtx < d->rdy/d->rdx))
    {
        zoomdelta = d->z-d->rdy/d->rdx;
    }

    if ((d->rtx >= d->rty) && (d->rty/d->rtx > d->rdy/d->rdx))
    {
        zoomdelta = d->z-d->rty;
    }

    QPoint p = QPoint(d->display_x/2, d->display_y/2);
    zoom(1.0 - zoomdelta, p);

    calcVertex();
}

/*!
    \fn Texture::setSize(QSize size)
    \param size desired texture size. QSize(0,0) will take the full image
    \return true if size has changed, false otherwise
    set new texture size in order to reduce AGP bandwidth
 */
bool Texture::setSize(QSize size)
{
    //don't allow larger textures than the original image. the image will be upsampled by
    //OpenGL if necessary and not by QImage::scale
    size = size.boundedTo(d->qimage.size());

    if (d->glimage.width() == size.width())
    {
        return false;
    }

    int w = size.width();
    int h = size.height();

    if (w == 0)
    {
        d->glimage = QGLWidget::convertToGLFormat(d->qimage);
    }
    else
    {
        d->glimage = QGLWidget::convertToGLFormat(d->qimage.scaled(w, h, Qt::KeepAspectRatio, Qt::FastTransformation));
    }

    //recalculate half-texel offset
    calcVertex();

    return true;
}

/*!
    \fn Texture::rotate()
    \brief smart image rotation
    since the two most frequent usecases are a CW or CCW rotation of 90,
    perform these rotation with one (+90) or two (-90) calls of rotation()
 */
void Texture::rotate()
{
    QMatrix matrix = RotationMatrix::toMatrix(d->rotate_list[d->rotate_idx%4]);
    d->qimage      = d->qimage.transformed(matrix);
    loadInternal();

    //save new rotation in exif header
    KPImageInfo info(d->filename);
    info.setOrientation(d->rotate_list[d->rotate_idx%4]);

    reset();
    d->rotate_idx++;
}

/*!
    \fn Texture::setToOriginalSize()
    zoom image such that each pixel of the screen corresponds to a pixel in the jpg
    remember that OpenGL is not a pixel exact specification, and the image will still be filtered by OpenGL
 */
void Texture::zoomToOriginal()
{
    float zoomfactorToOriginal;
    reset();

    if (float(d->qimage.width())/float(d->qimage.height()) > float(d->display_x)/float(d->display_y))
    {
        //image touches right and left edge of window
        zoomfactorToOriginal = float(d->display_x)/d->qimage.width();
    }
    else
    {
        //image touches upper and lower edge of window
        zoomfactorToOriginal = float(d->display_y)/d->qimage.height();
    }

    zoom(zoomfactorToOriginal,QPoint(d->display_x/2,d->display_y/2));
}

} // namespace KIPIViewerPlugin
