1 // Copyright (C) 2003 CEA/DEN, EDF R&D
5 // File : VISU_TimeAnimation.cxx
6 // Author : Vitaly SMETANNIKOV
9 #include "VISU_TimeAnimation.h"
10 #include "VISU_Result_i.hh"
11 #include "VISU_Prs3d_i.hh"
12 #include "VISU_Mesh_i.hh"
13 #include "VISU_ScalarMap_i.hh"
14 #include "VISU_IsoSurfaces_i.hh"
15 #include "VISU_DeformedShape_i.hh"
16 #include "VISU_CutPlanes_i.hh"
17 #include "VISU_CutLines_i.hh"
18 #include "VISU_Vectors_i.hh"
19 #include "VISU_StreamLines_i.hh"
20 #include "VISU_ViewManager_i.hh"
21 #include "VISU_ScalarBarActor.hxx"
22 #include "VISU_Actor.h"
24 #include "SALOME_Event.hxx"
26 #include "VTKViewer_ViewFrame.h"
28 #include "QAD_Config.h"
32 //QWaitCondition myCondition;
33 static int MYDELAY = 1;
36 //************************************************************************
37 VISU_TimeAnimation::VISU_TimeAnimation(SALOMEDS::Study_ptr theStudy, VISU::View3D_ptr theView3D) {
38 myStudy = SALOMEDS::Study::_duplicate(theStudy);
42 myProportional = false;
45 if(!CORBA::is_nil(theView3D)){
46 VISU::View3D_i* pView = dynamic_cast<VISU::View3D_i*>(GetServant(theView3D).in());
47 QAD_StudyFrame* aStudyFrame = pView->GetStudyFrame();
48 myView = VISU::GetViewFrame(aStudyFrame);
60 //************************************************************************
61 VISU_TimeAnimation::~VISU_TimeAnimation() {
62 for (int i = 0; i < getNbFields(); i++) {
63 clearData(myFieldsLst[i]);
68 //************************************************************************
69 void VISU_TimeAnimation::addField(SALOMEDS::SObject_ptr theField) {
71 aNewData.myField = SALOMEDS::SObject::_duplicate(theField);
72 aNewData.myNbFrames = 0;
73 aNewData.myPrsType = VISU::TSCALARMAP;
74 VISU::Storable::TRestoringMap aMap = getMapOfValue(aNewData.myField);
75 aNewData.myNbTimes = VISU::Storable::FindValue(aMap,"myNbTimeStamps").toLong();
76 myFieldsLst.append(aNewData);
78 //find Min/Max timestamps
79 if ((myTimeMin == 0) && (myTimeMax == 0)) {
80 SALOMEDS::ChildIterator_var anIter = myStudy->NewChildIterator(theField);
81 SALOMEDS::SObject_var aTimeStamp;
82 anIter->Next(); // First is reference on support
83 myTimeMin = getTimeValue(anIter->Value());
84 for(; anIter->More(); anIter->Next()) {
85 myTimeMax = getTimeValue(anIter->Value());
91 //************************************************************************
92 void VISU_TimeAnimation::clearData(FieldData& theData) {
93 theData.myTiming.clear();
94 vtkRenderer* aRen = myView->getRenderer();
95 if (!theData.myActors.empty()) {
96 for (int i = 0, iEnd = theData.myActors.size(); i < iEnd; i++) {
97 if (theData.myActors.at(i) != 0) {
98 theData.myActors.at(i)->RemoveFromRender(aRen);
99 theData.myActors.at(i)->Delete();
102 theData.myActors.clear();
104 if (!theData.myPrs.empty()) {
105 for (int i = 0, iEnd = theData.myPrs.size(); i < iEnd; i++)
106 if (theData.myPrs.at(i) != 0) {
107 theData.myPrs.at(i)->_remove_ref();
109 theData.myPrs.clear();
111 theData.myNbFrames = 0;
116 //************************************************************************
117 void VISU_TimeAnimation::generatePresentations(CORBA::Long theFieldNum) {
118 FieldData& aData = myFieldsLst[theFieldNum];
120 // Delete previous presentations
123 VISU::Result_i* pResult = createPresent(aData.myField);
124 VISU::Storable::TRestoringMap aMap = getMapOfValue(aData.myField);
125 aData.myNbFrames = aData.myNbTimes;
126 //VISU::Storable::FindValue(aMap,"myNbTimeStamps").toLong();
128 aData.myPrs.resize(aData.myNbTimes,NULL);
129 aData.myTiming.resize(aData.myNbTimes);
131 SALOMEDS::ChildIterator_var anIter = myStudy->NewChildIterator(aData.myField);
132 SALOMEDS::SObject_var aTimeStamp;
133 anIter->Next(); // First is reference on support
135 double aMin = VTK_LARGE_FLOAT, aMax = -VTK_LARGE_FLOAT;
136 for(;anIter->More();anIter->Next()) {
137 if (i == aData.myNbTimes) {
138 MESSAGE("There are extra timestamps in field");
141 aTimeStamp = anIter->Value();
142 if (aTimeStamp->_is_nil()) continue;
144 aData.myTiming.at(i) = getTimeValue(aTimeStamp);
145 if (isRangeDefined()) {
146 if (aData.myTiming.at(i) < myMinVal) continue;
147 if (aData.myTiming.at(i) > myMaxVal) break;
150 VISU::Storable::TRestoringMap aTimeMap = getMapOfValue(aTimeStamp);
151 QString aMeshName = VISU::Storable::FindValue(aTimeMap,"myMeshName");
152 VISU::Entity anEntity = (VISU::Entity) VISU::Storable::FindValue(aTimeMap,"myEntityId").toInt();
153 QString aFieldName = VISU::Storable::FindValue(aTimeMap,"myFieldName");
154 int aTimeStampId = VISU::Storable::FindValue(aTimeMap,"myTimeStampId").toInt();
156 switch (aData.myPrsType) {
157 case VISU::TSCALARMAP: // ScalarMap
159 VISU::ScalarMap_i* aPresent = new VISU::ScalarMap_i(pResult, false);
160 aPresent->Create(aMeshName.latin1(), anEntity,
161 aFieldName.latin1(), aTimeStampId);
162 //VISU::ScalarMap_var aTmp = aPresent->_this();
163 //aPresent->_remove_ref();
164 aData.myPrs.at(i) = aPresent;
168 case VISU::TISOSURFACE: // Iso Surfaces
170 VISU::IsoSurfaces_i* aPresent = new VISU::IsoSurfaces_i(pResult, false);
171 aPresent->Create(aMeshName.latin1(), anEntity,
172 aFieldName.latin1(), aTimeStampId);
173 //VISU::IsoSurfaces_var aTmp = aPresent->_this();
174 //aPresent->_remove_ref();
175 aData.myPrs.at(i) = aPresent;
179 case VISU::TCUTPLANES: // Cut Planes
181 VISU::CutPlanes_i* aPresent = new VISU::CutPlanes_i(pResult, false);
182 aPresent->Create(aMeshName.latin1(), anEntity,
183 aFieldName.latin1(), aTimeStampId);
184 //VISU::CutPlanes_var aTmp = aPresent->_this();
185 //aPresent->_remove_ref();
186 aData.myPrs.at(i) = aPresent;
190 case VISU::TDEFORMEDSHAPE: // Deformed Shape
192 VISU::DeformedShape_i* aPresent = new VISU::DeformedShape_i(pResult, false);
193 aPresent->Create(aMeshName.latin1(), anEntity,
194 aFieldName.latin1(), aTimeStampId);
195 //VISU::DeformedShape_var aTmp = aPresent->_this();
196 //aPresent->_remove_ref();
197 aData.myPrs.at(i) = aPresent;
201 case VISU::TVECTORS: // Vectors
203 VISU::Vectors_i* aPresent = new VISU::Vectors_i(pResult, false);
204 aPresent->Create(aMeshName.latin1(), anEntity,
205 aFieldName.latin1(), aTimeStampId);
206 //VISU::Vectors_var aTmp = aPresent->_this();
207 //aPresent->_remove_ref();
208 aData.myPrs.at(i) = aPresent;
212 case VISU::TSTREAMLINES: // Stream Lines
214 VISU::StreamLines_i* aPresent = new VISU::StreamLines_i(pResult, false);
215 aPresent->Create(aMeshName.latin1(), anEntity,
216 aFieldName.latin1(), aTimeStampId);
217 //VISU::StreamLines_var aTmp = aPresent->_this();
218 //aPresent->_remove_ref();
219 aData.myPrs.at(i) = aPresent;
223 if (aData.myPrs.at(i)->GetMin() < aMin) aMin = aData.myPrs.at(i)->GetMin();
224 if (aData.myPrs.at(i)->GetMax() > aMax) aMax = aData.myPrs.at(i)->GetMax();
227 aData.myNbFrames = i;
228 QString aFixRange = QAD_CONFIG->getSetting("Visu:SBImposeRange");
229 if (aFixRange.compare("true") != 0) {
230 for (i = 0; i < aData.myNbFrames; i++)
231 aData.myPrs.at(i)->SetRange(aMin, aMax);
232 if(aData.myPrsType == VISU::TISOSURFACE)
233 for (i = 0; i < aData.myNbFrames; i++)
234 if(VISU::IsoSurfaces_i* aPrs = dynamic_cast<VISU::IsoSurfaces_i*>(aData.myPrs.at(i)))
235 aPrs->SetSubRange(aMin, aMax);
240 //************************************************************************
241 CORBA::Boolean VISU_TimeAnimation::generateFrames(){
243 MESSAGE("Viewer is nod defined for animation");
247 myLastError = QString("Frame(s) for ");
248 bool aNoError = true;
250 vtkRenderer* aRen = myView->getRenderer();
252 for (int i = 0; i < getNbFields(); i++) {
253 FieldData& aData = myFieldsLst[i];
254 aData.myActors.resize(aData.myNbFrames,NULL);
255 for (long j = 0; j < aData.myNbFrames; j++) {
256 VISU_Actor* aActor = NULL;
258 aActor = aData.myPrs.at(j)->CreateActor();
259 myView->AddActor(aActor);
261 aActor->VisibilityOn();
263 aActor->VisibilityOff();
264 }catch(...){ //catch(std::runtime_error& exc){
266 myLastError += QString("%1 ").arg(aData.myTiming.at(j));
268 aData.myActors.at(j) = aActor;
272 myLastError += QString(" timestamp(s) cannot be created.");
273 emit frameChanged(myFrame, myFieldsLst[0].myTiming.at(myFrame));
278 //************************************************************************
279 void VISU_TimeAnimation::clearView() {
281 MESSAGE("Viewer is nod defined for animation");
284 vtkRenderer* aRen = myView->getRenderer();
285 for (int i = 0; i < getNbFields(); i++) {
286 FieldData& aData = myFieldsLst[i];
287 if (!aData.myActors.empty()) {
288 for (int i = 0, iEnd = aData.myActors.size(); i < iEnd; i++) {
289 if (aData.myActors.at(i) != 0) {
290 aData.myActors.at(i)->RemoveFromRender(aRen);
291 aData.myActors.at(i)->Delete();
294 aData.myActors.clear();
297 myView->getRW()->getRenderWindow()->Render();
301 //************************************************************************
302 void VISU_TimeAnimation::stopAnimation() {
306 //************************************************************************
307 void VISU_TimeAnimation::startAnimation() {
314 //************************************************************************
315 void VISU_TimeAnimation::nextFrame() {
317 if (myFrame < (myFieldsLst[0].myNbFrames-1)) {
319 for (i = 0; i < getNbFields(); i++)
320 if (myFieldsLst[i].myActors.at(myFrame) != 0)
321 myFieldsLst[i].myActors.at(myFrame)->VisibilityOff();
323 for (i = 0; i < getNbFields(); i++)
324 if (myFieldsLst[i].myActors.at(myFrame) != 0)
325 myFieldsLst[i].myActors.at(myFrame)->VisibilityOn();
327 emit frameChanged(myFrame, myFieldsLst[0].myTiming.at(myFrame));
332 //************************************************************************
333 void VISU_TimeAnimation::prevFrame() {
337 for (i = 0; i < getNbFields(); i++)
338 if (myFieldsLst[i].myActors.at(myFrame) != 0)
339 myFieldsLst[i].myActors.at(myFrame)->VisibilityOff();
341 for (i = 0; i < getNbFields(); i++)
342 if (myFieldsLst[i].myActors.at(myFrame) != 0)
343 myFieldsLst[i].myActors.at(myFrame)->VisibilityOn();
345 emit frameChanged(myFrame, myFieldsLst[0].myTiming.at(myFrame));
350 //************************************************************************
351 void VISU_TimeAnimation::firstFrame() {
354 for (i = 0; i < getNbFields(); i++)
355 if(!myFieldsLst[i].myActors.empty())
356 if (myFieldsLst[i].myActors.at(myFrame) != 0)
357 myFieldsLst[i].myActors.at(myFrame)->VisibilityOff();
359 for (i = 0; i < getNbFields(); i++)
360 if(!myFieldsLst[i].myActors.empty())
361 if (myFieldsLst[i].myActors.at(myFrame) != 0)
362 myFieldsLst[i].myActors.at(myFrame)->VisibilityOn();
363 if(!myFieldsLst[0].myTiming.empty()){
364 emit frameChanged(myFrame, myFieldsLst[0].myTiming.at(myFrame));
369 //************************************************************************
370 void VISU_TimeAnimation::lastFrame() {
373 for (i = 0; i < getNbFields(); i++)
374 if (myFieldsLst[i].myActors.at(myFrame) != 0)
375 myFieldsLst[i].myActors.at(myFrame)->VisibilityOff();
376 myFrame = myFieldsLst[0].myNbFrames-1;
377 for (i = 0; i < getNbFields(); i++)
378 if (myFieldsLst[i].myActors.at(myFrame) != 0)
379 myFieldsLst[i].myActors.at(myFrame)->VisibilityOn();
381 emit frameChanged(myFrame, myFieldsLst[0].myTiming.at(myFrame));
386 //************************************************************************
387 // For Batchmode using
388 void VISU_TimeAnimation::gotoFrame(CORBA::Long theFrame) {
389 if ((theFrame < 0) || (theFrame > (getNbFrames()-1)))
395 for (i = 0; i < getNbFields(); i++)
396 if (myFieldsLst[i].myActors.at(myFrame) != 0)
397 myFieldsLst[i].myActors.at(myFrame)->VisibilityOff();
399 for (i = 0; i < getNbFields(); i++)
400 if (myFieldsLst[i].myActors.at(myFrame) != 0)
401 myFieldsLst[i].myActors.at(myFrame)->VisibilityOn();
402 emit frameChanged(myFrame, myFieldsLst[0].myTiming.at(myFrame));
405 qApp->processEvents(3);
410 //************************************************************************
411 VISU::ScalarMap_ptr VISU_TimeAnimation::getPresentation(CORBA::Long theField, CORBA::Long theFrame) {
412 if ((theField > getNbFields()) || (theField < 0))
413 return VISU::ScalarMap::_nil();
414 if ((theFrame < 0) || (theFrame > (myFieldsLst[theField].myNbFrames - 1)))
415 return VISU::ScalarMap::_nil();
416 return myFieldsLst[theField].myPrs.at(theFrame)->_this();
420 //************************************************************************
421 CORBA::Long VISU_TimeAnimation::getNbFrames() {
422 return (getNbFields() > 0)? myFieldsLst[0].myNbFrames : 0;
426 //************************************************************************
427 void VISU_TimeAnimation::run() {
429 MESSAGE("Viewer is nod defined for animation");
434 bool isDumping = !myDumpPath.isEmpty();
435 aOneVal = (myFieldsLst[0].myNbFrames > 2) ?
436 myFieldsLst[0].myTiming.at(1) - myFieldsLst[0].myTiming.at(0) : 1;
439 emit frameChanged(myFrame, myFieldsLst[0].myTiming.at(myFrame));
440 for (int i = 0; i < getNbFields(); i++) {
441 FieldData& aData = myFieldsLst[i];
443 if (aData.myActors.at(myFrame-1) != 0)
444 aData.myActors.at(myFrame-1)->VisibilityOff();
446 if (aData.myActors.at(aData.myNbFrames-1) != 0)
447 aData.myActors.at(aData.myNbFrames-1)->VisibilityOff();
449 if (aData.myActors.at(myFrame) != 0) {
450 aData.myActors.at(myFrame)->VisibilityOn();
453 myView->Repaint(false);
455 QPixmap px = QPixmap::grabWindow(myView->getViewWidget()->winId());
456 QString aFile(myDumpPath);
457 QString aName = QString("%1").arg(myFieldsLst[0].myTiming.at(myFrame));
459 while ((aPos = aName.find(".")) > -1 )
460 aName.replace(aPos, 1, "_");
463 px.save(aFile, "JPEG");
466 if (myProportional) {
472 if (myFieldsLst[0].myNbFrames > 2)
473 k = (myFieldsLst[0].myTiming.at(myFrame+1) - myFieldsLst[0].myTiming.at(myFrame))/aOneVal;
478 k = (myFrame < (myFieldsLst[0].myNbFrames-1))?
479 (myFieldsLst[0].myTiming.at(myFrame+1) - myFieldsLst[0].myTiming.at(myFrame))/aOneVal : 1;
484 msleep((int)(1000.*k/mySpeed));
491 if (!myIsActive) break;
494 if (myFrame == myFieldsLst[0].myNbFrames) {
508 //************************************************************************
509 VISU::Result_i* VISU_TimeAnimation::createPresent(SALOMEDS::SObject_var theField) {
510 SALOMEDS::SObject_var aSObj = theField->GetFather();
511 aSObj = aSObj->GetFather();
512 aSObj = aSObj->GetFather();
513 CORBA::Object_var anObject = VISU::SObjectToObject(aSObj);
514 if(CORBA::is_nil(anObject)) return NULL;
515 return dynamic_cast<VISU::Result_i*>(VISU::GetServant(anObject).in());
519 //************************************************************************
520 VISU::Storable::TRestoringMap VISU_TimeAnimation::getMapOfValue(SALOMEDS::SObject_var theSObject) {
521 VISU::Storable::TRestoringMap aMap;
522 if(!theSObject->_is_nil()){
523 SALOMEDS::GenericAttribute_var anAttr;
524 if (theSObject->FindAttribute(anAttr, "AttributeComment")) {
525 SALOMEDS::AttributeComment_var aComment = SALOMEDS::AttributeComment::_narrow(anAttr);
526 CORBA::String_var aString = aComment->Value();
527 QString strIn(aString.in());
528 VISU::Storable::StrToMap(strIn,aMap);
534 //************************************************************************
535 double VISU_TimeAnimation::getTimeValue(SALOMEDS::SObject_var theTimeStamp) {
536 SALOMEDS::GenericAttribute_var anAttr;
537 if(theTimeStamp->FindAttribute(anAttr, "AttributeName")) {
538 SALOMEDS::AttributeName_var aName = SALOMEDS::AttributeName::_narrow(anAttr);
539 QString aNameString(aName->Value());
540 return aNameString.toDouble();
544 //************************************************************************
545 void VISU_TimeAnimation::setSpeed(CORBA::Long theSpeed) {
546 mySpeed = (theSpeed<1)? 1 : theSpeed;
550 //========================================================================
551 VISU_TimeAnimation_i::VISU_TimeAnimation_i(SALOMEDS::Study_ptr theStudy, VISU::View3D_ptr theView3D){
552 myAnim = new VISU_TimeAnimation(theStudy,theView3D);
556 VISU_TimeAnimation_i::~VISU_TimeAnimation_i(){
561 void VISU_TimeAnimation_i::addField(SALOMEDS::SObject_ptr theField){
562 myAnim->addField(theField);
566 CORBA::Boolean VISU_TimeAnimation_i::generateFrames(){
567 return ProcessEvent(new TMemFunEvent<VISU_TimeAnimation,bool>(myAnim,&VISU_TimeAnimation::generateFrames));
571 void VISU_TimeAnimation_i::generatePresentations(CORBA::Long theFieldNum){
572 myAnim->generatePresentations(theFieldNum);
576 void VISU_TimeAnimation_i::clearView(){
577 ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>(myAnim,&VISU_TimeAnimation::clearView));
581 void VISU_TimeAnimation_i::stopAnimation(){
582 ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>(myAnim,&VISU_TimeAnimation::stopAnimation));
586 void VISU_TimeAnimation_i::startAnimation(){
587 ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>(myAnim,&VISU_TimeAnimation::startAnimation));
591 void VISU_TimeAnimation_i::nextFrame(){
592 ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>(myAnim,&VISU_TimeAnimation::nextFrame));
596 void VISU_TimeAnimation_i::prevFrame(){
597 ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>(myAnim,&VISU_TimeAnimation::prevFrame));
601 void VISU_TimeAnimation_i::firstFrame(){
602 ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>(myAnim,&VISU_TimeAnimation::firstFrame));
606 void VISU_TimeAnimation_i::lastFrame(){
607 ProcessVoidEvent(new TVoidMemFunEvent<VISU_TimeAnimation>(myAnim,&VISU_TimeAnimation::lastFrame));
611 void VISU_TimeAnimation_i::gotoFrame(CORBA::Long theFrame){
612 ProcessVoidEvent(new TVoidMemFun1ArgEvent<VISU_TimeAnimation,CORBA::Long>(myAnim,&VISU_TimeAnimation::gotoFrame,theFrame));
616 CORBA::Long VISU_TimeAnimation_i::getNbFields(){
617 return myAnim->getNbFields();
621 CORBA::Long VISU_TimeAnimation_i::getNbFrames(){
622 return myAnim->getNbFrames();
626 CORBA::Boolean VISU_TimeAnimation_i::isRunning(){
627 return myAnim->isRunning();
631 CORBA::Long VISU_TimeAnimation_i::getCurrentFrame(){
632 return myAnim->getCurrentFrame();
636 VISU::ScalarMap_ptr VISU_TimeAnimation_i::getPresentation(CORBA::Long theField, CORBA::Long theFrame){
637 return myAnim->getPresentation(theField,theFrame);
641 void VISU_TimeAnimation_i::setPresentationType(CORBA::Long theFieldNum, VISU::VISUType theType){
642 myAnim->setPresentationType(theFieldNum,theType);
646 VISU::VISUType VISU_TimeAnimation_i::getPresentationType(CORBA::Long theFieldNum){
647 return myAnim->getPresentationType(theFieldNum);
651 void VISU_TimeAnimation_i::setSpeed(CORBA::Long theSpeed){
652 myAnim->setSpeed(theSpeed);
656 CORBA::Long VISU_TimeAnimation_i::getSpeed(){
657 return myAnim->getSpeed();
661 CORBA::Boolean VISU_TimeAnimation_i::isProportional(){
662 return myAnim->isProportional();
666 void VISU_TimeAnimation_i::setAnimationRange(CORBA::Double theMin, CORBA::Double theMax){
667 myAnim->setAnimationRange(theMin,theMax);
671 CORBA::Double VISU_TimeAnimation_i::getMinRange(){
672 return myAnim->getMinRange();
676 CORBA::Double VISU_TimeAnimation_i::getMaxRange(){
677 return myAnim->getMaxRange();
681 CORBA::Boolean VISU_TimeAnimation_i::isRangeDefined(){
682 return myAnim->isRangeDefined();
686 void VISU_TimeAnimation_i::dumpTo(const char* thePath){
687 myAnim->dumpTo(thePath);
691 CORBA::Boolean VISU_TimeAnimation_i::isCycling(){
692 return myAnim->isCycling();
696 CORBA::Double VISU_TimeAnimation_i::getMinTime(){
697 return myAnim->getMinTime();
701 CORBA::Double VISU_TimeAnimation_i::getMaxTime(){
702 return myAnim->getMaxTime();
706 void VISU_TimeAnimation_i::setProportional(CORBA::Boolean theProp){
707 myAnim->setProportional(theProp);
711 void VISU_TimeAnimation_i::setCycling(CORBA::Boolean theCycle){
712 myAnim->setCycling(theCycle);