]> git.donarmstrong.com Git - lilypond.git/blob - lily/beam.cc
release: 0.1.57
[lilypond.git] / lily / beam.cc
1 /*
2   beam.cc -- implement Beam
3
4   source file of the GNU LilyPond music typesetter
5
6   (c) 1997 Han-Wen Nienhuys <hanwen@stack.nl>
7
8   TODO
9
10   Less hairy code.  knee: ([\stem 1; c8 \stem -1; c8]
11
12 */
13
14 #include <math.h>
15
16 #include "p-col.hh"
17 #include "varray.hh"
18 #include "proto.hh"
19 #include "dimen.hh"
20 #include "beam.hh"
21 #include "abbreviation-beam.hh"
22 #include "misc.hh"
23 #include "debug.hh"
24 #include "atom.hh"
25 #include "molecule.hh"
26 #include "leastsquares.hh"
27 #include "stem.hh"
28 #include "paper-def.hh"
29 #include "lookup.hh"
30 #include "grouping.hh"
31 #include "stem-info.hh"
32 #include "main.hh"  // experimental features
33
34
35 IMPLEMENT_IS_TYPE_B1 (Beam, Spanner);
36
37 // ugh, hardcoded
38 const Real MINIMUM_STEMLEN[] = {
39   0, // just in case
40   5, 
41   4,
42   3,
43   2,
44   2,
45 };
46
47 Beam::Beam ()
48 {
49   slope_f_ = 0;
50   left_y_ = 0.0;
51   damping_i_ = 1;
52   quantisation_ = NORMAL;
53   multiple_i_ = 0;
54 }
55
56 void
57 Beam::add (Stem*s)
58 {
59   stems_.push (s);
60   s->add_dependency (this);
61   s->beam_l_ = this;
62
63   if (!spanned_drul_[LEFT])
64     set_bounds (LEFT,s);
65   else
66     set_bounds (RIGHT,s);
67 }
68
69 Molecule*
70 Beam::brew_molecule_p () const
71 {
72   Molecule *mol_p = new Molecule;
73   Real inter_f = paper ()->internote_f ();
74   Real x0 = stems_[0]->hpos_f ();
75   for (int j=0; j <stems_.size (); j++)
76     {
77       Stem *i = stems_[j];
78       Stem * prev = (j > 0)? stems_[j-1] : 0;
79       Stem * next = (j < stems_.size ()-1) ? stems_[j+1] :0;
80
81       Molecule sb = stem_beams (i, next, prev);
82       Real  x = i->hpos_f ()-x0;
83       sb.translate (Offset (x, (x * slope_f_  + left_y_)* inter_f));
84       mol_p->add (sb);
85     }
86   mol_p->translate_axis (x0 - spanned_drul_[LEFT]->absolute_coordinate (X_AXIS), X_AXIS);
87   return mol_p;
88 }
89
90 Offset
91 Beam::center () const
92 {
93   Real w= (paper ()->note_width () + width ().length ())/2.0;
94   return Offset (w, (left_y_ + w* slope_f_)*paper ()->internote_f ());
95 }
96
97 void
98 Beam::do_pre_processing ()
99 {
100   if (!dir_)
101     set_default_dir ();
102 }
103
104 void
105 Beam::do_print () const
106 {
107 #ifndef NPRINT
108   DOUT << "slope_f_ " <<slope_f_ << "left ypos " << left_y_;
109   Spanner::do_print ();
110 #endif
111 }
112
113 void
114 Beam::do_post_processing ()
115 {
116   if (stems_.size () < 2)
117     {
118       warning (_ ("Beam with less than 2 stems"));
119       transparent_b_ = true;
120       return ;
121     }
122   solve_slope ();
123   set_stemlens ();
124 }
125
126 void
127 Beam::do_substitute_dependent (Score_elem*o,Score_elem*n)
128 {
129   if (o->is_type_b (Stem::static_name ()))
130       stems_.substitute ((Stem*)o->item (),  n? (Stem*) n->item ():0);
131 }
132
133 Interval
134 Beam::do_width () const
135 {
136   return Interval (stems_[0]->hpos_f (),
137                    stems_.top ()->hpos_f ());
138 }
139
140 void
141 Beam::set_default_dir ()
142 {
143   Drul_array<int> total;
144   total[UP]  = total[DOWN] = 0;
145   Drul_array<int> count; 
146   count[UP]  = count[DOWN] = 0;
147   Direction d = DOWN;
148   
149   for (int i=0; i <stems_.size (); i++)
150     do {
151       Stem *s = stems_[i];
152       int current = s->dir_ 
153         ? (1 + d * s->dir_)/2
154         : s->get_center_distance ((Direction)-d);
155
156       if (current)
157         {
158           total[d] += current;
159           count[d] ++;
160         }
161
162     } while (flip(&d) != DOWN);
163   
164    do {
165     if (!total[d])
166       count[d] = 1;
167   } while (flip(&d) != DOWN);
168   
169   /* 
170
171      [Ross] states that the majority of the notes dictates the
172      direction (and not the mean of "center distance")
173   */
174   dir_ = (total[UP] > total[DOWN]) ? UP : DOWN;
175
176   for (int i=0; i <stems_.size (); i++)
177     {
178       Stem *sl = stems_[i];
179       sl->dir_ = dir_;
180     }
181 }
182
183 /*
184   should use minimum energy formulation (cf linespacing)
185 */
186 void
187 Beam::solve_slope ()
188 {
189   Array<Stem_info> sinfo;
190   for (int j=0; j <stems_.size (); j++)
191     {
192       Stem *i = stems_[j];
193
194       i->set_default_extents ();
195       if (i->invisible_b ())
196         continue;
197
198       Stem_info info (i);
199       sinfo.push (info);
200     }
201   if (! sinfo.size ())
202     slope_f_ = left_y_ = 0;
203   else if (sinfo.size () == 1)
204     {
205       slope_f_ = 0;
206       left_y_ = sinfo[0].idealy_f_;
207     }
208   else
209     {
210
211       Real leftx = sinfo[0].x;
212       Least_squares l;
213       for (int i=0; i < sinfo.size (); i++)
214         {
215           sinfo[i].x -= leftx;
216           l.input.push (Offset (sinfo[i].x, sinfo[i].idealy_f_));
217         }
218
219       l.minimise (slope_f_, left_y_);
220     }
221
222   Real dy = 0.0;
223   for (int i=0; i < sinfo.size (); i++)
224     {
225       Real y = sinfo[i].x * slope_f_ + left_y_;
226       Real my = sinfo[i].miny_f_;
227
228       if (my - y > dy)
229         dy = my -y;
230     }
231   left_y_ += dy;
232   left_y_ *= dir_;
233
234   slope_f_ *= dir_;
235
236   /*
237     This neat trick is by Werner Lemberg, damped = tanh (slope_f_) corresponds
238     with some tables in [Wanske]
239     */
240   if (damping_i_)
241     slope_f_ = 0.6 * tanh (slope_f_) / damping_i_;
242
243   quantise_yspan ();
244
245   // y-values traditionally use internote dimension: therefore slope = (y/in)/x
246   // but mf and beam-lookup use PT dimension for y (as used for x-values)
247   // ugh --- there goes our simplified but careful quantisation
248   Real sl = slope_f_ * paper ()->internote_f ();
249   paper ()->lookup_l ()->beam (sl, 20 PT, 1 PT);
250   slope_f_ = sl / paper ()->internote_f ();
251 }
252
253 void
254 Beam::quantise_yspan ()
255 {
256   /*
257     [Ross] (simplification of)
258     Try to set slope_f_ complying with y-span of:
259       - zero
260       - beam_thickness / 2 + staffline_thickness / 2
261       - beam_thickness + staffline_thickness
262     + n * interline
263     */
264
265   if (!quantisation_)
266     return;
267
268   Real interline_f = paper ()->interline_f ();
269   Real internote_f = interline_f / 2;
270   Real staffline_thickness = paper ()->rule_thickness ();
271   Real beam_thickness = 0.48 * (interline_f - staffline_thickness);
272
273   const int QUANTS = 3;
274   Real qdy[QUANTS] = {
275     0,
276     beam_thickness / 2 + staffline_thickness / 2,
277     beam_thickness + staffline_thickness
278   };
279
280   Real xspan_f = stems_.top ()->hpos_f () - stems_[0]->hpos_f ();
281   // y-values traditionally use internote dimension: therefore slope = (y/in)/x
282   Real yspan_f = xspan_f * abs (slope_f_ * internote_f);
283   int yspan_i = (int)(yspan_f / interline_f);
284   Real q = (yspan_f / interline_f - yspan_i) * interline_f;
285   int i = 0;
286   for (; i < QUANTS - 1; i++)
287     if ((q >= qdy[i]) && (q <= qdy[i + 1]))
288       {
289         if (q - qdy[i] < qdy[i + 1] - q)
290           break;
291         else
292         { 
293           i++;
294           break;
295         }
296       }
297   q = qdy[i];
298
299   yspan_f = (Real)yspan_i * interline_f + q;
300   // y-values traditionally use internote dimension: therefore slope = (y/in)/x
301   slope_f_ = yspan_f / xspan_f / internote_f * sign (slope_f_);
302 }
303
304 void
305 Beam::quantise_left_y (Beam::Pos pos, bool extend_b)
306 {
307   /*
308    quantising left y should suffice, as slope is quantised too
309    if extend then stems must not get shorter
310    */
311
312   if (!quantisation_)
313     return;
314
315   Real interline_f = paper ()->interline_f ();
316   Real internote_f = interline_f / 2;
317   Real staffline_thickness = paper ()->rule_thickness ();
318   Real beam_thickness = 0.48 * (interline_f - staffline_thickness);
319
320   const int QUANTS = 7;
321   Real qy[QUANTS] = 
322   {
323     0,
324     beam_thickness / 2,
325     beam_thickness,
326     interline_f / 2 + beam_thickness / 2 + staffline_thickness / 2,
327     interline_f,
328     interline_f + beam_thickness / 2,
329     interline_f + beam_thickness
330   };
331   /* 
332    ugh, using i triggers gcc 2.7.2.1 internal compiler error (far down):
333    for (int i = 0; i < QUANTS; i++)
334    */
335   
336   // fixme!
337   for (int ii = 0; ii < QUANTS; ii++)
338     qy[ii] -= 0.5 *beam_thickness;
339   Pos qpos[QUANTS] = 
340   {
341     HANG,
342     STRADDLE,
343     SIT,
344     INTER,
345     HANG,
346     STRADDLE,
347     SIT
348   };
349
350   // y-values traditionally use internote dimension
351   Real y = left_y_ * internote_f;
352   int y_i = (int)floor(y / interline_f);
353   y = (y / interline_f - y_i) * interline_f;
354
355   if (y < 0)
356     for (int ii = 0; ii < QUANTS; ii++)
357       qy[ii] -= interline_f;
358
359   int lower_i = 0;
360   int i = 0;
361   for (; i < QUANTS; i++)
362     {
363       if (qy[i] > y)
364         break;
365       // found if lower_i is allowed, and nearer (from below) y than new pos
366       if ((pos & qpos[lower_i]) && (y - qy[lower_i] < y - qy[i]))
367         break;
368       // if new pos is allowed or old pos isn't: assign new pos
369       if ((pos & qpos[i]) || !(pos & qpos[lower_i]))
370         lower_i = i;
371     }
372
373   int upper_i = QUANTS - 1;
374   for (i = QUANTS - 1; i >= 0; i--)
375     {
376       if (qy[i] < y)
377         break;
378       // found if upper_i is allowed, and nearer (from above) y than new pos
379       if ((pos & qpos[upper_i]) && (qy[upper_i] - y < qy[i] - y))
380         break;
381       // if new pos is allowed or old pos isn't: assign new pos
382       if ((pos & qpos[i]) || !(pos & qpos[upper_i]))
383         upper_i = i;
384     }
385
386   // y-values traditionally use internote dimension
387   Real upper_y = (qy[upper_i] + interline_f * y_i) / internote_f;
388   Real lower_y = (qy[lower_i] + interline_f * y_i) / internote_f;
389
390   if (extend_b)
391     left_y_ = (dir_ > 0 ? upper_y : lower_y);
392   else
393     left_y_ = (upper_y - y < y - lower_y ? upper_y : lower_y);
394 }
395
396 void
397 Beam::set_stemlens ()
398 {
399   Real x0 = stems_[0]->hpos_f ();
400   Real dy = 0;
401
402   Real interline_f = paper ()->interline_f ();
403   Real internote_f = interline_f / 2;
404   Real staffline_thickness = paper ()->rule_thickness ();
405   Real beam_thickness = 0.48 * (interline_f - staffline_thickness);
406   Real interbeam_f = paper ()->interbeam_f ();
407   if (multiple_i_ > 3)
408     interbeam_f += 2.0 * staffline_thickness / 4;
409   Real xspan_f = stems_.top ()->hpos_f () - stems_[0]->hpos_f ();
410   /*
411    ugh, y values are in "internote" dimension
412    */
413   Real yspan_f = xspan_f * abs (slope_f_ * internote_f);
414   int yspan_i = (int)(yspan_f / interline_f);
415
416   Pos left_pos = NONE;
417
418   if ((yspan_f < staffline_thickness / 2) || (quantisation_ == NORMAL))
419     left_pos = (Pos)(STRADDLE | SIT | HANG);
420   else
421     left_pos = (Pos) (sign (slope_f_) > 0 ? STRADDLE | HANG 
422       : SIT | STRADDLE);
423
424   /* 
425    ugh, slope currently mangled by availability mf chars...
426    be more generous regarding beam position between stafflines
427    */
428   Real q = (yspan_f / interline_f - yspan_i) * interline_f;
429   if ((quantisation_ < NORMAL) && (q < interline_f / 3 - beam_thickness / 2))
430     left_pos = (Pos) (left_pos | INTER);
431
432
433   if (multiple_i_ > 1)
434     left_pos = (Pos) (dir_ > 0 ? HANG : SIT);
435
436   // ugh, rounding problems! (enge floots)
437   const Real EPSILON = interline_f / 10;
438   do
439     { 
440       left_y_ += dy * dir_;
441       quantise_left_y (left_pos, dy);
442       dy = 0;
443       for (int i=0; i < stems_.size (); i++)
444         {
445           Stem *s = stems_[i];
446           if (s->transparent_b_)
447             continue;
448
449           Real x = s->hpos_f () - x0;
450           s->set_stemend (left_y_ + slope_f_ * x);
451           Real y = s->stem_length_f ();
452           // duh: 
453           int mult_i = stems_[i]->beams_left_i_ >? stems_[i]->beams_right_i_;
454           if (mult_i > 1)
455               // dim(y) = internote
456               y -= (Real)(mult_i - 1) * interbeam_f / internote_f;
457           if (y < MINIMUM_STEMLEN[mult_i])
458             dy = dy >? (MINIMUM_STEMLEN[mult_i] - y);
459         }
460     } while (abs (dy) > EPSILON);
461 }
462
463 void
464 Beam::set_grouping (Rhythmic_grouping def, Rhythmic_grouping cur)
465 {
466   def.OK ();
467   cur.OK ();
468   assert (cur.children.size () == stems_.size ());
469
470   cur.split (def);
471
472   Array<int> b;
473   {
474     Array<int> flags;
475     for (int j=0; j <stems_.size (); j++)
476       {
477         Stem *s = stems_[j];
478
479         int f = s->flag_i_ - 2;
480         assert (f>0);
481         flags.push (f);
482       }
483     int fi =0;
484     b= cur.generate_beams (flags, fi);
485     b.insert (0,0);
486     b.push (0);
487     assert (stems_.size () == b.size ()/2);
488   }
489
490   for (int j=0, i=0; i < b.size () && j <stems_.size (); i+= 2, j++)
491     {
492       Stem *s = stems_[j];
493       s->beams_left_i_ = b[i];
494       s->beams_right_i_ = b[i+1];
495       multiple_i_ = multiple_i_ >? (b[i] >? b[i+1]);
496     }
497 }
498
499 /*
500   beams to go with one stem.
501   */
502 Molecule
503 Beam::stem_beams (Stem *here, Stem *next, Stem *prev) const
504 {
505   assert (!next || next->hpos_f () > here->hpos_f ());
506   assert (!prev || prev->hpos_f () < here->hpos_f ());
507
508   Real staffline_thickness = paper ()->rule_thickness ();
509   Real interbeam_f = paper ()->interbeam_f ();
510   Real internote_f =paper ()->internote_f (); 
511   Real interline_f = 2 * internote_f;
512   Real beamheight_f = 0.48 * (interline_f - staffline_thickness);
513   if (multiple_i_ > 3)
514     interbeam_f += 2.0 * staffline_thickness / 4;
515   Real dy = interbeam_f;
516   Real stemdx = staffline_thickness;
517   Real sl = slope_f_* internote_f;
518   paper ()->lookup_l ()->beam (sl, 20 PT, 1 PT);
519
520   Molecule leftbeams;
521   Molecule rightbeams;
522
523   /* half beams extending to the left. */
524   if (prev)
525     {
526       int lhalfs= lhalfs = here->beams_left_i_ - prev->beams_right_i_ ;
527       int lwholebeams= here->beams_left_i_ <? prev->beams_right_i_ ;
528       Real w = (here->hpos_f () - prev->hpos_f ())/4 <? paper ()->note_width ();;
529       Atom a;
530       if (lhalfs)               // generates warnings if not
531         a =  paper ()->lookup_l ()->beam (sl, w, beamheight_f);
532       a.translate (Offset (-w, -w * sl));
533       for (int j = 0; j  < lhalfs; j++)
534         {
535           Atom b (a);
536           b.translate_axis (-dir_ * dy * (lwholebeams+j), Y_AXIS);
537           leftbeams.add (b);
538         }
539     }
540
541   if (next)
542     {
543       int rhalfs = here->beams_right_i_ - next->beams_left_i_;
544       int rwholebeams = here->beams_right_i_ <? next->beams_left_i_;
545
546       Real w = next->hpos_f () - here->hpos_f ();
547       Atom a = paper ()->lookup_l ()->beam (sl, w + stemdx, beamheight_f);
548       a.translate_axis( - stemdx/2, X_AXIS);
549       int j = 0;
550       Real gap_f = 0;
551       if (here->beam_gap_i_)
552         {
553           int nogap = rwholebeams - here->beam_gap_i_;
554           for (; j  < nogap; j++)
555             {
556               Atom b (a);
557               b.translate_axis (-dir_ * dy * j, Y_AXIS);
558               rightbeams.add (b);
559             }
560           // TODO: notehead widths differ for different types
561           gap_f = paper ()->note_width () / 2;
562           w -= 2 * gap_f;
563           a = paper ()->lookup_l ()->beam (sl, w + stemdx, beamheight_f);
564         }
565
566       for (; j  < rwholebeams; j++)
567         {
568           Atom b (a);
569           b.translate (Offset (gap_f, -dir_ * dy * j));
570           rightbeams.add (b);
571         }
572
573       w = w/4 <? paper ()->note_width ();
574       if (rhalfs)
575         a = paper ()->lookup_l ()->beam (sl, w, beamheight_f);
576
577       for (; j  < rwholebeams + rhalfs; j++)
578         {
579           Atom b (a);
580           b.translate_axis (-dir_ * dy * j, Y_AXIS);
581           rightbeams.add (b);
582         }
583
584     }
585   leftbeams.add (rightbeams);
586
587   /*
588     Does beam quanting think  of the asymetry of beams? 
589     Refpoint is on bottom of symbol. (FIXTHAT) --hwn.
590    */
591   if (experimental_features_global_b && dir_ < 0)
592     leftbeams.translate_axis (-beamheight_f, Y_AXIS);
593   return leftbeams;
594 }