Subject: Completed Python Plotting

Description: Completed Python Plotting

1 files changed, 373 insertions, 346 deletions

diff --git a/src/ngspiceSimulation/pythonPlotting.py b/src/ngspiceSimulation/pythonPlotting.py
index fd98a0c8..0bca22be 100644
--- a/src/ngspiceSimulation/pythonPlotting.py
+++ b/src/ngspiceSimulation/pythonPlotting.py
@@ -1,54 +1,59 @@
 from __future__ import division         # Used for decimal division eg 2/3=0.66 and not '0' 6/2=3.0 and 6//2=3
+import os
 from PyQt4 import QtGui, QtCore
 from decimal import Decimal
 from matplotlib.backends.backend_qt4agg import FigureCanvasQTAgg as FigureCanvas
 from matplotlib.backends.backend_qt4agg import NavigationToolbar2QTAgg as NavigationToolbar
 from matplotlib.figure import Figure
-import tkMessageBox
-
 
 class plotWindow(QtGui.QMainWindow):
     def __init__(self,fpath,projectName):
         QtGui.QMainWindow.__init__(self)
-        self.fpath = fpath
-        self.projName = projectName
-        self.createMainFrame()
+        self.fpath = fpath#+".cir.out"
+        self.projectName = projectName
+        print "Path : ",self.fpath
+        print "Project Name : ",self.projectName
         self.combo = []
         self.combo1 = []
         self.combo1_rev = []
+        #Creating Frame
+        self.createMainFrame()        
         
     def createMainFrame(self):
-        self.main_frame = QtGui.QWidget()
+        self.mainFrame = QtGui.QWidget()
         self.dpi = 100
-        #Creating Figure Canvas
         self.fig = Figure((7.0, 7.0), dpi=self.dpi)
+        #Creating Canvas which will figure
         self.canvas = FigureCanvas(self.fig)
-        self.canvas.setParent(self.main_frame)
-        self.canvas.setSizePolicy(QtGui.QSizePolicy.Expanding,QtGui.QSizePolicy.Expanding)
-        self.canvas.updateGeometry()
-        
+        self.canvas.setParent(self.mainFrame)
         self.axes = self.fig.add_subplot(111)
-        self.mpl_toolbar = NavigationToolbar(self.canvas, self.main_frame)
-    
-        left_vbox = QtGui.QVBoxLayout()
-        left_vbox.addWidget(self.mpl_toolbar)
-        left_vbox.addWidget(self.canvas)
-
-        right_vbox = QtGui.QVBoxLayout()
-        right_grid = QtGui.QGridLayout()
-        top_grid = QtGui.QGridLayout()
-        
-        ##Processing data file to extract data in proper format
-        self.fobj = File_data()
-        plot_type = self.fobj.openFile(self.fpath)
-        print "Plot Type :",plot_type
-        self.fobj.computeAxes()
+        self.navToolBar = NavigationToolbar(self.canvas, self.mainFrame)
+        
+        #LeftVbox hold navigation tool bar and canvas
+        self.left_vbox = QtGui.QVBoxLayout()
+        self.left_vbox.addWidget(self.navToolBar)
+        self.left_vbox.addWidget(self.canvas)
+
+        #right VBOX is main Layout which hold right grid(bottom part) and top grid(top part)
+        self.right_vbox = QtGui.QVBoxLayout()
+        self.right_grid = QtGui.QGridLayout()
+        self.top_grid = QtGui.QGridLayout()
+        
+        #Get DataExtraction Details
+        self.obj_dataext = DataExtraction()
+        self.plotType = self.obj_dataext.openFile(self.fpath)
+        
+        self.obj_dataext.computeAxes()
+        self.a = self.obj_dataext.numVals()
+                
+        #Hard coded to be remove
+        #self.a = [4,2]
+        #self.plotType = [1,0]
+        
+        
         self.chkbox=[]
-        self.a = self.fobj.numVals()
-
-        print "A :",self.a
         
-        ########### Generating list of colours :
+        ########### Generating list of colors :
         self.full_colors = ['r','b','g','y','c','m','k']#,(0.4,0.5,0.2),(0.1,0.4,0.9),(0.4,0.9,0.2),(0.9,0.4,0.9)]
         self.color = []
         for i in range(0,self.a[0]-1):
@@ -67,171 +72,171 @@ class plotWindow(QtGui.QMainWindow):
             elif (i-6)%7 == 0:
                 self.color.append(self.full_colors[6])
 
-        ###########
-                
+        ###########Color generation ends here
+        
+        
+        #Total number of voltage source
         self.volts_length = self.a[1]
-        #print "I'm Volts length:",self.volts_length
-        self.heading1 = QtGui.QLabel()
-        top_grid.addWidget(self.heading1,1,0)
-        self.heading2 = QtGui.QLabel()
-        top_grid.addWidget(self.heading2,self.a[1]+2,0)
+        self.analysisType = QtGui.QLabel()
+        self.top_grid.addWidget(self.analysisType,0,0)
+        self.listNode = QtGui.QLabel()
+        self.top_grid.addWidget(self.listNode,1,0)
+        self.listBranch = QtGui.QLabel()
+        self.top_grid.addWidget(self.listBranch,self.a[1]+2,0)
         for i in range(0,self.a[1]):#a[0]-1
