from paraview.simple import *

from paraview import smtesting
from paraview.modules.vtkRemotingSettings import vtkPVGeneralSettings

smtesting.ProcessCommandLineArguments()

filename = smtesting.DataDir + '/Testing/Data/can.ex2'
can_ex2 = OpenDataFile(filename)

AnimationScene1 = GetAnimationScene()
AnimationScene1.EndTime = 0.004299988504499197
AnimationScene1.PlayMode = 'Snap To TimeSteps'

can_ex2.PointVariables = ['ACCL', 'DISPL', 'VEL']
can_ex2.ElementVariables = ['EQPS']
can_ex2.GlobalVariables = ['KE', 'NSTEPS', 'TMSTEP', 'XMOM', 'YMOM', 'ZMOM']
can_ex2.ApplyDisplacements = 0
can_ex2.ElementBlocks = ['Unnamed block ID: 1 Type: HEX Size: 4800', 'Unnamed block ID: 2 Type: HEX Size: 2352']

DataRepresentation1 = Show()
RenderView1 = GetRenderView()

RenderView1.CameraPosition = [0.21706008911132812, 4.0, 46.629626614745732]
RenderView1.CameraFocalPoint = [0.21706008911132812, 4.0, -5.1109471321105957]
RenderView1.CameraParallelScale = 13.391445890217907
RenderView1.CenterOfRotation = [0.21706008911132812, 4.0, -5.1109471321105957]

RenderView1.CameraViewUp = [0.0, 0.0, 1.0]
RenderView1.CameraPosition = [0.21706008911132812, 55.740573746856327, -5.1109471321105957]
RenderView1.CameraFocalPoint = [0.21706008911132812, 4.0, -5.1109471321105957]

Render()

update_counters = 0;
def __request_data_callback(*args):
    global update_counters
    update_counters += 1

oid = can_ex2.GetClientSideObject().AddObserver("StartEvent", __request_data_callback)

#---------------------------------------------------------
# Initial state, nothing cached, or expected to be cached.
update_counters = 0
AnimationScene1.GoToNext()
assert update_counters == 1

#---------------------------------------------------------
# nothing cached, or expected to be cached since we didn't enable caching.
update_counters = 0
AnimationScene1.GoToNext()
AnimationScene1.GoToNext()
AnimationScene1.GoToPrevious()
assert update_counters == 3

#---------------------------------------------------------
# Enable caching
vtkPVGeneralSettings.GetInstance().SetCacheGeometryForAnimation(True)

#---------------------------------------------------------
# Nothing in cache, but we expect the data to be cached for next time.

update_counters = 0
AnimationScene1.GoToFirst()
assert update_counters == 0 # it ends up being 0, since default cache key is 0

update_counters = 0
AnimationScene1.GoToNext()
assert update_counters == 1

#---------------------------------------------------------
# now we expect the cache to hit and have no misses.
update_counters = 0
assert update_counters == 0

#---------------------------------------------------------
# Let's fill up the cache fully. We'll have some hits and misses, but no skips.
update_counters = 0
AnimationScene1.GoToFirst()
AnimationScene1.Play()
assert update_counters > 0


#---------------------------------------------------------
# Play again, this time cache should be used.
update_counters = 0
AnimationScene1.GoToFirst()
AnimationScene1.Play()
assert update_counters == 0

#---------------------------------------------------------
# Modify properties and play. this time it should be a cache miss since
# the cache should have been cleared.
update_counters = 0
can_ex2.PointVariables = ['ACCL', 'DISPL']
Render()
AnimationScene1.GoToFirst()
AnimationScene1.Play()
assert update_counters > 0

#---------------------------------------------------------
# Modify representation properties and play.
# This time, it should have any cache misses or clears.
update_counters = 0
DataRepresentation1.LineWidth = 5
AnimationScene1.GoToFirst()
AnimationScene1.Play()
assert update_counters == 0

#---------------------------------------------------------
# Let's modified "Representation" and play.
# This time, since we now have a different representation active, we expect
# updates
update_counters = 0
DataRepresentation1.SetRepresentationType("Outline")
AnimationScene1.Play()
assert update_counters > 0

can_ex2.GetClientSideObject().RemoveObserver(oid)
print("All's well that ends well! Looks like the cache is working as expected.")
