from enigma import eComponentScan, eDVBFrontendParametersSatellite, eDVBFrontendParametersCable, eDVBFrontendParametersTerrestrial

class ServiceScan:
	
	Idle = 1
	Running = 2
	Done = 3
	Error = 4
	
	Errors = { 
		0: "error starting scanning",
		1: "error while scanning",
		2: "no resource manager",
		3: "no channel list"
		}
	
	def scanStatusChanged(self):
		if self.state == self.Running:
			self.progressbar.setValue(self.scan.getProgress())
			if self.scan.isDone():
				errcode = self.scan.getError()
				
				if errcode == 0:
					self.state = self.Done
				else:
					self.state = self.Error
					self.errorcode = errcode
			else:
				self.text.setText(_("scan in progress - %d %% done!\n%d services found!") % (self.scan.getProgress(), self.scan.getNumServices()))
		
		if self.state == self.Done:
			if self.scan.getNumServices() == 1:
				self.text.setText(_("scan done!\nOne service found!"))
			elif self.scan.getNumServices() == 0:
				self.text.setText(_("scan done!\nNo service found!"))
			else:
				self.text.setText(_("scan done!\n%d services found!") % (self.scan.getNumServices()))
		
		if self.state == self.Error:
			self.text.setText(_("ERROR - failed to scan (%s)!") % (self.Errors[self.errorcode]) )
	
	def __init__(self, progressbar, text, transponders, feid, flags):
		self.progressbar = progressbar
		self.text = text
		self.scan = eComponentScan()
		self.state = self.Idle
		self.feid = feid
		self.flags = flags
		self.scanStatusChanged()
		
#		if 1:
#			parm = eDVBFrontendParametersSatellite()
#
#			parm.frequency = 11817000
#			parm.symbol_rate = 27500000
#			parm.polarisation = 1 # eDVBFrontendParametersSatellite.Polarisation.Vertical
#			parm.fec = 3 # eDVBFrontendParametersSatellite.FEC.f3_4;
#			parm.inversion = 1 #eDVBFrontendParametersSatellite.Inversion.Off;
#			parm.orbital_position = 192
#			self.scan.addInitial(parm)
#		else:
#				# until we have a terrestrial.xml:
#			for x in [490000000, 530000000, 546000000, 570000000, 626000000]:
#				parm = eDVBFrontendParametersTerrestrial()
#				parm.frequency = x
#				parm.inversion = 2  # eDVBFrontendParametersTerrestrial.Inversion.Unknown;
#				parm.bandwidth = 0  #eDVBFrontendParametersTerrestrial.Bandwidth.Bw8MHz;
#				parm.code_rate_HP = parm.code_rate_LP = 6 #eDVBFrontendParametersTerrestrial.FEC.fAuto;
#				parm.modulation = 1 #eDVBFrontendParametersTerrestrial.Modulation.QAM16;
#				parm.transmission_mode = 1 # eDVBFrontendParametersTerrestrial.TransmissionMode.TM8k;
#				parm.guard_interval = 0 # eDVBFrontendParametersTerrestrial.GuardInterval.GI_1_32;
#				parm.hierarchy = 0 #eDVBFrontendParametersTerrestrial.Hierarchy.HNone;
#				self.scan.addInitial(parm)
		
		for x in transponders:
			self.scan.addInitial(x)
		
		#self.scan.addInitial(parm)
		
	def execBegin(self):
		self.scan.statusChanged.get().append(self.scanStatusChanged)
		self.state = self.Running
		err = self.scan.start(self.feid, self.flags)
		if err:
			self.state = self.Error
			self.errorcode = 0

		self.scanStatusChanged()
	
	def execEnd(self):
		self.scan.statusChanged.get().remove(self.scanStatusChanged)
		if not self.isDone():
			print "*** warning *** scan was not finished!"
		
		del self.scan

	def isDone(self):
		print "state is %d " % (self.state)
		return self.state == self.Done or self.state == self.Error