# -*- coding: utf-8 -*-
#
# Gramps - a GTK+/GNOME based genealogy program
#
# Copyright (C) 2004-2006  Donald N. Allingham
#
# This program is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
#

# $Id$

"""
Russian-specific classes for parsing and displaying dates.
"""

#-------------------------------------------------------------------------
#
# Python modules
#
#-------------------------------------------------------------------------
import re

#-------------------------------------------------------------------------
#
# GRAMPS modules
#
#-------------------------------------------------------------------------
from RelLib import Date
from _DateParser import DateParser
from _DateDisplay import DateDisplay
from _DateHandler import register_datehandler

#-------------------------------------------------------------------------
#
# Russian parser
#
#-------------------------------------------------------------------------
class DateParserRU(DateParser):

    modifier_to_int = {
        u'перед'    : Date.MOD_BEFORE, 
        u'по'    : Date.MOD_BEFORE,
        u'после' : Date.MOD_AFTER,
        u'п.'    : Date.MOD_AFTER,
        u'п'    : Date.MOD_AFTER,
        u'с'     : Date.MOD_AFTER,
        u'ок' : Date.MOD_ABOUT,
        u'ок.'   : Date.MOD_ABOUT,
        u'около'    : Date.MOD_ABOUT,
        u'примерно'  : Date.MOD_ABOUT,
        u'прим'     : Date.MOD_ABOUT,
        u'прим.'     : Date.MOD_ABOUT,
        u'приблизительно'  : Date.MOD_ABOUT,
        u'приб.'  : Date.MOD_ABOUT,
        u'прибл.'  : Date.MOD_ABOUT,
        u'приб'  : Date.MOD_ABOUT,
        u'прибл'  : Date.MOD_ABOUT,
        }

    calendar_to_int = {
        u'григорианский'   : Date.CAL_GREGORIAN,
        u'г'                 : Date.CAL_GREGORIAN,
        u'юлианский'            : Date.CAL_JULIAN,
        u'ю'                 : Date.CAL_JULIAN,
        u'еврейский'         : Date.CAL_HEBREW,
        u'е'         : Date.CAL_HEBREW,
        u'исламский'         : Date.CAL_ISLAMIC,
        u'и'                 : Date.CAL_ISLAMIC,
        u'республиканский': Date.CAL_FRENCH,
        u'р'                 : Date.CAL_FRENCH,
        u'персидский'             : Date.CAL_PERSIAN,
        u'п'             : Date.CAL_PERSIAN,
        }

    quality_to_int = {
        u'оценено'  : Date.QUAL_ESTIMATED,
        u'оцен.'       : Date.QUAL_ESTIMATED,
        u'оц.'        : Date.QUAL_ESTIMATED,
        u'оцен'       : Date.QUAL_ESTIMATED,
        u'оц'        : Date.QUAL_ESTIMATED,
        u'вычислено'      : Date.QUAL_CALCULATED,
        u'вычисл.'       : Date.QUAL_CALCULATED,
        u'выч.' : Date.QUAL_CALCULATED,
        u'вычисл'       : Date.QUAL_CALCULATED,
        u'выч' : Date.QUAL_CALCULATED,
        }

    hebrew_to_int = {
        u"тишрей":1,
        u"хешван":2,
        u"кислев":3,
        u"тевет":4,
        u"шеват":5,
        u"адар":6,
        u"адар бет":7,
        u"нисан":8,
        u"ияр":9,
        u"сиван":10,
        u"таммуз":11,
        u"ав":12,
        u"элул":13,
    }

    islamic_to_int = {
        u"мухаррам":1,
        u"сафар":2,
        u"раби-аль-авваль":3,
        u"раби-ассани":4,
        u"джумада-аль-уля":5,
        u"джумада-аль-ахира":6,
        u"раджаб":7,
        u"шаабан":8,
        u"рамадан":9,
        u"шавваль":10,
        u"зуль-каада":11,
        u"зуль-хиджжа":12,
        }

    persian_to_int = {
        u"фарвардин":1,
        u"урдбихишт":2,
        u"хурдад":3,
        u"тир":4,
        u"мурдад":5,
        u"шахривар":6,
        u"михр":7,
        u"абан":8,
        u"азар":9,
        u"дай":10,
        u"бахман":11,
        u"исфаидармуз":12,
        }

    french_to_int = {
        u"вандемьер":1,
        u"брюмер":2,
        u"фример":3,
        u"нивоз":4,
        u"плювиоз":5, 
        u"вантоз":6,
        u"жерминаль":7,
        u"флореаль":8,
        u"прериаль":9,
        u"мессидор":10,
        u"термидор":11,
        u"фрюктидор":12,
        u"дополнит.":13,
        }

    bce = [
        u'до нашей эры', u'до н. э.', u'до н.э.', 
        u'до н э', u'до нэ'] + DateParser.bce

    def init_strings(self):
        DateParser.init_strings(self)
        _span_1 = [u'с',u'от']
        _span_2 = [u'по']
        _range_1 = [u'между',u'меж\.',u'меж']
        _range_2 = [u'и']
        self._span     = re.compile("(%s)\s+(?P<start>.+)\s+(%s)\s+(?P<stop>.+)" % 
                                   ('|'.join(_span_1),'|'.join(_span_2)),
                           re.IGNORECASE)
        self._range    = re.compile("(%s)\s+(?P<start>.+)\s+(%s)\s+(?P<stop>.+)" %
                                   ('|'.join(_range_1),'|'.join(_range_2)),
                           re.IGNORECASE)

