mapmap/Mapper.cpp

/*
 * Mapper.cpp
 *
 * (c) 2013 Sofian Audry -- info(@)sofianaudry(.)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 3 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, see <http://www.gnu.org/licenses/>.
 */

#include "Mapper.h"

TextureMapper::TextureMapper(std::tr1::shared_ptr<TextureMapping> mapping)
  : Mapper(mapping)
{
  _propertyBrowser = new QtGroupBoxPropertyBrowser;
  _variantManager = new QtVariantPropertyManager;
  _variantFactory = new QtVariantEditorFactory;

  _propertyBrowser->setFactoryForManager(_variantManager, _variantFactory);

  std::tr1::shared_ptr<TextureMapping> textureMapping = std::tr1::static_pointer_cast<TextureMapping>(mapping);
  Q_CHECK_PTR(textureMapping);

  std::tr1::shared_ptr<Texture> texture = std::tr1::static_pointer_cast<Texture>(textureMapping->getPaint());
  Q_CHECK_PTR(texture);

  QtProperty *topItem = _variantManager->addProperty(QtVariantPropertyManager::groupTypeId(),
                                                     QObject::tr("Texture mapping"));
  // Input shape
  QtProperty *inputItem = _variantManager->addProperty(QtVariantPropertyManager::groupTypeId(),
                                                       QObject::tr("Input shape"));
  topItem->addSubProperty(inputItem);

  Shape::ptr inputShape = textureMapping->getInputShape();
  for (int i=0; i<inputShape->nVertices(); i++)
  {
    // Add point.
    QtVariantProperty* pointItem = _variantManager->addProperty(QVariant::PointF,
                                                                QString("Point %1").arg(i));

    inputItem->addSubProperty(pointItem);
    _propertyToVertex[pointItem] = std::make_pair(inputShape.get(), i);
  }

  connect(_variantManager, SIGNAL(valueChanged(QtProperty*, const QVariant&)),
          this,            SLOT(setValue(QtProperty*, const QVariant&)));

  _propertyBrowser->addProperty(topItem);

  qDebug() << "Creating mapper" << endl;
}

void TextureMapper::setValue(QtProperty* property, const QVariant& value)
{
  std::map<QtProperty*, std::pair<Shape*, int> >::iterator it = _propertyToVertex.find(property);
  if (it != _propertyToVertex.end())
  {
    QPointF p = value.toPointF();
    it->second.first->setVertex(it->second.second, Point(p.x(), p.y()));
    qDebug() << "Changing vertex: " << it->second.second << " to " << p.x() << "," << p.y() << endl;
  }

  emit valueChanged();
//  qDebug() << "Property changed to " << property->propertyName() << " " << value.toPointF().x() << ", " << value.toPointF().y() << endl;
}

void TextureMapper::updateShape(Shape* shape)
{
  std::tr1::shared_ptr<TextureMapping> textureMapping = std::tr1::static_pointer_cast<TextureMapping>(_mapping);
  Q_CHECK_PTR(textureMapping);

  std::tr1::shared_ptr<Texture> texture = std::tr1::static_pointer_cast<Texture>(textureMapping->getPaint());
  Q_CHECK_PTR(texture);

  Shape::ptr inputShape = textureMapping->getInputShape();
  if (shape == inputShape.get())
  {
    QtProperty* topItem   = _propertyBrowser->properties().first();
    QtProperty* inputItem = topItem->subProperties()[0];
    QList<QtProperty*> pointItems = inputItem->subProperties();
    for (int i=0; i<inputShape->nVertices(); i++)
    {
      QtVariantProperty* pointItem = (QtVariantProperty*)pointItems[i];
      Point p = inputShape->getVertex(i);
      pointItem->setValue(QPointF(p.x, p.y));
    }
  }
}

QWidget* TextureMapper::getPropertiesEditor()
{
  return _propertyBrowser;
}

void TextureMapper::draw()
{
  // FIXME: use typedefs, member of the class for type names that are too long to type:
  std::tr1::shared_ptr<TextureMapping> textureMapping = std::tr1::static_pointer_cast<TextureMapping>(_mapping);
  Q_CHECK_PTR(textureMapping);

  std::tr1::shared_ptr<Texture> texture = std::tr1::static_pointer_cast<Texture>(textureMapping->getPaint());
  Q_CHECK_PTR(texture);

  std::tr1::shared_ptr<Shape> outputShape = std::tr1::static_pointer_cast<Quad>(textureMapping->getShape());
  Q_CHECK_PTR(outputShape);

  std::tr1::shared_ptr<Quad> inputShape = std::tr1::static_pointer_cast<Quad>(textureMapping->getInputShape());
  Q_CHECK_PTR(inputShape);

  // Only works for similar shapes.
  Q_ASSERT( outputShape->nVertices() == outputShape->nVertices());

  printf("Texid: %d\n", texture->getTextureId());
  // Project source texture and sent it to destination.

  glEnable (GL_TEXTURE_2D);
  glBindTexture(GL_TEXTURE_2D, texture->getTextureId());

  glTexImage2D(GL_TEXTURE_2D, 0, GL_RGBA,
    texture->getWidth(), texture->getHeight(), 0, GL_RGBA,
    GL_UNSIGNED_BYTE, texture->getBits());

  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_S, GL_CLAMP);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_WRAP_T, GL_CLAMP);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MIN_FILTER, GL_NEAREST);
  glTexParameteri(GL_TEXTURE_2D, GL_TEXTURE_MAG_FILTER, GL_LINEAR);

  glColor4f(1.0f, 1.0f, 1.0f, 1.0f);
  if (inputShape->nVertices() == 4)
    glBegin(GL_QUADS);
  else if (inputShape->nVertices() == 3)
    glBegin(GL_TRIANGLES);
  else
    // TODO: untested
    glBegin(GL_POLYGON);

  {
    for (int i = 0; i < inputShape->nVertices(); i++)
    {
      Util::correctGlTexCoord(
        (inputShape->getVertex(i).x - texture->getX()) / (GLfloat) texture->getWidth(),
        (inputShape->getVertex(i).y - texture->getY()) / (GLfloat) texture->getHeight());
      glVertex2f(
        outputShape->getVertex(i).x,
        outputShape->getVertex(i).y
        );
    }
  }
  glEnd();

  glDisable(GL_TEXTURE_2D);
}