def __init__(self, plotName, fileName=None, dataClass=None, showPlots=True):
DataSet.__init__(self, fileName=fileName, dataClass=dataClass)
print('-'*70)
print('Establishing {} Plot'.format(plotName))
print('-'*70)
self.plotName = plotName
#TODO: fix this to acocmodate non-colorbar plots
self.figure = plt.figure(figsize=(6,5))
self.axes = self.figure.add_axes([0.1, 0.1, 0.825*5/6, 0.825])
self.colorBarAxes = self.figure.add_axes([0.825, 0.1, 0.05, 0.825])
self.animationImages = [[] for _ in range(len(self.blockData.keys()))]
if showPlots != True:
matplotlib.use('Agg')
time = min(self.blockData.keys())
self.blocks = self.blockData[time].keys()
self.zones = self.zoneData[time].keys()
self.corners = self.cornerData[time].keys()
self.contacts = self.contactData[time].keys()
self.gridPoints = self.gridPointData[time].keys()
self.domains = self.domainData[time].keys()
def __init__(self):
"""
Constructor.
@ In, None
@ Out, None
"""
DataSet.__init__(self)
self.name = 'PointSet'
self.type = 'PointSet'
self.printTag = self.name
self._neededForReload = [] # PointSet doesn't need anything to reload
def __init__(self, centre, radius, dataClass=None, fileName=None, ):
DataSet.__init__(self, dataClass=dataClass, fileName=fileName)
self.centre = centre
self.radius = radius
self.singleBlock = False
if len(self.contactData) == 0:
self.singleBlock = True
self.calculateHomogenizationParameters()
def __init__(self):
"""
Constructor.
@ In, None
@ Out, None
"""
DataSet.__init__(self)
self.name = 'HistorySet'
self.type = 'HistorySet'
self.printTag = self.name
self._tempPivotParam = None
self._neededForReload = [
] # HistorySet doesn't need anything special to load, since it's written in cluster-by-sample CSV format
def __init__(self):
"""
Constructor.
@ In, None
@ Out, None
"""
DataSet.__init__(self)
self.name = 'HistorySet'
self.type = 'HistorySet'
self.printTag = self.name
self._tempPivotParam = None
self._neededForReload = [] # HistorySet doesn't need anything special to load, since it's written in cluster-by-sample CSV format
self._inputMetaVars = [] # meta vars belong to the input of HistorySet, i.e. scalar
self._outputMetaVars = [] # meta vara belong to the output of HistorySet, i.e. vector
def __init__(self, mainDataFile, mediaTypeDataFile): self.mainDataFile = mainDataFile self.mediaTypeDataFile = mediaTypeDataFile DataSet.__init__(self)
