#!/usr/bin/env python
#--!-- coding: utf8 --!--
 



from qt import *
from enums import *

# Because I have trouble with QDataWidgetMapper and the checkbox, I don't know why.

class chkOutlineCompile(QCheckBox):
    
    def __init__(self, parent=None):
        QCheckBox.__init__(self, parent)
        self.stateChanged.connect(self.submit)
        self._column = Outline.compile.value
        self._index = None
        self._indexes = None
        self._model = None
        self._updating = False
        
    def setModel(self, model):
        self._model = model
        self._model.dataChanged.connect(self.update)
        
    def setCurrentModelIndex(self, index):
        self._indexes = None
        if index.column() != self._column:
            index = index.sibling(index.row(), self._column)
        self._index = index
        self.setTristate(False)
        self.updateCheckState()
        
    def setCurrentModelIndexes(self, indexes):
        self._index = None
        self._indexes = []
        for i in indexes:
            if i.column() != self._column:
                i = i.sibling(i.row(), self._column)
            self._indexes.append(i)
        self.updateCheckState()
        
    def getCheckedValue(self, index):
        item = index.internalPointer()
        c = item.data(Outline.compile)
        if c:
            c = int(c)
        else:
            c = Qt.Unchecked
        
        return c
        
    def update(self, topLeft, bottomRight):
        
        if self._updating:
            # We are currently putting data in the model, so no updates
            return
        
        if self._index:
            if topLeft.row() <= self._index.row() <= bottomRight.row():
                self.updateCheckState()
                
        elif self._indexes:
            update = False
            
            for i in self._indexes:
                if topLeft.row() <= i.row() <= bottomRight.row():
                    update = True
                    
            if update:
                self.updateCheckState()
        
    def updateCheckState(self):
        
        if self._index:
            self.setEnabled(True)
            c = self.getCheckedValue(self._index)
            self.setCheckState(c)
            
        elif self._indexes:
            self.setEnabled(True)
            values = []
            for i in self._indexes:
                values.append(self.getCheckedValue(i))
            
            same = True
            for v in values[1:]:
                if v != values[0]:
                    same = False
                    break
            
            if same:
                self.setCheckState(values[0])
            else:
                self._updating = True
                self.setCheckState(Qt.PartiallyChecked)
                self._updating = False
            
        else:
            self.setChecked(False)
            self.setEnabled(False)
        
    def submit(self, state):
        
        if self._updating:
            return
        
        if self._index:
            if self._model.data(self._index) != state:
                self._model.setData(self._index, state)
            
        elif self._indexes:
            for i in self._indexes:
                if self._model.data(i) != state:
                    self._model.setData(i, state)
        
        self.setTristate(False)