def setDisplayStatus(self, displayStatus):
if self.reader:
# Update display status #
if displayStatus:
self.representation = Show(self.reader, self.renderView)
else:
Hide(self.reader, self.renderView)
self.representation = None
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Data display status set')
else:
return createResponse(value=False,
code=-1,
message='Data display status not set')
def setDataArray(self, dataArray):
if self.representation:
# Set data array #
if dataArray['type'] == 'cell':
ColorBy(self.representation, ('CELLS', dataArray['name']))
else:
ColorBy(self.representation, ('POINTS', dataArray['name']))
# Update transfer function #
self.representation.RescaleTransferFunctionToDataRange(
True, False)
# Update view #
self.updateView()
# Return #
return createResponse(value=True, code=1, message='Data array set')
else:
return createResponse(value=False,
code=-1,
message='Data array not set')
def setGridDisplayStatus(self, gridDisplayStatus):
if self.representation:
# Set grid display status #
if gridDisplayStatus:
self.representation.DataAxesGrid.GridAxesVisibility = 1
else:
self.representation.DataAxesGrid.GridAxesVisibility = 0
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Grid display status set')
else:
return createResponse(value=False,
code=-1,
message='Grid display status not set')
def setGridTitle(self, gridAxis, gridTitle):
if self.representation:
# Update title #
if gridAxis == 'X':
self.representation.DataAxesGrid.XTitle = gridTitle
elif gridAxis == 'Y':
self.representation.DataAxesGrid.YTitle = gridTitle
elif gridAxis == 'Z':
self.representation.DataAxesGrid.ZTitle = gridTitle
# Update view #
self.updateView()
# Return #
return createResponse(value=True, code=1, message='Grid title set')
else:
return createResponse(value=False,
code=-1,
message='Grid title not set')
def setColorMapScale(self, inferiorValue, superiorValue):
if self.representation:
# Get current data array #
dataArrayName = self.representation.ColorArrayName[1]
# Update transfer functions #
colorTransferFunction = GetColorTransferFunction(dataArrayName)
opacityTransferFunction = GetOpacityTransferFunction(dataArrayName)
if colorTransferFunction and opacityTransferFunction:
colorTransferFunction.RescaleTransferFunction(
inferiorValue, superiorValue)
opacityTransferFunction.RescaleTransferFunction(
inferiorValue, superiorValue)
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Color map scale set')
else:
return createResponse(value=False,
code=-1,
message='Color map scale not set')
def setWarpByVectorFilterScaleFactor(self, scaleFactor):
if (self.filter == 'warpByVector') and self.warpByVectorFilter:
# Set ScaleFactor #
self.warpByVectorFilter.ScaleFactor = scaleFactor
# Update view #
self.updateView()
# Return #
return createResponse(
value=True,
code=1,
message='WarpByVector filter scale factor set')
else:
return createResponse(
value=False,
code=-1,
message='WarpByVector filter scale factor not set')
def setColorMapPreset(self, colorMapPreset):
colorMapPresetCodes = {
'coolToWarm': 'Cool to Warm',
'warmToCool': 'Warm to Cool',
'coldAndHot': 'Cold and Hot',
'blackBodyRadiation': 'Black-Body Radiation',
'rainbow': 'rainbow',
}
if colorMapPreset in colorMapPresetCodes:
colorMapPresetCode = colorMapPresetCodes[colorMapPreset]
if self.representation:
# Get current data array #
dataArrayName = self.representation.ColorArrayName[1]
# Set color map #
colorTransferFunction = GetColorTransferFunction(dataArrayName)
if colorTransferFunction:
colorTransferFunction.ApplyPreset(colorMapPresetCode, True)
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='ColorMap preset set')
else:
return createResponse(value=False,
code=-3,
message='ColorMap preset not set')
else:
return createResponse(value=False,
code=-2,
message='ColorMap preset not set')
else:
return createResponse(value=False,
code=-1,
message='ColorMap preset not valid')
def setLegendTitle(self, legendTitle):
if self.renderView and self.representation:
# Get current data array #
dataArrayName = self.representation.ColorArrayName[1]
# Set color map #
colorTransferFunction = GetColorTransferFunction(dataArrayName)
if colorTransferFunction:
calarBarWidgetRepresentation = GetScalarBar(
colorTransferFunction, self.renderView)
if calarBarWidgetRepresentation:
calarBarWidgetRepresentation.Title = legendTitle
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='ColorMap title set')
else:
return createResponse(value=False,
code=-3,
message='ColorMap title not set')
else:
return createResponse(value=False,
code=-2,
message='ColorMap title not set')
else:
return createResponse(value=False,
code=-1,
message='ColorMap title not set')
def enableWarpByVectorFilter(self):
# Disable current filter #
self.disableCurrentFilter()
# Create filter #
self.warpByVectorFilter = WarpByVector(Input=self.reader)
# Display filter #
Show(self.warpByVectorFilter, self.renderView)
# Update indicator #
self.filter = 'warpByVector'
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='WarpByVector filter enabled',
data={
'scaleFactor':
self.warpByVectorFilter.ScaleFactor,
'vectors':
self.warpByVectorFilter.Vectors[1],
})
def setContourFilterData(self, pointArrayName, field):
if (self.filter == 'contour'
) and self.contourFilterCalculator and self.contourFilter:
# Compute function #
function = None
if field == 'mag':
function = 'mag(' + pointArrayName + ')'
elif field == 'X':
function = pointArrayName + '_X'
elif field == 'Y':
function = pointArrayName + '_Y'
elif field == 'Z':
function = pointArrayName + '_Z'
# Set Function #
self.contourFilterCalculator.Function = function
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Contour filter function set')
else:
return createResponse(value=False,
code=-1,
message='Contour filter function not set')
def loadData(self, dataLoadSignature):
# Reset #
if self.reader:
Delete(self.reader)
# Extract filePath from signature #
dataLoadSignatureDecoderResponse = decodeDataLoadSignature(
self.dataLoadSignatureDecoder, dataLoadSignature)
if dataLoadSignatureDecoderResponse['value']:
self.fileName = computeFullFilePath(
self.dataDirectoryPath,
dataLoadSignatureDecoderResponse['data']['filePath'])
print('Loading: ' + self.fileName)
# Test file existence #
if os.path.isfile(self.fileName):
return self.initializeVisualization()
else:
print('Not loaded')
return createResponse(value=False,
code=-2,
message='Data loading failed')
else:
print('Invalid dataLoadSignature')
return createResponse(value=False,
code=-1,
message='DataLoadSignature decoding failed')
def setCameraPosition(self, cameraPosition):
# Compute camera parameters #
focalPoint = [0, 0, 0]
viewUp = [0, 0, 0]
if cameraPosition == '+X':
focalPoint[0] = 1
viewUp[2] = 1
elif cameraPosition == '-X':
focalPoint[0] = -1
viewUp[2] = 1
if cameraPosition == '+Y':
focalPoint[1] = 1
viewUp[2] = 1
elif cameraPosition == '-Y':
focalPoint[1] = -1
viewUp[2] = 1
if cameraPosition == '+Z':
focalPoint[2] = 1
viewUp[1] = 1
elif cameraPosition == '-Z':
focalPoint[2] = -1
viewUp[1] = 1
self.renderView.CameraPosition = [0, 0, 0]
self.renderView.CameraFocalPoint = focalPoint
self.renderView.CameraViewUp = viewUp
# Reset camera #
ResetCamera()
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Camera position set')
def setBackgroundColor(self, backgroundColor):
if self.renderView:
if len(backgroundColor) == 6:
# Extract components #
R = float(convertHexadecimalToDecimal(
backgroundColor[0:2])) / 255
G = float(convertHexadecimalToDecimal(
backgroundColor[2:4])) / 255
B = float(convertHexadecimalToDecimal(
backgroundColor[4:6])) / 255
# Set background color #
self.renderView.Background = [R, G, B]
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Background color set')
else:
return createResponse(value=False,
code=-2,
message='Background color invalid')
else:
return createResponse(value=False,
code=-1,
message='Background color not set')
def setColorMapLogScaleStatus(self, colorMapLogScaleStatus):
if self.representation:
# Get current data array #
dataArrayName = self.representation.ColorArrayName[1]
# Update color transfer function #
colorTransferFunction = GetColorTransferFunction(dataArrayName)
if colorTransferFunction:
if colorMapLogScaleStatus:
colorTransferFunction.MapControlPointsToLogSpace()
colorTransferFunction.UseLogScale = 1
else:
colorTransferFunction.MapControlPointsToLinearSpace()
colorTransferFunction.UseLogScale = 0
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Color map log scale status set')
else:
return createResponse(value=False,
code=-1,
message='Color map log scale status not set')
def setTimeStep(self, timeStep):
# Set time step #
GetAnimationScene().TimeKeeper.Time = timeStep
# Update view #
self.updateView()
# Return #
return createResponse(value=True, code=1, message='Time step set')
def setOrientationVisibility(self, orientationAxesVisibility):
if self.renderView:
# Set orientation visibility #
self.renderView.OrientationAxesVisibility = orientationAxesVisibility
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Orientation visibility set')
else:
return createResponse(value=False,
code=-1,
message='Orientation visibility not set')
def setRepresentationType(self, representationType):
if self.representation:
# Set representation type #
self.representation.SetRepresentationType(representationType)
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Representation type set')
else:
return createResponse(value=False,
code=-1,
message='Representation type not set')
def resetColorMapScale(self):
if self.representation:
# Set scale bar visibility #
self.representation.RescaleTransferFunctionToDataRange(False, True)
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Color map scale reset')
else:
return createResponse(value=False,
code=-1,
message='Color map scale not reset')
def setLegendDisplayStatus(self, legendDisplayStatus):
if self.representation:
# Set legend display status #
self.representation.SetScalarBarVisibility(self.renderView,
legendDisplayStatus)
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Legend display status set')
else:
return createResponse(value=False,
code=-1,
message='Legend display status not set')
def setWarpByVectorFilterVectors(self, vectors):
if (self.filter == 'warpByVector') and self.warpByVectorFilter:
# Set ScaleFactor #
self.warpByVectorFilter.Vectors = vectors
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='WarpByVector filter vectors set')
else:
return createResponse(
value=False,
code=-1,
message='WarpByVector filter vectors not set')
def setContourFilterValues(self, values):
if (self.filter == 'contour'
) and self.contourFilterCalculator and self.contourFilter:
# Set Function #
self.contourFilter.Isosurfaces = values
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Contour filter values set')
else:
return createResponse(value=False,
code=-1,
message='Contour filter values not set')
def resetView(self):
# Reset camera #
ResetCamera()
# Update center of rotation #
self.renderView.CenterOfRotation = self.renderView.CameraFocalPoint
# Update view #
self.updateView()
# Return #
return createResponse(value=True, code=1, message='View reset')
def disableWarpByVectorFilter(self):
# Hide filter #
Hide(self.warpByVectorFilter, self.renderView)
self.warpByVectorFilter = None
# Update indicator #
self.filter = None
# Update view #
self.updateView()
# Return #
return createResponse(
value=True,
code=1,
message='WarpByVector filter disabled',
)
def disableContourFilter(self):
# Hide filter #
Hide(self.contourFilter, self.renderView)
self.contourFilter = None
self.contourFilterCalculator = None
# Update indicator #
self.filter = None
# Update view #
self.updateView()
# Return #
return createResponse(
value=True,
code=1,
message='Contour filter disabled',
)
def decodeDataLoadSignature(dataLoadSignatureDecoderExecutablePath, dataLoadSignature):
if(True or os.path.isfile(dataLoadSignatureDecoderExecutablePath)):
if(dataLoadSignature):
commandParts = [
dataLoadSignatureDecoderExecutablePath,
'--data-load-signature',
str(dataLoadSignature)
]
commandLine = ' '.join(map(str, commandParts))
print('CommandLine: ', commandLine)
print('User: '******'status'] == True):
return createResponse(
value=True,
code=1,
message='DataLoadSignature decoded',
data={
'filePath': response['data']['filePath'],
}
)
else:
return createResponse(
value=False,
code=-6,
message='DataLoadSignature not valid'
)
else:
return createResponse(
value=False,
code=-3,
message='DataLoadSignature decoder executable response is not valid'
)
else:
return createResponse(
value=False,
code=-2,
message='No dataLoadSignature'
)
else:
return createResponse(
value=False,
code=-1,
message='Can not find dataLoadSignature decoder executable'
)
def enableContourFilter(self):
pointArrays = self.reader.PointArrays
if len(pointArrays):
# Disable current filter #
self.disableCurrentFilter()
# Create calculator #
self.contourFilterCalculator = Calculator(Input=self.reader)
self.contourFilterCalculator.ResultArrayName = 'Result'
self.contourFilterCalculator.Function = 'mag(' + pointArrays[
0] + ')'
# Get range #
self.contourFilterCalculator.UpdatePipeline()
range = self.contourFilterCalculator.GetDataInformation(
).DataInformation.GetPointDataInformation().GetArrayInformation(
'Result').GetComponentRange(0)
# Create filter #
self.contourFilter = Contour(Input=self.contourFilterCalculator)
self.contourFilter.Isosurfaces = [
round((range[0] + range[1]) / 2, 2)
]
self.contourFilter.ContourBy = ['POINTS', 'Result']
# Display filter #
Show(self.contourFilter, self.renderView)
# Update indicator #
self.filter = 'contour'
# Update view #
self.updateView()
# Return #
return createResponse(value=True,
code=1,
message='Contour filter enabled',
data={
'data': {
'pointArrayName': pointArrays[0],
'field': 'mag',
},
'values':
list(self.contourFilter.Isosurfaces),
})
else:
return createResponse(value=False,
code=-1,
message='Contour filter not enabled')
def initializeVisualization(self):
# Read data #
self.reader = EnSightReader(CaseFileName=self.fileName)
print('Loaded')
# Display data #
self.setDisplayStatus(True)
self.resetView()
# Data #
## Data arrays #
dataArrays = []
for pointArray in self.reader.PointArrays:
dataArrays.append({
'type': 'point',
'name': pointArray,
})
for cellArray in self.reader.CellArrays:
dataArrays.append({
'type': 'cell',
'name': cellArray,
})
dataArray = {
'type':
'point'
if self.representation.ColorArrayName[0] == 'POINTS' else 'cell',
'name':
self.representation.ColorArrayName[1]
}
## Representation type ##
representationTypes = formatPropertyValueAsList(
self.representation.GetPropertyValue('RepresentationTypesInfo'))
representationType = str(
self.representation.GetPropertyValue('Representation'))
## Time steps ##
timeSteps = formatPropertyValueAsList(
GetAnimationScene().TimeKeeper.TimestepValues)
timeStep = GetAnimationScene().TimeKeeper.Time
## Legend ##
legendDisplayStatus = self.representation.IsScalarBarVisible(
self.renderView)
## GridDisplayStatus ##
gridDisplayStatus = False
## Background color ##
R = convertDecimalToHexadecimal(
int(self.renderView.Background[0] * 255))
G = convertDecimalToHexadecimal(
int(self.renderView.Background[1] * 255))
B = convertDecimalToHexadecimal(
int(self.renderView.Background[2] * 255))
backgroundColor = R + G + B
# Return #
return createResponse(value=True,
code=1,
message='Data loading succeed',
data={
'dataDisplayStatus': True,
'dataArrays': dataArrays,
'dataArray': dataArray,
'representationTypes': representationTypes,
'representationType': representationType,
'timeSteps': timeSteps,
'timeStep': timeStep,
'legendDisplayStatus': legendDisplayStatus,
'gridDisplayStatus': gridDisplayStatus,
'backgroundColor': backgroundColor,
})
