source: git/src/mainfrm.cc @ 73c15d2

RELEASE/1.0
Last change on this file since 73c15d2 was 73c15d2, checked in by Olly Betts <olly@…>, 13 years ago

Backport change from 1.2.0:
src/mainfrm.cc: Fix potential reading off end of string - fix from
Stuart Bennett.

git-svn-id: file:///home/survex-svn/survex/branches/1.0@3691 4b37db11-9a0c-4f06-9ece-9ab7cdaee568

  • Property mode set to 100644
File size: 47.1 KB
Line 
1//
2//  mainfrm.cc
3//
4//  Main frame handling for Aven.
5//
6//  Copyright (C) 2000-2002 Mark R. Shinwell
7//  Copyright (C) 2001-2003,2004,2005,2010 Olly Betts
8//  Copyright (C) 2004 Philip Underwood
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 "mainfrm.h"
30#include "aven.h"
31#include "aboutdlg.h"
32
33#include "message.h"
34#include "img.h"
35#include "namecmp.h"
36#include "printwx.h"
37#include "filename.h"
38
39#include <wx/confbase.h>
40#include <wx/regex.h>
41
42#include <float.h>
43#include <functional>
44#include <stack>
45
46using namespace std;
47
48const int NUM_DEPTH_COLOURS = 13; // up to 13
49
50#include "avenpal.h"
51
52class AvenSplitterWindow : public wxSplitterWindow {
53    MainFrm *parent;
54
55    public:
56        AvenSplitterWindow(MainFrm *parent_)
57            : wxSplitterWindow(parent_, -1, wxDefaultPosition, wxDefaultSize,
58                               wxSP_3D | wxSP_LIVE_UPDATE),
59              parent(parent_)
60        {
61        }
62
63        void OnSplitterDClick(wxSplitterEvent &e) {
64#if wxCHECK_VERSION(2,3,0)
65            e.Veto();
66#endif
67#if defined(__UNIX__) && !wxCHECK_VERSION(2,3,5)
68            parent->m_SashPosition = GetSashPosition(); // save width of panel
69            // Calling Unsplit from OnSplitterDClick() doesn't work in debian
70            // wxGtk 2.3.3.2 (which calls itself 2.3.4) - it does work from CVS
71            // prior to the actual 2.3.4 though - FIXME: monitor this
72            // situation...
73            SetSashPosition(0);
74#else
75            parent->ToggleSidePanel();
76#endif
77        }
78
79    private:
80        DECLARE_EVENT_TABLE()
81};
82
83BEGIN_EVENT_TABLE(AvenSplitterWindow, wxSplitterWindow)
84    // The wx docs say "EVT_SPLITTER_DOUBLECLICKED" but the wx headers say
85    // "EVT_SPLITTER_DCLICK" (wx docs corrected to agree with headers in 2.3)
86#ifdef EVT_SPLITTER_DOUBLECLICKED
87    EVT_SPLITTER_DOUBLECLICKED(-1, AvenSplitterWindow::OnSplitterDClick)
88#else
89    EVT_SPLITTER_DCLICK(-1, AvenSplitterWindow::OnSplitterDClick)
90#endif
91END_EVENT_TABLE()
92
93BEGIN_EVENT_TABLE(MainFrm, wxFrame)
94    EVT_BUTTON(wxID_FIND, MainFrm::OnFind)
95    EVT_BUTTON(button_HIDE, MainFrm::OnHide)
96
97    EVT_MENU(wxID_OPEN, MainFrm::OnOpen)
98    EVT_MENU(wxID_PRINT, MainFrm::OnPrint)
99    EVT_MENU(menu_FILE_PAGE_SETUP, MainFrm::OnPageSetup)
100    EVT_MENU(menu_FILE_EXPORT, MainFrm::OnExport)
101    EVT_MENU(wxID_EXIT, MainFrm::OnQuit)
102    EVT_MENU_RANGE(wxID_FILE1, wxID_FILE9, MainFrm::OnMRUFile)
103
104    EVT_CLOSE(MainFrm::OnClose)
105    EVT_SET_FOCUS(MainFrm::OnSetFocus)
106
107    EVT_MENU(menu_ROTATION_START, MainFrm::OnStartRotation)
108    EVT_MENU(menu_ROTATION_TOGGLE, MainFrm::OnToggleRotation)
109    EVT_MENU(menu_ROTATION_STOP, MainFrm::OnStopRotation)
110    EVT_MENU(menu_ROTATION_SPEED_UP, MainFrm::OnSpeedUp)
111    EVT_MENU(menu_ROTATION_SLOW_DOWN, MainFrm::OnSlowDown)
112    EVT_MENU(menu_ROTATION_REVERSE, MainFrm::OnReverseDirectionOfRotation)
113    EVT_MENU(menu_ROTATION_STEP_CCW, MainFrm::OnStepOnceAnticlockwise)
114    EVT_MENU(menu_ROTATION_STEP_CW, MainFrm::OnStepOnceClockwise)
115    EVT_MENU(menu_ORIENT_MOVE_NORTH, MainFrm::OnMoveNorth)
116    EVT_MENU(menu_ORIENT_MOVE_EAST, MainFrm::OnMoveEast)
117    EVT_MENU(menu_ORIENT_MOVE_SOUTH, MainFrm::OnMoveSouth)
118    EVT_MENU(menu_ORIENT_MOVE_WEST, MainFrm::OnMoveWest)
119    EVT_MENU(menu_ORIENT_SHIFT_LEFT, MainFrm::OnShiftDisplayLeft)
120    EVT_MENU(menu_ORIENT_SHIFT_RIGHT, MainFrm::OnShiftDisplayRight)
121    EVT_MENU(menu_ORIENT_SHIFT_UP, MainFrm::OnShiftDisplayUp)
122    EVT_MENU(menu_ORIENT_SHIFT_DOWN, MainFrm::OnShiftDisplayDown)
123    EVT_MENU(menu_ORIENT_PLAN, MainFrm::OnPlan)
124    EVT_MENU(menu_ORIENT_ELEVATION, MainFrm::OnElevation)
125    EVT_MENU(menu_ORIENT_HIGHER_VP, MainFrm::OnHigherViewpoint)
126    EVT_MENU(menu_ORIENT_LOWER_VP, MainFrm::OnLowerViewpoint)
127    EVT_MENU(wxID_ZOOM_IN, MainFrm::OnZoomIn)
128    EVT_MENU(wxID_ZOOM_OUT, MainFrm::OnZoomOut)
129    EVT_MENU(menu_ORIENT_DEFAULTS, MainFrm::OnDefaults)
130    EVT_MENU(menu_VIEW_SHOW_LEGS, MainFrm::OnShowSurveyLegs)
131    EVT_MENU(menu_VIEW_SHOW_CROSSES, MainFrm::OnShowCrosses)
132    EVT_MENU(menu_VIEW_SHOW_ENTRANCES, MainFrm::OnShowEntrances)
133    EVT_MENU(menu_VIEW_SHOW_FIXED_PTS, MainFrm::OnShowFixedPts)
134    EVT_MENU(menu_VIEW_SHOW_EXPORTED_PTS, MainFrm::OnShowExportedPts)
135    EVT_MENU(menu_VIEW_SHOW_NAMES, MainFrm::OnShowStationNames)
136    EVT_MENU(menu_VIEW_SHOW_OVERLAPPING_NAMES, MainFrm::OnDisplayOverlappingNames)
137    EVT_MENU(menu_VIEW_SHOW_SURFACE, MainFrm::OnShowSurface)
138    EVT_MENU(menu_VIEW_SURFACE_DEPTH, MainFrm::OnShowSurfaceDepth)
139    EVT_MENU(menu_VIEW_SURFACE_DASHED, MainFrm::OnShowSurfaceDashed)
140    EVT_MENU(menu_VIEW_COMPASS, MainFrm::OnViewCompass)
141    EVT_MENU(menu_VIEW_CLINO, MainFrm::OnViewClino)
142    EVT_MENU(menu_VIEW_GRID, MainFrm::OnViewGrid)
143    EVT_MENU(menu_VIEW_DEPTH_BAR, MainFrm::OnToggleDepthbar)
144    EVT_MENU(menu_VIEW_SCALE_BAR, MainFrm::OnToggleScalebar)
145    EVT_MENU(menu_VIEW_SIDE_PANEL, MainFrm::OnViewSidePanel)
146    EVT_MENU(menu_VIEW_METRIC, MainFrm::OnToggleMetric)
147    EVT_MENU(menu_VIEW_DEGREES, MainFrm::OnToggleDegrees)
148    EVT_MENU(menu_CTL_REVERSE, MainFrm::OnReverseControls)
149    EVT_MENU(menu_CTL_CANCEL_DIST_LINE, MainFrm::OnCancelDistLine)
150    EVT_MENU(wxID_ABOUT, MainFrm::OnAbout)
151
152    EVT_UPDATE_UI(wxID_PRINT, MainFrm::OnPrintUpdate)
153    EVT_UPDATE_UI(menu_FILE_EXPORT, MainFrm::OnExportUpdate)
154    EVT_UPDATE_UI(menu_ROTATION_START, MainFrm::OnStartRotationUpdate)
155    EVT_UPDATE_UI(menu_ROTATION_TOGGLE, MainFrm::OnToggleRotationUpdate)
156    EVT_UPDATE_UI(menu_ROTATION_STOP, MainFrm::OnStopRotationUpdate)
157    EVT_UPDATE_UI(menu_ROTATION_SPEED_UP, MainFrm::OnSpeedUpUpdate)
158    EVT_UPDATE_UI(menu_ROTATION_SLOW_DOWN, MainFrm::OnSlowDownUpdate)
159    EVT_UPDATE_UI(menu_ROTATION_REVERSE, MainFrm::OnReverseDirectionOfRotationUpdate)
160    EVT_UPDATE_UI(menu_ROTATION_STEP_CCW, MainFrm::OnStepOnceAnticlockwiseUpdate)
161    EVT_UPDATE_UI(menu_ROTATION_STEP_CW, MainFrm::OnStepOnceClockwiseUpdate)
162    EVT_UPDATE_UI(menu_ORIENT_MOVE_NORTH, MainFrm::OnMoveNorthUpdate)
163    EVT_UPDATE_UI(menu_ORIENT_MOVE_EAST, MainFrm::OnMoveEastUpdate)
164    EVT_UPDATE_UI(menu_ORIENT_MOVE_SOUTH, MainFrm::OnMoveSouthUpdate)
165    EVT_UPDATE_UI(menu_ORIENT_MOVE_WEST, MainFrm::OnMoveWestUpdate)
166    EVT_UPDATE_UI(menu_ORIENT_SHIFT_LEFT, MainFrm::OnShiftDisplayLeftUpdate)
167    EVT_UPDATE_UI(menu_ORIENT_SHIFT_RIGHT, MainFrm::OnShiftDisplayRightUpdate)
168    EVT_UPDATE_UI(menu_ORIENT_SHIFT_UP, MainFrm::OnShiftDisplayUpUpdate)
169    EVT_UPDATE_UI(menu_ORIENT_SHIFT_DOWN, MainFrm::OnShiftDisplayDownUpdate)
170    EVT_UPDATE_UI(menu_ORIENT_PLAN, MainFrm::OnPlanUpdate)
171    EVT_UPDATE_UI(menu_ORIENT_ELEVATION, MainFrm::OnElevationUpdate)
172    EVT_UPDATE_UI(menu_ORIENT_HIGHER_VP, MainFrm::OnHigherViewpointUpdate)
173    EVT_UPDATE_UI(menu_ORIENT_LOWER_VP, MainFrm::OnLowerViewpointUpdate)
174    EVT_UPDATE_UI(wxID_ZOOM_IN, MainFrm::OnZoomInUpdate)
175    EVT_UPDATE_UI(wxID_ZOOM_OUT, MainFrm::OnZoomOutUpdate)
176    EVT_UPDATE_UI(menu_ORIENT_DEFAULTS, MainFrm::OnDefaultsUpdate)
177    EVT_UPDATE_UI(menu_VIEW_SHOW_LEGS, MainFrm::OnShowSurveyLegsUpdate)
178    EVT_UPDATE_UI(menu_VIEW_SHOW_CROSSES, MainFrm::OnShowCrossesUpdate)
179    EVT_UPDATE_UI(menu_VIEW_SHOW_ENTRANCES, MainFrm::OnShowEntrancesUpdate)
180    EVT_UPDATE_UI(menu_VIEW_SHOW_FIXED_PTS, MainFrm::OnShowFixedPtsUpdate)
181    EVT_UPDATE_UI(menu_VIEW_SHOW_EXPORTED_PTS, MainFrm::OnShowExportedPtsUpdate)
182    EVT_UPDATE_UI(menu_VIEW_SHOW_NAMES, MainFrm::OnShowStationNamesUpdate)
183    EVT_UPDATE_UI(menu_VIEW_SHOW_SURFACE, MainFrm::OnShowSurfaceUpdate)
184    EVT_UPDATE_UI(menu_VIEW_SURFACE_DEPTH, MainFrm::OnShowSurfaceDepthUpdate)
185    EVT_UPDATE_UI(menu_VIEW_SURFACE_DASHED, MainFrm::OnShowSurfaceDashedUpdate)
186    EVT_UPDATE_UI(menu_VIEW_SHOW_OVERLAPPING_NAMES, MainFrm::OnDisplayOverlappingNamesUpdate)
187    EVT_UPDATE_UI(menu_VIEW_COMPASS, MainFrm::OnViewCompassUpdate)
188    EVT_UPDATE_UI(menu_VIEW_CLINO, MainFrm::OnViewClinoUpdate)
189    EVT_UPDATE_UI(menu_VIEW_DEPTH_BAR, MainFrm::OnToggleDepthbarUpdate)
190    EVT_UPDATE_UI(menu_VIEW_SCALE_BAR, MainFrm::OnToggleScalebarUpdate)
191    EVT_UPDATE_UI(menu_VIEW_GRID, MainFrm::OnViewGridUpdate)
192    EVT_UPDATE_UI(menu_VIEW_INDICATORS, MainFrm::OnIndicatorsUpdate)
193    EVT_UPDATE_UI(menu_VIEW_SIDE_PANEL, MainFrm::OnViewSidePanelUpdate)
194    EVT_UPDATE_UI(menu_CTL_REVERSE, MainFrm::OnReverseControlsUpdate)
195    EVT_UPDATE_UI(menu_CTL_CANCEL_DIST_LINE, MainFrm::OnCancelDistLineUpdate)
196    EVT_UPDATE_UI(menu_VIEW_METRIC, MainFrm::OnToggleMetricUpdate)
197    EVT_UPDATE_UI(menu_VIEW_DEGREES, MainFrm::OnToggleDegreesUpdate)
198END_EVENT_TABLE()
199
200class LabelCmp : public greater<const LabelInfo*> {
201    int separator;
202public:
203    LabelCmp(int separator_) : separator(separator_) {}
204    bool operator()(const LabelInfo* pt1, const LabelInfo* pt2) {
205        return name_cmp(pt1->GetText(), pt2->GetText(), separator) < 0;
206    }
207};
208
209class LabelPlotCmp : public greater<const LabelInfo*> {
210    int separator;
211public:
212    LabelPlotCmp(int separator_) : separator(separator_) {}
213    bool operator()(const LabelInfo* pt1, const LabelInfo* pt2) {
214        int n = pt1->flags - pt2->flags;
215        if (n) return n > 0;
216        wxString l1 = pt1->text.AfterLast(separator);
217        wxString l2 = pt2->text.AfterLast(separator);
218        n = name_cmp(l1, l2, separator);
219        if (n) return n < 0;
220        // Prefer non-2-nodes...
221        // FIXME; implement
222        // if leaf names are the same, prefer shorter labels as we can
223        // display more of them
224        n = pt1->text.length() - pt2->text.length();
225        if (n) return n < 0;
226        // make sure that we don't ever compare different labels as equal
227        return name_cmp(pt1->text, pt2->text, separator) < 0;
228    }
229};
230
231#if wxUSE_DRAG_AND_DROP
232class DnDFile : public wxFileDropTarget {
233    public:
234        DnDFile(MainFrm *parent) : m_Parent(parent) { }
235        virtual bool OnDropFiles(wxCoord, wxCoord,
236                        const wxArrayString &filenames);
237
238    private:
239        MainFrm * m_Parent;
240};
241
242bool
243DnDFile::OnDropFiles(wxCoord, wxCoord, const wxArrayString &filenames)
244{
245    // Load a survey file by drag-and-drop.
246    assert(filenames.GetCount() > 0);
247
248    if (filenames.GetCount() != 1) {
249        wxGetApp().ReportError(msg(/*You may only view one 3d file at a time.*/336));
250        return FALSE;
251    }
252
253    m_Parent->OpenFile(filenames[0]);
254    return TRUE;
255}
256#endif
257
258MainFrm::MainFrm(const wxString& title, const wxPoint& pos, const wxSize& size) :
259    wxFrame(NULL, 101, title, pos, size, wxDEFAULT_FRAME_STYLE | wxNO_FULL_REPAINT_ON_RESIZE),
260    m_Gfx(NULL), m_NumEntrances(0), m_NumFixedPts(0), m_NumExportedPts(0)
261{
262    icon_path = msg_cfgpth();
263    icon_path += wxCONFIG_PATH_SEPARATOR;
264    icon_path += "icons";
265    icon_path += wxCONFIG_PATH_SEPARATOR;
266
267#ifdef _WIN32
268    // The peculiar name is so that the icon is the first in the file
269    // (required by Microsoft Windows for this type of icon)
270    SetIcon(wxIcon("aaaaaAven"));
271#else
272    SetIcon(wxIcon(icon_path + "aven.png", wxBITMAP_TYPE_PNG));
273#endif
274
275    InitialisePensAndBrushes();
276    CreateMenuBar();
277    CreateToolBar();
278    CreateSidePanel();
279
280#ifdef __X__ // wxMotif or wxX11
281    int x;
282    int y;
283    GetSize(&x, &y);
284    // X seems to require a forced resize.
285    SetSize(-1, -1, x, y);
286#endif
287
288#if wxUSE_DRAG_AND_DROP
289    SetDropTarget(new DnDFile(this));
290#endif
291}
292
293MainFrm::~MainFrm()
294{
295    ClearPointLists();
296    delete[] m_Points;
297    delete[] m_Pens;
298    delete[] m_Brushes;
299}
300
301void MainFrm::InitialisePensAndBrushes()
302{
303    m_Points = new list<PointInfo*>[NUM_DEPTH_COLOURS + 1];
304    m_Pens = new wxPen[NUM_DEPTH_COLOURS + 1];
305    m_Brushes = new wxBrush[NUM_DEPTH_COLOURS + 1];
306    for (int pen = 0; pen < NUM_DEPTH_COLOURS + 1; ++pen) {
307        m_Pens[pen].SetColour(REDS[pen], GREENS[pen], BLUES[pen]);
308        m_Brushes[pen].SetColour(REDS[pen], GREENS[pen], BLUES[pen]);
309    }
310}
311
312void MainFrm::CreateMenuBar()
313{
314    // Create the menus and the menu bar.
315
316    wxMenu* filemenu = new wxMenu;
317    // wxID_OPEN stock label lacks the ellipses
318    filemenu->Append(wxID_OPEN, GetTabMsg(/*@Open...##Ctrl+O*/220));
319    filemenu->AppendSeparator();
320    // wxID_PRINT stock label lacks the ellipses
321    filemenu->Append(wxID_PRINT, GetTabMsg(/*@Print...##Ctrl+P*/380));
322    filemenu->Append(menu_FILE_PAGE_SETUP, GetTabMsg(/*P@age Setup...*/381));
323    filemenu->AppendSeparator();
324    filemenu->Append(menu_FILE_EXPORT, GetTabMsg(/*@Export as...*/382));
325    filemenu->AppendSeparator();
326    filemenu->Append(wxID_EXIT);
327
328    m_history.UseMenu(filemenu);
329    m_history.Load(*wxConfigBase::Get());
330
331    wxMenu* rotmenu = new wxMenu;
332    rotmenu->Append(menu_ROTATION_START, GetTabMsg(/*@Start Rotation##Return*/230));
333    rotmenu->Append(menu_ROTATION_STOP, GetTabMsg(/*S@top Rotation##Space*/231));
334    rotmenu->AppendSeparator();
335    rotmenu->Append(menu_ROTATION_SPEED_UP, GetTabMsg(/*Speed @Up*/232));
336    rotmenu->Append(menu_ROTATION_SLOW_DOWN, GetTabMsg(/*Slow @Down*/233));
337    rotmenu->AppendSeparator();
338    rotmenu->Append(menu_ROTATION_REVERSE, GetTabMsg(/*@Reverse Direction*/234));
339    rotmenu->AppendSeparator();
340    rotmenu->Append(menu_ROTATION_STEP_CCW, GetTabMsg(/*Step Once @Anticlockwise*/235));
341    rotmenu->Append(menu_ROTATION_STEP_CW, GetTabMsg(/*Step Once @Clockwise*/236));
342
343    wxMenu* orientmenu = new wxMenu;
344    orientmenu->Append(menu_ORIENT_MOVE_NORTH, GetTabMsg(/*View @North*/240));
345    orientmenu->Append(menu_ORIENT_MOVE_EAST, GetTabMsg(/*View @East*/241));
346    orientmenu->Append(menu_ORIENT_MOVE_SOUTH, GetTabMsg(/*View @South*/242));
347    orientmenu->Append(menu_ORIENT_MOVE_WEST, GetTabMsg(/*View @West*/243));
348    orientmenu->AppendSeparator();
349    orientmenu->Append(menu_ORIENT_SHIFT_LEFT, GetTabMsg(/*Shift Survey @Left*/244));
350    orientmenu->Append(menu_ORIENT_SHIFT_RIGHT, GetTabMsg(/*Shift Survey @Right*/245));
351    orientmenu->Append(menu_ORIENT_SHIFT_UP, GetTabMsg(/*Shift Survey @Up*/246));
352    orientmenu->Append(menu_ORIENT_SHIFT_DOWN, GetTabMsg(/*Shift Survey @Down*/247));
353    orientmenu->AppendSeparator();
354    orientmenu->Append(menu_ORIENT_PLAN, GetTabMsg(/*@Plan View*/248));
355    orientmenu->Append(menu_ORIENT_ELEVATION, GetTabMsg(/*Ele@vation*/249));
356    orientmenu->AppendSeparator();
357    orientmenu->Append(menu_ORIENT_HIGHER_VP, GetTabMsg(/*@Higher Viewpoint*/250));
358    orientmenu->Append(menu_ORIENT_LOWER_VP, GetTabMsg(/*L@ower Viewpoint*/251));
359    orientmenu->AppendSeparator();
360    // Default labels for wxID_ZOOM_IN and wxID_ZOOM_OUT don't have accels.
361    orientmenu->Append(wxID_ZOOM_IN, GetTabMsg(/*@Zoom In##]*/252));
362    orientmenu->Append(wxID_ZOOM_OUT, GetTabMsg(/*Zoo@m Out##[*/253));
363    orientmenu->AppendSeparator();
364    orientmenu->Append(menu_ORIENT_DEFAULTS, GetTabMsg(/*Restore De@fault Settings*/254));
365
366    wxMenu* viewmenu = new wxMenu;
367    viewmenu->Append(menu_VIEW_SHOW_NAMES, GetTabMsg(/*Station @Names##Ctrl+N*/270), "", true);
368    viewmenu->Append(menu_VIEW_SHOW_CROSSES, GetTabMsg(/*@Crosses##Ctrl+X*/271), "", true);
369    viewmenu->Append(menu_VIEW_GRID, GetTabMsg(/*@Grid##Ctrl+G*/297), "", true);
370    viewmenu->AppendSeparator();
371    viewmenu->Append(menu_VIEW_SHOW_LEGS, GetTabMsg(/*@Underground Survey Legs##Ctrl+L*/272), "", true);
372    viewmenu->Append(menu_VIEW_SHOW_SURFACE, GetTabMsg(/*@Surface Survey Legs##Ctrl+F*/291), "", true);
373    viewmenu->AppendSeparator();
374    viewmenu->Append(menu_VIEW_SURFACE_DEPTH, GetTabMsg(/*@Altitude Colouring on Surface Surveys*/292), "", true);
375    viewmenu->Append(menu_VIEW_SURFACE_DASHED, GetTabMsg(/*@Dashed Surface Surveys*/293), "", true);
376    viewmenu->AppendSeparator();
377    viewmenu->Append(menu_VIEW_SHOW_OVERLAPPING_NAMES, GetTabMsg(/*@Overlapping Names*/273), "", true);
378    viewmenu->AppendSeparator();
379    viewmenu->Append(menu_VIEW_SHOW_ENTRANCES, GetTabMsg(/*Highlight @Entrances*/294), "", true);
380    viewmenu->Append(menu_VIEW_SHOW_FIXED_PTS, GetTabMsg(/*Highlight @Fixed Points*/295), "", true);
381    viewmenu->Append(menu_VIEW_SHOW_EXPORTED_PTS, GetTabMsg(/*Highlight E@xported Points*/296), "", true);
382
383    wxMenu* ctlmenu = new wxMenu;
384    ctlmenu->Append(menu_CTL_REVERSE, GetTabMsg(/*@Reverse Sense##Ctrl+R*/280), "", true);
385    ctlmenu->AppendSeparator();
386    ctlmenu->Append(menu_CTL_CANCEL_DIST_LINE, GetTabMsg(/*@Cancel Measuring Line##Escape*/281));
387    ctlmenu->AppendSeparator();
388    wxMenu* indmenu = new wxMenu;
389    indmenu->Append(menu_VIEW_COMPASS, GetTabMsg(/*@Compass*/274), "", true);
390    indmenu->Append(menu_VIEW_CLINO, GetTabMsg(/*C@linometer*/275), "", true);
391    indmenu->Append(menu_VIEW_DEPTH_BAR, GetTabMsg(/*@Depth Bar*/276), "", true);
392    indmenu->Append(menu_VIEW_SCALE_BAR, GetTabMsg(/*@Scale Bar*/277), "", true);
393    ctlmenu->Append(menu_VIEW_INDICATORS, GetTabMsg(/*@Indicators*/299), indmenu);
394    ctlmenu->Append(menu_VIEW_SIDE_PANEL, GetTabMsg(/*@Side Panel*/337), "", true);
395    ctlmenu->AppendSeparator();
396    ctlmenu->Append(menu_VIEW_METRIC, GetTabMsg(/*@Metric*/342), "", true);
397    ctlmenu->Append(menu_VIEW_DEGREES, GetTabMsg(/*@Degrees*/343), "", true);
398
399    wxMenu* helpmenu = new wxMenu;
400    helpmenu->Append(wxID_ABOUT);
401
402    wxMenuBar* menubar = new wxMenuBar(wxMB_DOCKABLE);
403    menubar->Append(filemenu, GetTabMsg(/*@File*/210));
404    menubar->Append(rotmenu, GetTabMsg(/*@Rotation*/211));
405    menubar->Append(orientmenu, GetTabMsg(/*@Orientation*/212));
406    menubar->Append(viewmenu, GetTabMsg(/*@View*/213));
407    menubar->Append(ctlmenu, GetTabMsg(/*@Controls*/214));
408    menubar->Append(helpmenu, GetTabMsg(/*@Help*/215));
409    SetMenuBar(menubar);
410}
411
412// ICON must be a literal string.
413#define TOOLBAR_BITMAP(ICON) wxBitmap(icon_path + ICON".png", wxBITMAP_TYPE_PNG)
414
415void MainFrm::CreateToolBar()
416{
417    // Create the toolbar.
418
419    wxToolBar* toolbar = wxFrame::CreateToolBar();
420
421#ifndef __WXGTK20__
422    toolbar->SetMargins(5, 5);
423#endif
424
425    toolbar->AddTool(wxID_OPEN, TOOLBAR_BITMAP("open"), "Open a 3D file for viewing");
426    toolbar->AddSeparator();
427    toolbar->AddTool(menu_ROTATION_TOGGLE, TOOLBAR_BITMAP("rotation"),
428                     wxNullBitmap, true, -1, -1, NULL, "Toggle rotation");
429    toolbar->AddSeparator();
430    toolbar->AddTool(menu_ORIENT_PLAN, TOOLBAR_BITMAP("plan"), "Switch to plan view");
431    toolbar->AddTool(menu_ORIENT_ELEVATION, TOOLBAR_BITMAP("elevation"), "Switch to elevation view");
432    toolbar->AddSeparator();
433    toolbar->AddTool(menu_ORIENT_DEFAULTS, TOOLBAR_BITMAP("defaults"), "Restore default view");
434    toolbar->AddSeparator();
435    toolbar->AddTool(menu_VIEW_SHOW_NAMES, TOOLBAR_BITMAP("names"), wxNullBitmap, true,
436                     -1, -1, NULL, "Show station names");
437    toolbar->AddTool(menu_VIEW_SHOW_CROSSES, TOOLBAR_BITMAP("crosses"), wxNullBitmap, true,
438                     -1, -1, NULL, "Show crosses on stations");
439    toolbar->AddTool(menu_VIEW_SHOW_ENTRANCES, TOOLBAR_BITMAP("entrances"), wxNullBitmap, true,
440                     -1, -1, NULL, "Highlight entrances");
441    toolbar->AddTool(menu_VIEW_SHOW_FIXED_PTS, TOOLBAR_BITMAP("fixed-pts"), wxNullBitmap, true,
442                     -1, -1, NULL, "Highlight fixed points");
443    toolbar->AddTool(menu_VIEW_SHOW_EXPORTED_PTS, TOOLBAR_BITMAP("exported-pts"), wxNullBitmap, true,
444                     -1, -1, NULL, "Highlight exported stations");
445    toolbar->AddSeparator();
446    toolbar->AddTool(menu_VIEW_SHOW_LEGS, TOOLBAR_BITMAP("ug-legs"), wxNullBitmap, true,
447                     -1, -1, NULL, "Show underground surveys");
448    toolbar->AddTool(menu_VIEW_SHOW_SURFACE, TOOLBAR_BITMAP("surface-legs"), wxNullBitmap, true,
449                     -1, -1, NULL, "Show surface surveys");
450    toolbar->AddSeparator();
451
452    toolbar->Realize();
453}
454
455void MainFrm::CreateSidePanel()
456{
457    m_Splitter = new AvenSplitterWindow(this);
458
459    m_Panel = new wxPanel(m_Splitter);
460    m_Tree = new AvenTreeCtrl(this, m_Panel);
461    wxPanel *find_panel = new wxPanel(m_Panel);
462    m_Panel->Show(false);
463
464    m_FindBox = new wxTextCtrl(find_panel, -1, "");
465    wxButton *find_button, *hide_button;
466    find_button = new wxButton(find_panel, wxID_FIND, msg(/*Find*/332));
467    find_button->SetDefault();
468    // SetDefaultItem() was moved from wxPanel to wxTopLevelWindow in wx2.8.
469    // No idea how to work around that, but it doesn't seem to be needed with
470    // wx2.8 (at least with wxGTK - perhaps the SetDefault() call above does
471    // the trick.  Seems safest to still call this for older version in case
472    // they need it.
473#if !wxCHECK_VERSION(2,8,0)
474    find_panel->SetDefaultItem(find_button);
475#endif
476    hide_button = new wxButton(find_panel, button_HIDE, msg(/*Hide*/333));
477    m_RegexpCheckBox = new wxCheckBox(find_panel, -1,
478                                      msg(/*Regular expression*/334));
479    m_Coords = new wxStaticText(find_panel, -1, "");
480    m_StnCoords = new wxStaticText(find_panel, -1, "");
481    //  m_MousePtr = new wxStaticText(find_panel, -1, "Mouse coordinates");
482    m_StnName = new wxStaticText(find_panel, -1, "");
483    m_StnAlt = new wxStaticText(find_panel, -1, "");
484    m_Dist1 = new wxStaticText(find_panel, -1, "");
485    m_Dist2 = new wxStaticText(find_panel, -1, "");
486    m_Dist3 = new wxStaticText(find_panel, -1, "");
487    m_Found = new wxStaticText(find_panel, -1, "");
488
489    wxBoxSizer *find_button_sizer = new wxBoxSizer(wxHORIZONTAL);
490    find_button_sizer->Add(m_FindBox, 1, wxALL, 2);
491#ifdef _WIN32
492    find_button_sizer->Add(find_button, 0, wxALL, 2);
493#else
494    // GTK+ (and probably Motif) default buttons have a thick external
495    // border we need to allow for
496    find_button_sizer->Add(find_button, 0, wxALL, 4);
497#endif
498
499    wxBoxSizer *hide_button_sizer = new wxBoxSizer(wxHORIZONTAL);
500    hide_button_sizer->Add(m_Found, 1, wxALL, 2);
501#ifdef _WIN32
502    hide_button_sizer->Add(hide_button, 0, wxALL, 2);
503#else
504    hide_button_sizer->Add(hide_button, 0, wxALL, 4);
505#endif
506
507    wxBoxSizer *find_sizer = new wxBoxSizer(wxVERTICAL);
508    find_sizer->Add(find_button_sizer, 0, wxALL | wxEXPAND, 2);
509    find_sizer->Add(hide_button_sizer, 0, wxALL | wxEXPAND, 2);
510    find_sizer->Add(m_RegexpCheckBox, 0, wxALL | wxEXPAND, 2);
511    find_sizer->Add(10, 5, 0, wxALL | wxEXPAND, 2);
512    //   find_sizer->Add(m_MousePtr, 0, wxALL | wxEXPAND, 2);
513    find_sizer->Add(m_Coords, 0, wxALL | wxEXPAND, 2);
514    find_sizer->Add(10, 5, 0, wxALL | wxEXPAND, 2);
515    find_sizer->Add(m_StnName, 0, wxALL | wxEXPAND, 2);
516    find_sizer->Add(m_StnCoords, 0, wxALL | wxEXPAND, 2);
517    find_sizer->Add(m_StnAlt, 0, wxALL | wxEXPAND, 2);
518    find_sizer->Add(10, 5, 0, wxALL | wxEXPAND, 2);
519    find_sizer->Add(m_Dist1, 0, wxALL | wxEXPAND, 2);
520    find_sizer->Add(m_Dist2, 0, wxALL | wxEXPAND, 2);
521    find_sizer->Add(m_Dist3, 0, wxALL | wxEXPAND, 2);
522
523    find_panel->SetAutoLayout(true);
524    find_panel->SetSizer(find_sizer);
525    find_sizer->Fit(find_panel);
526    find_sizer->SetSizeHints(find_panel);
527
528    wxBoxSizer *panel_sizer = new wxBoxSizer(wxVERTICAL);
529    panel_sizer->Add(m_Tree, 1, wxALL | wxEXPAND, 2);
530    panel_sizer->Add(find_panel, 0, wxALL | wxEXPAND, 2);
531    m_Panel->SetAutoLayout(true);
532    m_Panel->SetSizer(panel_sizer);
533//    panel_sizer->Fit(m_Panel);
534//    panel_sizer->SetSizeHints(m_Panel);
535
536    m_Gfx = new GfxCore(this, m_Splitter);
537
538    m_Splitter->Initialize(m_Gfx);
539}
540
541void MainFrm::ClearPointLists()
542{
543    // Free memory occupied by the contents of the point and label lists.
544
545    for (int band = 0; band < NUM_DEPTH_COLOURS + 1; band++) {
546        list<PointInfo*>::iterator pos = m_Points[band].begin();
547        list<PointInfo*>::iterator end = m_Points[band].end();
548        while (pos != end) {
549            PointInfo* point = *pos++;
550            delete point;
551        }
552        m_Points[band].clear();
553    }
554
555    list<LabelInfo*>::iterator pos = m_Labels.begin();
556    while (pos != m_Labels.end()) {
557        LabelInfo* label = *pos++;
558        delete label;
559    }
560    m_Labels.clear();
561}
562
563bool MainFrm::LoadData(const wxString& file, wxString prefix)
564{
565    // Load survey data from file, centre the dataset around the origin,
566    // chop legs such that no legs cross depth colour boundaries and prepare
567    // the data for drawing.
568
569#if 0
570    wxStopWatch timer;
571    timer.Start();
572#endif
573
574    // Load the survey data.
575
576    img* survey = img_open_survey(file, prefix.c_str());
577    if (!survey) {
578        wxString m = wxString::Format(msg(img_error()), file.c_str());
579        wxGetApp().ReportError(m);
580        return false;
581    }
582
583    m_File = survey->filename_opened;
584
585    m_Tree->DeleteAllItems();
586
587    // Create a list of all the leg vertices, counting them and finding the
588    // extent of the survey at the same time.
589
590    m_NumLegs = 0;
591    m_NumPoints = 0;
592    m_NumExtraLegs = 0;
593    m_NumCrosses = 0;
594    m_NumFixedPts = 0;
595    m_NumExportedPts = 0;
596    m_NumEntrances = 0;
597
598    // Delete any existing list entries.
599    ClearPointLists();
600
601    Double xmin = DBL_MAX;
602    Double xmax = -DBL_MAX;
603    Double ymin = DBL_MAX;
604    Double ymax = -DBL_MAX;
605    m_ZMin = DBL_MAX;
606    Double zmax = -DBL_MAX;
607
608    list<PointInfo*> points;
609
610    int result;
611    do {
612        img_point pt;
613        result = img_read_item(survey, &pt);
614        switch (result) {
615            case img_MOVE:
616            case img_LINE:
617            {
618                m_NumPoints++;
619
620                // Update survey extents.
621                if (pt.x < xmin) xmin = pt.x;
622                if (pt.x > xmax) xmax = pt.x;
623                if (pt.y < ymin) ymin = pt.y;
624                if (pt.y > ymax) ymax = pt.y;
625                if (pt.z < m_ZMin) m_ZMin = pt.z;
626                if (pt.z > zmax) zmax = pt.z;
627
628                PointInfo* info = new PointInfo;
629                info->x = pt.x;
630                info->y = pt.y;
631                info->z = pt.z;
632
633                if (result == img_LINE) {
634                    // Set flags to say this is a line rather than a move
635                    m_NumLegs++;
636                    info->isLine = true;
637                    info->isSurface = (survey->flags & img_FLAG_SURFACE);
638                } else {
639                    info->isLine = false;
640                }
641
642                // Add this point to the list.
643                points.push_back(info);
644                break;
645            }
646
647            case img_LABEL: {
648                LabelInfo* label = new LabelInfo;
649                label->text = survey->label;
650                label->x = pt.x;
651                label->y = pt.y;
652                label->z = pt.z;
653                if (survey->flags & img_SFLAG_ENTRANCE) {
654                    survey->flags ^= (img_SFLAG_ENTRANCE | LFLAG_ENTRANCE);
655                }
656                label->flags = survey->flags;
657                if (label->IsEntrance()) {
658                    m_NumEntrances++;
659                }
660                if (label->IsFixedPt()) {
661                    m_NumFixedPts++;
662                }
663                if (label->IsExportedPt()) {
664                    m_NumExportedPts++;
665                }
666                m_Labels.push_back(label);
667                m_NumCrosses++;
668
669                break;
670            }
671
672            case img_BAD: {
673                m_Labels.clear();
674
675                // FIXME: Do we need to reset all these? - Olly
676                m_NumLegs = 0;
677                m_NumPoints = 0;
678                m_NumExtraLegs = 0;
679                m_NumCrosses = 0;
680                m_NumFixedPts = 0;
681                m_NumExportedPts = 0;
682                m_NumEntrances = 0;
683
684                m_ZMin = DBL_MAX;
685
686                img_close(survey);
687
688                wxString m = wxString::Format(msg(img_error()), file.c_str());
689                wxGetApp().ReportError(m);
690
691                return false;
692            }
693
694            default:
695                break;
696        }
697    } while (result != img_STOP);
698
699    separator = survey->separator;
700    m_Title = survey->title;
701    m_DateStamp = survey->datestamp;
702    img_close(survey);
703
704    // Check we've actually loaded some legs or stations!
705    if (m_NumLegs == 0 && m_Labels.empty()) {
706        wxString m = wxString::Format(msg(/*No survey data in 3d file `%s'*/202), file.c_str());
707        wxGetApp().ReportError(m);
708        return false;
709    }
710
711    if (points.empty()) {
712        // No legs, so get survey extents from stations
713        list<LabelInfo*>::const_iterator i;
714        for (i = m_Labels.begin(); i != m_Labels.end(); ++i) {
715            if ((*i)->x < xmin) xmin = (*i)->x;
716            if ((*i)->x > xmax) xmax = (*i)->x;
717            if ((*i)->y < ymin) ymin = (*i)->y;
718            if ((*i)->y > ymax) ymax = (*i)->y;
719            if ((*i)->z < m_ZMin) m_ZMin = (*i)->z;
720            if ((*i)->z > zmax) zmax = (*i)->z;
721        }
722    } else {
723        // Delete any trailing move.
724        PointInfo* pt = points.back();
725        if (!pt->isLine) {
726            m_NumPoints--;
727            points.pop_back();
728            delete pt;
729        }
730    }
731
732    m_XExt = xmax - xmin;
733    m_YExt = ymax - ymin;
734    m_ZExt = zmax - m_ZMin;
735
736    // FIXME -- temporary bodge
737    m_XMin = xmin;
738    m_YMin = ymin;
739
740    // Sort the labels.
741    m_Labels.sort(LabelCmp(separator));
742
743    // Fill the tree of stations and prefixes.
744    FillTree();
745
746    // Sort labels so that entrances are displayed in preference,
747    // then fixed points, then exported points, then other points.
748    //
749    // Also sort by leaf name so that we'll tend to choose labels
750    // from different surveys, rather than labels from surveys which
751    // are earlier in the list.
752    m_Labels.sort(LabelPlotCmp(separator));
753
754    // Sort out depth colouring boundaries (before centering dataset!)
755    SortIntoDepthBands(points);
756
757    // Centre the dataset around the origin.
758    CentreDataset(xmin, ymin, m_ZMin);
759
760#if 0
761    printf("time to load = %.3f\n", (double)timer.Time());
762#endif
763
764    // Update window title.
765    SetTitle(wxString(APP_NAME" - [") + m_File + wxString("]"));
766
767    return true;
768}
769
770void MainFrm::FillTree()
771{
772    // Create the root of the tree.
773    wxTreeItemId treeroot = m_Tree->AddRoot(wxFileNameFromPath(m_File));
774
775    // Fill the tree of stations and prefixes.
776    stack<wxTreeItemId> previous_ids;
777    wxString current_prefix = "";
778    wxTreeItemId current_id = treeroot;
779
780    list<LabelInfo*>::iterator pos = m_Labels.begin();
781    while (pos != m_Labels.end()) {
782        LabelInfo* label = *pos++;
783
784        // Determine the current prefix.
785        wxString prefix = label->GetText().BeforeLast(separator);
786
787        // Determine if we're still on the same prefix.
788        if (prefix == current_prefix) {
789            // no need to fiddle with branches...
790        }
791        // If not, then see if we've descended to a new prefix.
792        else if (prefix.length() > current_prefix.length() &&
793                 prefix.StartsWith(current_prefix) &&
794                 (prefix[current_prefix.length()] == separator ||
795                  current_prefix == "")) {
796            // We have, so start as many new branches as required.
797            int current_prefix_length = current_prefix.length();
798            current_prefix = prefix;
799            if (current_prefix_length != 0) {
800                prefix = prefix.Mid(current_prefix_length + 1);
801            }
802            int next_dot;
803            do {
804                // Extract the next bit of prefix.
805                next_dot = prefix.Find(separator);
806
807                wxString bit = next_dot == -1 ? prefix : prefix.Left(next_dot);
808                assert(bit != "");
809
810                // Add the current tree ID to the stack.
811                previous_ids.push(current_id);
812
813                // Append the new item to the tree and set this as the current branch.
814                current_id = m_Tree->AppendItem(current_id, bit);
815                m_Tree->SetItemData(current_id, new TreeData(NULL));
816                prefix = prefix.Mid(next_dot + 1);
817            } while (next_dot != -1);
818        }
819        // Otherwise, we must have moved up, and possibly then down again.
820        else {
821            size_t count = 0;
822            bool ascent_only = (prefix.length() < current_prefix.length() &&
823                                current_prefix.StartsWith(prefix) &&
824                                (current_prefix[prefix.length()] == separator ||
825                                 prefix == ""));
826            if (!ascent_only) {
827                // Find out how much of the current prefix and the new prefix
828                // are the same.
829                // Note that we require a match of a whole number of parts
830                // between dots!
831                size_t n = min(prefix.length(), current_prefix.length());
832                size_t i;
833                for (i = 0; i < n && prefix[i] == current_prefix[i]; ++i) {
834                    if (prefix[i] == separator) count = i + 1;
835                }
836            } else {
837                count = prefix.length() + 1;
838            }
839
840            // Extract the part of the current prefix after the bit (if any)
841            // which has matched.
842            // This gives the prefixes to ascend over.
843            wxString prefixes_ascended = current_prefix.Mid(count);
844
845            // Count the number of prefixes to ascend over.
846            int num_prefixes = prefixes_ascended.Freq(separator);
847
848            // Reverse up over these prefixes.
849            for (int i = 1; i <= num_prefixes; i++) {
850                previous_ids.pop();
851            }
852            current_id = previous_ids.top();
853            previous_ids.pop();
854
855            if (!ascent_only) {
856                // Now extract the bit of new prefix.
857                wxString new_prefix = prefix.Mid(count);
858
859                // Add branches for this new part.
860                while (true) {
861                    // Extract the next bit of prefix.
862                    int next_dot = new_prefix.Find(separator);
863
864                    wxString bit;
865                    if (next_dot == -1) {
866                        bit = new_prefix;
867                    } else {
868                        bit = new_prefix.Left(next_dot);
869                    }
870
871                    // Add the current tree ID to the stack.
872                    previous_ids.push(current_id);
873
874                    // Append the new item to the tree and set this as the
875                    // current branch.
876                    current_id = m_Tree->AppendItem(current_id, bit);
877                    m_Tree->SetItemData(current_id, new TreeData(NULL));
878
879                    if (next_dot == -1) break;
880
881                    new_prefix = new_prefix.Mid(next_dot + 1);
882                }
883            }
884
885            current_prefix = prefix;
886        }
887
888        // Now add the leaf.
889        wxString bit = label->GetText().AfterLast(separator);
890        assert(bit != "");
891        wxTreeItemId id = m_Tree->AppendItem(current_id, bit);
892        m_Tree->SetItemData(id, new TreeData(label));
893        label->tree_id = id;
894        // Set the colour for an item in the survey tree.
895        if (label->IsEntrance()) {
896            // Entrances are green (like entrance blobs).
897            m_Tree->SetItemTextColour(id, wxColour(0, 255, 0));
898        } else if (label->IsSurface()) {
899            // Surface stations are dark green.
900            m_Tree->SetItemTextColour(id, wxColour(49, 158, 79));
901        }
902    }
903
904    m_Tree->Expand(treeroot);
905    m_Tree->SetEnabled();
906}
907
908void MainFrm::SelectTreeItem(LabelInfo* label)
909{
910    m_Tree->SelectItem(label->tree_id);
911}
912
913void MainFrm::CentreDataset(Double xmin, Double ymin, Double zmin)
914{
915    // Centre the dataset around the origin.
916
917    Double xoff = xmin + (m_XExt / 2.0);
918    Double yoff = ymin + (m_YExt / 2.0);
919    Double zoff = zmin + (m_ZExt / 2.0);
920    m_Offsets.set(xoff, yoff, zoff);
921
922    for (int band = 0; band < NUM_DEPTH_COLOURS + 1; band++) {
923        list<PointInfo*>::iterator pos = m_Points[band].begin();
924        list<PointInfo*>::iterator end = m_Points[band].end();
925        while (pos != end) {
926            PointInfo* point = *pos++;
927            point->x -= xoff;
928            point->y -= yoff;
929            point->z -= zoff;
930        }
931    }
932
933    list<LabelInfo*>::iterator lpos = m_Labels.begin();
934    while (lpos != m_Labels.end()) {
935        LabelInfo* label = *lpos++;
936        label->x -= xoff;
937        label->y -= yoff;
938        label->z -= zoff;
939    }
940}
941
942int MainFrm::GetDepthColour(Double z)
943{
944    // Return the (0-based) depth colour band index for a z-coordinate.
945    return int(((z - m_ZMin) / (m_ZExt == 0.0 ? 1.0 : m_ZExt)) * (NUM_DEPTH_COLOURS - 1));
946}
947
948Double MainFrm::GetDepthBoundaryBetweenBands(int a, int b)
949{
950    // Return the z-coordinate of the depth colour boundary between
951    // two adjacent depth colour bands (specified by 0-based indices).
952
953    assert((a == b - 1) || (a == b + 1));
954
955    int band = (a > b) ? a : b; // boundary N lies on the bottom of band N.
956    return m_ZMin + (m_ZExt * band / (NUM_DEPTH_COLOURS - 1));
957}
958
959void MainFrm::IntersectLineWithPlane(Double x0, Double y0, Double z0,
960                                     Double x1, Double y1, Double z1,
961                                     Double z, Double& x, Double& y)
962{
963    // Find the intersection point of the line (x0, y0, z0) -> (x1, y1, z1)
964    // with the plane parallel to the xy-plane with z-axis intersection z.
965    assert(z1 - z0 != 0.0);
966
967    Double t = (z - z0) / (z1 - z0);
968    // FIXME: this assertion fails on Windows - haven't investigated but I
969    // suspect it'll be crap rounding:
970    // assert(0.0 <= t && t <= 1.0);
971
972    x = x0 + t * (x1 - x0);
973    y = y0 + t * (y1 - y0);
974}
975
976void MainFrm::SortIntoDepthBands(list<PointInfo*>& points)
977{
978    // Split legs which cross depth colouring boundaries and classify all
979    // points into the correct depth bands.
980
981    list<PointInfo*>::iterator pos = points.begin();
982    PointInfo* prev_point = NULL;
983    while (pos != points.end()) {
984        PointInfo* point = *pos++;
985        assert(point);
986
987        // If this is a leg, then check if it intersects a depth
988        // colour boundary.
989        if (point->isLine) {
990            assert(prev_point);
991            int col1 = GetDepthColour(prev_point->z);
992            int col2 = GetDepthColour(point->z);
993            if (col1 != col2) {
994                // The leg does cross at least one boundary, so split it as
995                // many times as required...
996                int inc = (col1 > col2) ? -1 : 1;
997                for (int band = col1; band != col2; band += inc) {
998                    int next_band = band + inc;
999
1000                    // Determine the z-coordinate of the boundary being
1001                    // intersected.
1002                    Double split_at_z = GetDepthBoundaryBetweenBands(band, next_band);
1003                    Double split_at_x, split_at_y;
1004
1005                    // Find the coordinates of the intersection point.
1006                    IntersectLineWithPlane(prev_point->x, prev_point->y, prev_point->z,
1007                                           point->x, point->y, point->z,
1008                                           split_at_z, split_at_x, split_at_y);
1009
1010                    // Create a new leg only as far as this point.
1011                    PointInfo* info = new PointInfo;
1012                    info->x = split_at_x;
1013                    info->y = split_at_y;
1014                    info->z = split_at_z;
1015                    info->isLine = true;
1016                    info->isSurface = point->isSurface;
1017                    m_Points[band].push_back(info);
1018
1019                    // Create a move to this point in the next band.
1020                    info = new PointInfo;
1021                    info->x = split_at_x;
1022                    info->y = split_at_y;
1023                    info->z = split_at_z;
1024                    info->isLine = false;
1025                    info->isSurface = point->isSurface;
1026                    m_Points[next_band].push_back(info);
1027
1028                    m_NumExtraLegs++;
1029                    m_NumPoints += 2;
1030                }
1031            }
1032
1033            // Add the last point of the (possibly split) leg.
1034            m_Points[col2].push_back(point);
1035        }
1036        else {
1037            // The first point, a surface point, or another move: put it in the
1038            // correct list according to depth.
1039            int band = GetDepthColour(point->z);
1040            m_Points[band].push_back(point);
1041        }
1042
1043        prev_point = point;
1044    }
1045}
1046
1047void MainFrm::OnMRUFile(wxCommandEvent& event)
1048{
1049    wxString f(m_history.GetHistoryFile(event.GetId() - wxID_FILE1));
1050    if (!f.empty()) OpenFile(f);
1051}
1052
1053void MainFrm::OpenFile(const wxString& file, wxString survey, bool delay)
1054{
1055    wxBusyCursor hourglass;
1056    if (LoadData(file, survey)) {
1057        if (wxIsAbsolutePath(m_File)) {
1058            m_history.AddFileToHistory(m_File);
1059        } else {
1060            wxString abs = wxGetCwd() + wxString(FNM_SEP_LEV) + m_File;
1061            m_history.AddFileToHistory(abs);
1062        }
1063        wxConfigBase *b = wxConfigBase::Get();
1064        m_history.Save(*b);
1065        b->Flush();
1066
1067        int x;
1068        int y;
1069        GetClientSize(&x, &y);
1070        if (x < 600)
1071            x /= 3;
1072        else if (x < 1000)
1073            x = 200;
1074        else
1075            x /= 5;
1076
1077        m_Splitter->SplitVertically(m_Panel, m_Gfx, x);
1078        m_SashPosition = x; // Save width of panel.
1079
1080        if (delay) {
1081            m_Gfx->InitialiseOnNextResize();
1082        } else {
1083            m_Gfx->Initialise();
1084        }
1085        m_Panel->Show(true);
1086
1087        m_Gfx->SetFocus();
1088    }
1089}
1090
1091//
1092//  UI event handlers
1093//
1094
1095#undef FILEDIALOG_MULTIGLOBS
1096// MS Windows supports "*.abc;*.def" natively; wxGtk supports them as of 2.3
1097#if defined(_WIN32) || wxCHECK_VERSION(2,3,0)
1098# define FILEDIALOG_MULTIGLOBS
1099#endif
1100
1101void MainFrm::OnOpen(wxCommandEvent&)
1102{
1103#ifdef __WXMOTIF__
1104    wxFileDialog dlg (this, wxString(msg(/*Select a 3d file to view*/206)), "", "",
1105                      "*.3d", wxOPEN);
1106#else
1107    wxFileDialog dlg(this, wxString(msg(/*Select a 3d file to view*/206)), "", "",
1108                     wxString::Format("%s|*.3d"
1109#ifdef FILEDIALOG_MULTIGLOBS
1110                                      ";*.3D"
1111#endif
1112#ifdef FILEDIALOG_MULTIGLOBS
1113                                      "|%s|*.plt;*.plf"
1114#ifndef _WIN32
1115                                      ";*.PLT;*.PLF"
1116#endif
1117#else
1118                                      "|%s|*.pl?" // not ideal...
1119#endif
1120                                      "|%s|*.xyz"
1121#ifdef FILEDIALOG_MULTIGLOBS
1122#ifndef _WIN32
1123                                      ";*.XYZ"
1124#endif
1125#endif
1126                                      "|%s|%s",
1127                                      msg(/*Survex 3d files*/207),
1128                                      msg(/*Compass PLT files*/324),
1129                                      msg(/*CMAP XYZ files*/325),
1130                                      msg(/*All files*/208),
1131                                      wxFileSelectorDefaultWildcardStr),
1132                     wxFD_OPEN);
1133#endif
1134    if (dlg.ShowModal() == wxID_OK) {
1135        OpenFile(dlg.GetPath());
1136    }
1137}
1138
1139void MainFrm::OnPrint(wxCommandEvent&)
1140{
1141    m_Gfx->OnPrint(m_File, m_Title, m_DateStamp);
1142}
1143
1144void MainFrm::OnPageSetup(wxCommandEvent&)
1145{
1146    wxPageSetupDialog dlg(this, wxGetApp().GetPageSetupDialogData());
1147    if (dlg.ShowModal() == wxID_OK) {
1148        wxGetApp().SetPageSetupDialogData(dlg.GetPageSetupData());
1149    }
1150}
1151
1152void MainFrm::OnExport(wxCommandEvent&)
1153{
1154    char *baseleaf = baseleaf_from_fnm(m_File.c_str());
1155    wxFileDialog dlg(this, wxString("Export as:"), "",
1156                     wxString(baseleaf),
1157                     "DXF files|*.dxf|SVG files|*.svg|Sketch files|*.sk|EPS files|*.eps|Compass PLT for use with Carto|*.plt",
1158                     wxFD_SAVE|wxFD_OVERWRITE_PROMPT);
1159    free(baseleaf);
1160    if (dlg.ShowModal() == wxID_OK) {
1161        wxString fnm = dlg.GetPath();
1162        if (!m_Gfx->OnExport(fnm, m_Title)) {
1163            wxGetApp().ReportError(wxString::Format("Couldn't write file `%s'", fnm.c_str()));
1164        }
1165    }
1166}
1167
1168void MainFrm::OnQuit(wxCommandEvent&)
1169{
1170    exit(0);
1171}
1172
1173void MainFrm::OnClose(wxCloseEvent&)
1174{
1175    exit(0);
1176}
1177
1178void MainFrm::OnAbout(wxCommandEvent&)
1179{
1180    AboutDlg dlg(this, icon_path);
1181    dlg.Centre();
1182    dlg.ShowModal();
1183}
1184
1185void MainFrm::GetColour(int band, Double& r, Double& g, Double& b) const
1186{
1187    assert(band >= 0 && band < NUM_DEPTH_COLOURS);
1188    r = Double(REDS[band]) / 255.0;
1189    g = Double(GREENS[band]) / 255.0;
1190    b = Double(BLUES[band]) / 255.0;
1191}
1192
1193void MainFrm::ClearTreeSelection()
1194{
1195    m_Tree->UnselectAll();
1196    m_Gfx->SetThere();
1197}
1198
1199void MainFrm::ClearCoords()
1200{
1201    m_Coords->SetLabel("");
1202}
1203
1204void MainFrm::SetCoords(Double x, Double y)
1205{
1206    wxString s;
1207    if (m_Gfx->m_Metric) {
1208        s.Printf(msg(/*  %d E, %d N*/338), int(x), int(y));
1209    } else {
1210        s.Printf(msg(/*  %d E, %d N*/338),
1211                 int(x / METRES_PER_FOOT), int(y / METRES_PER_FOOT));
1212    }
1213    m_Coords->SetLabel(s);
1214}
1215
1216void MainFrm::SetAltitude(Double z)
1217{
1218    wxString s;
1219    if (m_Gfx->m_Metric) {
1220        s.Printf("  %s %dm", msg(/*Altitude*/335), int(z));
1221    } else {
1222        s.Printf("  %s %dft", msg(/*Altitude*/335), int(z / METRES_PER_FOOT));
1223    }
1224    m_Coords->SetLabel(s);
1225}
1226
1227void MainFrm::ShowInfo(const LabelInfo *label)
1228{
1229    assert(m_Gfx);
1230
1231    wxString str;
1232    if (m_Gfx->m_Metric) {
1233        str.Printf(msg(/*  %d E, %d N*/338),
1234                   int(label->x + m_Offsets.getX()),
1235                   int(label->y + m_Offsets.getY()));
1236    } else {
1237        str.Printf(msg(/*  %d E, %d N*/338),
1238                   int((label->x + m_Offsets.getX()) / METRES_PER_FOOT),
1239                   int((label->y + m_Offsets.getY()) / METRES_PER_FOOT));
1240    }
1241    m_StnCoords->SetLabel(str);
1242    m_StnName->SetLabel(label->text);
1243
1244    if (m_Gfx->m_Metric) {
1245        str.Printf("  %s %dm", msg(/*Altitude*/335),
1246                   int(label->z + m_Offsets.getZ()));
1247    } else {
1248        str.Printf("  %s %dft", msg(/*Altitude*/335),
1249                   int((label->z + m_Offsets.getZ()) / METRES_PER_FOOT));
1250    }
1251    m_StnAlt->SetLabel(str);
1252    m_Gfx->SetHere(label->x, label->y, label->z);
1253
1254    wxTreeItemData* sel_wx;
1255    bool sel = m_Tree->GetSelectionData(&sel_wx);
1256    if (sel) {
1257        TreeData *data = (TreeData*) sel_wx;
1258
1259        if (data->IsStation()) {
1260            const LabelInfo* label2 = data->GetLabel();
1261            assert(label2);
1262
1263            Double x0 = label2->x;
1264            Double x1 = label->x;
1265            Double dx = x1 - x0;
1266            Double y0 = label2->y;
1267            Double y1 = label->y;
1268            Double dy = y1 - y0;
1269            Double z0 = label2->z;
1270            Double z1 = label->z;
1271            Double dz = z1 - z0;
1272
1273            Double d_horiz = sqrt(dx*dx + dy*dy);
1274            Double dr = sqrt(dx*dx + dy*dy + dz*dz);
1275
1276            Double brg = deg(atan2(dx, dy));
1277            if (brg < 0) brg += 360;
1278
1279            str.Printf(msg(/*From %s*/339), label2->text.c_str());
1280            m_Dist1->SetLabel(str);
1281            if (m_Gfx->m_Metric) {
1282                str.Printf(msg(/*  H %d%s, V %d%s*/340),
1283                           int(d_horiz), "m",
1284                           int(dz), "m");
1285            } else {
1286                str.Printf(msg(/*  H %d%s, V %d%s*/340),
1287                           int(d_horiz / METRES_PER_FOOT), "ft",
1288                           int(dz / METRES_PER_FOOT), "ft");
1289            }
1290            m_Dist2->SetLabel(str);
1291            wxString brg_unit;
1292            if (m_Gfx->m_Degrees) {
1293                brg_unit = msg(/*&deg;*/344);
1294            } else {
1295                brg *= 400.0 / 360.0;
1296                brg_unit = msg(/*grad*/345);
1297            }
1298            if (m_Gfx->m_Metric) {
1299                str.Printf(msg(/*  Dist %d%s, Brg %03d%s*/341),
1300                           int(dr), "m", int(brg), brg_unit.c_str());
1301            } else {
1302                str.Printf(msg(/*  Dist %d%s, Brg %03d%s*/341),
1303                           int(dr / METRES_PER_FOOT), "ft", int(brg),
1304                           brg_unit.c_str());
1305            }
1306            m_Dist3->SetLabel(str);
1307            m_Gfx->SetThere(x0, y0, z0);
1308        } else {
1309            m_Gfx->SetThere(); // FIXME: not in SetMouseOverStation version?
1310        }
1311    }
1312}
1313
1314void MainFrm::DisplayTreeInfo(const wxTreeItemData* item)
1315{
1316    const TreeData* data = static_cast<const TreeData*>(item);
1317    if (data && data->IsStation()) {
1318        const LabelInfo * l = data->GetLabel();
1319        ShowInfo(l);
1320        m_Gfx->SetHere(l->x, l->y, l->z);
1321    } else {
1322        m_StnName->SetLabel("");
1323        m_StnCoords->SetLabel("");
1324        m_StnAlt->SetLabel("");
1325        m_Gfx->SetHere();
1326        m_Dist1->SetLabel("");
1327        m_Dist2->SetLabel("");
1328        m_Dist3->SetLabel("");
1329        m_Gfx->SetHere();
1330    }
1331}
1332
1333void MainFrm::TreeItemSelected(wxTreeItemData* item)
1334{
1335    TreeData* data = (TreeData*) item;
1336
1337    if (data && data->IsStation()) {
1338        const LabelInfo* label = data->GetLabel();
1339        m_Gfx->CentreOn(label->x, label->y, label->z);
1340        m_Gfx->SetThere(label->x, label->y, label->z);
1341    } else {
1342        m_Gfx->SetThere();
1343    }
1344
1345    m_Dist1->SetLabel("");
1346    m_Dist2->SetLabel("");
1347    m_Dist3->SetLabel("");
1348}
1349
1350void MainFrm::OnFind(wxCommandEvent&)
1351{
1352    wxBusyCursor hourglass;
1353    // Find stations specified by a string or regular expression.
1354
1355    wxString pattern = m_FindBox->GetValue();
1356    int re_flags = wxRE_NOSUB;
1357
1358    if (true /* case insensitive */) {
1359        re_flags |= wxRE_ICASE;
1360    }
1361
1362    bool substring = true;
1363    if (m_RegexpCheckBox->GetValue()) {
1364        re_flags |= wxRE_EXTENDED;
1365    } else if (false /* simple glob-style */) {
1366        wxString pat;
1367        for (size_t i = 0; i < pattern.size(); i++) {
1368           char ch = pattern[i];
1369           // ^ only special at start; $ at end.  But this is simpler...
1370           switch (ch) {
1371            case '^': case '$': case '.': case '[': case '\\':
1372              pat += '\\';
1373              pat += ch;
1374              break;
1375            case '*':
1376              pat += ".*";
1377              substring = false;
1378              break;
1379            case '?':
1380              pat += '.';
1381              substring = false;
1382              break;
1383            default:
1384              pat += ch;
1385           }
1386        }
1387        pattern = pat;
1388        re_flags |= wxRE_BASIC;
1389    } else {
1390        wxString pat;
1391        for (size_t i = 0; i < pattern.size(); i++) {
1392           char ch = pattern[i];
1393           // ^ only special at start; $ at end.  But this is simpler...
1394           switch (ch) {
1395            case '^': case '$': case '*': case '.': case '[': case '\\':
1396              pat += '\\';
1397           }
1398           pat += ch;
1399        }
1400        pattern = pat;
1401        re_flags |= wxRE_BASIC;
1402    }
1403
1404    if (!substring) {
1405        // FIXME "0u" required to avoid compilation error with g++-3.0
1406        if (pattern.empty() || pattern[0u] != '^') pattern = '^' + pattern;
1407        // FIXME: this fails to cope with "\$" at the end of pattern...
1408        if (pattern[pattern.size() - 1] != '$') pattern += '$';
1409    }
1410
1411    wxRegEx regex;
1412    if (!regex.Compile(pattern, re_flags)) {
1413        wxString m;
1414        m.Printf(msg(/*Invalid regular expression: %s*/404), pattern.c_str());
1415        wxGetApp().ReportError(m);
1416        return;
1417    }
1418
1419    list<LabelInfo*>::iterator pos = m_Labels.begin();
1420
1421    int found = 0;
1422    while (pos != m_Labels.end()) {
1423        LabelInfo* label = *pos++;
1424
1425        if (regex.Matches(label->text)) {
1426            label->flags |= LFLAG_HIGHLIGHTED;
1427            found++;
1428        } else {
1429            label->flags &= ~LFLAG_HIGHLIGHTED;
1430        }
1431    }
1432
1433    m_Found->SetLabel(wxString::Format(msg(/*%d found*/331), found));
1434#ifdef _WIN32
1435    m_Found->Refresh(); // FIXME
1436#endif
1437    // Re-sort so highlighted points get names in preference
1438    if (found) m_Labels.sort(LabelPlotCmp(separator));
1439    m_Gfx->ForceRefresh();
1440
1441#if 0
1442    if (!found) {
1443        wxGetApp().ReportError(msg(/*No matches were found.*/328));
1444    }
1445#endif
1446
1447    m_Gfx->SetFocus();
1448}
1449
1450void MainFrm::OnHide(wxCommandEvent&)
1451{
1452    // Hide any search result highlights.
1453    m_Found->SetLabel("");
1454    list<LabelInfo*>::iterator pos = m_Labels.begin();
1455    while (pos != m_Labels.end()) {
1456        LabelInfo* label = *pos++;
1457        label->flags &= ~LFLAG_HIGHLIGHTED;
1458    }
1459    m_Gfx->ForceRefresh();
1460}
1461
1462void MainFrm::SetMouseOverStation(LabelInfo* label)
1463{
1464    if (label) {
1465        ShowInfo(label);
1466    } else {
1467        m_StnName->SetLabel("");
1468        m_StnCoords->SetLabel("");
1469        m_StnAlt->SetLabel("");
1470        m_Gfx->SetHere();
1471        m_Dist1->SetLabel("");
1472        m_Dist2->SetLabel("");
1473        m_Dist3->SetLabel("");
1474    }
1475}
1476
1477void MainFrm::OnViewSidePanel(wxCommandEvent&)
1478{
1479    ToggleSidePanel();
1480}
1481
1482void MainFrm::ToggleSidePanel()
1483{
1484    // Toggle display of the side panel.
1485
1486    assert(m_Gfx);
1487
1488    if (m_Splitter->IsSplit()) {
1489        m_SashPosition = m_Splitter->GetSashPosition(); // save width of panel
1490        m_Splitter->Unsplit(m_Panel);
1491    } else {
1492        m_Panel->Show(true);
1493        m_Gfx->Show(true);
1494        m_Splitter->SplitVertically(m_Panel, m_Gfx, m_SashPosition);
1495    }
1496}
1497
1498void MainFrm::OnViewSidePanelUpdate(wxUpdateUIEvent& ui)
1499{
1500    ui.Enable(!m_File.empty());
1501    ui.Check(m_Splitter->IsSplit());
1502}
Note: See TracBrowser for help on using the repository browser.