#!/usr/bin/env python


"""#!/usr/bin/python"""

#This file create the gui to install code model in the ngspice.

import os
import sys
import shutil
import subprocess
from PyQt4 import QtGui
from PyQt4 import QtCore
from ConfigParser import SafeConfigParser
from Appconfig import Appconfig
from createKicadLibrary import AutoSchematic

class Mainwindow(QtGui.QWidget):
    def __init__(self):
        #super(Mainwindow, self).__init__()
        QtGui.QMainWindow.__init__(self)
        print "Initializing.........."
        self.home = os.path.expanduser("~")
        #Reading all varibale from config.ini
        self.parser = SafeConfigParser()
        self.parser.read(os.path.join(self.home, os.path.join('.nghdl','config.ini')))
        self.ngspice_home = self.parser.get('NGSPICE','NGSPICE_HOME')
        self.release_dir = self.parser.get('NGSPICE','RELEASE')
        self.src_home = self.parser.get('SRC','SRC_HOME')
        self.licensefile = self.parser.get('SRC','LICENSE')
        #Printing LICENCE file on terminal
        fileopen = open(self.licensefile, 'r')
        print fileopen.read()
        fileopen.close()
        self.file_list = []             #to keep the supporting files
        self.errorFlag = False          #to keep the check of "make install" errors
        self.initUI()

    def initUI(self):
        self.uploadbtn = QtGui.QPushButton('Upload')
        self.uploadbtn.clicked.connect(self.uploadModle)
        self.exitbtn = QtGui.QPushButton('Exit')
        self.exitbtn.clicked.connect(self.closeWindow)
        self.browsebtn = QtGui.QPushButton('Browse')
        self.browsebtn.clicked.connect(self.browseFile)
        self.addbtn = QtGui.QPushButton('Add Files')
        self.addbtn.clicked.connect(self.addFiles)
        self.removebtn = QtGui.QPushButton('Remove Files')
        self.removebtn.clicked.connect(self.removeFiles)
        self.ledit = QtGui.QLineEdit(self)
        self.sedit = QtGui.QTextEdit(self)
        self.process = QtCore.QProcess(self)
        self.termedit = QtGui.QTextEdit(self)
        self.termedit.setReadOnly(1)
        pal = QtGui.QPalette()
        bgc = QtGui.QColor(0, 0, 0)
        pal.setColor(QtGui.QPalette.Base, bgc)
        self.termedit.setPalette(pal)
        self.termedit.setStyleSheet("QTextEdit {color:white}")

        #Creating gridlayout
        grid = QtGui.QGridLayout()
        grid.setSpacing(5)
        grid.addWidget(self.ledit, 1, 0)
        grid.addWidget(self.browsebtn, 1, 1)
        grid.addWidget(self.sedit, 2, 0, 4, 1)
        grid.addWidget(self.addbtn, 2, 1)
        grid.addWidget(self.removebtn, 3, 1)
        grid.addWidget(self.termedit, 6, 0, 10, 1)
        grid.addWidget(self.uploadbtn, 17, 0)
        grid.addWidget(self.exitbtn,17, 1)

        self.setLayout(grid)
        self.setGeometry(300, 300, 600,600)
        self.setWindowTitle("Ngspice Digital Model Creator")
        #self.setWindowIcon(QtGui.QIcon('logo.png'))
        self.show()
	

    def closeWindow(self):
        try:
            self.process.close()
        except:
                pass
        print "Close button clicked"
        quit()


    def browseFile(self):
        print "Browse button clicked"
        self.filename = QtGui.QFileDialog.getOpenFileName(self, 'Open File', '.')
        self.ledit.setText(self.filename)
        print "Vhdl file uploaded to process :", self.filename


    def addFiles(self):
        print "Starts adding supporting files"
        title = self.addbtn.text()
        for file in QtGui.QFileDialog.getOpenFileNames(self, title):
                self.sedit.append(str(file))
                self.file_list.append(file)
        print "Supporting Files are :",self.file_list


    def removeFiles(self):
            self.fileRemover = FileRemover(self)


    #check extensions of all supporting files
    def checkSupportFiles(self):
        nonvhdl_count = 0
        for file in self.file_list:
            extension = os.path.splitext(str(file))[1]
            if extension != ".vhdl":
                nonvhdl_count += 1
                self.file_list.remove(file)

        if nonvhdl_count > 0:
             QtGui.QMessageBox.critical(self,'Critical','''<b>Important Message.</b><br/><br/>Supporting files should be <b>.vhdl</b> file ''')


    def createModelDirectory(self):
        print "Create Model Directory Called"
        self.digital_home=self.parser.get('NGSPICE','DIGITAL_MODEL')
        os.chdir(self.digital_home)
        print "Current Working Directory Changed to",os.getcwd()
        self.modelname = os.path.basename(str(self.filename)).split('.')[0]
        print "Model to be created :",self.modelname
        # Looking if model directory is present or not
        if os.path.isdir(self.modelname):
            print "Model Already present"
            ret = QtGui.QMessageBox.warning(self, "Warning",'''<b>This model already exist. Do you want to overwrite it?</b><br/>
                    If yes press ok, else cancel it and change the name of your vhdl file.''', QtGui.QMessageBox.Ok, QtGui.QMessageBox.Cancel)
            if ret == QtGui.QMessageBox.Ok:
                print "Overwriting existing model "+self.modelname
                cmd="rm -rf "+self.modelname
                #process = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE, shell=True)
                subprocess.call(cmd, shell=True)
                os.mkdir(self.modelname)
            else:
                print "Exiting application"
                quit()
        else:
            print "Creating model "+self.modelname+" directory"
            os.mkdir(self.modelname)
    

    def addingModelInModpath(self):
        print "Adding Model "+self.modelname+" in Modpath file "+self.digital_home
        #Adding name of model in the modpath file
        #Check if the string is already in the file
        with open(self.digital_home+"/modpath.lst",'a+') as f:
            flag = 0
            for line in f:
                if line.strip() == self.modelname:
                    print "Found model "+self.modelname+" in the modpath.lst"
                    flag = 1
                    break
                else:
                    pass

            if flag == 0:
                print "Adding model name "+self.modelname+" into modpath.lst"
                f.write(self.modelname+"\n")
            else:
                print "Model name is already into modpath.lst"


    def createModelFiles(self):
        print "Create Model Files Called"
        os.chdir(self.cur_dir)
        print "Current Working directory changed to "+self.cur_dir
        cmd = "python "+self.src_home+"/src/model_generation.py "+str(self.ledit.text())
        stdouterr = os.popen4(cmd)[1].read()
        print stdouterr
        #Moving file to model directory
        path=os.path.join(self.digital_home,self.modelname)
        shutil.move("cfunc.mod",path)
        shutil.move("ifspec.ifs",path)

        #Creating directory inside model directoy
        print "Creating DUT directory at "+os.path.join(path,"DUTghdl")
        os.mkdir(path+"/DUTghdl/")
        print "Copying required file to DUTghdl directory"
        shutil.move("connection_info.txt",path+"/DUTghdl/")
        shutil.move("start_server.sh",path+"/DUTghdl/")
        shutil.move("sock_pkg_create.sh",path+"/DUTghdl/")
        shutil.move(self.modelname+"_tb.vhdl",path+"/DUTghdl/")
        
        shutil.copy(str(self.filename),path+"/DUTghdl/")
        shutil.copy(os.path.join(self.home, self.src_home)+"/src/ghdlserver/compile.sh",path+"/DUTghdl/")
        shutil.copy(os.path.join(self.home, self.src_home)+"/src/ghdlserver/uthash.h",path+"/DUTghdl/")
        shutil.copy(os.path.join(self.home, self.src_home)+"/src/ghdlserver/ghdlserver.c",path+"/DUTghdl/")
        shutil.copy(os.path.join(self.home, self.src_home)+"/src/ghdlserver/ghdlserver.h",path+"/DUTghdl/")
        shutil.copy(os.path.join(self.home, self.src_home)+"/src/ghdlserver/Utility_Package.vhdl",path+"/DUTghdl/")
        shutil.copy(os.path.join(self.home, self.src_home)+"/src/ghdlserver/Vhpi_Package.vhdl",path+"/DUTghdl/")

        for file in self.file_list:
                shutil.copy(str(file), path+"/DUTghdl/")
        
        os.chdir(path+"/DUTghdl")
        subprocess.call("bash "+path+"/DUTghdl/compile.sh", shell=True)
        subprocess.call("chmod a+x start_server.sh",shell=True)
        subprocess.call("chmod a+x sock_pkg_create.sh",shell=True)
        os.remove("compile.sh")
        os.remove("ghdlserver.c")
        #os.remove("ghdlserver.h")
        #os.remove("Utility_Package.vhdl")
        #os.remove("Vhpi_Package.vhdl")
        
    
    #slot to redirect stdout and stderr to window console
    @QtCore.pyqtSlot()
    def readAllStandard(self):
        self.termedit.append(QtCore.QString(self.process.readAllStandardOutput()))
        stderror = QtCore.QString(self.process.readAllStandardError())
        if stderror.contains("ERROR", QtCore.Qt.CaseInsensitive):
            self.errorFlag = True        
        self.termedit.append(stderror)
        

    def runMake(self):
        print "run Make Called"
        self.release_home=self.parser.get('NGSPICE','RELEASE')
        os.chdir(self.release_home)
        try:
            cmd = " make"
            print "Running Make command in "+self.release_home
            path = os.getcwd()
            self.process = QtCore.QProcess(self)
            self.process.start(cmd)
            print "make command process pid ---------- >",self.process.pid()
            
        except:
            print "There is error in 'make' "
            quit()


    def runMakeInstall(self):
        print "run Make Install Called"
        try:
            cmd = " make install"
            print "Running Make Install"
            path = os.getcwd()
            try:
                self.process.close()
            except:
                pass
            
            self.process = QtCore.QProcess(self)
            self.process.start(cmd)
            self.process.finished.connect(self.createSchematicLib)
            QtCore.QObject.connect(self.process, QtCore.SIGNAL("readyReadStandardOutput()"), self, QtCore.SLOT("readAllStandard()"))
            os.chdir(self.cur_dir)

        except:
            print "There is error during in 'make install' "
            quit()


    def createSchematicLib(self):
        if Appconfig.esimFlag == 1:
            if not self.errorFlag:
                print 'Creating library files.................................'
                schematicLib = AutoSchematic(self.modelname)
                schematicLib.createKicadLibrary()
            else:
                QtGui.QMessageBox.critical(self,'Error','''Cannot create Schematic Library of your model. Resolve the <b>errors</b> shown on console of NGHDL window. ''')
        else:
            QtGui.QMessageBox.information(self,'Message','''<b>Important Message</b><br/><br/>To create Schematic Library of your model, use NGHDL through <b>eSim</b> ''')    


    def uploadModle(self):
        print "Upload button clicked"
        try:
            self.process.close()
        except:
                pass
        try:
            self.file_extension = os.path.splitext(str(self.filename))[1]
            print "Uploaded File extension :"+self.file_extension
            self.cur_dir = os.getcwd()
            print "Current Working Directory :"+self.cur_dir
            self.checkSupportFiles()
            if self.file_extension == ".vhdl":
                self.errorFlag = False
                self.createModelDirectory()
                self.addingModelInModpath()
                self.createModelFiles()
                self.runMake()
                self.runMakeInstall()
            else:
                QtGui.QMessageBox.information(self,'Message','''<b>Important Message.</b><br/><br/>This accepts only <b>.vhdl</b> file ''')
        except Exception as e:
            QtGui.QMessageBox.critical(self, 'Error', str(e))