-            self.chkbox.append(QtGui.QCheckBox(self.fobj.butnames[i]))
-            self.chkbox[i].setToolTip('<b>Tick Me!</b>' )
-            top_grid.addWidget(self.chkbox[i],i+2,0)
+            self.chkbox.append(QtGui.QCheckBox(self.obj_dataext.NBList[i]))
+            self.chkbox[i].setToolTip('<b>Check To Plot</b>' )
+            self.top_grid.addWidget(self.chkbox[i],i+2,0)
 
         for i in range(self.a[1],self.a[0]-1):#a[0]-1
-            self.chkbox.append(QtGui.QCheckBox(self.fobj.butnames[i]))
-            self.chkbox[i].setToolTip('<b>Tick Me!</b>' )
-            top_grid.addWidget(self.chkbox[i],i+3,0)
+            self.chkbox.append(QtGui.QCheckBox(self.obj_dataext.NBList[i]))
+            self.chkbox[i].setToolTip('<b>Check To Plot</b>' )
+            self.top_grid.addWidget(self.chkbox[i],i+3,0)
     
         self.clear = QtGui.QPushButton("Clear")
-        self.Note = QtGui.QLabel()
-        self.Note1 = QtGui.QLabel()
-        self.Note2 = QtGui.QLabel()
+        self.warnning = QtGui.QLabel()
+        self.funcName = QtGui.QLabel()
+        self.funcExample = QtGui.QLabel()
     
-        self.btn = QtGui.QPushButton("Plot")
-        self.btn.setToolTip('<b>Press</b> to Plot' )
+        self.plotbtn = QtGui.QPushButton("Plot")
+        self.plotbtn.setToolTip('<b>Press</b> to Plot' )
         self.text = QtGui.QLineEdit()
         self.funcLabel = QtGui.QLabel()
         self.palette1 = QtGui.QPalette()
         self.palette2 = QtGui.QPalette()
-        self.btn1 = QtGui.QPushButton("Plot Function")
-        self.btn1.setToolTip('<b>Press</b> to Plot the function' )
+        self.plotfuncbtn = QtGui.QPushButton("Plot Function")
+        self.plotfuncbtn.setToolTip('<b>Press</b> to Plot the function' )
 
         self.palette1.setColor(QtGui.QPalette.Foreground,QtCore.Qt.blue)
         self.palette2.setColor(QtGui.QPalette.Foreground,QtCore.Qt.red)
-        self.Note1.setPalette(self.palette1)
-        self.Note2.setPalette(self.palette2)
+        self.funcName.setPalette(self.palette1)
+        self.funcExample.setPalette(self.palette2)
 
-        right_vbox.addLayout(top_grid)
-        right_vbox.addWidget(self.btn)
+        self.right_vbox.addLayout(self.top_grid)
+        self.right_vbox.addWidget(self.plotbtn)
     
-        right_grid.addWidget(self.funcLabel,1,0)
-        right_grid.addWidget(self.text,1,1)
-        right_grid.addWidget(self.btn1,2,1)    
-        right_grid.addWidget(self.clear,2,0)
-        right_grid.addWidget(self.Note,3,0)
-        right_grid.addWidget(self.Note1,4,0)
-        right_grid.addWidget(self.Note2,4,1)
+        self.right_grid.addWidget(self.funcLabel,1,0)
+        self.right_grid.addWidget(self.text,1,1)
+        self.right_grid.addWidget(self.plotfuncbtn,2,1)    
+        self.right_grid.addWidget(self.clear,2,0)
+        self.right_grid.addWidget(self.warnning,3,0)
+        self.right_grid.addWidget(self.funcName,4,0)
+        self.right_grid.addWidget(self.funcExample,4,1)
+        self.right_vbox.addLayout(self.right_grid)
+        
+        self.hbox = QtGui.QHBoxLayout()
+        self.hbox.addLayout(self.left_vbox)
+        self.hbox.addLayout(self.right_vbox)
     
-        right_vbox.addLayout(right_grid)
-        hbox = QtGui.QHBoxLayout()
-        hbox.addLayout(left_vbox)
-        hbox.addLayout(right_vbox)
-
-        widget = QtGui.QWidget()
-        widget.setLayout(hbox)#finalvbox
+        self.widget = QtGui.QWidget()
+        self.widget.setLayout(self.hbox)#finalvbox
         self.scrollArea = QtGui.QScrollArea()
         self.scrollArea.setWidgetResizable(True)
-        self.scrollArea.setWidget(widget)
+        self.scrollArea.setWidget(self.widget)
 
