initial commit

2019-01-12 22:39:19 +01:00
parent b13a339717
commit 59cd11de7b
10 changed files with 32041 additions and 0 deletions
--- a/GcodeSplitterTest.py
+++ b/GcodeSplitterTest.py
@@ -0,0 +1,96 @@
+
+import unittest
+from GcodeFileSplitter import GcodeFileAnalyzer
+
+dummyGcode = [
+    'header01', 'header02',
+    ';LAYER_COUNT:4',
+    ';LAYER:0', 'G0 Z0.3', 'G1 E0', ';TIME_ELAPSED:1.11',
+    ';LAYER:1', 'M106 S255', 'G0 Z0.5', 'G1 E11.11', ';TIME_ELAPSED:2.11',
+    ';LAYER:2', 'G0 Z0.7', 'G1 E22.22', ';TIME_ELAPSED:3.11',
+    ';LAYER:3', 'G0 Z0.9', 'G1 E33.33', ';TIME_ELAPSED:4.11',
+    'G1 F1500 E30', 'footer02',
+]
+
+class GcodeTest(unittest.TestCase):
+    def test_Analyzer(self):
+        gcode = GcodeFileAnalyzer(dummyGcode)
+        self.assertEqual(gcode.header, ['header01', 'header02', ';LAYER_COUNT:4'])
+        self.assertEqual(gcode.footer, ['G1 F1500 E30', 'footer02'])
+        self.assertEqual(gcode.layers[0].lines, [';LAYER:0', 'G0 Z0.3', 'G1 E0', ';TIME_ELAPSED:1.11',])
+        self.assertEqual(gcode.layers[1].lines, [';LAYER:1', 'M106 S255', 'G0 Z0.5', 'G1 E11.11', ';TIME_ELAPSED:2.11',])
+        self.assertEqual(gcode.layers[2].lines, [';LAYER:2', 'G0 Z0.7', 'G1 E22.22', ';TIME_ELAPSED:3.11',])
+        self.assertEqual(gcode.layers[3].lines, [';LAYER:3', 'G0 Z0.9', 'G1 E33.33', ';TIME_ELAPSED:4.11',])
+        self.assertEqual(gcode.layerCount, 4)
+        self.assertEqual(gcode.fanSpeedCode, 'M106 S255')
+
+    def test_SplitInTwo(self):
+        gcode = GcodeFileAnalyzer(dummyGcode)
+        lines1, lines2 = gcode.ExportSplitted([2])
+
+        # Check first file
+        # header + layer 1 and 2
+        self.assertEqual(lines1[:lines1.index(';TIME_ELAPSED:2.11')], 
+                         dummyGcode[:dummyGcode.index(';TIME_ELAPSED:2.11')])
+        # footer
+        self.assertEqual(lines1[-6:], 
+                         ['', ';PATCHED_FOOTER', 'G1 F1500 E4.610', '']
+                         + ['; ' + dummyGcode[-2]] + dummyGcode[-1:])
+
+        # Check second file
+        # header
+        self.assertEqual(lines2[:lines2.index(';LAYER_COUNT:4')], 
+                         dummyGcode[:dummyGcode.index(';LAYER_COUNT:4')])
+        # extra init code
+        self.assertEqual(lines2[lines2.index(';LAYER_COUNT:4') + 1:lines2.index(';LAYER:2')], 
+                         [
+                             '', ';GENERATED_EXTRA_INIT', 
+                             'M106 S255', 'G0 Z1.700', 'G92 E12.220', ''
+                         ])
+        # layer 3 and 4 + footer
+        self.assertEqual(lines2[lines2.index(';LAYER:2'):], 
+                         dummyGcode[dummyGcode.index(';LAYER:2'):-2]
+                         + ['G1 F1500 E30', 'footer02'])
+
+    def test_SplitInThree(self):
+        gcode = GcodeFileAnalyzer(dummyGcode)
+        lines1, lines2, lines3 = gcode.ExportSplitted([2, 3])
+
+        # Check first file
+        # header + layer 1 and 2
+        self.assertEqual(lines1[:lines1.index(';TIME_ELAPSED:2.11')], 
+                         dummyGcode[:dummyGcode.index(';TIME_ELAPSED:2.11')])
+        # footer
+        self.assertEqual(lines1[-6:],
+                         ['', ';PATCHED_FOOTER', 'G1 F1500 E4.610', '']
+                         + ['; ' + dummyGcode[-2]] + dummyGcode[-1:])
+
+        # Check second file
+        # header
+        self.assertEqual(lines2[:lines2.index(';LAYER_COUNT:4')], 
+                         dummyGcode[:dummyGcode.index(';LAYER_COUNT:4')])
+        # extra init code
+        self.assertEqual(lines2[lines2.index(';LAYER_COUNT:4') + 1:lines2.index(';LAYER:2')], 
+                         [
+                             '', ';GENERATED_EXTRA_INIT', 
+                             'M106 S255', 'G0 Z1.700', 'G92 E12.220', ''
+                         ])
+        # layer 3 
+        self.assertEqual(lines2[lines2.index(';LAYER:2'):], 
+                         dummyGcode[dummyGcode.index(';LAYER:2'):dummyGcode.index(';LAYER:3')]
+                         + ['', ';PATCHED_FOOTER', 'G1 F1500 E15.720', '']
+                         + ['; G1 F1500 E30', 'footer02'])
+
+        # Check third file
+        # header
+        self.assertEqual(lines3[:lines3.index(';LAYER_COUNT:4')], 
+                         dummyGcode[:dummyGcode.index(';LAYER_COUNT:4')])
+        # extra init code
+        self.assertEqual(lines3[lines3.index(';LAYER_COUNT:4') + 1:lines3.index(';LAYER:3')], 
+                         [
+                             '', ';GENERATED_EXTRA_INIT', 
+                             'M106 S255', 'G0 Z1.900', 'G92 E23.330', ''
+                         ])
+        # layer 3 
+        self.assertEqual(lines3[lines3.index(';LAYER:3'):], 
+                         dummyGcode[dummyGcode.index(';LAYER:3'):])