src/VISU_I/VISU_TimeAnimation.cxx

   1 //  Copyright (C) 2003  CEA/DEN, EDF R&D
   2 //
   3 //
   4 //
   5 //  File   : VISU_TimeAnimation.cxx
   6 //  Author : Vitaly SMETANNIKOV
   7 //  Module : VISU
   8
   9 #include "VISU_TimeAnimation.h"
  10
  11 #include "VISUConfig.hh"
  12
  13 #include "VISU_Result_i.hh"
  14 #include "VISU_Prs3d_i.hh"
  15 #include "VISU_Mesh_i.hh"
  16 #include "VISU_ScalarMap_i.hh"
  17 #include "VISU_IsoSurfaces_i.hh"
  18 #include "VISU_DeformedShape_i.hh"
  19 #include "VISU_CutPlanes_i.hh"
  20 #include "VISU_Plot3D_i.hh"
  21 #include "VISU_CutLines_i.hh"
  22 #include "VISU_Vectors_i.hh"
  23 #include "VISU_StreamLines_i.hh"
  24 #include "VISU_ViewManager_i.hh"
  25 #include "VISU_View_i.hh"
  26
  27 #include "VISU_ScalarBarActor.hxx"
  28 #include "VISU_Actor.h"
  29
  30 #include "SalomeApp_Study.h"
  31
  32 #include "SVTK_ViewWindow.h"
  33
  34 #include "SALOME_Event.hxx"
  35
  36 #include "SUIT_ResourceMgr.h"
  37 #include "SUIT_Application.h"
  38 #include "SUIT_Session.h"
  39 #include "SUIT_Study.h"
  40
  41 #include "SALOMEDSClient_AttributeComment.hxx"
  42 #include "SALOMEDSClient_AttributeName.hxx"
  43
  44 #include <qpixmap.h>
  45
  46 using namespace std;
  47
  48
  49 //************************************************************************
  50 VISU_TimeAnimation::VISU_TimeAnimation (_PTR(Study) theStudy,
  51                                         VISU::View3D_ptr theView3D)
  52 {
  53   myStudy = theStudy;
  54   myIsActive = false;
  55   myFrame = 0;
  56   mySpeed = 1;
  57   myProportional = false;
  58   myView = 0;
  59
  60   if (!CORBA::is_nil(theView3D)) {
  61     VISU::View3D_i* pView = dynamic_cast<VISU::View3D_i*>(GetServant(theView3D).in());
  62     //QAD_StudyFrame* aStudyFrame = pView->GetStudyFrame();
  63     //myView = VISU::GetViewFrame(aStudyFrame);
  64     SUIT_ViewWindow* aVW = pView->myViewWindow;
  65     myView = VISU::GetViewWindow(aVW);
  66   }
  67
  68   myMaxVal = 0;
  69   myMinVal = 0;
  70   myTimeMin = 0;
  71   myTimeMax = 0;
  72   myLastError = "";
  73   myCycling = false;
  74
  75   myAnimEntry = "";
  76 }
  77
  78
  79 //************************************************************************
  80 VISU_TimeAnimation::~VISU_TimeAnimation()
  81 {
  82   for (int i = 0; i < getNbFields(); i++) {
  83     clearData(myFieldsLst[i]);
  84   }
  85 }
  86
  87
  88 //************************************************************************
  89 void VISU_TimeAnimation::addField (_PTR(SObject) theField)
  90 {
  91   FieldData aNewData;
  92   aNewData.myField = theField;
  93   aNewData.myNbFrames = 0;
  94   aNewData.myPrsType = VISU::TSCALARMAP;
  95   aNewData.myOffset[0] = aNewData.myOffset[1] = aNewData.myOffset[2] = 0;
  96   VISU::Storable::TRestoringMap aMap = getMapOfValue(aNewData.myField);
  97   aNewData.myNbTimes = VISU::Storable::FindValue(aMap,"myNbTimeStamps").toLong();
  98   myFieldsLst.append(aNewData);
  99
 100   //find Min/Max timestamps
 101   if ((myTimeMin == 0) && (myTimeMax == 0)) {
 102     _PTR(ChildIterator) anIter = myStudy->NewChildIterator(theField);
 103     anIter->Next(); // First is reference on support
 104     myTimeMin = getTimeValue(anIter->Value());
 105     for(; anIter->More(); anIter->Next()) {
 106       myTimeMax = getTimeValue(anIter->Value());
 107     }
 108   }
 109 }
 110
 111 //************************************************************************
 112 void VISU_TimeAnimation::addField (SALOMEDS::SObject_ptr theField)
 113 {
 114   _PTR(SObject) aField = VISU::GetClientSObject(theField, myStudy);
 115   addField(aField);
 116 }
 117
 118
 119 //************************************************************************
 120 void VISU_TimeAnimation::clearData(FieldData& theData) {
 121   theData.myTiming.clear();
 122   vtkRenderer* aRen = myView->getRenderer();
 123   if (!theData.myActors.empty()) {
 124     for (int i = 0, iEnd = theData.myActors.size(); i < iEnd; i++) {
 125       if (theData.myActors[i] != 0) {
 126         theData.myActors[i]->RemoveFromRender(aRen);
 127         theData.myActors[i]->Delete();
 128       }
 129     }
 130     theData.myActors.clear();
 131   }
 132   if (!theData.myPrs.empty()) {
 133     for (int i = 0, iEnd = theData.myPrs.size(); i < iEnd; i++)
 134       if (theData.myPrs[i] != 0) {
 135         theData.myPrs[i]->_remove_ref();
 136       }
 137     theData.myPrs.clear();
 138   }
 139   theData.myNbFrames = 0;
 140   myView->update();
 141 }
 142
 143
 144 //************************************************************************
 145 void VISU_TimeAnimation::generatePresentations(CORBA::Long theFieldNum) {
 146   FieldData& aData = myFieldsLst[theFieldNum];
 147
 148   // Delete previous presentations
 149   clearData(aData);
 150
 151   VISU::Result_i* pResult = createPresent(aData.myField);
 152   VISU::Storable::TRestoringMap aMap = getMapOfValue(aData.myField);
 153   aData.myNbFrames = aData.myNbTimes;
 154     //VISU::Storable::FindValue(aMap,"myNbTimeStamps").toLong();
 155
 156   aData.myPrs.resize(aData.myNbTimes,NULL);
 157   aData.myTiming.resize(aData.myNbTimes);
 158
 159   _PTR(ChildIterator) anIter = myStudy->NewChildIterator(aData.myField);
 160   _PTR(SObject) aTimeStamp;
 161   anIter->Next(); // First is reference on support
 162   long i = 0;
 163   double aMin = VTK_LARGE_FLOAT, aMax = -VTK_LARGE_FLOAT;
 164   for (; anIter->More(); anIter->Next()) {
 165     if (i == aData.myNbTimes) {
 166       MESSAGE("There are extra timestamps in field");
 167       return;
 168     }
 169     aTimeStamp = anIter->Value();
 170     if (!aTimeStamp) continue;
 171
 172     aData.myTiming[i] = getTimeValue(aTimeStamp);
 173     if (isRangeDefined()) {
 174       if (aData.myTiming[i] < myMinVal) continue;
 175       if (aData.myTiming[i] > myMaxVal) break;
 176     }
 177
 178     VISU::Storable::TRestoringMap aTimeMap = getMapOfValue(aTimeStamp);
 179     QString aMeshName = VISU::Storable::FindValue(aTimeMap,"myMeshName");
 180     VISU::Entity anEntity = (VISU::Entity) VISU::Storable::FindValue(aTimeMap,"myEntityId").toInt();
 181     QString aFieldName = VISU::Storable::FindValue(aTimeMap,"myFieldName");
 182     int aTimeStampId = VISU::Storable::FindValue(aTimeMap,"myTimeStampId").toInt();
 183
 184     switch (aData.myPrsType) {
 185     case VISU::TSCALARMAP: // ScalarMap
 186       {
 187         VISU::ScalarMap_i* aPresent = new VISU::ScalarMap_i(pResult, false);
 188         aPresent->Create(aMeshName.latin1(), anEntity,
 189                          aFieldName.latin1(), aTimeStampId);
 190         //VISU::ScalarMap_var aTmp = aPresent->_this();
 191         //aPresent->_remove_ref();
 192         aData.myPrs[i] = aPresent;
 193       }
 194       break;
 195
 196     case VISU::TISOSURFACE: // Iso Surfaces
 197       {
 198         VISU::IsoSurfaces_i* aPresent = new VISU::IsoSurfaces_i(pResult, false);
 199         aPresent->Create(aMeshName.latin1(), anEntity,
 200                          aFieldName.latin1(), aTimeStampId);
 201         //VISU::IsoSurfaces_var aTmp = aPresent->_this();
 202         //aPresent->_remove_ref();
 203         aData.myPrs[i] = aPresent;
 204       }
 205       break;
 206
 207     case VISU::TCUTPLANES: // Cut Planes
 208       {
 209         VISU::CutPlanes_i* aPresent = new VISU::CutPlanes_i(pResult, false);
 210         aPresent->Create(aMeshName.latin1(), anEntity,
 211                          aFieldName.latin1(), aTimeStampId);
 212         //VISU::CutPlanes_var aTmp = aPresent->_this();
 213         //aPresent->_remove_ref();
 214         aData.myPrs[i] = aPresent;
 215       }
 216       break;
 217
 218     case VISU::TPLOT3D: // Cut Planes
 219       {
 220         VISU::Plot3D_i* aPresent = new VISU::Plot3D_i (pResult, false);
 221         aPresent->Create(aMeshName.latin1(), anEntity,
 222                          aFieldName.latin1(), aTimeStampId);
 223         aData.myPrs[i] = aPresent;
 224       }
 225       break;
 226
 227     case VISU::TDEFORMEDSHAPE: // Deformed Shape
 228       {
 229         VISU::DeformedShape_i* aPresent = new VISU::DeformedShape_i(pResult, false);
 230         aPresent->Create(aMeshName.latin1(), anEntity,
 231                          aFieldName.latin1(), aTimeStampId);
 232         //VISU::DeformedShape_var aTmp = aPresent->_this();
 233         //aPresent->_remove_ref();
 234         aData.myPrs[i] = aPresent;
 235       }
 236       break;
 237
 238     case VISU::TVECTORS: // Vectors
 239       {
 240         VISU::Vectors_i* aPresent = new VISU::Vectors_i(pResult, false);
 241         aPresent->Create(aMeshName.latin1(), anEntity,
 242                          aFieldName.latin1(), aTimeStampId);
 243         //VISU::Vectors_var aTmp = aPresent->_this();
 244         //aPresent->_remove_ref();
 245         aData.myPrs[i] = aPresent;
 246       }
 247       break;
 248
 249     case VISU::TSTREAMLINES: // Stream Lines
 250       {
 251         VISU::StreamLines_i* aPresent = new VISU::StreamLines_i(pResult, false);
 252         aPresent->Create(aMeshName.latin1(), anEntity,
 253                          aFieldName.latin1(), aTimeStampId);
 254         //VISU::StreamLines_var aTmp = aPresent->_this();
 255         //aPresent->_remove_ref();
 256         aData.myPrs[i] = aPresent;
 257       }
 258       break;
 259     default:
 260       MESSAGE("Not implemented for this presentation type: " << aData.myPrsType);
 261       return;
 262     }
 263     if (aData.myPrs[i]->GetMin() < aMin) aMin = aData.myPrs[i]->GetMin();
 264     if (aData.myPrs[i]->GetMax() > aMax) aMax = aData.myPrs[i]->GetMax();
 265     i++;
 266   }
 267   aData.myNbFrames = i;
 268
 269   int rangeType = VISU::GetResourceMgr()->integerValue("VISU" , "scalar_range_type", 0);
 270   if ( rangeType != 1 ) {
 271     for (i = 0; i < aData.myNbFrames; i++) {
 272       aData.myPrs[i]->SetRange(aMin, aMax);
 273       aData.myPrs[i]->SetOffset(aData.myOffset);
 274     }
 275     if (aData.myPrsType == VISU::TISOSURFACE)
 276       for (i = 0; i < aData.myNbFrames; i++)
 277         if (VISU::IsoSurfaces_i* aPrs = dynamic_cast<VISU::IsoSurfaces_i*>(aData.myPrs[i]))
 278            aPrs->SetSubRange(aMin, aMax);
 279   }
 280 }
 281
 282
 283 //************************************************************************
 284 CORBA::Boolean VISU_TimeAnimation::generateFrames() {
 285   if (!myView) {
 286     MESSAGE("Viewer is not defined for animation");
 287     return false;
 288   }
 289
 290   myLastError = QString("Frame(s) for ");
 291   bool aNoError = true;
 292
 293   clearView();
 294
 295   for (int i = 0; i < getNbFields(); i++) {
 296     FieldData& aData = myFieldsLst[i];
 297     aData.myActors.resize(aData.myNbFrames,NULL);
 298     for (long j = 0; j < aData.myNbFrames; j++) {
 299       VISU_Actor* aActor = NULL;
 300       try{
 301         aData.myPrs[j]->SetOffset(aData.myOffset);
 302         aActor = aData.myPrs[j]->CreateActor();
 303         myView->AddActor(aActor);
 304         if(j == 0)
 305           aActor->VisibilityOn();
 306         else
 307           aActor->VisibilityOff();
 308       }catch(...){ //catch(std::runtime_error& exc){
 309         aNoError = false;
 310         myLastError += QString("%1 ").arg(aData.myTiming[j]);
 311       }
 312       aData.myActors[j] = aActor;
 313     }
 314   }
 315   myFrame = 0;
 316   myLastError += QString(" timestamp(s) cannot be created.");
 317   emit frameChanged(myFrame, myFieldsLst[0].myTiming[myFrame]);
 318   myView->Repaint();
 319   return aNoError;
 320 }
 321
 322 //************************************************************************
 323 void VISU_TimeAnimation::clearView() {
 324   if (!myView) {
 325     MESSAGE("Viewer is not defined for animation");
 326     return;
 327   }
 328   vtkRenderer* aRen = myView->getRenderer();
 329   for (int i = 0; i < getNbFields(); i++) {
 330     FieldData& aData = myFieldsLst[i];
 331     if (!aData.myActors.empty()) {
 332       for (int i = 0, iEnd = aData.myActors.size(); i < iEnd; i++) {
 333         if (aData.myActors[i] != 0) {
 334           aData.myActors[i]->RemoveFromRender(aRen);
 335           aData.myActors[i]->Delete();
 336         }
 337       }
 338       aData.myActors.clear();
 339     }
 340   }
 341   VISU::RepaintView(myView);
 342 }
 343
 344 //************************************************************************
 345 void VISU_TimeAnimation::stopAnimation() {
 346   myIsActive = false;
 347 }
 348
 349 //************************************************************************
 350 void VISU_TimeAnimation::startAnimation() {
 351   if (!myIsActive) {
 352     myIsActive = true;
 353     QThread::start();
 354   }
 355 }
 356
 357 //************************************************************************
 358 void VISU_TimeAnimation::nextFrame() {
 359   stopAnimation();
 360   if (myFrame < (myFieldsLst[0].myNbFrames-1)) {
 361     int i;
 362     for (i = 0; i < getNbFields(); i++)
 363       if (myFieldsLst[i].myActors[myFrame] != 0)
 364         myFieldsLst[i].myActors[myFrame]->VisibilityOff();
 365
 366     myFrame++;
 367     for (i = 0; i < getNbFields(); i++)
 368       if (myFieldsLst[i].myActors[myFrame] != 0)
 369         myFieldsLst[i].myActors[myFrame]->VisibilityOn();
 370
 371     emit frameChanged(myFrame, myFieldsLst[0].myTiming[myFrame]);
 372     myView->Repaint();
 373   }
 374 }
 375
 376 //************************************************************************
 377 void VISU_TimeAnimation::prevFrame() {
 378   stopAnimation();
 379   if (myFrame > 0) {
 380     int i;
 381     for (i = 0; i < getNbFields(); i++)
 382       if (myFieldsLst[i].myActors[myFrame] != 0)
 383         myFieldsLst[i].myActors[myFrame]->VisibilityOff();
 384
 385     myFrame--;
 386     for (i = 0; i < getNbFields(); i++)
 387       if (myFieldsLst[i].myActors[myFrame] != 0)
 388         myFieldsLst[i].myActors[myFrame]->VisibilityOn();
 389
 390     emit frameChanged(myFrame, myFieldsLst[0].myTiming[myFrame]);
 391     myView->Repaint();
 392   }
 393 }
 394
 395 //************************************************************************
 396 void VISU_TimeAnimation::firstFrame() {
 397   stopAnimation();
 398   int i;
 399   for (i = 0; i < getNbFields(); i++)
 400     if(!myFieldsLst[i].myActors.empty())
 401       if (myFieldsLst[i].myActors[myFrame] != 0)
 402         myFieldsLst[i].myActors[myFrame]->VisibilityOff();
 403   myFrame = 0;
 404   for (i = 0; i < getNbFields(); i++)
 405     if(!myFieldsLst[i].myActors.empty())
 406       if (myFieldsLst[i].myActors[myFrame] != 0)
 407         myFieldsLst[i].myActors[myFrame]->VisibilityOn();
 408   if(!myFieldsLst[0].myTiming.empty()){
 409     emit frameChanged(myFrame, myFieldsLst[0].myTiming[myFrame]);
 410     myView->Repaint();
 411   }
 412 }
 413
 414 //************************************************************************
 415 void VISU_TimeAnimation::lastFrame() {
 416   stopAnimation();
 417   int i;
 418   for (i = 0; i < getNbFields(); i++)
 419    if (myFieldsLst[i].myActors[myFrame] != 0)
 420       myFieldsLst[i].myActors[myFrame]->VisibilityOff();
 421
 422   myFrame = myFieldsLst[0].myNbFrames-1;
 423   for (i = 0; i < getNbFields(); i++)
 424     if (myFieldsLst[i].myActors[myFrame] != 0)
 425       myFieldsLst[i].myActors[myFrame]->VisibilityOn();
 426
 427   emit frameChanged(myFrame, myFieldsLst[0].myTiming[myFrame]);
 428   myView->Repaint();
 429 }
 430
 431
 432 //************************************************************************
 433 // For Batchmode using
 434 void VISU_TimeAnimation::gotoFrame(CORBA::Long theFrame) {
 435   if ((theFrame < 0) || (theFrame > (getNbFrames()-1)))
 436     return;
 437   stopAnimation();
 438   qApp->lock();
 439   qApp->syncX();
 440   int i;
 441   for (i = 0; i < getNbFields(); i++)
 442     if (myFieldsLst[i].myActors[myFrame] != 0)
 443       myFieldsLst[i].myActors[myFrame]->VisibilityOff();
 444
 445   myFrame = theFrame;
 446   for (i = 0; i < getNbFields(); i++)
 447     if (myFieldsLst[i].myActors[myFrame] != 0)
 448       myFieldsLst[i].myActors[myFrame]->VisibilityOn();
 449
 450   emit frameChanged(myFrame, myFieldsLst[0].myTiming[myFrame]);
 451   myView->Repaint();
 452   qApp->flushX();
 453   qApp->processEvents(3);
 454   qApp->unlock();
 455 }
 456
 457
 458 //************************************************************************
 459 VISU::ScalarMap_ptr VISU_TimeAnimation::getPresentation(CORBA::Long theField, CORBA::Long theFrame) {
 460   if ((theField > getNbFields()) || (theField < 0))
 461     return VISU::ScalarMap::_nil();
 462   if ((theFrame < 0) || (theFrame > (myFieldsLst[theField].myNbFrames - 1)))
 463     return VISU::ScalarMap::_nil();
 464   return myFieldsLst[theField].myPrs[theFrame]->_this();
 465 }
 466
 467
 468 //************************************************************************
 469 CORBA::Long VISU_TimeAnimation::getNbFrames() {
 470   return (getNbFields() > 0)? myFieldsLst[0].myNbFrames : 0;
 471 }
 472
 473
 474 //************************************************************************
 475 void VISU_TimeAnimation::run()
 476 {
 477   if (!myView) {
 478     MESSAGE("Viewer is not defined for animation");
 479     return;
 480   }
 481   double k = 1;
 482   bool   isDumping = !myDumpPath.isEmpty();
 483   double aOneVal = 1;
 484   if (myFieldsLst[0].myNbFrames > 2)
 485     aOneVal = myFieldsLst[0].myTiming[1] - myFieldsLst[0].myTiming[0];
 486   qApp->lock();
 487   while (myIsActive) {
 488     emit frameChanged(myFrame, myFieldsLst[0].myTiming[myFrame]);
 489     for (int i = 0; i < getNbFields(); i++) {
 490       FieldData& aData = myFieldsLst[i];
 491       if (myFrame > 0) {
 492         if (aData.myActors[myFrame-1] != 0)
 493           aData.myActors[myFrame-1]->VisibilityOff();
 494       } else {
 495         if (aData.myActors[aData.myNbFrames-1] != 0)
 496           aData.myActors[aData.myNbFrames-1]->VisibilityOff();
 497       }
 498       if (aData.myActors[myFrame] != 0) {
 499         aData.myActors[myFrame]->VisibilityOn();
 500       }
 501     }
 502     myView->Repaint(false);
 503
 504     int delay = 100;
 505     if (isDumping) {
 506       QPixmap px = QPixmap::grabWindow(myView->winId());
 507       QString aFile(myDumpPath);
 508       QString aName = QString("%1").arg(myFieldsLst[0].myTiming[myFrame]);
 509       int aPos = -1;
 510       while ((aPos = aName.find(".")) > -1 )
 511         aName.replace(aPos, 1, "_");
 512       aFile += aName;
 513       aFile += ".jpeg";
 514       px.save(aFile, "JPEG");
 515     } else {
 516       k = 1;
 517       if (myProportional) {
 518         switch (myFrame) {
 519         case 0:
 520           break;
 521         case 1:
 522           if (myFieldsLst[0].myNbFrames > 2)
 523             k = (myFieldsLst[0].myTiming[myFrame+1] -
 524                  myFieldsLst[0].myTiming[myFrame]) / aOneVal;
 525           break;
 526         default:
 527           if (myFrame < (myFieldsLst[0].myNbFrames - 1))
 528             k = (myFieldsLst[0].myTiming[myFrame+1] -
 529                  myFieldsLst[0].myTiming[myFrame]) / aOneVal;
 530         }
 531       }
 532       delay = (int)(1000. * k / mySpeed);
 533     }
 534     qApp->unlock();
 535     msleep(delay);
 536     qApp->lock();
 537
 538     if (!myIsActive) break;
 539
 540     myFrame++;
 541     if (myFrame == myFieldsLst[0].myNbFrames) {
 542       if (!myCycling) {
 543         myIsActive = false;
 544         myFrame--;
 545         break;
 546       } else
 547         myFrame = 0;
 548     }
 549   }
 550   emit stopped();
 551   qApp->unlock();
 552   QThread::exit();
 553 }
 554
 555 //************************************************************************
 556 VISU::Result_i* VISU_TimeAnimation::createPresent (_PTR(SObject) theField)
 557 {
 558   _PTR(SObject) aSObj = theField->GetFather();
 559   aSObj = aSObj->GetFather();
 560   aSObj = aSObj->GetFather();
 561   CORBA::Object_var anObject = VISU::ClientSObjectToObject(aSObj);
 562   if (CORBA::is_nil(anObject)) return NULL;
 563   return dynamic_cast<VISU::Result_i*>(VISU::GetServant(anObject).in());
 564 }
 565
 566 //************************************************************************
 567 VISU::Storable::TRestoringMap VISU_TimeAnimation::getMapOfValue (_PTR(SObject) theSObject)
 568 {
 569   VISU::Storable::TRestoringMap aMap;
 570   if (theSObject) {
 571     _PTR(GenericAttribute) anAttr;
 572     if (theSObject->FindAttribute(anAttr, "AttributeComment")) {
 573       _PTR(AttributeComment) aComment (anAttr);
 574       std::string aString = aComment->Value();
 575       QString strIn (aString.c_str());
 576       VISU::Storable::StrToMap(strIn, aMap);
 577     }
 578   }
 579   return aMap;
 580 }
 581
 582 //************************************************************************
 583 double VISU_TimeAnimation::getTimeValue (_PTR(SObject) theTimeStamp)
 584 {
 585   _PTR(GenericAttribute) anAttr;
 586   if (theTimeStamp->FindAttribute(anAttr, "AttributeName")) {
 587     _PTR(AttributeName) aName (anAttr);
 588     QString aNameString (aName->Value().c_str());
 589     int time_len = aNameString.find(',');
 590     if (time_len > -1)
 591       return aNameString.left(time_len).toDouble();
 592     else
 593       return aNameString.toDouble();
 594   }
 595   return -1.0;
 596 }
 597
 598 //************************************************************************
 599 void VISU_TimeAnimation::setSpeed(CORBA::Long theSpeed)
 600 {
 601   mySpeed = (theSpeed<1)? 1 : theSpeed;
 602 }
 603
 604 //************************************************************************
 605 int VISU_TimeAnimation::myNBAnimations = 0;
 606 QString VISU_TimeAnimation::GenerateName()
 607 {
 608   return VISU::GenerateName("Animation", myNBAnimations++);
 609 }
 610
 611 //************************************************************************
 612 std::string GetPresentationComment (VISU::VISUType thePrsType)
 613 {
 614   std::string aPrsCmt;
 615   switch (thePrsType) {
 616   case VISU::TSCALARMAP:
 617     aPrsCmt = VISU::ScalarMap_i::myComment;
 618     break;
 619   case VISU::TISOSURFACE:
 620     aPrsCmt = VISU::IsoSurfaces_i::myComment;
 621     break;
 622   case VISU::TCUTPLANES:
 623     aPrsCmt = VISU::CutPlanes_i::myComment;
 624     break;
 625   case VISU::TPLOT3D:
 626     aPrsCmt = VISU::Plot3D_i::myComment;
 627     break;
 628   case VISU::TDEFORMEDSHAPE:
 629     aPrsCmt = VISU::DeformedShape_i::myComment;
 630     break;
 631   case VISU::TVECTORS:
 632     aPrsCmt = VISU::Vectors_i::myComment;
 633     break;
 634   case VISU::TSTREAMLINES:
 635     aPrsCmt = VISU::StreamLines_i::myComment;
 636     break;
 637   default:
 638     aPrsCmt = "Unknown presentation";
 639     break;
 640   }
 641   return aPrsCmt;
 642 }
 643
 644 //************************************************************************
 645 SALOMEDS::SObject_ptr VISU_TimeAnimation::publishInStudy()
 646 {
 647   if (myStudy->GetProperties()->IsLocked())
 648     return SALOMEDS::SObject::_nil();
 649
 650   _PTR(StudyBuilder) aStudyBuilder = myStudy->NewBuilder();
 651   aStudyBuilder->NewCommand();  // There is a transaction
 652   _PTR(SComponent) aSComponent = VISU::ClientFindOrCreateVisuComponent(myStudy);
 653   std::string aSComponentEntry = aSComponent->GetID();
 654
 655   QString aComment;
 656   aComment.sprintf("myComment=ANIMATION;myType=%d;myMinVal=%g;myMaxVal=%g",
 657                    VISU::TANIMATION,myMinVal,myMaxVal);
 658
 659   string anEntry = VISU::CreateAttributes(myStudy,aSComponentEntry.c_str(),"","",
 660                                           GenerateName(),"",aComment,true);
 661   myAnimEntry = anEntry.c_str();
 662   _PTR(SObject) aAnimSObject = myStudy->FindObjectID(anEntry.c_str());
 663
 664   for (int i = 0; i < getNbFields(); i++) {
 665     FieldData& aData = myFieldsLst[i];
 666     _PTR(SObject) newObj = aStudyBuilder->NewObject(aAnimSObject);
 667     aStudyBuilder->Addreference(newObj, aData.myField);
 668     if (aData.myPrs.empty()) {
 669       VISU::CreateAttributes(myStudy, newObj->GetID().c_str(),"","",
 670                              GetPresentationComment(aData.myPrsType).c_str(),"","",true);
 671     } else {
 672       ostringstream strOut;
 673       aData.myPrs[0]->ToStream(strOut);
 674       string aPrsComment = strOut.str();
 675       VISU::CreateAttributes(myStudy, newObj->GetID().c_str(),"","",
 676                              aData.myPrs[0]->GetComment(),"",aPrsComment.c_str(),true);
 677     }
 678   }
 679   aStudyBuilder->CommitCommand();
 680
 681   return VISU::GetSObject(aAnimSObject);
 682 }
 683
 684 //************************************************************************
 685 void VISU_TimeAnimation::saveAnimation()
 686 {
 687   if (myStudy->GetProperties()->IsLocked())  return;
 688   if (myAnimEntry.isEmpty()) return;
 689
 690   _PTR(SObject) aAnimSObject = myStudy->FindObjectID(myAnimEntry.latin1());
 691   if (!aAnimSObject) return;
 692
 693   _PTR(StudyBuilder) aStudyBuilder = myStudy->NewBuilder();
 694   aStudyBuilder->NewCommand();  // There is a transaction
 695   _PTR(SComponent) aSComponent = VISU::ClientFindOrCreateVisuComponent(myStudy);
 696   std::string aSComponentEntry = aSComponent->GetID();
 697
 698   QString aComment;
 699   aComment.sprintf("myComment=ANIMATION;myType=%d;myMinVal=%g;myMaxVal=%g",
 700                    VISU::TANIMATION,myMinVal,myMaxVal);
 701
 702   _PTR(GenericAttribute) anAttr;
 703   anAttr = aStudyBuilder->FindOrCreateAttribute(aAnimSObject, "AttributeComment");
 704   _PTR(AttributeComment) aCmnt (anAttr);
 705   aCmnt->SetValue(aComment.latin1());
 706
 707   _PTR(ChildIterator) anIter = myStudy->NewChildIterator(aAnimSObject);
 708   int i;
 709   for (i = 0, anIter->Init(); anIter->More(); anIter->Next(), i++) {
 710     FieldData& aData = myFieldsLst[i];
 711
 712     _PTR(SObject) aRefObj = anIter->Value();
 713     _PTR(ChildIterator) anPrsIter = myStudy->NewChildIterator(aRefObj);
 714     anPrsIter->Init();
 715
 716     string aPrsComment, aPrsNameTxt;
 717     if (aData.myPrs.empty()) {
 718       aPrsComment = "";
 719       aPrsNameTxt = GetPresentationComment(aData.myPrsType);
 720     } else {
 721       ostringstream strOut;
 722       aData.myPrs[0]->ToStream(strOut);
 723       aPrsComment = strOut.str();
 724
 725       aPrsNameTxt = aData.myPrs[0]->GetComment();
 726     }
 727
 728     if (anPrsIter->More()) {
 729       _PTR(SObject) aPrsObj = anPrsIter->Value();
 730       anAttr = aStudyBuilder->FindOrCreateAttribute(aPrsObj, "AttributeComment");
 731       aCmnt = _PTR(AttributeComment)(anAttr);
 732       aCmnt->SetValue(aPrsComment.c_str());
 733
 734       anAttr = aStudyBuilder->FindOrCreateAttribute(aPrsObj, "AttributeName");
 735       _PTR(AttributeName) aPrsName (anAttr);
 736       aPrsName->SetValue(aPrsNameTxt);
 737
 738     } else {
 739       VISU::CreateAttributes(myStudy, aRefObj->GetID().c_str(),"","",
 740                              aPrsNameTxt.c_str(),"",aPrsComment.c_str(),true);
 741     }
 742   }
 743   aStudyBuilder->CommitCommand();
 744 }
 745
 746 //************************************************************************
 747 void VISU_TimeAnimation::restoreFromStudy(SALOMEDS::SObject_ptr theField)
 748 {
 749   _PTR(SObject) aAnimSObject = VISU::GetClientSObject(theField, myStudy);
 750   restoreFromStudy(aAnimSObject);
 751 }
 752
 753 void VISU_TimeAnimation::restoreFromStudy(_PTR(SObject) theField)
 754 {
 755   _PTR(SObject) aAnimSObject = theField;
 756
 757   VISU::Storable::TRestoringMap aMap;
 758   _PTR(GenericAttribute) anAttr;
 759   if (!aAnimSObject->FindAttribute(anAttr, "AttributeComment")) return;
 760
 761   _PTR(AttributeComment) aComment (anAttr);
 762   string aComm = aComment->Value();
 763   QString strIn (aComm.c_str());
 764   VISU::Storable::StrToMap(strIn,aMap);
 765   bool isExist;
 766
 767   myMinVal = VISU::Storable::FindValue(aMap,"myMinVal",&isExist).toDouble();
 768   myMaxVal = VISU::Storable::FindValue(aMap,"myMaxVal",&isExist).toDouble();
 769
 770   _PTR(ChildIterator) anIter = myStudy->NewChildIterator(aAnimSObject);
 771   for (anIter->Init(); anIter->More(); anIter->Next()) {
 772     _PTR(SObject) aRefObj = anIter->Value();
 773     _PTR(SObject) aFieldObj;
 774     if (!aRefObj->ReferencedObject(aFieldObj) ) continue;
 775     addField(aFieldObj);
 776     FieldData& aData = getFieldData(getNbFields()-1);
 777
 778     // Get Presentation object
 779     _PTR(ChildIterator) anPrsIter = myStudy->NewChildIterator(aRefObj);
 780     anPrsIter->Init();
 781     if (!anPrsIter->More()) continue;
 782     _PTR(SObject) aPrsObj = anPrsIter->Value();
 783     if (!aPrsObj->FindAttribute(anAttr, "AttributeName")) continue;
 784     _PTR(AttributeName) aName (anAttr);
 785     string aStr = aName->Value();
 786     QString strName (aStr.c_str());
 787
 788     if (strName == VISU::ScalarMap_i::myComment.c_str())
 789       aData.myPrsType = VISU::TSCALARMAP;
 790     else if (strName == VISU::IsoSurfaces_i::myComment.c_str())
 791       aData.myPrsType = VISU::TISOSURFACE;
 792     else if (strName == VISU::CutPlanes_i::myComment.c_str())
 793       aData.myPrsType = VISU::TCUTPLANES;
 794     else if (strName == VISU::Plot3D_i::myComment.c_str())
 795       aData.myPrsType = VISU::TPLOT3D;
 796     else if (strName == VISU::DeformedShape_i::myComment.c_str())
 797       aData.myPrsType = VISU::TDEFORMEDSHAPE;
 798     else if (strName == VISU::Vectors_i::myComment.c_str())
 799       aData.myPrsType = VISU::TVECTORS;
 800     else if (strName == VISU::StreamLines_i::myComment.c_str())
 801       aData.myPrsType = VISU::TSTREAMLINES;
 802     else
 803       continue;
 804     generatePresentations(getNbFields()-1);
 805
 806     if (!aPrsObj->FindAttribute(anAttr, "AttributeComment")) continue;
 807     _PTR(AttributeComment) aPrsComment (anAttr);
 808     string aPrsComm = aPrsComment->Value();
 809     if (aPrsComm.length() > 0) {
 810       QString strPrsIn (aPrsComm.c_str());
 811       VISU::Storable::TRestoringMap aPrsMap;
 812       VISU::Storable::StrToMap(strPrsIn,aPrsMap);
 813
 814       aData.myPrs[0]->Restore(aPrsMap);
 815     }
 816     aData.myPrs[0]->GetOffset(aData.myOffset);
 817     for (int i = 1; i < aData.myNbFrames; i++) {
 818       aData.myPrs[i]->SameAs(aData.myPrs[0]);
 819     }
 820   }
 821   string aStr = aAnimSObject->GetID();
 822   myAnimEntry = aStr.c_str();
 823 }
 824
 825
 826 //========================================================================
 827 //========================================================================
 828 //========================================================================
 829 struct TNewAnimationEvent: public SALOME_Event
 830 {
 831   std::string myStudyName;
 832   VISU::View3D_ptr myView3D;
 833
 834   typedef VISU_TimeAnimation* TResult;
 835   TResult myResult;
 836
 837   TNewAnimationEvent (std::string theStudyName, VISU::View3D_ptr theView3D):
 838     myStudyName(theStudyName),
 839     myView3D(VISU::View3D::_duplicate(theView3D)),
 840     myResult(NULL)
 841   {}
 842
 843   virtual
 844   void
 845   Execute()
 846   {
 847     SUIT_Session* aSession = SUIT_Session::session();
 848     QPtrList<SUIT_Application> anApplications = aSession->applications();
 849     QPtrListIterator<SUIT_Application> anIter (anApplications);
 850     while (SUIT_Application* anApp = anIter.current()) {
 851       ++anIter;
 852       if (SUIT_Study* aSStudy = anApp->activeStudy()) {
 853         if (SalomeApp_Study* aStudy = dynamic_cast<SalomeApp_Study*>(aSStudy)) {
 854           if (_PTR(Study) aCStudy = aStudy->studyDS()) {
 855             if (myStudyName == aCStudy->Name()) {
 856               myResult = new VISU_TimeAnimation (aCStudy, myView3D);
 857               break;
 858             }
 859           }
 860         }
 861       }
 862     }
 863   }
 864 };
 865
 866 VISU_TimeAnimation_i::VISU_TimeAnimation_i (SALOMEDS::Study_ptr theStudy,
 867                                             VISU::View3D_ptr theView3D)
 868 {
 869   std::string aStudyName = theStudy->Name();
 870   myAnim = ProcessEvent(new TNewAnimationEvent (aStudyName, theView3D));
 871 }
 872
 873 VISU_TimeAnimation_i::~VISU_TimeAnimation_i()
 874 {
 875   delete myAnim;
 876 }
 877
 878 void VISU_TimeAnimation_i::addField (SALOMEDS::SObject_ptr theField)
 879 {
 880   myAnim->addField(theField);
 881 }
 882
 883 CORBA::Boolean VISU_TimeAnimation_i::generateFrames()
 884 {
 885   return ProcessEvent(new TMemFunEvent<VISU_TimeAnimation,bool>
 886                       (myAnim,&VISU_TimeAnimation::generateFrames));
 887 }
 888
 889 void VISU_TimeAnimation_i::generatePresentations (CORBA::Long theFieldNum)
 890 {
 891   myAnim->generatePresentations(theFieldNum);
 892 }
 893
 894 void VISU_TimeAnimation_i::clearView()
 895 {
 896   ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>
 897                    (myAnim,&VISU_TimeAnimation::clearView));
 898 }
 899
 900 void VISU_TimeAnimation_i::stopAnimation()
 901 {
 902   ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>
 903                    (myAnim,&VISU_TimeAnimation::stopAnimation));
 904 }
 905
 906 void VISU_TimeAnimation_i::startAnimation()
 907 {
 908   ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>
 909                    (myAnim,&VISU_TimeAnimation::startAnimation));
 910 }
 911
 912 void VISU_TimeAnimation_i::nextFrame()
 913 {
 914   ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>
 915                    (myAnim,&VISU_TimeAnimation::nextFrame));
 916 }
 917
 918 void VISU_TimeAnimation_i::prevFrame()
 919 {
 920   ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>
 921                    (myAnim,&VISU_TimeAnimation::prevFrame));
 922 }
 923
 924 void VISU_TimeAnimation_i::firstFrame()
 925 {
 926   ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>
 927                    (myAnim,&VISU_TimeAnimation::firstFrame));
 928 }
 929
 930 void VISU_TimeAnimation_i::lastFrame()
 931 {
 932   ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>
 933                    (myAnim,&VISU_TimeAnimation::lastFrame));
 934 }
 935
 936 void VISU_TimeAnimation_i::gotoFrame(CORBA::Long theFrame)
 937 {
 938   ProcessVoidEvent(new TVoidMemFun1ArgEvent<VISU_TimeAnimation,CORBA::Long>
 939                    (myAnim,&VISU_TimeAnimation::gotoFrame,theFrame));
 940 }
 941
 942 CORBA::Long VISU_TimeAnimation_i::getNbFields()
 943 {
 944   return myAnim->getNbFields();
 945 }
 946
 947 CORBA::Long VISU_TimeAnimation_i::getNbFrames()
 948 {
 949   return myAnim->getNbFrames();
 950 }
 951
 952 CORBA::Boolean VISU_TimeAnimation_i::isRunning()
 953 {
 954   return myAnim->isRunning();
 955 }
 956
 957 CORBA::Long VISU_TimeAnimation_i::getCurrentFrame()
 958 {
 959   return myAnim->getCurrentFrame();
 960 }
 961
 962 VISU::ScalarMap_ptr VISU_TimeAnimation_i::getPresentation
 963                     (CORBA::Long theField, CORBA::Long theFrame)
 964 {
 965   return myAnim->getPresentation(theField,theFrame);
 966 }
 967
 968 void VISU_TimeAnimation_i::setPresentationType (CORBA::Long theFieldNum,
 969                                                 VISU::VISUType theType)
 970 {
 971   myAnim->setPresentationType(theFieldNum,theType);
 972 }
 973
 974 VISU::VISUType VISU_TimeAnimation_i::getPresentationType (CORBA::Long theFieldNum)
 975 {
 976   return myAnim->getPresentationType(theFieldNum);
 977 }
 978
 979 void VISU_TimeAnimation_i::setSpeed(CORBA::Long theSpeed)
 980 {
 981   myAnim->setSpeed(theSpeed);
 982 }
 983
 984 CORBA::Long VISU_TimeAnimation_i::getSpeed()
 985 {
 986   return myAnim->getSpeed();
 987 }
 988
 989 CORBA::Boolean VISU_TimeAnimation_i::isProportional()
 990 {
 991   return myAnim->isProportional();
 992 }
 993
 994 void VISU_TimeAnimation_i::setAnimationRange (CORBA::Double theMin,
 995                                               CORBA::Double theMax)
 996 {
 997   myAnim->setAnimationRange(theMin,theMax);
 998 }
 999