-        finalhbox = QtGui.QHBoxLayout()
-        finalhbox.addWidget(self.scrollArea)
-        self.main_frame.setLayout(finalhbox)
+        self.finalhbox = QtGui.QHBoxLayout()
+        self.finalhbox.addWidget(self.scrollArea)
 
+        self.mainFrame.setLayout(self.finalhbox)
         self.showMaximized()
     
-        self.heading1.setText("<font color='indigo'>List of Nodes:</font>")
-
-        self.heading2.setText("<font color='indigo'>List of Branches:</font>")    
-
+        self.listNode.setText("<font color='indigo'>List of Nodes:</font>")
+        self.listBranch.setText("<font color='indigo'>List of Branches:</font>")    
         self.funcLabel.setText("<font color='indigo'>Function:</font>")
-    
-        self.Note1.setText("<font color='indigo'>Examples:</font>\
-        <br><br>Addition:<br>Subtraction:<br>Multiplication:<br>Division:<br>Comparison:")
-
-        self.Note2.setText("\n\n"+self.fobj.butnames[0]+" + "+self.fobj.butnames[1]+"\n"+self.fobj.butnames[0]+" - "+self.fobj.butnames[1]+ \
-                           "\n"+self.fobj.butnames[0]+" * "+self.fobj.butnames[1]+"\n"+self.fobj.butnames[0]+" / "+self.fobj.butnames[1]+ \
-                           "\n"+self.fobj.butnames[0]+" vs "+self.fobj.butnames[1])
-    
+        self.funcName.setText("<font color='indigo'>Examples:</font>\
+                <br><br>Addition:<br>Subtraction:<br>Multiplication:<br>Division:<br>Comparison:")
+        self.funcExample.setText("\n\nV(1) + V(2)\nV(1) - V(2)\nV(1) * V(2)\nV(1) / V(2)\nV(1) vs V(2)")
 
+        #Connecting to plot and clear function
         self.connect(self.clear,QtCore.SIGNAL('clicked()'),self.pushedClear)
+        self.connect(self.plotfuncbtn,QtCore.SIGNAL('clicked()'), self.pushedPlotFunc)
 
-        self.connect(self.btn1,QtCore.SIGNAL('clicked()'), self.pushedPlotFunc)
-
-        if plot_type[0]==0:
-            self.setWindowTitle('AC Analysis')
-            if plot_type[1]==1:
-                self.connect(self.btn,QtCore.SIGNAL('clicked()'), self.onPush_decade)
+        if self.plotType[0]==0:
+            self.analysisType.setText("<b>AC Analysis</b>")
+            if self.plotType[1]==1:
+                self.connect(self.plotbtn, QtCore.SIGNAL('clicked()'), self.onPush_decade)
             else:
-                self.connect(self.btn,QtCore.SIGNAL('clicked()'), self.onPush_ac)
+                self.connect(self.plotbtn, QtCore.SIGNAL('clicked()'), self.onPush_ac)
 
-        elif plot_type[0]==1:
+        elif self.plotType[0]==1:
+            self.analysisType.setText("<b>Transient Analysis</b>")
+            self.connect(self.plotbtn, QtCore.SIGNAL('clicked()'), self.onPush_trans)
 
-            self.setWindowTitle('Transient Analysis')
-            self.connect(self.btn,QtCore.SIGNAL('clicked()'), self.onPush_trans)
         else:
-            self.setWindowTitle('DC Analysis')
-            self.connect(self.btn,QtCore.SIGNAL('clicked()'), self.onPush_dc)
-        
-        self.setCentralWidget(self.main_frame)    
-    
+            self.analysisType.setText("<b>DC Analysis</b>")
+            self.connect(self.plotbtn, QtCore.SIGNAL('clicked()'), self.onPush_dc)
 
+        self.setCentralWidget(self.mainFrame)    
+        
+    def pushedClear(self):
+        #print "Calling Clear Canvas function"
+        self.text.clear()
+        self.axes.cla()
+        self.canvas.draw()
+        QtCore.SLOT('quit()')
+        
     def pushedPlotFunc(self):
+        #print "Calling Plot function"
         self.parts = str(self.text.text())
         self.parts = self.parts.split(" ")
-        #print self.parts
+        #print "Parts :",self.parts
+        
         if self.parts[len(self.parts)-1] == '':
             self.parts = self.parts[0:-1]
-        #print self.parts
+        
         self.values = self.parts
         self.comboAll = []
-        self.axes.cla()
-        plot_type2 = self.fobj.openFile(self.fpath)
-
+        self.axes.cla()   
+        
+        self.plotType2 = self.obj_dataext.openFile(self.fpath)
+        
         if len(self.parts) <= 2:
-            self.Note.setText("Too few arguments!\nRefer syntax below!")
+            self.warnning.setText("Too few arguments!\nRefer syntax below!")
             QtGui.QMessageBox.about(self, "Warning!!", "Too Few Arguments/SYNTAX Error!\n Refer Examples")
         else:
