geo/tcShadowClassification.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 tcShadowClassification.cpp
  22  * @brief Shadow classification channel.
  23  */
  24
  25 #include "tcShadowClassification.h"
  26 #include "tcIlluminantDiscontinuity.h"
  27
  28 #include <QObject>
  29
  30 /*
  31  * Constructors + destructor
  32  */
  33
  34 /// Primary constructor.
  35 tcShadowClassification::tcShadowClassification(tcIlluminantDiscontinuity* discontinuity, tcChannel* midIr)
  36 : tcChannel(tr("Shadow classification"),
  37             tr("Tries to classify shadows using illuminant discontinuities and a mid IR channel"))
  38 , m_discontinuity()
  39 , m_midIr(midIr)
  40 {
  41   for (int i = 0; i < 2; ++i)
  42   {
  43     m_discontinuity[i] = discontinuity->channels()[i];
  44     m_discontinuity[i]->addDerivitive(this);
  45   }
  46   m_midIr->addDerivitive(this);
  47 }
  48
  49 /// Destructor.
  50 tcShadowClassification::~tcShadowClassification()
  51 {
  52   for (int i = 0; i < 2; ++i)
  53   {
  54     m_discontinuity[i]->removeDerivitive(this);
  55   }
  56   m_midIr->removeDerivitive(this);
  57 }
  58
  59 /*
  60  * Interface for derived class to implement
  61  */
  62
  63 void tcShadowClassification::roundPortion(double* x1, double* y1, double* x2, double* y2)
  64 {
  65   m_midIr->roundPortion(x1, y1, x2, y2);
  66 }
  67
  68 tcAbstractPixelData* tcShadowClassification::loadPortion(double x1, double y1, double x2, double y2, bool changed)
  69 {
  70   // Get portions from both the discontinuity images and the mid ir
  71   // multiply discontinuity by (1-midIr)
  72
  73   Reference<tcPixelData<float> > disc[2] = {
  74     dynamic_cast<tcPixelData<float>*>(m_discontinuity[0]->loadPortion(x1, y1, x2, y2, changed)),
  75     dynamic_cast<tcPixelData<float>*>(m_discontinuity[1]->loadPortion(x1, y1, x2, y2, changed))
  76   };
  77   Reference<tcPixelData<GLubyte> > midIr = dynamic_cast<tcPixelData<GLubyte>*>(m_midIr->loadPortion(x1, y1, x2, y2, changed));
  78
  79   tcPixelData<GLubyte>* data = 0;
  80   if (0 != disc[0] && 0 != disc[1] && 0 != midIr)
  81   {
  82     int width = midIr->width();
  83     int height = midIr->height();
  84     Q_ASSERT(width == disc[0]->width());
  85     Q_ASSERT(height == disc[0]->height());
  86     Q_ASSERT(width == disc[1]->width());
  87     Q_ASSERT(height == disc[1]->height());
  88     data = new tcPixelData<GLubyte>(width, height);
  89     for (int i = 0; i < width*height; ++i)
  90     {
  91       float disc1Val = disc[0]->buffer()[i];
  92       float disc2Val = disc[1]->buffer()[i];
  93       float midIrVal = (float)midIr->buffer()[i]/255.0f;
  94       float shadowiness = fabs(disc1Val + disc1Val) * 0.5f
  95                             * (1.0f - midIrVal)
  96                             * (1.0f - midIrVal);
  97       //shadowiness = (shadowiness > 0.25f ? shadowiness : 0.0f);
  98       data->buffer()[i] = shadowiness;
  99       /// @todo Have a parameter threshold
 100       //data->buffer()[i] = (shadowiness > 0.25f ? 255 : 0);
 101     }
 102
 103     // Linearly fill in the gaps
 104     /// @todo fillSize should be a parameter
 105     const int fillSize = 2;
 106     for (int d = 0; d < 0; ++d)
 107     {
 108       int dir = d % 2;
 109       int jMax = (dir ? width : height);
 110       int iMax = (dir ? height : width);
 111       for (int j = 0; j < jMax; ++j)
 112       {
 113         int blankCount = 0;
 114         for (int i = 0; i < iMax; ++i)
 115         {
 116           int row = (dir ? i : j);
 117           int col = (dir ? j : i);
 118           int index = row*width + col;
 119           if (0 != data->buffer()[index])
 120           {
 121             if (blankCount > 0 && blankCount <= fillSize)
 122             {
 123               int min = i-blankCount;
 124               if (min > 0)
 125               {
 126                 for (int k = min; k < i; ++k)
 127                 {
 128                   int kIndex = (dir?k:row)*width + (dir?col:k);
 129                   data->buffer()[kIndex] = 255;
 130                 }
 131               }
 132             }
 133             blankCount = 0;
 134           }
 135           else
 136           {
 137             ++blankCount;
 138           }
 139         }
 140       }
 141     }
 142   }
 143   return data;
 144 }