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