source: git/src/model.h @ 46a9883

Last change on this file since 46a9883 was 46a9883, checked in by Olly Betts <olly@…>, 11 months ago

survexport: Fix exporting of passages and xsects

We now run the code which decides how much to rotate each cross-section
when exporting using survexport. Previous all cross-sections were
aligned West-East. Reported by Robert Jones.

  • Property mode set to 100644
File size: 8.0 KB
Line 
1//
2//  model.h
3//
4//  Cave survey model.
5//
6//  Copyright (C) 2000-2003,2005 Mark R. Shinwell
7//  Copyright (C) 2001-2003,2004,2005,2006,2010,2011,2012,2013,2014,2015,2016,2018 Olly Betts
8//
9//  This program is free software; you can redistribute it and/or modify
10//  it under the terms of the GNU General Public License as published by
11//  the Free Software Foundation; either version 2 of the License, or
12//  (at your option) any later version.
13//
14//  This program is distributed in the hope that it will be useful,
15//  but WITHOUT ANY WARRANTY; without even the implied warranty of
16//  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
17//  GNU General Public License for more details.
18//
19//  You should have received a copy of the GNU General Public License
20//  along with this program; if not, write to the Free Software
21//  Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301  USA
22//
23
24#ifndef model_h
25#define model_h
26
27#include "wx.h"
28
29#include "labelinfo.h"
30#include "vector3.h"
31
32#include <ctime>
33#include <list>
34#include <set>
35#include <vector>
36
37using namespace std;
38
39class MainFrm;
40
41class PointInfo : public Point {
42    int date;
43
44public:
45    PointInfo() : Point(), date(-1) { }
46    explicit PointInfo(const img_point & pt) : Point(pt), date(-1) { }
47    PointInfo(const img_point & pt, int date_) : Point(pt), date(date_) { }
48    PointInfo(const Point & p, int date_) : Point(p), date(date_) { }
49    int GetDate() const { return date; }
50};
51
52class XSect {
53    friend class MainFrm;
54    const LabelInfo* stn;
55    int date;
56    double l, r, u, d;
57    double right_bearing;
58
59public:
60    XSect(const LabelInfo* stn_, int date_,
61          double l_, double r_, double u_, double d_)
62        : stn(stn_), date(date_), l(l_), r(r_), u(u_), d(d_), right_bearing(0) { }
63    double GetL() const { return l; }
64    double GetR() const { return r; }
65    double GetU() const { return u; }
66    double GetD() const { return d; }
67    double get_right_bearing() const { return right_bearing; }
68    void set_right_bearing(double right_bearing_) {
69        right_bearing = right_bearing_;
70    }
71    int GetDate() const { return date; }
72    const wxString& GetLabel() const { return stn->GetText(); }
73    const Point& GetPoint() const { return *stn; }
74    double GetX() const { return stn->GetX(); }
75    double GetY() const { return stn->GetY(); }
76    double GetZ() const { return stn->GetZ(); }
77    friend Vector3 operator-(const XSect& a, const XSect& b);
78};
79
80inline Vector3 operator-(const XSect& a, const XSect& b) {
81    return *(a.stn) - *(b.stn);
82}
83
84class traverse : public vector<PointInfo> {
85  public:
86    int n_legs;
87    // Bitmask of img_FLAG_SURFACE, img_FLAG_SPLAY and img_FLAG_DUPLICATE.
88    int flags;
89    double length;
90    double E, H, V;
91    wxString name;
92
93    explicit
94    traverse(const char* name_)
95        : n_legs(0), flags(0),
96          length(0), E(-1), H(-1), V(-1),
97          name(name_, wxConvUTF8) {
98        if (name.empty() && !name_[0]) {
99            // If name isn't valid UTF-8 then this conversion will
100            // give an empty string.  In this case, assume that the
101            // label is CP1252 (the Microsoft superset of ISO8859-1).
102            static wxCSConv ConvCP1252(wxFONTENCODING_CP1252);
103            name = wxString(name_, ConvCP1252);
104            if (name.empty()) {
105                // Or if that doesn't work (ConvCP1252 doesn't like
106                // strings with some bytes in) let's just go for
107                // ISO8859-1.
108                name = wxString(name_, wxConvISO8859_1);
109            }
110        }
111    }
112};
113
114class SurveyFilter {
115    std::set<wxString, std::greater<wxString>> filters;
116    std::set<wxString, std::greater<wxString>> redundant_filters;
117    // Default to the Survex standard separator - then a filter created before
118    // the survey separator is known is likely to not need rebuilding.
119    wxChar separator = '.';
120
121  public:
122    SurveyFilter() {}
123
124    void add(const wxString& survey);
125
126    void remove(const wxString& survey);
127
128    void clear() { filters.clear(); redundant_filters.clear(); }
129
130    bool empty() const { return filters.empty(); }
131
132    void SetSeparator(wxChar separator_);
133
134    bool CheckVisible(const wxString& name) const;
135};
136
137/// Cave model.
138class Model {
139    list<traverse> traverses[8];
140    mutable list<vector<XSect>> tubes;
141
142  public: // FIXME
143    list<LabelInfo*> m_Labels;
144
145  private:
146    Vector3 m_Ext;
147    double m_DepthMin, m_DepthExt;
148    int m_DateMin, m_DateExt;
149    bool complete_dateinfo;
150    int m_NumEntrances = 0;
151    int m_NumFixedPts = 0;
152    int m_NumExportedPts = 0;
153    bool m_HasUndergroundLegs = false;
154    bool m_HasSplays = false;
155    bool m_HasDupes = false;
156    bool m_HasSurfaceLegs = false;
157    bool m_HasErrorInformation = false;
158    bool m_IsExtendedElevation = false;
159    mutable bool m_TubesPrepared = false;
160
161    // Character separating survey levels (often '.')
162    wxChar m_separator;
163
164    wxString m_Title, m_cs_proj, m_DateStamp;
165
166    time_t m_DateStamp_numeric;
167
168    Vector3 m_Offset;
169
170    void do_prepare_tubes() const;
171
172  public:
173    int Load(const wxString& file, const wxString& prefix);
174
175    void CentreDataset(const Vector3& vmin);
176
177    const Vector3& GetExtent() const { return m_Ext; }
178
179    const wxString& GetSurveyTitle() const { return m_Title; }
180
181    const wxString& GetDateString() const { return m_DateStamp; }
182
183    time_t GetDateStamp() const { return m_DateStamp_numeric; }
184
185    double GetDepthExtent() const { return m_DepthExt; }
186    double GetDepthMin() const { return m_DepthMin; }
187
188    bool HasCompleteDateInfo() const { return complete_dateinfo; }
189    int GetDateExtent() const { return m_DateExt; }
190    int GetDateMin() const { return m_DateMin; }
191
192    int GetNumFixedPts() const { return m_NumFixedPts; }
193    int GetNumExportedPts() const { return m_NumExportedPts; }
194    int GetNumEntrances() const { return m_NumEntrances; }
195
196    bool HasUndergroundLegs() const { return m_HasUndergroundLegs; }
197    bool HasSplays() const { return m_HasSplays; }
198    bool HasDupes() const { return m_HasDupes; }
199    bool HasSurfaceLegs() const { return m_HasSurfaceLegs; }
200    bool HasTubes() const { return !tubes.empty(); }
201    bool HasErrorInformation() const { return m_HasErrorInformation; }
202
203    bool IsExtendedElevation() const { return m_IsExtendedElevation; }
204
205    wxChar GetSeparator() const { return m_separator; }
206
207    const wxString& GetCSProj() const { return m_cs_proj; }
208
209    const Vector3& GetOffset() const { return m_Offset; }
210
211    list<traverse>::const_iterator
212    traverses_begin(unsigned flags, const SurveyFilter* filter) const {
213        if (flags >= sizeof(traverses)) return traverses[0].end();
214        auto it = traverses[flags].begin();
215        if (filter) {
216            while (it != traverses[flags].end() &&
217                   !filter->CheckVisible(it->name)) {
218                ++it;
219            }
220        }
221        return it;
222    }
223
224    list<traverse>::const_iterator
225    traverses_next(unsigned flags, const SurveyFilter* filter,
226                   list<traverse>::const_iterator it) const {
227        ++it;
228        if (filter) {
229            while (it != traverses[flags].end() &&
230                   !filter->CheckVisible(it->name)) {
231                ++it;
232            }
233        }
234        return it;
235    }
236
237    list<traverse>::const_iterator traverses_end(unsigned flags) const {
238        if (flags >= sizeof(traverses)) flags = 0;
239        return traverses[flags].end();
240    }
241
242    list<vector<XSect>>::const_iterator tubes_begin() const {
243        prepare_tubes();
244        return tubes.begin();
245    }
246
247    list<vector<XSect>>::const_iterator tubes_end() const {
248        return tubes.end();
249    }
250
251    list<vector<XSect>>::iterator tubes_begin() {
252        prepare_tubes();
253        return tubes.begin();
254    }
255
256    list<vector<XSect>>::iterator tubes_end() {
257        return tubes.end();
258    }
259
260    list<LabelInfo*>::const_iterator GetLabels() const {
261        return m_Labels.begin();
262    }
263
264    list<LabelInfo*>::const_iterator GetLabelsEnd() const {
265        return m_Labels.end();
266    }
267
268    list<LabelInfo*>::const_reverse_iterator GetRevLabels() const {
269        return m_Labels.rbegin();
270    }
271
272    list<LabelInfo*>::const_reverse_iterator GetRevLabelsEnd() const {
273        return m_Labels.rend();
274    }
275
276    list<LabelInfo*>::iterator GetLabelsNC() {
277        return m_Labels.begin();
278    }
279
280    list<LabelInfo*>::iterator GetLabelsNCEnd() {
281        return m_Labels.end();
282    }
283
284    void prepare_tubes() const {
285        if (!m_TubesPrepared) {
286            do_prepare_tubes();
287            m_TubesPrepared = true;
288        }
289    }
290};
291
292#endif
Note: See TracBrowser for help on using the repository browser.