#include "dbpoi.h"
#include <QtGui/QApplication>
#include <QtCore/QCoreApplication>
#include <QsqlDatabase>
#include <QsqlQuery>
#include <QsqlError>
#include <QsqlDriver>
#include <qDebug>
#include <QString>
#include <QVector>
#include <QList>
#include <QFile>
#include <QSet>


int main(int argc,char *argv[])
{
	QApplication a(argc,argv);
	//dbPOI w;
	
	 //QCoreApplication a(argc,argv);
  //  QTextCodec::setCodecForCStrings(QTextCodec::codecForName("utf8"));
	QVector<int> tempFromNode;
	QVector<int> tempToNode;
	QSet <int> tempSort;

    QsqlDatabase db = QsqlDatabase::addDatabase("QsqlITE"); 
    db.setDatabaseName("NEW_AR_POI.sqlite"); // 数据库名与路径,此时是放在同目录下
	// db.setDatabaseName("offset.db"); // 数据库名与路径,此时是放在同目录下
    bool ok = db.open(); // 连接数据库,然后就可以使用了.
	if(ok){
		QsqlQuery query;
	  		if (query.exec("select * from LinkNode_Table_bark "))   //尝试列出  表的所有记录
			{  //本次查询成功
				int numRows = 0;  //询问数据库驱动，是否驱动含有某种特性 
	 
					if (db.driver()->hasFeature(QsqlDriver::QuerySize))
					{
						numRows = query.size();  //如果支持结果影响的行数，那么直接记录下来
					}
					else
					{
						query.last(); //否则定位到结果最后，qt 文档说，这个方法非常慢
						numRows = query.at() + 1;
						query.seek(-1);
					}
			}
							//尝试列出  表的所有记录
			while (query.next()) {

				int num =0;
				/*QString FNode_ID = query.value(4).toString();
				QString TNode_ID = query.value(5).toString();*/
				int FNode_ID = query.value(3).toInt();
				int TNode_ID = query.value(4).toInt();

				tempFromNode.append(FNode_ID);
				tempToNode.append(TNode_ID);				

			}
		}
		else{
		  //qDebug() << "cannot open database.";
			printf( "cannot open database.");
	}
		
		QFile file( "file.txt" );
			if ( file.open(QIODevice::WriteOnly | QIODevice::Text)) {
				QTextStream stream( &file ); 
				for ( int index =0; index < tempFromNode.size(); index ++ ){
					stream << tempFromNode.at(index);
					tempSort.insert(tempFromNode.at(index));
						tempSort.insert(tempToNode.at(index));
					stream<<"\t"<< tempToNode.at(index) <<endl ;
									}
				file.close();
			}
		
		//	qSort(tempSort.begin(),tempSort.end());
			//tempSort.unite(tempSort);
QFile filetTempSort( "tempSort.txt" );
			if ( filetTempSort.open(QIODevice::WriteOnly | QIODevice::Text)) {
				QTextStream stream( &filetTempSort ); 
				
						 foreach (const int &value,tempSort)
						 {
							 stream << value<<endl;	 
							 filetTempSort.close();
					}
	
			}

			QFile fileTp( "TP.txt" );
			if ( fileTp.open(QIODevice::WriteOnly | QIODevice::Text)) {
				QTextStream stream( &fileTp); 
				
				 foreach (const int &value,tempSort)
				 {
					 stream << value;

					 for ( int index =0; index < tempFromNode.size(); index ++ ){
						 if (tempFromNode.at(index) == value)  
							 stream << "\t"<<tempToNode.at(index);
					 }
					
					for ( int index =0; index < tempFromNode.size(); index ++ ){
						if (tempToNode.at(index) == value)	
							 stream << "\t"<<tempFromNode.at(index);
					 }
						 stream << endl;
					}
				
				 fileTp.close();
			}

		//w.show();
		//return a.exec();
		return 0;

}