From 73b8be746d0d3412fa0f42286c16f6482a8084e5 Mon Sep 17 00:00:00 2001
From: Chirag Kumar <34834917+Chiral13@users.noreply.github.com>
Date: Mon, 6 Nov 2023 15:04:24 +0530
Subject: [PATCH] Create coord_init.py

---
 DobotAPI/coord_init.py | 54 ++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 54 insertions(+)
 create mode 100644 DobotAPI/coord_init.py

diff --git a/DobotAPI/coord_init.py b/DobotAPI/coord_init.py
new file mode 100644
index 0000000..ee2bd10
--- /dev/null
+++ b/DobotAPI/coord_init.py
@@ -0,0 +1,54 @@
+import time
+import csv
+from dobotapi.dobot import Dobot
+from dobotapi.utils import get_coms_port
+from dobotapi.effectors.suctioncup import SuctionCup
+
+bot = Dobot()
+bot.connect()
+cup = SuctionCup(bot = bot)
+fields = ['pos','x','y','z']
+def main():
+    print('dobot connected')
+    for x in range(4):
+        print('Select option:')
+        if(input() == '1'):
+            home = bot.get_pose()
+        elif(input() == '2'):
+            PickUP = bot.get_pose()
+        elif(input() == '3'):
+            pickDown = bot.get_pose()
+        elif(input() == '4'):
+            placeUp = bot.get_pose()
+        elif(input() == '5'):
+            placeDown = bot.get_pose()
+    filename = "coords.csv"
+ 
+    mydict =[{'pos': 'home', 'x': home.position.x, 
+            'y': home.position.y, 'z': home.position.y},
+            {'pos': 'PickUP', 'x': PickUP.position.x, 
+            'y': PickUP.position.y, 'z': PickUP.position.z},
+            {'pos': 'pickDown', 'x': pickDown.position.x, 
+            'y': pickDown.position.y, 'z': pickDown.position.z},
+            {'pos': 'placeUp', 'x': placeUp.position.x, 
+            'y': placeUp.position.y, 'z': placeUp.position.z},
+            {'pos': 'placeDown', 'x': placeDown.position.x, 
+            'y': placeDown.position.y, 'z': placeDown.position.z}]
+
+# writing to csv file
+    with open(filename, 'w') as csvfile:
+        # creating a csv dict writer object
+        writer = csv.DictWriter(csvfile, fieldnames = fields)
+     
+    # writing headers (field names)
+        writer.writeheader()
+     
+    # writing data rows
+        writer.writerows(mydict)
+    time.sleep(2)
+
+try:
+    main()
+except KeyboardInterrupt:
+    bot.close()
+    print("Dobot disconnected")