]> git.tdb.fi Git - r2c2.git/blob - source/designer/manipulator.cpp
6aca114aa5538ecfbceebf5ab328009c4b5a675b
[r2c2.git] / source / designer / manipulator.cpp
1 #include <algorithm>
2 #include <cmath>
3 #include <msp/strings/format.h>
4 #include "libr2c2/tracktype.h"
5 #include "designer.h"
6 #include "manipulator.h"
7 #include "selection.h"
8
9 using namespace std;
10 using namespace R2C2;
11 using namespace Msp;
12
13 Manipulator::Manipulator(Designer &d, Input::Mouse &m, Selection &s):
14         designer(d),
15         mouse(m),
16         selection(s),
17         mode(NONE),
18         angle(0)
19 {
20         mouse.signal_button_press.connect(sigc::bind_return(sigc::mem_fun(this, &Manipulator::button_press), false));
21         mouse.signal_axis_motion.connect(sigc::bind_return(sigc::mem_fun(this, &Manipulator::axis_motion), false));
22         selection.signal_changed.connect(sigc::mem_fun(this, &Manipulator::selection_changed));
23 }
24
25 void Manipulator::start_move()
26 {
27         if(mode)
28                 cancel();
29
30         move_origin = gpointer;
31
32         mode = MOVE;
33 }
34
35 void Manipulator::start_rotate()
36 {
37         if(mode)
38                 cancel();
39
40         rot_origin = atan2(gpointer.y-center.y, gpointer.x-center.x);
41
42         mode = ROTATE;
43 }
44
45 void Manipulator::start_elevate()
46 {
47         if(mode)
48                 cancel();
49
50         mode = ELEVATE;
51
52         elev_origin = pointer.y;
53 }
54
55 bool Manipulator::start_extend()
56 {
57         if(mode)
58                 cancel();
59
60         bool ok = false;
61         const set<Track *> &stracks = selection.get_objects<Track>();
62         for(set<Track *>::const_iterator i=stracks.begin(); (!ok && i!=stracks.end()); ++i)
63         {
64                 const vector<Track *> &links = (*i)->get_links();
65                 for(vector<Track *>::const_iterator j=links.begin(); (!ok && j!=links.end()); ++j)
66                         ok = !*j;
67         }
68
69         if(ok)
70                 mode = EXTEND;
71         else
72                 signal_status.emit("No free endpoints");
73
74         return ok;
75 }
76
77 void Manipulator::duplicate()
78 {
79         if(mode)
80                 cancel();
81
82         list<Object *> new_objs;
83         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
84         {
85                 Object *obj = i->object->clone(&designer.get_layout());
86                 if(Track *track = dynamic_cast<Track *>(obj))
87                 {
88                         for(list<Object *>::iterator j=new_objs.begin(); j!=new_objs.end(); ++j)
89                                 if(Track *track2 = dynamic_cast<Track *>(*j))
90                                         track->snap_to(*track2, true);
91                 }
92                 new_objs.push_back(obj);
93         }
94
95         selection.replace(new_objs.begin(), new_objs.end());
96 }
97
98 void Manipulator::flatten()
99 {
100         if(mode)
101                 cancel();
102
103         if(objects.empty())
104                 return;
105
106         float z = 0;
107         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
108         {
109                 z += i->object->get_position().z;
110                 if(Track *track = dynamic_cast<Track *>(i->object))
111                         z += track->get_slope()/2;
112         }
113         z /= static_cast<int>(objects.size());
114
115         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
116         {
117                 Vector p = i->object->get_position();
118                 i->object->set_position(Vector(p.x, p.y, z));
119                 if(Track *track = dynamic_cast<Track *>(i->object))
120                         track->set_slope(0);
121         }
122
123         for(set<Track *>::iterator i=neighbors.begin(); i!=neighbors.end(); ++i)
124                 (*i)->check_slope();
125
126         update_objects();
127 }
128
129 void Manipulator::even_slope(bool smooth)
130 {
131         if(mode)
132                 cancel();
133
134         if(neighbors.size()!=2)
135                 return;
136
137         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
138                 if(Track *track = dynamic_cast<Track *>(i->object))
139                         if(track->get_type().get_endpoints().size()!=2)
140                                 return;
141
142         list<Track *> tracks2;
143         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
144                 if(Track *track = dynamic_cast<Track *>(i->object))
145                         tracks2.push_back(track);
146
147         float total_len = 0;
148
149         list<TrackOrder> order;
150         Track *cur = *neighbors.begin();
151         while(tracks2.size())
152         {
153                 bool rev = false;
154                 for(list<Track *>::iterator i=tracks2.begin(); i!=tracks2.end(); ++i)
155                 {
156                         const vector<Track *> &links = (*i)->get_links();
157                         if(links[0]==cur)
158                         {
159                                 cur = *i;
160                                 tracks2.erase(i);
161                                 break;
162                         }
163                         else if(links[1]==cur)
164                         {
165                                 cur = *i;
166                                 rev = true;
167                                 tracks2.erase(i);
168                                 break;
169                         }
170                 }
171                 order.push_back(TrackOrder(cur, rev));
172                 total_len += cur->get_type().get_total_length();
173         }
174
175         set<Track *>::iterator nb = neighbors.begin();
176         int epi = (*nb)->get_endpoint_by_link(*order.front().track);
177         float start_z = (*nb)->get_snap_node(epi).position.z;
178         ++nb;
179         epi = (*nb)->get_endpoint_by_link(*order.back().track);
180         float end_z = (*nb)->get_snap_node(epi).position.z;
181
182         if(smooth)
183         {
184                 float dir = (end_z>start_z)?1:-1;
185                 float cur_slope = 0;
186                 while((end_z-start_z)*dir/total_len>cur_slope+0.025 && order.size()>2)
187                 {
188                         cur_slope += 0.025;
189
190                         float dz = order.front().track->get_type().get_total_length()*dir*cur_slope;
191                         set_slope(order.front(), start_z, dz);
192                         start_z += dz;
193                         total_len -= order.front().track->get_type().get_total_length();
194                         order.erase(order.begin());
195
196                         dz = order.back().track->get_type().get_total_length()*dir*cur_slope;
197                         set_slope(order.back(), end_z-dz, dz);
198                         end_z -= dz;
199                         total_len -= order.back().track->get_type().get_total_length();
200                         order.erase(--order.end());
201                 }
202         }
203
204         float cur_z = start_z;
205         for(list<TrackOrder>::iterator i=order.begin(); i!=order.end(); ++i)
206         {
207                 float dz = i->track->get_type().get_total_length()*(end_z-start_z)/total_len;
208                 set_slope(*i, cur_z, dz);
209                 cur_z += dz;
210         }
211
212         for(set<Track *>::iterator i=neighbors.begin(); i!=neighbors.end(); ++i)
213                 (*i)->check_slope();
214
215         update_objects();
216 }
217
218 void Manipulator::connect()
219 {
220         if(objects.size()!=2)
221         {
222                 signal_status.emit("Exactly two tracks must be selected");
223                 return;
224         }
225
226         Track *track1 = dynamic_cast<Track *>(objects.front().object);
227         Track *track2 = dynamic_cast<Track *>(objects.back().object);
228         if(!track1 || !track2)
229         {
230                 signal_status.emit("Exactly two tracks must be selected");
231                 return;
232         }
233
234         float limit = designer.get_layout().get_catalogue().get_gauge()/10;
235
236         Snap sn1;
237         bool ok = false;
238         float gap = 0;
239         for(unsigned i=0; i<track1->get_type().get_endpoints().size(); ++i)
240         {
241                 if(track1->get_link(i))
242                         continue;
243
244                 sn1 = track1->get_snap_node(i);
245
246                 for(unsigned j=0; j<track2->get_type().get_endpoints().size(); ++j)
247                 {
248                         if(track2->get_link(j))
249                                 continue;
250
251                         Snap sn2 = track2->get_snap_node(j);
252
253                         float dz = sn2.position.z-sn1.position.z;
254                         if(abs(dz)>0.02)
255                                 continue;
256
257                         float adiff = sn1.rotation+M_PI-sn2.rotation;
258                         while(adiff<-M_PI)
259                                 adiff += M_PI*2;
260                         while(adiff>M_PI)
261                                 adiff -= M_PI*2;
262                         if(abs(adiff)>0.01)
263                                 continue;
264
265                         float c = cos(sn1.rotation);
266                         float s = sin(sn1.rotation);
267                         float dx = sn2.position.x-sn1.position.x;
268                         float dy = sn2.position.y-sn1.position.y;
269                         if(abs(dx*s-dy*c)>limit)
270                                 continue;
271
272                         gap = dx*c+dy*s;
273                         if(gap<0)
274                                 continue;
275
276                         ok = true;
277                 }
278
279                 if(ok)
280                         break;
281         }
282
283         if(!ok)
284         {
285                 signal_status.emit("No aligned endpoints found");
286                 return;
287         }
288
289         vector<Track *> trks = create_straight(sn1.position, sn1.rotation, gap, limit);
290
291         if(trks.empty())
292         {
293                 signal_status.emit("No connection possible");
294                 return;
295         }
296
297         trks.front()->snap_to(*track1, true);
298         trks.back()->snap_to(*track2, true);
299
300         selection.replace(trks.begin(), trks.end());
301 }
302
303 void Manipulator::cancel()
304 {
305         if(!mode)
306                 return;
307         mode = NONE;
308
309         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
310         {
311                 i->object->set_position(Vector(center.x+i->pos.x, center.y+i->pos.y, center.z+i->pos.z));
312                 i->object->set_rotation(i->rot);
313         }
314
315         for(set<Track *>::iterator i=neighbors.begin(); i!=neighbors.end(); ++i)
316                 (*i)->check_slope();
317
318         for(vector<Track *>::iterator i=extend_tracks.begin(); i!=extend_tracks.end(); ++i)
319                 delete *i;
320         extend_tracks.clear();
321
322         angle = 0;
323
324         signal_done.emit(false);
325 }
326
327 void Manipulator::button_press(unsigned btn)
328 {
329         if(!mode)
330                 return;
331
332         if(btn==3)
333                 cancel();
334         else if(btn==1)
335         {
336                 Mode m = mode;
337                 mode = NONE;
338                 angle = 0;
339
340                 if(m!=EXTEND)
341                 {
342                         for(set<Track *>::iterator i=neighbors.begin(); i!=neighbors.end(); ++i)
343                                 for(vector<MObject>::iterator j=objects.begin(); j!=objects.end(); ++j)
344                                         if(Track *track = dynamic_cast<Track *>(j->object))
345                                                 track->break_link(**i);
346                 }
347
348                 const set<Track *> &ltracks = designer.get_layout().get_tracks();
349                 for(set<Track *>::const_iterator i=ltracks.begin(); i!=ltracks.end(); ++i)
350                 {
351                         bool ok = true;
352                         for(vector<MObject>::iterator j=objects.begin(); (j!=objects.end() && ok); ++j)
353                                 ok = (j->object!=*i);
354                         if(!ok) continue;
355
356                         for(vector<MObject>::iterator j=objects.begin(); j!=objects.end(); ++j)
357                                 if(Track *track = dynamic_cast<Track *>(j->object))
358                                         track->snap_to(**i, true);
359                 }
360
361                 if(m==EXTEND)
362                 {
363                         selection.replace(extend_tracks.begin(), extend_tracks.end());
364                         extend_tracks.clear();
365                 }
366                 else
367                 {
368                         update_objects();
369                         update_neighbors();
370                 }
371
372                 signal_done.emit(true);
373         }
374 }
375
376 void Manipulator::axis_motion(unsigned axis, float value, float)
377 {
378         if(axis==0)
379                 pointer.x = value;
380         else if(axis==1)
381                 pointer.y = value;
382         gpointer = designer.map_pointer_to_ground(pointer);
383
384         if(mode==MOVE)
385         {
386                 Vector delta(gpointer.x-move_origin.x, gpointer.y-move_origin.y, 0);
387                 Vector offset(center.x+delta.x, center.y+delta.y, center.z);
388                 for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
389                 {
390                         i->object->set_position(Vector(offset.x+i->pos.x, offset.y+i->pos.y, offset.z+i->pos.z));
391                         i->object->set_rotation(i->rot);
392                 }
393
394                 const set<Track *> &ltracks = designer.get_layout().get_tracks();
395                 float limit = max(designer.get_layout().get_catalogue().get_gauge(),
396                         designer.get_camera_controller().get_view_scale()/100.0f);
397                 MObject *snapped = 0;
398                 for(set<Track *>::const_iterator i=ltracks.begin(); (i!=ltracks.end() && !snapped); ++i)
399                 {
400                         bool ok = true;
401                         for(vector<MObject>::iterator j=objects.begin(); (j!=objects.end() && ok); ++j)
402                                 ok = (j->object!=*i);
403                         if(!ok)
404                                 continue;
405
406                         for(vector<MObject>::iterator j=objects.begin(); (j!=objects.end() && !snapped); ++j)
407                                 if(Track *track = dynamic_cast<Track *>(j->object))
408                                         if(track->snap_to(**i, false, limit))
409                                                 snapped = &*j;
410                 }
411
412                 if(snapped)
413                 {
414                         float da = snapped->object->get_rotation()-snapped->rot;
415                         float c = cos(da);
416                         float s = sin(da);
417                         const Vector &sp = snapped->object->get_position();
418                         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
419                         {
420                                 if(&*i==snapped)
421                                         continue;
422
423                                 Vector dp(i->pos.x-snapped->pos.x, i->pos.y-snapped->pos.y, 0);
424                                 i->object->set_position(Vector(sp.x+c*dp.x-s*dp.y, sp.y+s*dp.x+c*dp.y, sp.z+i->pos.z-snapped->pos.z));
425                                 i->object->set_rotation(i->rot+da);
426                         }
427                 }
428         }
429         else if(mode==ROTATE)
430         {
431                 float a = atan2(gpointer.y-center.y, gpointer.x-center.x);
432                 angle += a-rot_origin;
433                 rot_origin = a;
434
435                 for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
436                 {
437                         float c = cos(angle);
438                         float s = sin(angle);
439                         i->object->set_position(Vector(center.x+c*i->pos.x-s*i->pos.y, center.y+s*i->pos.x+c*i->pos.y, center.z+i->pos.z));
440                         i->object->set_rotation(angle+i->rot);
441                 }
442         }
443         else if(mode==ELEVATE && axis==1)
444         {
445                 float dz = pointer.y-elev_origin;
446
447                 signal_status.emit(format("Elevation: %+.0fmm (%.0fmm)", dz*1000, (center.z+dz)*1000));
448
449                 for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
450                         i->object->set_position(Vector(center.x+i->pos.x, center.y+i->pos.y, center.z+i->pos.z+dz));
451
452                 for(set<Track *>::iterator i=neighbors.begin(); i!=neighbors.end(); ++i)
453                         (*i)->check_slope();
454         }
455         else if(mode==EXTEND)
456         {
457                 Vector pos;
458                 float dir = 0;
459                 float length = 0;
460                 for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
461                 {
462                         Track *track = dynamic_cast<Track *>(i->object);
463                         if(!track)
464                                 continue;
465
466                         unsigned n_endpoints = track->get_type().get_endpoints().size();
467                         for(unsigned j=0; j<n_endpoints; ++j)
468                         {
469                                 if(track->get_link(j))
470                                         continue;
471
472                                 Snap sn = track->get_snap_node(j);
473                                 float c = cos(sn.rotation);
474                                 float s = sin(sn.rotation);
475                                 float dx = gpointer.x-sn.position.x;
476                                 float dy = gpointer.y-sn.position.y;
477
478                                 float len = dx*c+dy*s;
479                                 if(len<length)
480                                         continue;
481
482                                 pos = sn.position;
483                                 dir = sn.rotation;
484                                 length = len;
485                         }
486                 }
487
488                 if(length)
489                 {
490                         vector<Track *> trks = create_straight(pos, dir, length, max(length/500, 0.001f));
491
492                         if(!trks.empty())
493                         {
494                                 for(vector<Track *>::iterator i=extend_tracks.begin(); i!=extend_tracks.end(); ++i)
495                                         delete *i;
496                                 extend_tracks = trks;
497
498                                 map<ArticleNumber, unsigned> counts;
499                                 length = 0;
500                                 for(vector<Track *>::iterator i=extend_tracks.begin(); i!=extend_tracks.end(); ++i)
501                                 {
502                                         length += (*i)->get_type().get_total_length();
503                                         ++counts[(*i)->get_type().get_article_number()];
504                                 }
505
506                                 string detail;
507                                 for(map<ArticleNumber, unsigned>::const_iterator i=counts.begin(); i!=counts.end(); ++i)
508                                 {
509                                         if(!detail.empty())
510                                                 detail += ", ";
511                                         detail += format("%dx %s", i->second, i->first);
512                                 }
513
514                                 signal_status.emit(format("Extend: %.0fmm (%s)", length*1000, detail));
515                         }
516                 }
517         }
518 }
519
520 void Manipulator::selection_changed()
521 {
522         if(mode)
523                 cancel();
524
525         objects.clear();
526         set<Object *> pending = selection.get_objects();
527         while(!pending.empty())
528         {
529                 for(set<Object *>::iterator i=pending.begin(); i!=pending.end(); )
530                 {
531                         if((*i)->get_parent() && pending.count((*i)->get_parent()))
532                                 ++i;
533                         else
534                         {
535                                 objects.push_back(*i);
536                                 pending.erase(i++);
537                         }
538                 }
539         }
540
541         update_neighbors();
542         update_objects();
543 }
544
545 void Manipulator::update_objects()
546 {
547         Vector minp, maxp;
548         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
549         {
550                 // XXX Use generic bounding box when it is implemented
551                 if(Track *track = dynamic_cast<Track *>(i->object))
552                 {
553                         unsigned n_endpoints = track->get_type().get_endpoints().size();
554                         for(unsigned j=0; j<n_endpoints; ++j)
555                         {
556                                 Vector p = track->get_snap_node(j).position;
557                                 if(i==objects.begin() && j==0)
558                                         minp = maxp = p;
559                                 else
560                                 {
561                                         minp.x = min(minp.x, p.x);
562                                         maxp.x = max(maxp.x, p.x);
563                                         minp.y = min(minp.y, p.y);
564                                         maxp.y = max(maxp.y, p.y);
565                                         minp.z = min(minp.z, p.z);
566                                 }
567                         }
568                 }
569                 else
570                 {
571                         const Vector &p = i->object->get_position();
572                         minp.x = min(minp.x, p.x);
573                         maxp.x = max(maxp.x, p.x);
574                         minp.y = min(minp.y, p.y);
575                         maxp.y = max(maxp.y, p.y);
576                         minp.z = min(minp.z, p.z);
577                 }
578         }
579
580         center = Vector((minp.x+maxp.x)/2, (minp.y+maxp.y)/2, minp.z);
581         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
582         {
583                 const Vector &p = i->object->get_position();
584                 i->pos = Vector(p.x-center.x, p.y-center.y, p.z-center.z);
585                 i->rot = i->object->get_rotation();
586         }
587 }
588
589 void Manipulator::update_neighbors()
590 {
591         neighbors.clear();
592         for(vector<MObject>::iterator i=objects.begin(); i!=objects.end(); ++i)
593         {
594                 Track *track = dynamic_cast<Track *>(i->object);
595                 if(!track)
596                         continue;
597
598                 const vector<Track *> &links = track->get_links();
599                 for(vector<Track *>::const_iterator j=links.begin(); j!=links.end(); ++j)
600                 {
601                         if(!*j)
602                                 continue;
603                         if(neighbors.count(*j))
604                                 continue;
605
606                         bool ok = true;
607                         for(vector<MObject>::iterator k=objects.begin(); (k!=objects.end() && ok); ++k)
608                                 ok = (k->object!=*j);
609
610                         if(ok)
611                                 neighbors.insert(*j);
612                 }
613         }
614 }
615
616 void Manipulator::set_slope(TrackOrder &track, float z, float dz)
617 {
618         const Vector &p = track.track->get_position();
619         if(track.rev)
620         {
621                 track.track->set_position(Vector(p.x, p.y, z+dz));
622                 track.track->set_slope(-dz);
623         }
624         else
625         {
626                 track.track->set_position(Vector(p.x, p.y, z));
627                 track.track->set_slope(dz);
628         }
629 }
630
631 vector<Track *> Manipulator::create_straight(const R2C2::Vector &start, float dir, float length, float limit)
632 {
633         const Catalogue::TrackMap &track_types = designer.get_catalogue().get_tracks();
634         std::map<float, const TrackType *> types_by_length;
635         unsigned preference = 0;
636         for(Catalogue::TrackMap::const_iterator i=track_types.begin(); i!=track_types.end(); ++i)
637         {
638                 const vector<TrackPart> &parts = i->second->get_parts();
639                 if(parts.size()!=1)
640                         continue;
641                 if(parts.front().is_curved() || parts.front().is_dead_end())
642                         continue;
643
644                 types_by_length[parts.front().get_length()] = i->second;
645                 preference = max(preference, i->second->get_autofit_preference());
646         }
647
648         vector<float> lengths;
649         float removed = 0;
650         while(length>limit)
651         {
652                 bool found = false;
653                 for(map<float, const TrackType *>::iterator i=types_by_length.end(); i!=types_by_length.begin(); )
654                 {
655                         --i;
656                         if(i->second->get_autofit_preference()<preference)
657                                 continue;
658                         if((!removed || i->first<removed) && i->first<length+limit)
659                         {
660                                 unsigned n = static_cast<unsigned>((length+limit)/i->first);
661                                 lengths.insert(lengths.end(), n, i->first);
662                                 length -= n*i->first;
663                                 found = true;
664                                 break;
665                         }
666                 }
667
668                 if(found)
669                         continue;
670
671                 if(lengths.empty())
672                 {
673                         if(preference>0)
674                         {
675                                 --preference;
676                                 removed = 0;
677                                 continue;
678                         }
679                         break;
680                 }
681
682                 length += lengths.back();
683                 removed = lengths.back();
684                 lengths.pop_back();
685         }
686
687         vector<Track *> trks;
688
689         if(!lengths.empty())
690         {
691                 Vector pos = start;
692                 float c = cos(dir);
693                 float s = sin(dir);
694                 for(vector<float>::iterator i=lengths.begin(); i!=lengths.end(); ++i)
695                 {
696                         Track *track = new Track(designer.get_layout(), *get_item(types_by_length, *i));
697                         track->set_position(pos);
698                         track->set_rotation(dir);
699
700                         if(!trks.empty())
701                                 track->snap_to(*trks.back(), true);
702                         trks.push_back(track);
703
704                         pos.x += c**i;
705                         pos.y += s**i;
706                 }
707         }
708
709         return trks;
710 }
711
712
713 Manipulator::MObject::MObject(Object *o):
714         object(o),
715         pos(object->get_position()),
716         rot(object->get_rotation())
717 { }