(dataType, reader, writer, ext, numTris, useSubdir=False)
| 150 | return pf |
| 151 | |
| 152 | def TestDataType(dataType, reader, writer, ext, numTris, useSubdir=False): |
| 153 | s = GetSource(dataType) |
| 154 | |
| 155 | filename = VTK_TEMP_DIR + "/%s.p%s" % (dataType, ext) |
| 156 | |
| 157 | writer.SetInputConnection(s.GetOutputPort()) |
| 158 | npieces = 16 |
| 159 | writer.SetNumberOfPieces(npieces) |
| 160 | pperrank = npieces // nranks |
| 161 | start = pperrank * rank |
| 162 | end = start + pperrank - 1 |
| 163 | writer.SetStartPiece(start) |
| 164 | writer.SetEndPiece(end) |
| 165 | writer.SetFileName(filename) |
| 166 | if useSubdir: |
| 167 | writer.SetUseSubdirectory(True) |
| 168 | #writer.SetDataModeToAscii() |
| 169 | writer.Write() |
| 170 | |
| 171 | if contr: |
| 172 | contr.Barrier() |
| 173 | |
| 174 | reader.SetFileName(filename) |
| 175 | |
| 176 | ntris = 0 |
| 177 | |
| 178 | if dataType != "Table": |
| 179 | cf = vtkContourFilter() |
| 180 | cf.SetValue(0, 130) |
| 181 | cf.SetComputeNormals(0) |
| 182 | cf.SetComputeGradients(0) |
| 183 | cf.SetInputConnection(reader.GetOutputPort()) |
| 184 | cf.UpdateInformation() |
| 185 | cf.GetOutputInformation(0).Set(vtkStreamingDemandDrivenPipeline.UPDATE_NUMBER_OF_PIECES(), nranks) |
| 186 | cf.GetOutputInformation(0).Set(vtkStreamingDemandDrivenPipeline.UPDATE_PIECE_NUMBER(), rank) |
| 187 | cf.Update() |
| 188 | |
| 189 | ntris = cf.GetOutput().GetNumberOfCells() |
| 190 | else: |
| 191 | reader.UpdateInformation() |
| 192 | reader.GetOutputInformation(0).Set(vtkStreamingDemandDrivenPipeline.UPDATE_NUMBER_OF_PIECES(), nranks) |
| 193 | reader.GetOutputInformation(0).Set(vtkStreamingDemandDrivenPipeline.UPDATE_PIECE_NUMBER(), rank) |
| 194 | reader.Update() |
| 195 | ntris = reader.GetOutput().GetNumberOfRows() |
| 196 | |
| 197 | da = vtkIntArray() |
| 198 | da.InsertNextValue(ntris) |
| 199 | |
| 200 | da2 = vtkIntArray() |
| 201 | da2.SetNumberOfTuples(1) |
| 202 | contr.AllReduce(da, da2, vtkCommunicator.SUM_OP) |
| 203 | |
| 204 | if rank == 0: |
| 205 | print(da2.GetValue(0)) |
| 206 | import os |
| 207 | os.remove(filename) |
| 208 | for i in range(npieces): |
| 209 | if not useSubdir: |
no test coverage detected