aboutsummaryrefslogtreecommitdiffstats
path: root/firmware/simulator/circuits/pwm.py
blob: dc330f5b9baa33d8b916c5b0e8a86a09dbaf5319 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
"""
# Xytronic LF-1600
# Open Source firmware
# PWM calculation
#
# Copyright (c) 2019 Michael Buesch <m@bues.ch>
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License along
# with this program; if not, write to the Free Software Foundation, Inc.,
# 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
"""

__all__ = [
	"PWMGenerator",
]

class PWMGenerator(object):
	__slots__ = (
		"f",
		"duty",
		"cycleT",
		"out",
		"__errors",
	)

	def __init__(self, f, duty=0.0):
		self.f = max(f, 0.0)
		self.duty = max(min(duty, 1.0), 0.0)
		self.cycleT = 0.0
		self.out = False
		self.__errors = 0

	@property
	def cycleDur(self):
		return 1.0 / self.f

	def calc(self, dt):
		if self.f:
			self.cycleT += dt
			dur = self.cycleDur
			wrap = False
			if self.out:
				if self.cycleT >= dur:
					if self.duty < 1.0:
						self.out = False
					wrap = True
			else:
				if self.cycleT >= dur * (1.0 - self.duty):
					if self.duty > 0.0:
						self.out = True
					else:
						wrap = True
			if wrap:
				self.cycleT -= dur
				if self.cycleT >= dur:
					if self.__errors < 10:
						print("The PWM frequency %f Hz is "
						      "too fast. Loosing steps." % (
						      self.f))
						self.__errors += 1
					self.cycleT = 0.0
		return self.out

if __name__ == "__main__":
	import time
	pwm = PWMGenerator(f=0.5, duty=0.3)
	dt = 0.1
	i = 0.0
	while 1:
		out = pwm.calc(dt)
		print("t=%.1f, out=%d, cycleT=%.3f" % (i, int(out), pwm.cycleT))
		time.sleep(dt)
		i += dt
bues.ch cgit interface