# -*- coding: ISO-8859-1 -*-
""" capellaScript -- Copyright (c) 2006 Peter Becker
>>> FaulenzerZaehler : Faulenzer durchnummerieren.
    Die zu nummerierenden Faulenzer mit dem Cursor markieren,
    und dann das Script aufrufen||
    Bitte nur aufeinanderfolgende Faulenzer markieren !||
    Bei Problemen bitte eine Mail an peter_becker@freenet.de
<<<
<?xml version="1.0" encoding="ISO-8859-1"?>
<info>
  <lang id="en">
    <title>Numbering consecutive measure repetitions</title>
    <descr>
      <p>Mark the measure repetitions to be numbered with the cursor,
and then open the script.</p> <p>
Please mark only consecutive measure repetitions!</p> <p>
In case of problems please send a mail to peter_becker@freenet.de</p>
    </descr>
  </lang>

  <lang id="de">
    <title>Faulenzerzaehler : Faulenzer durchnummerieren.</title>
    <descr>
      <p>Die zu nummerierenden Faulenzer mit dem Cursor markieren,
und dann das Script aufrufen.</p> <p>
Bitte nur aufeinanderfolgende Faulenzer markieren !</p> <p>
Bei Problemen bitte eine Mail an peter_becker@freenet.de</p>
    </descr>
  </lang>

  <lang id="nl">
    <title>Maatherhalingen doornummeren</title>
    <descr>
      <p>Markeer de te nummeren maatherhalingen met de cursor en
open daarna het script.</p> <p>
S.v.p alleen opeenvolgende maatherhalingen markeren!</p> <p>
Bij problemen graag een mail naar peter_becker@freenet.de</p>
    </descr>
  </lang>
</info>

Tags   :  2086-60
History:  25.09.06 - Erste Ausgabe
          07.02.07 - bessere Positionierung des Counts
          04.12.07 - freie Schriftwahl
          22.12.18 - WW Internationalisierung (en-de-nl) + 'help'(=Infotext)
          07.07.21 - WW Font 'Capella3' auch verwendbar (sodann 'y'= -3)
	  13.07.21 - PB Absturz behoben bei bestimmten Zeichen im Font Namen
	                Notmassnahme für MAC in getFonts() 
	  27.12.21   PB Schreibweise Times New Roman korrigiert
	  07.03.23   WW - Verikale Position ('y') mit Font 'Capella3'  nach -3.2  und sonst nach -2.35 :
                          vertikale Position wie bei 'grafische Mehrtakt-Pausen'
                        - 'Capella3' an Fontlist für MAC zugefügt
"""
english = {
    'error'         :'Error',
    'noScore'       :'No active score',
    'standard'      :'Standard',
    'italic'        :'Italic',
    'bold'          :'Bold',
    'boldItalic'    :'Bold Italic',
    'counter1stMeas':'Number for the first measure repetition    ',
    'charStyle'     :'Character style',
    'fontSize'      :'Font size',
    'font'          :'Font',
    'format'        :'Format',
    'counter'       :'Numbering measure repetitions',
    'title'         :'Numbering consecutive measure repetitions',
    'help'          :'Mark the measure repetitions to be numbered with the \
cursor, and then open the script.\n\
\nPlease mark only consecutive measure repetitions!\n\
\nIn addition to fonts for texts the font \'Capella3\' can also be used.\n\
\nIn case of problems please send a mail to peter_becker@freenet.de'
}
german = {
    'error'         :'Fehler',
    'noScore'       :'keine aktive Partitur',
    'standard'      :'Standard',
    'italic'        :'Kursiv',
    'bold'          :'Fett',
    'boldItalic'    :'Fett Kursiv',
    'counter1stMeas':'Zähler für ersten Faulenzer    ',
    'charStyle'     :'Schriftschnitt',
    'fontSize'      :'Schriftgröße',
    'font'          :'Schriftart',
    'format'        :'Schrift',
    'counter'       :'Faulenzerzähler',
    'title'         :'Faulenzerzaehler : Faulenzer durchnummerieren',
    'help'          :'Die zu nummerierenden Faulenzer mit dem Cursor markieren,\
\nund dann das Script aufrufen.\n\
\nBitte nur aufeinanderfolgende Faulenzer markieren !\n\
\nNeben Schriftarten für Texte kann auch die Schriftart \'Capella3\' verwendet werden.\n\
\nBei Problemen bitte eine Mail an peter_becker@freenet.de'
}
dutch = {
    'error'         :'fout',
    'noScore'       :'Geen actieve partituur',
    'standard'      :'Standaard',
    'italic'        :'Cursief',
    'bold'          :'Vet',
    'boldItalic'    :'Vet Cursief',
    'counter1stMeas':'Nummer boven de eerste maatherhaling    ',
    'charStyle'     :'Tekenstijl',
    'fontSize'      :'Lettergrootte',
    'font'          :'Lettertype',
    'format'        :'Opmaak',
    'counter'       :'Nummering maatherhalingen',
    'title'         :'Maatherhalingen doornummeren',
    'help'          :'Markeer de te nummeren maatherhalingen met de cursor\
\nen open daarna het script.\n\
\nS.v.p alleen opeenvolgende maatherhalingen markeren!\n\
\nBehalve lettertypen voor teksten kan men ook het lettertype \'Capella3\' gebruiken.\n\
\nBij problemen graag een mail naar peter_becker@freenet.de'
}

