Skip to content
This repository has been archived by the owner on Jan 25, 2024. It is now read-only.

Commit

Permalink
feat: read extracted data instead of the original
Browse files Browse the repository at this point in the history
  • Loading branch information
seanwu1105 committed Apr 15, 2023
1 parent e24b6ba commit 2d6b211
Showing 1 changed file with 3 additions and 14 deletions.
17 changes: 3 additions & 14 deletions main.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,7 +6,7 @@
from vtkmodules.vtkIOXML import vtkXMLStructuredGridReader
from vtkmodules.vtkRenderingCore import vtkRenderer

from src.data import DATA_DIR, FILE_ID_MIN, to_filename
from src.data import EXTRACTED_DIR, FILE_ID_MIN, to_filename
from src.temporal import build_temporal_gui
from src.vegetation import get_vegetation_actor
from src.vtk_side_effects import import_for_rendering_core, import_for_rendering_volume
Expand Down Expand Up @@ -35,27 +35,16 @@
######## VTK Widget

reader = vtkXMLStructuredGridReader()
reader.SetFileName(str(DATA_DIR / to_filename(FILE_ID_MIN)))
reader.SetFileName(str(EXTRACTED_DIR / to_filename(FILE_ID_MIN)))


def on_time_changed(value: str):
reader.SetFileName(str(DATA_DIR / to_filename(int(value))))
reader.SetFileName(str(EXTRACTED_DIR / to_filename(int(value))))
vtk_widget.GetRenderWindow().Render()


temporal_spinbox.textChanged.connect(on_time_changed)

# Disable the arrays we don't need to save memory.
reader.SetPointArrayStatus("u", 0)
reader.SetPointArrayStatus("v", 0)
reader.SetPointArrayStatus("w", 0)
reader.SetPointArrayStatus("theta", 0)
reader.SetPointArrayStatus("O2", 0)
reader.SetPointArrayStatus("rhowatervapor", 0)
reader.SetPointArrayStatus("rhof_1", 1)
reader.SetPointArrayStatus("convht_1", 0)
reader.SetPointArrayStatus("frhosiesrad_1", 0)

renderer = vtkRenderer()
renderer.AddActor(get_vegetation_actor(reader.GetOutputPort()))
colors = vtkNamedColors()
Expand Down

0 comments on commit 2d6b211

Please sign in to comment.