Added tcpService tool to Add/Remove objects through TCP
git-svn-id: http://find-object.googlecode.com/svn/trunk/find_object@420 620bd6b2-0a58-f614-fd9a-1bd335dccda9
This commit is contained in:
matlabbe
@@ -1,4 +1,5 @@
|
||||
ADD_SUBDIRECTORY( tcpClient )
|
||||
ADD_SUBDIRECTORY( tcpImagesServer )
|
||||
ADD_SUBDIRECTORY( tcpRequest )
|
||||
ADD_SUBDIRECTORY( tcpService )
|
||||
ADD_SUBDIRECTORY( similarity )
|
||||
@@ -0,0 +1,33 @@
|
||||
|
||||
SET(SRC_FILES
|
||||
main.cpp
|
||||
)
|
||||
|
||||
SET(INCLUDE_DIRS
|
||||
${CMAKE_CURRENT_SOURCE_DIR}/../../include
|
||||
${CMAKE_CURRENT_SOURCE_DIR}
|
||||
${OpenCV_INCLUDE_DIRS}
|
||||
)
|
||||
|
||||
INCLUDE(${QT_USE_FILE})
|
||||
|
||||
SET(LIBRARIES
|
||||
${OpenCV_LIBS}
|
||||
${QT_LIBRARIES}
|
||||
)
|
||||
|
||||
# Make sure the compiler can find include files from our library.
|
||||
INCLUDE_DIRECTORIES(${INCLUDE_DIRS})
|
||||
|
||||
# Add binary called "example" that is built from the source file "main.cpp".
|
||||
# The extension is automatically found.
|
||||
ADD_EXECUTABLE(tcpService ${SRC_FILES})
|
||||
TARGET_LINK_LIBRARIES(tcpService find_object ${LIBRARIES})
|
||||
|
||||
SET_TARGET_PROPERTIES( tcpService
|
||||
PROPERTIES OUTPUT_NAME ${PROJECT_PREFIX}-tcpService)
|
||||
|
||||
INSTALL(TARGETS tcpService
|
||||
RUNTIME DESTINATION bin COMPONENT runtime
|
||||
BUNDLE DESTINATION "${CMAKE_BUNDLE_LOCATION}" COMPONENT runtime)
|
||||
|
||||
@@ -0,0 +1,208 @@
|
||||
/*
|
||||
Copyright (c) 2011-2014, Mathieu Labbe - IntRoLab - Universite de Sherbrooke
|
||||
All rights reserved.
|
||||
|
||||
Redistribution and use in source and binary forms, with or without
|
||||
modification, are permitted provided that the following conditions are met:
|
||||
* Redistributions of source code must retain the above copyright
|
||||
notice, this list of conditions and the following disclaimer.
|
||||
* Redistributions in binary form must reproduce the above copyright
|
||||
notice, this list of conditions and the following disclaimer in the
|
||||
documentation and/or other materials provided with the distribution.
|
||||
* Neither the name of the Universite de Sherbrooke nor the
|
||||
names of its contributors may be used to endorse or promote products
|
||||
derived from this software without specific prior written permission.
|
||||
|
||||
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
||||
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
||||
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
||||
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY
|
||||
DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
||||
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
||||
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
|
||||
ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
||||
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
||||
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
||||
*/
|
||||
|
||||
#include <QtNetwork/QNetworkInterface>
|
||||
#include <QtNetwork/QTcpSocket>
|
||||
#include <QtCore/QCoreApplication>
|
||||
#include <QtCore/QFile>
|
||||
#include <QtCore/QFileInfo>
|
||||
#include <QtCore/QTime>
|
||||
#include <opencv2/opencv.hpp>
|
||||
#include <find_object/TcpServer.h>
|
||||
|
||||
void showUsage()
|
||||
{
|
||||
printf("\ntcpService [options] port\n"
|
||||
" Options:\n"
|
||||
" --add \"image.png\" # Add object (file name + id). Set id=0 "
|
||||
" will make the server generating an id.\n"
|
||||
" --remove # Remove object by ID.\n"
|
||||
" --host #.#.#.# Set host address.\n"
|
||||
" --help Show this help.\n"
|
||||
" Examples:\n"
|
||||
" Add: $ tcpService --add image.png 1 --host 127.0.0.1 4000\n"
|
||||
" Remove: $ tcpService --remove 1 --host 127.0.0.1 4000\n");
|
||||
exit(-1);
|
||||
}
|
||||
|
||||
int main(int argc, char * argv[])
|
||||
{
|
||||
QString ipAddress;
|
||||
QString fileName;
|
||||
int addId = 0;
|
||||
int removeId = -1;
|
||||
quint16 port = 0;
|
||||
|
||||
for(int i=1; i<argc-1; ++i)
|
||||
{
|
||||
if(strcmp(argv[i], "--host") == 0 || strcmp(argv[i], "-host") == 0)
|
||||
{
|
||||
++i;
|
||||
if(i < argc)
|
||||
{
|
||||
ipAddress = argv[i];
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("error parsing --host\n");
|
||||
showUsage();
|
||||
}
|
||||
continue;
|
||||
}
|
||||
if(strcmp(argv[i], "--add") == 0 || strcmp(argv[i], "-add") == 0)
|
||||
{
|
||||
++i;
|
||||
if(i < argc-1)
|
||||
{
|
||||
fileName = argv[i];
|
||||
++i;
|
||||
if(i < argc)
|
||||
{
|
||||
addId = atoi(argv[i]);
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("error parsing --add\n");
|
||||
}
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("error parsing --add\n");
|
||||
showUsage();
|
||||
}
|
||||
continue;
|
||||
}
|
||||
if(strcmp(argv[i], "--remove") == 0 || strcmp(argv[i], "-remove") == 0)
|
||||
{
|
||||
++i;
|
||||
if(i < argc)
|
||||
{
|
||||
removeId = std::atoi(argv[i]);
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("error parsing --remove\n");
|
||||
showUsage();
|
||||
}
|
||||
continue;
|
||||
}
|
||||
if(strcmp(argv[i], "-help") == 0 ||
|
||||
strcmp(argv[i], "--help") == 0)
|
||||
{
|
||||
showUsage();
|
||||
}
|
||||
|
||||
printf("Unrecognized option: %s\n", argv[i]);
|
||||
showUsage();
|
||||
}
|
||||
|
||||
port = atoi(argv[argc-1]);
|
||||
|
||||
if(fileName.isEmpty() && removeId == -1)
|
||||
{
|
||||
printf("Arguments --add or --remove should be set.\n");
|
||||
showUsage();
|
||||
}
|
||||
else if(port <=0)
|
||||
{
|
||||
printf("Port should be set!\n");
|
||||
showUsage();
|
||||
}
|
||||
|
||||
if(ipAddress.isEmpty())
|
||||
{
|
||||
ipAddress = QHostAddress(QHostAddress::LocalHost).toString();
|
||||
}
|
||||
|
||||
cv::Mat image;
|
||||
if(!fileName.isEmpty())
|
||||
{
|
||||
image = cv::imread(fileName.toStdString());
|
||||
if(image.empty())
|
||||
{
|
||||
printf("Cannot read image from \"%s\".\n", fileName.toStdString().c_str());
|
||||
showUsage();
|
||||
}
|
||||
fileName = QFileInfo(fileName).fileName();
|
||||
}
|
||||
|
||||
QCoreApplication app(argc, argv);
|
||||
QTcpSocket request;
|
||||
|
||||
QObject::connect(&request, SIGNAL(disconnected()), &app, SLOT(quit()));
|
||||
QObject::connect(&request, SIGNAL(error(QAbstractSocket::SocketError)), &app, SLOT(quit()));
|
||||
QObject::connect(&request, SIGNAL(bytesWritten(quint64)), &app, SLOT(quit()));
|
||||
|
||||
printf("Connecting to \"%s:%d\"...\n", ipAddress.toStdString().c_str(), port);
|
||||
request.connectToHost(ipAddress, port);
|
||||
|
||||
if(!request.waitForConnected())
|
||||
{
|
||||
printf("Connecting to \"%s:%d\"... connection failed!\n", ipAddress.toStdString().c_str(), port);
|
||||
return -1;
|
||||
}
|
||||
else
|
||||
{
|
||||
printf("Connecting to \"%s:%d\"... connected!\n", ipAddress.toStdString().c_str(), port);
|
||||
}
|
||||
|
||||
QByteArray block;
|
||||
QDataStream out(&block, QIODevice::WriteOnly);
|
||||
out.setVersion(QDataStream::Qt_4_0);
|
||||
out << (quint64)0;
|
||||
|
||||
if(!image.empty())
|
||||
{
|
||||
// publish image
|
||||
std::vector<unsigned char> buf;
|
||||
cv::imencode(".png", image, buf);
|
||||
|
||||
out << (quint32)find_object::TcpServer::kAddObject;
|
||||
out << addId;
|
||||
out << fileName;
|
||||
quint64 imageSize = buf.size();
|
||||
out << imageSize;
|
||||
out.writeRawData((char*)buf.data(), (int)buf.size());
|
||||
printf("Add object %d \"%s\"\n", addId, fileName.toStdString().c_str());
|
||||
}
|
||||
else if(removeId)
|
||||
{
|
||||
out << (quint32)find_object::TcpServer::kRemoveObject;
|
||||
out << removeId;
|
||||
printf("Remove object %d\n", removeId);
|
||||
}
|
||||
|
||||
out.device()->seek(0);
|
||||
out << (quint64)(block.size() - sizeof(quint64));
|
||||
|
||||
qint64 bytes = request.write(block);
|
||||
printf("Service published (%d bytes)!\n", (int)bytes);
|
||||
request.waitForReadyRead();
|
||||
|
||||
return 0;
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user