Source code for diaGrabber.source._dimension

# -*- coding: utf-8 *-*
from diaGrabber.methods.merge import mean
import numpy as np

class dimension(object):
	def __init__(self,name, index):
		self.name = name
		self.index = index
		self._is_counter = False # take value from source by default
		self.calc = []
		self.exclude = []
		self.merge = mean()#standard
		self.transform = False#can be loaded up with some methods.transform....
		self._take_all_values = True
		self._include_chronic = False
		self.include_from_to = [None, None]
		self._plot_range = slice(None,None,None)
		self._plotOnlyRecentPosition = False
		self._recent_position = 0 #will be updated by target-class


	def setPlotOnlyRecentPosition(self):
		self._plotOnlyRecentPosition = True

	def setCounter(self, start = 0, delta = 1):
		self._is_counter = True
		self.index = 0 #to prevent error when reading out values from sources
		#self._recent_number = int(0)
		self._recent_value = start#-delta
		self._delta = delta
	
	def setPlotRange(self, start, stop, step):
		"""type 'None' if you don't want to define the input"""
		self._plot_range = slice(start,stop,step)


	def _updateCounter(self):
		#self._recent_number += 1
		self._recent_value += self._delta


	def includeAll(self, resolution):
		'''Include all values and sort in a range from min to max values
		dissolved by the resolution
		!!!: not all sources can define the min and max values '''
		self._take_all_values = True
		self.resolution = int(resolution)
		###########get min max from source ##############
		#!!!!!!!!!!!!!!!!!!!!!!!!!!!
		self._initSortRange()

	
	def includeFromTo(self,from_value,to_value, resolution):
		self.include_from_to = [from_value,to_value]
		self._take_all_values = False
		self.resolution = int(resolution)
		self._initSortRange()

	def includeChronic(self, resolution):
		'''Only take the last [dimension.resolution] values
		override when read more than that'''
		self._take_all_values = False
		self._include_chronic = True
		self.resolution = int(resolution)
		#set the first include_from_to ... this has to be updated
		self.include_from_to = [-self.resolution+1,self.resolution-1]
		self._chronic_step = 0
		#self._updateRange = self._updateChronic()
		self._initSortRange()

	def _updateChronic(self, ):
		self._chronic_step += 1
		if self._chronic_step >= self.resolution:
			#update range
			#self.include_from_to[0] += self.resolution
			#self.include_from_to[1] += self.resolution
			#self.include_from_to[0] = self.resolution
			#self.include_from_to[1] = self.resolution
			self._chronic_step = 0
		#print self.include_from_to, self._chronic_step, self._sort_range
			
		self._sort_range[self._chronic_step] = self._recent_value
		#the smallest value in the chronic is the value infront of the recent one
		try:
			self.include_from_to[0] = self._sort_range[self._chronic_step+1]
		# except the revent value ist the last value in range - than its the first one
		except IndexError:
			self.include_from_to[0] = self._sort_range[0]
		self.include_from_to[1] = self._recent_value
		#print self._sort_range,self._chronic_step, self._recent_value

	def _initSortRange(self):
		self._sort_range = np.linspace(self.include_from_to[0],self.include_from_to[1],
			self.resolution)
			



	def _update(self, in_range, source_value):
		if self._is_counter:
			self._updateCounter()
		else:
			self._recent_value = source_value

		#in_range = True
		if self._include_chronic:
			self._updateChronic()
		elif self._take_all_values:
			#update self.include_from_to with max in min values of the dim
			if source_value < self.include_from_to[0] or self.include_from_to[0] == None:
				self.include_from_to[0] = source_value
			if source_value > self.include_from_to[1] or self.include_from_to[1] == None:
				self.include_from_to[1] = source_value
		else:
			#check whether in range
			if self.include_from_to[0] > self._recent_value or \
					self.include_from_to[1] < self._recent_value:
				in_range = False

		if in_range:
			#calc values bounded to dim
			for c in self.calc:
				c._get(self._recent_value)
			for c in self.exclude:
				if c._get():
					in_range = False

			if self._plotOnlyRecentPosition:
				self._plot_range = slice(self._recent_position,self._recent_position+1,1)

		return in_range, self._recent_value