You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
346 lines
8.2 KiB
346 lines
8.2 KiB
/*=========================================================================
|
|
|
|
Program: Visualization Toolkit
|
|
Module: $RCSfile: SocketClient.cxx,v $
|
|
|
|
Copyright (c) Ken Martin, Will Schroeder, Bill Lorensen
|
|
All rights reserved.
|
|
See Copyright.txt or http://www.kitware.com/Copyright.htm for details.
|
|
|
|
This software is distributed WITHOUT ANY WARRANTY; without even
|
|
the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
|
|
PURPOSE. See the above copyright notice for more information.
|
|
|
|
=========================================================================*/
|
|
#include "vtkActor.h"
|
|
#include "vtkContourFilter.h"
|
|
#include "vtkDataSetMapper.h"
|
|
#include "vtkDebugLeaks.h"
|
|
#include "vtkDoubleArray.h"
|
|
#include "vtkPolyData.h"
|
|
#include "vtkPolyDataMapper.h"
|
|
#include "vtkRectilinearGrid.h"
|
|
#include "vtkRegressionTestImage.h"
|
|
#include "vtkRenderWindow.h"
|
|
#include "vtkRenderer.h"
|
|
#include "vtkSocketCommunicator.h"
|
|
#include "vtkSocketController.h"
|
|
#include "vtkStructuredGrid.h"
|
|
#include "vtkImageData.h"
|
|
#include "vtkUnstructuredGrid.h"
|
|
#include "vtkCamera.h"
|
|
#include "vtkImageActor.h"
|
|
|
|
#include "vtkRenderWindowInteractor.h"
|
|
|
|
static const int scMsgLength = 10;
|
|
|
|
static void CleanUp(vtkSocketCommunicator* comm, vtkSocketController* contr)
|
|
{
|
|
// This will close the connection as well as delete
|
|
// the communicator
|
|
comm->Delete();
|
|
contr->Delete();
|
|
}
|
|
|
|
int main(int argc, char** argv)
|
|
{
|
|
vtkSocketController* contr = vtkSocketController::New();
|
|
contr->Initialize();
|
|
|
|
vtkSocketCommunicator* comm = vtkSocketCommunicator::New();
|
|
|
|
int i;
|
|
|
|
// Get the host name from the command line arguments
|
|
char* hostname = new char[30];
|
|
strcpy(hostname, "localhost");
|
|
int dataIndex=-1;
|
|
for (i=0; i<argc; i++)
|
|
{
|
|
if ( strcmp("-H", argv[i]) == 0 )
|
|
{
|
|
if ( i < argc-1 )
|
|
{
|
|
dataIndex = i+1;
|
|
}
|
|
}
|
|
}
|
|
if (dataIndex != -1)
|
|
{
|
|
delete[] hostname;
|
|
hostname = new char[strlen(argv[dataIndex])+1];
|
|
strcpy(hostname, argv[dataIndex]);
|
|
}
|
|
|
|
// Get the port from the command line arguments
|
|
int port=11111;
|
|
|
|
dataIndex=-1;
|
|
for (i=0; i<argc; i++)
|
|
{
|
|
if ( strcmp("-P", argv[i]) == 0 )
|
|
{
|
|
if ( i < argc-1 )
|
|
{
|
|
dataIndex = i+1;
|
|
}
|
|
}
|
|
}
|
|
if (dataIndex != -1)
|
|
{
|
|
port = atoi(argv[dataIndex]);
|
|
}
|
|
|
|
// Establish connection
|
|
if (!comm->ConnectTo(hostname, port))
|
|
{
|
|
cerr << "Client error: Could not connect to the server."
|
|
<< endl;
|
|
comm->Delete();
|
|
contr->Delete();
|
|
delete[] hostname;
|
|
return 1;
|
|
}
|
|
delete[] hostname;
|
|
|
|
// Test sending all supported types of arrays
|
|
int datai[scMsgLength];
|
|
for (i=0; i<scMsgLength; i++)
|
|
{
|
|
datai[i] = i;
|
|
}
|
|
if (!comm->Send(datai, scMsgLength, 1, 11))
|
|
{
|
|
cerr << "Client error: Error sending data." << endl;
|
|
CleanUp(comm, contr);
|
|
return 1;
|
|
}
|
|
|
|
unsigned long dataul[scMsgLength];
|
|
for (i=0; i<scMsgLength; i++)
|
|
{
|
|
dataul[i] = static_cast<unsigned long>(i);
|
|
}
|
|
if (!comm->Send(dataul, scMsgLength, 1, 22))
|
|
{
|
|
cerr << "Client error: Error sending data." << endl;
|
|
CleanUp(comm, contr);
|
|
return 1;
|
|
}
|
|
|
|
char datac[scMsgLength];
|
|
for (i=0; i<scMsgLength; i++)
|
|
{
|
|
datac[i] = static_cast<char>(i);
|
|
}
|
|
if (!comm->Send(datac, scMsgLength, 1, 33))
|
|
{
|
|
cerr << "Client error: Error sending data." << endl;
|
|
CleanUp(comm, contr);
|
|
return 1;
|
|
}
|
|
|
|
unsigned char datauc[scMsgLength];
|
|
for (i=0; i<scMsgLength; i++)
|
|
{
|
|
datauc[i] = static_cast<unsigned char>(i);
|
|
}
|
|
if (!comm->Send(datauc, scMsgLength, 1, 44))
|
|
{
|
|
cerr << "Client error: Error sending data." << endl;
|
|
CleanUp(comm, contr);
|
|
return 1;
|
|
}
|
|
|
|
float dataf[scMsgLength];
|
|
for (i=0; i<scMsgLength; i++)
|
|
{
|
|
dataf[i] = static_cast<float>(i);
|
|
}
|
|
if (!comm->Send(dataf, scMsgLength, 1, 7))
|
|
{
|
|
cerr << "Client error: Error sending data." << endl;
|
|
CleanUp(comm, contr);
|
|
return 1;
|
|
}
|
|
|
|
|
|
double datad[scMsgLength];
|
|
for (i=0; i<scMsgLength; i++)
|
|
{
|
|
datad[i] = static_cast<double>(i);
|
|
}
|
|
if (!comm->Send(datad, scMsgLength, 1, 7))
|
|
{
|
|
cerr << "Client error: Error sending data." << endl;
|
|
CleanUp(comm, contr);
|
|
return 1;
|
|
}
|
|
|
|
vtkIdType datait[scMsgLength];
|
|
for (i=0; i<scMsgLength; i++)
|
|
{
|
|
datait[i] = static_cast<vtkIdType>(i);
|
|
}
|
|
if (!comm->Send(datait, scMsgLength, 1, 7))
|
|
{
|
|
cerr << "Client error: Error sending data." << endl;
|
|
CleanUp(comm, contr);
|
|
return 1;
|
|
}
|
|
|
|
// Test receiving vtkDataObject
|
|
vtkUnstructuredGrid* ugrid = vtkUnstructuredGrid::New();
|
|
|
|
if (!comm->Receive(ugrid, 1, 9))
|
|
{
|
|
cerr << "Client error: Error receiving data." << endl;
|
|
CleanUp(comm, contr);
|
|
ugrid->Delete();
|
|
return 1;
|
|
}
|
|
|
|
vtkDataSetMapper* umapper = vtkDataSetMapper::New();
|
|
umapper->SetInput(ugrid);
|
|
|
|
vtkActor* uactor = vtkActor::New();
|
|
uactor->SetMapper(umapper);
|
|
uactor->SetPosition(5, 0, 0);
|
|
uactor->SetScale(0.2, 0.2, 0.2);
|
|
umapper->UnRegister(0);
|
|
|
|
// Test receiving vtkDataArray
|
|
vtkDoubleArray* da = vtkDoubleArray::New();
|
|
if (!comm->Receive(da, 1, 9))
|
|
{
|
|
cerr << "Client error: Error receiving data." << endl;
|
|
CleanUp(comm, contr);
|
|
ugrid->Delete();
|
|
uactor->Delete();
|
|
da->Delete();
|
|
return 1;
|
|
}
|
|
for (i=0; i<40; i++)
|
|
{
|
|
if (da->GetValue(i) != static_cast<double>(i))
|
|
{
|
|
cerr << "Server error: Corrupt vtkDoubleArray." << endl;
|
|
CleanUp(comm, contr);
|
|
ugrid->Delete();
|
|
uactor->Delete();
|
|
da->Delete();
|
|
return 1;
|
|
}
|
|
}
|
|
|
|
da->Delete();
|
|
|
|
// Test receiving null vtkDataArray
|
|
vtkDoubleArray *da2 = vtkDoubleArray::New();
|
|
if (!comm->Receive(da2, 1, 9))
|
|
{
|
|
cerr << "Client error: Error receiving data." << endl;
|
|
CleanUp(comm, contr);
|
|
ugrid->Delete();
|
|
uactor->Delete();
|
|
da2->Delete();
|
|
return 1;
|
|
}
|
|
if (da2->GetNumberOfTuples() == 0)
|
|
{
|
|
cout << "receive null data array successful" << endl;
|
|
}
|
|
else
|
|
{
|
|
cout << "receive null data array failed" << endl;
|
|
}
|
|
da2->Delete();
|
|
|
|
contr->SetCommunicator(comm);
|
|
|
|
// The following lines were added for coverage
|
|
// These methods have empty implementations
|
|
contr->SingleMethodExecute();
|
|
contr->MultipleMethodExecute();
|
|
contr->CreateOutputWindow();
|
|
contr->Barrier();
|
|
contr->Finalize();
|
|
|
|
vtkPolyDataMapper* pmapper = vtkPolyDataMapper::New();
|
|
vtkPolyData* pd = vtkPolyData::New();
|
|
comm->Receive(pd, 1, 11);
|
|
pmapper->SetInput(pd);
|
|
pd->Delete();
|
|
|
|
vtkActor* pactor = vtkActor::New();
|
|
pactor->SetMapper(pmapper);
|
|
pmapper->UnRegister(0);
|
|
|
|
vtkDataSetMapper* rgmapper = vtkDataSetMapper::New();
|
|
vtkRectilinearGrid* rg = vtkRectilinearGrid::New();
|
|
comm->Receive(rg, 1, 11);
|
|
rgmapper->SetInput(rg);
|
|
rg->Delete();
|
|
|
|
vtkActor* rgactor = vtkActor::New();
|
|
rgactor->SetMapper(rgmapper);
|
|
rgactor->SetPosition(0, -5, 0);
|
|
rgactor->SetScale(2, 2, 2);
|
|
rgmapper->UnRegister(0);
|
|
|
|
vtkContourFilter* iso2 = vtkContourFilter::New();
|
|
vtkStructuredGrid* sg = vtkStructuredGrid::New();
|
|
comm->Receive(sg, 1, 11);
|
|
iso2->SetInput(sg);
|
|
sg->Delete();
|
|
iso2->SetValue(0, .205);
|
|
|
|
vtkPolyDataMapper* sgmapper = vtkPolyDataMapper::New();
|
|
sgmapper->SetInputConnection(0, iso2->GetOutputPort());
|
|
iso2->UnRegister(0);
|
|
|
|
vtkActor* sgactor = vtkActor::New();
|
|
sgactor->SetMapper(sgmapper);
|
|
sgactor->SetPosition(10, -5, -40);
|
|
sgmapper->UnRegister(0);
|
|
|
|
|
|
vtkImageData* id = vtkImageData::New();
|
|
comm->Receive(id, 1, 11);
|
|
|
|
vtkImageActor* imactor = vtkImageActor::New();
|
|
imactor->SetInput(id);
|
|
id->Delete();
|
|
imactor->SetPosition(10, 0, 10);
|
|
imactor->SetScale(0.02, 0.02, 0.02);
|
|
|
|
vtkRenderer* ren = vtkRenderer::New();
|
|
ren->AddActor(uactor);
|
|
ren->AddActor(pactor);
|
|
ren->AddActor(rgactor);
|
|
ren->AddActor(sgactor);
|
|
ren->AddActor(imactor);
|
|
uactor->UnRegister(0);
|
|
pactor->UnRegister(0);
|
|
rgactor->UnRegister(0);
|
|
sgactor->UnRegister(0);
|
|
imactor->UnRegister(0);
|
|
|
|
vtkRenderWindow* renWin = vtkRenderWindow::New();
|
|
renWin->SetSize(500,400);
|
|
renWin->AddRenderer(ren);
|
|
ren->ResetCamera();
|
|
ren->GetActiveCamera()->Zoom(2.2);
|
|
ren->UnRegister(0);
|
|
|
|
renWin->Render();
|
|
|
|
int retVal = vtkRegressionTestImage( renWin );
|
|
|
|
renWin->Delete();
|
|
ugrid->Delete();
|
|
CleanUp(comm, contr);
|
|
|
|
return !retVal;
|
|
}
|
|
|