try:
    setStringTable(
        ("en", english),
        ("de", german),
        ("nl", dutch))
except:
    def tr(s):
        return german[s]
#--------------------------------------------------
import xml.dom, tempfile
#import tkFont
#import Tkinter
from caplib.capDOM import ScoreChange
from xml.dom.minidom import NodeList, Node, Element

doc        = [] # parentNode von score

# auf Mac Schriftauswahl einschränken, um TkInter zu vermeiden
# Sind wir auf Mac?
onMac = False
onWin = False
try:
    import _winreg
    onWin = True
except:
    try:
	if sys.platform == "darwin":
	    onMac = True
    except:
	messageBox(tr('FEHLER'))
    pass

if onWin:
    import tkFont
    import Tkinter

def latin1_e(u):                                  # 13.7.21 PB unwirksam, da sonst bei bestimmten Zeichen Absturz
    return u                                 
    #return u.encode('Latin-1')                    

def getCursor():
    sel = curSelection()
    result = None
    if sel == 0:
        messageBox(tr('error'), tr('noScore'))
        return result
    return sel

def getElementObjects(objList):  # returns a List
    newList = NodeList()
    for n in range(objList.length):
        if objList[n].nodeType == objList[n].ELEMENT_NODE:
            newList.append(objList[n])
    return newList

def findElementNode(el,tagName):
    global doc
    childs = el.childNodes
    for n in range(childs.length):
        if childs[n].nodeType ==childs[n].ELEMENT_NODE and childs[n].tagName == tagName:
            return childs[n]
    return 'NONE'

def addElementNode(el,tagName):
    # Neue Node zu "el" hinzufügen wenn Node "tagName" nicht existiert
    # ansonsten existierende Node zurückmelden
    global doc
    childs = el.childNodes
    for n in range(childs.length):
        if childs[n].nodeType ==childs[n].ELEMENT_NODE and childs[n].tagName == tagName:
            return childs[n]
    newChild = doc.createElement(tagName)
    el.appendChild(newChild)
    return newChild

def addNewElementNode(el,tagName):
    # add new Node with tagName "tagName" to el 
    global doc
    newChild = doc.createElement(tagName)
    el.appendChild(newChild)
    return newChild

def getElementObjects(objList):  # returns a List
    newList = NodeList()         
    for n in range(objList.length):
        if objList[n].nodeType == objList[n].ELEMENT_NODE:
            newList.append(objList[n])
    return newList

    
def addCount(score,chord,count):
    global doc, Font, Hoehe, Schriftschnitt

    drawObjects = addElementNode(chord,'drawObjects')
    drawObjectL = getElementObjects(drawObjects.childNodes)
    ix = 0
    while ix < len(drawObjectL): 
        basic = findElementNode(drawObjectL[ix],'basic')
        if basic <> 'NONE' and basic.hasAttribute('tag'):
            tag = basic.getAttribute('tag')
            if tag == '2086-60':
                drawObjectL[ix].parentNode.removeChild(drawObjectL[ix]) 
        ix = ix+1        
    drawObj     = addNewElementNode(drawObjects,'drawObj')
    text        = addNewElementNode(drawObj,'text')
    font        = addNewElementNode(text,'font')
    content     = addNewElementNode(text,'content')
    font.setAttribute('face',Font)
    font.setAttribute('height',str(Hoehe))
    font.setAttribute('pitchAndFamily','0')
    if Schriftschnitt == 0:
        weight = '0'
    elif Schriftschnitt == 1:
        weight = '0'
        font.setAttribute('italic','true')
    elif Schriftschnitt == 2:
        weight = '700'
    elif Schriftschnitt == 3:
        weight = '700'
        font.setAttribute('italic','true')
    font.setAttribute('weight',weight)
    #text.setAttribute('align','center')
    if Font == 'Capella3':
        text.setAttribute('y','-3.2')       # war -3 (WW)       
    else:
        text.setAttribute('y','-2.35')      # war -2.5 (WW) 
    text.setAttribute('x','0.1')
    if int(count) >= 10:
        text.setAttribute('x','-0.3125')        
    basic       = addNewElementNode(drawObj,'basic')
    basic.setAttribute('tag','2086-60')
    #messageBox('ADD',str(score) + '\n' + str(chord) + '\n' + str(count))
    content.appendChild(score.parentNode.createTextNode(str(count)))

    return


