visu2d.cpp

Go to the documentation of this file.

#include <QtGui/qpainter.h>
#include <QtGui/qmessagebox.h>
#include <QtGui/qevent.h>

#include <iostream>
#include "visu2d.h"
#include "palette.h"
#include "selection.h"
#include "shape.h"
#include "image.h"

/*-------------------------------------------------------------------------*/
/* Class for the image visualization                                       */
/*-------------------------------------------------------------------------*/
Visu2D::Visu2D(QWidget *parent)
  : QWidget(parent) 
{
  m_pImage = NULL;
  m_selection = NULL;
  m_tmpShape = NULL;
  m_visuScaleFactor = 2;
  
  setFocusPolicy(Qt::StrongFocus);
  setMouseTracking(TRUE);
  
  setMinimumSize(512, 512);
}

Visu2D::~Visu2D()
{
  if (m_selection) {
    delete m_selection;
    m_selection = 0;
  }
  if (m_tmpShape) {
    delete m_tmpShape;
    m_tmpShape = 0;
  }
}

void Visu2D::paintEvent(QPaintEvent*) 
{
  QPainter p(this);
  p.drawPixmap(0, 0, m_pxScaled);
  DrawShapes(&p);       
}

void Visu2D::DrawShapes(QPainter *p) 
{
  if (m_selection) {
    p->setPen(Qt::red);
    m_selection->Draw(p, m_visuScaleFactor);
  }
  if (m_tmpShape) {
    p->setPen(Qt::blue);
    m_tmpShape->Draw(p, m_visuScaleFactor);
  }
}

void Visu2D::SetShapeType(SHAPE_TYPE type) 
{
  m_shapeType = type;
}

void Visu2D::UpdateDataAndPixmap(Image* im, Palette *pal) 
{
  m_pImage = im;
  
  if (!m_pImage || !pal) {
    return;
  }  
  // the visualization is resized depending on the size of the data
  // and the visualization scale factor
  setFixedSize(m_pImage->GetSizeX()*m_visuScaleFactor, 
               m_pImage->GetSizeY()*m_visuScaleFactor);
  
  // instantiate a QImage object with the correct data size
  QImage *pQImage = new QImage(m_pImage->GetSizeX(),
                               m_pImage->GetSizeY(),
                               QImage::Format_Indexed8);
  
  // initialize the object's colors from the palette's ones
  pQImage->setNumColors(NB_COLORS); 
  for (int i = 0; i < pQImage->numColors(); i++) {
    pQImage->setColor(i, pal->GetColor(i).rgb());
  }

  // initialize the object's data with the image's values and the palette's
  uchar *ptr;
  for (int y = 0; y < pQImage->height(); y++) {
    ptr = pQImage->scanLine(y);
    for (int x = 0; x < pQImage->width(); x++) {
      *ptr++ = pal->DataToIndex(m_pImage->GetData(x, y));
    }
  }

  // convert the QImage object into a QPixmap
  QPixmap px;
  px = px.fromImage(*pQImage);
 
  delete pQImage;

  // initialize the pixmap rescaling matrix (to fit into the visualization)
  QMatrix matrix;
  matrix.scale((double)m_visuScaleFactor, (double)m_visuScaleFactor);
  
  // rescale the pixmap
  m_pxScaled = px.transformed(matrix); 
  
  // eventually calls paintEvent()
  update();
}

void Visu2D::mouseMoveEvent(QMouseEvent *e) 
{
  HandleMouse(MOVE, e);
}

void Visu2D::mousePressEvent(QMouseEvent *e) 
{
  HandleMouse(PRESS, e);
}

void Visu2D::mouseReleaseEvent(QMouseEvent *e) 
{
  HandleMouse(RELEASE, e);
}

void Visu2D::HandleMouse(int type, QMouseEvent *e) 
{
  // no image to be drawn
  if (!m_pImage) {
    return;
  }

  // computes the 'data' coordinates from the 'visualization' ones.
  // the 'visualization' coordinates are given by the position of the mouse
  int xData = e->x() / m_visuScaleFactor;
  int yData = e->y() / m_visuScaleFactor;
  
  if (!m_pImage->IsPointInside(xData, yData)) {
    return;
  }
  
  SetTextInfo(xData, yData);
  
  if (e->button()  & Qt::LeftButton || 
      e->buttons() & Qt::LeftButton) {
    GenerateShape(type, xData, yData);
  }
}

void Visu2D::SetTextInfo(int xData, int yData) 
{
  QString txt;
  if (m_pImage->IsPointInside(xData, yData)) {
    txt.sprintf("(%d, %d) : %d", 
                xData, 
                yData, 
                m_pImage->GetData(xData, yData));
  }
  emit(UpdateInfoBar(txt));
}

void Visu2D::GenerateShape(int type, int xData, int yData)
{
  switch (type) {
  case PRESS :
    if (!m_selection) {
      m_selection = new Selection;
    }
    m_tmpShape = m_selection->AllocateShape(m_shapeType);
    if (m_tmpShape == NULL) {
      QMessageBox msgBox;
      msgBox.setText("Visu2D::GenerateShape\n Warning : No shape allocated !");
      msgBox.exec();
      return;
    }
    m_tmpShape->Init(xData, yData);
    break;

  case MOVE :
    if (!m_tmpShape) {
      return;
    }
    m_tmpShape->ModifyEndShape(xData, yData);
    break;

  case RELEASE :
    if (!m_tmpShape) {
      return;
    }
    if (m_tmpShape->Validate()) {
      m_selection->HandleNewShape(m_tmpShape);
    } else {
      delete m_tmpShape;
    }
    m_tmpShape = NULL;
    emit(UpdateStats(m_selection));
    break;
  }
  update();
}

void Visu2D::SaveSelection() 
{
  if (m_selection == NULL) {
    QMessageBox msgBox;
    msgBox.setText("Visu2D::SaveSelection\n Warning : No selection allocated (check the value of m_selection)");
    msgBox.exec();
    return;
  }

}

void Visu2D::ReadSelection() 
{
  if (m_selection == NULL) {
    QMessageBox msgBox;
    msgBox.setText("Visu2D::SaveSelection\n Warning : No selection allocated (check the value of m_selection)");
    msgBox.exec();
    return;
  }

}

---