-            self.Note.setText("")
-
-        a = []
+            self.warnning.setText("")
+            
+        a = []  
         finalResult  = []
         p = 0
-        #print "values:",self.values
-        #print "parts:",self.parts
-
+        
         for i in range(len(self.parts)):
-            #print "hello"
+            #print "I",i
             if i%2 == 0:
                 #print "I'm in:"
-                for j in range(len(self.fobj.butnames)):
-                    if self.parts[i]==self.fobj.butnames[j]:
-                        #print "I got you:",i
+                for j in range(len(self.obj_dataext.NBList)):
+                    if self.parts[i]==self.obj_dataext.NBList[j]:
+                        #print "I got you:",self.parts[i]
                         a.append(j)
-
-        #print "positions",a
-
+    
         if len(a) != len(self.parts)//2 + 1:
-            QtGui.QMessageBox.about(self, "Warning!!", "One of the operands doesn't belong to the above list!!")        
-
+            QtGui.QMessageBox.about(self, "Warning!!", "One of the operands doesn't belong to the above list of Nodes!!")
+            
         for i in a:
-            self.comboAll.append(self.fobj.y[i])
-
-        #print self.comboAll        
-
+            self.comboAll.append(self.obj_dataext.y[i])
+            
+        #print "ComboAll : ",self.comboAll
+        
         for i in range(len(a)):
-            if a[i] == len(self.fobj.butnames):
+            
+            if a[i] == len(self.obj_dataext.NBList):
                 QtGui.QMessageBox.about(self, "Warning!!", "One of the operands doesn't belong to the above list!!")
-                self.Note.setText("<font color='red'>To Err Is Human!<br>One of the operands doesn't belong to the above list!!</font>")
-
+                self.warnning.setText("<font color='red'>To Err Is Human!<br>One of the operands doesn't belong to the above list!!</font>")
+            
         if self.parts[1] == 'vs':
             if len(self.parts) > 3:
-                self.Note.setText("Enter two operands only!!")
-                QtGui.QMessageBox.about(self, "Warning!!", "Re-check the expression syntax!")
+                self.warnning.setText("Enter two operands only!!")
+                QtGui.QMessageBox.about(self, "Warning!!", "Recheck the expression syntax!")
+
             else:
                 self.axes.cla()
-                #print "plotting wait"
-                for i in range(len(self.fobj.y[a[0]])):
-                    self.combo.append(self.fobj.y[a[0]][i]) 
-                    self.combo1.append(self.fobj.y[a[1]][i])
             
-                '''for i in reversed(self.combo1):
-                    self.combo1_rev.append(i)'''     
-                #print self.combo
-                #print "\ncombo1_rev\n",self.combo1_rev
+                for i in range(len(self.obj_dataext.y[a[0]])):
+                    self.combo.append(self.obj_dataext.y[a[0]][i]) 
+                    self.combo1.append(self.obj_dataext.y[a[1]][i])
+                
                 self.axes.plot(self.combo,self.combo1,c=self.color[1],label=str(2))#_rev
+            
                 if max(a) < self.volts_length:
                     self.axes.set_ylabel('Voltage(V)-->')
                     self.axes.set_xlabel('Voltage(V)-->')
@@ -239,202 +244,215 @@ class plotWindow(QtGui.QMainWindow):
                     self.axes.set_ylabel('Current(I)-->')
                     self.axes.set_ylabel('Current(I)-->')        
 
-
         elif max(a) >= self.volts_length and min(a) < self.volts_length:
             QtGui.QMessageBox.about(self, "Warning!!", "Do not combine Voltage and Current!!")
 
         else:
