1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129
|
/*=========================================================================
Program: ParaView
Module: vtkSMViewExportHelper.cxx
Copyright (c) Kitware, Inc.
All rights reserved.
See Copyright.txt or http://www.paraview.org/HTML/Copyright.html for details.
This software is distributed WITHOUT ANY WARRANTY; without even
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
PURPOSE. See the above copyright notice for more information.
=========================================================================*/
#include "vtkSMViewExportHelper.h"
#include "vtkNew.h"
#include "vtkObjectFactory.h"
#include "vtkPVProxyDefinitionIterator.h"
#include "vtkSMDocumentation.h"
#include "vtkSMExporterProxy.h"
#include "vtkSMParaViewPipelineController.h"
#include "vtkSMPropertyHelper.h"
#include "vtkSMProxyDefinitionManager.h"
#include "vtkSMSessionProxyManager.h"
#include "vtkSMViewProxy.h"
#include "vtkSmartPointer.h"
#include <sstream>
#include <vtksys/RegularExpression.hxx>
#include <vtksys/SystemTools.hxx>
vtkObjectFactoryNewMacro(vtkSMViewExportHelper);
//----------------------------------------------------------------------------
vtkSMViewExportHelper::vtkSMViewExportHelper()
{
}
//----------------------------------------------------------------------------
vtkSMViewExportHelper::~vtkSMViewExportHelper()
{
}
//----------------------------------------------------------------------------
vtkStdString vtkSMViewExportHelper::GetSupportedFileTypes(vtkSMViewProxy* view)
{
if (!view)
{
return vtkStdString();
}
std::ostringstream stream;
int count = 0;
vtkSMSessionProxyManager* pxm = view->GetSessionProxyManager();
vtkSmartPointer<vtkPVProxyDefinitionIterator> iter;
iter.TakeReference(pxm->GetProxyDefinitionManager()->NewSingleGroupIterator("exporters"));
for (iter->InitTraversal(); !iter->IsDoneWithTraversal(); iter->GoToNextItem())
{
vtkSMExporterProxy* prototype =
vtkSMExporterProxy::SafeDownCast(pxm->GetPrototypeProxy("exporters", iter->GetProxyName()));
if (prototype && prototype->CanExport(view) && prototype->GetFileExtension() != NULL)
{
vtkSMDocumentation* doc = prototype->GetDocumentation();
std::ostringstream helpstream;
if (doc && doc->GetShortHelp())
{
helpstream << doc->GetShortHelp();
}
else
{
helpstream << vtksys::SystemTools::UpperCase(prototype->GetFileExtension()).c_str()
<< " Files";
}
stream << (count > 0 ? ";;" : "") << helpstream.str().c_str() << " (*."
<< prototype->GetFileExtension() << ")";
count++;
}
}
return stream.str();
}
//----------------------------------------------------------------------------
vtkSMExporterProxy* vtkSMViewExportHelper::CreateExporter(
const char* filename, vtkSMViewProxy* view)
{
if (!view || filename == NULL || filename[0] == '\0')
{
vtkErrorMacro("Invalid input arguments to Export.");
return NULL;
}
vtkSMSessionProxyManager* pxm = view->GetSessionProxyManager();
vtkSmartPointer<vtkPVProxyDefinitionIterator> iter;
iter.TakeReference(pxm->GetProxyDefinitionManager()->NewSingleGroupIterator("exporters"));
for (iter->InitTraversal(); !iter->IsDoneWithTraversal(); iter->GoToNextItem())
{
vtkSMExporterProxy* prototype =
vtkSMExporterProxy::SafeDownCast(pxm->GetPrototypeProxy("exporters", iter->GetProxyName()));
if (prototype && prototype->CanExport(view) && prototype->GetFileExtension() != NULL)
{
std::ostringstream reStream;
reStream << "^" // start
<< ".*" // leading text
<< "\\." // extension separator
<< prototype->GetFileExtension() << "$"; // end
vtksys::RegularExpression re(reStream.str().c_str());
if (re.find(filename))
{
vtkSMExporterProxy* exporter =
vtkSMExporterProxy::SafeDownCast(pxm->NewProxy("exporters", iter->GetProxyName()));
vtkNew<vtkSMParaViewPipelineController> controller;
controller->PreInitializeProxy(exporter);
exporter->SetView(view);
vtkSMPropertyHelper(exporter, "FileName").Set(filename);
controller->PostInitializeProxy(exporter);
exporter->UpdateVTKObjects();
return exporter;
}
}
}
return NULL;
}
//----------------------------------------------------------------------------
void vtkSMViewExportHelper::PrintSelf(ostream& os, vtkIndent indent)
{
this->Superclass::PrintSelf(os, indent);
}
|