1 /******************************************************************************
2 *
3 * Project: Arc/Info Generate Translator
4 * Purpose: Implements OGRARCGENDataSource class
5 * Author: Even Rouault, even dot rouault at spatialys.com
6 *
7 ******************************************************************************
8 * Copyright (c) 2011-2013, Even Rouault <even dot rouault at spatialys.com>
9 *
10 * Permission is hereby granted, free of charge, to any person obtaining a
11 * copy of this software and associated documentation files (the "Software"),
12 * to deal in the Software without restriction, including without limitation
13 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
14 * and/or sell copies of the Software, and to permit persons to whom the
15 * Software is furnished to do so, subject to the following conditions:
16 *
17 * The above copyright notice and this permission notice shall be included
18 * in all copies or substantial portions of the Software.
19 *
20 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS
21 * OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
22 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
23 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMARCGENS OR OTHER
24 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
25 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
26 * DEALINGS IN THE SOFTWARE.
27 ****************************************************************************/
28
29 #include "ogr_arcgen.h"
30 #include "cpl_conv.h"
31 #include "cpl_string.h"
32
33 CPL_CVSID("$Id: ograrcgendatasource.cpp 355b41831cd2685c85d1aabe5b95665a2c6e99b7 2019-06-19 17:07:04 +0200 Even Rouault $")
34
35 /************************************************************************/
36 /* OGRARCGENDataSource() */
37 /************************************************************************/
38
OGRARCGENDataSource()39 OGRARCGENDataSource::OGRARCGENDataSource() :
40 pszName(nullptr),
41 papoLayers(nullptr),
42 nLayers(0)
43 {}
44
45 /************************************************************************/
46 /* ~OGRARCGENDataSource() */
47 /************************************************************************/
48
~OGRARCGENDataSource()49 OGRARCGENDataSource::~OGRARCGENDataSource()
50
51 {
52 for( int i = 0; i < nLayers; i++ )
53 delete papoLayers[i];
54 CPLFree( papoLayers );
55
56 CPLFree( pszName );
57 }
58
59 /************************************************************************/
60 /* TestCapability() */
61 /************************************************************************/
62
TestCapability(const char *)63 int OGRARCGENDataSource::TestCapability( const char * /* pszCap */ )
64 {
65 return FALSE;
66 }
67
68 /************************************************************************/
69 /* GetLayer() */
70 /************************************************************************/
71
GetLayer(int iLayer)72 OGRLayer *OGRARCGENDataSource::GetLayer( int iLayer )
73
74 {
75 if( iLayer < 0 || iLayer >= nLayers )
76 return nullptr;
77
78 return papoLayers[iLayer];
79 }
80
81 /************************************************************************/
82 /* Open() */
83 /************************************************************************/
84
Open(const char * pszFilename)85 int OGRARCGENDataSource::Open( const char * pszFilename )
86
87 {
88 pszName = CPLStrdup( pszFilename );
89
90 // --------------------------------------------------------------------
91 // Does this appear to be a Arc/Info generate file?
92 // --------------------------------------------------------------------
93
94 VSILFILE* fp = VSIFOpenL(pszFilename, "rb");
95 if (fp == nullptr)
96 return FALSE;
97
98 /* Go to end of file, and count the number of END keywords */
99 /* If there's 1, it is a point layer */
100 /* If there's 2, it is a linestring or polygon layer */
101 VSIFSeekL( fp, 0, SEEK_END );
102 vsi_l_offset nSize = VSIFTellL(fp);
103 if (nSize < 10)
104 {
105 VSIFCloseL(fp);
106 return FALSE;
107 }
108 char szBuffer[10+1];
109 VSIFSeekL( fp, nSize - 10, SEEK_SET );
110 VSIFReadL( szBuffer, 1, 10, fp );
111 szBuffer[10] = '\0';
112
113 VSIFSeekL( fp, 0, SEEK_SET );
114
115 const char* szPtr = szBuffer;
116 const char* szEnd = strstr(szPtr, "END");
117 if (szEnd == nullptr) szEnd = strstr(szPtr, "end");
118 if (szEnd == nullptr)
119 {
120 VSIFCloseL(fp);
121 return FALSE;
122 }
123 szPtr = szEnd + 3;
124 szEnd = strstr(szPtr, "END");
125 if (szEnd == nullptr) szEnd = strstr(szPtr, "end");
126
127 OGRwkbGeometryType eType;
128 if (szEnd == nullptr)
129 {
130 const char* pszLine = CPLReadLine2L(fp,256,nullptr);
131 if (pszLine == nullptr)
132 {
133 VSIFCloseL(fp);
134 return FALSE;
135 }
136 char** papszTokens = CSLTokenizeString2( pszLine, " ,", 0 );
137 int nTokens = CSLCount(papszTokens);
138 CSLDestroy(papszTokens);
139
140 if (nTokens == 3)
141 eType = wkbPoint;
142 else if (nTokens == 4)
143 eType = wkbPoint25D;
144 else
145 {
146 VSIFCloseL(fp);
147 return FALSE;
148 }
149 }
150 else
151 {
152 int nLineNumber = 0;
153 eType = wkbUnknown;
154 CPLString osFirstX, osFirstY;
155 CPLString osLastX, osLastY;
156 bool bIs3D = false;
157 const char* pszLine = nullptr;
158 while( (pszLine = CPLReadLine2L(fp,256,nullptr)) != nullptr )
159 {
160 nLineNumber ++;
161 if (nLineNumber == 2)
162 {
163 char** papszTokens = CSLTokenizeString2( pszLine, " ,", 0 );
164 int nTokens = CSLCount(papszTokens);
165 if (nTokens == 2 || nTokens == 3)
166 {
167 if (nTokens == 3)
168 bIs3D = true;
169 osFirstX = papszTokens[0];
170 osFirstY = papszTokens[1];
171 }
172 CSLDestroy(papszTokens);
173 if (nTokens != 2 && nTokens != 3)
174 break;
175 }
176 else if (nLineNumber > 2)
177 {
178 if (EQUAL(pszLine, "END"))
179 {
180 if (osFirstX.compare(osLastX) == 0 &&
181 osFirstY.compare(osLastY) == 0)
182 eType = bIs3D ? wkbPolygon25D : wkbPolygon;
183 else
184 eType = bIs3D ? wkbLineString25D : wkbLineString;
185 break;
186 }
187
188 char** papszTokens = CSLTokenizeString2( pszLine, " ,", 0 );
189 int nTokens = CSLCount(papszTokens);
190 if (nTokens == 2 || nTokens == 3)
191 {
192 osLastX = papszTokens[0];
193 osLastY = papszTokens[1];
194 }
195 CSLDestroy(papszTokens);
196 if (nTokens != 2 && nTokens != 3)
197 break;
198 }
199 }
200 if (eType == wkbUnknown)
201 {
202 VSIFCloseL(fp);
203 return FALSE;
204 }
205 }
206
207 VSIFSeekL( fp, 0, SEEK_SET );
208
209 nLayers = 1;
210 papoLayers = static_cast<OGRLayer**>( CPLMalloc( sizeof(OGRLayer*) ) );
211 papoLayers[0] = new OGRARCGENLayer(pszName, fp, eType);
212
213 return TRUE;
214 }
215