src/CurveCreator/CurveCreator_TreeView.cxx

   1 // Copyright (C) 2013-2016  CEA/DEN, EDF R&D, OPEN CASCADE
   2 //
   3 // This library is free software; you can redistribute it and/or
   4 // modify it under the terms of the GNU Lesser General Public
   5 // License as published by the Free Software Foundation; either
   6 // version 2.1 of the License, or (at your option) any later version.
   7 //
   8 // This library is distributed in the hope that it will be useful,
   9 // but WITHOUT ANY WARRANTY; without even the implied warranty of
  10 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  11 // Lesser General Public License for more details.
  12 //
  13 // You should have received a copy of the GNU Lesser General Public
  14 // License along with this library; if not, write to the Free Software
  15 // Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
  16 //
  17 // See http://www.salome-platform.org/ or email : webmaster.salome@opencascade.com
  18 //
  19
  20 #include "CurveCreator_TreeView.h"
  21 #include "CurveCreator_ICurve.hxx"
  22
  23 #include <SUIT_Session.h>
  24 #include <SUIT_ResourceMgr.h>
  25
  26 #include <QHeaderView>
  27 #include <QtAlgorithms>
  28
  29 #define ID_SECTION -1
  30
  31 CurveCreator_TreeViewModel::CurveCreator_TreeViewModel( CurveCreator_ICurve* theCurve, QObject* parent ) :
  32   QAbstractItemModel(parent), myCurve(theCurve)
  33 {
  34   SUIT_ResourceMgr* aResMgr = SUIT_Session::session()->resourceMgr();
  35   QPixmap aSplineIcon(aResMgr->loadPixmap("GEOM", tr("ICON_CC_SPLINE")));
  36   QPixmap aPolylineIcon(aResMgr->loadPixmap("GEOM", tr("ICON_CC_POLYLINE")));
  37   QPixmap aClosedSplineIcon(aResMgr->loadPixmap("GEOM", tr("ICON_CC_CLOSED_SPLINE")));
  38   QPixmap aClosedPolylineIcon(aResMgr->loadPixmap("GEOM", tr("ICON_CC_CLOSED_POLYLINE")));
  39   QPixmap aPointIcon(aResMgr->loadPixmap("GEOM", tr("ICON_CC_POINT")));
  40
  41 /*  QPixmap aSplineIcon(tr(":images/ICON_SPLINE"));
  42   QPixmap aPolylineIcon(tr(":images/ICON_POLYLINE"));
  43   QPixmap aClosedPolylineIcon(tr(":images/ICON_CLOSED_POLYLINE"));
  44   QPixmap aClosedSplineIcon(tr(":images/ICON_CLOSED_SPLINE"));
  45   QPixmap aPointIcon(tr(":images/ICON_POINT")); */
  46
  47   if( !aSplineIcon.isNull() )
  48     myCachedIcons[ICON_SPLINE] = aSplineIcon;
  49
  50   if( !aPolylineIcon.isNull() )
  51     myCachedIcons[ICON_POLYLINE] = aPolylineIcon;
  52
  53   if( !aPolylineIcon.isNull() )
  54     myCachedIcons[ICON_CLOSED_POLYLINE] = aClosedPolylineIcon;
  55
  56   if( !aPolylineIcon.isNull() )
  57     myCachedIcons[ICON_CLOSED_SPLINE] = aClosedSplineIcon;
  58
  59   if( !aPointIcon.isNull() )
  60     myCachedIcons[ICON_POINT] = aPointIcon;
  61
  62   setHeaderData(1, Qt::Horizontal, QVariant("Name"), Qt::DisplayRole);
  63   setHeaderData(2, Qt::Horizontal, QVariant("Nb points"), Qt::DisplayRole);
  64 }
  65
  66 int     CurveCreator_TreeViewModel::columnCount(const QModelIndex & parent ) const
  67 {
  68   if( parent.internalId() == ID_SECTION )
  69     return 2;
  70   else
  71     return 2;
  72 }
  73
  74 QVariant        CurveCreator_TreeViewModel::data(const QModelIndex & index, int role ) const
  75 {
  76   int aRow = index.row();
  77   int aColumn = index.column();
  78   if( myCurve ){
  79     if( index.internalId() == ID_SECTION ){
  80       if( role == Qt::DisplayRole ){
  81         if( aColumn == 0 )
  82           return QString::fromStdString(myCurve->getSectionName(aRow));
  83         else if( aColumn == 1 )
  84           return QString::number(myCurve->getNbPoints(aRow));
  85         return QVariant();
  86       }
  87       else if( role == Qt::DecorationRole ){
  88         if( aColumn == 0 ){
  89           CurveCreator::SectionType aSectionType = myCurve->getSectionType(aRow);
  90           if( aSectionType == CurveCreator::Polyline ){
  91             if( myCurve->isClosed(aRow) ){
  92               return myCachedIcons[ICON_CLOSED_POLYLINE];
  93             }
  94             else{
  95               return myCachedIcons[ICON_POLYLINE];
  96             }
  97           }
  98           else{
  99             if( myCurve->isClosed(aRow) ){
 100               return myCachedIcons[ICON_CLOSED_SPLINE];
 101             }
 102             else{
 103               return myCachedIcons[ICON_SPLINE];
 104             }
 105           }
 106         }
 107       }
 108     }
 109 /*    else{
 110       if( role == Qt::DisplayRole ){
 111         if( aColumn == 1 )
 112           return QVariant();
 113         //                    return "Point";
 114         else if( aColumn == 0 ){
 115           CurveCreator::Coordinates aCoords = myCurve->getCoordinates(index.internalId(),index.row() );
 116           QString anOut;
 117           if( myCurve->getDimension() == CurveCreator::Dim2d ){
 118             anOut = QString(tr("X=%1, Y=%2")).arg(aCoords[0]).arg(aCoords[1]);
 119           }
 120           else{
 121             anOut = QString(tr("X=%1, Y=%2, Z=%3")).arg(aCoords[0]).arg(aCoords[1]).arg(aCoords[2]);
 122           }
 123           return anOut;
 124         }
 125       }
 126       else if( role == Qt::DecorationRole ){
 127         if( aColumn == 0 ){
 128           return myCachedIcons[ICON_POINT];
 129         }
 130       }
 131     }*/
 132   }
 133   return QVariant();
 134 }
 135
 136 QModelIndex     CurveCreator_TreeViewModel::index(int row, int column, const QModelIndex & parent ) const
 137 {
 138   if( parent.isValid() ){
 139     return createIndex(row, column, parent.row() );
 140   }
 141   else{
 142     QModelIndex aParent = createIndex(row, column, ID_SECTION );
 143     return aParent;
 144   }
 145   return QModelIndex();
 146 }
 147
 148 QModelIndex     CurveCreator_TreeViewModel::parent(const QModelIndex & theIndex) const
 149 {
 150   if( !theIndex.isValid() )
 151     return QModelIndex();
 152
 153   if( theIndex.internalId() == ID_SECTION ){
 154     return QModelIndex();
 155   }
 156   return createIndex( theIndex.internalId(), 0, ID_SECTION  );
 157 }
 158
 159 int     CurveCreator_TreeViewModel::rowCount(const QModelIndex & parent ) const
 160 {
 161   int aRowCnt = 0;
 162   if( myCurve != NULL ){
 163     if( !parent.isValid() ){
 164       //Section level
 165       aRowCnt =  myCurve->getNbSections();
 166     }
 167     else{
 168       if( parent.internalId() == ID_SECTION ){
 169         //Points level
 170         aRowCnt = myCurve->getNbPoints(parent.row());
 171       }
 172     }
 173   }
 174   return aRowCnt;
 175 }
 176
 177 QModelIndex CurveCreator_TreeViewModel::sectionIndex( int theSection ) const
 178 {
 179   return createIndex( theSection, 0, ID_SECTION );
 180 }
 181
 182 QModelIndex CurveCreator_TreeViewModel::nbPointsIndex( int theSection ) const
 183 {
 184   return createIndex( theSection, 1, ID_SECTION );
 185 }
 186
 187 QModelIndex CurveCreator_TreeViewModel::pointIndex( int theSection, int thePoint ) const
 188 {
 189   return createIndex( thePoint, 0, theSection );
 190 }
 191
 192 bool CurveCreator_TreeViewModel::isSection( const QModelIndex& theIndx ) const
 193 {
 194   if( theIndx.internalId() == ID_SECTION )
 195     return true;
 196   return false;
 197 }
 198
 199 int CurveCreator_TreeViewModel::getSection( const QModelIndex& theIndx ) const
 200 {
 201   if( theIndx.internalId() == ID_SECTION )
 202     return theIndx.row();
 203   return theIndx.internalId();
 204 }
 205
 206 int CurveCreator_TreeViewModel::getPoint( const QModelIndex& theIndx ) const
 207 {
 208   if( theIndx.internalId() == ID_SECTION )
 209     return -1;
 210   return theIndx.row();
 211 }
 212
 213 void CurveCreator_TreeViewModel::setCurve( CurveCreator_ICurve* theCurve )
 214 {
 215   myCurve = theCurve;
 216 #if QT_VERSION < QT_VERSION_CHECK(5, 0, 0)
 217   reset();
 218 #endif
 219 }
 220
 221 /*****************************************************************************************/
 222 CurveCreator_TreeView::CurveCreator_TreeView( CurveCreator_ICurve* theCurve, QWidget *parent) :
 223   QTreeView(parent)
 224 {
 225   header()->hide();
 226 #if QT_VERSION < QT_VERSION_CHECK(5, 0, 0)
 227   header()->setResizeMode(QHeaderView::ResizeToContents);
 228 #else
 229   header()->setSectionResizeMode(QHeaderView::ResizeToContents);
 230 #endif
 231   setUniformRowHeights(true);
 232   setContextMenuPolicy( Qt::CustomContextMenu );
 233   CurveCreator_TreeViewModel* aModel = new CurveCreator_TreeViewModel(theCurve, this);
 234   setModel(aModel);
 235   setSelectionBehavior(SelectRows);
 236   setSelectionMode(SingleSelection);
 237   setRootIsDecorated(false);
 238   setItemsExpandable(false);
 239   setAllColumnsShowFocus(true);
 240   connect( selectionModel(), SIGNAL(selectionChanged(QItemSelection,QItemSelection)),
 241            this, SIGNAL(selectionChanged()) );
 242   connect( this, SIGNAL(activated(QModelIndex)), this, SLOT(onActivated(QModelIndex)));
 243 }
 244
 245 QList<int> CurveCreator_TreeView::getSelectedSections() const
 246 {
 247   QList<int> aSect;
 248   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 249   if( !aModel )
 250     return aSect;
 251 //  QModelIndexList anIndxs = selectionModel()->selectedIndexes();
 252   QModelIndexList anIndxs = selectionModel()->selectedRows();
 253   for( int i = 0 ; i < anIndxs.size() ; i++ ){
 254     if( aModel->isSection(anIndxs[i]) ){
 255       aSect << aModel->getSection( anIndxs[i] );
 256     }
 257   }
 258   return aSect;
 259 }
 260
 261 void CurveCreator_TreeView::pointsAdded( int theSection, int thePoint, int thePointsCnt )
 262 {
 263   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 264   if( aModel ){
 265     QModelIndex aSectIndx = aModel->sectionIndex( theSection );
 266     rowsInserted(aSectIndx, thePoint, thePoint + thePointsCnt - 1 );
 267 //    expand( aSectIndx );
 268     update( aModel->nbPointsIndex( theSection ) );
 269   }
 270 }
 271
 272 void CurveCreator_TreeView::pointDataChanged( int theSection, int thePoint )
 273 {
 274   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 275   if( aModel ){
 276     QModelIndex aPointIndx = aModel->pointIndex( theSection, thePoint );
 277     dataChanged( aPointIndx, aPointIndx );
 278   }
 279 }
 280
 281 void CurveCreator_TreeView::pointsRemoved( int theSection, int thePoint, int thePointsCnt )
 282 {
 283   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 284   if( aModel ){
 285     for( int i = 0 ; i < thePointsCnt ; i++ ){
 286       QModelIndex aSectIndx = aModel->pointIndex(theSection, thePoint + i);
 287       selectionModel()->select(aSectIndx,QItemSelectionModel::Deselect);
 288     }
 289     QModelIndex aSectIndx = aModel->sectionIndex( theSection );
 290     rowsRemoved(aSectIndx, thePoint, thePoint + thePointsCnt - 1 );
 291   }
 292 }
 293
 294 void CurveCreator_TreeView::sectionAdded( int theSection )
 295 {
 296   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 297   if( aModel ){
 298     int nbRows = aModel->rowCount();
 299     int aSection = (theSection == -1 ? (nbRows==0 ? 0 : nbRows-1) : theSection);
 300     rowsInserted(QModelIndex(), aSection, aSection );
 301     QModelIndex aSectIndx = aModel->sectionIndex(aSection);
 302     selectionModel()->select(aSectIndx, QItemSelectionModel::Rows | QItemSelectionModel::ClearAndSelect);
 303   }
 304 }
 305
 306 void CurveCreator_TreeView::sectionChanged( int theSection, int aSectCnt )
 307 {
 308   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 309   if( aModel ){
 310     QModelIndex aFirstSectIndx = aModel->sectionIndex( theSection );
 311     QModelIndex aLastSectIndx = aModel->sectionIndex( theSection + aSectCnt - 1);
 312     dataChanged( aFirstSectIndx, aLastSectIndx );
 313   }
 314 }
 315
 316 void CurveCreator_TreeView::sectionsRemoved( int theSection, int theSectionCnt )
 317 {
 318   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 319   if( aModel ){
 320     for( int i = 0 ; i < theSectionCnt ; i++ ){
 321       QModelIndex aSectIndx = aModel->sectionIndex(theSection + i);
 322       this->selectionModel()->select(aSectIndx,QItemSelectionModel::Deselect);
 323     }
 324     rowsRemoved( QModelIndex(), theSection, theSection+theSectionCnt-1 );
 325   }
 326 }
 327
 328 void CurveCreator_TreeView::setIndexState( const QModelIndex& theIndx, bool& isExpanded, bool& isSelected, bool& isCurrent )
 329 {
 330   setExpanded( theIndx, isExpanded );
 331   QItemSelectionModel::SelectionFlags aFlag = QItemSelectionModel::Select;
 332   if( !isSelected ){
 333     aFlag = QItemSelectionModel::Deselect;
 334   }
 335   selectionModel()->select( theIndx, aFlag );
 336 }
 337
 338 void CurveCreator_TreeView::getIndexInfo( const QModelIndex& theIndx, bool& isExpand, bool& isSelected, bool& isCurrent )
 339 {
 340   isExpand = isExpanded(theIndx);
 341   isSelected = selectionModel()->isSelected(theIndx);
 342   isCurrent = (theIndx == selectionModel()->currentIndex());
 343 }
 344
 345 void CurveCreator_TreeView::swapIndexes( const QModelIndex& theFirst, const QModelIndex& theSecond )
 346 {
 347   bool isFirstSelected;
 348   bool isFirstExpanded;
 349   bool isFirstCurrent;
 350   getIndexInfo( theFirst, isFirstExpanded, isFirstSelected, isFirstCurrent );
 351
 352   bool isSecondSelected;
 353   bool isSecondExpanded;
 354   bool isSecondCurrent;
 355   getIndexInfo( theSecond, isSecondExpanded, isSecondSelected, isSecondCurrent );
 356
 357   setIndexState( theFirst, isSecondExpanded, isSecondSelected, isSecondCurrent );
 358   setIndexState( theSecond, isFirstExpanded, isFirstSelected, isFirstCurrent );
 359   dataChanged(theFirst,theFirst);
 360   dataChanged(theSecond,theSecond);
 361 }
 362
 363 void CurveCreator_TreeView::sectionsSwapped( int theSection, int theOffset )
 364 {
 365   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 366   if( aModel ){
 367     QModelIndex aFirstIndex = aModel->sectionIndex( theSection );
 368     QModelIndex aSecondIndex = aModel->sectionIndex( theSection + theOffset );
 369     swapIndexes( aFirstIndex, aSecondIndex );
 370   }
 371 }
 372
 373 void CurveCreator_TreeView::setSelectedSections( const QList<int>& theList )
 374 {
 375   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 376   if( aModel ){
 377     selectionModel()->clearSelection();
 378     for( int i = 0 ; i < theList.size() ; i++ ){
 379       QModelIndex aSectIndx = aModel->sectionIndex(theList[i]);
 380       selectionModel()->select(aSectIndx, QItemSelectionModel::Select | QItemSelectionModel::Rows );
 381     }
 382   }
 383 }
 384
 385 bool pointLessThan(const QPair<int,int> &s1, const QPair<int,int> &s2)
 386 {
 387   if( s1.first < s2.first )
 388     return true;
 389   if( s1.first > s2.first )
 390     return false;
 391   return s1.second < s2.second;
 392 }
 393
 394 CurveCreator_TreeView::SelectionType CurveCreator_TreeView::getSelectionType() const
 395 {
 396   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 397   if( !aModel )
 398     return ST_NOSEL;
 399   bool isPointSel = false;
 400   bool isSectSel = false;
 401   bool isOneSection = true;
 402   int aSectNum = -1;
 403   QModelIndexList aLst = selectionModel()->selectedIndexes();
 404   for( int i = 0 ; i < aLst.size() ; i++ ){
 405     if( aModel->isSection( aLst[i] ) ){
 406       isSectSel = true;
 407     }
 408     else{
 409       isPointSel = true;
 410       if( aSectNum == -1 ){
 411         aSectNum = aModel->getSection(aLst[i]);
 412       }
 413       else{
 414         if( aSectNum != aModel->getSection( aLst[i] ) ){
 415           isOneSection = false;
 416         }
 417       }
 418     }
 419   }
 420   if( isSectSel && !isPointSel )
 421     return ST_SECTIONS;
 422   if( isPointSel && !isSectSel ){
 423     if( isOneSection ){
 424       return ST_POINTS_ONE_SECTION;
 425     }
 426     return ST_POINTS;
 427   }
 428   if( isPointSel && isSectSel )
 429     return ST_MIXED;
 430   return ST_NOSEL;
 431 }
 432
 433 void CurveCreator_TreeView::onActivated( QModelIndex theIndx )
 434 {
 435   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 436   if( !aModel )
 437     return;
 438   int aSect = aModel->getSection(theIndx);
 439   if( aModel->isSection(theIndx) ){
 440     emit sectionEntered( aSect );
 441   }
 442 }
 443
 444 void CurveCreator_TreeView::setCurve( CurveCreator_ICurve* theCurve )
 445 {
 446   CurveCreator_TreeViewModel* aModel = dynamic_cast<CurveCreator_TreeViewModel*>(model());
 447   if( aModel )
 448     aModel->setCurve(theCurve);
 449   reset();
 450 }
 451
 452 void CurveCreator_TreeView::reset()
 453 {
 454   QList<int> aSelSections = getSelectedSections();
 455   QTreeView::reset();
 456   setSelectedSections(aSelSections);
 457 }