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
|
/*=========================================================================
Program: Visualization Toolkit
Module: TestStructuredGridPartitioner.cxx
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
All rights reserved.
See Copyright.txt or http://www.kitware.com/Copyright.htm 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.
=========================================================================*/
// .NAME TestStructuredGridPartitioner.cxx -- Simple test for partitioning
// structured grids.
//
// .SECTION Description
// Simple test for structured grid partitioner
#include <iostream>
#include <sstream>
#include <cassert>
#include "vtkStructuredGridPartitioner.h"
#include "vtkStructuredGrid.h"
#include "vtkMultiBlockDataSet.h"
#include "vtkXMLStructuredGridReader.h"
#include "vtkXMLMultiBlockDataWriter.h"
//------------------------------------------------------------------------------
// Description:
// Writes multi-block dataset to grid
void WriteMultiBlock(const std::string &file, vtkMultiBlockDataSet *mbds)
{
assert("pre: NULL multi-block dataset!" && (mbds != NULL) );
std::ostringstream oss;
vtkXMLMultiBlockDataWriter *writer = vtkXMLMultiBlockDataWriter::New();
oss << file << "." << writer->GetDefaultFileExtension();
writer->SetFileName( oss.str().c_str() );
writer->SetInputData( mbds );
writer->Update();
writer->Delete();
}
//------------------------------------------------------------------------------
// Description:
// Get grid from file
vtkStructuredGrid* GetGridFromFile( std::string &file )
{
vtkXMLStructuredGridReader *reader = vtkXMLStructuredGridReader::New();
reader->SetFileName( file.c_str() );
reader->Update();
vtkStructuredGrid* myGrid = vtkStructuredGrid::New();
myGrid->DeepCopy(reader->GetOutput());
reader->Delete();
return( myGrid );
}
//------------------------------------------------------------------------------
// Description:
// Program Main
int TestStructuredGridPartitioner( int argc, char *argv[] )
{
if( argc != 3)
{
std::cout << "Usage: ./TestStructuredGridPartitioner <vtsfile> <N>\n";
std::cout.flush();
return -1;
}
std::string fileName = std::string( argv[1] );
int NumPartitions = atoi( argv[2] );
vtkStructuredGrid *grid = GetGridFromFile( fileName );
assert("pre: grid is not NULL" && (grid != NULL) );
vtkStructuredGridPartitioner *gridPartitioner =
vtkStructuredGridPartitioner::New();
gridPartitioner->SetInputData(grid);
gridPartitioner->SetNumberOfPartitions( NumPartitions );
gridPartitioner->Update();
vtkMultiBlockDataSet *mbds = gridPartitioner->GetOutput();
WriteMultiBlock( "PartitionedGrid", mbds );
grid->Delete();
gridPartitioner->Delete();
return 0;
}
|
