/*========================================================================= Program: ParaView Module: pqObjectBuilder.cxx Copyright (c) 2005-2008 Sandia Corporation, Kitware Inc. All rights reserved. ParaView is a free software; you can redistribute it and/or modify it under the terms of the ParaView license version 1.2. See License_v1.2.txt for the full ParaView license. A copy of this license can be obtained by contacting Kitware Inc. 28 Corporate Drive Clifton Park, NY 12065 USA THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE AUTHORS OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. ========================================================================*/ #include "pqObjectBuilder.h" #include "vtkDebugLeaksManager.h" #include "vtkNew.h" #include "vtkProcessModule.h" #include "vtkPVXMLElement.h" #include "vtkSmartPointer.h" #include "vtkSMCoreUtilities.h" #include "vtkSMDomain.h" #include "vtkSMDomainIterator.h" #include "vtkSMInputProperty.h" #include "vtkSMParaViewPipelineController.h" #include "vtkSMPropertyHelper.h" #include "vtkSMPropertyIterator.h" #include "vtkSMRenderViewProxy.h" #include "vtkSMRepresentationProxy.h" #include "vtkSMSession.h" #include "vtkSMSessionProxyManager.h" #include "vtkSMSettings.h" #include "vtkSMSourceProxy.h" #include "vtkSMStringVectorProperty.h" #include "vtkSMTransferFunctionManager.h" #include #include #include #include "pqAnimationCue.h" #include "pqApplicationCore.h" #include "pqDataRepresentation.h" #include "pqInterfaceTracker.h" #include "pqOutputPort.h" #include "pqPipelineFilter.h" #include "pqPipelineSource.h" #include "pqProxyModifiedStateUndoElement.h" #include "pqRenderView.h" #include "pqSMAdaptor.h" #include "pqScalarBarRepresentation.h" #include "pqScalarsToColors.h" #include "pqServer.h" #include "pqServerManagerModel.h" #include "pqUndoStack.h" #include "pqView.h" #include "vtkSMAnimationSceneProxy.h" namespace pqObjectBuilderNS { vtkNew Controller; static bool ContinueWaiting = true; static bool processEvents() { QApplication::processEvents(); return ContinueWaiting; } //----------------------------------------------------------------------------- bool preCreatePipelineProxy(vtkSMProxy* proxy) { return Controller->PreInitializeProxy(proxy); } //----------------------------------------------------------------------------- pqPipelineSource* postCreatePipelineProxy( vtkSMProxy* proxy, pqServer* server, const QString& regName=QString()) { // since there are no properties to set, nothing to do here. Controller->PostInitializeProxy(proxy); if (regName.isEmpty()) { Controller->RegisterPipelineProxy(proxy); } else { Controller->RegisterPipelineProxy(proxy, regName.toLatin1().data()); } pqPipelineSource* source = pqApplicationCore::instance()-> getServerManagerModel()->findItem(proxy); source->setModifiedState(pqProxy::UNINITIALIZED); // Manage Modified state in Undo/Redo only if not a collaborative server if (!server->session()->IsMultiClients()) { pqProxyModifiedStateUndoElement* elem = pqProxyModifiedStateUndoElement::New(); elem->SetSession(server->session()); elem->MadeUninitialized(source); ADD_UNDO_ELEM(elem); elem->Delete(); } // Update animation scene. vtkSMAnimationSceneProxy::UpdateAnimationUsingDataTimeSteps( Controller->GetAnimationScene(server->session())); return source; } } //----------------------------------------------------------------------------- pqObjectBuilder::pqObjectBuilder(QObject* _parent/*=0*/) : QObject(_parent), WaitingForConnection(false) { } //----------------------------------------------------------------------------- pqObjectBuilder::~pqObjectBuilder() { } //----------------------------------------------------------------------------- pqPipelineSource* pqObjectBuilder::createSource(const QString& sm_group, const QString& sm_name, pqServer* server) { vtkSMSessionProxyManager* pxm = server->proxyManager(); vtkSmartPointer proxy; proxy.TakeReference( pxm->NewProxy(sm_group.toLatin1().data(), sm_name.toLatin1().data())); if (!pqObjectBuilderNS::preCreatePipelineProxy(proxy)) { return NULL; } pqPipelineSource* source = pqObjectBuilderNS::postCreatePipelineProxy(proxy, server); emit this->sourceCreated(source); emit this->proxyCreated(source); return source; } //----------------------------------------------------------------------------- pqPipelineSource* pqObjectBuilder::createFilter( const QString& sm_group, const QString& sm_name, QMap > namedInputs, pqServer* server) { vtkSMSessionProxyManager* pxm = server->proxyManager(); vtkSmartPointer proxy; proxy.TakeReference( pxm->NewProxy(sm_group.toLatin1().data(), sm_name.toLatin1().data())); if (!pqObjectBuilderNS::preCreatePipelineProxy(proxy)) { return NULL; } // Now for every input port, connect the inputs. QMap >::iterator mapIter; for (mapIter = namedInputs.begin(); mapIter != namedInputs.end(); ++mapIter) { QString input_port_name = mapIter.key(); QList &inputs = mapIter.value(); vtkSMProperty* prop = proxy->GetProperty(input_port_name.toLatin1().data()); if (!prop) { qCritical() << "Failed to locate input property "<< input_port_name; continue; } vtkSMPropertyHelper helper(prop); foreach (pqOutputPort* opPort, inputs) { helper.Add(opPort->getSource()->getProxy(), opPort->getPortNumber()); } } pqPipelineSource* filter = pqObjectBuilderNS::postCreatePipelineProxy(proxy, server); emit this->filterCreated(filter); emit this->proxyCreated(filter); return filter; } //----------------------------------------------------------------------------- pqPipelineSource* pqObjectBuilder::createFilter( const QString& group, const QString& name, pqPipelineSource* input, int output_port) { QMap > namedInputs; QList inputs; inputs.push_back(input->getOutputPort(output_port)); namedInputs["Input"] = inputs; return this->createFilter(group, name, namedInputs, input->getServer()); } //----------------------------------------------------------------------------- inline QString pqObjectBuilderGetPath(const QString& filename, bool use_dir) { if (use_dir) { return QFileInfo(filename).path(); } return filename; } //----------------------------------------------------------------------------- pqPipelineSource* pqObjectBuilder::createReader(const QString& sm_group, const QString& sm_name, const QStringList& files, pqServer* server) { if (files.empty()) { return 0; } unsigned int numFiles = files.size(); QString reg_name = QFileInfo(files[0]).fileName(); if (numFiles > 1) { // Find the largest prefix that matches all filenames, and then append a '*' // to signify that it is a collection of files. If they all start with // something different, just give up and add the '*' anyway. for (unsigned int i = 1; i < numFiles; i++) { QString nextFile = QFileInfo(files[i]).fileName(); if (nextFile.startsWith(reg_name)) continue; QString commonPrefix = reg_name; do { commonPrefix.chop(1); } while (!nextFile.startsWith(commonPrefix) && !commonPrefix.isEmpty()); if (commonPrefix.isEmpty()) break; reg_name = commonPrefix; } reg_name += '*'; } vtkSMSessionProxyManager* pxm = server->proxyManager(); vtkSmartPointer proxy; proxy.TakeReference( pxm->NewProxy(sm_group.toLatin1().data(), sm_name.toLatin1().data())); if (!pqObjectBuilderNS::preCreatePipelineProxy(proxy)) { return NULL; } QString pname = this->getFileNamePropertyName(proxy); if (!pname.isEmpty()) { vtkSMStringVectorProperty* prop = vtkSMStringVectorProperty::SafeDownCast( proxy->GetProperty(pname.toLatin1().data())); if (!prop) { return 0; } // If there's a hint on the property indicating that this property expects a // directory name, then, we will set the directory name on it. bool use_dir = false; if (prop->GetHints() && prop->GetHints()->FindNestedElementByName("UseDirectoryName")) { use_dir = true; } if (numFiles == 1 || !prop->GetRepeatCommand()) { pqSMAdaptor::setElementProperty(prop, pqObjectBuilderGetPath(files[0], use_dir)); } else { QList values; foreach (QString file, files) { values.push_back(pqObjectBuilderGetPath(file, use_dir)); } pqSMAdaptor::setMultipleElementProperty(prop, values); } proxy->UpdateVTKObjects(); } pqPipelineSource* reader = pqObjectBuilderNS::postCreatePipelineProxy(proxy, server, reg_name); emit this->readerCreated(reader, files[0]); emit this->readerCreated(reader, files); emit this->sourceCreated(reader); emit this->proxyCreated(reader); return reader; } //----------------------------------------------------------------------------- void pqObjectBuilder::destroy(pqPipelineSource* source) { if (!source) { qDebug() << "Cannot remove null source."; return; } if (source->getAllConsumers().size() > 0) { qDebug() << "Cannot remove source with consumers."; return; } emit this->destroying(source); pqObjectBuilderNS::Controller->UnRegisterProxy(source->getProxy()); } //----------------------------------------------------------------------------- pqView* pqObjectBuilder::createView(const QString& type, pqServer* server) { if (!server) { qDebug() << "Cannot create view without server."; return NULL; } vtkSMSessionProxyManager* pxm = server->proxyManager(); vtkSmartPointer proxy; proxy.TakeReference(pxm->NewProxy("views", type.toLatin1().data())); if (!proxy) { qDebug() << "Failed to create a proxy for the requested view type:" << type; return NULL; } // notify the world that we may create a new view. applications may handle // this by setting up layouts, etc. emit this->aboutToCreateView(server); pqObjectBuilderNS::Controller->PreInitializeProxy(proxy); pqObjectBuilderNS::Controller->PostInitializeProxy(proxy); pqObjectBuilderNS::Controller->RegisterViewProxy(proxy); pqServerManagerModel* model = pqApplicationCore::instance()->getServerManagerModel(); pqView* view = model->findItem(proxy); if (view) { emit this->viewCreated(view); emit this->proxyCreated(view); } else { qDebug() << "Cannot locate the pqView for the view proxy of type" << type; } return view; } //----------------------------------------------------------------------------- void pqObjectBuilder::destroy(pqView* view) { if (!view) { return; } emit this->destroying(view); pqObjectBuilderNS::Controller->UnRegisterProxy(view->getProxy()); } //----------------------------------------------------------------------------- pqDataRepresentation* pqObjectBuilder::createDataRepresentation( pqOutputPort* opPort, pqView* view, const QString &representationType) { if (!opPort || !view) { qCritical() <<"Missing required attribute."; return NULL; } if(!view->canDisplay(opPort)) { // View cannot display this source, nothing to do here. return NULL; } vtkSmartPointer reprProxy; pqPipelineSource* source = opPort->getSource(); vtkSMSessionProxyManager* pxm = source->proxyManager(); // HACK to create correct representation for text sources/filters. QString srcProxyName = source->getProxy()->GetXMLName(); if (representationType != "") { reprProxy.TakeReference(pxm->NewProxy( "representations", representationType.toLatin1().data())); } else { reprProxy.TakeReference(view->getViewProxy()->CreateDefaultRepresentation( source->getProxy(), opPort->getPortNumber())); } // Could not determine representation proxy to create. if (!reprProxy) { return NULL; } pqObjectBuilderNS::Controller->PreInitializeProxy(reprProxy); // Set the reprProxy's input. pqSMAdaptor::setInputProperty(reprProxy->GetProperty("Input"), source->getProxy(), opPort->getPortNumber()); pqObjectBuilderNS::Controller->PostInitializeProxy(reprProxy); pqObjectBuilderNS::Controller->RegisterRepresentationProxy(reprProxy); // Add the reprProxy to render module. vtkSMProxy* viewModuleProxy = view->getProxy(); vtkSMPropertyHelper(viewModuleProxy, "Representations").Add(reprProxy); viewModuleProxy->UpdateVTKObjects(); pqApplicationCore* core= pqApplicationCore::instance(); pqDataRepresentation* repr = core->getServerManagerModel()-> findItem(reprProxy); if (repr ) { emit this->dataRepresentationCreated(repr); emit this->proxyCreated(repr); } return repr; } //----------------------------------------------------------------------------- vtkSMProxy* pqObjectBuilder::createProxy(const QString& sm_group, const QString& sm_name, pqServer* server, const QString& reg_group) { if (!server) { qDebug() << "server cannot be null"; return 0; } if (sm_group.isEmpty() || sm_name.isEmpty()) { qCritical() << "Group name and proxy name must be non empty."; return 0; } vtkSMSessionProxyManager* pxm = server->proxyManager(); vtkSmartPointer proxy; proxy.TakeReference( pxm->NewProxy(sm_group.toLatin1().data(), sm_name.toLatin1().data())); if (!proxy.GetPointer()) { qCritical() << "Failed to create proxy: " << sm_group << ", " << sm_name; return NULL; } else if (reg_group.contains("prototypes")) { // Mark as prototype to prevent them from behing saved in undo stack and // managed through the state proxy->SetPrototype(true); } pxm->RegisterProxy(reg_group.toLatin1().data(), proxy); return proxy; } //----------------------------------------------------------------------------- void pqObjectBuilder::destroy(pqRepresentation* repr) { if (!repr) { return; } emit this->destroying(repr); // Remove repr from the view module. pqView* view = repr->getView(); if (view) { vtkSMProxyProperty* pp = vtkSMProxyProperty::SafeDownCast( view->getProxy()->GetProperty("Representations")); pp->RemoveProxy(repr->getProxy()); view->getProxy()->UpdateVTKObjects(); } // If this repr has a lookuptable, we hide all scalar bars for that // lookup table unless there is some other repr that's using it. pqScalarsToColors* stc =0; if (pqDataRepresentation* dataRepr = qobject_cast(repr)) { stc = dataRepr->getLookupTable(); } this->destroyProxyInternal(repr); if (stc) { // this hides scalar bars only if the LUT is not used by // any other repr. This must happen after the repr has // been deleted. stc->hideUnusedScalarBars(); } } //----------------------------------------------------------------------------- void pqObjectBuilder::destroy(pqProxy* proxy) { emit this->destroying(proxy); this->destroyProxyInternal(proxy); } //----------------------------------------------------------------------------- void pqObjectBuilder::destroySources(pqServer* server) { pqServerManagerModel* model = pqApplicationCore::instance()->getServerManagerModel(); pqObjectBuilder* builder = pqApplicationCore::instance()->getObjectBuilder(); QList sources = model->findItems(server); while(!sources.isEmpty()) { for(int i=0; igetAllConsumers().size() == 0) { builder->destroy(sources[i]); sources[i] = NULL; } } sources.removeAll(NULL); } } //----------------------------------------------------------------------------- void pqObjectBuilder::destroyLookupTables(pqServer* server) { pqServerManagerModel* model = pqApplicationCore::instance()->getServerManagerModel(); pqObjectBuilder* builder = pqApplicationCore::instance()->getObjectBuilder(); QList luts = model->findItems(server); foreach (pqScalarsToColors* lut, luts) { builder->destroy(lut); } QList scalarbars = model->findItems(server); foreach (pqScalarBarRepresentation* sb, scalarbars) { builder->destroy(sb); } } //----------------------------------------------------------------------------- void pqObjectBuilder::destroyPipelineProxies(pqServer* server) { this->destroySources(server); this->destroyLookupTables(server); } //----------------------------------------------------------------------------- void pqObjectBuilder::destroyAllProxies(pqServer* server) { if (!server) { qDebug() << "Server cannot be NULL."; return; } server->proxyManager()->UnRegisterProxies(); } //----------------------------------------------------------------------------- void pqObjectBuilder::destroyProxyInternal(pqProxy* proxy) { if (proxy) { vtkSMSessionProxyManager* pxm = proxy->proxyManager(); pxm->UnRegisterProxy(proxy->getSMGroup().toLatin1().data(), proxy->getSMName().toLatin1().data(), proxy->getProxy()); } } //----------------------------------------------------------------------------- QString pqObjectBuilder::getFileNamePropertyName(vtkSMProxy* proxy) { const char* fname = vtkSMCoreUtilities::GetFileNameProperty(proxy); return fname? QString(fname): QString::Null(); } //----------------------------------------------------------------------------- void pqObjectBuilder::abortPendingConnections() { pqObjectBuilderNS::ContinueWaiting = false; } //----------------------------------------------------------------------------- pqServer* pqObjectBuilder::createServer(const pqServerResource& resource) { if (this->WaitingForConnection) { qCritical() << "createServer called while waiting for previous connection " "to be established."; return NULL; } // Create a modified version of the resource that only contains server information const pqServerResource server_resource = resource.schemeHostsPorts(); pqServerManagerModel *smModel = pqApplicationCore::instance()->getServerManagerModel(); if (!vtkProcessModule::GetProcessModule()->GetMultipleSessionsSupport()) { // If multiple connections are not supported, then we only connect to the // new server if no already connected and ensure that any previously // connected servers are disconnected. // determine if we're already connected to this server. pqServer* server = smModel->findServer(server_resource); if (server) { return server; } if (smModel->getNumberOfItems() > 0) { this->removeServer(smModel->getItemAtIndex(0)); } } this->WaitingForConnection = true; // Let the pqServerManagerModel know the resource to use for the connection // to be created. smModel->setActiveResource(resource); // Based on the server resource, create the correct type of server ... vtkIdType id = 0; if (server_resource.scheme() == "builtin") { id = vtkSMSession::ConnectToSelf(); } else if(server_resource.scheme() == "cs") { id = vtkSMSession::ConnectToRemote( resource.host().toLatin1().data(), resource.port(11111)); } else if(server_resource.scheme() == "csrc") { pqObjectBuilderNS::ContinueWaiting = true; id = vtkSMSession::ReverseConnectToRemote(server_resource.port(11111), &pqObjectBuilderNS::processEvents); } else if(server_resource.scheme() == "cdsrs") { id = vtkSMSession::ConnectToRemote( server_resource.dataServerHost().toLatin1().data(), server_resource.dataServerPort(11111), server_resource.renderServerHost().toLatin1().data(), server_resource.renderServerPort(22221)); } else if(server_resource.scheme() == "cdsrsrc") { pqObjectBuilderNS::ContinueWaiting = true; id = vtkSMSession::ReverseConnectToRemote( server_resource.dataServerPort(11111), server_resource.renderServerPort(22221), &pqObjectBuilderNS::processEvents); } else if(server_resource.scheme() == "catalyst") { id = vtkSMSession::ConnectToCatalyst(); } else { qCritical() << "Unknown server type: " << server_resource.scheme() << "\n"; } pqServer* server = NULL; if (id != 0) { server = smModel->findServer(id); emit this->finishedAddingServer(server); } this->WaitingForConnection = false; return server; } //----------------------------------------------------------------------------- void pqObjectBuilder::removeServer(pqServer* server) { if (!server) { qDebug() << "No server to remove."; return; } pqApplicationCore* core = pqApplicationCore::instance(); pqServerManagerModel* sModel = core->getServerManagerModel(); sModel->beginRemoveServer(server); vtkSMSession::Disconnect(server->sessionId()); sModel->endRemoveServer(); } //----------------------------------------------------------------------------- pqServer* pqObjectBuilder::resetServer(pqServer* server) { Q_ASSERT(server); pqServerResource resource = server->getResource(); vtkSmartPointer session = server->session(); vtkProcessModule* pm = vtkProcessModule::GetProcessModule(); // simulate disconnect. pqApplicationCore* core = pqApplicationCore::instance(); pqServerManagerModel* smModel = core->getServerManagerModel(); smModel->beginRemoveServer(server); // Unregister all proxies. this->destroyAllProxies(server); // Unregister session. pm->UnRegisterSession(session); smModel->endRemoveServer(); server = NULL; // Let the pqServerManagerModel know the resource to use for the connection // to be created. smModel->setActiveResource(resource); // re-register session. vtkIdType id = pm->RegisterSession(session); pqServer* newServer = NULL; if (id != 0) { newServer = smModel->findServer(id); emit this->finishedAddingServer(newServer); } return newServer; } //----------------------------------------------------------------------------- void pqObjectBuilder::destroy(pqAnimationCue* cue) { if (!cue) { return; } vtkSMSessionProxyManager* pxm = cue->proxyManager(); QList keyframes = cue->getKeyFrames(); // unregister all the keyframes. foreach (vtkSMProxy* kf, keyframes) { pxm->UnRegisterProxy("animation", pxm->GetProxyName("animation", kf), kf); } this->destroy(static_cast(cue)); }