| 338 | } |
| 339 | |
| 340 | void G4VtkViewer::ExportScreenShot(G4String path, G4String format) |
| 341 | { |
| 342 | vtkImageWriter* imWriter = nullptr; |
| 343 | |
| 344 | if (format == "bmp") { |
| 345 | imWriter = vtkBMPWriter::New(); |
| 346 | } |
| 347 | else if (format == "jpg") { |
| 348 | imWriter = vtkJPEGWriter::New(); |
| 349 | } |
| 350 | else if (format == "pnm") { |
| 351 | imWriter = vtkPNMWriter::New(); |
| 352 | } |
| 353 | else if (format == "png") { |
| 354 | imWriter = vtkPNGWriter::New(); |
| 355 | } |
| 356 | else if (format == "tiff") { |
| 357 | imWriter = vtkTIFFWriter::New(); |
| 358 | } |
| 359 | else if (format == "ps") { |
| 360 | imWriter = vtkPostScriptWriter::New(); |
| 361 | } |
| 362 | else { |
| 363 | return; |
| 364 | } |
| 365 | |
| 366 | _renderWindow->Render(); |
| 367 | |
| 368 | vtkSmartPointer<vtkWindowToImageFilter> winToImage = |
| 369 | vtkSmartPointer<vtkWindowToImageFilter>::New(); |
| 370 | winToImage->SetInput(_renderWindow); |
| 371 | winToImage->SetScale(1); |
| 372 | if (format == "ps") { |
| 373 | winToImage->SetInputBufferTypeToRGB(); |
| 374 | winToImage->ReadFrontBufferOff(); |
| 375 | winToImage->Update(); |
| 376 | } |
| 377 | else { |
| 378 | winToImage->SetInputBufferTypeToRGBA(); |
| 379 | } |
| 380 | |
| 381 | imWriter->SetFileName((path + "." + format).c_str()); |
| 382 | imWriter->SetInputConnection(winToImage->GetOutputPort()); |
| 383 | imWriter->Write(); |
| 384 | } |
| 385 | |
| 386 | void G4VtkViewer::ExportOBJScene(G4String path) |
| 387 | { |
no test coverage detected