1000 CORBA::Double VISU_TimeAnimation_i::getMinRange()
1001 {
1002   return myAnim->getMinRange();
1003 }
1004
1005 CORBA::Double VISU_TimeAnimation_i::getMaxRange()
1006 {
1007   return myAnim->getMaxRange();
1008 }
1009
1010 CORBA::Boolean VISU_TimeAnimation_i::isRangeDefined()
1011 {
1012   return myAnim->isRangeDefined();
1013 }
1014
1015 void VISU_TimeAnimation_i::dumpTo (const char* thePath)
1016 {
1017   myAnim->dumpTo(thePath);
1018 }
1019
1020 CORBA::Boolean VISU_TimeAnimation_i::isCycling()
1021 {
1022   return myAnim->isCycling();
1023 }
1024
1025 CORBA::Double VISU_TimeAnimation_i::getMinTime()
1026 {
1027   return myAnim->getMinTime();
1028 }
1029
1030 CORBA::Double VISU_TimeAnimation_i::getMaxTime()
1031 {
1032   return myAnim->getMaxTime();
1033 }
1034
1035 void VISU_TimeAnimation_i::setProportional (CORBA::Boolean theProp)
1036 {
1037   myAnim->setProportional(theProp);
1038 }
1039
1040 void VISU_TimeAnimation_i::setCycling (CORBA::Boolean theCycle)
1041 {
1042   myAnim->setCycling(theCycle);
1043 }
1044
1045 SALOMEDS::SObject_ptr VISU_TimeAnimation_i::publishInStudy()
1046 {
1047   return myAnim->publishInStudy();
1048 }
1049
1050 void VISU_TimeAnimation_i::restoreFromStudy(SALOMEDS::SObject_ptr theObj)
1051 {
1052   myAnim->restoreFromStudy(theObj);
1053 }
1054
1055 CORBA::Boolean VISU_TimeAnimation_i::isSavedInStudy()
1056 {
1057   return myAnim->isSavedInStudy();
1058 }
1059
1060 void VISU_TimeAnimation_i::saveAnimation()
1061 {
1062   myAnim->saveAnimation();
1063 }