lily/figured-bass-engraver.cc

   1 /*
   2   figured-bass-engraver.cc -- implement Figured_bass_engraver
   3
   4   source file of the GNU LilyPond music typesetter
   5
   6   (c) 2005--2006 Han-Wen Nienhuys <hanwen@xs4all.nl>
   7
   8 */
   9
  10 #include "engraver.hh"
  11
  12 #include "context.hh"
  13 #include "music.hh"
  14 #include "item.hh"
  15 #include "spanner.hh"
  16 #include "axis-group-interface.hh"
  17 #include "align-interface.hh"
  18 #include "pointer-group-interface.hh"
  19 #include "text-interface.hh"
  20 #include "grob-array.hh"
  21
  22
  23 #include "translator.icc"
  24
  25 struct Figure_group
  26 {
  27   Spanner *group_;
  28   Spanner *continuation_line_;
  29
  30   SCM number_;
  31   SCM alteration_;
  32
  33   Item *figure_item_;
  34   Music *current_music_;
  35   bool force_no_continuation_;
  36
  37   Figure_group ()
  38   {
  39     figure_item_ = 0;
  40     force_no_continuation_ = false;
  41     continuation_line_ = 0;
  42     number_ = SCM_EOL;
  43     alteration_ = SCM_EOL;
  44     group_ = 0;
  45     current_music_ = 0;
  46   }
  47   bool is_continuation () const
  48   {
  49     return
  50       current_music_
  51       && !force_no_continuation_
  52       && ly_is_equal (number_,
  53                       current_music_->get_property ("figure"))
  54       && ly_is_equal (alteration_,
  55                       current_music_->get_property ("alteration"));
  56   }
  57 };
  58
  59 struct Figured_bass_engraver : public Engraver
  60 {
  61   TRANSLATOR_DECLARATIONS(Figured_bass_engraver);
  62   void clear_spanners();
  63   void add_brackets ();
  64   void create_grobs ();
  65
  66   void center_continuations (vector<Spanner*> const &consecutive_lines);
  67   void center_repeated_continuations ();
  68 protected:
  69   vector<Figure_group> groups_;
  70   Spanner *alignment_;
  71   vector<Music*> new_musics_;
  72   bool continuation_;
  73   bool new_music_found_;
  74
  75   Moment stop_moment_;
  76   Music *rest_event_;
  77
  78   virtual bool try_music (Music *);
  79   virtual void derived_mark () const;
  80
  81   void start_translation_timestep ();
  82   void stop_translation_timestep ();
  83   void process_music ();
  84 };
  85
  86 void
  87 Figured_bass_engraver::derived_mark () const
  88 {
  89   for (vsize i = 0; i < groups_.size (); i++)
  90     {
  91       scm_gc_mark (groups_[i].number_);
  92       scm_gc_mark (groups_[i].alteration_);
  93     }
  94 }
  95
  96 void
  97 Figured_bass_engraver::stop_translation_timestep ()
  98 {
  99   if (groups_.empty ()
 100       || now_mom ().main_part_ < stop_moment_.main_part_
 101       || now_mom ().grace_part_ < Rational (0))
 102     return ;
 103
 104   bool found = false;
 105   for (vsize i = 0; !found && i < groups_.size (); i++)
 106     found  = found  || groups_[i].current_music_;
 107
 108   if (!found)
 109     clear_spanners ();
 110 }
 111
 112 Figured_bass_engraver::Figured_bass_engraver ()
 113 {
 114   alignment_ = 0;
 115   continuation_ = false;
 116   rest_event_ = 0;
 117   new_music_found_ = false;
 118 }
 119
 120 void
 121 Figured_bass_engraver::start_translation_timestep ()
 122 {
 123   if (now_mom ().main_part_ < stop_moment_.main_part_
 124       || now_mom ().grace_part_ < Rational (0))
 125     return ;
 126
 127   rest_event_ = 0;
 128   new_musics_.clear ();
 129   for (vsize i = 0; i < groups_.size (); i++)
 130     groups_[i].current_music_ = 0;
 131   continuation_ = false;
 132 }
 133
 134 bool
 135 Figured_bass_engraver::try_music (Music *m)
 136 {
 137   new_music_found_ = true;
 138   if (m->is_mus_type ("rest-event"))
 139     {
 140       rest_event_ = m;
 141       return true;
 142     }
 143   else
 144     {
 145       stop_moment_ = now_mom () + m->get_length ();
 146
 147       SCM fig = m->get_property ("figure");
 148       for (vsize i = 0; i < groups_.size (); i++)
 149         {
 150           if (!groups_[i].current_music_
 151               && ly_is_equal (groups_[i].number_, fig))
 152             {
 153               groups_[i].current_music_ = m;
 154               groups_[i].force_no_continuation_
 155                 = to_boolean (m->get_property ("no-continuation"));
 156               continuation_ = true;
 157               return true;
 158             }
 159         }
 160
 161       new_musics_.push_back (m);
 162
 163       return true;
 164     }
 165 }
 166
 167 void
 168 Figured_bass_engraver::center_continuations (vector<Spanner*> const &consecutive_lines)
 169 {
 170   if (consecutive_lines.size () == 2)
 171     {
 172       vector<Grob*> left_figs;
 173       for (vsize j = consecutive_lines.size(); j--;)
 174         left_figs.push_back (consecutive_lines[j]->get_bound (LEFT));
 175
 176       SCM  ga = Grob_array::make_array ();
 177       unsmob_grob_array (ga)->set_array (left_figs);
 178
 179       for (vsize j = consecutive_lines.size(); j--;)
 180         consecutive_lines[j]->set_object ("figures",
 181                                           unsmob_grob_array (ga)->smobbed_copy ());
 182     }
 183 }
 184
 185 void
 186 Figured_bass_engraver::center_repeated_continuations ()
 187 {
 188   vector<Spanner*> consecutive_lines;
 189   for (vsize i = 0; i <= groups_.size(); i++)
 190     {
 191       if (i < groups_.size ()
 192           && groups_[i].continuation_line_
 193           && (consecutive_lines.empty ()
 194               || (consecutive_lines[0]->get_bound(LEFT)->get_column ()
 195                   == groups_[i].continuation_line_->get_bound (LEFT)->get_column ()
 196                   && consecutive_lines[0]->get_bound(RIGHT)->get_column ()
 197                   == groups_[i].continuation_line_->get_bound (RIGHT)->get_column ())))
 198         consecutive_lines.push_back (groups_[i].continuation_line_);
 199       else
 200         {
 201           center_continuations (consecutive_lines);
 202           consecutive_lines.clear ();
 203         }
 204     }
 205 }
 206
 207 void
 208 Figured_bass_engraver::clear_spanners ()
 209 {
 210   if (!alignment_)
 211     return;
 212
 213   alignment_ = 0;
 214   if (to_boolean (get_property ("figuredBassCenterContinuations")))
 215     center_repeated_continuations();
 216
 217   groups_.clear ();
 218 }
 219
 220 void
 221 Figured_bass_engraver::add_brackets ()
 222 {
 223   vector<Grob*> encompass;
 224   bool inside = false;
 225   for (vsize i = 0; i < groups_.size (); i ++)
 226     {
 227       if (!groups_[i].current_music_)
 228         continue;
 229
 230       if (to_boolean (groups_[i].current_music_->get_property ("bracket-start")))
 231         inside = true;
 232
 233       if (inside && groups_[i].figure_item_)
 234         encompass.push_back (groups_[i].figure_item_);
 235
 236        if (to_boolean (groups_[i].current_music_->get_property ("bracket-stop")))
 237         {
 238           inside = false;
 239
 240           Item * brack = make_item ("BassFigureBracket", groups_[i].current_music_->self_scm ());
 241           for (vsize j = 0; j < encompass.size (); j++)
 242             {
 243               Pointer_group_interface::add_grob (brack,
 244                                                  ly_symbol2scm ("elements"),
 245                                                  encompass[j]);
 246             }
 247           encompass.clear ();
 248         }
 249     }
 250 }
 251
 252 void
 253 Figured_bass_engraver::process_music ()
 254 {
 255   if (rest_event_)
 256     {
 257       clear_spanners ();
 258       return;
 259     }
 260
 261   if (!continuation_
 262       && new_musics_.empty ())
 263     {
 264       clear_spanners ();
 265       return;
 266     }
 267
 268   if (!new_music_found_)
 269     return;
 270
 271   new_music_found_ = false;
 272
 273   /*
 274     Don't need to sync alignments, if we're not using extenders.
 275    */
 276   bool use_extenders = to_boolean (get_property ("useBassFigureExtenders"));
 277   if (!use_extenders)
 278     {
 279       if (to_boolean (get_property ("figuredBassCenterContinuations")))
 280         center_repeated_continuations ();
 281
 282       alignment_ = 0;
 283       for (vsize i = 0; i < groups_.size (); i++)
 284         {
 285           groups_[i].group_ = 0;
 286           groups_[i].continuation_line_ = 0;
 287         }
 288     }
 289
 290   if (!continuation_)
 291     clear_spanners ();
 292
 293   vsize k = 0;
 294   for (vsize i = 0; i < new_musics_.size (); i++)
 295     {
 296       while (k < groups_.size ()
 297              && groups_[k].current_music_)
 298         k++;
 299
 300       if (k >= groups_.size ())
 301         {
 302           Figure_group group;
 303           groups_.push_back (group);
 304         }
 305
 306       groups_[k].current_music_ = new_musics_[i];
 307       groups_[k].figure_item_ = 0;
 308       k++;
 309     }
 310
 311   for (vsize i = 0; i < groups_.size (); i++)
 312     {
 313       if (!groups_[i].is_continuation ())
 314         {
 315           groups_[i].number_ = SCM_BOOL_F;
 316           groups_[i].alteration_ = SCM_BOOL_F;
 317         }
 318     }
 319
 320   if (use_extenders)
 321     {
 322       vector<int> junk_continuations;
 323       for (vsize i = 0; i < groups_.size(); i++)
 324         {
 325                 Figure_group &group = groups_[i];
 326
 327           if (group.is_continuation ())
 328             {
 329               if (!group.continuation_line_)
 330                 {
 331                   Spanner * line = make_spanner ("BassFigureContinuation", SCM_EOL);
 332                   Item * item = group.figure_item_;
 333                   group.continuation_line_ = line;
 334                   line->set_bound (LEFT, item);
 335
 336                   /*
 337                     Don't add as child. This will cache the wrong
 338                     (pre-break) stencil when callbacks are triggered.
 339                   */
 340                   line->set_parent (group.group_, Y_AXIS);
 341                   Pointer_group_interface::add_grob (line, ly_symbol2scm ("figures"), item);
 342
 343                   group.figure_item_ = 0;
 344                 }
 345             }
 346           else if (group.continuation_line_)
 347             junk_continuations.push_back (i);
 348         }
 349
 350       /*
 351         Ugh, repeated code.
 352        */
 353       vector<Spanner*> consecutive;
 354       if (to_boolean (get_property ("figuredBassCenterContinuations")))
 355         {
 356           for (vsize i = 0; i <= junk_continuations.size (); i++)
 357             {
 358               if (i < junk_continuations.size()
 359                   && (i == 0 || junk_continuations[i-1] == junk_continuations[i] - 1))
 360                 consecutive.push_back (groups_[junk_continuations[i]].continuation_line_);
 361               else
 362                 {
 363                   center_continuations (consecutive);
 364                   consecutive.clear ();
 365                   if (i < junk_continuations.size ())
 366                     consecutive.push_back (groups_[junk_continuations[i]].continuation_line_);
 367                 }
 368             }
 369         }
 370       for (vsize i = 0; i < junk_continuations.size (); i++)
 371         groups_[junk_continuations[i]].continuation_line_ = 0;
 372     }
 373
 374   create_grobs ();
 375   add_brackets ();
 376 }
 377
 378 void
 379 Figured_bass_engraver::create_grobs ()
 380 {
 381   Grob *muscol = dynamic_cast<Item*> (unsmob_grob (get_property ("currentMusicalColumn")));
 382   if (!alignment_)
 383     {
 384       alignment_ = make_spanner ("BassFigureAlignment", SCM_EOL);
 385       alignment_->set_bound (LEFT, muscol);
 386     }
 387   alignment_->set_bound (RIGHT, muscol);
 388
 389   SCM proc = get_property ("figuredBassFormatter");
 390   for (vsize i = 0; i < groups_.size(); i++)
 391     {
 392       Figure_group &group = groups_[i];
 393
 394       if (group.current_music_)
 395         {
 396           Item *item
 397             = make_item ("BassFigure",
 398                          group.current_music_->self_scm ());
 399
 400
 401           SCM fig = group.current_music_->get_property ("figure");
 402           if (!group.group_)
 403             {
 404               group.group_ = make_spanner ("BassFigureLine", SCM_EOL);
 405               group.group_->set_bound (LEFT, muscol);
 406               Align_interface::add_element (alignment_,
 407                                             group.group_);
 408             }
 409
 410           if (scm_memq (fig, get_property ("implicitBassFigures")) != SCM_BOOL_F)
 411             {
 412               item->set_property ("transparent", SCM_BOOL_T);
 413               item->set_property ("implicit", SCM_BOOL_T);
 414             }
 415
 416           group.number_ = fig;
 417           group.alteration_ = group.current_music_->get_property ("alteration");
 418
 419           SCM text = group.current_music_->get_property ("text");
 420           if (!Text_interface::is_markup (text)
 421               && ly_is_procedure (proc))
 422             {
 423               text = scm_call_3 (proc, fig, group.current_music_->self_scm (),
 424                                  context ()->self_scm ());
 425             }
 426
 427           item->set_property ("text", text);
 428
 429           Axis_group_interface::add_element (group.group_, item);
 430           group.figure_item_ = item;
 431         }
 432
 433       if (group.continuation_line_)
 434         {
 435           /*
 436             UGH should connect to the bass staff, and get the note heads.
 437           */
 438           group.figure_item_->set_property ("transparent", SCM_BOOL_T);
 439           group.continuation_line_->set_bound (RIGHT, group.figure_item_);
 440         }
 441
 442
 443       if (groups_[i].group_)
 444         groups_[i].group_->set_bound (RIGHT, muscol);
 445     }
 446
 447 }
 448
 449 ADD_TRANSLATOR (Figured_bass_engraver,
 450                 /* doc */
 451
 452                 "Make figured bass numbers.",
 453                 /* create */
 454                 "BassFigure "
 455                 "BassFigureAlignment "
 456                 "BassFigureBracket "
 457                 "BassFigureContinuation "
 458                 "BassFigureLine "
 459                 ,
 460                 /* accept */
 461                 "bass-figure-event rest-event",
 462
 463                 /* read */
 464                 "figuredBassAlterationDirection "
 465                 "figuredBassCenterContinuations "
 466                 "figuredBassFormatter "
 467                 "implicitBassFigures "
 468                 "useBassFigureExtenders "
 469                 ,
 470
 471                 /* write */
 472                 "");