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