from paraview.simple import *
from paraview import coprocessing


# --------------------------------------------------------------
# Code generated from cpstate.py to create the CoProcessor.
# ParaView 5.2.0 64 bits


# ----------------------- CoProcessor definition -----------------------


def CreateCoProcessor():
    def _CreatePipeline(coprocessor, datadescription):
        class Pipeline:
            # state file generated using paraview version 5.2.0

            # ----------------------------------------------------------------
            # setup the data processing pipelines
            # ----------------------------------------------------------------

            #### disable automatic camera reset on 'Show'
            paraview.simple._DisableFirstRenderCameraReset()

            # create a new 'XML Unstructured Grid Reader'
            # create a producer from a simulation input
            cube_1e0_pcs_0_ts_0_t_0000000vtu = coprocessor.CreateProducer(
                datadescription, "input"
            )

            # create a new 'Contour'
            contour1 = Contour(Input=cube_1e0_pcs_0_ts_0_t_0000000vtu)
            contour1.ContourBy = ["POINTS", "D1_left_front_N1_right"]
            contour1.ComputeScalars = 1
            contour1.Isosurfaces = [
                1.0,
                1.0750344444444444,
                1.1500688888888888,
                1.2251033333333334,
                1.3001377777777778,
                1.3751722222222222,
                1.4502066666666669,
                1.5252411111111113,
                1.6002755555555557,
                1.67531,
            ]
            contour1.PointMergeMethod = "Uniform Binning"

            # create a new 'Parallel PolyData Writer'
            parallelPolyDataWriter1 = servermanager.writers.XMLPPolyDataWriter(
                Input=contour1
            )

            # register the writer with coprocessor
            # and provide it with information such as the filename to use,
            # how frequently to write the data, etc.
            coprocessor.RegisterWriter(
                parallelPolyDataWriter1, filename="cube_1e1_%t.pvtp", freq=1
            )

            # ----------------------------------------------------------------
            # finally, restore active source
            SetActiveSource(contour1)
            # ----------------------------------------------------------------

        return Pipeline()

    class CoProcessor(coprocessing.CoProcessor):
        def CreatePipeline(self, datadescription):
            self.Pipeline = _CreatePipeline(self, datadescription)

    coprocessor = CoProcessor()
    # these are the frequencies at which the coprocessor updates.
    freqs = {"input": [1]}
    coprocessor.SetUpdateFrequencies(freqs)
    return coprocessor


# --------------------------------------------------------------
# Global variables that will hold the pipeline for each timestep
# Creating the CoProcessor object, doesn't actually create the ParaView pipeline.
# It will be automatically setup when coprocessor.UpdateProducers() is called the
# first time.
coprocessor = CreateCoProcessor()

# --------------------------------------------------------------
# Enable Live-Visualizaton with ParaView
coprocessor.EnableLiveVisualization(False, 1)


# ---------------------- Data Selection method ----------------------


def RequestDataDescription(datadescription):
    "Callback to populate the request for current timestep"
    global coprocessor
    if datadescription.GetForceOutput() == True:
        # We are just going to request all fields and meshes from the simulation
        # code/adaptor.
        for i in range(datadescription.GetNumberOfInputDescriptions()):
            datadescription.GetInputDescription(i).AllFieldsOn()
            datadescription.GetInputDescription(i).GenerateMeshOn()
        return

    # setup requests for all inputs based on the requirements of the
    # pipeline.
    coprocessor.LoadRequestedData(datadescription)


# ------------------------ Processing method ------------------------


def DoCoProcessing(datadescription):
    "Callback to do co-processing for current timestep"
    global coprocessor

    # Update the coprocessor by providing it the newly generated simulation data.
    # If the pipeline hasn't been setup yet, this will setup the pipeline.
    coprocessor.UpdateProducers(datadescription)

    # Write output data, if appropriate.
    coprocessor.WriteData(datadescription)

    # Write image capture (Last arg: rescale lookup table), if appropriate.
    coprocessor.WriteImages(datadescription, rescale_lookuptable=False)

    # Live Visualization, if enabled.
    coprocessor.DoLiveVisualization(datadescription, "localhost", 22222)