class FileRemover(QtGui.QWidget):

        def __init__(self, main_obj):
                super(FileRemover, self).__init__()
                self.row = 0
                self.col = 0
                self.cb_dict = {}
                self.marked_list = []
                self.files = main_obj.file_list
                self.sedit = main_obj.sedit

                print self.files

                self.grid = QtGui.QGridLayout()
                removebtn = QtGui.QPushButton('Remove', self)
                removebtn.clicked.connect(self.removeFiles)

                self.grid.addWidget(self.createCheckBox(), 0, 0)
                self.grid.addWidget(removebtn, 1, 1)

                self.setLayout(self.grid)
                self.show()

        def createCheckBox(self):

                self.checkbox = QtGui.QGroupBox()
                self.checkbox.setTitle('Remove Files')
                self.checkgrid = QtGui.QGridLayout()

                self.checkgroupbtn = QtGui.QButtonGroup()

                for path in self.files:

                        print path

                        self.cb_dict[path] = QtGui.QCheckBox(path)
                        self.checkgroupbtn.addButton(self.cb_dict[path])
                        self.checkgrid.addWidget(self.cb_dict[path], self.row, self.col)
                        self.row += 1

                self.checkgroupbtn.setExclusive(False)
                self.checkgroupbtn.buttonClicked.connect(self.mark_file)
                self.checkbox.setLayout(self.checkgrid)

                return self.checkbox

        def mark_file(self):

                for path in self.cb_dict:
                        if self.cb_dict[path].isChecked():
                                if path not in self.marked_list:
                                        self.marked_list.append(path)

                        else:
                                if path in self.marked_list:
                                        self.marked_list.remove(path)

        def removeFiles(self):

                for path in self.marked_list:
                        print path +" is removed"
                        self.sedit.append(path + " removed")
                        self.files.remove(path)

                self.sedit.clear()
                for path in self.files:
                        self.sedit.append(path)

                self.marked_list[:] = []
                self.files[:] = []
                self.close()



def main():
    app = QtGui.QApplication(sys.argv)
    if len(sys.argv) > 1:
        if sys.argv[1] == '-e':
            Appconfig.esimFlag = 1
    w = Mainwindow()
    sys.exit(app.exec_())

if __name__=='__main__':
    main()