import sys
import os

#### import the simple module from the paraview
from paraview.simple import *

cwd = os.getcwd() # current working directory where input and output files are saved

filename1 = sys.argv[1]

pngoutname = "test.png"
resolution = [1920,1080] # Full HD resolution

# get active view
renderView1 = GetActiveViewOrCreate('RenderView')
renderView1.Background = [0.0, 0.0, 0.0] # black background

# camera parameters for setting up the view
camera_posx = -4.7946006
camera_posy =  2.0
camera_posz = -6.91633375

focalpoint_x = 0.0
focalpoint_y = 0.0
focalpoint_z = 4.0

camera_viewup_x = 0.0
camera_viewup_y = 1.0
camera_viewup_z = 0.0

view_angle = 22.5
eye_angle  = 2.0
focal_distance = 0.0
focal_disk = 1

camera_parallel_scale = 25.0

# render fixed params
renderView1.CameraPosition      = [camera_posx ,camera_posy ,camera_posz]
renderView1.CameraFocalPoint    = [focalpoint_x,focalpoint_y,focalpoint_z]
renderView1.CameraViewUp        = [camera_viewup_x,camera_viewup_y,camera_viewup_z]
renderView1.CameraParallelScale = camera_parallel_scale
renderView1.CameraParallelProjection = 0

#######################################################################################################################
################################################## FILES READING ######################################################
#######################################################################################################################
# create a new 'PLOT3D Reader'
plane = PLOT3DReader(registrationName=filename1, FileName=cwd+"/"+filename1,  # filename1 is the grid file (e.g. ".x")
    QFileName=[''],
    FunctionFileName=cwd+"/"+filename1.split('.x')[0]+".q")                     # remove ".x" to filename1 and append ".q" for field reading
plane.Functions = []
plane.PreserveIntermediateFunctions = 0
#######################################################################################################################
############################################## render data in view ####################################################
#######################################################################################################################
planeDisplay = Show(plane, renderView1, 'StructuredGridRepresentation')
planeDisplay.Representation = 'Surface'
planeDisplay.ColorArrayName = [None, '']
# set scalar coloring
ColorBy(planeDisplay, ('POINTS', 'Function0'))
renderView1.InteractionMode = '3D'
# Hide orientation axes
renderView1.OrientationAxesVisibility = 0
# Hide color bar/color legend
planeDisplay.SetScalarBarVisibility(renderView1, False)

# get color transfer function/color map for 'Function0'
function0LUT = GetColorTransferFunction('Function0')
# Rescale transfer function
function0LUT.RescaleTransferFunction(0.0, 30.0)
# Apply colormap
function0LUT.ApplyPreset('Cool to Warm (Extended)', True)
# Set text to Black
function0LUTbar = GetScalarBar(function0LUT,renderView1)
function0LUTbar.TitleColor = [0.0, 0.0, 0.0]
function0LUTbar.LabelColor = [0.0, 0.0, 0.0]
# get opacity transfer function/opacity map for 'Function0'
function0PWF = GetOpacityTransferFunction('Function0')
# Rescale transfer function
function0PWF.RescaleTransferFunction(0.0, 30.0)

# hide color bar/color legend, False = hide, True = show
planeDisplay.SetScalarBarVisibility(renderView1, False)

layout1 = GetLayout()
layout1.SetSize(resolution) # set render layout equal to picture resolution

# Create a new 'Light' to improve color contrast
light1 = AddLight(view=renderView1)
Show3DWidgets(proxy=light1)
light1.Intensity = 0.2
light1.Enable = 1
light1.FocalPoint = [focalpoint_x,focalpoint_y,focalpoint_z]
light1.Position   = [camera_posx ,camera_posy ,camera_posz]
Show3DWidgets(proxy=light1)

#######################################################################################################################
################################################### save screenshot ###################################################
#######################################################################################################################
SaveScreenshot(cwd+"/"+pngoutname, renderView1, ImageResolution=resolution, 
    # PNG options
    CompressionLevel='0')

#######################################################################################################################
################################################### clean memory ######################################################
#######################################################################################################################
Delete(plane)