# -*- coding: utf-8 -*-
"""novelWriter GUI Document Word Counter

 novelWriter – GUI Document Word Counter
===================================
 A thread for counting words and characters in a document

 File History:
 Created: 2019-04-22 [0.0.1]

"""

import logging
import nw

from time import time

from PyQt5.QtWidgets     import QWidget, QTextEdit, QHBoxLayout, QVBoxLayout, QFrame, QSplitter, QToolBar, QAction, QScrollArea
from PyQt5.QtCore        import Qt, QSize, QSizeF, QTimer, QThread, pyqtSignal
from PyQt5.QtGui         import QIcon, QFont, QTextCursor, QTextFormat, QTextBlockFormat

from nw.gui.dochighlight import GuiDocHighlighter

logger = logging.getLogger(__name__)

class WordCounter(QThread):

    def __init__(self, theParent):
        QThread.__init__(self, theParent)
        self.theParent = theParent
        self.charCount = 0
        self.wordCount = 0
        return

    def run(self):

        self.charCount = 0
        self.wordCount = 0

        for n in range(self.theParent.theDoc.blockCount()):

            theBlock = self.theParent.theDoc.findBlockByNumber(n)
            if not theBlock.isValid():
                continue

            theText = theBlock.text()
            theLen  = len(theText)

            if theLen == 0:
                continue
            if theText[0] == "@" or theText[0] == "%":
                continue

            if   theText[0:5] == "#### ":
                self.wordCount -= 1
                self.charCount -= 5
            elif theText[0:4] == "### ":
                self.wordCount -= 1
                self.charCount -= 4
            elif theText[0:3] == "## ":
                self.wordCount -= 1
                self.charCount -= 3
            elif theText[0:2] == "# ":
                self.wordCount -= 1
                self.charCount -= 2

            theBuff = theText.replace("–"," ").replace("—"," ")
            self.wordCount += len(theBuff.split())
            self.charCount += theLen

        pass

## END Class _WordCounter