curveeditor.cpp

/*****************************************************************************
 *                                                                           *
 *  Elmer, A Finite Element Software for Multiphysical Problems              *
 *                                                                           *
 *  Copyright 1st April 1995 - , CSC - IT Center for Science Ltd., Finland   *
 *                                                                           *
 *  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           *
 *  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 (in file fem/GPL-2); if not, write to the        *
 *  Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,         *
 *  Boston, MA 02110-1301, USA.                                              *
 *                                                                           *
 *****************************************************************************/

/*****************************************************************************
 *                                                                           *
 *  ElmerGUI CurveEditor                                                     *
 *                                                                           *
 *****************************************************************************
 *                                                                           *
 *  Authors: Mikko Lyly, Juha Ruokolainen and Peter Råback                   *
 *  Email:   [email protected]                                         *
 *  Web:     http://www.csc.fi/elmer                                         *
 *  Address: CSC - IT Center for Science Ltd.                                *
 *           Keilaranta 14                                                   *
 *           02101 Espoo, Finland                                            *
 *                                                                           *
 *  Original Date: 15 Mar 2008                                               *
 *                                                                           *
 *****************************************************************************/
#include <QTableWidget>
#include <QModelIndex>
#include <iostream>
#include "curveeditor.h"
#include "renderarea.h"

using namespace std;

CurveEditor::CurveEditor(QWidget *parent)
  : QTabWidget(parent)
{
  pTable = new QTableWidget(0, 3, this);
  cTable = new QTableWidget(0, 6, this);

  addTab(pTable, tr("Points"));
  addTab(cTable, tr("Curves"));

  connect(cTable, SIGNAL(cellChanged(int, int)), this, SLOT(cCellChanged(int, int)));
  connect(pTable, SIGNAL(cellChanged(int, int)), this, SLOT(pCellChanged(int, int)));

  clearAll();
}

CurveEditor::~CurveEditor()
{
}

void CurveEditor::addPoint(int idx, double x, double y)
{
  int i = pTable->rowCount();
  
  pTable->insertRow(i);
  pTable->setRowHeight(i, 20);
  QTableWidgetItem *item;

  item = new QTableWidgetItem;
  item->setText(QString::number(idx));
  pTable->setItem(i, 0, item);

  item = new QTableWidgetItem;
  item->setText(QString::number(x));
  pTable->setItem(i, 1, item);

  item = new QTableWidgetItem;
  item->setText(QString::number(y));
  pTable->setItem(i, 2, item);
}

void CurveEditor::addCurve(int in, int out, int pts, int *p)
{
  int i = cTable->rowCount();
  
  cTable->insertRow(i);
  cTable->setRowHeight(i, 20);
  QTableWidgetItem *item;

  item = new QTableWidgetItem;
  item->setText(QString::number(in));
  cTable->setItem(i, 0, item);

  item = new QTableWidgetItem;
  item->setText(QString::number(out));
  cTable->setItem(i, 1, item);

  item = new QTableWidgetItem;
  item->setText(QString::number(pts));
  cTable->setItem(i, 2, item);

  item = new QTableWidgetItem;
  item->setText(QString::number(p[0]));
  cTable->setItem(i, 3, item);

  item = new QTableWidgetItem;
  item->setText(QString::number(p[1]));
  cTable->setItem(i, 4, item);

  item = new QTableWidgetItem;
  if(pts == 3) {
    item->setText(QString::number(p[2]));
  } else {
    item->setText("-");
  }
  cTable->setItem(i, 5, item);
}

void CurveEditor::clearAll()
{
  pTable->clear();
  pTable->setRowCount(0);

  cTable->clear();
  cTable->setRowCount(0);

  QStringList pHeaders;
  pHeaders << "idx" << "x" << "y";
  pTable->setHorizontalHeaderLabels(pHeaders);

  pTable->setColumnWidth(0, 40);
  pTable->setColumnWidth(1, 80);
  pTable->setColumnWidth(2, 80);

  QStringList cHeaders;
  cHeaders << "out" << "in" << "pts" << "p1" << "p2" << "p3";
  cTable->setHorizontalHeaderLabels(cHeaders);
  cTable->setColumnWidth(0, 40);
  cTable->setColumnWidth(1, 40);
  cTable->setColumnWidth(2, 40);
  cTable->setColumnWidth(3, 40);
  cTable->setColumnWidth(4, 40);
  cTable->setColumnWidth(5, 40);
}