-
             for j in range(len(self.comboAll[0])):
                 for i in range(len(self.values)):
                     if i%2==0:
                         self.values[i] = str(self.comboAll[i//2][j])
                         re = " ".join(self.values[:])
-                    #print re
-            try:
-                finalResult.append(eval(re))
-            except ArithmeticError:
-                QtGui.QMessageBox.about(self, "Warning!!", "Dividing by zero!!")
-        ############################################
-            if plot_type2[0]==0:
-                self.setWindowTitle('AC Analysis')
-                if plot_type2[1]==1:
-                    self.axes.semilogx(self.fobj.x,finalResult,c=self.color[0],label=str(1))
-        
+                try:
+                    finalResult.append(eval(re))
+                except ArithmeticError:
+                    QtGui.QMessageBox.about(self, "Warning!!", "Dividing by zero!!")
+                    
+            if self.plotType2[0]==0:
+                #self.setWindowTitle('AC Analysis')
+                if self.plotType2[1]==1:
+                    self.axes.semilogx(self.obj_dataext.x,finalResult,c=self.color[0],label=str(1))
                 else:
-    
-                    self.axes.plot(self.fobj.x,finalResult,c=self.color[0],label=str(1))        
+                    self.axes.plot(self.obj_dataext.x,finalResult,c=self.color[0],label=str(1))
+                    
                 self.axes.set_xlabel('freq-->')
+                
                 if max(a) < self.volts_length:
                     self.axes.set_ylabel('Voltage(V)-->')
                 else:
-                    self.axes.set_ylabel('Current(I)-->')        
-
-
-            elif plot_type2[0]==1:
-                self.setWindowTitle('Transient Analysis')
-                self.axes.plot(self.fobj.x,finalResult,c=self.color[0],label=str(1))    
+                    self.axes.set_ylabel('Current(I)-->')
+                    
+            elif self.plotType2[0]==1:
+                #self.setWindowTitle('Transient Analysis')
+                self.axes.plot(self.obj_dataext.x,finalResult,c=self.color[0],label=str(1))    
                 self.axes.set_xlabel('time-->')
                 if max(a) < self.volts_length:
                     self.axes.set_ylabel('Voltage(V)-->')
                 else:
-                    self.axes.set_ylabel('Current(I)-->')        
-    
-
+                    self.axes.set_ylabel('Current(I)-->')
+                    
             else:
-
-                self.setWindowTitle('DC Analysis')
-                self.axes.plot(self.fobj.x,finalResult,c=self.color[0],label=str(1))
+                #self.setWindowTitle('DC Analysis')
+                self.axes.plot(self.obj_dataext.x,finalResult,c=self.color[0],label=str(1))
                 self.axes.set_xlabel('I/P Voltage-->')
                 if max(a) < self.volts_length:
                     self.axes.set_ylabel('Voltage(V)-->')
                 else:
-                    self.axes.set_ylabel('Current(I)-->')        
-        
-               
-
-
+                    self.axes.set_ylabel('Current(I)-->')
+                    
+                    
         self.axes.grid(True)
         self.canvas.draw()
         self.combo = []
         self.combo1 = []
         self.combo1_rev = []
-        #print "succes:",self.parts
-
-    def pushedClear(self):
-
-        self.text.clear()
+            
+            
+       
+    
+    def onPush_decade(self):
+        #print "Calling on push Decade"
+        boxCheck = 0
         self.axes.cla()
-        self.canvas.draw()
-        QtCore.SLOT('quit()')
 
-    def onPush_ac(self):
-        self.axes.cla()
-        boxCheck = 0
         for i,j in zip(self.chkbox,range(len(self.chkbox))):
             if i.isChecked():
                 boxCheck += 1
-                self.axes.plot(self.fobj.x,self.fobj.y[j],c=self.color[j],label=str(j+1))
+                self.axes.semilogx(self.obj_dataext.x,self.obj_dataext.y[j],c=self.color[j],label=str(j+1))
                 self.axes.set_xlabel('freq-->')
                 if j < self.volts_length:
                     self.axes.set_ylabel('Voltage(V)-->')
                 else:
                     self.axes.set_ylabel('Current(I)-->')
+        
                 self.axes.grid(True)
         if boxCheck == 0:
-            QtGui.QMessageBox.about(self, "Warning!!","Please select atleast one Node OR Branch")
+            QtGui.QMessageBox.about(self, "Warning!!","Please select at least one Node OR Branch")        
         self.canvas.draw()
-                     
 
-    def onPush_decade(self):
-        boxCheck = 0
+        
+    def onPush_ac(self):
+        #print "Calling on push ac"
         self.axes.cla()
-
+        boxCheck = 0
         for i,j in zip(self.chkbox,range(len(self.chkbox))):
             if i.isChecked():
                 boxCheck += 1
-                self.axes.semilogx(self.fobj.x,self.fobj.y[j],c=self.color[j],label=str(j+1))
+                self.axes.plot(self.obj_dataext.x,self.obj_dataext.y[j],c=self.color[j],label=str(j+1))
+                self.axes.set_xlabel('freq-->')
                 if j < self.volts_length:
                     self.axes.set_ylabel('Voltage(V)-->')
                 else:
                     self.axes.set_ylabel('Current(I)-->')
-                self.axes.set_xlabel('freq-->')
                 self.axes.grid(True)
         if boxCheck == 0:
-            QtGui.QMessageBox.about(self, "Warning!!","Please select atleast one Node OR Branch")        
+            QtGui.QMessageBox.about(self, "Warning!!","Please select at least one Node OR Branch")
         self.canvas.draw()
-
+        
     def onPush_trans(self):
-        boxCheck = 0
+        #print "Calling on push trans"
         self.axes.cla()
+        boxCheck = 0
         for i,j in zip(self.chkbox,range(len(self.chkbox))):
             if i.isChecked():
                 boxCheck += 1
-                #print self.fobj.y[j]
-                self.axes.plot(self.fobj.x,self.fobj.y[j],c=self.color[j],label=str(j+1))
+                self.axes.plot(self.obj_dataext.x,self.obj_dataext.y[j],c=self.color[j],label=str(j+1))
                 self.axes.set_xlabel('time-->')
-
-            if j < self.volts_length:
-                self.axes.set_ylabel('Voltage(V)-->')
-
-            else:
-                self.axes.set_ylabel('Current(I)-->')
-            self.axes.grid(True)
+                if j < self.volts_length:
+                    self.axes.set_ylabel('Voltage(V)-->')
+                else:
+                    self.axes.set_ylabel('Current(I)-->')
+                self.axes.grid(True)
         if boxCheck == 0:
-            QtGui.QMessageBox.about(self,"Warning!!", "Please select atleast one Node OR Branch")
+            QtGui.QMessageBox.about(self, "Warning!!","Please select at least one Node OR Branch")
         self.canvas.draw()
-
+        
+        
     def onPush_dc(self):
+        #print "Calling on push dc"
         boxCheck = 0
         self.axes.cla()    
         for i,j in zip(self.chkbox,range(len(self.chkbox))):
             if i.isChecked():
                 boxCheck += 1
-                self.axes.plot(self.fobj.x,self.fobj.y[j],c=self.color[j],label=str(j+1))
+                self.axes.plot(self.obj_dataext.x,self.obj_dataext.y[j],c=self.color[j],label=str(j+1))
                 self.axes.set_xlabel('Voltage Sweep(V)-->')
-
-            if j < self.volts_length:
-                self.axes.set_ylabel('Voltage(V)-->')
-            else:
-                self.axes.set_ylabel('Current(I)-->')
-            self.axes.grid(True)
+                
+                if j < self.volts_length:
+                    self.axes.set_ylabel('Voltage(V)-->')
+                else:
+                    self.axes.set_ylabel('Current(I)-->')
+                self.axes.grid(True)
         if boxCheck == 0:
             QtGui.QMessageBox.about(self,"Warning!!", "Please select atleast one Node OR Branch")
         self.canvas.draw() 
+
         
-class File_data:
-    def __init__(self,parent=None):
-        self.data=[]        #consists of all the columns of data belonging to nodes and branches
+    
+    
+
+class DataExtraction:
+    def __init__(self):
+        print "Initialization"
+        self.data=[]     #consists of all the columns of data belonging to nodes and branches
         self.y=[]        #stores y-axis data
         self.x=[]        #stores x-axis data
-
+                
+        
     def numberFinder(self,fpath):
-        """
-        New function for finding no of points to be plotted
-        """
-        with open (fpath+"/analysis") as f3:
-            info = f3.read()
-            
-        with open (fpath+"/plot_data_v.txt") as f2:
-            ilines = f2.read()
-            
+        #Opening ANalysis file
+        with open(os.path.join(fpath,"analysis")) as f3:
+            self.analysisInfo = f3.read()
+        self.analysisInfo = self.analysisInfo.split(" ")
+        
+        #print "Analysis :",analysisInfo
+        
+        #Reading data file for voltage
+        with open(os.path.join(fpath,"plot_data_v.txt")) as f2:
+            self.voltData = f2.read()
+        
+        self.voltData = self.voltData.split("\n")
+        
+        #Initializing variable 
+        #'p' gives no. of lines of data for each node/branch 
+        # 'l' gives the no of partitions for a single voltage node
+        #'vnumber' gives total number of voltage
+        #'inumber' gives total number of current   
+        
         p = l = vnumber = inumber = 0
-        ilines = ilines.split("\n")
+        #print "VoltsData : ",self.voltData
         
-        for i in ilines[3:]:
-            if "V(" in i or "x1" in i or "u3" in i:    #it has possible names of voltage nodes in ngspice
+        #Finding totla number of voltage node
+        for i in self.voltData[3:]:
+            #it has possible names of voltage nodes in NgSpice
+            if "V(" in i or "x1" in i or "u3" in i:
                 vnumber+=1    
-                
-        # for finding no of branches:
-        with open (fpath+"/plot_data_i.txt") as f2:
-            current = f2.read()
         
-        current = current.split("\n")
+        #print "Voltage Number :",vnumber
         
-        for i in current[3:]:
-            if "#branch" in i:
-                inumber+=1    
-                #print "current no:",inumber
+        #Reading Current Source Data
+        with open (os.path.join(fpath,"plot_data_i.txt")) as f1:
+            self.currentData = f1.read()
+        self.currentData = self.currentData.split("\n")  
+        
+        #print "CurrentData : ",self.currentData  
     
-        dec = 0 
+        #Finding Number of Branch
+        for i in self.currentData[3:]:
+            if "#branch" in i:
+                inumber+=1
+                
+        #print "Current Number :",inumber    
         
-        # For AC:
-        if info[0][-3:]==".ac":
-            if "dec" in info:
-                dec = 1
-
-            for i in ilines[3:]:
+        self.dec = 0 
+        
+        #For AC
+        if self.analysisInfo[0][-3:]==".ac":
+            self.analysisType = 0
+            if "dec" in self.analysisInfo:
+                self.dec = 1
+        
+            for i in self.voltData[3:]:
                 p+=1            #'p' gives no. of lines of data for each node/branch 
-                if "Index" in i:      
+                if "Index" in i:
                     l+=1        # 'l' gives the no of partitions for a single voltage node
                 #print "l:",l
                 if "AC" in i:    #DC for dc files and AC for ac ones
                     break
-            analysis_type = 0
-
-        elif ".tran" in info:
-            analysis_type = 1
-            for i in ilines[3:]:
+        
+        elif ".tran" in self.analysisInfo:
+            self.analysisType = 1
+            for i in self.voltData[3:]:
                 p+=1
                 if "Index" in i:      
                     l+=1        # 'l' gives the no of partitions for a single voltage node
@@ -445,106 +463,111 @@ class File_data:
 
         # For DC:
         else:
-            for i in ilines[3:]:
+            self.analysisType = 2
+            for i in self.voltData[3:]:
                 p+=1
                 if "Index" in i:      
                     l+=1        # 'l' gives the no of partitions for a single voltage node
                 #print "l:",l
                 if "DC" in i:    #DC for dc files and AC for ac ones
                     break
-            analysis_type = 2
-            
-        #if ac!=1:
+        
+
+
+             
+        #print "VoltNumber",vnumber
+        #print "CurrentNumber",inumber
         vnumber = vnumber//l        #vnumber gives the no of voltage nodes
         inumber  = inumber//l        #inumber gives the no of branches
-        #print "i'm p:",p
-        p=[p,vnumber,analysis_type,dec,inumber]
-        #print p
+        
+        #print "VoltNumber",vnumber
+        #print "CurrentNumber",inumber
+        
+        p=[p,vnumber,self.analysisType,self.dec,inumber]
+        
         return p
     
-    
     def openFile(self,fpath):
-        # For Current:
+        #print "Calling Open File"
+        
         try:
-            with open (fpath+"/plot_data_i.txt") as f2:       #Checking whether the files Plot_data_i.txt
-                I = f2.read()                    #  and plot_data_v.txt are present or not
-            I = I.split("\n")
-            self.butnamesi = []
+            with open (os.path.join(fpath,"plot_data_i.txt")) as f2:
+                alli = f2.read()
+                
+            alli = alli.split("\n")
+            self.NBIList = []
 
-            with open (fpath+"/plot_data_v.txt") as f1:
-                idata = f1.read()
+            with open (os.path.join(fpath,"plot_data_v.txt")) as f1:
+                allv = f1.read()
 
         except Exception as e:
-            tkMessageBox.showinfo("Warning!!", "Click on KI->Ng button before simulation ")
-            print "Excpetion MSG :",str(e)
-            #exit(1)
-    
+            print "Exception Message : ",str(e)
+            self.msg = QtGui.QErrorMessage(None)
+            self.msg.showMessage('Unable to open plot data files.')
+            self.msg.setWindowTitle("Error Message:openFile")
+            
         try:
-            for l in I[3].split(" "):
+            for l in alli[3].split(" "):
                 if len(l)>0:
-                    self.butnamesi.append(l)
-            self.butnamesi=self.butnamesi[2:]
-            len_bnamesi = len(self.butnamesi)
-            #print "length_new",len_bnamesi
-            #print self.butnamesi
-        except:
-            tkMessageBox.showinfo("Warning!!", "Error in Analysis File")    
+                    self.NBIList.append(l)
+            self.NBIList = self.NBIList[2:]
+            len_NBIList = len(self.NBIList)
+            #print "NBILIST : ",self.NBIList
+        except Exception as e:
+            print "Exception Message : ",str(e)
+            self.msg = QtGui.QErrorMessage(None)
+            self.msg.showMessage('Error in Analysis File.')
+            self.msg.setWindowTitle("Error Message:openFile")
+            
             
-   
         d = self.numberFinder(fpath)
         d1 = int(d[0] + 1)
-        #print "I'm D1:", d1  #for debugging
         d2 = int(d[1])
         d3 = d[2]
         d4 = d[4]
-        #print "I'm D4:", d4  #for debugging
+        
         dec = [d3,d[3]]
         #print "No. of Nodes:", d2
-        self.butnames=[]
-        idata=idata.split("\n")
-        
-        for l in idata[3].split(" "):
+        self.NBList = []
+        allv=allv.split("\n")
+        for l in allv[3].split(" "):
             if len(l)>0:
-                self.butnames.append(l)
-        self.butnames=self.butnames[2:]
-        len_bnames = len(self.butnames)
-        #print len_bnames
-        #print self.butnames
-
+                self.NBList.append(l)
+        self.NBList=self.NBList[2:]
+        len_NBList = len(self.NBList)
+        #print "NBLIST",self.NBList
+        
         ivals=[]
-        inum = len(idata[5].split("\t"))
-        inum_i = len(I[5].split("\t"))
-        #print inum
-    
+        inum = len(allv[5].split("\t"))
+        inum_i = len(alli[5].split("\t"))
+        
+        
+        
         full_data = []
         
         # Creating list of data:
         if d3 < 3 :
             for i in range(1,d2):
-                for l in idata[3+i*d1].split(" "):
+                for l in allv[3+i*d1].split(" "):
                     if len(l)>0:
-                        self.butnames.append(l)
-                self.butnames.pop(len_bnames)
-                self.butnames.pop(len_bnames)
-                len_bnames = len(self.butnames)
-                #print "volts:",self.butnames
-
+                        self.NBList.append(l)
+                self.NBList.pop(len_NBList)
+                self.NBList.pop(len_NBList)
+                len_NBList = len(self.NBList)
+            
             for n in range(1,d4):
-                for l in I[3+n*d1].split(" "):
+                for l in alli[3+n*d1].split(" "):
                     if len(l)>0:
-                        self.butnamesi.append(l)
-                #print "names:",self.butnamesi
-                self.butnamesi.pop(len_bnamesi)
-                self.butnamesi.pop(len_bnamesi)
-                len_bnamesi = len(self.butnamesi)
-                #print "current",self.butnamesi
+                        self.NBIList.append(l)
+                self.NBIList.pop(len_NBIList)
+                self.NBIList.pop(len_NBIList)
+                len_NBIList = len(self.NBIList)
             
             p=0
             k = 0
             m=0
-
-            for i in I[5:d1-1]:
-                #print "hello:"
+            
+            for i in alli[5:d1-1]:
                 if len(i.split("\t"))==inum_i:
                     j2=i.split("\t")
                     #print j2
@@ -554,42 +577,43 @@ class File_data:
                     if d3 == 0:       #not in trans
                         j2.pop()
                     #print j2
-
+                    
                     for l in range(1,d4):
-                        j3 = I[5+l*d1+k].split("\t")
+                        j3 = alli[5+l*d1+k].split("\t")
                         j3.pop(0)
                         j3.pop(0)
-                        if d3==0:
+                        if d3==0:            
                             j3.pop()     #not required for dc
-                            j3.pop()
-                            j2 = j2 + j3
-                            #print j2
+                        j3.pop()
+                        j2 = j2 + j3
+                        #print j2
+
+
                     full_data.append(j2)
+                    
                 k+=1
-                #print full_data        
-        
+                
+            #print "FULL DATA :",full_data
 
-            for i in idata[5:d1-1]:
+            
+            for i in allv[5:d1-1]:
                 if len(i.split("\t"))==inum:
                     j=i.split("\t")
                     j.pop()
                     if d3==0:
                         j.pop()
-
                     for l in range(1,d2):
-                        j1 = idata[5+l*d1+p].split("\t")
+                        j1 = allv[5+l*d1+p].split("\t")
                         j1.pop(0)
                         j1.pop(0)
                         if d3==0:
                             j1.pop()     #not required for dc
-                        if self.butnames[len(self.butnames)-1] == 'v-sweep':
-                            self.butnames.pop()
+                        if self.NBList[len(self.NBList)-1] == 'v-sweep':         
+                            self.NBList.pop()
                             j1.pop()
-                        #if l==d2-1 and d3==2:
-                            #j1.pop()
+            
                         j1.pop()
                         j = j + j1 
-                    #self.volts_length = len(j)-2        
                     j = j + full_data[m]
                     #print j
                     m+=1
@@ -597,28 +621,27 @@ class File_data:
                     j = "\t".join(j[1:])
                     j = j.replace(",","")
                     ivals.append(j)
-
+                
                 p+=1
-
+            
             self.data = ivals
-            #print self.data
-
+        
         #print "volts:",self.butnames
-        self.volts_length = len(self.butnames)
-        #print "volts_length:",self.volts_length
-        self.butnames = self.butnames + self.butnamesi
-        #print "new butnames:",self.butnames 
+        self.volts_length = len(self.NBList)
+        self.NBList = self.NBList + self.NBIList
+         
     
-        #print self.data    
+        print dec    
         return dec
-
-
+    
+    
     def numVals(self):
         a = self.volts_length        # No of voltage nodes
         b = len(self.data[0].split("\t"))
         #print "numvals:",b
         return [b,a]
-    
+
+
     def computeAxes(self):
         nums = len(self.data[0].split("\t"))
         #print "i'm nums:",nums
@@ -626,21 +649,25 @@ class File_data:
         var=self.data[0].split("\t")
         for i in range(1,nums):
             self.y.append([Decimal(var[i])])
-        #print self.y
-        #print y,nums
         for i in self.data[1:]:
             temp=i.split("\t")
             for j in range(1,nums):
                 self.y[j-1].append(Decimal(temp[j]))
-        #print len(self.y)
-        #print self.y[3]
-
         for i in self.data:
             temp=i.split("\t")
             self.x.append(Decimal(temp[0]))
-"""
-app = QtGui.QApplication(sys.argv)
-main = plotWindow()
-main.show()
-sys.exit(app.exec_())
-"""
\ No newline at end of file
+
+    
+        
+            
+        
+            
+            
+        
+            
+        
+        
+        
+        
+        
+        
\ No newline at end of file