Merge SPLAY_SHOW_* and DUPES_SHOW_*
[survex.git] / src / gfxcore.cc
blobc3e234b25a6610be5f320d0808ec1f08e2c9f86a
1 //
2 // gfxcore.cc
3 //
4 // Core drawing code for Aven.
5 //
6 // Copyright (C) 2000-2003,2005,2006 Mark R. Shinwell
7 // Copyright (C) 2001-2003,2004,2005,2006,2007,2010,2011,2012,2014,2015,2016 Olly Betts
8 // Copyright (C) 2005 Martin Green
9 //
10 // This program is free software; you can redistribute it and/or modify
11 // it under the terms of the GNU General Public License as published by
12 // the Free Software Foundation; either version 2 of the License, or
13 // (at your option) any later version.
15 // This program is distributed in the hope that it will be useful,
16 // but WITHOUT ANY WARRANTY; without even the implied warranty of
17 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
18 // GNU General Public License for more details.
20 // You should have received a copy of the GNU General Public License
21 // along with this program; if not, write to the Free Software
22 // Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA
25 #ifdef HAVE_CONFIG_H
26 #include <config.h>
27 #endif
29 #include <assert.h>
30 #include <float.h>
32 #include "aven.h"
33 #include "date.h"
34 #include "filename.h"
35 #include "gfxcore.h"
36 #include "mainfrm.h"
37 #include "message.h"
38 #include "useful.h"
39 #include "printing.h"
40 #include "guicontrol.h"
41 #include "moviemaker.h"
43 #include <wx/confbase.h>
44 #include <wx/wfstream.h>
45 #include <wx/image.h>
46 #include <wx/zipstrm.h>
48 #include <proj_api.h>
50 const unsigned long DEFAULT_HGT_DIM = 3601;
51 const unsigned long DEFAULT_HGT_SIZE = sqrd(DEFAULT_HGT_DIM) * 2;
53 // Values for m_SwitchingTo
54 #define PLAN 1
55 #define ELEVATION 2
56 #define NORTH 3
57 #define EAST 4
58 #define SOUTH 5
59 #define WEST 6
61 // Any error value higher than this is clamped to this.
62 #define MAX_ERROR 12.0
64 // Any length greater than pow(10, LOG_LEN_MAX) will be clamped to this.
65 const Double LOG_LEN_MAX = 1.5;
67 // How many bins per letter height to use when working out non-overlapping
68 // labels.
69 const unsigned int QUANTISE_FACTOR = 2;
71 #include "avenpal.h"
73 static const int INDICATOR_BOX_SIZE = 60;
74 static const int INDICATOR_GAP = 2;
75 static const int INDICATOR_MARGIN = 5;
76 static const int INDICATOR_OFFSET_X = 15;
77 static const int INDICATOR_OFFSET_Y = 15;
78 static const int INDICATOR_RADIUS = INDICATOR_BOX_SIZE / 2 - INDICATOR_MARGIN;
79 static const int KEY_OFFSET_X = 10;
80 static const int KEY_OFFSET_Y = 10;
81 static const int KEY_EXTRA_LEFT_MARGIN = 2;
82 static const int KEY_BLOCK_WIDTH = 20;
83 static const int KEY_BLOCK_HEIGHT = 16;
84 static const int TICK_LENGTH = 4;
85 static const int SCALE_BAR_OFFSET_X = 15;
86 static const int SCALE_BAR_OFFSET_Y = 12;
87 static const int SCALE_BAR_HEIGHT = 12;
89 static const gla_colour TEXT_COLOUR = col_GREEN;
90 static const gla_colour HERE_COLOUR = col_WHITE;
91 static const gla_colour NAME_COLOUR = col_GREEN;
92 static const gla_colour SEL_COLOUR = col_WHITE;
94 // Number of entries across and down the hit-test grid:
95 #define HITTEST_SIZE 20
97 // How close the pointer needs to be to a station to be considered:
98 #define MEASURE_THRESHOLD 7
100 // vector for lighting angle
101 static const Vector3 light(.577, .577, .577);
103 BEGIN_EVENT_TABLE(GfxCore, GLACanvas)
104 EVT_PAINT(GfxCore::OnPaint)
105 EVT_LEFT_DOWN(GfxCore::OnLButtonDown)
106 EVT_LEFT_UP(GfxCore::OnLButtonUp)
107 EVT_MIDDLE_DOWN(GfxCore::OnMButtonDown)
108 EVT_MIDDLE_UP(GfxCore::OnMButtonUp)
109 EVT_RIGHT_DOWN(GfxCore::OnRButtonDown)
110 EVT_RIGHT_UP(GfxCore::OnRButtonUp)
111 EVT_MOUSEWHEEL(GfxCore::OnMouseWheel)
112 EVT_MOTION(GfxCore::OnMouseMove)
113 EVT_LEAVE_WINDOW(GfxCore::OnLeaveWindow)
114 EVT_SIZE(GfxCore::OnSize)
115 EVT_IDLE(GfxCore::OnIdle)
116 EVT_CHAR(GfxCore::OnKeyPress)
117 END_EVENT_TABLE()
119 GfxCore::GfxCore(MainFrm* parent, wxWindow* parent_win, GUIControl* control) :
120 GLACanvas(parent_win, 100),
121 m_Scale(0.0),
122 initial_scale(1.0),
123 m_ScaleBarWidth(0),
124 m_Control(control),
125 m_LabelGrid(NULL),
126 m_Parent(parent),
127 m_DoneFirstShow(false),
128 m_TiltAngle(0.0),
129 m_PanAngle(0.0),
130 m_Rotating(false),
131 m_RotationStep(0.0),
132 m_SwitchingTo(0),
133 m_Crosses(false),
134 m_Legs(true),
135 m_Splays(SHOW_FADED),
136 m_Dupes(SHOW_DASHED),
137 m_Names(false),
138 m_Scalebar(true),
139 m_ColourKey(true),
140 m_OverlappingNames(false),
141 m_Compass(true),
142 m_Clino(true),
143 m_Tubes(false),
144 m_ColourBy(COLOUR_BY_DEPTH),
145 m_HaveData(false),
146 m_HaveTerrain(true),
147 m_MouseOutsideCompass(false),
148 m_MouseOutsideElev(false),
149 m_Surface(false),
150 m_Entrances(false),
151 m_FixedPts(false),
152 m_ExportedPts(false),
153 m_Grid(false),
154 m_BoundingBox(false),
155 m_Terrain(false),
156 m_Degrees(false),
157 m_Metric(false),
158 m_Percent(false),
159 m_HitTestDebug(false),
160 m_RenderStats(false),
161 m_PointGrid(NULL),
162 m_HitTestGridValid(false),
163 m_here(NULL),
164 m_there(NULL),
165 presentation_mode(0),
166 pres_reverse(false),
167 pres_speed(0.0),
168 movie(NULL),
169 current_cursor(GfxCore::CURSOR_DEFAULT),
170 sqrd_measure_threshold(sqrd(MEASURE_THRESHOLD)),
171 dem(NULL),
172 last_time(0),
173 n_tris(0)
175 AddQuad = &GfxCore::AddQuadrilateralDepth;
176 AddPoly = &GfxCore::AddPolylineDepth;
177 wxConfigBase::Get()->Read(wxT("metric"), &m_Metric, true);
178 wxConfigBase::Get()->Read(wxT("degrees"), &m_Degrees, true);
179 wxConfigBase::Get()->Read(wxT("percent"), &m_Percent, false);
181 for (int pen = 0; pen < NUM_COLOUR_BANDS + 1; ++pen) {
182 m_Pens[pen].SetColour(REDS[pen] / 255.0,
183 GREENS[pen] / 255.0,
184 BLUES[pen] / 255.0);
187 timer.Start();
190 GfxCore::~GfxCore()
192 TryToFreeArrays();
194 delete[] m_PointGrid;
197 void GfxCore::TryToFreeArrays()
199 // Free up any memory allocated for arrays.
200 delete[] m_LabelGrid;
201 m_LabelGrid = NULL;
205 // Initialisation methods
208 void GfxCore::Initialise(bool same_file)
210 // Initialise the view from the parent holding the survey data.
212 TryToFreeArrays();
214 m_DoneFirstShow = false;
216 m_HitTestGridValid = false;
217 m_here = NULL;
218 m_there = NULL;
220 m_MouseOutsideCompass = m_MouseOutsideElev = false;
222 if (!same_file) {
223 // Apply default parameters unless reloading the same file.
224 DefaultParameters();
227 m_HaveData = true;
229 // Clear any cached OpenGL lists which depend on the data.
230 InvalidateList(LIST_SCALE_BAR);
231 InvalidateList(LIST_DEPTH_KEY);
232 InvalidateList(LIST_DATE_KEY);
233 InvalidateList(LIST_ERROR_KEY);
234 InvalidateList(LIST_GRADIENT_KEY);
235 InvalidateList(LIST_LENGTH_KEY);
236 InvalidateList(LIST_UNDERGROUND_LEGS);
237 InvalidateList(LIST_TUBES);
238 InvalidateList(LIST_SURFACE_LEGS);
239 InvalidateList(LIST_BLOBS);
240 InvalidateList(LIST_CROSSES);
241 InvalidateList(LIST_GRID);
242 InvalidateList(LIST_SHADOW);
243 InvalidateList(LIST_TERRAIN);
245 // Set diameter of the viewing volume.
246 double cave_diameter = sqrt(sqrd(m_Parent->GetXExtent()) +
247 sqrd(m_Parent->GetYExtent()) +
248 sqrd(m_Parent->GetZExtent()));
250 // Allow for terrain.
251 double diameter = max(1000.0 * 2, cave_diameter * 2);
253 if (!same_file) {
254 SetVolumeDiameter(diameter);
256 // Set initial scale based on the size of the cave.
257 initial_scale = diameter / cave_diameter;
258 SetScale(initial_scale);
259 } else {
260 // Adjust the position when restricting the view to a subsurvey (or
261 // expanding the view to show the whole survey).
262 AddTranslation(m_Parent->GetOffset() - offsets);
264 // Try to keep the same scale, allowing for the
265 // cave having grown (or shrunk).
266 double rescale = GetVolumeDiameter() / diameter;
267 SetVolumeDiameter(diameter);
268 SetScale(GetScale() / rescale); // ?
269 initial_scale = initial_scale * rescale;
272 offsets = m_Parent->GetOffset();
274 ForceRefresh();
277 void GfxCore::FirstShow()
279 GLACanvas::FirstShow();
281 const unsigned int quantise(GetFontSize() / QUANTISE_FACTOR);
282 list<LabelInfo*>::iterator pos = m_Parent->GetLabelsNC();
283 while (pos != m_Parent->GetLabelsNCEnd()) {
284 LabelInfo* label = *pos++;
285 // Calculate and set the label width for use when plotting
286 // none-overlapping labels.
287 int ext_x;
288 GLACanvas::GetTextExtent(label->GetText(), &ext_x, NULL);
289 label->set_width(unsigned(ext_x) / quantise + 1);
292 m_DoneFirstShow = true;
296 // Recalculating methods
299 void GfxCore::SetScale(Double scale)
301 if (scale < 0.05) {
302 scale = 0.05;
303 } else if (scale > GetVolumeDiameter()) {
304 scale = GetVolumeDiameter();
307 m_Scale = scale;
308 m_HitTestGridValid = false;
309 if (m_here && m_here == &temp_here) SetHere();
311 GLACanvas::SetScale(scale);
314 bool GfxCore::HasUndergroundLegs() const
316 return m_Parent->HasUndergroundLegs();
319 bool GfxCore::HasSplays() const
321 return m_Parent->HasSplays();
324 bool GfxCore::HasDupes() const
326 return m_Parent->HasDupes();
329 bool GfxCore::HasSurfaceLegs() const
331 return m_Parent->HasSurfaceLegs();
334 bool GfxCore::HasTubes() const
336 return m_Parent->HasTubes();
339 void GfxCore::UpdateBlobs()
341 InvalidateList(LIST_BLOBS);
345 // Event handlers
348 void GfxCore::OnLeaveWindow(wxMouseEvent&) {
349 SetHere();
350 ClearCoords();
353 void GfxCore::OnIdle(wxIdleEvent& event)
355 // Handle an idle event.
356 if (Animating()) {
357 Animate();
358 // If still animating, we want more idle events.
359 if (Animating())
360 event.RequestMore();
361 } else {
362 // If we're idle, don't show a bogus FPS next time we render.
363 last_time = 0;
367 void GfxCore::OnPaint(wxPaintEvent&)
369 // Redraw the window.
371 // Get a graphics context.
372 wxPaintDC dc(this);
374 if (m_HaveData) {
375 // Make sure we're initialised.
376 bool first_time = !m_DoneFirstShow;
377 if (first_time) {
378 FirstShow();
381 StartDrawing();
383 // Clear the background.
384 Clear();
386 // Set up model transformation matrix.
387 SetDataTransform();
389 if (m_Legs || m_Tubes) {
390 if (m_Tubes) {
391 EnableSmoothPolygons(true); // FIXME: allow false for wireframe view
392 DrawList(LIST_TUBES);
393 DisableSmoothPolygons();
396 // Draw the underground legs. Do this last so that anti-aliasing
397 // works over polygons.
398 SetColour(col_GREEN);
399 DrawList(LIST_UNDERGROUND_LEGS);
402 if (m_Surface) {
403 // Draw the surface legs.
404 DrawList(LIST_SURFACE_LEGS);
407 if (m_BoundingBox) {
408 DrawShadowedBoundingBox();
410 if (m_Grid) {
411 // Draw the grid.
412 DrawList(LIST_GRID);
415 DrawList(LIST_BLOBS);
417 if (m_Crosses) {
418 DrawList(LIST_CROSSES);
421 if (m_Terrain) {
422 // This is needed if blobs and/or crosses are drawn using lines -
423 // otherwise the terrain doesn't appear when they are enabled.
424 SetDataTransform();
426 // We don't want to be able to see the terrain through itself, so
427 // do a "Z-prepass" - plot the terrain once only updating the
428 // Z-buffer, then again with Z-clipping only plotting where the
429 // depth matches the value in the Z-buffer.
430 DrawListZPrepass(LIST_TERRAIN);
433 SetIndicatorTransform();
435 // Draw station names.
436 if (m_Names /*&& !m_Control->MouseDown() && !Animating()*/) {
437 SetColour(NAME_COLOUR);
439 if (m_OverlappingNames) {
440 SimpleDrawNames();
441 } else {
442 NattyDrawNames();
446 if (m_HitTestDebug) {
447 // Show the hit test grid bucket sizes...
448 SetColour(m_HitTestGridValid ? col_LIGHT_GREY : col_DARK_GREY);
449 if (m_PointGrid) {
450 for (int i = 0; i != HITTEST_SIZE; ++i) {
451 int x = (GetXSize() + 1) * i / HITTEST_SIZE + 2;
452 for (int j = 0; j != HITTEST_SIZE; ++j) {
453 int square = i + j * HITTEST_SIZE;
454 unsigned long bucket_size = m_PointGrid[square].size();
455 if (bucket_size) {
456 int y = (GetYSize() + 1) * (HITTEST_SIZE - 1 - j) / HITTEST_SIZE;
457 DrawIndicatorText(x, y, wxString::Format(wxT("%lu"), bucket_size));
463 EnableDashedLines();
464 BeginLines();
465 for (int i = 0; i != HITTEST_SIZE; ++i) {
466 int x = (GetXSize() + 1) * i / HITTEST_SIZE;
467 PlaceIndicatorVertex(x, 0);
468 PlaceIndicatorVertex(x, GetYSize());
470 for (int j = 0; j != HITTEST_SIZE; ++j) {
471 int y = (GetYSize() + 1) * (HITTEST_SIZE - 1 - j) / HITTEST_SIZE;
472 PlaceIndicatorVertex(0, y);
473 PlaceIndicatorVertex(GetXSize(), y);
475 EndLines();
476 DisableDashedLines();
479 long now = timer.Time();
480 if (m_RenderStats) {
481 // Show stats about rendering.
482 SetColour(col_TURQUOISE);
483 int y = GetYSize() - GetFontSize();
484 if (last_time != 0.0) {
485 // timer.Time() measure in milliseconds.
486 double fps = 1000.0 / (now - last_time);
487 DrawIndicatorText(1, y, wxString::Format(wxT("FPS:% 5.1f"), fps));
489 y -= GetFontSize();
490 DrawIndicatorText(1, y, wxString::Format(wxT("▲:%lu"), (unsigned long)n_tris));
492 last_time = now;
494 // Draw indicators.
496 // There's no advantage in generating an OpenGL list for the
497 // indicators since they change with almost every redraw (and
498 // sometimes several times between redraws). This way we avoid
499 // the need to track when to update the indicator OpenGL list,
500 // and also avoid indicator update bugs when we don't quite get this
501 // right...
502 DrawIndicators();
504 if (zoombox.active()) {
505 SetColour(SEL_COLOUR);
506 EnableDashedLines();
507 BeginPolyline();
508 glaCoord Y = GetYSize();
509 PlaceIndicatorVertex(zoombox.x1, Y - zoombox.y1);
510 PlaceIndicatorVertex(zoombox.x1, Y - zoombox.y2);
511 PlaceIndicatorVertex(zoombox.x2, Y - zoombox.y2);
512 PlaceIndicatorVertex(zoombox.x2, Y - zoombox.y1);
513 PlaceIndicatorVertex(zoombox.x1, Y - zoombox.y1);
514 EndPolyline();
515 DisableDashedLines();
516 } else if (MeasuringLineActive()) {
517 // Draw "here" and "there".
518 double hx, hy;
519 SetColour(HERE_COLOUR);
520 if (m_here) {
521 double dummy;
522 Transform(*m_here, &hx, &hy, &dummy);
523 if (m_here != &temp_here) DrawRing(hx, hy);
525 if (m_there) {
526 double tx, ty;
527 double dummy;
528 Transform(*m_there, &tx, &ty, &dummy);
529 if (m_here) {
530 BeginLines();
531 PlaceIndicatorVertex(hx, hy);
532 PlaceIndicatorVertex(tx, ty);
533 EndLines();
535 BeginBlobs();
536 DrawBlob(tx, ty);
537 EndBlobs();
541 FinishDrawing();
542 } else {
543 dc.SetBackground(wxSystemSettings::GetColour(wxSYS_COLOUR_WINDOWFRAME));
544 dc.Clear();
548 void GfxCore::DrawBoundingBox()
550 const Vector3 v = 0.5 * m_Parent->GetExtent();
552 SetColour(col_BLUE);
553 EnableDashedLines();
554 BeginPolyline();
555 PlaceVertex(-v.GetX(), -v.GetY(), v.GetZ());
556 PlaceVertex(-v.GetX(), v.GetY(), v.GetZ());
557 PlaceVertex(v.GetX(), v.GetY(), v.GetZ());
558 PlaceVertex(v.GetX(), -v.GetY(), v.GetZ());
559 PlaceVertex(-v.GetX(), -v.GetY(), v.GetZ());
560 EndPolyline();
561 BeginPolyline();
562 PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
563 PlaceVertex(-v.GetX(), v.GetY(), -v.GetZ());
564 PlaceVertex(v.GetX(), v.GetY(), -v.GetZ());
565 PlaceVertex(v.GetX(), -v.GetY(), -v.GetZ());
566 PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
567 EndPolyline();
568 BeginLines();
569 PlaceVertex(-v.GetX(), -v.GetY(), v.GetZ());
570 PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
571 PlaceVertex(-v.GetX(), v.GetY(), v.GetZ());
572 PlaceVertex(-v.GetX(), v.GetY(), -v.GetZ());
573 PlaceVertex(v.GetX(), v.GetY(), v.GetZ());
574 PlaceVertex(v.GetX(), v.GetY(), -v.GetZ());
575 PlaceVertex(v.GetX(), -v.GetY(), v.GetZ());
576 PlaceVertex(v.GetX(), -v.GetY(), -v.GetZ());
577 EndLines();
578 DisableDashedLines();
581 void GfxCore::DrawShadowedBoundingBox()
583 const Vector3 v = 0.5 * m_Parent->GetExtent();
585 DrawBoundingBox();
587 PolygonOffset(true);
588 SetColour(col_DARK_GREY);
589 BeginQuadrilaterals();
590 PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
591 PlaceVertex(-v.GetX(), v.GetY(), -v.GetZ());
592 PlaceVertex(v.GetX(), v.GetY(), -v.GetZ());
593 PlaceVertex(v.GetX(), -v.GetY(), -v.GetZ());
594 EndQuadrilaterals();
595 PolygonOffset(false);
597 DrawList(LIST_SHADOW);
600 void GfxCore::DrawGrid()
602 // Draw the grid.
603 SetColour(col_RED);
605 // Calculate the extent of the survey, in metres across the screen plane.
606 Double m_across_screen = SurveyUnitsAcrossViewport();
607 // Calculate the length of the scale bar in metres.
608 //--move this elsewhere
609 Double size_snap = pow(10.0, floor(log10(0.75 * m_across_screen)));
610 Double t = m_across_screen * 0.75 / size_snap;
611 if (t >= 5.0) {
612 size_snap *= 5.0;
614 else if (t >= 2.0) {
615 size_snap *= 2.0;
618 Double grid_size = size_snap * 0.1;
619 Double edge = grid_size * 2.0;
620 Double grid_z = -m_Parent->GetZExtent() * 0.5 - grid_size;
621 Double left = -m_Parent->GetXExtent() * 0.5 - edge;
622 Double right = m_Parent->GetXExtent() * 0.5 + edge;
623 Double bottom = -m_Parent->GetYExtent() * 0.5 - edge;
624 Double top = m_Parent->GetYExtent() * 0.5 + edge;
625 int count_x = (int) ceil((right - left) / grid_size);
626 int count_y = (int) ceil((top - bottom) / grid_size);
627 Double actual_right = left + count_x*grid_size;
628 Double actual_top = bottom + count_y*grid_size;
630 BeginLines();
632 for (int xc = 0; xc <= count_x; xc++) {
633 Double x = left + xc*grid_size;
635 PlaceVertex(x, bottom, grid_z);
636 PlaceVertex(x, actual_top, grid_z);
639 for (int yc = 0; yc <= count_y; yc++) {
640 Double y = bottom + yc*grid_size;
641 PlaceVertex(left, y, grid_z);
642 PlaceVertex(actual_right, y, grid_z);
645 EndLines();
648 int GfxCore::GetClinoOffset() const
650 int result = INDICATOR_OFFSET_X;
651 if (m_Compass) {
652 result += 6 + GetCompassWidth() + INDICATOR_GAP;
654 return result;
657 void GfxCore::DrawTick(int angle_cw)
659 const Double theta = rad(angle_cw);
660 const wxCoord length1 = INDICATOR_RADIUS;
661 const wxCoord length0 = length1 + TICK_LENGTH;
662 wxCoord x0 = wxCoord(length0 * sin(theta));
663 wxCoord y0 = wxCoord(length0 * cos(theta));
664 wxCoord x1 = wxCoord(length1 * sin(theta));
665 wxCoord y1 = wxCoord(length1 * cos(theta));
667 PlaceIndicatorVertex(x0, y0);
668 PlaceIndicatorVertex(x1, y1);
671 void GfxCore::DrawArrow(gla_colour col1, gla_colour col2) {
672 Vector3 p1(0, INDICATOR_RADIUS, 0);
673 Vector3 p2(INDICATOR_RADIUS/2, INDICATOR_RADIUS*-.866025404, 0); // 150deg
674 Vector3 p3(-INDICATOR_RADIUS/2, INDICATOR_RADIUS*-.866025404, 0); // 210deg
675 Vector3 pc(0, 0, 0);
677 DrawTriangle(col_LIGHT_GREY, col1, p2, p1, pc);
678 DrawTriangle(col_LIGHT_GREY, col2, p3, p1, pc);
681 void GfxCore::DrawCompass() {
682 // Ticks.
683 BeginLines();
684 for (int angle = 315; angle > 0; angle -= 45) {
685 DrawTick(angle);
687 SetColour(col_GREEN);
688 DrawTick(0);
689 EndLines();
691 // Compass background.
692 DrawCircle(col_LIGHT_GREY_2, col_GREY, 0, 0, INDICATOR_RADIUS);
694 // Compass arrow.
695 DrawArrow(col_INDICATOR_1, col_INDICATOR_2);
698 // Draw the non-rotating background to the clino.
699 void GfxCore::DrawClinoBack() {
700 BeginLines();
701 for (int angle = 0; angle <= 180; angle += 90) {
702 DrawTick(angle);
705 SetColour(col_GREY);
706 PlaceIndicatorVertex(0, INDICATOR_RADIUS);
707 PlaceIndicatorVertex(0, -INDICATOR_RADIUS);
708 PlaceIndicatorVertex(0, 0);
709 PlaceIndicatorVertex(INDICATOR_RADIUS, 0);
711 EndLines();
714 void GfxCore::DrawClino() {
715 // Ticks.
716 SetColour(col_GREEN);
717 BeginLines();
718 DrawTick(0);
719 EndLines();
721 // Clino background.
722 DrawSemicircle(col_LIGHT_GREY_2, col_GREY, 0, 0, INDICATOR_RADIUS, 0);
724 // Elevation arrow.
725 DrawArrow(col_INDICATOR_2, col_INDICATOR_1);
728 void GfxCore::Draw2dIndicators()
730 // Draw the compass and elevation indicators.
732 const int centre_y = INDICATOR_BOX_SIZE / 2 + INDICATOR_OFFSET_Y;
734 const int comp_centre_x = GetCompassXPosition();
736 if (m_Compass && !m_Parent->IsExtendedElevation()) {
737 // If the user is dragging the compass with the pointer outside the
738 // compass, we snap to 45 degree multiples, and the ticks go white.
739 SetColour(m_MouseOutsideCompass ? col_WHITE : col_LIGHT_GREY_2);
740 DrawList2D(LIST_COMPASS, comp_centre_x, centre_y, -m_PanAngle);
743 const int elev_centre_x = GetClinoXPosition();
745 if (m_Clino) {
746 // If the user is dragging the clino with the pointer outside the
747 // clino, we snap to 90 degree multiples, and the ticks go white.
748 SetColour(m_MouseOutsideElev ? col_WHITE : col_LIGHT_GREY_2);
749 DrawList2D(LIST_CLINO_BACK, elev_centre_x, centre_y, 0);
750 DrawList2D(LIST_CLINO, elev_centre_x, centre_y, 90 - m_TiltAngle);
753 SetColour(TEXT_COLOUR);
755 static int triple_zero_width = 0;
756 static int height = 0;
757 if (!triple_zero_width) {
758 GetTextExtent(wxT("000"), &triple_zero_width, &height);
760 const int y_off = INDICATOR_OFFSET_Y + INDICATOR_BOX_SIZE + height / 2;
762 if (m_Compass && !m_Parent->IsExtendedElevation()) {
763 wxString str;
764 int value;
765 int brg_unit;
766 if (m_Degrees) {
767 value = int(m_PanAngle);
768 /* TRANSLATORS: degree symbol - probably should be translated to
769 * itself. */
770 brg_unit = /*°*/344;
771 } else {
772 value = int(m_PanAngle * 200.0 / 180.0);
773 /* TRANSLATORS: symbol for grad (400 grad = 360 degrees = full
774 * circle). */
775 brg_unit = /*ᵍ*/76;
777 str.Printf(wxT("%03d"), value);
778 str += wmsg(brg_unit);
779 DrawIndicatorText(comp_centre_x - triple_zero_width / 2, y_off, str);
781 // TRANSLATORS: Used in aven above the compass indicator at the lower
782 // right of the display, with a bearing below "Facing". This indicates the
783 // direction the viewer is "facing" in.
785 // Try to keep this translation short - ideally at most 10 characters -
786 // as otherwise the compass and clino will be moved further apart to
787 // make room. */
788 str = wmsg(/*Facing*/203);
789 int w;
790 GetTextExtent(str, &w, NULL);
791 DrawIndicatorText(comp_centre_x - w / 2, y_off + height, str);
794 if (m_Clino) {
795 if (m_TiltAngle == -90.0) {
796 // TRANSLATORS: Label used for "clino" in Aven when the view is
797 // from directly above.
799 // Try to keep this translation short - ideally at most 10
800 // characters - as otherwise the compass and clino will be moved
801 // further apart to make room. */
802 wxString str = wmsg(/*Plan*/432);
803 static int width = 0;
804 if (!width) {
805 GetTextExtent(str, &width, NULL);
807 int x = elev_centre_x - width / 2;
808 DrawIndicatorText(x, y_off + height / 2, str);
809 } else if (m_TiltAngle == 90.0) {
810 // TRANSLATORS: Label used for "clino" in Aven when the view is
811 // from directly below.
813 // Try to keep this translation short - ideally at most 10
814 // characters - as otherwise the compass and clino will be moved
815 // further apart to make room. */
816 wxString str = wmsg(/*Kiwi Plan*/433);
817 static int width = 0;
818 if (!width) {
819 GetTextExtent(str, &width, NULL);
821 int x = elev_centre_x - width / 2;
822 DrawIndicatorText(x, y_off + height / 2, str);
823 } else {
824 int angle;
825 wxString str;
826 int width;
827 int unit;
828 if (m_Percent) {
829 static int zero_width = 0;
830 if (!zero_width) {
831 GetTextExtent(wxT("0"), &zero_width, NULL);
833 width = zero_width;
834 if (m_TiltAngle > 89.99) {
835 angle = 1000000;
836 } else if (m_TiltAngle < -89.99) {
837 angle = -1000000;
838 } else {
839 angle = int(100 * tan(rad(m_TiltAngle)));
841 if (angle > 99999 || angle < -99999) {
842 str = angle > 0 ? wxT("+") : wxT("-");
843 /* TRANSLATORS: infinity symbol - used for the percentage gradient on
844 * vertical angles. */
845 str += wmsg(/*∞*/431);
846 } else {
847 str = angle ? wxString::Format(wxT("%+03d"), angle) : wxT("0");
849 /* TRANSLATORS: symbol for percentage gradient (100% = 45
850 * degrees = 50 grad). */
851 unit = /*%*/96;
852 } else if (m_Degrees) {
853 static int zero_zero_width = 0;
854 if (!zero_zero_width) {
855 GetTextExtent(wxT("00"), &zero_zero_width, NULL);
857 width = zero_zero_width;
858 angle = int(m_TiltAngle);
859 str = angle ? wxString::Format(wxT("%+03d"), angle) : wxT("00");
860 unit = /*°*/344;
861 } else {
862 width = triple_zero_width;
863 angle = int(m_TiltAngle * 200.0 / 180.0);
864 str = angle ? wxString::Format(wxT("%+04d"), angle) : wxT("000");
865 unit = /*ᵍ*/76;
868 int sign_offset = 0;
869 if (unit == /*%*/96) {
870 // Right align % since the width changes so much.
871 GetTextExtent(str, &sign_offset, NULL);
872 sign_offset -= width;
873 } else if (angle < 0) {
874 // Adjust horizontal position so the left of the first digit is
875 // always in the same place.
876 static int minus_width = 0;
877 if (!minus_width) {
878 GetTextExtent(wxT("-"), &minus_width, NULL);
880 sign_offset = minus_width;
881 } else if (angle > 0) {
882 // Adjust horizontal position so the left of the first digit is
883 // always in the same place.
884 static int plus_width = 0;
885 if (!plus_width) {
886 GetTextExtent(wxT("+"), &plus_width, NULL);
888 sign_offset = plus_width;
891 str += wmsg(unit);
892 DrawIndicatorText(elev_centre_x - sign_offset - width / 2, y_off, str);
894 // TRANSLATORS: Label used for "clino" in Aven when the view is
895 // neither from directly above nor from directly below. It is
896 // also used in the dialog for editing a marked position in a
897 // presentation.
899 // Try to keep this translation short - ideally at most 10
900 // characters - as otherwise the compass and clino will be moved
901 // further apart to make room. */
902 str = wmsg(/*Elevation*/118);
903 static int elevation_width = 0;
904 if (!elevation_width) {
905 GetTextExtent(str, &elevation_width, NULL);
907 int x = elev_centre_x - elevation_width / 2;
908 DrawIndicatorText(x, y_off + height, str);
913 void GfxCore::NattyDrawNames()
915 // Draw station names, without overlapping.
917 const unsigned int quantise(GetFontSize() / QUANTISE_FACTOR);
918 const unsigned int quantised_x = GetXSize() / quantise;
919 const unsigned int quantised_y = GetYSize() / quantise;
920 const size_t buffer_size = quantised_x * quantised_y;
922 if (!m_LabelGrid) m_LabelGrid = new char[buffer_size];
924 memset((void*) m_LabelGrid, 0, buffer_size);
926 list<LabelInfo*>::const_iterator label = m_Parent->GetLabels();
927 for ( ; label != m_Parent->GetLabelsEnd(); ++label) {
928 if (!((m_Surface && (*label)->IsSurface()) ||
929 (m_Legs && (*label)->IsUnderground()) ||
930 (!(*label)->IsSurface() && !(*label)->IsUnderground()))) {
931 // if this station isn't to be displayed, skip to the next
932 // (last case is for stns with no legs attached)
933 continue;
936 double x, y, z;
938 Transform(**label, &x, &y, &z);
939 // Check if the label is behind us (in perspective view).
940 if (z <= 0.0 || z >= 1.0) continue;
942 // Apply a small shift so that translating the view doesn't make which
943 // labels are displayed change as the resulting twinkling effect is
944 // distracting.
945 double tx, ty, tz;
946 Transform(Vector3(), &tx, &ty, &tz);
947 tx -= floor(tx / quantise) * quantise;
948 ty -= floor(ty / quantise) * quantise;
950 tx = x - tx;
951 if (tx < 0) continue;
953 ty = y - ty;
954 if (ty < 0) continue;
956 unsigned int iy = unsigned(ty) / quantise;
957 if (iy >= quantised_y) continue;
958 unsigned int width = (*label)->get_width();
959 unsigned int ix = unsigned(tx) / quantise;
960 if (ix + width >= quantised_x) continue;
962 char * test = m_LabelGrid + ix + iy * quantised_x;
963 if (memchr(test, 1, width)) continue;
965 x += 3;
966 y -= GetFontSize() / 2;
967 DrawIndicatorText((int)x, (int)y, (*label)->GetText());
969 if (iy > QUANTISE_FACTOR) iy = QUANTISE_FACTOR;
970 test -= quantised_x * iy;
971 iy += 4;
972 while (--iy && test < m_LabelGrid + buffer_size) {
973 memset(test, 1, width);
974 test += quantised_x;
979 void GfxCore::SimpleDrawNames()
981 // Draw all station names, without worrying about overlaps
982 list<LabelInfo*>::const_iterator label = m_Parent->GetLabels();
983 for ( ; label != m_Parent->GetLabelsEnd(); ++label) {
984 if (!((m_Surface && (*label)->IsSurface()) ||
985 (m_Legs && (*label)->IsUnderground()) ||
986 (!(*label)->IsSurface() && !(*label)->IsUnderground()))) {
987 // if this station isn't to be displayed, skip to the next
988 // (last case is for stns with no legs attached)
989 continue;
992 double x, y, z;
993 Transform(**label, &x, &y, &z);
995 // Check if the label is behind us (in perspective view).
996 if (z <= 0) continue;
998 x += 3;
999 y -= GetFontSize() / 2;
1000 DrawIndicatorText((int)x, (int)y, (*label)->GetText());
1004 void GfxCore::DrawColourKey(int num_bands, const wxString & other, const wxString & units)
1006 int total_block_height =
1007 KEY_BLOCK_HEIGHT * (num_bands == 1 ? num_bands : num_bands - 1);
1008 if (!other.empty()) total_block_height += KEY_BLOCK_HEIGHT * 2;
1009 if (!units.empty()) total_block_height += KEY_BLOCK_HEIGHT;
1011 const int bottom = -total_block_height;
1013 int size = 0;
1014 if (!other.empty()) GetTextExtent(other, &size, NULL);
1015 int band;
1016 for (band = 0; band < num_bands; ++band) {
1017 int x;
1018 GetTextExtent(key_legends[band], &x, NULL);
1019 if (x > size) size = x;
1022 int left = -KEY_BLOCK_WIDTH - size;
1024 key_lowerleft[m_ColourBy].x = left - KEY_EXTRA_LEFT_MARGIN;
1025 key_lowerleft[m_ColourBy].y = bottom;
1027 int y = bottom;
1028 if (!units.empty()) y += KEY_BLOCK_HEIGHT;
1030 if (!other.empty()) {
1031 DrawShadedRectangle(GetSurfacePen(), GetSurfacePen(), left, y,
1032 KEY_BLOCK_WIDTH, KEY_BLOCK_HEIGHT);
1033 SetColour(col_BLACK);
1034 BeginPolyline();
1035 PlaceIndicatorVertex(left, y);
1036 PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, y);
1037 PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, y + KEY_BLOCK_HEIGHT);
1038 PlaceIndicatorVertex(left, y + KEY_BLOCK_HEIGHT);
1039 PlaceIndicatorVertex(left, y);
1040 EndPolyline();
1041 y += KEY_BLOCK_HEIGHT * 2;
1044 int start = y;
1045 if (num_bands == 1) {
1046 DrawShadedRectangle(GetPen(0), GetPen(0), left, y,
1047 KEY_BLOCK_WIDTH, KEY_BLOCK_HEIGHT);
1048 y += KEY_BLOCK_HEIGHT;
1049 } else {
1050 for (band = 0; band < num_bands - 1; ++band) {
1051 DrawShadedRectangle(GetPen(band), GetPen(band + 1), left, y,
1052 KEY_BLOCK_WIDTH, KEY_BLOCK_HEIGHT);
1053 y += KEY_BLOCK_HEIGHT;
1057 SetColour(col_BLACK);
1058 BeginPolyline();
1059 PlaceIndicatorVertex(left, y);
1060 PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, y);
1061 PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, start);
1062 PlaceIndicatorVertex(left, start);
1063 PlaceIndicatorVertex(left, y);
1064 EndPolyline();
1066 SetColour(TEXT_COLOUR);
1068 y = bottom;
1069 if (!units.empty()) {
1070 GetTextExtent(units, &size, NULL);
1071 DrawIndicatorText(left + (KEY_BLOCK_WIDTH - size) / 2, y, units);
1072 y += KEY_BLOCK_HEIGHT;
1074 y -= GetFontSize() / 2;
1075 left += KEY_BLOCK_WIDTH + 5;
1077 if (!other.empty()) {
1078 y += KEY_BLOCK_HEIGHT / 2;
1079 DrawIndicatorText(left, y, other);
1080 y += KEY_BLOCK_HEIGHT * 2 - KEY_BLOCK_HEIGHT / 2;
1083 if (num_bands == 1) {
1084 y += KEY_BLOCK_HEIGHT / 2;
1085 DrawIndicatorText(left, y, key_legends[0]);
1086 } else {
1087 for (band = 0; band < num_bands; ++band) {
1088 DrawIndicatorText(left, y, key_legends[band]);
1089 y += KEY_BLOCK_HEIGHT;
1094 void GfxCore::DrawDepthKey()
1096 Double z_ext = m_Parent->GetDepthExtent();
1097 int num_bands = 1;
1098 int sf = 0;
1099 if (z_ext > 0.0) {
1100 num_bands = GetNumColourBands();
1101 Double z_range = z_ext;
1102 if (!m_Metric) z_range /= METRES_PER_FOOT;
1103 sf = max(0, 1 - (int)floor(log10(z_range)));
1106 Double z_min = m_Parent->GetDepthMin() + m_Parent->GetOffset().GetZ();
1107 for (int band = 0; band < num_bands; ++band) {
1108 Double z = z_min;
1109 if (band)
1110 z += z_ext * band / (num_bands - 1);
1112 if (!m_Metric)
1113 z /= METRES_PER_FOOT;
1115 key_legends[band].Printf(wxT("%.*f"), sf, z);
1118 DrawColourKey(num_bands, wxString(), wmsg(m_Metric ? /*m*/424: /*ft*/428));
1121 void GfxCore::DrawDateKey()
1123 int num_bands;
1124 if (!HasDateInformation()) {
1125 num_bands = 0;
1126 } else {
1127 int date_ext = m_Parent->GetDateExtent();
1128 if (date_ext == 0) {
1129 num_bands = 1;
1130 } else {
1131 num_bands = GetNumColourBands();
1133 for (int band = 0; band < num_bands; ++band) {
1134 int y, m, d;
1135 int days = m_Parent->GetDateMin();
1136 if (band)
1137 days += date_ext * band / (num_bands - 1);
1138 ymd_from_days_since_1900(days, &y, &m, &d);
1139 key_legends[band].Printf(wxT("%04d-%02d-%02d"), y, m, d);
1143 wxString other;
1144 if (!m_Parent->HasCompleteDateInfo()) {
1145 /* TRANSLATORS: Used in the "colour key" for "colour by date" if there
1146 * are surveys without date information. Try to keep this fairly short.
1148 other = wmsg(/*Undated*/221);
1151 DrawColourKey(num_bands, other, wxString());
1154 void GfxCore::DrawErrorKey()
1156 int num_bands;
1157 if (HasErrorInformation()) {
1158 // Use fixed colours for each error factor so it's directly visually
1159 // comparable between surveys.
1160 num_bands = GetNumColourBands();
1161 for (int band = 0; band < num_bands; ++band) {
1162 double E = MAX_ERROR * band / (num_bands - 1);
1163 key_legends[band].Printf(wxT("%.2f"), E);
1165 } else {
1166 num_bands = 0;
1169 // Always show the "Not in loop" legend for now (FIXME).
1170 /* TRANSLATORS: Used in the "colour key" for "colour by error" for surveys
1171 * which aren’t part of a loop and so have no error information. Try to keep
1172 * this fairly short. */
1173 DrawColourKey(num_bands, wmsg(/*Not in loop*/290), wxString());
1176 void GfxCore::DrawGradientKey()
1178 int num_bands;
1179 // Use fixed colours for each gradient so it's directly visually comparable
1180 // between surveys.
1181 num_bands = GetNumColourBands();
1182 wxString units = wmsg(m_Degrees ? /*°*/344 : /*ᵍ*/76);
1183 for (int band = 0; band < num_bands; ++band) {
1184 double gradient = double(band) / (num_bands - 1);
1185 if (m_Degrees) {
1186 gradient *= 90.0;
1187 } else {
1188 gradient *= 100.0;
1190 key_legends[band].Printf(wxT("%.f%s"), gradient, units);
1193 DrawColourKey(num_bands, wxString(), wxString());
1196 void GfxCore::DrawLengthKey()
1198 int num_bands;
1199 // Use fixed colours for each length so it's directly visually comparable
1200 // between surveys.
1201 num_bands = GetNumColourBands();
1202 for (int band = 0; band < num_bands; ++band) {
1203 double len = pow(10, LOG_LEN_MAX * band / (num_bands - 1));
1204 if (!m_Metric) {
1205 len /= METRES_PER_FOOT;
1207 key_legends[band].Printf(wxT("%.1f"), len);
1210 DrawColourKey(num_bands, wxString(), wmsg(m_Metric ? /*m*/424: /*ft*/428));
1213 void GfxCore::DrawScaleBar()
1215 // Draw the scalebar.
1216 if (GetPerspective()) return;
1218 // Calculate how many metres of survey are currently displayed across the
1219 // screen.
1220 Double across_screen = SurveyUnitsAcrossViewport();
1222 double f = double(GetClinoXPosition() - INDICATOR_BOX_SIZE / 2 - SCALE_BAR_OFFSET_X) / GetXSize();
1223 if (f > 0.75) {
1224 f = 0.75;
1225 } else if (f < 0.5) {
1226 // Stop it getting squeezed to nothing.
1227 // FIXME: In this case we should probably move the compass and clino up
1228 // to make room rather than letting stuff overlap.
1229 f = 0.5;
1232 // Convert to imperial measurements if required.
1233 Double multiplier = 1.0;
1234 if (!m_Metric) {
1235 across_screen /= METRES_PER_FOOT;
1236 multiplier = METRES_PER_FOOT;
1237 if (across_screen >= 5280.0 / f) {
1238 across_screen /= 5280.0;
1239 multiplier *= 5280.0;
1243 // Calculate the length of the scale bar.
1244 Double size_snap = pow(10.0, floor(log10(f * across_screen)));
1245 Double t = across_screen * f / size_snap;
1246 if (t >= 5.0) {
1247 size_snap *= 5.0;
1248 } else if (t >= 2.0) {
1249 size_snap *= 2.0;
1252 if (!m_Metric) size_snap *= multiplier;
1254 // Actual size of the thing in pixels:
1255 int size = int((size_snap / SurveyUnitsAcrossViewport()) * GetXSize());
1256 m_ScaleBarWidth = size;
1258 // Draw it...
1259 const int end_y = SCALE_BAR_OFFSET_Y + SCALE_BAR_HEIGHT;
1260 int interval = size / 10;
1262 gla_colour col = col_WHITE;
1263 for (int ix = 0; ix < 10; ix++) {
1264 int x = SCALE_BAR_OFFSET_X + int(ix * ((Double) size / 10.0));
1266 DrawRectangle(col, col, x, end_y, interval + 2, SCALE_BAR_HEIGHT);
1268 col = (col == col_WHITE) ? col_GREY : col_WHITE;
1271 // Add labels.
1272 wxString str;
1273 int units;
1274 if (m_Metric) {
1275 Double km = size_snap * 1e-3;
1276 if (km >= 1.0) {
1277 size_snap = km;
1278 /* TRANSLATORS: abbreviation for "kilometres" (unit of length),
1279 * used e.g. "5km".
1281 * If there should be a space between the number and this, include
1282 * one in the translation. */
1283 units = /*km*/423;
1284 } else if (size_snap >= 1.0) {
1285 /* TRANSLATORS: abbreviation for "metres" (unit of length), used
1286 * e.g. "10m".
1288 * If there should be a space between the number and this, include
1289 * one in the translation. */
1290 units = /*m*/424;
1291 } else {
1292 size_snap *= 1e2;
1293 /* TRANSLATORS: abbreviation for "centimetres" (unit of length),
1294 * used e.g. "50cm".
1296 * If there should be a space between the number and this, include
1297 * one in the translation. */
1298 units = /*cm*/425;
1300 } else {
1301 size_snap /= METRES_PER_FOOT;
1302 Double miles = size_snap / 5280.0;
1303 if (miles >= 1.0) {
1304 size_snap = miles;
1305 if (size_snap >= 2.0) {
1306 /* TRANSLATORS: abbreviation for "miles" (unit of length,
1307 * plural), used e.g. "2 miles".
1309 * If there should be a space between the number and this,
1310 * include one in the translation. */
1311 units = /* miles*/426;
1312 } else {
1313 /* TRANSLATORS: abbreviation for "mile" (unit of length,
1314 * singular), used e.g. "1 mile".
1316 * If there should be a space between the number and this,
1317 * include one in the translation. */
1318 units = /* mile*/427;
1320 } else if (size_snap >= 1.0) {
1321 /* TRANSLATORS: abbreviation for "feet" (unit of length), used e.g.
1322 * as "10ft".
1324 * If there should be a space between the number and this, include
1325 * one in the translation. */
1326 units = /*ft*/428;
1327 } else {
1328 size_snap *= 12.0;
1329 /* TRANSLATORS: abbreviation for "inches" (unit of length), used
1330 * e.g. as "6in".
1332 * If there should be a space between the number and this, include
1333 * one in the translation. */
1334 units = /*in*/429;
1337 if (size_snap >= 1.0) {
1338 str.Printf(wxT("%.f%s"), size_snap, wmsg(units).c_str());
1339 } else {
1340 int sf = -(int)floor(log10(size_snap));
1341 str.Printf(wxT("%.*f%s"), sf, size_snap, wmsg(units).c_str());
1344 int text_width, text_height;
1345 GetTextExtent(str, &text_width, &text_height);
1346 const int text_y = end_y - text_height + 1;
1347 SetColour(TEXT_COLOUR);
1348 DrawIndicatorText(SCALE_BAR_OFFSET_X, text_y, wxT("0"));
1349 DrawIndicatorText(SCALE_BAR_OFFSET_X + size - text_width, text_y, str);
1352 bool GfxCore::CheckHitTestGrid(const wxPoint& point, bool centre)
1354 if (Animating()) return false;
1356 if (point.x < 0 || point.x >= GetXSize() ||
1357 point.y < 0 || point.y >= GetYSize()) {
1358 return false;
1361 SetDataTransform();
1363 if (!m_HitTestGridValid) CreateHitTestGrid();
1365 int grid_x = point.x * HITTEST_SIZE / (GetXSize() + 1);
1366 int grid_y = point.y * HITTEST_SIZE / (GetYSize() + 1);
1368 LabelInfo *best = NULL;
1369 int dist_sqrd = sqrd_measure_threshold;
1370 int square = grid_x + grid_y * HITTEST_SIZE;
1371 list<LabelInfo*>::iterator iter = m_PointGrid[square].begin();
1373 while (iter != m_PointGrid[square].end()) {
1374 LabelInfo *pt = *iter++;
1376 double cx, cy, cz;
1378 Transform(*pt, &cx, &cy, &cz);
1380 cy = GetYSize() - cy;
1382 int dx = point.x - int(cx);
1383 int ds = dx * dx;
1384 if (ds >= dist_sqrd) continue;
1385 int dy = point.y - int(cy);
1387 ds += dy * dy;
1388 if (ds >= dist_sqrd) continue;
1390 dist_sqrd = ds;
1391 best = pt;
1393 if (ds == 0) break;
1396 if (best) {
1397 m_Parent->ShowInfo(best, m_there);
1398 if (centre) {
1399 // FIXME: allow Ctrl-Click to not set there or something?
1400 CentreOn(*best);
1401 WarpPointer(GetXSize() / 2, GetYSize() / 2);
1402 SetThere(best);
1403 m_Parent->SelectTreeItem(best);
1405 } else {
1406 // Left-clicking not on a survey cancels the measuring line.
1407 if (centre) {
1408 ClearTreeSelection();
1409 } else {
1410 m_Parent->ShowInfo(best, m_there);
1411 double x, y, z;
1412 ReverseTransform(point.x, GetYSize() - point.y, &x, &y, &z);
1413 temp_here.assign(Vector3(x, y, z));
1414 SetHere(&temp_here);
1418 return best;
1421 void GfxCore::OnSize(wxSizeEvent& event)
1423 // Handle a change in window size.
1424 wxSize size = event.GetSize();
1426 if (size.GetWidth() <= 0 || size.GetHeight() <= 0) {
1427 // Before things are fully initialised, we sometimes get a bogus
1428 // resize message...
1429 // FIXME have changes in MainFrm cured this? It still happens with
1430 // 1.0.32 and wxGTK 2.5.2 (load a file from the command line).
1431 // With 1.1.6 and wxGTK 2.4.2 we only get negative sizes if MainFrm
1432 // is resized such that the GfxCore window isn't visible.
1433 //printf("OnSize(%d,%d)\n", size.GetWidth(), size.GetHeight());
1434 return;
1437 event.Skip();
1439 if (m_DoneFirstShow) {
1440 TryToFreeArrays();
1442 m_HitTestGridValid = false;
1444 ForceRefresh();
1448 void GfxCore::DefaultParameters()
1450 // Set default viewing parameters.
1452 m_Surface = false;
1453 if (!m_Parent->HasUndergroundLegs()) {
1454 if (m_Parent->HasSurfaceLegs()) {
1455 // If there are surface legs, but no underground legs, turn
1456 // surface surveys on.
1457 m_Surface = true;
1458 } else {
1459 // If there are no legs (e.g. after loading a .pos file), turn
1460 // crosses on.
1461 m_Crosses = true;
1465 m_PanAngle = 0.0;
1466 if (m_Parent->IsExtendedElevation()) {
1467 m_TiltAngle = 0.0;
1468 } else {
1469 m_TiltAngle = -90.0;
1472 SetRotation(m_PanAngle, m_TiltAngle);
1473 SetTranslation(Vector3());
1475 m_RotationStep = 30.0;
1476 m_Rotating = false;
1477 m_SwitchingTo = 0;
1478 m_Entrances = false;
1479 m_FixedPts = false;
1480 m_ExportedPts = false;
1481 m_Grid = false;
1482 m_BoundingBox = false;
1483 m_Tubes = false;
1484 if (GetPerspective()) TogglePerspective();
1486 // Set the initial scale.
1487 SetScale(initial_scale);
1490 void GfxCore::Defaults()
1492 // Restore default scale, rotation and translation parameters.
1493 DefaultParameters();
1495 // Invalidate all the cached lists.
1496 GLACanvas::FirstShow();
1498 ForceRefresh();
1501 void GfxCore::Animate()
1503 // Don't show pointer coordinates while animating.
1504 // FIXME : only do this when we *START* animating! Use a static copy
1505 // of the value of "Animating()" last time we were here to track this?
1506 // MainFrm now checks if we're trying to clear already cleared labels
1507 // and just returns, but it might be simpler to check here!
1508 ClearCoords();
1509 m_Parent->ShowInfo();
1511 long t;
1512 if (movie) {
1513 ReadPixels(movie->GetWidth(), movie->GetHeight(), movie->GetBuffer());
1514 if (!movie->AddFrame()) {
1515 wxGetApp().ReportError(wxString(movie->get_error_string(), wxConvUTF8));
1516 delete movie;
1517 movie = NULL;
1518 presentation_mode = 0;
1519 return;
1521 t = 1000 / 25; // 25 frames per second
1522 } else {
1523 static long t_prev = 0;
1524 t = timer.Time();
1525 // Avoid redrawing twice in the same frame.
1526 long delta_t = (t_prev == 0 ? 1000 / MAX_FRAMERATE : t - t_prev);
1527 if (delta_t < 1000 / MAX_FRAMERATE)
1528 return;
1529 t_prev = t;
1530 if (presentation_mode == PLAYING && pres_speed != 0.0)
1531 t = delta_t;
1534 if (presentation_mode == PLAYING && pres_speed != 0.0) {
1535 // FIXME: It would probably be better to work relative to the time we
1536 // passed the last mark, but that's complicated by the speed
1537 // potentially changing (or even the direction of playback reversing)
1538 // at any point during playback.
1539 Double tick = t * 0.001 * fabs(pres_speed);
1540 while (tick >= next_mark_time) {
1541 tick -= next_mark_time;
1542 this_mark_total = 0;
1543 PresentationMark prev_mark = next_mark;
1544 if (prev_mark.angle < 0) prev_mark.angle += 360.0;
1545 else if (prev_mark.angle >= 360.0) prev_mark.angle -= 360.0;
1546 if (pres_reverse)
1547 next_mark = m_Parent->GetPresMark(MARK_PREV);
1548 else
1549 next_mark = m_Parent->GetPresMark(MARK_NEXT);
1550 if (!next_mark.is_valid()) {
1551 SetView(prev_mark);
1552 presentation_mode = 0;
1553 if (movie && !movie->Close()) {
1554 wxGetApp().ReportError(wxString(movie->get_error_string(), wxConvUTF8));
1556 delete movie;
1557 movie = NULL;
1558 break;
1561 double tmp = (pres_reverse ? prev_mark.time : next_mark.time);
1562 if (tmp > 0) {
1563 next_mark_time = tmp;
1564 } else {
1565 double d = (next_mark - prev_mark).magnitude();
1566 // FIXME: should ignore component of d which is unseen in
1567 // non-perspective mode?
1568 next_mark_time = sqrd(d / 30.0);
1569 double a = next_mark.angle - prev_mark.angle;
1570 if (a > 180.0) {
1571 next_mark.angle -= 360.0;
1572 a = 360.0 - a;
1573 } else if (a < -180.0) {
1574 next_mark.angle += 360.0;
1575 a += 360.0;
1576 } else {
1577 a = fabs(a);
1579 next_mark_time += sqrd(a / 60.0);
1580 double ta = fabs(next_mark.tilt_angle - prev_mark.tilt_angle);
1581 next_mark_time += sqrd(ta / 60.0);
1582 double s = fabs(log(next_mark.scale) - log(prev_mark.scale));
1583 next_mark_time += sqrd(s / 2.0);
1584 next_mark_time = sqrt(next_mark_time);
1585 // was: next_mark_time = max(max(d / 30, s / 2), max(a, ta) / 60);
1586 //printf("*** %.6f from (\nd: %.6f\ns: %.6f\na: %.6f\nt: %.6f )\n",
1587 // next_mark_time, d/30.0, s/2.0, a/60.0, ta/60.0);
1588 if (tmp < 0) next_mark_time /= -tmp;
1592 if (presentation_mode) {
1593 // Advance position towards next_mark
1594 double p = tick / next_mark_time;
1595 double q = 1 - p;
1596 PresentationMark here = GetView();
1597 if (next_mark.angle < 0) {
1598 if (here.angle >= next_mark.angle + 360.0)
1599 here.angle -= 360.0;
1600 } else if (next_mark.angle >= 360.0) {
1601 if (here.angle <= next_mark.angle - 360.0)
1602 here.angle += 360.0;
1604 here.assign(q * here + p * next_mark);
1605 here.angle = q * here.angle + p * next_mark.angle;
1606 if (here.angle < 0) here.angle += 360.0;
1607 else if (here.angle >= 360.0) here.angle -= 360.0;
1608 here.tilt_angle = q * here.tilt_angle + p * next_mark.tilt_angle;
1609 here.scale = exp(q * log(here.scale) + p * log(next_mark.scale));
1610 SetView(here);
1611 this_mark_total += tick;
1612 next_mark_time -= tick;
1615 ForceRefresh();
1616 return;
1619 // When rotating...
1620 if (m_Rotating) {
1621 Double step = base_pan + (t - base_pan_time) * 1e-3 * m_RotationStep - m_PanAngle;
1622 TurnCave(step);
1625 if (m_SwitchingTo == PLAN) {
1626 // When switching to plan view...
1627 Double step = base_tilt - (t - base_tilt_time) * 1e-3 * 90.0 - m_TiltAngle;
1628 TiltCave(step);
1629 if (m_TiltAngle == -90.0) {
1630 m_SwitchingTo = 0;
1632 } else if (m_SwitchingTo == ELEVATION) {
1633 // When switching to elevation view...
1634 Double step;
1635 if (m_TiltAngle > 0.0) {
1636 step = base_tilt - (t - base_tilt_time) * 1e-3 * 90.0 - m_TiltAngle;
1637 } else {
1638 step = base_tilt + (t - base_tilt_time) * 1e-3 * 90.0 - m_TiltAngle;
1640 if (fabs(step) >= fabs(m_TiltAngle)) {
1641 m_SwitchingTo = 0;
1642 step = -m_TiltAngle;
1644 TiltCave(step);
1645 } else if (m_SwitchingTo) {
1646 // Rotate the shortest way around to the destination angle. If we're
1647 // 180 off, we favour turning anticlockwise, as auto-rotation does by
1648 // default.
1649 Double target = (m_SwitchingTo - NORTH) * 90;
1650 Double diff = target - m_PanAngle;
1651 diff = fmod(diff, 360);
1652 if (diff <= -180)
1653 diff += 360;
1654 else if (diff > 180)
1655 diff -= 360;
1656 if (m_RotationStep < 0 && diff == 180.0)
1657 diff = -180.0;
1658 Double step = base_pan - m_PanAngle;
1659 Double delta = (t - base_pan_time) * 1e-3 * fabs(m_RotationStep);
1660 if (diff > 0) {
1661 step += delta;
1662 } else {
1663 step -= delta;
1665 step = fmod(step, 360);
1666 if (step <= -180)
1667 step += 360;
1668 else if (step > 180)
1669 step -= 360;
1670 if (fabs(step) >= fabs(diff)) {
1671 m_SwitchingTo = 0;
1672 step = diff;
1674 TurnCave(step);
1677 ForceRefresh();
1680 // How much to allow around the box - this is because of the ring shape
1681 // at one end of the line.
1682 static const int HIGHLIGHTED_PT_SIZE = 2; // FIXME: tie in to blob and ring size
1683 #define MARGIN (HIGHLIGHTED_PT_SIZE * 2 + 1)
1684 void GfxCore::RefreshLine(const Point *a, const Point *b, const Point *c)
1686 #ifdef __WXMSW__
1687 (void)a;
1688 (void)b;
1689 (void)c;
1690 // FIXME: We get odd redraw artifacts if we just update the line, and
1691 // redrawing the whole scene doesn't actually seem to be measurably
1692 // slower. That may not be true with software rendering though...
1693 ForceRefresh();
1694 #else
1695 // Best of all might be to copy the window contents before we draw the
1696 // line, then replace each time we redraw.
1698 // Calculate the minimum rectangle which includes the old and new
1699 // measuring lines to minimise the redraw time
1700 int l = INT_MAX, r = INT_MIN, u = INT_MIN, d = INT_MAX;
1701 double X, Y, Z;
1702 if (a) {
1703 if (!Transform(*a, &X, &Y, &Z)) {
1704 printf("oops\n");
1705 } else {
1706 int x = int(X);
1707 int y = GetYSize() - 1 - int(Y);
1708 l = x;
1709 r = x;
1710 u = y;
1711 d = y;
1714 if (b) {
1715 if (!Transform(*b, &X, &Y, &Z)) {
1716 printf("oops\n");
1717 } else {
1718 int x = int(X);
1719 int y = GetYSize() - 1 - int(Y);
1720 l = min(l, x);
1721 r = max(r, x);
1722 u = max(u, y);
1723 d = min(d, y);
1726 if (c) {
1727 if (!Transform(*c, &X, &Y, &Z)) {
1728 printf("oops\n");
1729 } else {
1730 int x = int(X);
1731 int y = GetYSize() - 1 - int(Y);
1732 l = min(l, x);
1733 r = max(r, x);
1734 u = max(u, y);
1735 d = min(d, y);
1738 l -= MARGIN;
1739 r += MARGIN;
1740 u += MARGIN;
1741 d -= MARGIN;
1742 RefreshRect(wxRect(l, d, r - l, u - d), false);
1743 #endif
1746 void GfxCore::SetHereFromTree(const LabelInfo * p)
1748 SetHere(p);
1749 m_Parent->ShowInfo(m_here, m_there);
1752 void GfxCore::SetHere(const LabelInfo *p)
1754 if (p == m_here) return;
1755 bool line_active = MeasuringLineActive();
1756 const LabelInfo * old = m_here;
1757 m_here = p;
1758 if (line_active || MeasuringLineActive())
1759 RefreshLine(old, m_there, m_here);
1762 void GfxCore::SetThere(const LabelInfo * p)
1764 if (p == m_there) return;
1765 const LabelInfo * old = m_there;
1766 m_there = p;
1767 RefreshLine(m_here, old, m_there);
1770 void GfxCore::CreateHitTestGrid()
1772 if (!m_PointGrid) {
1773 // Initialise hit-test grid.
1774 m_PointGrid = new list<LabelInfo*>[HITTEST_SIZE * HITTEST_SIZE];
1775 } else {
1776 // Clear hit-test grid.
1777 for (int i = 0; i < HITTEST_SIZE * HITTEST_SIZE; i++) {
1778 m_PointGrid[i].clear();
1782 // Fill the grid.
1783 list<LabelInfo*>::const_iterator pos = m_Parent->GetLabels();
1784 list<LabelInfo*>::const_iterator end = m_Parent->GetLabelsEnd();
1785 while (pos != end) {
1786 LabelInfo* label = *pos++;
1788 if (!((m_Surface && label->IsSurface()) ||
1789 (m_Legs && label->IsUnderground()) ||
1790 (!label->IsSurface() && !label->IsUnderground()))) {
1791 // if this station isn't to be displayed, skip to the next
1792 // (last case is for stns with no legs attached)
1793 continue;
1796 // Calculate screen coordinates.
1797 double cx, cy, cz;
1798 Transform(*label, &cx, &cy, &cz);
1799 if (cx < 0 || cx >= GetXSize()) continue;
1800 if (cy < 0 || cy >= GetYSize()) continue;
1802 cy = GetYSize() - cy;
1804 // On-screen, so add to hit-test grid...
1805 int grid_x = int(cx * HITTEST_SIZE / (GetXSize() + 1));
1806 int grid_y = int(cy * HITTEST_SIZE / (GetYSize() + 1));
1808 m_PointGrid[grid_x + grid_y * HITTEST_SIZE].push_back(label);
1811 m_HitTestGridValid = true;
1815 // Methods for controlling the orientation of the survey
1818 void GfxCore::TurnCave(Double angle)
1820 // Turn the cave around its z-axis by a given angle.
1822 m_PanAngle += angle;
1823 // Wrap to range [0, 360):
1824 m_PanAngle = fmod(m_PanAngle, 360.0);
1825 if (m_PanAngle < 0.0) {
1826 m_PanAngle += 360.0;
1829 m_HitTestGridValid = false;
1830 if (m_here && m_here == &temp_here) SetHere();
1832 SetRotation(m_PanAngle, m_TiltAngle);
1835 void GfxCore::TurnCaveTo(Double angle)
1837 if (m_Rotating) {
1838 // If we're rotating, jump to the specified angle.
1839 TurnCave(angle - m_PanAngle);
1840 SetPanBase();
1841 return;
1844 int new_switching_to = ((int)angle) / 90 + NORTH;
1845 if (new_switching_to == m_SwitchingTo) {
1846 // A second order to switch takes us there right away
1847 TurnCave(angle - m_PanAngle);
1848 m_SwitchingTo = 0;
1849 ForceRefresh();
1850 } else {
1851 SetPanBase();
1852 m_SwitchingTo = new_switching_to;
1856 void GfxCore::TiltCave(Double tilt_angle)
1858 // Tilt the cave by a given angle.
1859 if (m_TiltAngle + tilt_angle > 90.0) {
1860 m_TiltAngle = 90.0;
1861 } else if (m_TiltAngle + tilt_angle < -90.0) {
1862 m_TiltAngle = -90.0;
1863 } else {
1864 m_TiltAngle += tilt_angle;
1867 m_HitTestGridValid = false;
1868 if (m_here && m_here == &temp_here) SetHere();
1870 SetRotation(m_PanAngle, m_TiltAngle);
1873 void GfxCore::TranslateCave(int dx, int dy)
1875 AddTranslationScreenCoordinates(dx, dy);
1876 m_HitTestGridValid = false;
1878 if (m_here && m_here == &temp_here) SetHere();
1880 ForceRefresh();
1883 void GfxCore::DragFinished()
1885 m_MouseOutsideCompass = m_MouseOutsideElev = false;
1886 ForceRefresh();
1889 void GfxCore::ClearCoords()
1891 m_Parent->ClearCoords();
1894 void GfxCore::SetCoords(wxPoint point)
1896 // We can't work out 2D coordinates from a perspective view, and it
1897 // doesn't really make sense to show coordinates while we're animating.
1898 if (GetPerspective() || Animating()) return;
1900 // Update the coordinate or altitude display, given the (x, y) position in
1901 // window coordinates. The relevant display is updated depending on
1902 // whether we're in plan or elevation view.
1904 double cx, cy, cz;
1906 SetDataTransform();
1907 ReverseTransform(point.x, GetYSize() - 1 - point.y, &cx, &cy, &cz);
1909 if (ShowingPlan()) {
1910 m_Parent->SetCoords(cx + m_Parent->GetOffset().GetX(),
1911 cy + m_Parent->GetOffset().GetY(),
1912 m_there);
1913 } else if (ShowingElevation()) {
1914 m_Parent->SetAltitude(cz + m_Parent->GetOffset().GetZ(),
1915 m_there);
1916 } else {
1917 m_Parent->ClearCoords();
1921 int GfxCore::GetCompassWidth() const
1923 static int result = 0;
1924 if (result == 0) {
1925 result = INDICATOR_BOX_SIZE;
1926 int width;
1927 const wxString & msg = wmsg(/*Facing*/203);
1928 GetTextExtent(msg, &width, NULL);
1929 if (width > result) result = width;
1931 return result;
1934 int GfxCore::GetClinoWidth() const
1936 static int result = 0;
1937 if (result == 0) {
1938 result = INDICATOR_BOX_SIZE;
1939 int width;
1940 const wxString & msg1 = wmsg(/*Plan*/432);
1941 GetTextExtent(msg1, &width, NULL);
1942 if (width > result) result = width;
1943 const wxString & msg2 = wmsg(/*Kiwi Plan*/433);
1944 GetTextExtent(msg2, &width, NULL);
1945 if (width > result) result = width;
1946 const wxString & msg3 = wmsg(/*Elevation*/118);
1947 GetTextExtent(msg3, &width, NULL);
1948 if (width > result) result = width;
1950 return result;
1953 int GfxCore::GetCompassXPosition() const
1955 // Return the x-coordinate of the centre of the compass in window
1956 // coordinates.
1957 return GetXSize() - INDICATOR_OFFSET_X - GetCompassWidth() / 2;
1960 int GfxCore::GetClinoXPosition() const
1962 // Return the x-coordinate of the centre of the compass in window
1963 // coordinates.
1964 return GetXSize() - GetClinoOffset() - GetClinoWidth() / 2;
1967 int GfxCore::GetIndicatorYPosition() const
1969 // Return the y-coordinate of the centre of the indicators in window
1970 // coordinates.
1971 return GetYSize() - INDICATOR_OFFSET_Y - INDICATOR_BOX_SIZE / 2;
1974 int GfxCore::GetIndicatorRadius() const
1976 // Return the radius of each indicator.
1977 return (INDICATOR_BOX_SIZE - INDICATOR_MARGIN * 2) / 2;
1980 bool GfxCore::PointWithinCompass(wxPoint point) const
1982 // Determine whether a point (in window coordinates) lies within the
1983 // compass.
1984 if (!ShowingCompass()) return false;
1986 glaCoord dx = point.x - GetCompassXPosition();
1987 glaCoord dy = point.y - GetIndicatorYPosition();
1988 glaCoord radius = GetIndicatorRadius();
1990 return (dx * dx + dy * dy <= radius * radius);
1993 bool GfxCore::PointWithinClino(wxPoint point) const
1995 // Determine whether a point (in window coordinates) lies within the clino.
1996 if (!ShowingClino()) return false;
1998 glaCoord dx = point.x - GetClinoXPosition();
1999 glaCoord dy = point.y - GetIndicatorYPosition();
2000 glaCoord radius = GetIndicatorRadius();
2002 return (dx * dx + dy * dy <= radius * radius);
2005 bool GfxCore::PointWithinScaleBar(wxPoint point) const
2007 // Determine whether a point (in window coordinates) lies within the scale
2008 // bar.
2009 if (!ShowingScaleBar()) return false;
2011 return (point.x >= SCALE_BAR_OFFSET_X &&
2012 point.x <= SCALE_BAR_OFFSET_X + m_ScaleBarWidth &&
2013 point.y <= GetYSize() - SCALE_BAR_OFFSET_Y - SCALE_BAR_HEIGHT &&
2014 point.y >= GetYSize() - SCALE_BAR_OFFSET_Y - SCALE_BAR_HEIGHT*2);
2017 bool GfxCore::PointWithinColourKey(wxPoint point) const
2019 // Determine whether a point (in window coordinates) lies within the key.
2020 point.x -= GetXSize() - KEY_OFFSET_X;
2021 point.y = KEY_OFFSET_Y - point.y;
2022 return (point.x >= key_lowerleft[m_ColourBy].x && point.x <= 0 &&
2023 point.y >= key_lowerleft[m_ColourBy].y && point.y <= 0);
2026 void GfxCore::SetCompassFromPoint(wxPoint point)
2028 // Given a point in window coordinates, set the heading of the survey. If
2029 // the point is outside the compass, it snaps to 45 degree intervals;
2030 // otherwise it operates as normal.
2032 wxCoord dx = point.x - GetCompassXPosition();
2033 wxCoord dy = point.y - GetIndicatorYPosition();
2034 wxCoord radius = GetIndicatorRadius();
2036 double angle = deg(atan2(double(dx), double(dy))) - 180.0;
2037 if (dx * dx + dy * dy <= radius * radius) {
2038 TurnCave(angle - m_PanAngle);
2039 m_MouseOutsideCompass = false;
2040 } else {
2041 TurnCave(int(angle / 45.0) * 45.0 - m_PanAngle);
2042 m_MouseOutsideCompass = true;
2045 ForceRefresh();
2048 void GfxCore::SetClinoFromPoint(wxPoint point)
2050 // Given a point in window coordinates, set the elevation of the survey.
2051 // If the point is outside the clino, it snaps to 90 degree intervals;
2052 // otherwise it operates as normal.
2054 glaCoord dx = point.x - GetClinoXPosition();
2055 glaCoord dy = point.y - GetIndicatorYPosition();
2056 glaCoord radius = GetIndicatorRadius();
2058 if (dx >= 0 && dx * dx + dy * dy <= radius * radius) {
2059 TiltCave(-deg(atan2(double(dy), double(dx))) - m_TiltAngle);
2060 m_MouseOutsideElev = false;
2061 } else if (dy >= INDICATOR_MARGIN) {
2062 TiltCave(-90.0 - m_TiltAngle);
2063 m_MouseOutsideElev = true;
2064 } else if (dy <= -INDICATOR_MARGIN) {
2065 TiltCave(90.0 - m_TiltAngle);
2066 m_MouseOutsideElev = true;
2067 } else {
2068 TiltCave(-m_TiltAngle);
2069 m_MouseOutsideElev = true;
2072 ForceRefresh();
2075 void GfxCore::SetScaleBarFromOffset(wxCoord dx)
2077 // Set the scale of the survey, given an offset as to how much the mouse has
2078 // been dragged over the scalebar since the last scale change.
2080 SetScale((m_ScaleBarWidth + dx) * m_Scale / m_ScaleBarWidth);
2081 ForceRefresh();
2084 void GfxCore::RedrawIndicators()
2086 // Redraw the compass and clino indicators.
2088 int total_width = GetCompassWidth() + INDICATOR_GAP + GetClinoWidth();
2089 RefreshRect(wxRect(GetXSize() - INDICATOR_OFFSET_X - total_width,
2090 GetYSize() - INDICATOR_OFFSET_Y - INDICATOR_BOX_SIZE,
2091 total_width,
2092 INDICATOR_BOX_SIZE), false);
2095 void GfxCore::StartRotation()
2097 // Start the survey rotating.
2099 if (m_SwitchingTo >= NORTH)
2100 m_SwitchingTo = 0;
2101 m_Rotating = true;
2102 SetPanBase();
2105 void GfxCore::ToggleRotation()
2107 // Toggle the survey rotation on/off.
2109 if (m_Rotating) {
2110 StopRotation();
2111 } else {
2112 StartRotation();
2116 void GfxCore::StopRotation()
2118 // Stop the survey rotating.
2120 m_Rotating = false;
2121 ForceRefresh();
2124 bool GfxCore::IsExtendedElevation() const
2126 return m_Parent->IsExtendedElevation();
2129 void GfxCore::ReverseRotation()
2131 // Reverse the direction of rotation.
2133 m_RotationStep = -m_RotationStep;
2134 if (m_Rotating)
2135 SetPanBase();
2138 void GfxCore::RotateSlower(bool accel)
2140 // Decrease the speed of rotation, optionally by an increased amount.
2141 if (fabs(m_RotationStep) == 1.0)
2142 return;
2144 m_RotationStep *= accel ? (1 / 1.44) : (1 / 1.2);
2146 if (fabs(m_RotationStep) < 1.0) {
2147 m_RotationStep = (m_RotationStep > 0 ? 1.0 : -1.0);
2149 if (m_Rotating)
2150 SetPanBase();
2153 void GfxCore::RotateFaster(bool accel)
2155 // Increase the speed of rotation, optionally by an increased amount.
2156 if (fabs(m_RotationStep) == 180.0)
2157 return;
2159 m_RotationStep *= accel ? 1.44 : 1.2;
2160 if (fabs(m_RotationStep) > 180.0) {
2161 m_RotationStep = (m_RotationStep > 0 ? 180.0 : -180.0);
2163 if (m_Rotating)
2164 SetPanBase();
2167 void GfxCore::SwitchToElevation()
2169 // Perform an animated switch to elevation view.
2171 if (m_SwitchingTo != ELEVATION) {
2172 SetTiltBase();
2173 m_SwitchingTo = ELEVATION;
2174 } else {
2175 // A second order to switch takes us there right away
2176 TiltCave(-m_TiltAngle);
2177 m_SwitchingTo = 0;
2178 ForceRefresh();
2182 void GfxCore::SwitchToPlan()
2184 // Perform an animated switch to plan view.
2186 if (m_SwitchingTo != PLAN) {
2187 SetTiltBase();
2188 m_SwitchingTo = PLAN;
2189 } else {
2190 // A second order to switch takes us there right away
2191 TiltCave(-90.0 - m_TiltAngle);
2192 m_SwitchingTo = 0;
2193 ForceRefresh();
2197 void GfxCore::SetViewTo(Double xmin, Double xmax, Double ymin, Double ymax, Double zmin, Double zmax)
2200 SetTranslation(-Vector3((xmin + xmax) / 2, (ymin + ymax) / 2, (zmin + zmax) / 2));
2201 Double scale = HUGE_VAL;
2202 const Vector3 ext = m_Parent->GetExtent();
2203 if (xmax > xmin) {
2204 Double s = ext.GetX() / (xmax - xmin);
2205 if (s < scale) scale = s;
2207 if (ymax > ymin) {
2208 Double s = ext.GetY() / (ymax - ymin);
2209 if (s < scale) scale = s;
2211 if (!ShowingPlan() && zmax > zmin) {
2212 Double s = ext.GetZ() / (zmax - zmin);
2213 if (s < scale) scale = s;
2215 if (scale != HUGE_VAL) SetScale(scale);
2216 ForceRefresh();
2219 bool GfxCore::CanRaiseViewpoint() const
2221 // Determine if the survey can be viewed from a higher angle of elevation.
2223 return GetPerspective() ? (m_TiltAngle < 90.0) : (m_TiltAngle > -90.0);
2226 bool GfxCore::CanLowerViewpoint() const
2228 // Determine if the survey can be viewed from a lower angle of elevation.
2230 return GetPerspective() ? (m_TiltAngle > -90.0) : (m_TiltAngle < 90.0);
2233 bool GfxCore::HasDepth() const
2235 return m_Parent->GetDepthExtent() == 0.0;
2238 bool GfxCore::HasErrorInformation() const
2240 return m_Parent->HasErrorInformation();
2243 bool GfxCore::HasDateInformation() const
2245 return m_Parent->GetDateMin() >= 0;
2248 bool GfxCore::ShowingPlan() const
2250 // Determine if the survey is in plan view.
2252 return (m_TiltAngle == -90.0);
2255 bool GfxCore::ShowingElevation() const
2257 // Determine if the survey is in elevation view.
2259 return (m_TiltAngle == 0.0);
2262 bool GfxCore::ShowingMeasuringLine() const
2264 // Determine if the measuring line is being shown. Only check if "there"
2265 // is valid, since that means the measuring line anchor is out.
2267 return m_there;
2270 void GfxCore::ToggleFlag(bool* flag, int update)
2272 *flag = !*flag;
2273 if (update == UPDATE_BLOBS) {
2274 UpdateBlobs();
2275 } else if (update == UPDATE_BLOBS_AND_CROSSES) {
2276 UpdateBlobs();
2277 InvalidateList(LIST_CROSSES);
2278 m_HitTestGridValid = false;
2280 ForceRefresh();
2283 int GfxCore::GetNumEntrances() const
2285 return m_Parent->GetNumEntrances();
2288 int GfxCore::GetNumFixedPts() const
2290 return m_Parent->GetNumFixedPts();
2293 int GfxCore::GetNumExportedPts() const
2295 return m_Parent->GetNumExportedPts();
2298 void GfxCore::ToggleTerrain()
2300 ToggleFlag(&m_Terrain);
2301 if (m_Terrain && !dem) {
2302 wxCommandEvent dummy;
2303 m_Parent->OnOpenTerrain(dummy);
2307 void GfxCore::ToggleFatFinger()
2309 if (sqrd_measure_threshold == sqrd(MEASURE_THRESHOLD)) {
2310 sqrd_measure_threshold = sqrd(5 * MEASURE_THRESHOLD);
2311 wxMessageBox(wxT("Fat finger enabled"), wxT("Aven Debug"), wxOK | wxICON_INFORMATION);
2312 } else {
2313 sqrd_measure_threshold = sqrd(MEASURE_THRESHOLD);
2314 wxMessageBox(wxT("Fat finger disabled"), wxT("Aven Debug"), wxOK | wxICON_INFORMATION);
2318 void GfxCore::ClearTreeSelection()
2320 m_Parent->ClearTreeSelection();
2323 void GfxCore::CentreOn(const Point &p)
2325 SetTranslation(-p);
2326 m_HitTestGridValid = false;
2328 ForceRefresh();
2331 void GfxCore::ForceRefresh()
2333 Refresh(false);
2336 void GfxCore::GenerateList(unsigned int l)
2338 assert(m_HaveData);
2340 switch (l) {
2341 case LIST_COMPASS:
2342 DrawCompass();
2343 break;
2344 case LIST_CLINO:
2345 DrawClino();
2346 break;
2347 case LIST_CLINO_BACK:
2348 DrawClinoBack();
2349 break;
2350 case LIST_SCALE_BAR:
2351 DrawScaleBar();
2352 break;
2353 case LIST_DEPTH_KEY:
2354 DrawDepthKey();
2355 break;
2356 case LIST_DATE_KEY:
2357 DrawDateKey();
2358 break;
2359 case LIST_ERROR_KEY:
2360 DrawErrorKey();
2361 break;
2362 case LIST_GRADIENT_KEY:
2363 DrawGradientKey();
2364 break;
2365 case LIST_LENGTH_KEY:
2366 DrawLengthKey();
2367 break;
2368 case LIST_UNDERGROUND_LEGS:
2369 GenerateDisplayList();
2370 break;
2371 case LIST_TUBES:
2372 GenerateDisplayListTubes();
2373 break;
2374 case LIST_SURFACE_LEGS:
2375 GenerateDisplayListSurface();
2376 break;
2377 case LIST_BLOBS:
2378 GenerateBlobsDisplayList();
2379 break;
2380 case LIST_CROSSES: {
2381 BeginCrosses();
2382 SetColour(col_LIGHT_GREY);
2383 list<LabelInfo*>::const_iterator pos = m_Parent->GetLabels();
2384 while (pos != m_Parent->GetLabelsEnd()) {
2385 const LabelInfo* label = *pos++;
2387 if ((m_Surface && label->IsSurface()) ||
2388 (m_Legs && label->IsUnderground()) ||
2389 (!label->IsSurface() && !label->IsUnderground())) {
2390 // Check if this station should be displayed
2391 // (last case is for stns with no legs attached)
2392 DrawCross(label->GetX(), label->GetY(), label->GetZ());
2395 EndCrosses();
2396 break;
2398 case LIST_GRID:
2399 DrawGrid();
2400 break;
2401 case LIST_SHADOW:
2402 GenerateDisplayListShadow();
2403 break;
2404 case LIST_TERRAIN:
2405 DrawTerrain();
2406 break;
2407 default:
2408 assert(false);
2409 break;
2413 void GfxCore::ToggleSmoothShading()
2415 GLACanvas::ToggleSmoothShading();
2416 InvalidateList(LIST_TUBES);
2417 ForceRefresh();
2420 void GfxCore::GenerateDisplayList()
2422 // Generate the display list for the underground legs.
2423 list<traverse>::const_iterator trav = m_Parent->traverses_begin();
2424 list<traverse>::const_iterator tend = m_Parent->traverses_end();
2426 if (m_Splays == SHOW_FADED) {
2427 SetAlpha(0.4);
2428 while (trav != tend) {
2429 if ((*trav).isSplay)
2430 (this->*AddPoly)(*trav);
2431 ++trav;
2433 SetAlpha(1.0);
2434 trav = m_Parent->traverses_begin();
2437 if (m_Dupes == SHOW_DASHED) {
2438 EnableDashedLines();
2439 while (trav != tend) {
2440 if ((*trav).isDupe)
2441 (this->*AddPoly)(*trav);
2442 ++trav;
2444 DisableDashedLines();
2445 trav = m_Parent->traverses_begin();
2448 while (trav != tend) {
2449 if (((*trav).isSplay && m_Splays == SHOW_NORMAL) ||
2450 ((*trav).isDupe && m_Dupes == SHOW_NORMAL) ||
2451 (!(*trav).isSplay && !(*trav).isDupe))
2452 (this->*AddPoly)(*trav);
2453 ++trav;
2457 void GfxCore::GenerateDisplayListTubes()
2459 // Generate the display list for the tubes.
2460 list<vector<XSect> >::iterator trav = m_Parent->tubes_begin();
2461 list<vector<XSect> >::iterator tend = m_Parent->tubes_end();
2462 while (trav != tend) {
2463 SkinPassage(*trav);
2464 ++trav;
2468 void GfxCore::GenerateDisplayListSurface()
2470 // Generate the display list for the surface legs.
2471 EnableDashedLines();
2472 list<traverse>::const_iterator trav = m_Parent->surface_traverses_begin();
2473 list<traverse>::const_iterator tend = m_Parent->surface_traverses_end();
2474 while (trav != tend) {
2475 if (m_ColourBy == COLOUR_BY_ERROR) {
2476 AddPolylineError(*trav);
2477 } else {
2478 AddPolyline(*trav);
2480 ++trav;
2482 DisableDashedLines();
2485 void GfxCore::GenerateDisplayListShadow()
2487 SetColour(col_BLACK);
2488 list<traverse>::const_iterator trav = m_Parent->traverses_begin();
2489 list<traverse>::const_iterator tend = m_Parent->traverses_end();
2490 while (trav != tend) {
2491 AddPolylineShadow(*trav);
2492 ++trav;
2496 void
2497 GfxCore::parse_hgt_filename(const wxString & lc_name)
2499 char * leaf = leaf_from_fnm(lc_name.utf8_str());
2500 const char * p = leaf;
2501 char * q;
2502 char dirn = *p++;
2503 o_y = strtoul(p, &q, 10);
2504 p = q;
2505 if (dirn == 's')
2506 o_y = -o_y;
2507 ++o_y;
2508 dirn = *p++;
2509 o_x = strtoul(p, &q, 10);
2510 if (dirn == 'w')
2511 o_x = -o_x;
2512 bigendian = true;
2513 nodata_value = -32768;
2514 osfree(leaf);
2517 size_t
2518 GfxCore::parse_hdr(wxInputStream & is, unsigned long & skipbytes)
2520 // ESRI docs say NBITS defaults to 8.
2521 unsigned long nbits = 8;
2522 // ESRI docs say NBANDS defaults to 1.
2523 unsigned long nbands = 1;
2524 unsigned long bandrowbytes = 0;
2525 unsigned long totalrowbytes = 0;
2526 // ESRI docs say ULXMAP defaults to 0.
2527 o_x = 0.0;
2528 // ESRI docs say ULYMAP defaults to NROWS - 1.
2529 o_y = HUGE_VAL;
2530 // ESRI docs say XDIM and YDIM default to 1.
2531 step_x = step_y = 1.0;
2532 while (!is.Eof()) {
2533 wxString line;
2534 int ch;
2535 while ((ch = is.GetC()) != wxEOF) {
2536 if (ch == '\n' || ch == '\r') break;
2537 line += wxChar(ch);
2539 #define CHECK(X, COND) \
2540 } else if (line.StartsWith(wxT(X " "))) { \
2541 size_t v = line.find_first_not_of(wxT(' '), sizeof(X)); \
2542 if (v == line.npos || !(COND)) { \
2543 err += wxT("Unexpected value for " X); \
2545 wxString err;
2546 if (false) {
2547 // I = little-endian; M = big-endian
2548 CHECK("BYTEORDER", (bigendian = (line[v] == 'M')) || line[v] == 'I')
2549 // ESRI docs say LAYOUT defaults to BIL if not specified.
2550 CHECK("LAYOUT", line.substr(v) == wxT("BIL"))
2551 CHECK("NROWS", line.substr(v).ToCULong(&dem_height))
2552 CHECK("NCOLS", line.substr(v).ToCULong(&dem_width))
2553 // ESRI docs say NBANDS defaults to 1 if not specified.
2554 CHECK("NBANDS", line.substr(v).ToCULong(&nbands) && nbands == 1)
2555 CHECK("NBITS", line.substr(v).ToCULong(&nbits) && nbits == 16)
2556 CHECK("BANDROWBYTES", line.substr(v).ToCULong(&bandrowbytes))
2557 CHECK("TOTALROWBYTES", line.substr(v).ToCULong(&totalrowbytes))
2558 // PIXELTYPE is a GDAL extension, so may not be present.
2559 CHECK("PIXELTYPE", line.substr(v) == wxT("SIGNEDINT"))
2560 CHECK("ULXMAP", line.substr(v).ToCDouble(&o_x))
2561 CHECK("ULYMAP", line.substr(v).ToCDouble(&o_y))
2562 CHECK("XDIM", line.substr(v).ToCDouble(&step_x))
2563 CHECK("YDIM", line.substr(v).ToCDouble(&step_y))
2564 CHECK("NODATA", line.substr(v).ToCLong(&nodata_value))
2565 CHECK("SKIPBYTES", line.substr(v).ToCULong(&skipbytes))
2567 if (!err.empty()) {
2568 wxMessageBox(err);
2571 if (o_y == HUGE_VAL) {
2572 o_y = dem_height - 1;
2574 if (bandrowbytes != 0) {
2575 if (nbits * dem_width != bandrowbytes * 8) {
2576 wxMessageBox("BANDROWBYTES setting indicates unused bits after each band - not currently supported");
2579 if (totalrowbytes != 0) {
2580 // This is the ESRI default for BIL, for BIP it would be
2581 // nbands * bandrowbytes.
2582 if (nbands * nbits * dem_width != totalrowbytes * 8) {
2583 wxMessageBox("TOTALROWBYTES setting indicates unused bits after "
2584 "each row - not currently supported");
2587 return ((nbits * dem_width + 7) / 8) * dem_height;
2590 bool
2591 GfxCore::read_bil(wxInputStream & is, size_t size, unsigned long skipbytes)
2593 bool know_size = true;
2594 if (!size) {
2595 // If the stream doesn't know its size, GetSize() returns 0.
2596 size = is.GetSize();
2597 if (!size) {
2598 size = DEFAULT_HGT_SIZE;
2599 know_size = false;
2602 dem = new unsigned short[size / 2];
2603 if (skipbytes) {
2604 if (is.SeekI(skipbytes, wxFromStart) == ::wxInvalidOffset) {
2605 while (skipbytes) {
2606 unsigned long to_read = skipbytes;
2607 if (size < to_read) to_read = size;
2608 is.Read(reinterpret_cast<char *>(dem), to_read);
2609 size_t c = is.LastRead();
2610 if (c == 0) {
2611 wxMessageBox(wxT("Failed to skip terrain data header"));
2612 break;
2614 skipbytes -= c;
2619 #if wxCHECK_VERSION(2,9,5)
2620 if (!is.ReadAll(dem, size)) {
2621 if (know_size) {
2622 // FIXME: On __WXMSW__ currently we fail to
2623 // read any data from files in zips.
2624 delete [] dem;
2625 dem = NULL;
2626 wxMessageBox(wxT("Failed to read terrain data"));
2627 return false;
2629 size = is.LastRead();
2631 #else
2632 char * p = reinterpret_cast<char *>(dem);
2633 while (size) {
2634 is.Read(p, size);
2635 size_t c = is.LastRead();
2636 if (c == 0) {
2637 if (!know_size) {
2638 size = DEFAULT_HGT_SIZE - size;
2639 if (size)
2640 break;
2642 delete [] dem;
2643 dem = NULL;
2644 wxMessageBox(wxT("Failed to read terrain data"));
2645 return false;
2647 p += c;
2648 size -= c;
2650 #endif
2652 if (dem_width == 0 && dem_height == 0) {
2653 dem_width = dem_height = sqrt(size / 2);
2654 if (dem_width * dem_height * 2 != size) {
2655 delete [] dem;
2656 dem = NULL;
2657 wxMessageBox(wxT("HGT format data doesn't form a square"));
2658 return false;
2660 step_x = step_y = 1.0 / dem_width;
2663 return true;
2666 bool GfxCore::LoadDEM(const wxString & file)
2668 if (m_Parent->m_cs_proj.empty()) {
2669 wxMessageBox(wxT("No coordinate system specified in survey data"));
2670 return false;
2673 delete [] dem;
2674 dem = NULL;
2676 size_t size = 0;
2677 // Default is to not skip any bytes.
2678 unsigned long skipbytes = 0;
2679 // For .hgt files, default to using filesize to determine.
2680 dem_width = dem_height = 0;
2681 // ESRI say "The default byte order is the same as that of the host machine
2682 // executing the software", but that's stupid so we default to
2683 // little-endian.
2684 bigendian = false;
2686 wxFileInputStream fs(file);
2687 if (!fs.IsOk()) {
2688 wxMessageBox(wxT("Failed to open DEM file"));
2689 return false;
2692 const wxString & lc_file = file.Lower();
2693 if (lc_file.EndsWith(wxT(".hgt"))) {
2694 parse_hgt_filename(lc_file);
2695 read_bil(fs, size, skipbytes);
2696 } else if (lc_file.EndsWith(wxT(".bil"))) {
2697 wxString hdr_file = file;
2698 hdr_file.replace(file.size() - 4, 4, wxT(".hdr"));
2699 wxFileInputStream hdr_is(hdr_file);
2700 if (!hdr_is.IsOk()) {
2701 wxMessageBox(wxT("Failed to open HDR file '") + hdr_file + wxT("'"));
2702 return false;
2704 size = parse_hdr(hdr_is, skipbytes);
2705 read_bil(fs, size, skipbytes);
2706 } else if (lc_file.EndsWith(wxT(".zip"))) {
2707 wxZipEntry * ze_data = NULL;
2708 wxZipInputStream zs(fs);
2709 wxZipEntry * ze;
2710 while ((ze = zs.GetNextEntry()) != NULL) {
2711 if (!ze->IsDir()) {
2712 const wxString & lc_name = ze->GetName().Lower();
2713 if (!ze_data && lc_name.EndsWith(wxT(".hgt"))) {
2714 // SRTM .hgt files are raw binary data, with the filename
2715 // encoding the coordinates.
2716 parse_hgt_filename(lc_name);
2717 read_bil(zs, size, skipbytes);
2718 delete ze;
2719 break;
2722 if (!ze_data && lc_name.EndsWith(wxT(".bil"))) {
2723 if (size) {
2724 read_bil(zs, size, skipbytes);
2725 break;
2727 ze_data = ze;
2728 continue;
2731 if (lc_name.EndsWith(wxT(".hdr"))) {
2732 size = parse_hdr(zs, skipbytes);
2733 if (ze_data) {
2734 if (!zs.OpenEntry(*ze_data)) {
2735 wxMessageBox(wxT("Couldn't read DEM data from .zip file"));
2736 break;
2738 read_bil(zs, size, skipbytes);
2740 } else if (lc_name.EndsWith(wxT(".prj"))) {
2741 //FIXME: check this matches the datum string we use
2742 //Projection GEOGRAPHIC
2743 //Datum WGS84
2744 //Zunits METERS
2745 //Units DD
2746 //Spheroid WGS84
2747 //Xshift 0.0000000000
2748 //Yshift 0.0000000000
2749 //Parameters
2752 delete ze;
2754 delete ze_data;
2757 if (!dem) {
2758 return false;
2761 InvalidateList(LIST_TERRAIN);
2762 ForceRefresh();
2763 return true;
2766 void GfxCore::DrawTerrainTriangle(const Vector3 & a, const Vector3 & b, const Vector3 & c)
2768 Vector3 n = (b - a) * (c - a);
2769 n.normalise();
2770 Double factor = dot(n, light) * .95 + .05;
2771 SetColour(col_WHITE, factor);
2772 PlaceVertex(a);
2773 PlaceVertex(b);
2774 PlaceVertex(c);
2775 ++n_tris;
2778 // Like wxBusyCursor, but you can cancel it early.
2779 class AvenBusyCursor {
2780 bool active;
2782 public:
2783 AvenBusyCursor() : active(true) {
2784 wxBeginBusyCursor();
2787 void stop() {
2788 if (active) {
2789 active = false;
2790 wxEndBusyCursor();
2794 ~AvenBusyCursor() {
2795 stop();
2799 void GfxCore::DrawTerrain()
2801 if (!dem) return;
2803 AvenBusyCursor hourglass;
2805 // Draw terrain to twice the extent, or at least 1km.
2806 double r_sqrd = sqrd(max(m_Parent->GetExtent().magnitude(), 1000.0));
2807 #define WGS84_DATUM_STRING "+proj=longlat +ellps=WGS84 +datum=WGS84"
2808 static projPJ pj_in = pj_init_plus(WGS84_DATUM_STRING);
2809 if (!pj_in) {
2810 ToggleTerrain();
2811 delete [] dem;
2812 dem = NULL;
2813 hourglass.stop();
2814 error(/*Failed to initialise input coordinate system “%s”*/287, WGS84_DATUM_STRING);
2815 return;
2817 static projPJ pj_out = pj_init_plus(m_Parent->m_cs_proj.c_str());
2818 if (!pj_out) {
2819 ToggleTerrain();
2820 delete [] dem;
2821 dem = NULL;
2822 hourglass.stop();
2823 error(/*Failed to initialise output coordinate system “%s”*/288, (const char *)m_Parent->m_cs_proj.c_str());
2824 return;
2826 n_tris = 0;
2827 SetAlpha(0.3);
2828 BeginTriangles();
2829 const Vector3 & off = m_Parent->GetOffset();
2830 vector<Vector3> prevcol(dem_height + 1);
2831 for (size_t x = 0; x < dem_width; ++x) {
2832 double X_ = (o_x + x * step_x) * DEG_TO_RAD;
2833 Vector3 prev;
2834 for (size_t y = 0; y < dem_height; ++y) {
2835 unsigned short elev = dem[x + y * dem_width];
2836 #ifdef WORDS_BIGENDIAN
2837 const bool MACHINE_BIGENDIAN = true;
2838 #else
2839 const bool MACHINE_BIGENDIAN = false;
2840 #endif
2841 if (bigendian != MACHINE_BIGENDIAN) {
2842 #if defined __GNUC__ && (__GNUC__ * 100 + __GNUC_MINOR__ >= 408)
2843 elev = __builtin_bswap16(elev);
2844 #else
2845 elev = (elev >> 8) | (elev << 8);
2846 #endif
2848 double Z = (short)elev;
2849 Vector3 pt;
2850 if (Z == nodata_value) {
2851 pt = Vector3(DBL_MAX, DBL_MAX, DBL_MAX);
2852 } else {
2853 double X = X_;
2854 double Y = (o_y - y * step_y) * DEG_TO_RAD;
2855 pj_transform(pj_in, pj_out, 1, 1, &X, &Y, &Z);
2856 pt = Vector3(X, Y, Z) - off;
2857 double dist_2 = sqrd(pt.GetX()) + sqrd(pt.GetY());
2858 if (dist_2 > r_sqrd) {
2859 pt = Vector3(DBL_MAX, DBL_MAX, DBL_MAX);
2862 if (x > 0 && y > 0) {
2863 const Vector3 & a = prevcol[y - 1];
2864 const Vector3 & b = prevcol[y];
2865 // If all points are valid, split the quadrilateral into
2866 // triangles along the shorter 3D diagonal, which typically
2867 // looks better:
2869 // ----->
2870 // prev---a x prev---a
2871 // | |P /| |\ S|
2872 // y | | / | or | \ |
2873 // V | / | | \ |
2874 // |/ Q| |R \|
2875 // b----pt b----pt
2877 // FORWARD BACKWARD
2878 enum { NONE = 0, P = 1, Q = 2, R = 4, S = 8, ALL = P|Q|R|S };
2879 int valid =
2880 ((prev.GetZ() != DBL_MAX)) |
2881 ((a.GetZ() != DBL_MAX) << 1) |
2882 ((b.GetZ() != DBL_MAX) << 2) |
2883 ((pt.GetZ() != DBL_MAX) << 3);
2884 static const int tris_map[16] = {
2885 NONE, // nothing valid
2886 NONE, // prev
2887 NONE, // a
2888 NONE, // a, prev
2889 NONE, // b
2890 NONE, // b, prev
2891 NONE, // b, a
2892 P, // b, a, prev
2893 NONE, // pt
2894 NONE, // pt, prev
2895 NONE, // pt, a
2896 S, // pt, a, prev
2897 NONE, // pt, b
2898 R, // pt, b, prev
2899 Q, // pt, b, a
2900 ALL, // pt, b, a, prev
2902 int tris = tris_map[valid];
2903 if (tris == ALL) {
2904 // All points valid.
2905 if ((a - b).magnitude() < (prev - pt).magnitude()) {
2906 tris = P | Q;
2907 } else {
2908 tris = R | S;
2911 if (tris & P)
2912 DrawTerrainTriangle(a, prev, b);
2913 if (tris & Q)
2914 DrawTerrainTriangle(a, b, pt);
2915 if (tris & R)
2916 DrawTerrainTriangle(pt, prev, b);
2917 if (tris & S)
2918 DrawTerrainTriangle(a, prev, pt);
2920 prev = prevcol[y];
2921 prevcol[y].assign(pt);
2924 EndTriangles();
2925 SetAlpha(1.0);
2926 if (n_tris == 0) {
2927 ToggleTerrain();
2928 delete [] dem;
2929 dem = NULL;
2930 hourglass.stop();
2931 /* TRANSLATORS: Aven shows a circle of terrain covering the area
2932 * of the survey plus a bit, but the terrain data file didn't
2933 * contain any data inside that circle.
2935 error(/*No terrain data near area of survey*/161);
2939 // Plot blobs.
2940 void GfxCore::GenerateBlobsDisplayList()
2942 if (!(m_Entrances || m_FixedPts || m_ExportedPts ||
2943 m_Parent->GetNumHighlightedPts()))
2944 return;
2946 // Plot blobs.
2947 gla_colour prev_col = col_BLACK; // not a colour used for blobs
2948 list<LabelInfo*>::const_iterator pos = m_Parent->GetLabels();
2949 BeginBlobs();
2950 while (pos != m_Parent->GetLabelsEnd()) {
2951 const LabelInfo* label = *pos++;
2953 // When more than one flag is set on a point:
2954 // search results take priority over entrance highlighting
2955 // which takes priority over fixed point
2956 // highlighting, which in turn takes priority over exported
2957 // point highlighting.
2959 if (!((m_Surface && label->IsSurface()) ||
2960 (m_Legs && label->IsUnderground()) ||
2961 (!label->IsSurface() && !label->IsUnderground()))) {
2962 // if this station isn't to be displayed, skip to the next
2963 // (last case is for stns with no legs attached)
2964 continue;
2967 gla_colour col;
2969 if (label->IsHighLighted()) {
2970 col = col_YELLOW;
2971 } else if (m_Entrances && label->IsEntrance()) {
2972 col = col_GREEN;
2973 } else if (m_FixedPts && label->IsFixedPt()) {
2974 col = col_RED;
2975 } else if (m_ExportedPts && label->IsExportedPt()) {
2976 col = col_TURQUOISE;
2977 } else {
2978 continue;
2981 // Stations are sorted by blob type, so colour changes are infrequent.
2982 if (col != prev_col) {
2983 SetColour(col);
2984 prev_col = col;
2986 DrawBlob(label->GetX(), label->GetY(), label->GetZ());
2988 EndBlobs();
2991 void GfxCore::DrawIndicators()
2993 // Draw colour key.
2994 if (m_ColourKey) {
2995 drawing_list key_list = LIST_LIMIT_;
2996 switch (m_ColourBy) {
2997 case COLOUR_BY_DEPTH:
2998 key_list = LIST_DEPTH_KEY; break;
2999 case COLOUR_BY_DATE:
3000 key_list = LIST_DATE_KEY; break;
3001 case COLOUR_BY_ERROR:
3002 key_list = LIST_ERROR_KEY; break;
3003 case COLOUR_BY_GRADIENT:
3004 key_list = LIST_GRADIENT_KEY; break;
3005 case COLOUR_BY_LENGTH:
3006 key_list = LIST_LENGTH_KEY; break;
3008 if (key_list != LIST_LIMIT_) {
3009 DrawList2D(key_list, GetXSize() - KEY_OFFSET_X,
3010 GetYSize() - KEY_OFFSET_Y, 0);
3014 // Draw compass or elevation/heading indicators.
3015 if (m_Compass || m_Clino) {
3016 if (!m_Parent->IsExtendedElevation()) Draw2dIndicators();
3019 // Draw scalebar.
3020 if (m_Scalebar) {
3021 DrawList2D(LIST_SCALE_BAR, 0, 0, 0);
3025 void GfxCore::PlaceVertexWithColour(const Vector3 & v,
3026 glaTexCoord tex_x, glaTexCoord tex_y,
3027 Double factor)
3029 SetColour(col_WHITE, factor);
3030 PlaceVertex(v, tex_x, tex_y);
3033 void GfxCore::SetDepthColour(Double z, Double factor) {
3034 // Set the drawing colour based on the altitude.
3035 Double z_ext = m_Parent->GetDepthExtent();
3037 z -= m_Parent->GetDepthMin();
3038 // points arising from tubes may be slightly outside the limits...
3039 if (z < 0) z = 0;
3040 if (z > z_ext) z = z_ext;
3042 if (z == 0) {
3043 SetColour(GetPen(0), factor);
3044 return;
3047 assert(z_ext > 0.0);
3048 Double how_far = z / z_ext;
3049 assert(how_far >= 0.0);
3050 assert(how_far <= 1.0);
3052 int band = int(floor(how_far * (GetNumColourBands() - 1)));
3053 GLAPen pen1 = GetPen(band);
3054 if (band < GetNumColourBands() - 1) {
3055 const GLAPen& pen2 = GetPen(band + 1);
3057 Double interval = z_ext / (GetNumColourBands() - 1);
3058 Double into_band = z / interval - band;
3060 // printf("%g z_offset=%g interval=%g band=%d\n", into_band,
3061 // z_offset, interval, band);
3062 // FIXME: why do we need to clamp here? Is it because the walls can
3063 // extend further up/down than the centre-line?
3064 if (into_band < 0.0) into_band = 0.0;
3065 if (into_band > 1.0) into_band = 1.0;
3066 assert(into_band >= 0.0);
3067 assert(into_band <= 1.0);
3069 pen1.Interpolate(pen2, into_band);
3071 SetColour(pen1, factor);
3074 void GfxCore::PlaceVertexWithDepthColour(const Vector3 &v, Double factor)
3076 SetDepthColour(v.GetZ(), factor);
3077 PlaceVertex(v);
3080 void GfxCore::PlaceVertexWithDepthColour(const Vector3 &v,
3081 glaTexCoord tex_x, glaTexCoord tex_y,
3082 Double factor)
3084 SetDepthColour(v.GetZ(), factor);
3085 PlaceVertex(v, tex_x, tex_y);
3088 void GfxCore::SplitLineAcrossBands(int band, int band2,
3089 const Vector3 &p, const Vector3 &q,
3090 Double factor)
3092 const int step = (band < band2) ? 1 : -1;
3093 for (int i = band; i != band2; i += step) {
3094 const Double z = GetDepthBoundaryBetweenBands(i, i + step);
3096 // Find the intersection point of the line p -> q
3097 // with the plane parallel to the xy-plane with z-axis intersection z.
3098 assert(q.GetZ() - p.GetZ() != 0.0);
3100 const Double t = (z - p.GetZ()) / (q.GetZ() - p.GetZ());
3101 // assert(0.0 <= t && t <= 1.0); FIXME: rounding problems!
3103 const Double x = p.GetX() + t * (q.GetX() - p.GetX());
3104 const Double y = p.GetY() + t * (q.GetY() - p.GetY());
3106 PlaceVertexWithDepthColour(Vector3(x, y, z), factor);
3110 int GfxCore::GetDepthColour(Double z) const
3112 // Return the (0-based) depth colour band index for a z-coordinate.
3113 Double z_ext = m_Parent->GetDepthExtent();
3114 z -= m_Parent->GetDepthMin();
3115 // We seem to get rounding differences causing z to sometimes be slightly
3116 // less than GetDepthMin() here, and it can certainly be true for passage
3117 // tubes, so just clamp the value to 0.
3118 if (z <= 0) return 0;
3119 // We seem to get rounding differences causing z to sometimes exceed z_ext
3120 // by a small amount here (see: https://trac.survex.com/ticket/26) and it
3121 // can certainly be true for passage tubes, so just clamp the value.
3122 if (z >= z_ext) return GetNumColourBands() - 1;
3123 return int(z / z_ext * (GetNumColourBands() - 1));
3126 Double GfxCore::GetDepthBoundaryBetweenBands(int a, int b) const
3128 // Return the z-coordinate of the depth colour boundary between
3129 // two adjacent depth colour bands (specified by 0-based indices).
3131 assert((a == b - 1) || (a == b + 1));
3132 if (GetNumColourBands() == 1) return 0;
3134 int band = (a > b) ? a : b; // boundary N lies on the bottom of band N.
3135 Double z_ext = m_Parent->GetDepthExtent();
3136 return (z_ext * band / (GetNumColourBands() - 1)) + m_Parent->GetDepthMin();
3139 void GfxCore::AddPolyline(const traverse & centreline)
3141 BeginPolyline();
3142 SetColour(col_WHITE);
3143 vector<PointInfo>::const_iterator i = centreline.begin();
3144 PlaceVertex(*i);
3145 ++i;
3146 while (i != centreline.end()) {
3147 PlaceVertex(*i);
3148 ++i;
3150 EndPolyline();
3153 void GfxCore::AddPolylineShadow(const traverse & centreline)
3155 BeginPolyline();
3156 const double z = -0.5 * m_Parent->GetZExtent();
3157 vector<PointInfo>::const_iterator i = centreline.begin();
3158 PlaceVertex(i->GetX(), i->GetY(), z);
3159 ++i;
3160 while (i != centreline.end()) {
3161 PlaceVertex(i->GetX(), i->GetY(), z);
3162 ++i;
3164 EndPolyline();
3167 void GfxCore::AddPolylineDepth(const traverse & centreline)
3169 BeginPolyline();
3170 vector<PointInfo>::const_iterator i, prev_i;
3171 i = centreline.begin();
3172 int band0 = GetDepthColour(i->GetZ());
3173 PlaceVertexWithDepthColour(*i);
3174 prev_i = i;
3175 ++i;
3176 while (i != centreline.end()) {
3177 int band = GetDepthColour(i->GetZ());
3178 if (band != band0) {
3179 SplitLineAcrossBands(band0, band, *prev_i, *i);
3180 band0 = band;
3182 PlaceVertexWithDepthColour(*i);
3183 prev_i = i;
3184 ++i;
3186 EndPolyline();
3189 void GfxCore::AddQuadrilateral(const Vector3 &a, const Vector3 &b,
3190 const Vector3 &c, const Vector3 &d)
3192 Vector3 normal = (a - c) * (d - b);
3193 normal.normalise();
3194 Double factor = dot(normal, light) * .3 + .7;
3195 glaTexCoord w(ceil(((b - a).magnitude() + (d - c).magnitude()) * .5));
3196 glaTexCoord h(ceil(((b - c).magnitude() + (d - a).magnitude()) * .5));
3197 // FIXME: should plot triangles instead to avoid rendering glitches.
3198 BeginQuadrilaterals();
3199 PlaceVertexWithColour(a, 0, 0, factor);
3200 PlaceVertexWithColour(b, w, 0, factor);
3201 PlaceVertexWithColour(c, w, h, factor);
3202 PlaceVertexWithColour(d, 0, h, factor);
3203 EndQuadrilaterals();
3206 void GfxCore::AddQuadrilateralDepth(const Vector3 &a, const Vector3 &b,
3207 const Vector3 &c, const Vector3 &d)
3209 Vector3 normal = (a - c) * (d - b);
3210 normal.normalise();
3211 Double factor = dot(normal, light) * .3 + .7;
3212 int a_band, b_band, c_band, d_band;
3213 a_band = GetDepthColour(a.GetZ());
3214 a_band = min(max(a_band, 0), GetNumColourBands());
3215 b_band = GetDepthColour(b.GetZ());
3216 b_band = min(max(b_band, 0), GetNumColourBands());
3217 c_band = GetDepthColour(c.GetZ());
3218 c_band = min(max(c_band, 0), GetNumColourBands());
3219 d_band = GetDepthColour(d.GetZ());
3220 d_band = min(max(d_band, 0), GetNumColourBands());
3221 // All this splitting is incorrect - we need to make a separate polygon
3222 // for each depth band...
3223 glaTexCoord w(ceil(((b - a).magnitude() + (d - c).magnitude()) * .5));
3224 glaTexCoord h(ceil(((b - c).magnitude() + (d - a).magnitude()) * .5));
3225 BeginPolygon();
3226 //// PlaceNormal(normal);
3227 PlaceVertexWithDepthColour(a, 0, 0, factor);
3228 if (a_band != b_band) {
3229 SplitLineAcrossBands(a_band, b_band, a, b, factor);
3231 PlaceVertexWithDepthColour(b, w, 0, factor);
3232 if (b_band != c_band) {
3233 SplitLineAcrossBands(b_band, c_band, b, c, factor);
3235 PlaceVertexWithDepthColour(c, w, h, factor);
3236 if (c_band != d_band) {
3237 SplitLineAcrossBands(c_band, d_band, c, d, factor);
3239 PlaceVertexWithDepthColour(d, 0, h, factor);
3240 if (d_band != a_band) {
3241 SplitLineAcrossBands(d_band, a_band, d, a, factor);
3243 EndPolygon();
3246 void GfxCore::SetColourFromDate(int date, Double factor)
3248 // Set the drawing colour based on a date.
3250 if (date == -1) {
3251 // Undated.
3252 SetColour(col_WHITE, factor);
3253 return;
3256 int date_offset = date - m_Parent->GetDateMin();
3257 if (date_offset == 0) {
3258 // Earliest date - handle as a special case for the single date case.
3259 SetColour(GetPen(0), factor);
3260 return;
3263 int date_ext = m_Parent->GetDateExtent();
3264 Double how_far = (Double)date_offset / date_ext;
3265 assert(how_far >= 0.0);
3266 assert(how_far <= 1.0);
3267 SetColourFrom01(how_far, factor);
3270 void GfxCore::AddPolylineDate(const traverse & centreline)
3272 BeginPolyline();
3273 vector<PointInfo>::const_iterator i, prev_i;
3274 i = centreline.begin();
3275 int date = i->GetDate();
3276 SetColourFromDate(date, 1.0);
3277 PlaceVertex(*i);
3278 prev_i = i;
3279 while (++i != centreline.end()) {
3280 int newdate = i->GetDate();
3281 if (newdate != date) {
3282 EndPolyline();
3283 BeginPolyline();
3284 date = newdate;
3285 SetColourFromDate(date, 1.0);
3286 PlaceVertex(*prev_i);
3288 PlaceVertex(*i);
3289 prev_i = i;
3291 EndPolyline();
3294 static int static_date_hack; // FIXME
3296 void GfxCore::AddQuadrilateralDate(const Vector3 &a, const Vector3 &b,
3297 const Vector3 &c, const Vector3 &d)
3299 Vector3 normal = (a - c) * (d - b);
3300 normal.normalise();
3301 Double factor = dot(normal, light) * .3 + .7;
3302 int w = int(ceil(((b - a).magnitude() + (d - c).magnitude()) / 2));
3303 int h = int(ceil(((b - c).magnitude() + (d - a).magnitude()) / 2));
3304 // FIXME: should plot triangles instead to avoid rendering glitches.
3305 BeginQuadrilaterals();
3306 //// PlaceNormal(normal);
3307 SetColourFromDate(static_date_hack, factor);
3308 PlaceVertex(a, 0, 0);
3309 PlaceVertex(b, w, 0);
3310 PlaceVertex(c, w, h);
3311 PlaceVertex(d, 0, h);
3312 EndQuadrilaterals();
3315 static double static_E_hack; // FIXME
3317 void GfxCore::SetColourFromError(double E, Double factor)
3319 // Set the drawing colour based on an error value.
3321 if (E < 0) {
3322 SetColour(col_WHITE, factor);
3323 return;
3326 Double how_far = E / MAX_ERROR;
3327 assert(how_far >= 0.0);
3328 if (how_far > 1.0) how_far = 1.0;
3329 SetColourFrom01(how_far, factor);
3332 void GfxCore::AddQuadrilateralError(const Vector3 &a, const Vector3 &b,
3333 const Vector3 &c, const Vector3 &d)
3335 Vector3 normal = (a - c) * (d - b);
3336 normal.normalise();
3337 Double factor = dot(normal, light) * .3 + .7;
3338 int w = int(ceil(((b - a).magnitude() + (d - c).magnitude()) / 2));
3339 int h = int(ceil(((b - c).magnitude() + (d - a).magnitude()) / 2));
3340 // FIXME: should plot triangles instead to avoid rendering glitches.
3341 BeginQuadrilaterals();
3342 //// PlaceNormal(normal);
3343 SetColourFromError(static_E_hack, factor);
3344 PlaceVertex(a, 0, 0);
3345 PlaceVertex(b, w, 0);
3346 PlaceVertex(c, w, h);
3347 PlaceVertex(d, 0, h);
3348 EndQuadrilaterals();
3351 void GfxCore::AddPolylineError(const traverse & centreline)
3353 BeginPolyline();
3354 SetColourFromError(centreline.E, 1.0);
3355 vector<PointInfo>::const_iterator i;
3356 for(i = centreline.begin(); i != centreline.end(); ++i) {
3357 PlaceVertex(*i);
3359 EndPolyline();
3362 // gradient is in *radians*.
3363 void GfxCore::SetColourFromGradient(double gradient, Double factor)
3365 // Set the drawing colour based on the gradient of the leg.
3367 const Double GRADIENT_MAX = M_PI_2;
3368 gradient = fabs(gradient);
3369 Double how_far = gradient / GRADIENT_MAX;
3370 SetColourFrom01(how_far, factor);
3373 void GfxCore::AddPolylineGradient(const traverse & centreline)
3375 vector<PointInfo>::const_iterator i, prev_i;
3376 i = centreline.begin();
3377 prev_i = i;
3378 while (++i != centreline.end()) {
3379 BeginPolyline();
3380 SetColourFromGradient((*i - *prev_i).gradient(), 1.0);
3381 PlaceVertex(*prev_i);
3382 PlaceVertex(*i);
3383 prev_i = i;
3384 EndPolyline();
3388 static double static_gradient_hack; // FIXME
3390 void GfxCore::AddQuadrilateralGradient(const Vector3 &a, const Vector3 &b,
3391 const Vector3 &c, const Vector3 &d)
3393 Vector3 normal = (a - c) * (d - b);
3394 normal.normalise();
3395 Double factor = dot(normal, light) * .3 + .7;
3396 int w = int(ceil(((b - a).magnitude() + (d - c).magnitude()) / 2));
3397 int h = int(ceil(((b - c).magnitude() + (d - a).magnitude()) / 2));
3398 // FIXME: should plot triangles instead to avoid rendering glitches.
3399 BeginQuadrilaterals();
3400 //// PlaceNormal(normal);
3401 SetColourFromGradient(static_gradient_hack, factor);
3402 PlaceVertex(a, 0, 0);
3403 PlaceVertex(b, w, 0);
3404 PlaceVertex(c, w, h);
3405 PlaceVertex(d, 0, h);
3406 EndQuadrilaterals();
3409 void GfxCore::SetColourFromLength(double length, Double factor)
3411 // Set the drawing colour based on log(length_of_leg).
3413 Double log_len = log10(length);
3414 Double how_far = log_len / LOG_LEN_MAX;
3415 how_far = max(how_far, 0.0);
3416 how_far = min(how_far, 1.0);
3417 SetColourFrom01(how_far, factor);
3420 void GfxCore::SetColourFrom01(double how_far, Double factor)
3422 double b;
3423 double into_band = modf(how_far * (GetNumColourBands() - 1), &b);
3424 int band(b);
3425 GLAPen pen1 = GetPen(band);
3426 // With 24bit colour, interpolating by less than this can have no effect.
3427 if (into_band >= 1.0 / 512.0) {
3428 const GLAPen& pen2 = GetPen(band + 1);
3429 pen1.Interpolate(pen2, into_band);
3431 SetColour(pen1, factor);
3434 void GfxCore::AddPolylineLength(const traverse & centreline)
3436 vector<PointInfo>::const_iterator i, prev_i;
3437 i = centreline.begin();
3438 prev_i = i;
3439 while (++i != centreline.end()) {
3440 BeginPolyline();
3441 SetColourFromLength((*i - *prev_i).magnitude(), 1.0);
3442 PlaceVertex(*prev_i);
3443 PlaceVertex(*i);
3444 prev_i = i;
3445 EndPolyline();
3449 static double static_length_hack; // FIXME
3451 void GfxCore::AddQuadrilateralLength(const Vector3 &a, const Vector3 &b,
3452 const Vector3 &c, const Vector3 &d)
3454 Vector3 normal = (a - c) * (d - b);
3455 normal.normalise();
3456 Double factor = dot(normal, light) * .3 + .7;
3457 int w = int(ceil(((b - a).magnitude() + (d - c).magnitude()) / 2));
3458 int h = int(ceil(((b - c).magnitude() + (d - a).magnitude()) / 2));
3459 // FIXME: should plot triangles instead to avoid rendering glitches.
3460 BeginQuadrilaterals();
3461 //// PlaceNormal(normal);
3462 SetColourFromLength(static_length_hack, factor);
3463 PlaceVertex(a, 0, 0);
3464 PlaceVertex(b, w, 0);
3465 PlaceVertex(c, w, h);
3466 PlaceVertex(d, 0, h);
3467 EndQuadrilaterals();
3470 void
3471 GfxCore::SkinPassage(vector<XSect> & centreline, bool draw)
3473 assert(centreline.size() > 1);
3474 Vector3 U[4];
3475 XSect prev_pt_v;
3476 Vector3 last_right(1.0, 0.0, 0.0);
3478 // FIXME: it's not simple to set the colour of a tube based on error...
3479 // static_E_hack = something...
3480 vector<XSect>::iterator i = centreline.begin();
3481 vector<XSect>::size_type segment = 0;
3482 while (i != centreline.end()) {
3483 // get the coordinates of this vertex
3484 XSect & pt_v = *i++;
3486 bool cover_end = false;
3488 Vector3 right, up;
3490 const Vector3 up_v(0.0, 0.0, 1.0);
3492 if (segment == 0) {
3493 assert(i != centreline.end());
3494 // first segment
3496 // get the coordinates of the next vertex
3497 const XSect & next_pt_v = *i;
3499 // calculate vector from this pt to the next one
3500 Vector3 leg_v = next_pt_v - pt_v;
3502 // obtain a vector in the LRUD plane
3503 right = leg_v * up_v;
3504 if (right.magnitude() == 0) {
3505 right = last_right;
3506 // Obtain a second vector in the LRUD plane,
3507 // perpendicular to the first.
3508 //up = right * leg_v;
3509 up = up_v;
3510 } else {
3511 last_right = right;
3512 up = up_v;
3515 cover_end = true;
3516 static_date_hack = next_pt_v.GetDate();
3517 } else if (segment + 1 == centreline.size()) {
3518 // last segment
3520 // Calculate vector from the previous pt to this one.
3521 Vector3 leg_v = pt_v - prev_pt_v;
3523 // Obtain a horizontal vector in the LRUD plane.
3524 right = leg_v * up_v;
3525 if (right.magnitude() == 0) {
3526 right = Vector3(last_right.GetX(), last_right.GetY(), 0.0);
3527 // Obtain a second vector in the LRUD plane,
3528 // perpendicular to the first.
3529 //up = right * leg_v;
3530 up = up_v;
3531 } else {
3532 last_right = right;
3533 up = up_v;
3536 cover_end = true;
3537 static_date_hack = pt_v.GetDate();
3538 } else {
3539 assert(i != centreline.end());
3540 // Intermediate segment.
3542 // Get the coordinates of the next vertex.
3543 const XSect & next_pt_v = *i;
3545 // Calculate vectors from this vertex to the
3546 // next vertex, and from the previous vertex to
3547 // this one.
3548 Vector3 leg1_v = pt_v - prev_pt_v;
3549 Vector3 leg2_v = next_pt_v - pt_v;
3551 // Obtain horizontal vectors perpendicular to
3552 // both legs, then normalise and average to get
3553 // a horizontal bisector.
3554 Vector3 r1 = leg1_v * up_v;
3555 Vector3 r2 = leg2_v * up_v;
3556 r1.normalise();
3557 r2.normalise();
3558 right = r1 + r2;
3559 if (right.magnitude() == 0) {
3560 // This is the "mid-pitch" case...
3561 right = last_right;
3563 if (r1.magnitude() == 0) {
3564 up = up_v;
3566 // Rotate pitch section to minimise the
3567 // "tortional stress" - FIXME: use
3568 // triangles instead of rectangles?
3569 int shift = 0;
3570 Double maxdotp = 0;
3572 // Scale to unit vectors in the LRUD plane.
3573 right.normalise();
3574 up.normalise();
3575 Vector3 vec = up - right;
3576 for (int orient = 0; orient <= 3; ++orient) {
3577 Vector3 tmp = U[orient] - prev_pt_v;
3578 tmp.normalise();
3579 Double dotp = dot(vec, tmp);
3580 if (dotp > maxdotp) {
3581 maxdotp = dotp;
3582 shift = orient;
3585 if (shift) {
3586 if (shift != 2) {
3587 Vector3 temp(U[0]);
3588 U[0] = U[shift];
3589 U[shift] = U[2];
3590 U[2] = U[shift ^ 2];
3591 U[shift ^ 2] = temp;
3592 } else {
3593 swap(U[0], U[2]);
3594 swap(U[1], U[3]);
3597 #if 0
3598 // Check that the above code actually permuted
3599 // the vertices correctly.
3600 shift = 0;
3601 maxdotp = 0;
3602 for (int j = 0; j <= 3; ++j) {
3603 Vector3 tmp = U[j] - prev_pt_v;
3604 tmp.normalise();
3605 Double dotp = dot(vec, tmp);
3606 if (dotp > maxdotp) {
3607 maxdotp = dotp + 1e-6; // Add small tolerance to stop 45 degree offset cases being flagged...
3608 shift = j;
3611 if (shift) {
3612 printf("New shift = %d!\n", shift);
3613 shift = 0;
3614 maxdotp = 0;
3615 for (int j = 0; j <= 3; ++j) {
3616 Vector3 tmp = U[j] - prev_pt_v;
3617 tmp.normalise();
3618 Double dotp = dot(vec, tmp);
3619 printf(" %d : %.8f\n", j, dotp);
3622 #endif
3623 } else {
3624 up = up_v;
3626 last_right = right;
3627 static_date_hack = pt_v.GetDate();
3630 // Scale to unit vectors in the LRUD plane.
3631 right.normalise();
3632 up.normalise();
3634 Double l = fabs(pt_v.GetL());
3635 Double r = fabs(pt_v.GetR());
3636 Double u = fabs(pt_v.GetU());
3637 Double d = fabs(pt_v.GetD());
3639 // Produce coordinates of the corners of the LRUD "plane".
3640 Vector3 v[4];
3641 v[0] = pt_v - right * l + up * u;
3642 v[1] = pt_v + right * r + up * u;
3643 v[2] = pt_v + right * r - up * d;
3644 v[3] = pt_v - right * l - up * d;
3646 if (draw) {
3647 const Vector3 & delta = pt_v - prev_pt_v;
3648 static_length_hack = delta.magnitude();
3649 static_gradient_hack = delta.gradient();
3650 if (segment > 0) {
3651 (this->*AddQuad)(v[0], v[1], U[1], U[0]);
3652 (this->*AddQuad)(v[2], v[3], U[3], U[2]);
3653 (this->*AddQuad)(v[1], v[2], U[2], U[1]);
3654 (this->*AddQuad)(v[3], v[0], U[0], U[3]);
3657 if (cover_end) {
3658 if (segment == 0) {
3659 (this->*AddQuad)(v[0], v[1], v[2], v[3]);
3660 } else {
3661 (this->*AddQuad)(v[3], v[2], v[1], v[0]);
3666 prev_pt_v = pt_v;
3667 U[0] = v[0];
3668 U[1] = v[1];
3669 U[2] = v[2];
3670 U[3] = v[3];
3672 pt_v.set_right_bearing(deg(atan2(right.GetY(), right.GetX())));
3674 ++segment;
3678 void GfxCore::FullScreenMode()
3680 m_Parent->ViewFullScreen();
3683 bool GfxCore::IsFullScreen() const
3685 return m_Parent->IsFullScreen();
3688 bool GfxCore::FullScreenModeShowingMenus() const
3690 return m_Parent->FullScreenModeShowingMenus();
3693 void GfxCore::FullScreenModeShowMenus(bool show)
3695 m_Parent->FullScreenModeShowMenus(show);
3698 void
3699 GfxCore::MoveViewer(double forward, double up, double right)
3701 double cT = cos(rad(m_TiltAngle));
3702 double sT = sin(rad(m_TiltAngle));
3703 double cP = cos(rad(m_PanAngle));
3704 double sP = sin(rad(m_PanAngle));
3705 Vector3 v_forward(cT * sP, cT * cP, sT);
3706 Vector3 v_up(sT * sP, sT * cP, -cT);
3707 Vector3 v_right(-cP, sP, 0);
3708 assert(fabs(dot(v_forward, v_up)) < 1e-6);
3709 assert(fabs(dot(v_forward, v_right)) < 1e-6);
3710 assert(fabs(dot(v_right, v_up)) < 1e-6);
3711 Vector3 move = v_forward * forward + v_up * up + v_right * right;
3712 AddTranslation(-move);
3713 // Show current position.
3714 m_Parent->SetCoords(m_Parent->GetOffset() - GetTranslation());
3715 ForceRefresh();
3718 PresentationMark GfxCore::GetView() const
3720 return PresentationMark(GetTranslation() + m_Parent->GetOffset(),
3721 m_PanAngle, -m_TiltAngle, m_Scale);
3724 void GfxCore::SetView(const PresentationMark & p)
3726 m_SwitchingTo = 0;
3727 SetTranslation(p - m_Parent->GetOffset());
3728 m_PanAngle = p.angle;
3729 m_TiltAngle = -p.tilt_angle; // FIXME: nasty reversed sense (and above)
3730 SetRotation(m_PanAngle, m_TiltAngle);
3731 SetScale(p.scale);
3732 ForceRefresh();
3735 void GfxCore::PlayPres(double speed, bool change_speed) {
3736 if (!change_speed || presentation_mode == 0) {
3737 if (speed == 0.0) {
3738 presentation_mode = 0;
3739 return;
3741 presentation_mode = PLAYING;
3742 next_mark = m_Parent->GetPresMark(MARK_FIRST);
3743 SetView(next_mark);
3744 next_mark_time = 0; // There already!
3745 this_mark_total = 0;
3746 pres_reverse = (speed < 0);
3749 if (change_speed) pres_speed = speed;
3751 if (speed != 0.0) {
3752 bool new_pres_reverse = (speed < 0);
3753 if (new_pres_reverse != pres_reverse) {
3754 pres_reverse = new_pres_reverse;
3755 if (pres_reverse) {
3756 next_mark = m_Parent->GetPresMark(MARK_PREV);
3757 } else {
3758 next_mark = m_Parent->GetPresMark(MARK_NEXT);
3760 swap(this_mark_total, next_mark_time);
3765 void GfxCore::SetColourBy(int colour_by) {
3766 m_ColourBy = colour_by;
3767 switch (colour_by) {
3768 case COLOUR_BY_DEPTH:
3769 AddQuad = &GfxCore::AddQuadrilateralDepth;
3770 AddPoly = &GfxCore::AddPolylineDepth;
3771 break;
3772 case COLOUR_BY_DATE:
3773 AddQuad = &GfxCore::AddQuadrilateralDate;
3774 AddPoly = &GfxCore::AddPolylineDate;
3775 break;
3776 case COLOUR_BY_ERROR:
3777 AddQuad = &GfxCore::AddQuadrilateralError;
3778 AddPoly = &GfxCore::AddPolylineError;
3779 break;
3780 case COLOUR_BY_GRADIENT:
3781 AddQuad = &GfxCore::AddQuadrilateralGradient;
3782 AddPoly = &GfxCore::AddPolylineGradient;
3783 break;
3784 case COLOUR_BY_LENGTH:
3785 AddQuad = &GfxCore::AddQuadrilateralLength;
3786 AddPoly = &GfxCore::AddPolylineLength;
3787 break;
3788 default: // case COLOUR_BY_NONE:
3789 AddQuad = &GfxCore::AddQuadrilateral;
3790 AddPoly = &GfxCore::AddPolyline;
3791 break;
3794 InvalidateList(LIST_UNDERGROUND_LEGS);
3795 InvalidateList(LIST_SURFACE_LEGS);
3796 InvalidateList(LIST_TUBES);
3798 ForceRefresh();
3801 bool GfxCore::ExportMovie(const wxString & fnm)
3803 FILE* fh = wxFopen(fnm.fn_str(), wxT("wb"));
3804 if (fh == NULL) {
3805 wxGetApp().ReportError(wxString::Format(wmsg(/*Failed to open output file “%s”*/47), fnm.c_str()));
3806 return false;
3809 wxString ext;
3810 wxFileName::SplitPath(fnm, NULL, NULL, NULL, &ext, wxPATH_NATIVE);
3812 int width;
3813 int height;
3814 GetSize(&width, &height);
3815 // Round up to next multiple of 2 (required by ffmpeg).
3816 width += (width & 1);
3817 height += (height & 1);
3819 movie = new MovieMaker();
3821 // movie takes ownership of fh.
3822 if (!movie->Open(fh, ext.utf8_str(), width, height)) {
3823 wxGetApp().ReportError(wxString(movie->get_error_string(), wxConvUTF8));
3824 delete movie;
3825 movie = NULL;
3826 return false;
3829 PlayPres(1);
3830 return true;
3833 void
3834 GfxCore::OnPrint(const wxString &filename, const wxString &title,
3835 const wxString &datestamp, time_t datestamp_numeric,
3836 const wxString &cs_proj,
3837 bool close_after_print)
3839 svxPrintDlg * p;
3840 p = new svxPrintDlg(m_Parent, filename, title, cs_proj,
3841 datestamp, datestamp_numeric,
3842 m_PanAngle, m_TiltAngle,
3843 m_Names, m_Crosses, m_Legs, m_Surface, m_Splays,
3844 m_Tubes, m_Entrances, m_FixedPts, m_ExportedPts,
3845 true, close_after_print);
3846 p->Show(true);
3849 void
3850 GfxCore::OnExport(const wxString &filename, const wxString &title,
3851 const wxString &datestamp, time_t datestamp_numeric,
3852 const wxString &cs_proj)
3854 // Fill in "right_bearing" for each cross-section.
3855 list<vector<XSect> >::iterator trav = m_Parent->tubes_begin();
3856 list<vector<XSect> >::iterator tend = m_Parent->tubes_end();
3857 while (trav != tend) {
3858 SkinPassage(*trav, false);
3859 ++trav;
3862 svxPrintDlg * p;
3863 p = new svxPrintDlg(m_Parent, filename, title, cs_proj,
3864 datestamp, datestamp_numeric,
3865 m_PanAngle, m_TiltAngle,
3866 m_Names, m_Crosses, m_Legs, m_Surface, m_Splays,
3867 m_Tubes, m_Entrances, m_FixedPts, m_ExportedPts,
3868 false);
3869 p->Show(true);
3872 static wxCursor
3873 make_cursor(const unsigned char * bits, const unsigned char * mask,
3874 int hotx, int hoty)
3876 #if defined __WXMSW__ || defined __WXMAC__
3877 # ifdef __WXMAC__
3878 // The default Mac cursor is black with a white edge, so
3879 // invert our custom cursors to match.
3880 char b[128];
3881 for (int i = 0; i < 128; ++i)
3882 b[i] = bits[i] ^ 0xff;
3883 # else
3884 const char * b = reinterpret_cast<const char *>(bits);
3885 # endif
3886 wxBitmap cursor_bitmap(b, 32, 32);
3887 wxBitmap mask_bitmap(reinterpret_cast<const char *>(mask), 32, 32);
3888 cursor_bitmap.SetMask(new wxMask(mask_bitmap, *wxWHITE));
3889 wxImage cursor_image = cursor_bitmap.ConvertToImage();
3890 cursor_image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_X, hotx);
3891 cursor_image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y, hoty);
3892 return wxCursor(cursor_image);
3893 #else
3894 return wxCursor((const char *)bits, 32, 32, hotx, hoty,
3895 (const char *)mask, wxBLACK, wxWHITE);
3896 #endif
3899 const
3900 #include "hand.xbm"
3901 const
3902 #include "handmask.xbm"
3904 const
3905 #include "brotate.xbm"
3906 const
3907 #include "brotatemask.xbm"
3909 const
3910 #include "vrotate.xbm"
3911 const
3912 #include "vrotatemask.xbm"
3914 const
3915 #include "rotate.xbm"
3916 const
3917 #include "rotatemask.xbm"
3919 const
3920 #include "rotatezoom.xbm"
3921 const
3922 #include "rotatezoommask.xbm"
3924 void
3925 GfxCore::UpdateCursor(GfxCore::cursor new_cursor)
3927 // Check if we're already showing that cursor.
3928 if (current_cursor == new_cursor) return;
3930 current_cursor = new_cursor;
3931 switch (current_cursor) {
3932 case GfxCore::CURSOR_DEFAULT:
3933 GLACanvas::SetCursor(wxNullCursor);
3934 break;
3935 case GfxCore::CURSOR_POINTING_HAND:
3936 GLACanvas::SetCursor(wxCursor(wxCURSOR_HAND));
3937 break;
3938 case GfxCore::CURSOR_DRAGGING_HAND:
3939 GLACanvas::SetCursor(make_cursor(hand_bits, handmask_bits, 12, 18));
3940 break;
3941 case GfxCore::CURSOR_HORIZONTAL_RESIZE:
3942 GLACanvas::SetCursor(wxCursor(wxCURSOR_SIZEWE));
3943 break;
3944 case GfxCore::CURSOR_ROTATE_HORIZONTALLY:
3945 GLACanvas::SetCursor(make_cursor(rotate_bits, rotatemask_bits, 15, 15));
3946 break;
3947 case GfxCore::CURSOR_ROTATE_VERTICALLY:
3948 GLACanvas::SetCursor(make_cursor(vrotate_bits, vrotatemask_bits, 15, 15));
3949 break;
3950 case GfxCore::CURSOR_ROTATE_EITHER_WAY:
3951 GLACanvas::SetCursor(make_cursor(brotate_bits, brotatemask_bits, 15, 15));
3952 break;
3953 case GfxCore::CURSOR_ZOOM:
3954 GLACanvas::SetCursor(wxCursor(wxCURSOR_MAGNIFIER));
3955 break;
3956 case GfxCore::CURSOR_ZOOM_ROTATE:
3957 GLACanvas::SetCursor(make_cursor(rotatezoom_bits, rotatezoommask_bits, 15, 15));
3958 break;
3962 bool GfxCore::MeasuringLineActive() const
3964 if (Animating()) return false;
3965 return HereIsReal() || m_there;
3968 bool GfxCore::HandleRClick(wxPoint point)
3970 if (PointWithinCompass(point)) {
3971 // Pop up menu.
3972 wxMenu menu;
3973 /* TRANSLATORS: View *looking* North */
3974 menu.Append(menu_ORIENT_MOVE_NORTH, wmsg(/*View &North*/240));
3975 /* TRANSLATORS: View *looking* East */
3976 menu.Append(menu_ORIENT_MOVE_EAST, wmsg(/*View &East*/241));
3977 /* TRANSLATORS: View *looking* South */
3978 menu.Append(menu_ORIENT_MOVE_SOUTH, wmsg(/*View &South*/242));
3979 /* TRANSLATORS: View *looking* West */
3980 menu.Append(menu_ORIENT_MOVE_WEST, wmsg(/*View &West*/243));
3981 menu.AppendSeparator();
3982 /* TRANSLATORS: Menu item which turns off the "north arrow" in aven. */
3983 menu.AppendCheckItem(menu_IND_COMPASS, wmsg(/*&Hide Compass*/387));
3984 /* TRANSLATORS: tickable menu item in View menu.
3986 * Degrees are the angular measurement where there are 360 in a full
3987 * circle. */
3988 menu.AppendCheckItem(menu_CTL_DEGREES, wmsg(/*&Degrees*/343));
3989 menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
3990 PopupMenu(&menu);
3991 return true;
3994 if (PointWithinClino(point)) {
3995 // Pop up menu.
3996 wxMenu menu;
3997 menu.Append(menu_ORIENT_PLAN, wmsg(/*&Plan View*/248));
3998 menu.Append(menu_ORIENT_ELEVATION, wmsg(/*Ele&vation*/249));
3999 menu.AppendSeparator();
4000 /* TRANSLATORS: Menu item which turns off the tilt indicator in aven. */
4001 menu.AppendCheckItem(menu_IND_CLINO, wmsg(/*&Hide Clino*/384));
4002 /* TRANSLATORS: tickable menu item in View menu.
4004 * Degrees are the angular measurement where there are 360 in a full
4005 * circle. */
4006 menu.AppendCheckItem(menu_CTL_DEGREES, wmsg(/*&Degrees*/343));
4007 /* TRANSLATORS: tickable menu item in View menu.
4009 * Show the tilt of the survey as a percentage gradient (100% = 45
4010 * degrees = 50 grad). */
4011 menu.AppendCheckItem(menu_CTL_PERCENT, wmsg(/*&Percent*/430));
4012 menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
4013 PopupMenu(&menu);
4014 return true;
4017 if (PointWithinScaleBar(point)) {
4018 // Pop up menu.
4019 wxMenu menu;
4020 /* TRANSLATORS: Menu item which turns off the scale bar in aven. */
4021 menu.AppendCheckItem(menu_IND_SCALE_BAR, wmsg(/*&Hide scale bar*/385));
4022 /* TRANSLATORS: tickable menu item in View menu.
4024 * "Metric" here means metres, km, etc (rather than feet, miles, etc)
4026 menu.AppendCheckItem(menu_CTL_METRIC, wmsg(/*&Metric*/342));
4027 menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
4028 PopupMenu(&menu);
4029 return true;
4032 if (PointWithinColourKey(point)) {
4033 // Pop up menu.
4034 wxMenu menu;
4035 menu.AppendCheckItem(menu_COLOUR_BY_DEPTH, wmsg(/*Colour by &Depth*/292));
4036 menu.AppendCheckItem(menu_COLOUR_BY_DATE, wmsg(/*Colour by D&ate*/293));
4037 menu.AppendCheckItem(menu_COLOUR_BY_ERROR, wmsg(/*Colour by &Error*/289));
4038 menu.AppendCheckItem(menu_COLOUR_BY_GRADIENT, wmsg(/*Colour by &Gradient*/85));
4039 menu.AppendCheckItem(menu_COLOUR_BY_LENGTH, wmsg(/*Colour by &Length*/82));
4040 menu.AppendSeparator();
4041 /* TRANSLATORS: Menu item which turns off the colour key.
4042 * The "Colour Key" is the thing in aven showing which colour
4043 * corresponds to which depth, date, survey closure error, etc. */
4044 menu.AppendCheckItem(menu_IND_COLOUR_KEY, wmsg(/*&Hide colour key*/386));
4045 if (m_ColourBy == COLOUR_BY_DEPTH || m_ColourBy == COLOUR_BY_LENGTH)
4046 menu.AppendCheckItem(menu_CTL_METRIC, wmsg(/*&Metric*/342));
4047 else if (m_ColourBy == COLOUR_BY_GRADIENT)
4048 menu.AppendCheckItem(menu_CTL_DEGREES, wmsg(/*&Degrees*/343));
4049 menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
4050 PopupMenu(&menu);
4051 return true;
4054 return false;
4057 void GfxCore::SetZoomBox(wxPoint p1, wxPoint p2, bool centred, bool aspect)
4059 if (centred) {
4060 p1.x = p2.x + (p1.x - p2.x) * 2;
4061 p1.y = p2.y + (p1.y - p2.y) * 2;
4063 if (aspect) {
4064 #if 0 // FIXME: This needs more work.
4065 int sx = GetXSize();
4066 int sy = GetYSize();
4067 int dx = p1.x - p2.x;
4068 int dy = p1.y - p2.y;
4069 int dy_new = dx * sy / sx;
4070 if (abs(dy_new) >= abs(dy)) {
4071 p1.y += (dy_new - dy) / 2;
4072 p2.y -= (dy_new - dy) / 2;
4073 } else {
4074 int dx_new = dy * sx / sy;
4075 p1.x += (dx_new - dx) / 2;
4076 p2.x -= (dx_new - dx) / 2;
4078 #endif
4080 zoombox.set(p1, p2);
4081 ForceRefresh();
4084 void GfxCore::ZoomBoxGo()
4086 if (!zoombox.active()) return;
4088 int width = GetXSize();
4089 int height = GetYSize();
4091 TranslateCave(-0.5 * (zoombox.x1 + zoombox.x2 - width),
4092 -0.5 * (zoombox.y1 + zoombox.y2 - height));
4093 int box_w = abs(zoombox.x1 - zoombox.x2);
4094 int box_h = abs(zoombox.y1 - zoombox.y2);
4096 double factor = min(double(width) / box_w, double(height) / box_h);
4098 zoombox.unset();
4100 SetScale(GetScale() * factor);