#-------------------------------------------------------------------------
#
# Russian displayer
#
#-------------------------------------------------------------------------
class DateDisplayRU(DateDisplay):

    calendar = (
        u"",
        u" (юлианский)", 
        u" (еврейский)", 
        u" (республиканский)", 
        u" (персидский)", 
        u" (исламский)"
        )

    _mod_str = (
        u"",
        u"перед ",
        u"после ",
        u"около ",
        u"",u"",u"")
    
    _qual_str = (u"",u"оцен ",u"вычисл ")

    _bce_str = u"%s до н.э."

    formats = (
        "ГГГГ-ММ-ДД (ISO)", "Численный", "Месяц День, Год",
        "МЕС ДД, ГГГГГ", "День Месяц, Год", "ДД МЕС, ГГГГГ"
        )

    _hebrew = ( u"",
        u"Тишрей",
        u"Хешван",
        u"Кислев",
        u"Тевет",
        u"Шеват",
        u"Адар",
        u"Адар бет",
        u"Нисан",
        u"Ияр",
        u"Сиван",
        u"Таммуз",
        u"Ав",
        u"Элул",
        )

    _islamic = ( u"",
        u"Мухаррам",
        u"Сафар",
        u"Раби-аль-авваль",
        u"Раби-ассани",
        u"Джумада-аль-уля",
        u"Джумада-аль-ахира",
        u"Раджаб",
        u"Шаабан",
        u"Рамадан",
        u"Шавваль",
        u"Зуль-каада",
        u"Зуль-хиджжа",
        )

    _persian = ( u"",
        u"Фарвардин",
        u"Урдбихишт",
        u"Хурдад",
        u"Тир",
        u"Мурдад",
        u"Шахривар",
        u"Михр",
        u"Абан",
        u"Азар",
        u"Дай",
        u"Бахман",
        u"Исфаидармуз",
        )

    _french = ( u"",
        u"Вандемьер",
        u"Брюмер",
        u"Фример",
        u"Нивоз",
        u"Плювиоз", 
        u"Вантоз",
        u"Жерминаль",
        u"Флореаль",
        u"Прериаль",
        u"Мессидор",
        u"Термидор",
        u"Фрюктидор",
        u"Дополнит."
        )

    def display(self,date):
        """
        Returns a text string representing the date.
        """
        mod = date.get_modifier()
        cal = date.get_calendar()
        qual = date.get_quality()
        start = date.get_start_date()

        qual_str = self._qual_str[qual]
        
        if mod == Date.MOD_TEXTONLY:
            return date.get_text()
        elif start == Date.EMPTY:
            return ""
        elif mod == Date.MOD_SPAN:
            d1 = self.display_cal[cal](start)
            d2 = self.display_cal[cal](date.get_stop_date())
            return "%sс %s %s %s%s" % (qual_str,d1,u'по',d2,self.calendar[cal])
        elif mod == Date.MOD_RANGE:
            d1 = self.display_cal[cal](start)
            d2 = self.display_cal[cal](date.get_stop_date())
            return "%s%s %s %s %s%s" % (qual_str,u'между',d1,u'и',d2,self.calendar[cal])
        else:
            text = self.display_cal[date.get_calendar()](start)
            return "%s%s%s%s" % (qual_str,self._mod_str[mod],text,self.calendar[cal])

#-------------------------------------------------------------------------
#
# Register classes
#
#-------------------------------------------------------------------------
register_datehandler(('ru_RU','ru','russian'),DateParserRU, DateDisplayRU)