Starting Plotting Templates

jonathan.hess · jonathan.hess · commit cec255783821 · 2024-06-03T16:15:43.000-07:00
diff --git a/.vscode/launch.json b/.vscode/launch.json
@@ -5,6 +5,7 @@
     "version": "0.2.0",
     "configurations": [
 
+
         {
             "name": "Python: Schematic Test",
             "type": "python",
@@ -29,6 +30,14 @@
             "console": "integratedTerminal",
             "justMyCode": true
         },
+        {
+            "name": "Python: Plotting Test",
+            "type": "python",
+            "request": "launch",
+            "program": "tests/plottingtemplatetest.py",
+            "console": "integratedTerminal",
+            "justMyCode": true
+        },
         {
             "name": "Python: Module",
             "type": "python",
diff --git a/environment.yml b/environment.yml
@@ -5,15 +5,17 @@ channels:
 dependencies:
   # Runtime
   - attrdict3=2.0.2
-  - wxPython=4.2.1
+  - wxPython=4.2.1 #GUI
   - python=3.11
   - pydantic=2.0.3
   - pandas=2.0.3
   - rich=13.4.2
-  - skillbridge=1.5.0
+  - skillbridge=1.5.0 #python-skill communication
+  - beautifulsoup4 #XML 
+  - lxml
  # Development and Testing
-  - PyInstaller=5.13.0
-  - conda=23.5.2
+  - PyInstaller=5.13.0 #build
+  - conda=23.5.2 #ENV
   - mamba=1.4.9
   - pip=23.2
   - flit=3.9.0
@@ -23,3 +25,4 @@ dependencies:
   # documentation
   - mkdocs=1.4.3
   - mkdocs-material=9.1.19
+
diff --git a/morpheus/CadenceManager.py b/morpheus/CadenceManager.py
@@ -19,18 +19,22 @@
 
 
 class cadenceManager:
-    def __init__(self,id):
+    def __init__(self,id, cwd=None, precommand= ""):
         self.timeout= 300 #5 minute timeout
-        self.log_file = os.path.join(os.getcwd(), "virtuoso_log.txt")
-        self.createNewCadence(id)
+        if(cwd is None): cwd=os.getcwd()
+        self.log_file = os.path.join(cwd, "virtuoso_log.txt")
+        self.createNewCadence(id,cwd, precommand)
         self.checkInactive()
         #self.loop()
         
-    def createNewCadence(self, id):
+    def createNewCadence(self, id,cwd, precommand):
         
         
         # Run virtuoso with input from a string
-        virtuoso_process = subprocess.Popen(["virtuoso", "-nograph"],cwd=os.getcwd(), stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
+        print("start cadence at: "+ cwd)
+        command = precommand.split() + [ ";icfb", "-nograph"]
+        #virtuoso_process = subprocess.Popen(command, capture_output=True, shell=True)
+        virtuoso_process = subprocess.Popen(command ,cwd=cwd, stdin=subprocess.PIPE, stdout=subprocess.PIPE, stderr=subprocess.PIPE,shell=True) #TODO SUPPORT BOTH VIRTUOSO AND ICFB
 
         def log_stdout():
             with open(self.log_file, 'w') as log_file:
diff --git a/morpheus/PlottingTemplate.py b/morpheus/PlottingTemplate.py
@@ -0,0 +1,119 @@
+from distutils.dir_util import copy_tree
+from bs4 import BeautifulSoup
+from lxml import objectify, etree
+
+from xml.dom import minidom 
+import os
+
+class PlottingTemplate:
+    def __init__(self,ws,lib,cell,view) -> None: #for pin terminals
+        self.ws = ws
+        self.lib = lib
+        self.cell = cell
+        self.view = view
+
+        pass
+    
+    #Import ploting template from existing maestro plotting template  
+    def import_plotting_template(self, pt_lib, pt_cell, pt_view, pTemplate):
+        
+        
+        #copy the exisitng template
+        print(self.ws['getWorkingDir']());
+        cellview = self.ws.dd.GetObj(pt_lib,pt_cell,pt_view) #get the cellview for a filelocation on disk
+        path = cellview.write_path #get filepath from cellview
+        print(path);
+        filepath  = path +"/plottingTemplates/"+ pTemplate + "/" + pTemplate+ "_0.cmpt";
+        with open(filepath, 'r') as f:
+            file = f.read()
+        soup = BeautifulSoup(bytes(file, encoding='utf8'),"xml")
+        xml_object = objectify.parse(filepath) #get_etree
+        
+        self.root = objectify.fromstring(bytes(file, encoding='utf8'))
+        attrib = self.root.attrib
+        #Now that we have the XML we should convert to (remove excess data) a YML/Object format for saving
+        self.root
+        self.soup = soup
+
+            #get Markers?
+            
+        print(soup);
+        
+
+
+    #export the loaded template to new maestro cellview
+    def export_plotting_template(self,pt_lib,pt_cell,pt_view,pTemplate):
+        
+        cellview = self.ws.dd.GetObj(pt_lib,pt_cell,pt_view) #get the cellview for a filelocation on disk
+        path = cellview.write_path #get filepath from cellview
+        #MIN DATA FOR CMPT.group file
+
+        # <?xml version="1.0" encoding="UTF-8" ?>
+        # <!--StateFileType saveLoadState-->
+        # <topLevel>
+        #     <IndexedProp Name="graphFileList">
+        #         <SimpleProp Value="Template2_0.cmpt" Type="stringValue" />
+        #     </IndexedProp>
+        # </topLevel>
+
+        # file = open(filename, 'w',encoding='utf8') 
+        # file.write(str(self.soup)) 
+        # file.close() 
+
+        filename  = path +"/plottingTemplates/"+ pTemplate + "/" + pTemplate+ "_0.cmpt";
+        os.makedirs(os.path.dirname(filename), exist_ok=True)
+        #for now just relay the template
+        file = open(filename, 'w',encoding='utf8') 
+        file.write(str(self.soup)) 
+        file.close() 
+
+        etree.tostring(self.root, pretty_print=True)
+        self.updateMaestroXML(pt_lib,pt_cell,pt_view,pTemplate)
+        
+        # root = minidom.Document()
+  
+        # xml = root.createElement('root')  
+        # root.appendChild(xml) 
+        
+        # productChild = root.createElement('product') 
+        # productChild.setAttribute('name', 'Geeks for Geeks') 
+        
+        # xml.appendChild(productChild) 
+        
+        # xml_str = root.toprettyxml(indent ="\t")  
+        
+        # save_path_file = "gfg.xml"
+
+    def updateMaestroXML(self,pt_lib,pt_cell,pt_view,pTemplate):
+        cellview = self.ws.dd.GetObj(pt_lib,pt_cell,pt_view) #get the cellview for a filelocation on disk
+        path = cellview.write_path #get filepath from cellview
+        filepath = path + "/maestro.sdb"
+        with open(filepath, 'r') as f:
+            file = f.read()
+
+        maestroXML = BeautifulSoup(bytes(file, encoding='utf8'),"xml")
+        plottingOptions = maestroXML.find_all("plottingoption")
+        #check if plotting templates are an option
+        plottingTemplates = [option for option in plottingOptions if "allplottingtemplate" in option.next] #find the plotting templates option
+
+        #TODO check if there isnt a plotting templates already
+        if(plottingTemplates is None):
+            pass
+        plottingTemplates = plottingTemplates[0]
+        #plottingTemplates.append("<")
+
+        new_plottingtemplate = maestroXML.new_tag("plottingtemplate")
+        #<templatelocation>/orpheus_tests/templatetest1/maestro/plottingTemplates/</templatelocation>
+        new_plottingtemplate.string = pTemplate
+        location =maestroXML.new_tag("templatelocation")
+        templates_filepath  = path +"/plottingTemplates/";
+        location.string = templates_filepath
+        new_plottingtemplate.append(location)
+        plottingTemplates.append(new_plottingtemplate)
+
+        print(maestroXML)
+        print(plottingOptions)
+
+        file = open(filepath, 'w') 
+        file.write(str(maestroXML)) 
+        file.close() 
diff --git a/tests/plottingtemplatetest.py b/tests/plottingtemplatetest.py
@@ -0,0 +1,39 @@
+
+
+import asyncio
+from skillbridge import Workspace
+import sys
+import os.path
+parentddir = os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir))
+sys.path.append(parentddir)
+from morpheus.Schematic import schematic
+from morpheus import Config
+#sys.path.append(S
+#    os.path.abspath(os.path.join(os.path.dirname(__file__), os.path.pardir)))
+from morpheus import *
+from morpheus.PlottingTemplate import PlottingTemplate
+
+
+from morpheus.CadenceManager import cadenceManager
+
+
+
+id = "test"
+lib = "morpheus_tests"
+cell = "templatetest2"
+view = "maestro"
+template = "Template1"
+
+manager = cadenceManager(id) #, cwd = "", precommand="")
+ws = Workspace.open(id)
+
+
+pt = PlottingTemplate(ws,lib,cell,view) #setup template2 as the maestro to edit
+
+
+cell = "templatetest1" #copy from template1
+
+pt.import_plotting_template(lib,cell,view,template)
+template = "Template2" 
+pt.export_plotting_template(lib,cell,view,template) #add same template to same cell
+#manager.killCadence()py
diff --git a/virtuoso_log.txt b/virtuoso_log.txt
