File: importCSV.cpp

package info (click to toggle)
libterralib 4.0.0-4
  • links: PTS
  • area: main
  • in suites: wheezy
  • size: 170,672 kB
  • sloc: cpp: 254,330; ansic: 82,910; makefile: 808; sh: 80; xml: 37
file content (202 lines) | stat: -rw-r--r-- 6,374 bytes parent folder | download | duplicates (8)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
/************************************************************************************
TerraLib - a library for developing GIS applications.
Copyright  2001-2004 INPE and Tecgraf/PUC-Rio.

This code is part of the TerraLib library.
This library is free software; you can redistribute it and/or
modify it under the terms of the GNU Lesser General Public
License as published by the Free Software Foundation; either
version 2.1 of the License, or (at your option) any later version.

You should have received a copy of the GNU Lesser General Public
License along with this library.

The authors reassure the license terms regarding the warranties.
They specifically disclaim any warranties, including, but not limited to,
the implied warranties of merchantability and fitness for a particular purpose.
The library provided hereunder is on an "as is" basis, and the authors have no
obligation to provide maintenance, support, updates, enhancements, or modifications.
In no event shall INPE and Tecgraf / PUC-Rio be held liable to any party for direct,
indirect, special, incidental, or consequential damages arising out of the use
of this library and its documentation.
*************************************************************************************/

/* 
	This file shows how to import a MID/MIF data file into a new layer
	and a CSV file as a second attribute table of the layer

	Author: Lubia Vinhas  
*/

#include <TeAsciiFile.h>
#include <TeDriverMIDMIF.h>
#include <TeMySQL.h>

int main()
{
     // Database server parameters
     string host = "localhost";
     string dbname = "DB320RC1";
     string user = "root";
     string password = "vinhas";

     // Connect to the database
     TeDatabase* db = new TeMySQL();
     if (!db->connect(host, user, password, dbname))
     {
		cout << "Error: " << db->errorMessage() << endl << endl;
		cout << "Press Enter\n";
		getchar();
		return 1;
     }

	string filename = "../data/BairrosRecife.mif";
	// Check whether there is a layer with the name BairrosRecife in the database
	TeLayer *layer = 0;
	if (db->layerExist("BairrosRecife"))
	{
		cout << "The MID/MIF file was not imported because it was previously\n";
		cout << "imported into the database!\n\n";
		if (db->loadLayerSet() == true)
		{
			// Get the layer named "BairrosRecife" from the cache
			TeLayerMap& layerMap = db->layerMap();
			map<int, TeLayer*>::iterator it;
			for (it = layerMap.begin(); it != layerMap.end(); ++it)
			{
				layer = it->second;
				if (layer->name() == "BairrosRecife")
					break;
			}
		}
	}
	else
	{
		// Import the MID/MIF file
		layer = new TeLayer("BairrosRecife", db, 0);

		// Select the filed ID_ to be the identifier of each neighbourhood
		if (TeImportMIF(layer, filename,"BairrosRecife", "ID_")) 
			cout << "MID/MIF file imported to the TerraLib database successfully!\n" << endl;
		else
		{
			cout << "Fail to import the MID/MIF file!\n";
			db->close();
			delete db;
			cout << "Press Enter\n";
			getchar();
			return 1;
		}
	}


	// Import the csv file as a second attribute table of the layer
	filename = "../data/BairrosRecife2.csv";
	if (db->tableExist("BairrosRecife2"))
	{
		cout << "The csv file was not imported because it was previously\n";
		cout << "imported as an attribute table into the database!\n\n";
	}
	else
	{
		cout << "Importing a second attribute table from a CSV file...\n";
	
		// Creates a definition of the columns contained in the CSV file
		TeAttributeList attList;
		TeAttribute column1;
		column1.rep_.name_  = "BAIRRO_ID";
		column1.rep_.type_ = TeSTRING;
		column1.rep_.isPrimaryKey_ = true;
		column1.rep_.numChar_ = 32;
		attList.push_back(column1);
	
		TeAttribute column2;
		column2.rep_.name_  = "ORDEM";
		column2.rep_.type_ = TeINT;
		attList.push_back(column2);
	
	
		TeAttribute column3;
		column3.rep_.name_  = "NOME";
		column3.rep_.type_ = TeSTRING;
		column3.rep_.numChar_ = 32;
		attList.push_back(column3);
	
		TeAttribute column4;
		column4.rep_.name_  = "NOME_PADRAO";
		column4.rep_.type_ = TeSTRING;
		column4.rep_.numChar_ = 32;
		attList.push_back(column4);
	
		TeTable attTable2("BairrosRecife2",attList, "BAIRRO_ID","BAIRRO_ID", TeAttrStatic);
		if (!layer->createAttributeTable(attTable2))
		{
			cout << "Fail to create the attribute table from the csv file in the TerraLib database!\n\n";
			db->close();
			delete db;
			cout << "Press Enter\n";
			getchar();
			return 1;
		}
	
		TeAsciiFile csvFile(filename);
		while (csvFile.isNotAtEOF())
		{
			TeTableRow trow;
			csvFile.readNStringCSV (trow, 4, ';');
			if (trow.empty())
				break;
	
			csvFile.findNewLine();
			attTable2.add(trow);
		}

		if (attTable2.size() > 0)
		{
			if (layer->saveAttributeTable(attTable2) == false)
			{
				cout << "Fail to save the attribute table generated\n";
				cout << "through the csv file in the database!\n\n";
				cout << "Press Enter\n";
				db->close();
				delete db;
				getchar();
				return 1;
			}
			attTable2.clear();
		}
		cout << "Table successfully imported!\n";
	}

	// Show some information about the layer associated to the MID/MIF file
	cout << "Layer: " << layer->name() << ", id: " << layer->id() << endl;

	// Geometries information
	cout << "Projection: " << layer->projection()->name();              
	cout << "/" << layer->projection()->datum().name();          
	TeBox bb = layer->box();
	char message[255];
	sprintf(message,"[%.15f,%.15f,%.15f,%.15f]\n",bb.x1_, bb.y1_, bb.x2_, bb.y2_);
	cout << "\nBounding box: " << message;
	cout << "Number of polygons: " << layer->nGeometries(TePOLYGONS) << endl;
	cout << "Number of lines: "    << layer->nGeometries(TeLINES) << endl;
	cout << "Number of points: "   << layer->nGeometries(TePOINTS) << endl;

	// Information of the Tables of Attributes of the Layer
	TeAttrTableVector tableVec;
	layer->getAttrTables(tableVec);
	cout << "Number of Attribute Tables: " << tableVec.size() << endl;
	
	for (unsigned int i = 0; i < tableVec.size(); ++i)
	{
		cout << "Attribute Table " << "\"" << tableVec[i].name() << "\": ";
		cout << layer->nObjects(tableVec[i].name()) << " objects" << endl << endl;
	}
	
	db->close();
	delete db;
	cout << "Press Enter\n";
	getchar();
	return 0;
}