]> git.donarmstrong.com Git - lilypond.git/blob - lily/accidental-placement.cc
release commit
[lilypond.git] / lily / accidental-placement.cc
1 /*   
2 accidental-placement.cc --  implement Accidental_placement
3
4 source file of the GNU LilyPond music typesetter
5
6 (c) 2002--2003 Han-Wen Nienhuys <hanwen@cs.uu.nl>
7
8  */
9
10 #include <math.h>
11 #include <libc-extension.hh>
12 #include "item.hh"
13 #include "skyline.hh"
14 #include "music.hh"
15 #include "pitch.hh"
16 #include "warn.hh"
17 #include "accidental-placement.hh"
18 #include "note-column.hh"
19 #include "group-interface.hh"
20 #include "note-collision.hh"
21 #include "accidental-interface.hh"
22
23 MAKE_SCHEME_CALLBACK(Accidental_placement,alignment_callback, 2);
24 SCM
25 Accidental_placement::alignment_callback(SCM s, SCM )
26 {
27   Grob * me =unsmob_grob (s);
28
29   Grob * par = me->get_parent (X_AXIS);
30   if (!to_boolean (par->get_grob_property ("alignment-done")))
31     {
32       par->set_grob_property ("alignment-done", SCM_BOOL_T);
33       position_accidentals (par);
34     }
35
36   return scm_int2num (0);
37 }
38
39
40 void
41 Accidental_placement::add_accidental (Grob* me, Grob* a)
42 {
43   a->set_parent (me, X_AXIS);
44   a->add_offset_callback (alignment_callback_proc, X_AXIS);
45   SCM cause = a->get_parent (Y_AXIS)->get_grob_property("cause");
46
47   Music *mcause =unsmob_music (cause); 
48   if (!mcause)
49     {
50       programming_error ("Note head has no music cause!");
51       return; 
52     }
53
54   Pitch *p= unsmob_pitch (mcause->get_mus_property ("pitch"));
55
56   int n = p->get_notename ();
57
58   SCM accs = me->get_grob_property ("accidental-grobs");
59   SCM key = scm_int2num (n);
60   SCM entry = scm_assq (key, accs);
61   if (entry == SCM_BOOL_F)
62     {
63       entry = SCM_EOL;
64     }
65   else
66     entry = gh_cdr (entry);
67
68   entry = gh_cons (a->self_scm (), entry);
69
70   accs = scm_assq_set_x (accs,  key, entry);
71
72   me->set_grob_property ("accidental-grobs", accs);
73 }
74
75 /*
76   Split into break reminders.
77  */
78 void
79 Accidental_placement::split_accidentals (Grob * accs,
80                                          Link_array<Grob> *break_reminder,
81                                          Link_array<Grob> *real_acc)
82 {
83   for (SCM acs =accs->get_grob_property ("accidental-grobs"); gh_pair_p (acs);
84        acs =gh_cdr (acs))
85     for (SCM s = gh_cdar (acs); gh_pair_p (s); s = gh_cdr (s))
86       {
87         Grob *a = unsmob_grob (gh_car (s));
88
89         if (unsmob_grob (a->get_grob_property ("tie")))
90           break_reminder->push (a);
91         else
92           real_acc->push (a);
93       }
94 }
95
96 /*
97   Accidentals are special, because they appear and disappear after
98   ties at will.
99 */
100 Interval
101 Accidental_placement::get_relevant_accidental_extent (Grob *me,
102                                                       Item *item_col,
103                                                       Grob *left_object)
104 {
105   Link_array<Grob> br, ra;
106   Link_array<Grob> *which = 0;
107
108   Accidental_placement::split_accidentals (me, &br, &ra);
109   br.concat (ra);
110   
111   if (dynamic_cast<Item*>(left_object)->break_status_dir () == RIGHT)
112     which = & br;
113   else
114     which = & ra;
115   
116   Interval extent;
117   for (int i = 0; i < which->size(); i++)
118     {
119       extent.unite (which->elem(i)->extent (item_col, X_AXIS));
120     }
121
122   if (!extent.empty_b())
123     {
124       Real p = gh_scm2double (me->get_grob_property ("left-padding"));
125       extent[LEFT] -= p;
126     }
127   
128   return extent;
129 }
130
131
132
133 struct Accidental_placement_entry
134 {
135   Array<Skyline_entry> left_skyline_;
136   Array<Skyline_entry> right_skyline_;
137   Interval vertical_extent_;
138   Array<Box> extents_;
139   Link_array<Grob> grobs_;
140   Real offset_; 
141   int notename_;
142   Accidental_placement_entry()
143   {
144     offset_ =0.0;
145     notename_ = -1;
146   }
147 };
148
149 static Interval all_accidental_vertical_extent;
150 Real ape_priority (Accidental_placement_entry const * a)
151 {
152   return a->vertical_extent_[UP];
153 }
154
155   
156
157 int ape_compare (Accidental_placement_entry *const &a,
158                  Accidental_placement_entry *const &b)
159 {
160   return sign (ape_priority (a) - ape_priority(b));
161 }
162
163 int ape_rcompare (Accidental_placement_entry *const &a,
164                  Accidental_placement_entry *const &b)
165 {
166   return -sign (ape_priority (a) - ape_priority(b));
167 }
168
169
170 /*
171
172 TODO: should favor
173
174   b
175  b
176
177 placement
178  
179 */
180 void
181 stagger_apes (Link_array<Accidental_placement_entry> *apes)
182 {
183   Link_array<Accidental_placement_entry> asc = *apes;
184
185
186   asc.sort (&ape_compare);
187
188   apes->clear();
189
190   int i =0;
191   int parity = 1;
192   while (i < asc.size())
193     {
194       Accidental_placement_entry * a = 0;      
195       if (parity)
196         a = asc.pop();
197       else
198         a = asc[i++];
199
200       apes->push (a);
201       parity = !parity;
202     }
203
204   apes->reverse();
205 }
206
207   
208
209 /*
210
211   This routine computes placements of accidentals. During
212   add_accidental(), accidentals are already grouped by note, so that
213   octaves are placed above each other; they form columns. Then the
214   columns are sorted: the biggest columns go closest to the note.
215   Then the columns are spaced as closely as possible (using skyline
216   spacing).
217   
218   
219   TODO: more advanced placement. Typically, the accs should be placed
220   to form a C shape, like this
221
222   
223            ##
224         b b
225        # #
226         b
227           b b
228
229    The naturals should be left of the C as well; they should
230    be separate accs.
231
232    Note that this placement problem looks NP hard, so we just use a
233    simple strategy, not an optimal choice.
234
235 */
236
237 SCM
238 Accidental_placement::position_accidentals (Grob * me)
239 {
240   if (!me->live ())
241     return SCM_UNSPECIFIED;
242   
243   SCM accs = me->get_grob_property ("accidental-grobs");
244
245   /*
246     TODO: there is a bug in this code. If two accs are on the same
247     Y-position, they share an Ape, and will be pritned in overstrike.
248    */
249   Link_array<Accidental_placement_entry> apes;
250   for (SCM s = accs; gh_pair_p (s); s =gh_cdr (s))
251     {
252       Accidental_placement_entry *ape = new Accidental_placement_entry;
253       ape->notename_ = gh_scm2int (gh_caar (s));
254       
255       for (SCM t = gh_cdar (s); gh_pair_p (t); t =gh_cdr (t))
256         ape->grobs_.push (unsmob_grob (gh_car (t)));
257
258       apes.push (ape);
259     }
260
261
262   Grob *common[] = {me, 0};
263
264   /*
265     First we must extract *all* pointers. We can only determine
266     extents if we're sure that we've found the right common refpoint
267    */
268   Link_array<Grob> note_cols, heads;
269   for (int i= apes.size (); i--;)
270     { 
271       Accidental_placement_entry * ape = apes[i];
272       for (int j = ape->grobs_.size(); j--;)
273         {
274           Grob * a = ape->grobs_[j];
275
276           if (common[Y_AXIS])
277             common[Y_AXIS] = common[Y_AXIS]->common_refpoint (a, Y_AXIS);
278           else
279             common[Y_AXIS] = a;
280           
281           Grob *head = a->get_parent (Y_AXIS);
282
283           Grob * col = head->get_parent (X_AXIS);
284           if (Note_column::has_interface (col))
285             note_cols.push (col);
286           else
287             heads.push (head);
288         }
289     }
290
291   /*
292     This is a little kludgy: to get all notes, we look if there are
293     collisions as well.
294    */
295   for (int i = note_cols.size() ; i--;)
296     {
297       Grob *c = note_cols[i]->get_parent (X_AXIS);
298       if (Note_collision_interface::has_interface (c))
299         {
300           Link_array<Grob> gs =
301             Pointer_group_interface__extract_grobs (c, (Grob*)0, "elements");
302       
303           note_cols.concat (gs);
304         }
305     }
306   
307   for (int i = note_cols.size() ; i--;)
308     {
309       heads.concat (Pointer_group_interface__extract_grobs (note_cols[i],
310                                                             (Grob*)0,
311                                                             "note-heads"));
312       
313     }
314   heads.default_sort();
315   heads.uniq();
316   common[Y_AXIS] = common_refpoint_of_array (heads, common[Y_AXIS], Y_AXIS);
317
318   
319   for (int i= apes.size (); i--;)
320     {
321       Accidental_placement_entry * ape = apes[i];
322       ape->left_skyline_ = empty_skyline (LEFT);
323       ape->right_skyline_ = empty_skyline (RIGHT);
324    
325       for (int j = apes[i]->grobs_.size(); j--;)
326         {
327           Grob * a = apes[i]->grobs_[j];
328
329           Array<Box> boxes = Accidental_interface::accurate_boxes (a, common);
330           
331           ape->extents_.concat (boxes);
332           for (int j  = boxes.size(); j--;)
333             {
334               insert_extent_into_skyline (&ape->left_skyline_, boxes[j], Y_AXIS, LEFT);
335               insert_extent_into_skyline (&ape->right_skyline_ , boxes[j], Y_AXIS, RIGHT);
336             }
337         }
338     }
339
340
341   Interval total;
342   for (int i = apes.size(); i--;)
343     {
344       Interval y ;
345       
346       for (int j = apes[i]->extents_.size(); j--;)
347         {
348           y.unite (apes[i]->extents_[j][Y_AXIS]);
349         }
350       apes[i]->vertical_extent_ = y;
351       total.unite (y);
352     }
353   all_accidental_vertical_extent = total;
354   stagger_apes (&apes);
355
356   Accidental_placement_entry * head_ape = new Accidental_placement_entry;
357   common[X_AXIS] = common_refpoint_of_array (heads, common[X_AXIS], X_AXIS);  
358   Array<Skyline_entry> head_skyline (empty_skyline (LEFT));
359   Array<Box> head_extents;
360   for (int i = heads.size(); i--;)
361     {
362       Box b(heads[i]->extent (common[X_AXIS] , X_AXIS),
363             heads[i]->extent (common[Y_AXIS], Y_AXIS));
364
365       insert_extent_into_skyline (&head_skyline, b , Y_AXIS, LEFT);
366     }
367
368   head_ape-> left_skyline_ = head_skyline;
369   head_ape->offset_ = 0.0;
370
371   SCM rs = me->get_grob_property ("right-padding");
372   if (gh_number_p (rs))
373     head_ape->offset_ -= gh_scm2double (rs);
374
375   
376   Real padding = 0.2;
377   SCM spad = me->get_grob_property ("padding");
378   if (gh_number_p (spad))
379     padding = gh_scm2double (spad);
380
381
382   /*
383     TODO:
384
385     There is a bug in this code: the left_skylines should be
386     accumulated, otherwise the b will collide with bb in
387
388       bb
389     b 
390       n 
391     
392    */
393   apes.push (head_ape);
394   for (int i= apes.size () -1 ; i-- > 0;)
395     {
396       Accidental_placement_entry *ape = apes[i];
397       Real d = 0.0;
398       int j = i+1;
399       do {
400         d = - skyline_meshing_distance (ape->right_skyline_,
401                                         apes[j]->left_skyline_);
402
403         if (!isinf(d)
404             || j + 1 == apes.size())
405           break;
406         
407         j = j+ 1;
408       } while (1);
409
410       if (isinf(d))
411         d = 0.0;
412       
413       d -= padding;
414       ape->offset_ += apes[j]->offset_ + d;
415     }
416
417   apes.pop();
418   for (int i = apes.size(); i--;)
419     {
420       Accidental_placement_entry* ape = apes[i];
421       for (int j  = ape->grobs_.size(); j--;)
422         {
423           ape->grobs_[j]->translate_axis (ape->offset_, X_AXIS);
424         }
425     }
426
427   
428   Interval left_extent, right_extent;
429   Accidental_placement_entry *ape = apes[0];
430
431   for (int i = ape->extents_.size(); i--;)
432     left_extent.unite (ape->offset_ +  ape->extents_[i][X_AXIS]);
433
434   ape = apes.top();
435   for (int i = ape->extents_.size(); i--;)
436     right_extent.unite (ape->offset_  + ape->extents_[i][X_AXIS]);
437
438   SCM ls = me->get_grob_property ("left-padding");
439   if (gh_number_p (rs))
440     left_extent[LEFT] -= gh_scm2double (ls);
441
442   
443   Interval width(left_extent[LEFT], right_extent[RIGHT]);
444
445   SCM scm_width = ly_interval2scm (width);
446   me->set_extent (scm_width, X_AXIS);
447   
448   for (int i = apes.size(); i--;)
449     delete apes[i];
450
451   return SCM_UNSPECIFIED;
452 }
453
454 ADD_INTERFACE(Accidental_placement,
455               "accidental-placement-interface",
456               "Take care of complex accidental collisions.",
457               "left-padding padding right-padding accidental-grobs alignment-done")