void CurveEditor::modifyPoint(int idx, double x, double y)
{
  for(int i = 0; i < pTable->rowCount(); i++) {
    QTableWidgetItem *item = pTable->item(i, 0);

    if(item) {
      if(item->text().toInt() == idx) {
	QTableWidgetItem *itemX = pTable->item(i, 1);
	if(itemX)
	  itemX->setText(QString::number(x));

	QTableWidgetItem *itemY = pTable->item(i, 2);
	if(itemY)
	  itemY->setText(QString::number(y));
      }
    }
  }
}

void CurveEditor::setRenderArea(RenderArea *renderArea)
{
  this->renderArea = renderArea;
}

void CurveEditor::pCellChanged(int row, int col)
{
  QTableWidgetItem *item0 = pTable->item(row, 0);
  QTableWidgetItem *item1 = pTable->item(row, 1);
  QTableWidgetItem *item2 = pTable->item(row, 2);

  int idx;
  double x, y;

  if(item0)
    idx = item0->text().toInt();
  else
    return;

  if(item1)
    x = item1->text().toDouble();
  else
    return;

  if(item2)
    y = item2->text().toDouble();
  else
    return;

  renderArea->modifyPoint(idx, x, y);
}

void CurveEditor::cCellChanged(int row, int col)
{
  QTableWidgetItem *item0 = cTable->item(row, 0);
  QTableWidgetItem *item1 = cTable->item(row, 1);
  QTableWidgetItem *item2 = cTable->item(row, 2);
  QTableWidgetItem *item3 = cTable->item(row, 3);
  QTableWidgetItem *item4 = cTable->item(row, 4);
  QTableWidgetItem *item5 = cTable->item(row, 5);

  int in, out, np, p0, p1, p2;

  if(item0)
    in = item0->text().toInt();
  else
    return;

  if(item1)
    out = item1->text().toInt();
  else
    return;

  if(item2)
    np = item2->text().toInt();
  else
    return;

  if(item3)
    p0 = item3->text().toInt();
  else
    return;

  if(item4)
    p1 = item4->text().toInt();
  else
    return;

  p2 = 0;
  if(item5)
    p2 = item5->text().toInt();

  renderArea->modifyCurve(row, in, out, np, p0, p1, p2);
}

void CurveEditor::addPoint()
{
  int i = pTable->rowCount();
  
  pTable->insertRow(i);
  pTable->setRowHeight(i, 20);
  QTableWidgetItem *item;

  for(int j = 0; j < 3; j++) {
    item = new QTableWidgetItem;
    item->setText("-");
    pTable->setItem(i, j, item);
  }
}

void CurveEditor::addCurve()
{
  int i = cTable->rowCount();
  
  cTable->insertRow(i);
  cTable->setRowHeight(i, 20);
  QTableWidgetItem *item;

  for(int j = 0; j < 6; j++) {
    item = new QTableWidgetItem;
    item->setText("-");
    cTable->setItem(i, j, item);
  }
}

void CurveEditor::deletePoint()
{
  QModelIndex index = pTable->currentIndex();
  int row = index.row();

  // Check if the point is attached to a curve:
  //--------------------------------------------
  bool attached = false;
  int curve = -1;
  int idx = pTable->item(row, 0)->text().toInt();
  for(int i = 0; i < cTable->rowCount(); i++) {
    int p0 = cTable->item(i, 3)->text().toInt();
    int p1 = cTable->item(i, 4)->text().toInt();
    int p2 = cTable->item(i, 5)->text().toInt();
    if((p0 == idx) || (p1 == idx) || (p2 == idx)) {
      curve = i + 1;
      attached = true;
      break;
    }
  }

  // Do not delete if attached:
  //---------------------------
  if(attached) {
    QString message = "Point is attached to curve " + QString::number(curve);
    #if  WITH_QT5 || WITH_QT6
    cout << message.toLatin1().data() << endl;
    #else
    cout << message.toAscii().data() << endl;
    #endif
    emit(statusMessage(message));
    return;
  }

  pTable->removeRow(row);
  renderArea->updatePoints(pTable);
}

void CurveEditor::deleteCurve()
{
  QModelIndex index = cTable->currentIndex();
  int row = index.row();
  cTable->removeRow(row);
  renderArea->updateCurves(cTable);
}