def changeDoc(score):
    global doc, startcnt 

    doc    = score.parentNode

    sel = getCursor()
    if sel == None:
        #
        return
    else:
        selvon = sel[0]
        selbis = sel[1]
        ix       = 0
        while selvon[3]+ix < selbis[3]: 
            system = score.getElementsByTagName('system')[selvon[0]]
            staff = system.getElementsByTagName('staff')[selvon[1]]
            voice = staff.getElementsByTagName('voice')[selvon[2]]
            noteObject = voice.getElementsByTagName('noteObjects')[0]
            objList = getElementObjects(noteObject.childNodes)
            if objList.length <= selvon[3]:
                ix = ix + 1
                return
            obj = objList[selvon[3]+ix]
            ix = ix + 1
            dos = findElementNode(obj,'drawObjects')
            if dos <> 'NONE':
                objList2 = getElementObjects(dos.childNodes)
                doix = 0
                while doix < len(objList2): 
                    objList3 = getElementObjects(objList2[doix].childNodes)
                    cont = findElementNode(objList3[0],'content')
                    doix = doix+1
                    #messageBox('CONT',str(objList3) + '\n' + str(cont))
                    if cont <> 'NONE':                          # RTF überspringen
                        if cont.firstChild.nodeValue == 'W':                
                            addCount(score, obj, str(startcnt))
                            startcnt = startcnt + 1
    return

def getFonts():
    if onWin:
	root = Tkinter.Tk()
	FontListT= tkFont.families()
	FontList  = []
	FontList2  = []
	i=0
	while i < len(FontListT):
	    try:
		FontList += [latin1_e(FontListT[i])]
	    except:
		FontList += ['FEHLER']
	    i+=1
	j=0
	while j < len(FontList):
	    FontName = FontList[j]
	    FontList[j] = FontName[0].upper() + FontName[1:]
	    j+=1
	FontList.sort()
    if onMac:
	FontList = ["Arial", "Arial Narrow", "Capella3", "Chalkboard", "Courier", "Courier New", "Helvetica", "Times New Roman"]

    return FontList

def pqDialog(FontList):
    global startcnt, Hoehe, Font, Schriftschnitt

    sc = '2'                # Startcount
    options = ScriptOptions() 
    opt = options.get()
    Font = 'Times New Roman'
    Hoehe = '8'
    Schriftschnitt = '0'
    if opt:
        Font = opt.get('FontTyp',Font)
        Hoehe = opt.get('FontHigh',Hoehe)
        Schriftschnitt = opt.get('FontWeight',Schriftschnitt)
        FontList = [Font] + FontList
        
    Schriftschnitte = [tr('standard'),tr('italic'),tr('bold'),tr('boldItalic')]    
    placeholder1   = Label('       ',width=1)
    placeholder2   = Label('     ',width=1)
    placeholder3   = Label('   ',width=1)
    placeholder4   = Label(' ',width=1)
  
    lablSc         = Label(tr('counter1stMeas'),width=6)
    editSc         = Edit (sc,width=10)
    subboxSc       = HBox([lablSc,editSc,placeholder4],padding=0)
    box1           = HBox([subboxSc,placeholder4],padding=4)
    radSchnitt = Radio(Schriftschnitte, value=int(opt.get(tr('charStyle'),str(Schriftschnitt))), padding = 0)    
    editHeight = Edit(Hoehe, width=4, min = 1, max = 99)
    comboFont  = ComboBox(FontList, width=20, value=0)
    box1ba     = HBox([radSchnitt], padding = 0,text=tr('charStyle'))   
    box1bb     = VBox([editHeight], padding = 0,text='')
    box1bb     = HBox([box1bb, placeholder2], padding = 5,text=tr('fontSize'))
    box1bc     = VBox([comboFont], padding = 0,text='')
    box1bc     = HBox([box1bc], padding = 50,text=tr('font'))
    box1bd     = HBox([box1ba,box1bb], padding = 8, text = '') 
    box1be     = VBox([box1bd, box1bc], padding = 8,text='')
    box1b      = HBox([box1be], padding = 8,text=tr('format'))  
    mainboxSc  = VBox([box1],text=tr('counter'))


    box        = VBox([mainboxSc,box1b,placeholder1],padding=1)
    dlg  = Dialog(tr('title'), box, help=tr('help'))
 
    if dlg.run():
        startcnt = int(editSc.value())
        Hoehe = editHeight.value()
        Font = FontList[comboFont.value()]
        Schriftschnitt = radSchnitt.value()
        opt  = dict(
                  FontTyp    = str(Font),
                  FontHigh  = str(Hoehe),
                  FontWeight = str(Schriftschnitt)
                 )
        options.set(opt)
        return True
    else:
        return False
                       
# Hauptprogramm:

from caplib.capDOM import ScoreChange
import tempfile


class ScoreChange(ScoreChange):
    
    def changeScore(self, score):
        changeDoc(score)                         

if activeScore():
    FontList = getFonts()
    if pqDialog(FontList):
        global tempInput, tempOutput
        activeScore().registerUndo("FaulenzerZaehler")
        tempInput = tempfile.mktemp('.capx')
        tempOutput = tempfile.mktemp('.capx')
        activeScore().write(tempInput)

        ScoreChange(tempInput, tempOutput)
    
        activeScore().read(tempOutput)
        os.remove(tempInput)
        os.remove(tempOutput)