]> git.lyx.org Git - lyx.git/blobdiff - src/Counters.cpp
Update my email and status.
[lyx.git] / src / Counters.cpp
index e3596477a7fe0d8febb18b3e417c293b802a4561..21331e8e1878ba0333c49dec58232f83cab225f9 100644 (file)
 #include <config.h>
 
 #include "Counters.h"
-
+#include "Layout.h"
 #include "Lexer.h"
 
 #include "support/convert.h"
 #include "support/debug.h"
-#include "support/lstrings.h"
-
+#include "support/gettext.h"
 #include "support/lassert.h"
+#include "support/lstrings.h"
 
+#include <algorithm>
 #include <sstream>
 
 using namespace std;
@@ -37,8 +38,8 @@ Counter::Counter()
 }
 
 
-Counter::Counter(docstring const & mc, docstring const & ls, 
-                docstring const & lsa)
+Counter::Counter(docstring const & mc, docstring const & ls,  
+                docstring const & lsa)
        : master_(mc), labelstring_(ls), labelstringappendix_(lsa)
 {
        reset();
@@ -51,6 +52,7 @@ bool Counter::read(Lexer & lex)
                CT_WITHIN = 1,
                CT_LABELSTRING,
                CT_LABELSTRING_APPENDIX,
+               CT_PRETTYFORMAT,
                CT_END
        };
 
@@ -58,6 +60,7 @@ bool Counter::read(Lexer & lex)
                { "end", CT_END },
                { "labelstring", CT_LABELSTRING },
                { "labelstringappendix", CT_LABELSTRING_APPENDIX },
+               { "prettyformat", CT_PRETTYFORMAT },
                { "within", CT_WITHIN }
        };
 
@@ -80,6 +83,10 @@ bool Counter::read(Lexer & lex)
                                if (master_ == "none")
                                        master_.erase();
                                break;
+                       case CT_PRETTYFORMAT:
+                               lex.next();
+                               prettyformat_ = lex.getDocString();
+                               break;
                        case CT_LABELSTRING:
                                lex.next();
                                labelstring_ = lex.getDocString();
@@ -102,6 +109,7 @@ bool Counter::read(Lexer & lex)
        return getout;
 }
 
+
 void Counter::set(int v)
 {
        value_ = v;
@@ -138,15 +146,31 @@ docstring const & Counter::master() const
 }
 
 
-docstring const & Counter::labelString() const
+bool Counter::checkAndRemoveMaster(docstring const & cnt)
+{
+       if (master_ != cnt)
+               return false;
+       master_ = docstring();
+       return true;
+}
+
+
+docstring const & Counter::labelString(bool in_appendix) const
 {
-       return labelstring_;
+       return in_appendix ? labelstringappendix_ : labelstring_;
 }
 
 
-docstring const & Counter::labelStringAppendix() const
+Counter::StringMap & Counter::flatLabelStrings(bool in_appendix) const
 {
-       return labelstringappendix_;
+       return in_appendix ? flatlabelstringappendix_ : flatlabelstring_;
+}
+
+
+Counters::Counters() : appendix_(false), subfloat_(false)
+{
+       layout_stack_.push_back(0);
+       counter_stack_.push_back(from_ascii(""));
 }
 
 
@@ -171,18 +195,20 @@ bool Counters::hasCounter(docstring const & c) const
 }
 
 
-bool Counters::read(Lexer & lex, docstring const & name)
+bool Counters::read(Lexer & lex, docstring const & name, bool makenew)
 {
        if (hasCounter(name)) {
                LYXERR(Debug::TCLASS, "Reading existing counter " << to_utf8(name));
                return counterList_[name].read(lex);
        }
+
        LYXERR(Debug::TCLASS, "Reading new counter " << to_utf8(name));
        Counter cnt;
        bool success = cnt.read(lex);
-       if (success)
+       // if makenew is false, we will just discard what we read
+       if (success && makenew)
                counterList_[name] = cnt;
-       else
+       else if (!success)
                LYXERR0("Error reading counter `" << name << "'!");
        return success;
 }
@@ -224,7 +250,7 @@ int Counters::value(docstring const & ctr) const
 }
 
 
-void Counters::step(docstring const & ctr)
+void Counters::step(docstring const & ctr, UpdateType utype)
 {
        CounterList::iterator it = counterList_.find(ctr);
        if (it == counterList_.end()) {
@@ -234,6 +260,11 @@ void Counters::step(docstring const & ctr)
        }
 
        it->second.step();
+       if (utype == OutputUpdate) {
+               LASSERT(!counter_stack_.empty(), /* */);
+               counter_stack_.pop_back();
+               counter_stack_.push_back(ctr);
+       }
        it = counterList_.begin();
        CounterList::iterator const end = counterList_.end();
        for (; it != end; ++it) {
@@ -251,9 +282,12 @@ void Counters::reset()
        current_float_.erase();
        CounterList::iterator it = counterList_.begin();
        CounterList::iterator const end = counterList_.end();
-       for (; it != end; ++it) {
+       for (; it != end; ++it)
                it->second.reset();
-       }
+       counter_stack_.clear();
+       counter_stack_.push_back(from_ascii(""));
+       layout_stack_.clear();
+       layout_stack_.push_back(0);
 }
 
 
@@ -270,6 +304,22 @@ void Counters::reset(docstring const & match)
 }
 
 
