(
self,
location,
sampling_dimesions,
fields_to_keep,
custom_probing_bounds=None,
metadata={},
)
| 326 | # ----------------------------------------------------------------------------- |
| 327 | class DataProberDataSetBuilder(DataSetBuilder): |
| 328 | def __init__( |
| 329 | self, |
| 330 | location, |
| 331 | sampling_dimesions, |
| 332 | fields_to_keep, |
| 333 | custom_probing_bounds=None, |
| 334 | metadata={}, |
| 335 | ): |
| 336 | DataSetBuilder.__init__(self, location, None, metadata) |
| 337 | self.fieldsToWrite = fields_to_keep |
| 338 | self.resamplerFilter = vtkPResampleFilter() |
| 339 | self.resamplerFilter.SetSamplingDimension(sampling_dimesions) |
| 340 | if custom_probing_bounds: |
| 341 | self.resamplerFilter.SetUseInputBounds(0) |
| 342 | self.resamplerFilter.SetCustomSamplingBounds(custom_probing_bounds) |
| 343 | else: |
| 344 | self.resamplerFilter.SetUseInputBounds(1) |
| 345 | |
| 346 | # Register all fields |
| 347 | self.dataHandler.addTypes("data-prober", "binary") |
| 348 | self.DataProber = { |
| 349 | "types": {}, |
| 350 | "dimensions": sampling_dimesions, |
| 351 | "ranges": {}, |
| 352 | "spacing": [1, 1, 1], |
| 353 | } |
| 354 | for field in self.fieldsToWrite: |
| 355 | self.dataHandler.registerData( |
| 356 | name=field, type="array", fileName="/%s.array" % field |
| 357 | ) |
| 358 | |
| 359 | def setDataToProbe(self, dataset): |
| 360 | self.resamplerFilter.SetInputData(dataset) |
nothing calls this directly
no test coverage detected