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