+bool Counters::remove(docstring const & cnt)
+{
+       bool retval = counterList_.erase(cnt);
+       if (!retval)
+               return false;
+       CounterList::iterator it = counterList_.begin();
+       CounterList::iterator end = counterList_.end();
+       for (; it != end; ++it) {
+               if (it->second.checkAndRemoveMaster(cnt))
+                       LYXERR(Debug::TCLASS, "Removed master counter `" +
+                                       to_utf8(cnt) + "' from counter: " + to_utf8(it->first));
+       }
+       return retval;
+}
+
+
 void Counters::copy(Counters & from, Counters & to, docstring const & match)
 {
        CounterList::iterator it = counterList_.begin();
@@ -314,11 +364,10 @@ char hebrewCounter(int const n)
 }
 
 
-
-//On the special cases, see http://mathworld.wolfram.com/RomanNumerals.html
-//and for a list of roman numerals up to and including 3999, see 
-//http://www.research.att.com/~njas/sequences/a006968.txt. (Thanks to Joost
-//for this info.)
+// On the special cases, see http://mathworld.wolfram.com/RomanNumerals.html
+// and for a list of roman numerals up to and including 3999, see 
+// http://www.research.att.com/~njas/sequences/a006968.txt. (Thanks to Joost
+// for this info.)
 docstring const romanCounter(int const n)
 {
        static char const * const ones[9] = {
@@ -415,57 +464,89 @@ docstring Counters::labelItem(docstring const & ctr,
 }
 
 
-docstring Counters::theCounter(docstring const & counter) const
+docstring Counters::theCounter(docstring const & counter,
+                              string const & lang) const
 {
-       std::set<docstring> callers;
-       return theCounter(counter, callers);
+       CounterList::const_iterator it = counterList_.find(counter); 
+       if (it == counterList_.end())
+               return from_ascii("#");
+       Counter const & ctr = it->second;
+       Counter::StringMap & sm = ctr.flatLabelStrings(appendix());
+       Counter::StringMap::iterator smit = sm.find(lang);
+       if (smit != sm.end())
+               return counterLabel(smit->second, lang);
+
+       vector<docstring> callers;
+       docstring const & fls = flattenLabelString(counter, appendix(),
+                                                  lang, callers);
+       sm[lang] = fls;
+       return counterLabel(fls, lang);
 }
 
-docstring Counters::theCounter(docstring const & counter,
-                               std::set<docstring> & callers) const
+
+docstring Counters::flattenLabelString(docstring const & counter, 
+                                      bool in_appendix,
+                                      string const & lang,
+                                      vector<docstring> & callers) const
 {
        docstring label;
 
-       if (callers.find(counter) == callers.end()) {
-               
-               CounterList::const_iterator it = counterList_.find(counter); 
-               if (it == counterList_.end())
-                       return from_ascii("??");
-               Counter const & c = it->second;
-
-               docstring ls = appendix() ? c.labelStringAppendix() : c.labelString();
-
-               if (ls.empty()) {
-                       if (!c.master().empty())
-                               ls = from_ascii("\\the") + c.master() + from_ascii(".");
-                       ls += from_ascii("\\arabic{") + counter + "}";
-               }
-
-               pair<std::set<docstring>::iterator, bool> const result = callers.insert(counter);
-               label = counterLabel(ls, &callers);
-               callers.erase(result.first);
-       } else {
+       if (find(callers.begin(), callers.end(), counter) != callers.end()) {
                // recursion detected
                lyxerr << "Warning: Recursion in label for counter `"
-                          << counter << "' detected"
-                          << endl;
+                      << counter << "' detected"
+                      << endl;
+               return from_ascii("??");
+       }
+               
+       CounterList::const_iterator it = counterList_.find(counter); 
+       if (it == counterList_.end())
+               return from_ascii("#");
+       Counter const & c = it->second;
+
+       docstring ls = translateIfPossible(c.labelString(in_appendix), lang);
+
+       callers.push_back(counter);
+       if (ls.empty()) {
+               if (!c.master().empty())
+                       ls = flattenLabelString(c.master(), in_appendix, lang, callers) 
+                               + from_ascii(".");
+               callers.pop_back();
+               return ls + from_ascii("\\arabic{") + counter + "}";
        }
 
-       return label;
+       while (true) {
+               //lyxerr << "ls=" << to_utf8(ls) << endl;
+               size_t const i = ls.find(from_ascii("\\the"), 0);
+               if (i == docstring::npos)
+                       break;
+               size_t const j = i + 4;
+               size_t k = j;
+               while (k < ls.size() && lowercase(ls[k]) >= 'a' 
+                      && lowercase(ls[k]) <= 'z')
+                       ++k;
+               docstring const newc = ls.substr(j, k - j);
+               docstring const repl = flattenLabelString(newc, in_appendix,
+                                                         lang, callers);
+               ls.replace(i, k - j + 4, repl);
+       }
+       callers.pop_back();
+
+       return ls;
 }
 
 
 docstring Counters::counterLabel(docstring const & format,
-                                 std::set<docstring> * callers) const
+                                string const & lang) const
 {
        docstring label = format;
 
        // FIXME: Using regexps would be better, but we compile boost without
        // wide regexps currently.
-
+       docstring const the = from_ascii("\\the");
        while (true) {
-               //lyxerr << "label=" << to_utf8(label) << endl;
-               size_t const i = label.find(from_ascii("\\the"), 0);
+               //lyxerr << "label=" << label << endl;
+               size_t const i = label.find(the, 0);
                if (i == docstring::npos)
                        break;
                size_t const j = i + 4;
@@ -473,14 +554,11 @@ docstring Counters::counterLabel(docstring const & format,
                while (k < label.size() && lowercase(label[k]) >= 'a' 
                       && lowercase(label[k]) <= 'z')
                        ++k;
-               docstring const counter = label.substr(j, k - j);
-               docstring const repl = callers? theCounter(counter, *callers): 
-                                             theCounter(counter);
-               label.replace(i, k - j + 4, repl);
+               docstring const newc(label, j, k - j);
+               label.replace(i, k - i, theCounter(newc, lang));
        }
-
        while (true) {
-               //lyxerr << "label=" << to_utf8(label) << endl;
+               //lyxerr << "label=" << label << endl;
 
                size_t const i = label.find('\\', 0);
                if (i == docstring::npos)
@@ -493,13 +571,76 @@ docstring Counters::counterLabel(docstring const & format,
                        break;
                docstring const numbertype(label, i + 1, j - i - 1);
                docstring const counter(label, j + 1, k - j - 1);
-               docstring const rep = labelItem(counter, numbertype);
-               label = docstring(label, 0, i) + rep
-                       + docstring(label, k + 1, docstring::npos);
+               label.replace(i, k + 1 - i, labelItem(counter, numbertype));
        }
-       //lyxerr << "DONE! label=" << to_utf8(label) << endl;
+       //lyxerr << "DONE! label=" << label << endl;
        return label;
 }
 
 
+docstring Counters::prettyCounter(docstring const & name,
+                              string const & lang) const
+{
+       CounterList::const_iterator it = counterList_.find(name); 
+       if (it == counterList_.end())
+               return from_ascii("#");
+       Counter const & ctr = it->second;
+
+       docstring const value = theCounter(name, lang);
+       docstring const & format =
+           translateIfPossible(ctr.prettyFormat(), lang);
+       if (format.empty())
+               return value;
+       return subst(format, from_ascii("##"), value);
+}
+
+
+docstring Counters::currentCounter() const
+{ 
+       LASSERT(!counter_stack_.empty(), /* */);
+       return counter_stack_.back(); 
+}
+
+
+void Counters::setActiveLayout(Layout const & lay)
+{
+       LASSERT(!layout_stack_.empty(), return);
+       Layout const * const lastlay = layout_stack_.back();
+       // we want to check whether the layout has changed and, if so,
+       // whether we are coming out of or going into an environment.
+       if (!lastlay) { 
+               layout_stack_.pop_back();
+               layout_stack_.push_back(&lay);
+               if (lay.isEnvironment())
+                       beginEnvironment();
+       } else if (lastlay->name() != lay.name()) {
+               layout_stack_.pop_back();
+               layout_stack_.push_back(&lay);
+               if (lastlay->isEnvironment()) {
+                       // we are coming out of an environment
+                       // LYXERR0("Out: " << lastlay->name());
+                       endEnvironment();
+               }
+               if (lay.isEnvironment()) {
+                       // we are going into a new environment
+                       // LYXERR0("In: " << lay.name());
+                       beginEnvironment();
+               }
+       } 
+}
+
+
+void Counters::beginEnvironment()
+{
+       counter_stack_.push_back(counter_stack_.back());
+}
+
+
+void Counters::endEnvironment()
+{
+       LASSERT(!counter_stack_.empty(), return);
+       counter_stack_.pop_back();
+}
+
+
 } // namespace lyx