source/filter.cpp

   1 #include <msp/core/algorithm.h>
   2 #include <msp/core/maputils.h>
   3 #include <msp/io/print.h>
   4 #include "category.h"
   5 #include "filter.h"
   6 #include "poefilter.h"
   7
   8 using namespace std;
   9 using namespace Msp;
  10
  11 FilterStatement::FilterStatement():
  12         show(true)
  13 { }
  14
  15 void FilterStatement::set_show(bool s)
  16 {
  17         show = s;
  18 }
  19
  20 void FilterStatement::add_line(const string &l)
  21 {
  22         lines.push_back(l);
  23 }
  24
  25 void FilterStatement::add_line(list<FilterStatement> &st, const string &line)
  26 {
  27         for(list<FilterStatement>::iterator i=st.begin(); i!=st.end(); ++i)
  28                 i->add_line(line);
  29 }
  30
  31 void FilterStatement::write(IO::Base &out) const
  32 {
  33         if(show)
  34                 out.write("Show\n");
  35         else
  36                 out.write("Hide\n");
  37
  38         for(list<string>::const_iterator i=lines.begin(); i!=lines.end(); ++i)
  39                 IO::print(out, "\t%s\n", *i);
  40 }
  41
  42
  43 Filter::Filter():
  44         abstract(false)
  45 { }
  46
  47 void Filter::write(IO::Base &out) const
  48 {
  49         for(list<Block>::const_iterator i=blocks.begin(); i!=blocks.end(); ++i)
  50         {
  51                 list<FilterStatement> st;
  52                 st.push_back(FilterStatement());
  53                 st.back().set_show(i->show);
  54                 i->appearance.add_lines(st.back());
  55                 i->category->create_statements(st);
  56
  57                 IO::print(out, "# %s\n", i->category->get_name());
  58                 for(list<FilterStatement>::const_iterator j=st.begin(); j!=st.end(); ++j)
  59                         j->write(out);
  60         }
  61 }
  62
  63
  64 Filter::Loader::Loader(Filter &f, const PoeFilter &p):
  65         DataFile::ObjectLoader<Filter>(f),
  66         poe(p)
  67 {
  68         add("abstract", &Filter::abstract);
  69         add("cancel", &Loader::cancel);
  70         add("hide", &Loader::hide);
  71         add("include", &Loader::include);
  72         add("show", &Loader::show);
  73 }
  74
  75 string Filter::Loader::glob_to_re(const string &glob)
  76 {
  77         string result = "^";
  78         for(string::const_iterator i=glob.begin(); i!=glob.end(); ++i)
  79         {
  80                 if(*i=='*')
  81                         result += '.';
  82                 else if(!isalnum(*i))
  83                         result += '\\';
  84                 result += *i;
  85         }
  86         result += '$';
  87
  88         return result;
  89 }
  90
  91 bool Filter::Loader::category_order(const Category *c1, const Category *c2)
  92 {
  93         return c1->get_order()<c2->get_order();
  94 }
  95
  96 void Filter::Loader::add_categories(const string &name, bool show)
  97 {
  98         vector<const Category *> categs;
  99         if(name.find('*')!=string::npos)
 100         {
 101                 poe.find_categories(glob_to_re(name), categs);
 102
 103                 if(categs.empty())
 104                         throw key_error(name);
 105
 106                 sort(categs, &category_order);
 107         }
 108         else
 109                 categs.push_back(&poe.get_category(name));
 110
 111         for(vector<const Category *>::const_iterator i=categs.begin(); i!=categs.end(); ++i)
 112         {
 113                 bool found = false;
 114                 for(list<Block>::const_iterator j=obj.blocks.begin(); (!found && j!=obj.blocks.end()); ++j)
 115                         found = (j->category==*i);
 116
 117                 if(!found)
 118                 {
 119                         Block blk;
 120                         blk.show = show;
 121                         blk.category = *i;
 122                         blk.appearance = (*i)->get_appearance();
 123                         if(const Appearance *app = poe.get_theme().find_appearance((*i)->get_name()))
 124                                 blk.appearance.merge_from(*app);
 125                         obj.blocks.push_back(blk);
 126                 }
 127         }
 128 }
 129
 130 void Filter::Loader::cancel(const string &name)
 131 {
 132         vector<const Category *> categs;
 133         if(name.find('*')!=string::npos)
 134         {
 135                 poe.find_categories(glob_to_re(name), categs);
 136
 137                 if(categs.empty())
 138                         throw key_error(name);
 139         }
 140         else
 141                 categs.push_back(&poe.get_category(name));
 142
 143         for(vector<const Category *>::const_iterator i=categs.begin(); i!=categs.end(); ++i)
 144                 for(list<Block>::iterator j=obj.blocks.begin(); j!=obj.blocks.end(); ++j)
 145                         if(j->category==*i)
 146                         {
 147                                 obj.blocks.erase(j);
 148                                 break;
 149                         }
 150 }
 151
 152 void Filter::Loader::hide(const string &name)
 153 {
 154         add_categories(name, false);
 155 }
 156
 157 void Filter::Loader::include(const string &name)
 158 {
 159         const Filter &base = poe.get_filter(name);
 160         obj.blocks.insert(obj.blocks.end(), base.blocks.begin(), base.blocks.end());
 161 }
 162
 163 void Filter::Loader::show(const string &name)
 164 {
 165         add_categories(name, true);
 166 }