geo/tcIlluminantDiscontinuity.cpp

   1 /***************************************************************************
   2  *   This file is part of Tecorrec.                                        *
   3  *   Copyright 2008 James Hogan <james@albanarts.com>                      *
   4  *                                                                         *
   5  *   Tecorrec is free software: you can redistribute it and/or modify      *
   6  *   it under the terms of the GNU General Public License as published by  *
   7  *   the Free Software Foundation, either version 2 of the License, or     *
   8  *   (at your option) any later version.                                   *
   9  *                                                                         *
  10  *   Tecorrec is distributed in the hope that it will be useful,           *
  11  *   but WITHOUT ANY WARRANTY; without even the implied warranty of        *
  12  *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the         *
  13  *   GNU General Public License for more details.                          *
  14  *                                                                         *
  15  *   You should have received a copy of the GNU General Public License     *
  16  *   along with Tecorrec.  If not, write to the Free Software Foundation,  *
  17  *   Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.   *
  18  ***************************************************************************/
  19
  20 /**
  21  * @file tcIlluminantDiscontinuity.cpp
  22  * @brief Illuminant discontinuity in both dimentions.
  23  */
  24
  25 #include "tcIlluminantDiscontinuity.h"
  26 #include "tcChannel.h"
  27
  28 #include <QObject>
  29
  30 #include <cmath>
  31
  32 /*
  33  * Constructors + destructor
  34  */
  35
  36 /// Primary constructor.
  37 tcIlluminantDiscontinuity::tcIlluminantDiscontinuity(const QList<tcChannel*>& chromaticities)
  38 : tcIlluminantDirection(chromaticities, 2, QObject::tr("Illuminant discontinuity measure"), QObject::tr(""))
  39 {
  40   channels()[0]->setName(QObject::tr("idm.x"));
  41   channels()[1]->setName(QObject::tr("idm.y"));
  42   channels()[0]->setDescription(QObject::tr("Illuminant discontinuity measure in the x direction"));
  43   channels()[1]->setDescription(QObject::tr("Illuminant discontinuity measure in the y direction"));
  44 }
  45
  46 /// Destructor.
  47 tcIlluminantDiscontinuity::~tcIlluminantDiscontinuity()
  48 {
  49 }
  50
  51 /*
  52  * Interface for derived class to implement
  53  */
  54
  55 void tcIlluminantDiscontinuity::loadPortions(const QList< Reference< tcPixelData<float> > >& chromaticities, int width, int height)
  56 {
  57   Reference< tcPixelData<float> > discontinuityData[2];
  58   for (int i = 0; i < 2; ++i)
  59   {
  60     discontinuityData[i] = new tcPixelData<GLfloat>(width, height);
  61     m_portions += discontinuityData[i];
  62   }
  63
  64   // Go through the pixels
  65   maths::VarVector<float> lastLogChromaticity(numChromaticities());
  66   maths::VarVector<float> logChromaticity(numChromaticities());
  67   maths::VarVector<float> temp(numChromaticities());
  68   float lastDisc = 0.0f;
  69   for (int i = 0; i < width; ++i)
  70   {
  71     for (int j = 0; j < height; ++j)
  72     {
  73       int index = j*width + i;
  74       // Fill log chromaticity vector
  75       for (int channel = 0; channel < numChromaticities(); ++channel)
  76       {
  77         logChromaticity[channel] = logf(chromaticities[channel]->buffer()[index]);
  78       }
  79
  80       temp = logChromaticity;
  81       temp -= lastLogChromaticity;
  82
  83       float discontinuity = lastDisc;
  84       float sqr = temp.sqr();
  85       if (sqr > 0.0f)
  86       {
  87         float mag = sqrtf(sqr);
  88         if (mag > 0.0f)
  89         {
  90           discontinuity = /*fabs*/(temp * illuminantDirection()) /*/ mag*/;
  91         }
  92       }
  93       lastDisc = discontinuity*0.5f;
  94
  95       lastLogChromaticity = logChromaticity;
  96
  97       discontinuityData[0]->buffer()[index] = 0.5f + discontinuity;
  98     }
  99   }
 100   lastDisc = 0.0f;
 101   for (int j = 0; j < height; ++j)
 102   {
 103     for (int i = 0; i < width; ++i)
 104     {
 105       int index = j*width + i;
 106       // Fill log chromaticity vector
 107       for (int channel = 0; channel < numChromaticities(); ++channel)
 108       {
 109         logChromaticity[channel] = logf(chromaticities[channel]->buffer()[index]);
 110       }
 111
 112       temp = logChromaticity;
 113       temp -= lastLogChromaticity;
 114
 115       float discontinuity = lastDisc;
 116       float sqr = temp.sqr();
 117       if (sqr > 0.0f)
 118       {
 119         float mag = sqrtf(sqr);
 120         if (mag > 0.0f)
 121         {
 122           discontinuity = /*fabs*/(temp * illuminantDirection()) /*/ mag*/;
 123         }
 124       }
 125       lastDisc = discontinuity*0.5f;
 126
 127       lastLogChromaticity = logChromaticity;
 128
 129       discontinuityData[1]->buffer()[index] = 0.5f + discontinuity;
 130     }
 131   }
 132 }