source: git/src/gfxcore.cc @ 53496ab3

RELEASE/1.2debug-cidebug-ci-sanitisersstereowalls-data
Last change on this file since 53496ab3 was f10cf8f, checked in by Olly Betts <olly@…>, 10 years ago

src/: Pass the numeric datestamp to ExportFilter::header(), and use
it to set a timestamp in exported GPX files.

  • Property mode set to 100644
File size: 80.3 KB
Line 
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 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.
14//
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.
19//
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
23//
24
25#ifdef HAVE_CONFIG_H
26#include <config.h>
27#endif
28
29#include <assert.h>
30#include <float.h>
31
32#include "aven.h"
33#include "date.h"
34#include "gfxcore.h"
35#include "mainfrm.h"
36#include "message.h"
37#include "useful.h"
38#include "printwx.h"
39#include "guicontrol.h"
40#include "moviemaker.h"
41
42#include <wx/confbase.h>
43#include <wx/image.h>
44
45#define MAX(a, b) ((a) > (b) ? (a) : (b))
46#define MAX3(a, b, c) ((a) > (b) ? MAX(a, c) : MAX(b, c))
47
48// Values for m_SwitchingTo
49#define PLAN 1
50#define ELEVATION 2
51#define NORTH 3
52#define EAST 4
53#define SOUTH 5
54#define WEST 6
55
56// Any error value higher than this is clamped to this.
57#define MAX_ERROR 12.0
58
59// How many bins per letter height to use when working out non-overlapping
60// labels.
61const unsigned int QUANTISE_FACTOR = 2;
62
63#include "avenpal.h"
64
65static const int COMPASS_OFFSET_X = 60;
66static const int COMPASS_OFFSET_Y = 80;
67static const int INDICATOR_BOX_SIZE = 60;
68static const int INDICATOR_GAP = 2;
69static const int INDICATOR_MARGIN = 5;
70static const int INDICATOR_OFFSET_X = 15;
71static const int INDICATOR_OFFSET_Y = 15;
72static const int INDICATOR_RADIUS = INDICATOR_BOX_SIZE / 2 - INDICATOR_MARGIN;
73static const int CLINO_OFFSET_X = 6 + INDICATOR_OFFSET_X +
74                                  INDICATOR_BOX_SIZE + INDICATOR_GAP;
75static const int KEY_OFFSET_X = 10;
76static const int KEY_OFFSET_Y = 10;
77static const int KEY_EXTRA_LEFT_MARGIN = 2;
78static const int KEY_BLOCK_WIDTH = 20;
79static const int KEY_BLOCK_HEIGHT = 16;
80static const int TICK_LENGTH = 4;
81static const int SCALE_BAR_OFFSET_X = 15;
82static const int SCALE_BAR_OFFSET_Y = 12;
83static const int SCALE_BAR_HEIGHT = 12;
84
85static const gla_colour TEXT_COLOUR = col_GREEN;
86static const gla_colour HERE_COLOUR = col_WHITE;
87static const gla_colour NAME_COLOUR = col_GREEN;
88
89#define HITTEST_SIZE 20
90
91// vector for lighting angle
92static const Vector3 light(.577, .577, .577);
93
94BEGIN_EVENT_TABLE(GfxCore, GLACanvas)
95    EVT_PAINT(GfxCore::OnPaint)
96    EVT_LEFT_DOWN(GfxCore::OnLButtonDown)
97    EVT_LEFT_UP(GfxCore::OnLButtonUp)
98    EVT_MIDDLE_DOWN(GfxCore::OnMButtonDown)
99    EVT_MIDDLE_UP(GfxCore::OnMButtonUp)
100    EVT_RIGHT_DOWN(GfxCore::OnRButtonDown)
101    EVT_RIGHT_UP(GfxCore::OnRButtonUp)
102    EVT_MOUSEWHEEL(GfxCore::OnMouseWheel)
103    EVT_MOTION(GfxCore::OnMouseMove)
104    EVT_LEAVE_WINDOW(GfxCore::OnLeaveWindow)
105    EVT_SIZE(GfxCore::OnSize)
106    EVT_IDLE(GfxCore::OnIdle)
107    EVT_CHAR(GfxCore::OnKeyPress)
108END_EVENT_TABLE()
109
110GfxCore::GfxCore(MainFrm* parent, wxWindow* parent_win, GUIControl* control) :
111    GLACanvas(parent_win, 100),
112    m_Scale(0.0),
113    m_ScaleBarWidth(0),
114    m_Control(control),
115    m_LabelGrid(NULL),
116    m_Parent(parent),
117    m_DoneFirstShow(false),
118    m_TiltAngle(0.0),
119    m_PanAngle(0.0),
120    m_Rotating(false),
121    m_RotationStep(0.0),
122    m_SwitchingTo(0),
123    m_Crosses(false),
124    m_Legs(true),
125    m_Splays(SPLAYS_SHOW_FADED),
126    m_Names(false),
127    m_Scalebar(true),
128    m_ColourKey(true),
129    m_OverlappingNames(false),
130    m_Compass(true),
131    m_Clino(true),
132    m_Tubes(false),
133    m_ColourBy(COLOUR_BY_DEPTH),
134    m_HaveData(false),
135    m_MouseOutsideCompass(false),
136    m_MouseOutsideElev(false),
137    m_Surface(false),
138    m_Entrances(false),
139    m_FixedPts(false),
140    m_ExportedPts(false),
141    m_Grid(false),
142    m_BoundingBox(false),
143    m_Degrees(false),
144    m_Metric(false),
145    m_HitTestGridValid(false),
146    m_here_is_temporary(false),
147    presentation_mode(0),
148    pres_reverse(false),
149    pres_speed(0.0),
150    movie(NULL),
151    current_cursor(GfxCore::CURSOR_DEFAULT)
152{
153    AddQuad = &GfxCore::AddQuadrilateralDepth;
154    AddPoly = &GfxCore::AddPolylineDepth;
155    wxConfigBase::Get()->Read(wxT("metric"), &m_Metric, true);
156    wxConfigBase::Get()->Read(wxT("degrees"), &m_Degrees, true);
157    m_here.Invalidate();
158    m_there.Invalidate();
159
160    // Initialise grid for hit testing.
161    m_PointGrid = new list<LabelInfo*>[HITTEST_SIZE * HITTEST_SIZE];
162
163    m_Pens = new GLAPen[NUM_COLOUR_BANDS + 1];
164    for (int pen = 0; pen < NUM_COLOUR_BANDS + 1; ++pen) {
165        m_Pens[pen].SetColour(REDS[pen] / 255.0,
166                              GREENS[pen] / 255.0,
167                              BLUES[pen] / 255.0);
168    }
169}
170
171GfxCore::~GfxCore()
172{
173    TryToFreeArrays();
174
175    delete[] m_Pens;
176    delete[] m_PointGrid;
177}
178
179void GfxCore::TryToFreeArrays()
180{
181    // Free up any memory allocated for arrays.
182    delete[] m_LabelGrid;
183    m_LabelGrid = NULL;
184}
185
186//
187//  Initialisation methods
188//
189
190void GfxCore::Initialise(bool same_file)
191{
192    // Initialise the view from the parent holding the survey data.
193
194    TryToFreeArrays();
195
196    m_DoneFirstShow = false;
197
198    m_HitTestGridValid = false;
199    m_here_is_temporary = true;
200    m_here.Invalidate();
201    m_there.Invalidate();
202
203    m_MouseOutsideCompass = m_MouseOutsideElev = false;
204
205    if (!same_file) {
206        // Apply default parameters unless reloading the same file.
207        DefaultParameters();
208
209        // Set the initial scale.
210        SetScale(1.0);
211    }
212
213    m_HaveData = true;
214
215    // Clear any cached OpenGL lists which depend on the data.
216    InvalidateList(LIST_SCALE_BAR);
217    InvalidateList(LIST_DEPTH_KEY);
218    InvalidateList(LIST_DATE_KEY);
219    InvalidateList(LIST_ERROR_KEY);
220    InvalidateList(LIST_UNDERGROUND_LEGS);
221    InvalidateList(LIST_TUBES);
222    InvalidateList(LIST_SURFACE_LEGS);
223    InvalidateList(LIST_BLOBS);
224    InvalidateList(LIST_CROSSES);
225    InvalidateList(LIST_GRID);
226    InvalidateList(LIST_SHADOW);
227
228    ForceRefresh();
229}
230
231void GfxCore::FirstShow()
232{
233    GLACanvas::FirstShow();
234
235    const unsigned int quantise(GetFontSize() / QUANTISE_FACTOR);
236    list<LabelInfo*>::iterator pos = m_Parent->GetLabelsNC();
237    while (pos != m_Parent->GetLabelsNCEnd()) {
238        LabelInfo* label = *pos++;
239        // Calculate and set the label width for use when plotting
240        // none-overlapping labels.
241        int ext_x;
242        GLACanvas::GetTextExtent(label->GetText(), &ext_x, NULL);
243        label->set_width(unsigned(ext_x) / quantise + 1);
244    }
245
246    // Set diameter of the viewing volume.
247    SetVolumeDiameter(sqrt(sqrd(m_Parent->GetXExtent()) +
248                           sqrd(m_Parent->GetYExtent()) +
249                           sqrd(m_Parent->GetZExtent())));
250
251    m_DoneFirstShow = true;
252}
253
254//
255//  Recalculating methods
256//
257
258void GfxCore::SetScale(Double scale)
259{
260    if (scale < 0.05) {
261        scale = 0.05;
262    } else if (scale > GetVolumeDiameter()) {
263        scale = GetVolumeDiameter();
264    }
265
266    m_Scale = scale;
267    m_HitTestGridValid = false;
268    if (m_here_is_temporary) SetHere();
269
270    GLACanvas::SetScale(scale);
271}
272
273bool GfxCore::HasUndergroundLegs() const
274{
275    return m_Parent->HasUndergroundLegs();
276}
277
278bool GfxCore::HasSplays() const
279{
280    return m_Parent->HasSplays();
281}
282
283bool GfxCore::HasSurfaceLegs() const
284{
285    return m_Parent->HasSurfaceLegs();
286}
287
288bool GfxCore::HasTubes() const
289{
290    return m_Parent->HasTubes();
291}
292
293void GfxCore::UpdateBlobs()
294{
295    InvalidateList(LIST_BLOBS);
296}
297
298//
299//  Event handlers
300//
301
302void GfxCore::OnLeaveWindow(wxMouseEvent&) {
303    SetHere();
304    ClearCoords();
305}
306
307void GfxCore::OnIdle(wxIdleEvent& event)
308{
309    // Handle an idle event.
310    if (Animate()) {
311        ForceRefresh();
312        event.RequestMore();
313    }
314}
315
316void GfxCore::OnPaint(wxPaintEvent&)
317{
318    // Redraw the window.
319
320    // Get a graphics context.
321    wxPaintDC dc(this);
322
323    if (m_HaveData) {
324        // Make sure we're initialised.
325        bool first_time = !m_DoneFirstShow;
326        if (first_time) {
327            FirstShow();
328        }
329
330        StartDrawing();
331
332        // Clear the background.
333        Clear();
334
335        // Set up model transformation matrix.
336        SetDataTransform();
337
338        timer.Start(); // reset timer
339
340        if (m_Legs || m_Tubes) {
341            if (m_Tubes) {
342                EnableSmoothPolygons(true); // FIXME: allow false for wireframe view
343                DrawList(LIST_TUBES);
344                DisableSmoothPolygons();
345            }
346
347            // Draw the underground legs.  Do this last so that anti-aliasing
348            // works over polygons.
349            SetColour(col_GREEN);
350            DrawList(LIST_UNDERGROUND_LEGS);
351        }
352
353        if (m_Surface) {
354            // Draw the surface legs.
355            DrawList(LIST_SURFACE_LEGS);
356        }
357
358        if (m_BoundingBox) {
359            DrawShadowedBoundingBox();
360        }
361        if (m_Grid) {
362            // Draw the grid.
363            DrawList(LIST_GRID);
364        }
365
366        DrawList(LIST_BLOBS);
367
368        if (m_Crosses) {
369            DrawList(LIST_CROSSES);
370        }
371
372        SetIndicatorTransform();
373
374        // Draw station names.
375        if (m_Names /*&& !m_Control->MouseDown() && !Animating()*/) {
376            SetColour(NAME_COLOUR);
377
378            if (m_OverlappingNames) {
379                SimpleDrawNames();
380            } else {
381                NattyDrawNames();
382            }
383        }
384
385        // Draw indicators.
386        //
387        // There's no advantage in generating an OpenGL list for the
388        // indicators since they change with almost every redraw (and
389        // sometimes several times between redraws).  This way we avoid
390        // the need to track when to update the indicator OpenGL list,
391        // and also avoid indicator update bugs when we don't quite get this
392        // right...
393        DrawIndicators();
394
395        if (MeasuringLineActive()) {
396            // Draw "here" and "there".
397            double hx, hy;
398            SetColour(HERE_COLOUR);
399            if (m_here.IsValid()) {
400                double dummy;
401                Transform(m_here, &hx, &hy, &dummy);
402                if (!m_here_is_temporary) DrawRing(hx, hy);
403            }
404            if (m_there.IsValid()) {
405                double tx, ty;
406                double dummy;
407                Transform(m_there, &tx, &ty, &dummy);
408                if (m_here.IsValid()) {
409                    BeginLines();
410                    PlaceIndicatorVertex(hx, hy);
411                    PlaceIndicatorVertex(tx, ty);
412                    EndLines();
413                }
414                BeginBlobs();
415                DrawBlob(tx, ty);
416                EndBlobs();
417            }
418        }
419
420        FinishDrawing();
421
422        drawtime = timer.Time();
423    } else {
424        dc.SetBackground(wxSystemSettings::GetColour(wxSYS_COLOUR_WINDOWFRAME));
425        dc.Clear();
426    }
427}
428
429void GfxCore::DrawBoundingBox()
430{
431    const Vector3 v = 0.5 * m_Parent->GetExtent();
432
433    SetColour(col_BLUE);
434    EnableDashedLines();
435    BeginPolyline();
436    PlaceVertex(-v.GetX(), -v.GetY(), v.GetZ());
437    PlaceVertex(-v.GetX(), v.GetY(), v.GetZ());
438    PlaceVertex(v.GetX(), v.GetY(), v.GetZ());
439    PlaceVertex(v.GetX(), -v.GetY(), v.GetZ());
440    PlaceVertex(-v.GetX(), -v.GetY(), v.GetZ());
441    EndPolyline();
442    BeginPolyline();
443    PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
444    PlaceVertex(-v.GetX(), v.GetY(), -v.GetZ());
445    PlaceVertex(v.GetX(), v.GetY(), -v.GetZ());
446    PlaceVertex(v.GetX(), -v.GetY(), -v.GetZ());
447    PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
448    EndPolyline();
449    BeginLines();
450    PlaceVertex(-v.GetX(), -v.GetY(), v.GetZ());
451    PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
452    PlaceVertex(-v.GetX(), v.GetY(), v.GetZ());
453    PlaceVertex(-v.GetX(), v.GetY(), -v.GetZ());
454    PlaceVertex(v.GetX(), v.GetY(), v.GetZ());
455    PlaceVertex(v.GetX(), v.GetY(), -v.GetZ());
456    PlaceVertex(v.GetX(), -v.GetY(), v.GetZ());
457    PlaceVertex(v.GetX(), -v.GetY(), -v.GetZ());
458    EndLines();
459    DisableDashedLines();
460}
461
462void GfxCore::DrawShadowedBoundingBox()
463{
464    const Vector3 v = 0.5 * m_Parent->GetExtent();
465
466    DrawBoundingBox();
467
468    PolygonOffset(true);
469    SetColour(col_DARK_GREY);
470    BeginQuadrilaterals();
471    PlaceVertex(-v.GetX(), -v.GetY(), -v.GetZ());
472    PlaceVertex(-v.GetX(), v.GetY(), -v.GetZ());
473    PlaceVertex(v.GetX(), v.GetY(), -v.GetZ());
474    PlaceVertex(v.GetX(), -v.GetY(), -v.GetZ());
475    EndQuadrilaterals();
476    PolygonOffset(false);
477
478    DrawList(LIST_SHADOW);
479}
480
481void GfxCore::DrawGrid()
482{
483    // Draw the grid.
484    SetColour(col_RED);
485
486    // Calculate the extent of the survey, in metres across the screen plane.
487    Double m_across_screen = SurveyUnitsAcrossViewport();
488    // Calculate the length of the scale bar in metres.
489    //--move this elsewhere
490    Double size_snap = pow(10.0, floor(log10(0.75 * m_across_screen)));
491    Double t = m_across_screen * 0.75 / size_snap;
492    if (t >= 5.0) {
493        size_snap *= 5.0;
494    }
495    else if (t >= 2.0) {
496        size_snap *= 2.0;
497    }
498
499    Double grid_size = size_snap * 0.1;
500    Double edge = grid_size * 2.0;
501    Double grid_z = -m_Parent->GetZExtent() * 0.5 - grid_size;
502    Double left = -m_Parent->GetXExtent() * 0.5 - edge;
503    Double right = m_Parent->GetXExtent() * 0.5 + edge;
504    Double bottom = -m_Parent->GetYExtent() * 0.5 - edge;
505    Double top = m_Parent->GetYExtent() * 0.5 + edge;
506    int count_x = (int) ceil((right - left) / grid_size);
507    int count_y = (int) ceil((top - bottom) / grid_size);
508    Double actual_right = left + count_x*grid_size;
509    Double actual_top = bottom + count_y*grid_size;
510
511    BeginLines();
512
513    for (int xc = 0; xc <= count_x; xc++) {
514        Double x = left + xc*grid_size;
515
516        PlaceVertex(x, bottom, grid_z);
517        PlaceVertex(x, actual_top, grid_z);
518    }
519
520    for (int yc = 0; yc <= count_y; yc++) {
521        Double y = bottom + yc*grid_size;
522        PlaceVertex(left, y, grid_z);
523        PlaceVertex(actual_right, y, grid_z);
524    }
525
526    EndLines();
527}
528
529int GfxCore::GetClinoOffset() const
530{
531    return m_Compass ? CLINO_OFFSET_X : INDICATOR_OFFSET_X;
532}
533
534void GfxCore::DrawTick(int angle_cw)
535{
536    const Double theta = rad(angle_cw);
537    const wxCoord length1 = INDICATOR_RADIUS;
538    const wxCoord length0 = length1 + TICK_LENGTH;
539    wxCoord x0 = wxCoord(length0 * sin(theta));
540    wxCoord y0 = wxCoord(length0 * cos(theta));
541    wxCoord x1 = wxCoord(length1 * sin(theta));
542    wxCoord y1 = wxCoord(length1 * cos(theta));
543
544    PlaceIndicatorVertex(x0, y0);
545    PlaceIndicatorVertex(x1, y1);
546}
547
548void GfxCore::DrawArrow(gla_colour col1, gla_colour col2) {
549    Vector3 p1(0, INDICATOR_RADIUS, 0);
550    Vector3 p2(INDICATOR_RADIUS/2, INDICATOR_RADIUS*-.866025404, 0); // 150deg
551    Vector3 p3(-INDICATOR_RADIUS/2, INDICATOR_RADIUS*-.866025404, 0); // 210deg
552    Vector3 pc(0, 0, 0);
553
554    DrawTriangle(col_LIGHT_GREY, col1, p2, p1, pc);
555    DrawTriangle(col_LIGHT_GREY, col2, p3, p1, pc);
556}
557
558void GfxCore::DrawCompass() {
559    // Ticks.
560    BeginLines();
561    for (int angle = 315; angle > 0; angle -= 45) {
562        DrawTick(angle);
563    }
564    SetColour(col_GREEN);
565    DrawTick(0);
566    EndLines();
567
568    // Compass background.
569    DrawCircle(col_LIGHT_GREY_2, col_GREY, 0, 0, INDICATOR_RADIUS);
570
571    // Compass arrow.
572    DrawArrow(col_INDICATOR_1, col_INDICATOR_2);
573}
574
575// Draw the non-rotating background to the clino.
576void GfxCore::DrawClinoBack() {
577    BeginLines();
578    for (int angle = 0; angle <= 180; angle += 90) {
579        DrawTick(angle);
580    }
581
582    SetColour(col_GREY);
583    PlaceIndicatorVertex(0, INDICATOR_RADIUS);
584    PlaceIndicatorVertex(0, -INDICATOR_RADIUS);
585    PlaceIndicatorVertex(0, 0);
586    PlaceIndicatorVertex(INDICATOR_RADIUS, 0);
587
588    EndLines();
589}
590
591void GfxCore::DrawClino() {
592    // Ticks.
593    SetColour(col_GREEN);
594    BeginLines();
595    DrawTick(0);
596    EndLines();
597
598    // Clino background.
599    DrawSemicircle(col_LIGHT_GREY_2, col_GREY, 0, 0, INDICATOR_RADIUS, 0);
600
601    // Elevation arrow.
602    DrawArrow(col_INDICATOR_2, col_INDICATOR_1);
603}
604
605void GfxCore::Draw2dIndicators()
606{
607    // Draw the compass and elevation indicators.
608
609    const int centre_y = INDICATOR_BOX_SIZE / 2 + INDICATOR_OFFSET_Y;
610
611    const int comp_centre_x = GetCompassXPosition();
612
613    if (m_Compass && !m_Parent->IsExtendedElevation()) {
614        // If the user is dragging the compass with the pointer outside the
615        // compass, we snap to 45 degree multiples, and the ticks go white.
616        SetColour(m_MouseOutsideCompass ? col_WHITE : col_LIGHT_GREY_2);
617        DrawList2D(LIST_COMPASS, comp_centre_x, centre_y, -m_PanAngle);
618    }
619
620    const int elev_centre_x = GetClinoXPosition();
621
622    if (m_Clino) {
623        // If the user is dragging the clino with the pointer outside the
624        // clino, we snap to 90 degree multiples, and the ticks go white.
625        SetColour(m_MouseOutsideElev ? col_WHITE : col_LIGHT_GREY_2);
626        DrawList2D(LIST_CLINO_BACK, elev_centre_x, centre_y, 0);
627        DrawList2D(LIST_CLINO, elev_centre_x, centre_y, 90 - m_TiltAngle);
628    }
629
630    SetColour(TEXT_COLOUR);
631
632    static int triple_zero_width = 0;
633    static int height = 0;
634    if (!triple_zero_width) {
635        GetTextExtent(wxT("000"), &triple_zero_width, &height);
636    }
637    const int y_off = INDICATOR_OFFSET_Y + INDICATOR_BOX_SIZE + height / 2;
638
639    if (m_Compass && !m_Parent->IsExtendedElevation()) {
640        wxString str;
641        int value;
642        if (m_Degrees) {
643            value = int(m_PanAngle);
644        } else {
645            value = int(m_PanAngle * 200.0 / 180.0);
646        }
647        str.Printf(wxT("%03d"), value);
648        DrawIndicatorText(comp_centre_x - triple_zero_width / 2, y_off, str);
649
650        str = wmsg(/*Facing*/203);
651        int w;
652        GetTextExtent(str, &w, NULL);
653        DrawIndicatorText(comp_centre_x - w / 2, y_off + height, str);
654    }
655
656    if (m_Clino) {
657        int angle;
658        wxString str;
659        int width;
660        if (m_Degrees) {
661            static int zero_zero_width = 0;
662            if (!zero_zero_width) {
663                GetTextExtent(wxT("00"), &zero_zero_width, NULL);
664            }
665            width = zero_zero_width;
666            angle = int(m_TiltAngle);
667            str = angle ? wxString::Format(wxT("%+03d"), angle) : wxT("00");
668        } else {
669            width = triple_zero_width;
670            angle = int(m_TiltAngle * 200.0 / 180.0);
671            str = angle ? wxString::Format(wxT("%+04d"), angle) : wxT("000");
672        }
673
674        // Adjust horizontal position so the left of the first digit is
675        // always in the same place.
676        int sign_offset = 0;
677        if (angle < 0) {
678            static int minus_width = 0;
679            if (!minus_width) {
680                GetTextExtent(wxT("-"), &minus_width, NULL);
681            }
682            sign_offset = minus_width + 1;
683        } else if (angle > 0) {
684            static int plus_width = 0;
685            if (!plus_width) {
686                GetTextExtent(wxT("+"), &plus_width, NULL);
687            }
688            sign_offset = plus_width + 1;
689        }
690        DrawIndicatorText(elev_centre_x - sign_offset - width / 2, y_off, str);
691
692        str = wmsg(/*Elevation*/118);
693        int w;
694        GetTextExtent(str, &w, NULL);
695        DrawIndicatorText(elev_centre_x - w / 2, y_off + height, str);
696    }
697}
698
699void GfxCore::NattyDrawNames()
700{
701    // Draw station names, without overlapping.
702
703    const unsigned int quantise(GetFontSize() / QUANTISE_FACTOR);
704    const unsigned int quantised_x = GetXSize() / quantise;
705    const unsigned int quantised_y = GetYSize() / quantise;
706    const size_t buffer_size = quantised_x * quantised_y;
707
708    if (!m_LabelGrid) m_LabelGrid = new char[buffer_size];
709
710    memset((void*) m_LabelGrid, 0, buffer_size);
711
712    list<LabelInfo*>::const_iterator label = m_Parent->GetLabels();
713    for ( ; label != m_Parent->GetLabelsEnd(); ++label) {
714        if (!((m_Surface && (*label)->IsSurface()) ||
715              (m_Legs && (*label)->IsUnderground()) ||
716              (!(*label)->IsSurface() && !(*label)->IsUnderground()))) {
717            // if this station isn't to be displayed, skip to the next
718            // (last case is for stns with no legs attached)
719            continue;
720        }
721
722        double x, y, z;
723
724        Transform(**label, &x, &y, &z);
725        // Check if the label is behind us (in perspective view).
726        if (z <= 0.0 || z >= 1.0) continue;
727
728        // Apply a small shift so that translating the view doesn't make which
729        // labels are displayed change as the resulting twinkling effect is
730        // distracting.
731        double tx, ty, tz;
732        Transform(Vector3(), &tx, &ty, &tz);
733        tx -= floor(tx / quantise) * quantise;
734        ty -= floor(ty / quantise) * quantise;
735
736        tx = x - tx;
737        if (tx < 0) continue;
738
739        ty = y - ty;
740        if (ty < 0) continue;
741
742        unsigned int iy = unsigned(ty) / quantise;
743        if (iy >= quantised_y) continue;
744        unsigned int width = (*label)->get_width();
745        unsigned int ix = unsigned(tx) / quantise;
746        if (ix + width >= quantised_x) continue;
747
748        char * test = m_LabelGrid + ix + iy * quantised_x;
749        if (memchr(test, 1, width)) continue;
750
751        x += 3;
752        y -= GetFontSize() / 2;
753        DrawIndicatorText((int)x, (int)y, (*label)->GetText());
754
755        if (iy > QUANTISE_FACTOR) iy = QUANTISE_FACTOR;
756        test -= quantised_x * iy;
757        iy += 4;
758        while (--iy && test < m_LabelGrid + buffer_size) {
759            memset(test, 1, width);
760            test += quantised_x;
761        }
762    }
763}
764
765void GfxCore::SimpleDrawNames()
766{
767    // Draw all station names, without worrying about overlaps
768    list<LabelInfo*>::const_iterator label = m_Parent->GetLabels();
769    for ( ; label != m_Parent->GetLabelsEnd(); ++label) {
770        if (!((m_Surface && (*label)->IsSurface()) ||
771              (m_Legs && (*label)->IsUnderground()) ||
772              (!(*label)->IsSurface() && !(*label)->IsUnderground()))) {
773            // if this station isn't to be displayed, skip to the next
774            // (last case is for stns with no legs attached)
775            continue;
776        }
777
778        double x, y, z;
779        Transform(**label, &x, &y, &z);
780
781        // Check if the label is behind us (in perspective view).
782        if (z <= 0) continue;
783
784        x += 3;
785        y -= GetFontSize() / 2;
786        DrawIndicatorText((int)x, (int)y, (*label)->GetText());
787    }
788}
789
790void GfxCore::DrawColourKey(int num_bands, const wxString & other, const wxString & units)
791{
792    int total_block_height =
793        KEY_BLOCK_HEIGHT * (num_bands == 1 ? num_bands : num_bands - 1);
794    if (!other.empty()) total_block_height += KEY_BLOCK_HEIGHT * 2;
795    if (!units.empty()) total_block_height += KEY_BLOCK_HEIGHT;
796
797    const int bottom = -total_block_height;
798
799    int size = 0;
800    if (!other.empty()) GetTextExtent(other, &size, NULL);
801    int band;
802    for (band = 0; band < num_bands; ++band) {
803        int x;
804        GetTextExtent(key_legends[band], &x, NULL);
805        if (x > size) size = x;
806    }
807
808    int left = -KEY_BLOCK_WIDTH - size;
809
810    key_lowerleft[m_ColourBy].x = left - KEY_EXTRA_LEFT_MARGIN;
811    key_lowerleft[m_ColourBy].y = bottom;
812
813    int y = bottom;
814    if (!units.empty()) y += KEY_BLOCK_HEIGHT;
815
816    if (!other.empty()) {
817        DrawShadedRectangle(GetSurfacePen(), GetSurfacePen(), left, y,
818                KEY_BLOCK_WIDTH, KEY_BLOCK_HEIGHT);
819        SetColour(col_BLACK);
820        BeginPolyline();
821        PlaceIndicatorVertex(left, y);
822        PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, y);
823        PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, y + KEY_BLOCK_HEIGHT);
824        PlaceIndicatorVertex(left, y + KEY_BLOCK_HEIGHT);
825        PlaceIndicatorVertex(left, y);
826        EndPolyline();
827        y += KEY_BLOCK_HEIGHT * 2;
828    }
829
830    int start = y;
831    if (num_bands == 1) {
832        DrawShadedRectangle(GetPen(0), GetPen(0), left, y,
833                            KEY_BLOCK_WIDTH, KEY_BLOCK_HEIGHT);
834        y += KEY_BLOCK_HEIGHT;
835    } else {
836        for (band = 0; band < num_bands - 1; ++band) {
837            DrawShadedRectangle(GetPen(band), GetPen(band + 1), left, y,
838                                KEY_BLOCK_WIDTH, KEY_BLOCK_HEIGHT);
839            y += KEY_BLOCK_HEIGHT;
840        }
841    }
842
843    SetColour(col_BLACK);
844    BeginPolyline();
845    PlaceIndicatorVertex(left, y);
846    PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, y);
847    PlaceIndicatorVertex(left + KEY_BLOCK_WIDTH, start);
848    PlaceIndicatorVertex(left, start);
849    PlaceIndicatorVertex(left, y);
850    EndPolyline();
851
852    SetColour(TEXT_COLOUR);
853
854    y = bottom;
855    if (!units.empty()) {
856        GetTextExtent(units, &size, NULL);
857        DrawIndicatorText(left + (KEY_BLOCK_WIDTH - size) / 2, y, units);
858        y += KEY_BLOCK_HEIGHT;
859    }
860    y -= GetFontSize() / 2;
861    left += KEY_BLOCK_WIDTH + 5;
862
863    if (!other.empty()) {
864        y += KEY_BLOCK_HEIGHT / 2;
865        DrawIndicatorText(left, y, other);
866        y += KEY_BLOCK_HEIGHT * 2 - KEY_BLOCK_HEIGHT / 2;
867    }
868
869    if (num_bands == 1) {
870        y += KEY_BLOCK_HEIGHT / 2;
871        DrawIndicatorText(left, y, key_legends[0]);
872    } else {
873        for (band = 0; band < num_bands; ++band) {
874            DrawIndicatorText(left, y, key_legends[band]);
875            y += KEY_BLOCK_HEIGHT;
876        }
877    }
878}
879
880void GfxCore::DrawDepthKey()
881{
882    Double z_ext = m_Parent->GetDepthExtent();
883    int num_bands = 1;
884    int sf = 0;
885    if (z_ext > 0.0) {
886        num_bands = GetNumColourBands();
887        Double z_range = z_ext;
888        if (!m_Metric) z_range /= METRES_PER_FOOT;
889        sf = max(0, 1 - (int)floor(log10(z_range)));
890    }
891
892    Double z_min = m_Parent->GetDepthMin() + m_Parent->GetOffset().GetZ();
893    for (int band = 0; band < num_bands; ++band) {
894        Double z = z_min;
895        if (band)
896            z += z_ext * band / (num_bands - 1);
897
898        if (!m_Metric)
899            z /= METRES_PER_FOOT;
900
901        key_legends[band].Printf(wxT("%.*f"), sf, z);
902    }
903
904    DrawColourKey(num_bands, wxString(), m_Metric ? wxT("m") : wxT("ft"));
905}
906
907void GfxCore::DrawDateKey()
908{
909    int num_bands;
910    if (!HasDateInformation()) {
911        num_bands = 0;
912    } else {
913        int date_ext = m_Parent->GetDateExtent();
914        if (date_ext == 0) {
915            num_bands = 1;
916        } else {
917            num_bands = GetNumColourBands();
918        }
919        for (int band = 0; band < num_bands; ++band) {
920            int y, m, d;
921            int days = m_Parent->GetDateMin();
922            if (band)
923                days += date_ext * band / (num_bands - 1);
924            ymd_from_days_since_1900(days, &y, &m, &d);
925            key_legends[band].Printf(wxT("%04d-%02d-%02d"), y, m, d);
926        }
927    }
928
929    wxString other;
930    if (!m_Parent->HasCompleteDateInfo()) {
931        other = wmsg(/*Undated*/221);
932    }
933
934    DrawColourKey(num_bands, other, wxString());
935}
936
937void GfxCore::DrawErrorKey()
938{
939    int num_bands;
940    if (HasErrorInformation()) {
941        // Use fixed colours for each error factor so it's directly visually
942        // comparable between surveys.
943        num_bands = GetNumColourBands();
944        for (int band = 0; band < num_bands; ++band) {
945            double E = MAX_ERROR * band / (num_bands - 1);
946            key_legends[band].Printf(wxT("%.2f"), E);
947        }
948    } else {
949        num_bands = 0;
950    }
951
952    // Always show the "Not in loop" legend for now (FIXME).
953    DrawColourKey(num_bands, wmsg(/*Not in loop*/290), wxString());
954}
955
956void GfxCore::DrawScaleBar()
957{
958    // Draw the scalebar.
959    if (GetPerspective()) return;
960
961    // Calculate how many metres of survey are currently displayed across the
962    // screen.
963    Double across_screen = SurveyUnitsAcrossViewport();
964
965    double f = double(GetClinoXPosition() - INDICATOR_BOX_SIZE / 2 - SCALE_BAR_OFFSET_X) / GetXSize();
966    if (f > 0.75) {
967        f = 0.75;
968    } else if (f < 0.5) {
969        // Stop it getting squeezed to nothing.
970        // FIXME: In this case we should probably move the compass and clino up
971        // to make room rather than letting stuff overlap.
972        f = 0.5;
973    }
974
975    // Convert to imperial measurements if required.
976    Double multiplier = 1.0;
977    if (!m_Metric) {
978        across_screen /= METRES_PER_FOOT;
979        multiplier = METRES_PER_FOOT;
980        if (across_screen >= 5280.0 / f) {
981            across_screen /= 5280.0;
982            multiplier *= 5280.0;
983        }
984    }
985
986    // Calculate the length of the scale bar.
987    Double size_snap = pow(10.0, floor(log10(f * across_screen)));
988    Double t = across_screen * f / size_snap;
989    if (t >= 5.0) {
990        size_snap *= 5.0;
991    } else if (t >= 2.0) {
992        size_snap *= 2.0;
993    }
994
995    if (!m_Metric) size_snap *= multiplier;
996
997    // Actual size of the thing in pixels:
998    int size = int((size_snap / SurveyUnitsAcrossViewport()) * GetXSize());
999    m_ScaleBarWidth = size;
1000
1001    // Draw it...
1002    const int end_y = SCALE_BAR_OFFSET_Y + SCALE_BAR_HEIGHT;
1003    int interval = size / 10;
1004
1005    gla_colour col = col_WHITE;
1006    for (int ix = 0; ix < 10; ix++) {
1007        int x = SCALE_BAR_OFFSET_X + int(ix * ((Double) size / 10.0));
1008
1009        DrawRectangle(col, col, x, end_y, interval + 2, SCALE_BAR_HEIGHT);
1010
1011        col = (col == col_WHITE) ? col_GREY : col_WHITE;
1012    }
1013
1014    // Add labels.
1015    wxString str;
1016    if (m_Metric) {
1017        if (size_snap >= 1e3) {
1018            str.Printf(wxT("%.fkm"), size_snap * 1e-3);
1019        } else if (size_snap >= 1.0) {
1020            str.Printf(wxT("%.fm"), size_snap);
1021        } else {
1022            int sf = -(int)floor(log10(size_snap));
1023            str.Printf(wxT("%.*fm"), sf, size_snap);
1024        }
1025    } else {
1026        size_snap /= METRES_PER_FOOT;
1027        Double miles = size_snap / 5280.0;
1028        if (miles >= 2.0) {
1029            str.Printf(wxT("%.f miles"), miles);
1030        } else if (miles >= 1.0) {
1031            str.Printf(wxT("%.f mile"), miles);
1032        } else if (size_snap >= 1.0) {
1033            str.Printf(wxT("%.fft"), size_snap);
1034        } else {
1035            Double inches = size_snap * 12;
1036            if (inches >= 1.0) {
1037                str.Printf(wxT("%.fin"), inches);
1038            } else {
1039                int sf = -(int)floor(log10(inches));
1040                str.Printf(wxT("%.*fin"), sf, inches);
1041            }
1042        }
1043    }
1044
1045    int text_width, text_height;
1046    GetTextExtent(str, &text_width, &text_height);
1047    const int text_y = end_y - text_height + 1;
1048    SetColour(TEXT_COLOUR);
1049    DrawIndicatorText(SCALE_BAR_OFFSET_X, text_y, wxT("0"));
1050    DrawIndicatorText(SCALE_BAR_OFFSET_X + size - text_width, text_y, str);
1051}
1052
1053bool GfxCore::CheckHitTestGrid(const wxPoint& point, bool centre)
1054{
1055    if (Animating()) return false;
1056
1057    if (point.x < 0 || point.x >= GetXSize() ||
1058        point.y < 0 || point.y >= GetYSize()) {
1059        return false;
1060    }
1061
1062    SetDataTransform();
1063
1064    if (!m_HitTestGridValid) CreateHitTestGrid();
1065
1066    int grid_x = (point.x * (HITTEST_SIZE - 1)) / GetXSize();
1067    int grid_y = (point.y * (HITTEST_SIZE - 1)) / GetYSize();
1068
1069    LabelInfo *best = NULL;
1070    int dist_sqrd = 25;
1071    int square = grid_x + grid_y * HITTEST_SIZE;
1072    list<LabelInfo*>::iterator iter = m_PointGrid[square].begin();
1073
1074    while (iter != m_PointGrid[square].end()) {
1075        LabelInfo *pt = *iter++;
1076
1077        double cx, cy, cz;
1078
1079        Transform(*pt, &cx, &cy, &cz);
1080
1081        cy = GetYSize() - cy;
1082
1083        int dx = point.x - int(cx);
1084        int ds = dx * dx;
1085        if (ds >= dist_sqrd) continue;
1086        int dy = point.y - int(cy);
1087
1088        ds += dy * dy;
1089        if (ds >= dist_sqrd) continue;
1090
1091        dist_sqrd = ds;
1092        best = pt;
1093
1094        if (ds == 0) break;
1095    }
1096
1097    if (best) {
1098        m_Parent->ShowInfo(best);
1099        if (centre) {
1100            // FIXME: allow Ctrl-Click to not set there or something?
1101            CentreOn(*best);
1102            WarpPointer(GetXSize() / 2, GetYSize() / 2);
1103            SetThere(*best);
1104            m_Parent->SelectTreeItem(best);
1105        }
1106    } else {
1107        // Left-clicking not on a survey cancels the measuring line.
1108        if (centre) {
1109            ClearTreeSelection();
1110        } else {
1111            m_Parent->ShowInfo(best);
1112            double x, y, z;
1113            ReverseTransform(point.x, GetYSize() - point.y, &x, &y, &z);
1114            SetHere(Point(Vector3(x, y, z)));
1115            m_here_is_temporary = true;
1116        }
1117    }
1118
1119    return best;
1120}
1121
1122void GfxCore::OnSize(wxSizeEvent& event)
1123{
1124    // Handle a change in window size.
1125    wxSize size = event.GetSize();
1126
1127    if (size.GetWidth() <= 0 || size.GetHeight() <= 0) {
1128        // Before things are fully initialised, we sometimes get a bogus
1129        // resize message...
1130        // FIXME have changes in MainFrm cured this?  It still happens with
1131        // 1.0.32 and wxGTK 2.5.2 (load a file from the command line).
1132        // With 1.1.6 and wxGTK 2.4.2 we only get negative sizes if MainFrm
1133        // is resized such that the GfxCore window isn't visible.
1134        //printf("OnSize(%d,%d)\n", size.GetWidth(), size.GetHeight());
1135        return;
1136    }
1137
1138    event.Skip();
1139
1140    if (m_DoneFirstShow) {
1141        TryToFreeArrays();
1142
1143        m_HitTestGridValid = false;
1144
1145        ForceRefresh();
1146    }
1147}
1148
1149void GfxCore::DefaultParameters()
1150{
1151    // Set default viewing parameters.
1152
1153    m_Surface = false;
1154    if (!m_Parent->HasUndergroundLegs()) {
1155        if (m_Parent->HasSurfaceLegs()) {
1156            // If there are surface legs, but no underground legs, turn
1157            // surface surveys on.
1158            m_Surface = true;
1159        } else {
1160            // If there are no legs (e.g. after loading a .pos file), turn
1161            // crosses on.
1162            m_Crosses = true;
1163        }
1164    }
1165
1166    m_PanAngle = 0.0;
1167    if (m_Parent->IsExtendedElevation()) {
1168        m_TiltAngle = 0.0;
1169    } else {
1170        m_TiltAngle = -90.0;
1171    }
1172
1173    SetRotation(m_PanAngle, m_TiltAngle);
1174    SetTranslation(Vector3());
1175
1176    m_RotationStep = 30.0;
1177    m_Rotating = false;
1178    m_SwitchingTo = 0;
1179    m_Entrances = false;
1180    m_FixedPts = false;
1181    m_ExportedPts = false;
1182    m_Grid = false;
1183    m_BoundingBox = false;
1184    m_Tubes = false;
1185    if (GetPerspective()) TogglePerspective();
1186}
1187
1188void GfxCore::Defaults()
1189{
1190    // Restore default scale, rotation and translation parameters.
1191    DefaultParameters();
1192    SetScale(1.0);
1193
1194    // Invalidate all the cached lists.
1195    GLACanvas::FirstShow();
1196
1197    ForceRefresh();
1198}
1199
1200// return: true if animation occured (and ForceRefresh() needs to be called)
1201bool GfxCore::Animate()
1202{
1203    if (!Animating()) return false;
1204
1205    // Don't show pointer coordinates while animating.
1206    // FIXME : only do this when we *START* animating!  Use a static copy
1207    // of the value of "Animating()" last time we were here to track this?
1208    // MainFrm now checks if we're trying to clear already cleared labels
1209    // and just returns, but it might be simpler to check here!
1210    ClearCoords();
1211    m_Parent->ShowInfo(NULL);
1212
1213    static double last_t = 0;
1214    double t;
1215    if (movie) {
1216        ReadPixels(movie->GetWidth(), movie->GetHeight(), movie->GetBuffer());
1217        if (!movie->AddFrame()) {
1218            wxGetApp().ReportError(wxString(movie->get_error_string(), wxConvUTF8));
1219            delete movie;
1220            movie = NULL;
1221            presentation_mode = 0;
1222            return false;
1223        }
1224        t = 1.0 / 25.0; // 25 frames per second
1225    } else {
1226        t = timer.Time() * 1.0e-3;
1227        if (t == 0) t = 0.001;
1228        else if (t > 1.0) t = 1.0;
1229        if (last_t > 0) t = (t + last_t) / 2;
1230        last_t = t;
1231    }
1232
1233    if (presentation_mode == PLAYING && pres_speed != 0.0) {
1234        t *= fabs(pres_speed);
1235        while (t >= next_mark_time) {
1236            t -= next_mark_time;
1237            this_mark_total = 0;
1238            PresentationMark prev_mark = next_mark;
1239            if (prev_mark.angle < 0) prev_mark.angle += 360.0;
1240            else if (prev_mark.angle >= 360.0) prev_mark.angle -= 360.0;
1241            if (pres_reverse)
1242                next_mark = m_Parent->GetPresMark(MARK_PREV);
1243            else
1244                next_mark = m_Parent->GetPresMark(MARK_NEXT);
1245            if (!next_mark.is_valid()) {
1246                SetView(prev_mark);
1247                presentation_mode = 0;
1248                if (movie && !movie->Close()) {
1249                    wxGetApp().ReportError(wxString(movie->get_error_string(), wxConvUTF8));
1250                }
1251                delete movie;
1252                movie = NULL;
1253                break;
1254            }
1255
1256            double tmp = (pres_reverse ? prev_mark.time : next_mark.time);
1257            if (tmp > 0) {
1258                next_mark_time = tmp;
1259            } else {
1260                double d = (next_mark - prev_mark).magnitude();
1261                // FIXME: should ignore component of d which is unseen in
1262                // non-perspective mode?
1263                next_mark_time = sqrd(d / 30.0);
1264                double a = next_mark.angle - prev_mark.angle;
1265                if (a > 180.0) {
1266                    next_mark.angle -= 360.0;
1267                    a = 360.0 - a;
1268                } else if (a < -180.0) {
1269                    next_mark.angle += 360.0;
1270                    a += 360.0;
1271                } else {
1272                    a = fabs(a);
1273                }
1274                next_mark_time += sqrd(a / 60.0);
1275                double ta = fabs(next_mark.tilt_angle - prev_mark.tilt_angle);
1276                next_mark_time += sqrd(ta / 60.0);
1277                double s = fabs(log(next_mark.scale) - log(prev_mark.scale));
1278                next_mark_time += sqrd(s / 2.0);
1279                next_mark_time = sqrt(next_mark_time);
1280                // was: next_mark_time = max(max(d / 30, s / 2), max(a, ta) / 60);
1281                //printf("*** %.6f from (\nd: %.6f\ns: %.6f\na: %.6f\nt: %.6f )\n",
1282                //       next_mark_time, d/30.0, s/2.0, a/60.0, ta/60.0);
1283                if (tmp < 0) next_mark_time /= -tmp;
1284            }
1285        }
1286
1287        if (presentation_mode) {
1288            // Advance position towards next_mark
1289            double p = t / next_mark_time;
1290            double q = 1 - p;
1291            PresentationMark here = GetView();
1292            if (next_mark.angle < 0) {
1293                if (here.angle >= next_mark.angle + 360.0)
1294                    here.angle -= 360.0;
1295            } else if (next_mark.angle >= 360.0) {
1296                if (here.angle <= next_mark.angle - 360.0)
1297                    here.angle += 360.0;
1298            }
1299            here.assign(q * here + p * next_mark);
1300            here.angle = q * here.angle + p * next_mark.angle;
1301            if (here.angle < 0) here.angle += 360.0;
1302            else if (here.angle >= 360.0) here.angle -= 360.0;
1303            here.tilt_angle = q * here.tilt_angle + p * next_mark.tilt_angle;
1304            here.scale = exp(q * log(here.scale) + p * log(next_mark.scale));
1305            SetView(here);
1306            this_mark_total += t;
1307            next_mark_time -= t;
1308        }
1309    }
1310
1311    // When rotating...
1312    if (m_Rotating) {
1313        TurnCave(m_RotationStep * t);
1314    }
1315
1316    if (m_SwitchingTo == PLAN) {
1317        // When switching to plan view...
1318        TiltCave(-90.0 * t);
1319        if (m_TiltAngle == -90.0) {
1320            m_SwitchingTo = 0;
1321        }
1322    } else if (m_SwitchingTo == ELEVATION) {
1323        // When switching to elevation view...
1324        if (fabs(m_TiltAngle) < 90.0 * t) {
1325            m_SwitchingTo = 0;
1326            TiltCave(-m_TiltAngle);
1327        } else if (m_TiltAngle > 0.0) {
1328            TiltCave(-90.0 * t);
1329        } else {
1330            TiltCave(90.0 * t);
1331        }
1332    } else if (m_SwitchingTo) {
1333        int angle = (m_SwitchingTo - NORTH) * 90;
1334        double diff = angle - m_PanAngle;
1335        if (diff < -180) diff += 360;
1336        if (diff > 180) diff -= 360;
1337        double move = 90.0 * t;
1338        if (move >= fabs(diff)) {
1339            TurnCave(diff);
1340            m_SwitchingTo = 0;
1341        } else if (diff < 0) {
1342            TurnCave(-move);
1343        } else {
1344            TurnCave(move);
1345        }
1346    }
1347
1348    return true;
1349}
1350
1351// How much to allow around the box - this is because of the ring shape
1352// at one end of the line.
1353static const int HIGHLIGHTED_PT_SIZE = 2; // FIXME: tie in to blob and ring size
1354#define MARGIN (HIGHLIGHTED_PT_SIZE * 2 + 1)
1355void GfxCore::RefreshLine(const Point &a, const Point &b, const Point &c)
1356{
1357    // Best of all might be to copy the window contents before we draw the
1358    // line, then replace each time we redraw.
1359
1360    // Calculate the minimum rectangle which includes the old and new
1361    // measuring lines to minimise the redraw time
1362    int l = INT_MAX, r = INT_MIN, u = INT_MIN, d = INT_MAX;
1363    double X, Y, Z;
1364    if (a.IsValid()) {
1365        if (!Transform(a, &X, &Y, &Z)) {
1366            printf("oops\n");
1367        } else {
1368            int x = int(X);
1369            int y = GetYSize() - 1 - int(Y);
1370            l = x - MARGIN;
1371            r = x + MARGIN;
1372            u = y + MARGIN;
1373            d = y - MARGIN;
1374        }
1375    }
1376    if (b.IsValid()) {
1377        if (!Transform(b, &X, &Y, &Z)) {
1378            printf("oops\n");
1379        } else {
1380            int x = int(X);
1381            int y = GetYSize() - 1 - int(Y);
1382            l = min(l, x - MARGIN);
1383            r = max(r, x + MARGIN);
1384            u = max(u, y + MARGIN);
1385            d = min(d, y - MARGIN);
1386        }
1387    }
1388    if (c.IsValid()) {
1389        if (!Transform(c, &X, &Y, &Z)) {
1390            printf("oops\n");
1391        } else {
1392            int x = int(X);
1393            int y = GetYSize() - 1 - int(Y);
1394            l = min(l, x - MARGIN);
1395            r = max(r, x + MARGIN);
1396            u = max(u, y + MARGIN);
1397            d = min(d, y - MARGIN);
1398        }
1399    }
1400    RefreshRect(wxRect(l, d, r - l, u - d), false);
1401}
1402
1403void GfxCore::SetHere()
1404{
1405    if (!m_here.IsValid()) return;
1406    bool line_active = MeasuringLineActive();
1407    Point old = m_here;
1408    m_here.Invalidate();
1409    if (line_active || MeasuringLineActive())
1410        RefreshLine(old, m_there, m_here);
1411}
1412
1413void GfxCore::SetHere(const Point &p)
1414{
1415    bool line_active = MeasuringLineActive();
1416    Point old = m_here;
1417    m_here = p;
1418    if (line_active || MeasuringLineActive())
1419        RefreshLine(old, m_there, m_here);
1420    m_here_is_temporary = false;
1421}
1422
1423void GfxCore::SetThere()
1424{
1425    if (!m_there.IsValid()) return;
1426    Point old = m_there;
1427    m_there.Invalidate();
1428    RefreshLine(m_here, old, m_there);
1429}
1430
1431void GfxCore::SetThere(const Point &p)
1432{
1433    Point old = m_there;
1434    m_there = p;
1435    RefreshLine(m_here, old, m_there);
1436}
1437
1438void GfxCore::CreateHitTestGrid()
1439{
1440    // Clear hit-test grid.
1441    for (int i = 0; i < HITTEST_SIZE * HITTEST_SIZE; i++) {
1442        m_PointGrid[i].clear();
1443    }
1444
1445    // Fill the grid.
1446    list<LabelInfo*>::const_iterator pos = m_Parent->GetLabels();
1447    list<LabelInfo*>::const_iterator end = m_Parent->GetLabelsEnd();
1448    while (pos != end) {
1449        LabelInfo* label = *pos++;
1450
1451        if (!((m_Surface && label->IsSurface()) ||
1452              (m_Legs && label->IsUnderground()) ||
1453              (!label->IsSurface() && !label->IsUnderground()))) {
1454            // if this station isn't to be displayed, skip to the next
1455            // (last case is for stns with no legs attached)
1456            continue;
1457        }
1458
1459        // Calculate screen coordinates.
1460        double cx, cy, cz;
1461        Transform(*label, &cx, &cy, &cz);
1462        if (cx < 0 || cx >= GetXSize()) continue;
1463        if (cy < 0 || cy >= GetYSize()) continue;
1464
1465        cy = GetYSize() - cy;
1466
1467        // On-screen, so add to hit-test grid...
1468        int grid_x = int((cx * (HITTEST_SIZE - 1)) / GetXSize());
1469        int grid_y = int((cy * (HITTEST_SIZE - 1)) / GetYSize());
1470
1471        m_PointGrid[grid_x + grid_y * HITTEST_SIZE].push_back(label);
1472    }
1473
1474    m_HitTestGridValid = true;
1475}
1476
1477//
1478//  Methods for controlling the orientation of the survey
1479//
1480
1481void GfxCore::TurnCave(Double angle)
1482{
1483    // Turn the cave around its z-axis by a given angle.
1484
1485    m_PanAngle += angle;
1486    if (m_PanAngle >= 360.0) {
1487        m_PanAngle -= 360.0;
1488    } else if (m_PanAngle < 0.0) {
1489        m_PanAngle += 360.0;
1490    }
1491
1492    m_HitTestGridValid = false;
1493    if (m_here_is_temporary) SetHere();
1494
1495    SetRotation(m_PanAngle, m_TiltAngle);
1496}
1497
1498void GfxCore::TurnCaveTo(Double angle)
1499{
1500    timer.Start(drawtime);
1501    int new_switching_to = ((int)angle) / 90 + NORTH;
1502    if (new_switching_to == m_SwitchingTo) {
1503        // A second order to switch takes us there right away
1504        TurnCave(angle - m_PanAngle);
1505        m_SwitchingTo = 0;
1506        ForceRefresh();
1507    } else {
1508        m_SwitchingTo = new_switching_to;
1509    }
1510}
1511
1512void GfxCore::TiltCave(Double tilt_angle)
1513{
1514    // Tilt the cave by a given angle.
1515    if (m_TiltAngle + tilt_angle > 90.0) {
1516        m_TiltAngle = 90.0;
1517    } else if (m_TiltAngle + tilt_angle < -90.0) {
1518        m_TiltAngle = -90.0;
1519    } else {
1520        m_TiltAngle += tilt_angle;
1521    }
1522
1523    m_HitTestGridValid = false;
1524    if (m_here_is_temporary) SetHere();
1525
1526    SetRotation(m_PanAngle, m_TiltAngle);
1527}
1528
1529void GfxCore::TranslateCave(int dx, int dy)
1530{
1531    AddTranslationScreenCoordinates(dx, dy);
1532    m_HitTestGridValid = false;
1533
1534    if (m_here_is_temporary) SetHere();
1535
1536    ForceRefresh();
1537}
1538
1539void GfxCore::DragFinished()
1540{
1541    m_MouseOutsideCompass = m_MouseOutsideElev = false;
1542    ForceRefresh();
1543}
1544
1545void GfxCore::ClearCoords()
1546{
1547    m_Parent->ClearCoords();
1548}
1549
1550void GfxCore::SetCoords(wxPoint point)
1551{
1552    // We can't work out 2D coordinates from a perspective view, and it
1553    // doesn't really make sense to show coordinates while we're animating.
1554    if (GetPerspective() || Animating()) return;
1555
1556    // Update the coordinate or altitude display, given the (x, y) position in
1557    // window coordinates.  The relevant display is updated depending on
1558    // whether we're in plan or elevation view.
1559
1560    double cx, cy, cz;
1561
1562    SetDataTransform();
1563    ReverseTransform(point.x, GetYSize() - 1 - point.y, &cx, &cy, &cz);
1564
1565    if (ShowingPlan()) {
1566        m_Parent->SetCoords(cx + m_Parent->GetOffset().GetX(),
1567                            cy + m_Parent->GetOffset().GetY());
1568    } else if (ShowingElevation()) {
1569        m_Parent->SetAltitude(cz + m_Parent->GetOffset().GetZ());
1570    } else {
1571        m_Parent->ClearCoords();
1572    }
1573}
1574
1575int GfxCore::GetCompassXPosition() const
1576{
1577    // Return the x-coordinate of the centre of the compass in window
1578    // coordinates.
1579    return GetXSize() - INDICATOR_OFFSET_X - INDICATOR_BOX_SIZE / 2;
1580}
1581
1582int GfxCore::GetClinoXPosition() const
1583{
1584    // Return the x-coordinate of the centre of the compass in window
1585    // coordinates.
1586    return GetXSize() - GetClinoOffset() - INDICATOR_BOX_SIZE / 2;
1587}
1588
1589int GfxCore::GetIndicatorYPosition() const
1590{
1591    // Return the y-coordinate of the centre of the indicators in window
1592    // coordinates.
1593    return GetYSize() - INDICATOR_OFFSET_Y - INDICATOR_BOX_SIZE / 2;
1594}
1595
1596int GfxCore::GetIndicatorRadius() const
1597{
1598    // Return the radius of each indicator.
1599    return (INDICATOR_BOX_SIZE - INDICATOR_MARGIN * 2) / 2;
1600}
1601
1602bool GfxCore::PointWithinCompass(wxPoint point) const
1603{
1604    // Determine whether a point (in window coordinates) lies within the
1605    // compass.
1606    if (!ShowingCompass()) return false;
1607
1608    glaCoord dx = point.x - GetCompassXPosition();
1609    glaCoord dy = point.y - GetIndicatorYPosition();
1610    glaCoord radius = GetIndicatorRadius();
1611
1612    return (dx * dx + dy * dy <= radius * radius);
1613}
1614
1615bool GfxCore::PointWithinClino(wxPoint point) const
1616{
1617    // Determine whether a point (in window coordinates) lies within the clino.
1618    if (!ShowingClino()) return false;
1619
1620    glaCoord dx = point.x - GetClinoXPosition();
1621    glaCoord dy = point.y - GetIndicatorYPosition();
1622    glaCoord radius = GetIndicatorRadius();
1623
1624    return (dx * dx + dy * dy <= radius * radius);
1625}
1626
1627bool GfxCore::PointWithinScaleBar(wxPoint point) const
1628{
1629    // Determine whether a point (in window coordinates) lies within the scale
1630    // bar.
1631    if (!ShowingScaleBar()) return false;
1632
1633    return (point.x >= SCALE_BAR_OFFSET_X &&
1634            point.x <= SCALE_BAR_OFFSET_X + m_ScaleBarWidth &&
1635            point.y <= GetYSize() - SCALE_BAR_OFFSET_Y - SCALE_BAR_HEIGHT &&
1636            point.y >= GetYSize() - SCALE_BAR_OFFSET_Y - SCALE_BAR_HEIGHT*2);
1637}
1638
1639bool GfxCore::PointWithinColourKey(wxPoint point) const
1640{
1641    // Determine whether a point (in window coordinates) lies within the key.
1642    point.x -= GetXSize() - KEY_OFFSET_X;
1643    point.y = KEY_OFFSET_Y - point.y;
1644    return (point.x >= key_lowerleft[m_ColourBy].x && point.x <= 0 &&
1645            point.y >= key_lowerleft[m_ColourBy].y && point.y <= 0);
1646}
1647
1648void GfxCore::SetCompassFromPoint(wxPoint point)
1649{
1650    // Given a point in window coordinates, set the heading of the survey.  If
1651    // the point is outside the compass, it snaps to 45 degree intervals;
1652    // otherwise it operates as normal.
1653
1654    wxCoord dx = point.x - GetCompassXPosition();
1655    wxCoord dy = point.y - GetIndicatorYPosition();
1656    wxCoord radius = GetIndicatorRadius();
1657
1658    double angle = deg(atan2(double(dx), double(dy))) - 180.0;
1659    if (dx * dx + dy * dy <= radius * radius) {
1660        TurnCave(angle - m_PanAngle);
1661        m_MouseOutsideCompass = false;
1662    } else {
1663        TurnCave(int(angle / 45.0) * 45.0 - m_PanAngle);
1664        m_MouseOutsideCompass = true;
1665    }
1666
1667    ForceRefresh();
1668}
1669
1670void GfxCore::SetClinoFromPoint(wxPoint point)
1671{
1672    // Given a point in window coordinates, set the elevation of the survey.
1673    // If the point is outside the clino, it snaps to 90 degree intervals;
1674    // otherwise it operates as normal.
1675
1676    glaCoord dx = point.x - GetClinoXPosition();
1677    glaCoord dy = point.y - GetIndicatorYPosition();
1678    glaCoord radius = GetIndicatorRadius();
1679
1680    if (dx >= 0 && dx * dx + dy * dy <= radius * radius) {
1681        TiltCave(-deg(atan2(double(dy), double(dx))) - m_TiltAngle);
1682        m_MouseOutsideElev = false;
1683    } else if (dy >= INDICATOR_MARGIN) {
1684        TiltCave(-90.0 - m_TiltAngle);
1685        m_MouseOutsideElev = true;
1686    } else if (dy <= -INDICATOR_MARGIN) {
1687        TiltCave(90.0 - m_TiltAngle);
1688        m_MouseOutsideElev = true;
1689    } else {
1690        TiltCave(-m_TiltAngle);
1691        m_MouseOutsideElev = true;
1692    }
1693
1694    ForceRefresh();
1695}
1696
1697void GfxCore::SetScaleBarFromOffset(wxCoord dx)
1698{
1699    // Set the scale of the survey, given an offset as to how much the mouse has
1700    // been dragged over the scalebar since the last scale change.
1701
1702    SetScale((m_ScaleBarWidth + dx) * m_Scale / m_ScaleBarWidth);
1703    ForceRefresh();
1704}
1705
1706void GfxCore::RedrawIndicators()
1707{
1708    // Redraw the compass and clino indicators.
1709
1710    RefreshRect(wxRect(GetXSize() - INDICATOR_OFFSET_X - INDICATOR_BOX_SIZE*2 -
1711                       INDICATOR_GAP,
1712                       GetYSize() - INDICATOR_OFFSET_Y - INDICATOR_BOX_SIZE,
1713                       INDICATOR_BOX_SIZE*2 + INDICATOR_GAP,
1714                       INDICATOR_BOX_SIZE), false);
1715}
1716
1717void GfxCore::StartRotation()
1718{
1719    // Start the survey rotating.
1720
1721    m_Rotating = true;
1722    timer.Start(drawtime);
1723}
1724
1725void GfxCore::ToggleRotation()
1726{
1727    // Toggle the survey rotation on/off.
1728
1729    if (m_Rotating) {
1730        StopRotation();
1731    } else {
1732        StartRotation();
1733    }
1734}
1735
1736void GfxCore::StopRotation()
1737{
1738    // Stop the survey rotating.
1739
1740    m_Rotating = false;
1741    ForceRefresh();
1742}
1743
1744bool GfxCore::IsExtendedElevation() const
1745{
1746    return m_Parent->IsExtendedElevation();
1747}
1748
1749void GfxCore::ReverseRotation()
1750{
1751    // Reverse the direction of rotation.
1752
1753    m_RotationStep = -m_RotationStep;
1754}
1755
1756void GfxCore::RotateSlower(bool accel)
1757{
1758    // Decrease the speed of rotation, optionally by an increased amount.
1759
1760    m_RotationStep /= accel ? 1.44 : 1.2;
1761    if (m_RotationStep < 1.0) {
1762        m_RotationStep = 1.0;
1763    }
1764}
1765
1766void GfxCore::RotateFaster(bool accel)
1767{
1768    // Increase the speed of rotation, optionally by an increased amount.
1769
1770    m_RotationStep *= accel ? 1.44 : 1.2;
1771    if (m_RotationStep > 450.0) {
1772        m_RotationStep = 450.0;
1773    }
1774}
1775
1776void GfxCore::SwitchToElevation()
1777{
1778    // Perform an animated switch to elevation view.
1779
1780    switch (m_SwitchingTo) {
1781        case 0:
1782            timer.Start(drawtime);
1783            m_SwitchingTo = ELEVATION;
1784            break;
1785
1786        case PLAN:
1787            m_SwitchingTo = ELEVATION;
1788            break;
1789
1790        case ELEVATION:
1791            // a second order to switch takes us there right away
1792            TiltCave(-m_TiltAngle);
1793            m_SwitchingTo = 0;
1794            ForceRefresh();
1795    }
1796}
1797
1798void GfxCore::SwitchToPlan()
1799{
1800    // Perform an animated switch to plan view.
1801
1802    switch (m_SwitchingTo) {
1803        case 0:
1804            timer.Start(drawtime);
1805            m_SwitchingTo = PLAN;
1806            break;
1807
1808        case ELEVATION:
1809            m_SwitchingTo = PLAN;
1810            break;
1811
1812        case PLAN:
1813            // A second order to switch takes us there right away
1814            TiltCave(-90.0 - m_TiltAngle);
1815            m_SwitchingTo = 0;
1816            ForceRefresh();
1817    }
1818}
1819
1820void GfxCore::SetViewTo(Double xmin, Double xmax, Double ymin, Double ymax, Double zmin, Double zmax)
1821{
1822
1823    SetTranslation(-Vector3((xmin + xmax) / 2, (ymin + ymax) / 2, (zmin + zmax) / 2));
1824    Double scale = HUGE_VAL;
1825    const Vector3 ext = m_Parent->GetExtent();
1826    if (xmax > xmin) {
1827        Double s = ext.GetX() / (xmax - xmin);
1828        if (s < scale) scale = s;
1829    }
1830    if (ymax > ymin) {
1831        Double s = ext.GetY() / (ymax - ymin);
1832        if (s < scale) scale = s;
1833    }
1834    if (!ShowingPlan() && zmax > zmin) {
1835        Double s = ext.GetZ() / (zmax - zmin);
1836        if (s < scale) scale = s;
1837    }
1838    if (scale != HUGE_VAL) SetScale(scale);
1839    ForceRefresh();
1840}
1841
1842bool GfxCore::CanRaiseViewpoint() const
1843{
1844    // Determine if the survey can be viewed from a higher angle of elevation.
1845
1846    return GetPerspective() ? (m_TiltAngle < 90.0) : (m_TiltAngle > -90.0);
1847}
1848
1849bool GfxCore::CanLowerViewpoint() const
1850{
1851    // Determine if the survey can be viewed from a lower angle of elevation.
1852
1853    return GetPerspective() ? (m_TiltAngle > -90.0) : (m_TiltAngle < 90.0);
1854}
1855
1856bool GfxCore::HasDepth() const
1857{
1858    return m_Parent->GetDepthExtent() == 0.0;
1859}
1860
1861bool GfxCore::HasErrorInformation() const
1862{
1863    return m_Parent->HasErrorInformation();
1864}
1865
1866bool GfxCore::HasDateInformation() const
1867{
1868    return m_Parent->GetDateMin() >= 0;
1869}
1870
1871bool GfxCore::ShowingPlan() const
1872{
1873    // Determine if the survey is in plan view.
1874
1875    return (m_TiltAngle == -90.0);
1876}
1877
1878bool GfxCore::ShowingElevation() const
1879{
1880    // Determine if the survey is in elevation view.
1881
1882    return (m_TiltAngle == 0.0);
1883}
1884
1885bool GfxCore::ShowingMeasuringLine() const
1886{
1887    // Determine if the measuring line is being shown.
1888
1889    return (m_there.IsValid() && m_here.IsValid());
1890}
1891
1892void GfxCore::ToggleFlag(bool* flag, int update)
1893{
1894    *flag = !*flag;
1895    if (update == UPDATE_BLOBS) {
1896        UpdateBlobs();
1897    } else if (update == UPDATE_BLOBS_AND_CROSSES) {
1898        UpdateBlobs();
1899        InvalidateList(LIST_CROSSES);
1900    }
1901    ForceRefresh();
1902}
1903
1904int GfxCore::GetNumEntrances() const
1905{
1906    return m_Parent->GetNumEntrances();
1907}
1908
1909int GfxCore::GetNumFixedPts() const
1910{
1911    return m_Parent->GetNumFixedPts();
1912}
1913
1914int GfxCore::GetNumExportedPts() const
1915{
1916    return m_Parent->GetNumExportedPts();
1917}
1918
1919void GfxCore::ClearTreeSelection()
1920{
1921    m_Parent->ClearTreeSelection();
1922}
1923
1924void GfxCore::CentreOn(const Point &p)
1925{
1926    SetTranslation(-p);
1927    m_HitTestGridValid = false;
1928
1929    ForceRefresh();
1930}
1931
1932void GfxCore::ForceRefresh()
1933{
1934    Refresh(false);
1935}
1936
1937void GfxCore::GenerateList(unsigned int l)
1938{
1939    assert(m_HaveData);
1940
1941    switch (l) {
1942        case LIST_COMPASS:
1943            DrawCompass();
1944            break;
1945        case LIST_CLINO:
1946            DrawClino();
1947            break;
1948        case LIST_CLINO_BACK:
1949            DrawClinoBack();
1950            break;
1951        case LIST_SCALE_BAR:
1952            DrawScaleBar();
1953            break;
1954        case LIST_DEPTH_KEY:
1955            DrawDepthKey();
1956            break;
1957        case LIST_DATE_KEY:
1958            DrawDateKey();
1959            break;
1960        case LIST_ERROR_KEY:
1961            DrawErrorKey();
1962            break;
1963        case LIST_UNDERGROUND_LEGS:
1964            GenerateDisplayList();
1965            break;
1966        case LIST_TUBES:
1967            GenerateDisplayListTubes();
1968            break;
1969        case LIST_SURFACE_LEGS:
1970            GenerateDisplayListSurface();
1971            break;
1972        case LIST_BLOBS:
1973            GenerateBlobsDisplayList();
1974            break;
1975        case LIST_CROSSES: {
1976            BeginCrosses();
1977            SetColour(col_LIGHT_GREY);
1978            list<LabelInfo*>::const_iterator pos = m_Parent->GetLabels();
1979            while (pos != m_Parent->GetLabelsEnd()) {
1980                const LabelInfo* label = *pos++;
1981
1982                if ((m_Surface && label->IsSurface()) ||
1983                    (m_Legs && label->IsUnderground()) ||
1984                    (!label->IsSurface() && !label->IsUnderground())) {
1985                    // Check if this station should be displayed
1986                    // (last case is for stns with no legs attached)
1987                    DrawCross(label->GetX(), label->GetY(), label->GetZ());
1988                }
1989            }
1990            EndCrosses();
1991            break;
1992        }
1993        case LIST_GRID:
1994            DrawGrid();
1995            break;
1996        case LIST_SHADOW:
1997            GenerateDisplayListShadow();
1998            break;
1999        default:
2000            assert(false);
2001            break;
2002    }
2003}
2004
2005void GfxCore::ToggleSmoothShading()
2006{
2007    GLACanvas::ToggleSmoothShading();
2008    InvalidateList(LIST_TUBES);
2009    ForceRefresh();
2010}
2011
2012void GfxCore::GenerateDisplayList()
2013{
2014    // Generate the display list for the underground legs.
2015    list<traverse>::const_iterator trav = m_Parent->traverses_begin();
2016    list<traverse>::const_iterator tend = m_Parent->traverses_end();
2017
2018    if (m_Splays == SPLAYS_SHOW_FADED) {
2019        SetAlpha(0.4);
2020        while (trav != tend) {
2021            if ((*trav).isSplay)
2022                (this->*AddPoly)(*trav);
2023            ++trav;
2024        }
2025        SetAlpha(1.0);
2026        trav = m_Parent->traverses_begin();
2027    }
2028
2029    while (trav != tend) {
2030        if (m_Splays == SPLAYS_SHOW_NORMAL || !(*trav).isSplay)
2031            (this->*AddPoly)(*trav);
2032        ++trav;
2033    }
2034}
2035
2036void GfxCore::GenerateDisplayListTubes()
2037{
2038    // Generate the display list for the tubes.
2039    list<vector<XSect> >::iterator trav = m_Parent->tubes_begin();
2040    list<vector<XSect> >::iterator tend = m_Parent->tubes_end();
2041    while (trav != tend) {
2042        SkinPassage(*trav);
2043        ++trav;
2044    }
2045}
2046
2047void GfxCore::GenerateDisplayListSurface()
2048{
2049    // Generate the display list for the surface legs.
2050    EnableDashedLines();
2051    list<traverse>::const_iterator trav = m_Parent->surface_traverses_begin();
2052    list<traverse>::const_iterator tend = m_Parent->surface_traverses_end();
2053    while (trav != tend) {
2054        if (m_ColourBy == COLOUR_BY_ERROR) {
2055            AddPolylineError(*trav);
2056        } else {
2057            AddPolyline(*trav);
2058        }
2059        ++trav;
2060    }
2061    DisableDashedLines();
2062}
2063
2064void GfxCore::GenerateDisplayListShadow()
2065{
2066    SetColour(col_BLACK);
2067    list<traverse>::const_iterator trav = m_Parent->traverses_begin();
2068    list<traverse>::const_iterator tend = m_Parent->traverses_end();
2069    while (trav != tend) {
2070        AddPolylineShadow(*trav);
2071        ++trav;
2072    }
2073}
2074
2075// Plot blobs.
2076void GfxCore::GenerateBlobsDisplayList()
2077{
2078    if (!(m_Entrances || m_FixedPts || m_ExportedPts ||
2079          m_Parent->GetNumHighlightedPts()))
2080        return;
2081
2082    // Plot blobs.
2083    gla_colour prev_col = col_BLACK; // not a colour used for blobs
2084    list<LabelInfo*>::const_iterator pos = m_Parent->GetLabels();
2085    BeginBlobs();
2086    while (pos != m_Parent->GetLabelsEnd()) {
2087        const LabelInfo* label = *pos++;
2088
2089        // When more than one flag is set on a point:
2090        // search results take priority over entrance highlighting
2091        // which takes priority over fixed point
2092        // highlighting, which in turn takes priority over exported
2093        // point highlighting.
2094
2095        if (!((m_Surface && label->IsSurface()) ||
2096              (m_Legs && label->IsUnderground()) ||
2097              (!label->IsSurface() && !label->IsUnderground()))) {
2098            // if this station isn't to be displayed, skip to the next
2099            // (last case is for stns with no legs attached)
2100            continue;
2101        }
2102
2103        gla_colour col;
2104
2105        if (label->IsHighLighted()) {
2106            col = col_YELLOW;
2107        } else if (m_Entrances && label->IsEntrance()) {
2108            col = col_GREEN;
2109        } else if (m_FixedPts && label->IsFixedPt()) {
2110            col = col_RED;
2111        } else if (m_ExportedPts && label->IsExportedPt()) {
2112            col = col_TURQUOISE;
2113        } else {
2114            continue;
2115        }
2116
2117        // Stations are sorted by blob type, so colour changes are infrequent.
2118        if (col != prev_col) {
2119            SetColour(col);
2120            prev_col = col;
2121        }
2122        DrawBlob(label->GetX(), label->GetY(), label->GetZ());
2123    }
2124    EndBlobs();
2125}
2126
2127void GfxCore::DrawIndicators()
2128{
2129    // Draw colour key.
2130    if (m_ColourKey) {
2131        if (m_ColourBy == COLOUR_BY_DEPTH) {
2132            DrawList2D(LIST_DEPTH_KEY, GetXSize() - KEY_OFFSET_X,
2133                       GetYSize() - KEY_OFFSET_Y, 0);
2134        } else if (m_ColourBy == COLOUR_BY_DATE) {
2135            DrawList2D(LIST_DATE_KEY, GetXSize() - KEY_OFFSET_X,
2136                       GetYSize() - KEY_OFFSET_Y, 0);
2137        } else if (m_ColourBy == COLOUR_BY_ERROR) {
2138            DrawList2D(LIST_ERROR_KEY, GetXSize() - KEY_OFFSET_X,
2139                       GetYSize() - KEY_OFFSET_Y, 0);
2140        }
2141    }
2142
2143    // Draw compass or elevation/heading indicators.
2144    if (m_Compass || m_Clino) {
2145        if (!m_Parent->IsExtendedElevation()) Draw2dIndicators();
2146    }
2147
2148    // Draw scalebar.
2149    if (m_Scalebar) {
2150        DrawList2D(LIST_SCALE_BAR, 0, 0, 0);
2151    }
2152}
2153
2154void GfxCore::PlaceVertexWithColour(const Vector3 & v,
2155                                    glaTexCoord tex_x, glaTexCoord tex_y,
2156                                    Double factor)
2157{
2158    SetColour(GetSurfacePen(), factor); // FIXME : assumes surface pen is white!
2159    PlaceVertex(v, tex_x, tex_y);
2160}
2161
2162void GfxCore::SetDepthColour(Double z, Double factor) {
2163    // Set the drawing colour based on the altitude.
2164    Double z_ext = m_Parent->GetDepthExtent();
2165
2166    z -= m_Parent->GetDepthMin();
2167    // points arising from tubes may be slightly outside the limits...
2168    if (z < 0) z = 0;
2169    if (z > z_ext) z = z_ext;
2170
2171    if (z == 0) {
2172        SetColour(GetPen(0), factor);
2173        return;
2174    }
2175
2176    assert(z_ext > 0.0);
2177    Double how_far = z / z_ext;
2178    assert(how_far >= 0.0);
2179    assert(how_far <= 1.0);
2180
2181    int band = int(floor(how_far * (GetNumColourBands() - 1)));
2182    GLAPen pen1 = GetPen(band);
2183    if (band < GetNumColourBands() - 1) {
2184        const GLAPen& pen2 = GetPen(band + 1);
2185
2186        Double interval = z_ext / (GetNumColourBands() - 1);
2187        Double into_band = z / interval - band;
2188
2189//      printf("%g z_offset=%g interval=%g band=%d\n", into_band,
2190//             z_offset, interval, band);
2191        // FIXME: why do we need to clamp here?  Is it because the walls can
2192        // extend further up/down than the centre-line?
2193        if (into_band < 0.0) into_band = 0.0;
2194        if (into_band > 1.0) into_band = 1.0;
2195        assert(into_band >= 0.0);
2196        assert(into_band <= 1.0);
2197
2198        pen1.Interpolate(pen2, into_band);
2199    }
2200    SetColour(pen1, factor);
2201}
2202
2203void GfxCore::PlaceVertexWithDepthColour(const Vector3 &v, Double factor)
2204{
2205    SetDepthColour(v.GetZ(), factor);
2206    PlaceVertex(v);
2207}
2208
2209void GfxCore::PlaceVertexWithDepthColour(const Vector3 &v,
2210                                         glaTexCoord tex_x, glaTexCoord tex_y,
2211                                         Double factor)
2212{
2213    SetDepthColour(v.GetZ(), factor);
2214    PlaceVertex(v, tex_x, tex_y);
2215}
2216
2217void GfxCore::SplitLineAcrossBands(int band, int band2,
2218                                   const Vector3 &p, const Vector3 &q,
2219                                   Double factor)
2220{
2221    const int step = (band < band2) ? 1 : -1;
2222    for (int i = band; i != band2; i += step) {
2223        const Double z = GetDepthBoundaryBetweenBands(i, i + step);
2224
2225        // Find the intersection point of the line p -> q
2226        // with the plane parallel to the xy-plane with z-axis intersection z.
2227        assert(q.GetZ() - p.GetZ() != 0.0);
2228
2229        const Double t = (z - p.GetZ()) / (q.GetZ() - p.GetZ());
2230//      assert(0.0 <= t && t <= 1.0);           FIXME: rounding problems!
2231
2232        const Double x = p.GetX() + t * (q.GetX() - p.GetX());
2233        const Double y = p.GetY() + t * (q.GetY() - p.GetY());
2234
2235        PlaceVertexWithDepthColour(Vector3(x, y, z), factor);
2236    }
2237}
2238
2239int GfxCore::GetDepthColour(Double z) const
2240{
2241    // Return the (0-based) depth colour band index for a z-coordinate.
2242    Double z_ext = m_Parent->GetDepthExtent();
2243    z -= m_Parent->GetDepthMin();
2244    // We seem to get rounding differences causing z to sometimes be slightly
2245    // less than GetDepthMin() here, and it can certainly be true for passage
2246    // tubes, so just clamp the value to 0.
2247    if (z <= 0) return 0;
2248    assert(z_ext > 0.0);
2249    // We seem to get rounding differences causing z to sometimes exceed z_ext
2250    // by a small amount here (see: http://trac.survex.com/ticket/26) and it
2251    // can certainly be true for passage tubes, so just clamp the value.
2252    if (z >= z_ext) return GetNumColourBands() - 1;
2253    return int(z / z_ext * (GetNumColourBands() - 1));
2254}
2255
2256Double GfxCore::GetDepthBoundaryBetweenBands(int a, int b) const
2257{
2258    // Return the z-coordinate of the depth colour boundary between
2259    // two adjacent depth colour bands (specified by 0-based indices).
2260
2261    assert((a == b - 1) || (a == b + 1));
2262    if (GetNumColourBands() == 1) return 0;
2263
2264    int band = (a > b) ? a : b; // boundary N lies on the bottom of band N.
2265    Double z_ext = m_Parent->GetDepthExtent();
2266    return (z_ext * band / (GetNumColourBands() - 1)) + m_Parent->GetDepthMin();
2267}
2268
2269void GfxCore::AddPolyline(const traverse & centreline)
2270{
2271    BeginPolyline();
2272    SetColour(GetSurfacePen());
2273    vector<PointInfo>::const_iterator i = centreline.begin();
2274    PlaceVertex(*i);
2275    ++i;
2276    while (i != centreline.end()) {
2277        PlaceVertex(*i);
2278        ++i;
2279    }
2280    EndPolyline();
2281}
2282
2283void GfxCore::AddPolylineShadow(const traverse & centreline)
2284{
2285    BeginPolyline();
2286    const double z = -0.5 * m_Parent->GetZExtent();
2287    vector<PointInfo>::const_iterator i = centreline.begin();
2288    PlaceVertex(i->GetX(), i->GetY(), z);
2289    ++i;
2290    while (i != centreline.end()) {
2291        PlaceVertex(i->GetX(), i->GetY(), z);
2292        ++i;
2293    }
2294    EndPolyline();
2295}
2296
2297void GfxCore::AddPolylineDepth(const traverse & centreline)
2298{
2299    BeginPolyline();
2300    vector<PointInfo>::const_iterator i, prev_i;
2301    i = centreline.begin();
2302    int band0 = GetDepthColour(i->GetZ());
2303    PlaceVertexWithDepthColour(*i);
2304    prev_i = i;
2305    ++i;
2306    while (i != centreline.end()) {
2307        int band = GetDepthColour(i->GetZ());
2308        if (band != band0) {
2309            SplitLineAcrossBands(band0, band, *prev_i, *i);
2310            band0 = band;
2311        }
2312        PlaceVertexWithDepthColour(*i);
2313        prev_i = i;
2314        ++i;
2315    }
2316    EndPolyline();
2317}
2318
2319void GfxCore::AddQuadrilateral(const Vector3 &a, const Vector3 &b,
2320                               const Vector3 &c, const Vector3 &d)
2321{
2322    Vector3 normal = (a - c) * (d - b);
2323    normal.normalise();
2324    Double factor = dot(normal, light) * .3 + .7;
2325    glaTexCoord w(ceil(((b - a).magnitude() + (d - c).magnitude()) * .5));
2326    glaTexCoord h(ceil(((b - c).magnitude() + (d - a).magnitude()) * .5));
2327    // FIXME: should plot triangles instead to avoid rendering glitches.
2328    BeginQuadrilaterals();
2329    PlaceVertexWithColour(a, 0, 0, factor);
2330    PlaceVertexWithColour(b, w, 0, factor);
2331    PlaceVertexWithColour(c, w, h, factor);
2332    PlaceVertexWithColour(d, 0, h, factor);
2333    EndQuadrilaterals();
2334}
2335
2336void GfxCore::AddQuadrilateralDepth(const Vector3 &a, const Vector3 &b,
2337                                    const Vector3 &c, const Vector3 &d)
2338{
2339    Vector3 normal = (a - c) * (d - b);
2340    normal.normalise();
2341    Double factor = dot(normal, light) * .3 + .7;
2342    int a_band, b_band, c_band, d_band;
2343    a_band = GetDepthColour(a.GetZ());
2344    a_band = min(max(a_band, 0), GetNumColourBands());
2345    b_band = GetDepthColour(b.GetZ());
2346    b_band = min(max(b_band, 0), GetNumColourBands());
2347    c_band = GetDepthColour(c.GetZ());
2348    c_band = min(max(c_band, 0), GetNumColourBands());
2349    d_band = GetDepthColour(d.GetZ());
2350    d_band = min(max(d_band, 0), GetNumColourBands());
2351    // All this splitting is incorrect - we need to make a separate polygon
2352    // for each depth band...
2353    glaTexCoord w(ceil(((b - a).magnitude() + (d - c).magnitude()) * .5));
2354    glaTexCoord h(ceil(((b - c).magnitude() + (d - a).magnitude()) * .5));
2355    BeginPolygon();
2356////    PlaceNormal(normal);
2357    PlaceVertexWithDepthColour(a, 0, 0, factor);
2358    if (a_band != b_band) {
2359        SplitLineAcrossBands(a_band, b_band, a, b, factor);
2360    }
2361    PlaceVertexWithDepthColour(b, w, 0, factor);
2362    if (b_band != c_band) {
2363        SplitLineAcrossBands(b_band, c_band, b, c, factor);
2364    }
2365    PlaceVertexWithDepthColour(c, w, h, factor);
2366    if (c_band != d_band) {
2367        SplitLineAcrossBands(c_band, d_band, c, d, factor);
2368    }
2369    PlaceVertexWithDepthColour(d, 0, h, factor);
2370    if (d_band != a_band) {
2371        SplitLineAcrossBands(d_band, a_band, d, a, factor);
2372    }
2373    EndPolygon();
2374}
2375
2376void GfxCore::SetColourFromDate(int date, Double factor)
2377{
2378    // Set the drawing colour based on a date.
2379
2380    if (date == -1) {
2381        // Undated.
2382        SetColour(GetSurfacePen(), factor);
2383        return;
2384    }
2385
2386    int date_offset = date - m_Parent->GetDateMin();
2387    if (date_offset == 0) {
2388        // Earliest date - handle as a special case for the single date case.
2389        SetColour(GetPen(0), factor);
2390        return;
2391    }
2392
2393    int date_ext = m_Parent->GetDateExtent();
2394    Double how_far = (Double)date_offset / date_ext;
2395    assert(how_far >= 0.0);
2396    assert(how_far <= 1.0);
2397
2398    int band = int(floor(how_far * (GetNumColourBands() - 1)));
2399    GLAPen pen1 = GetPen(band);
2400    if (band < GetNumColourBands() - 1) {
2401        const GLAPen& pen2 = GetPen(band + 1);
2402
2403        Double interval = date_ext / (GetNumColourBands() - 1);
2404        Double into_band = date_offset / interval - band;
2405
2406//      printf("%g z_offset=%g interval=%g band=%d\n", into_band,
2407//             z_offset, interval, band);
2408        // FIXME: why do we need to clamp here?  Is it because the walls can
2409        // extend further up/down than the centre-line?
2410        if (into_band < 0.0) into_band = 0.0;
2411        if (into_band > 1.0) into_band = 1.0;
2412        assert(into_band >= 0.0);
2413        assert(into_band <= 1.0);
2414
2415        pen1.Interpolate(pen2, into_band);
2416    }
2417    SetColour(pen1, factor);
2418}
2419
2420void GfxCore::AddPolylineDate(const traverse & centreline)
2421{
2422    BeginPolyline();
2423    vector<PointInfo>::const_iterator i, prev_i;
2424    i = centreline.begin();
2425    int date = i->GetDate();
2426    SetColourFromDate(date, 1.0);
2427    PlaceVertex(*i);
2428    prev_i = i;
2429    while (++i != centreline.end()) {
2430        int newdate = i->GetDate();
2431        if (newdate != date) {
2432            EndPolyline();
2433            BeginPolyline();
2434            date = newdate;
2435            SetColourFromDate(date, 1.0);
2436            PlaceVertex(*prev_i);
2437        }
2438        PlaceVertex(*i);
2439        prev_i = i;
2440    }
2441    EndPolyline();
2442}
2443
2444static int static_date_hack; // FIXME
2445
2446void GfxCore::AddQuadrilateralDate(const Vector3 &a, const Vector3 &b,
2447                                   const Vector3 &c, const Vector3 &d)
2448{
2449    Vector3 normal = (a - c) * (d - b);
2450    normal.normalise();
2451    Double factor = dot(normal, light) * .3 + .7;
2452    int w = int(ceil(((b - a).magnitude() + (d - c).magnitude()) / 2));
2453    int h = int(ceil(((b - c).magnitude() + (d - a).magnitude()) / 2));
2454    // FIXME: should plot triangles instead to avoid rendering glitches.
2455    BeginQuadrilaterals();
2456////    PlaceNormal(normal);
2457    SetColourFromDate(static_date_hack, factor);
2458    PlaceVertex(a, 0, 0);
2459    PlaceVertex(b, w, 0);
2460    PlaceVertex(c, w, h);
2461    PlaceVertex(d, 0, h);
2462    EndQuadrilaterals();
2463}
2464
2465static double static_E_hack; // FIXME
2466
2467void GfxCore::SetColourFromError(double E, Double factor)
2468{
2469    // Set the drawing colour based on an error value.
2470
2471    if (E < 0) {
2472        SetColour(GetSurfacePen(), factor);
2473        return;
2474    }
2475
2476    Double how_far = E / MAX_ERROR;
2477    assert(how_far >= 0.0);
2478    if (how_far > 1.0) how_far = 1.0;
2479
2480    int band = int(floor(how_far * (GetNumColourBands() - 1)));
2481    GLAPen pen1 = GetPen(band);
2482    if (band < GetNumColourBands() - 1) {
2483        const GLAPen& pen2 = GetPen(band + 1);
2484
2485        Double interval = MAX_ERROR / (GetNumColourBands() - 1);
2486        Double into_band = E / interval - band;
2487
2488//      printf("%g z_offset=%g interval=%g band=%d\n", into_band,
2489//             z_offset, interval, band);
2490        // FIXME: why do we need to clamp here?  Is it because the walls can
2491        // extend further up/down than the centre-line?
2492        if (into_band < 0.0) into_band = 0.0;
2493        if (into_band > 1.0) into_band = 1.0;
2494        assert(into_band >= 0.0);
2495        assert(into_band <= 1.0);
2496
2497        pen1.Interpolate(pen2, into_band);
2498    }
2499    SetColour(pen1, factor);
2500}
2501
2502void GfxCore::AddQuadrilateralError(const Vector3 &a, const Vector3 &b,
2503                                    const Vector3 &c, const Vector3 &d)
2504{
2505    Vector3 normal = (a - c) * (d - b);
2506    normal.normalise();
2507    Double factor = dot(normal, light) * .3 + .7;
2508    int w = int(ceil(((b - a).magnitude() + (d - c).magnitude()) / 2));
2509    int h = int(ceil(((b - c).magnitude() + (d - a).magnitude()) / 2));
2510    // FIXME: should plot triangles instead to avoid rendering glitches.
2511    BeginQuadrilaterals();
2512////    PlaceNormal(normal);
2513    SetColourFromError(static_E_hack, factor);
2514    PlaceVertex(a, 0, 0);
2515    PlaceVertex(b, w, 0);
2516    PlaceVertex(c, w, h);
2517    PlaceVertex(d, 0, h);
2518    EndQuadrilaterals();
2519}
2520
2521void GfxCore::AddPolylineError(const traverse & centreline)
2522{
2523    BeginPolyline();
2524    SetColourFromError(centreline.E, 1.0);
2525    vector<PointInfo>::const_iterator i;
2526    for(i = centreline.begin(); i != centreline.end(); ++i) {
2527        PlaceVertex(*i);
2528    }
2529    EndPolyline();
2530}
2531
2532void
2533GfxCore::SkinPassage(vector<XSect> & centreline, bool draw)
2534{
2535    assert(centreline.size() > 1);
2536    Vector3 U[4];
2537    XSect prev_pt_v;
2538    Vector3 last_right(1.0, 0.0, 0.0);
2539
2540//  FIXME: it's not simple to set the colour of a tube based on error...
2541//    static_E_hack = something...
2542    vector<XSect>::iterator i = centreline.begin();
2543    vector<XSect>::size_type segment = 0;
2544    while (i != centreline.end()) {
2545        // get the coordinates of this vertex
2546        XSect & pt_v = *i++;
2547
2548        double z_pitch_adjust = 0.0;
2549        bool cover_end = false;
2550
2551        Vector3 right, up;
2552
2553        const Vector3 up_v(0.0, 0.0, 1.0);
2554
2555        if (segment == 0) {
2556            assert(i != centreline.end());
2557            // first segment
2558
2559            // get the coordinates of the next vertex
2560            const XSect & next_pt_v = *i;
2561
2562            // calculate vector from this pt to the next one
2563            Vector3 leg_v = next_pt_v - pt_v;
2564
2565            // obtain a vector in the LRUD plane
2566            right = leg_v * up_v;
2567            if (right.magnitude() == 0) {
2568                right = last_right;
2569                // Obtain a second vector in the LRUD plane,
2570                // perpendicular to the first.
2571                //up = right * leg_v;
2572                up = up_v;
2573            } else {
2574                last_right = right;
2575                up = up_v;
2576            }
2577
2578            cover_end = true;
2579            static_date_hack = next_pt_v.GetDate();
2580        } else if (segment + 1 == centreline.size()) {
2581            // last segment
2582
2583            // Calculate vector from the previous pt to this one.
2584            Vector3 leg_v = pt_v - prev_pt_v;
2585
2586            // Obtain a horizontal vector in the LRUD plane.
2587            right = leg_v * up_v;
2588            if (right.magnitude() == 0) {
2589                right = Vector3(last_right.GetX(), last_right.GetY(), 0.0);
2590                // Obtain a second vector in the LRUD plane,
2591                // perpendicular to the first.
2592                //up = right * leg_v;
2593                up = up_v;
2594            } else {
2595                last_right = right;
2596                up = up_v;
2597            }
2598
2599            cover_end = true;
2600            static_date_hack = pt_v.GetDate();
2601        } else {
2602            assert(i != centreline.end());
2603            // Intermediate segment.
2604
2605            // Get the coordinates of the next vertex.
2606            const XSect & next_pt_v = *i;
2607
2608            // Calculate vectors from this vertex to the
2609            // next vertex, and from the previous vertex to
2610            // this one.
2611            Vector3 leg1_v = pt_v - prev_pt_v;
2612            Vector3 leg2_v = next_pt_v - pt_v;
2613
2614            // Obtain horizontal vectors perpendicular to
2615            // both legs, then normalise and average to get
2616            // a horizontal bisector.
2617            Vector3 r1 = leg1_v * up_v;
2618            Vector3 r2 = leg2_v * up_v;
2619            r1.normalise();
2620            r2.normalise();
2621            right = r1 + r2;
2622            if (right.magnitude() == 0) {
2623                // This is the "mid-pitch" case...
2624                right = last_right;
2625            }
2626            if (r1.magnitude() == 0) {
2627                Vector3 n = leg1_v;
2628                n.normalise();
2629                z_pitch_adjust = n.GetZ();
2630                //up = Vector3(0, 0, leg1_v.GetZ());
2631                //up = right * up;
2632                up = up_v;
2633
2634                // Rotate pitch section to minimise the
2635                // "tortional stress" - FIXME: use
2636                // triangles instead of rectangles?
2637                int shift = 0;
2638                Double maxdotp = 0;
2639
2640                // Scale to unit vectors in the LRUD plane.
2641                right.normalise();
2642                up.normalise();
2643                Vector3 vec = up - right;
2644                for (int orient = 0; orient <= 3; ++orient) {
2645                    Vector3 tmp = U[orient] - prev_pt_v;
2646                    tmp.normalise();
2647                    Double dotp = dot(vec, tmp);
2648                    if (dotp > maxdotp) {
2649                        maxdotp = dotp;
2650                        shift = orient;
2651                    }
2652                }
2653                if (shift) {
2654                    if (shift != 2) {
2655                        Vector3 temp(U[0]);
2656                        U[0] = U[shift];
2657                        U[shift] = U[2];
2658                        U[2] = U[shift ^ 2];
2659                        U[shift ^ 2] = temp;
2660                    } else {
2661                        swap(U[0], U[2]);
2662                        swap(U[1], U[3]);
2663                    }
2664                }
2665#if 0
2666                // Check that the above code actually permuted
2667                // the vertices correctly.
2668                shift = 0;
2669                maxdotp = 0;
2670                for (int j = 0; j <= 3; ++j) {
2671                    Vector3 tmp = U[j] - prev_pt_v;
2672                    tmp.normalise();
2673                    Double dotp = dot(vec, tmp);
2674                    if (dotp > maxdotp) {
2675                        maxdotp = dotp + 1e-6; // Add small tolerance to stop 45 degree offset cases being flagged...
2676                        shift = j;
2677                    }
2678                }
2679                if (shift) {
2680                    printf("New shift = %d!\n", shift);
2681                    shift = 0;
2682                    maxdotp = 0;
2683                    for (int j = 0; j <= 3; ++j) {
2684                        Vector3 tmp = U[j] - prev_pt_v;
2685                        tmp.normalise();
2686                        Double dotp = dot(vec, tmp);
2687                        printf("    %d : %.8f\n", j, dotp);
2688                    }
2689                }
2690#endif
2691            } else if (r2.magnitude() == 0) {
2692                Vector3 n = leg2_v;
2693                n.normalise();
2694                z_pitch_adjust = n.GetZ();
2695                //up = Vector3(0, 0, leg2_v.GetZ());
2696                //up = right * up;
2697                up = up_v;
2698            } else {
2699                up = up_v;
2700            }
2701            last_right = right;
2702            static_date_hack = pt_v.GetDate();
2703        }
2704
2705        // Scale to unit vectors in the LRUD plane.
2706        right.normalise();
2707        up.normalise();
2708
2709        if (z_pitch_adjust != 0) up += Vector3(0, 0, fabs(z_pitch_adjust));
2710
2711        Double l = fabs(pt_v.GetL());
2712        Double r = fabs(pt_v.GetR());
2713        Double u = fabs(pt_v.GetU());
2714        Double d = fabs(pt_v.GetD());
2715
2716        // Produce coordinates of the corners of the LRUD "plane".
2717        Vector3 v[4];
2718        v[0] = pt_v - right * l + up * u;
2719        v[1] = pt_v + right * r + up * u;
2720        v[2] = pt_v + right * r - up * d;
2721        v[3] = pt_v - right * l - up * d;
2722
2723        if (draw) {
2724            if (segment > 0) {
2725                (this->*AddQuad)(v[0], v[1], U[1], U[0]);
2726                (this->*AddQuad)(v[2], v[3], U[3], U[2]);
2727                (this->*AddQuad)(v[1], v[2], U[2], U[1]);
2728                (this->*AddQuad)(v[3], v[0], U[0], U[3]);
2729            }
2730
2731            if (cover_end) {
2732                (this->*AddQuad)(v[3], v[2], v[1], v[0]);
2733            }
2734        }
2735
2736        prev_pt_v = pt_v;
2737        U[0] = v[0];
2738        U[1] = v[1];
2739        U[2] = v[2];
2740        U[3] = v[3];
2741
2742        pt_v.set_right_bearing(deg(atan2(right.GetY(), right.GetX())));
2743
2744        ++segment;
2745    }
2746}
2747
2748void GfxCore::FullScreenMode()
2749{
2750    m_Parent->ViewFullScreen();
2751}
2752
2753bool GfxCore::IsFullScreen() const
2754{
2755    return m_Parent->IsFullScreen();
2756}
2757
2758void
2759GfxCore::MoveViewer(double forward, double up, double right)
2760{
2761    double cT = cos(rad(m_TiltAngle));
2762    double sT = sin(rad(m_TiltAngle));
2763    double cP = cos(rad(m_PanAngle));
2764    double sP = sin(rad(m_PanAngle));
2765    Vector3 v_forward(cT * sP, cT * cP, sT);
2766    Vector3 v_up(sT * sP, sT * cP, -cT);
2767    Vector3 v_right(-cP, sP, 0);
2768    assert(fabs(dot(v_forward, v_up)) < 1e-6);
2769    assert(fabs(dot(v_forward, v_right)) < 1e-6);
2770    assert(fabs(dot(v_right, v_up)) < 1e-6);
2771    Vector3 move = v_forward * forward + v_up * up + v_right * right;
2772    AddTranslation(-move);
2773    // Show current position.
2774    m_Parent->SetCoords(m_Parent->GetOffset() - GetTranslation());
2775    ForceRefresh();
2776}
2777
2778PresentationMark GfxCore::GetView() const
2779{
2780    return PresentationMark(GetTranslation() + m_Parent->GetOffset(),
2781                            m_PanAngle, -m_TiltAngle, m_Scale);
2782}
2783
2784void GfxCore::SetView(const PresentationMark & p)
2785{
2786    m_SwitchingTo = 0;
2787    SetTranslation(p - m_Parent->GetOffset());
2788    m_PanAngle = p.angle;
2789    m_TiltAngle = -p.tilt_angle; // FIXME: nasty reversed sense (and above)
2790    SetRotation(m_PanAngle, m_TiltAngle);
2791    SetScale(p.scale);
2792    ForceRefresh();
2793}
2794
2795void GfxCore::PlayPres(double speed, bool change_speed) {
2796    if (!change_speed || presentation_mode == 0) {
2797        if (speed == 0.0) {
2798            presentation_mode = 0;
2799            return;
2800        }
2801        presentation_mode = PLAYING;
2802        next_mark = m_Parent->GetPresMark(MARK_FIRST);
2803        SetView(next_mark);
2804        next_mark_time = 0; // There already!
2805        this_mark_total = 0;
2806        pres_reverse = (speed < 0);
2807    }
2808
2809    if (change_speed) pres_speed = speed;
2810
2811    if (speed != 0.0) {
2812        bool new_pres_reverse = (speed < 0);
2813        if (new_pres_reverse != pres_reverse) {
2814            pres_reverse = new_pres_reverse;
2815            if (pres_reverse) {
2816                next_mark = m_Parent->GetPresMark(MARK_PREV);
2817            } else {
2818                next_mark = m_Parent->GetPresMark(MARK_NEXT);
2819            }
2820            swap(this_mark_total, next_mark_time);
2821        }
2822    }
2823}
2824
2825void GfxCore::SetColourBy(int colour_by) {
2826    m_ColourBy = colour_by;
2827    switch (colour_by) {
2828        case COLOUR_BY_DEPTH:
2829            AddQuad = &GfxCore::AddQuadrilateralDepth;
2830            AddPoly = &GfxCore::AddPolylineDepth;
2831            break;
2832        case COLOUR_BY_DATE:
2833            AddQuad = &GfxCore::AddQuadrilateralDate;
2834            AddPoly = &GfxCore::AddPolylineDate;
2835            break;
2836        case COLOUR_BY_ERROR:
2837            AddQuad = &GfxCore::AddQuadrilateralError;
2838            AddPoly = &GfxCore::AddPolylineError;
2839            break;
2840        default: // case COLOUR_BY_NONE:
2841            AddQuad = &GfxCore::AddQuadrilateral;
2842            AddPoly = &GfxCore::AddPolyline;
2843            break;
2844    }
2845
2846    InvalidateList(LIST_UNDERGROUND_LEGS);
2847    InvalidateList(LIST_SURFACE_LEGS);
2848    InvalidateList(LIST_TUBES);
2849
2850    ForceRefresh();
2851}
2852
2853bool GfxCore::ExportMovie(const wxString & fnm)
2854{
2855    int width;
2856    int height;
2857    GetSize(&width, &height);
2858    // Round up to next multiple of 2 (required by ffmpeg).
2859    width += (width & 1);
2860    height += (height & 1);
2861
2862    movie = new MovieMaker();
2863
2864    // FIXME: This should really use fn_str() - currently we probably can't
2865    // save to a Unicode path on wxmsw.
2866    if (!movie->Open(fnm.mb_str(), width, height)) {
2867        wxGetApp().ReportError(wxString(movie->get_error_string(), wxConvUTF8));
2868        delete movie;
2869        movie = NULL;
2870        return false;
2871    }
2872
2873    PlayPres(1);
2874    return true;
2875}
2876
2877void
2878GfxCore::OnPrint(const wxString &filename, const wxString &title,
2879                 const wxString &datestamp, time_t datestamp_numeric)
2880{
2881    svxPrintDlg * p;
2882    p = new svxPrintDlg(m_Parent, filename, title,
2883                        datestamp, datestamp_numeric,
2884                        m_PanAngle, m_TiltAngle,
2885                        m_Names, m_Crosses, m_Legs, m_Surface, m_Tubes,
2886                        m_Entrances, m_FixedPts, m_ExportedPts,
2887                        true);
2888    p->Show(true);
2889}
2890
2891void
2892GfxCore::OnExport(const wxString &filename, const wxString &title,
2893                  const wxString &datestamp, time_t datestamp_numeric)
2894{
2895    // Fill in "right_bearing" for each cross-section.
2896    list<vector<XSect> >::iterator trav = m_Parent->tubes_begin();
2897    list<vector<XSect> >::iterator tend = m_Parent->tubes_end();
2898    while (trav != tend) {
2899        SkinPassage(*trav, false);
2900        ++trav;
2901    }
2902
2903    svxPrintDlg * p;
2904    p = new svxPrintDlg(m_Parent, filename, title,
2905                        datestamp, datestamp_numeric,
2906                        m_PanAngle, m_TiltAngle,
2907                        m_Names, m_Crosses, m_Legs, m_Surface, m_Tubes,
2908                        m_Entrances, m_FixedPts, m_ExportedPts,
2909                        false);
2910    p->Show(true);
2911}
2912
2913static wxCursor
2914make_cursor(const unsigned char * bits, const unsigned char * mask,
2915            int hotx, int hoty)
2916{
2917#if defined __WXMSW__ || defined __WXMAC__
2918    wxBitmap cursor_bitmap(reinterpret_cast<const char *>(bits), 32, 32);
2919    wxBitmap mask_bitmap(reinterpret_cast<const char *>(mask), 32, 32);
2920    cursor_bitmap.SetMask(new wxMask(mask_bitmap, *wxWHITE));
2921    wxImage cursor_image = cursor_bitmap.ConvertToImage();
2922    cursor_image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_X, hotx);
2923    cursor_image.SetOption(wxIMAGE_OPTION_CUR_HOTSPOT_Y, hoty);
2924    return wxCursor(cursor_image);
2925#else
2926    return wxCursor((const char *)bits, 32, 32, hotx, hoty,
2927                    (const char *)mask, wxBLACK, wxWHITE);
2928#endif
2929}
2930
2931const
2932#include "hand.xbm"
2933const
2934#include "handmask.xbm"
2935
2936const
2937#include "brotate.xbm"
2938const
2939#include "brotatemask.xbm"
2940
2941const
2942#include "vrotate.xbm"
2943const
2944#include "vrotatemask.xbm"
2945
2946const
2947#include "rotate.xbm"
2948const
2949#include "rotatemask.xbm"
2950
2951const
2952#include "rotatezoom.xbm"
2953const
2954#include "rotatezoommask.xbm"
2955
2956void
2957GfxCore::SetCursor(GfxCore::cursor new_cursor)
2958{
2959    // Check if we're already showing that cursor.
2960    if (current_cursor == new_cursor) return;
2961
2962    current_cursor = new_cursor;
2963    switch (current_cursor) {
2964        case GfxCore::CURSOR_DEFAULT:
2965            GLACanvas::SetCursor(wxNullCursor);
2966            break;
2967        case GfxCore::CURSOR_POINTING_HAND:
2968            GLACanvas::SetCursor(wxCursor(wxCURSOR_HAND));
2969            break;
2970        case GfxCore::CURSOR_DRAGGING_HAND:
2971            GLACanvas::SetCursor(make_cursor(hand_bits, handmask_bits, 12, 18));
2972            break;
2973        case GfxCore::CURSOR_HORIZONTAL_RESIZE:
2974            GLACanvas::SetCursor(wxCursor(wxCURSOR_SIZEWE));
2975            break;
2976        case GfxCore::CURSOR_ROTATE_HORIZONTALLY:
2977            GLACanvas::SetCursor(make_cursor(rotate_bits, rotatemask_bits, 15, 15));
2978            break;
2979        case GfxCore::CURSOR_ROTATE_VERTICALLY:
2980            GLACanvas::SetCursor(make_cursor(vrotate_bits, vrotatemask_bits, 15, 15));
2981            break;
2982        case GfxCore::CURSOR_ROTATE_EITHER_WAY:
2983            GLACanvas::SetCursor(make_cursor(brotate_bits, brotatemask_bits, 15, 15));
2984            break;
2985        case GfxCore::CURSOR_ZOOM:
2986            GLACanvas::SetCursor(wxCursor(wxCURSOR_MAGNIFIER));
2987            break;
2988        case GfxCore::CURSOR_ZOOM_ROTATE:
2989            GLACanvas::SetCursor(make_cursor(rotatezoom_bits, rotatezoommask_bits, 15, 15));
2990            break;
2991    }
2992}
2993
2994bool GfxCore::MeasuringLineActive() const
2995{
2996    if (Animating()) return false;
2997    return (m_here.IsValid() && !m_here_is_temporary) || m_there.IsValid();
2998}
2999
3000bool GfxCore::HandleRClick(wxPoint point)
3001{
3002    if (PointWithinCompass(point)) {
3003        // Pop up menu.
3004        wxMenu menu;
3005        menu.Append(menu_ORIENT_MOVE_NORTH, wmsg(/*View &North*/240));
3006        menu.Append(menu_ORIENT_MOVE_EAST, wmsg(/*View &East*/241));
3007        menu.Append(menu_ORIENT_MOVE_SOUTH, wmsg(/*View &South*/242));
3008        menu.Append(menu_ORIENT_MOVE_WEST, wmsg(/*View &West*/243));
3009        menu.AppendSeparator();
3010        menu.AppendCheckItem(menu_IND_COMPASS, wmsg(/*&Hide Compass*/387));
3011        menu.AppendCheckItem(menu_CTL_DEGREES, wmsg(/*&Degrees*/343));
3012        menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
3013        PopupMenu(&menu);
3014        return true;
3015    }
3016
3017    if (PointWithinClino(point)) {
3018        // Pop up menu.
3019        wxMenu menu;
3020        menu.Append(menu_ORIENT_PLAN, wmsg(/*&Plan View*/248));
3021        menu.Append(menu_ORIENT_ELEVATION, wmsg(/*Ele&vation*/249));
3022        menu.AppendSeparator();
3023        menu.AppendCheckItem(menu_IND_CLINO, wmsg(/*&Hide Clino*/384));
3024        menu.AppendCheckItem(menu_CTL_DEGREES, wmsg(/*&Degrees*/343));
3025        menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
3026        PopupMenu(&menu);
3027        return true;
3028    }
3029
3030    if (PointWithinScaleBar(point)) {
3031        // Pop up menu.
3032        wxMenu menu;
3033        menu.AppendCheckItem(menu_IND_SCALE_BAR, wmsg(/*&Hide scale bar*/385));
3034        menu.AppendCheckItem(menu_CTL_METRIC, wmsg(/*&Metric*/342));
3035        menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
3036        PopupMenu(&menu);
3037        return true;
3038    }
3039
3040    if (PointWithinColourKey(point)) {
3041        // Pop up menu.
3042        wxMenu menu;
3043        menu.AppendCheckItem(menu_VIEW_COLOUR_BY_DEPTH, wmsg(/*Colour by &Depth*/292));
3044        menu.AppendCheckItem(menu_VIEW_COLOUR_BY_DATE, wmsg(/*Colour by D&ate*/293));
3045        menu.AppendCheckItem(menu_VIEW_COLOUR_BY_ERROR, wmsg(/*Colour by E&rror*/289));
3046        menu.AppendSeparator();
3047        menu.AppendCheckItem(menu_IND_COLOUR_KEY, wmsg(/*&Hide colour key*/386));
3048        if (m_ColourBy == COLOUR_BY_DEPTH)
3049            menu.AppendCheckItem(menu_CTL_METRIC, wmsg(/*&Metric*/342));
3050        menu.Connect(wxEVT_COMMAND_MENU_SELECTED, (wxObjectEventFunction)&wxEvtHandler::ProcessEvent, NULL, m_Parent->GetEventHandler());
3051        PopupMenu(&menu);
3052        return true;
3053    }
3054
3055    return false;
3056}
Note: See TracBrowser for help on using the repository browser.