gui.py

#! /usr/bin/env python
# -*- coding: utf-8 -*-
#
import sys
import time
import os
from PyQt4.QtGui import *
from PyQt4.QtCore import *
import join_fires
import merge_incendios
import procesar_incendios_grilla
import fire_phenology
class QProgBar(QProgressBar):
 
    value = 0
 
    @pyqtSlot()
    def increaseValue(progressBar):
        progressBar.setValue(progressBar.value)
        progressBar.value = progressBar.value+1
 

class GUI:
    def __init__(self):
        QApplication.setStyle(QStyleFactory.create('CleanLooks'))
        # Create an PyQT4 application object.
        self.a = QApplication(sys.argv)
         
        # The QWidget widget is the base class of all user interface objects in PyQt4.
        self.window = QWidget()

        # Set window size.
        self.window.resize(640/2,480/4)

        # Set window title
        self.window.setWindowTitle("Detección de eventos de incendio")
        self.macrolayout = QVBoxLayout(self.window)
        self.addIntroBlock()
        self.addFirstStageBlock()
        self.addSecondStageBlock()
        self.addThirdStageBlock()
        self.addFourthStageBlock()

    def addIntroBlock(self):
        label = QLabel("Detectar eventos de incendio")
        label.setAlignment(Qt.AlignCenter)
        self.macrolayout.addWidget(label)

    def addFirstStageBlock(self):
        # First stage: join_fires.py
        topgridlayout = QGridLayout()
        self.macrolayout.addLayout(topgridlayout)

        textbox = QLineEdit()
        textbox.setPlaceholderText("archivo csv")

        def openFileDialog():
            filename = QFileDialog.getOpenFileName(
                self.window, 
                'Abrir CSV', 
                os.getcwd()
            )
            textbox.setText(filename)

        btn = QPushButton('Cargar archivo de entrada')
        btn.clicked.connect(openFileDialog)
        btn.resize(btn.sizeHint())
        btn.move(100, 80)

        topgridlayout.addWidget(textbox,0,0)
        topgridlayout.addWidget(btn, 0,1)
        
        firstStageLayout = QGridLayout()
        self.macrolayout.addLayout(firstStageLayout)
        formato_fecha_label = QLabel("Formato de la fecha")
        formato_fecha_box = QLineEdit()
        formato_fecha_box.setPlaceholderText("%d (dia) %m (mes) %Y (año)")
        formato_fecha_ayuda = QLabel("e.g. %d/%m/%Y")
        firstStageLayout.addWidget(formato_fecha_label, 2,0)
        firstStageLayout.addWidget(formato_fecha_box, 2,1)
        firstStageLayout.addWidget(formato_fecha_ayuda, 2,2)

        separador_label = QLabel("Separador:")
        separador_radio_1 = QRadioButton(";")
        separador_radio_2 = QRadioButton(",")
        separador_radio_group = QButtonGroup()
        separador_radio_group.addButton(separador_radio_1,1)
        separador_radio_group.addButton(separador_radio_2,2)
        separador_radio_group.setExclusive(False)
        firstStageLayout.addWidget(separador_label,3,0)
        firstStageLayout.addWidget(separador_radio_1,3,1)
        firstStageLayout.addWidget(separador_radio_2,3,2)

        pdecimal_label = QLabel("Punto decimal:")
        pdecimal_radio_1 = QRadioButton(",")
        pdecimal_radio_2 = QRadioButton(".")
        pdecimal_radio_group = QButtonGroup()
        pdecimal_radio_group.addButton(pdecimal_radio_1,1)
        pdecimal_radio_group.addButton(pdecimal_radio_2,2)
        pdecimal_radio_group.setExclusive(True)
        firstStageLayout.addWidget(pdecimal_label,4,0)
        firstStageLayout.addWidget(pdecimal_radio_1,4,1)
        firstStageLayout.addWidget(pdecimal_radio_2,4,2)

        bar = QProgBar()
        self.bar_inserted = False
        def procesar_join_fires():
            if not self.bar_inserted:
                self.macrolayout.insertWidget(3,bar)
                self.bar_inserted = True
            bar.setValue(0)
            print(pdecimal_radio_group.checkedId())
            print(separador_radio_group.checkedId())
            def actualizar_barra(porc):
                bar.setValue(porc)
                self.a.processEvents()
            print("Archivo entrada = " + textbox.text())
            print("Formato de fecha = " + formato_fecha_box.text())
            pdecimal = "."
            if pdecimal_radio_group.checkedId() == 1:
                pdecimal = ","

            separador = ","
            if separador_radio_group.checkedId() == 1:
                separador = ";"

            print("pdecimal = " + pdecimal + " sep = " + separador)
            join_fires.SEPARADOR = separador
            join_fires.PUNTO_DECIMAL = pdecimal
            join_fires.FORMATO_FECHA = formato_fecha_box.text()
            app = join_fires.Difusion(actualizar_barra)
            app.cargar_de_archivo(textbox.text())
            app.correr()
            app.guardar_en_archivo(join_fires.ARCHIVO_SALIDA)
            del(app)

            
        procesar_1 = QPushButton('Procesar')
        procesar_1.clicked.connect(procesar_join_fires)
        firstStageLayout.addWidget(procesar_1, 5,0)
        self.macrolayout.addSpacing(10)

    def addSecondStageBlock(self):
        secondStageLayout = QGridLayout()
        self.macrolayout.addLayout(secondStageLayout)

        def procesar_agrupar():
            # It may look like it should be 5, but after a bar
            # is inserted in the first stage it shuold be 6.
            self.macrolayout.insertWidget(6, QLabel("Agrupando.."))
            merge_incendios.merge_incendios()
            self.macrolayout.insertWidget(7, QLabel("OK!"))


        secondStageTitle = QLabel("Agrupar focos de incendio")
        secondStageTitle.setAlignment(Qt.AlignCenter)
        secondStageFileOut = QLabel(
            "Archivo de salida:"
            + os.getcwd() + "/" + merge_incendios.DEFAULT_FILENAME_OUT
        )
        secondStageFileOut.setAlignment(Qt.AlignLeft)
        secondStageButton = QPushButton("Procesar")
        secondStageButton.clicked.connect(procesar_agrupar)
        secondStageLayout.addWidget(secondStageTitle)
        secondStageLayout.addWidget(secondStageFileOut)
        secondStageLayout.addWidget(secondStageButton)
    def addThirdStageBlock(self):
        thirdStageLayout = QGridLayout()
        self.macrolayout.addLayout(thirdStageLayout)
        bar = QProgBar()
        self.grid_bar_inserted = False
        def procesar_asignar_celdas():
            if not self.grid_bar_inserted:
                self.macrolayout.insertWidget(9, bar)
            bar.setValue(0)
            def actualizar_barra(porc):
                bar.setValue(porc)
                self.a.processEvents()

            procesar_incendios_grilla.asignar_grilla(actualizar_barra)

        thirdStageTitle = QLabel("Asignar celdas de grillas")
        thirdStageTitle.setAlignment(Qt.AlignCenter)
        thirdStageFileOut = QLabel(
            "Archivo de salida:"
            + os.getcwd() + "/"+procesar_incendios_grilla.AR_CSV_OUT
        )
        thirdStageFileOut.setAlignment(Qt.AlignLeft)
        thirdStageButton = QPushButton("Procesar")
        thirdStageButton.clicked.connect(procesar_asignar_celdas)
        thirdStageLayout.addWidget(thirdStageTitle)
        thirdStageLayout.addWidget(thirdStageFileOut)
        thirdStageLayout.addWidget(thirdStageButton)

    def addFourthStageBlock(self):
        fourthStageLayout = QGridLayout()
        self.macrolayout.addLayout(fourthStageLayout)

        def procesar_phenology():
            self.macrolayout.addWidget(QLabel("Analizando.."))
            fire_phenology.analizar_fenologia()
            self.macrolayout.addWidget(QLabel("OK!"))

        fourthStageTitle = QLabel("Analizar fenologia")
        fourthStageTitle.setAlignment(Qt.AlignCenter)
        fourthStageFileOut = QLabel(
            "Archivo de salida:"
            + os.getcwd() + "/"+fire_phenology.AR_OUT_ANUAL + "\n y "
            + os.getcwd() + "/"+fire_phenology.AR_OUT_MENSUAL
        )
        fourthStageFileOut.setAlignment(Qt.AlignLeft)
        fourthStageButton = QPushButton("Procesar")
        fourthStageButton.clicked.connect(procesar_phenology)
        fourthStageLayout.addWidget(fourthStageTitle)
        fourthStageLayout.addWidget(fourthStageFileOut)
        fourthStageLayout.addWidget(fourthStageButton)
    def run(self):
        
        self.window.show()
         
        sys.exit(self.a.exec_())


g = GUI()
g.run()

# Parte 2

# Show window