int main(int argc, char* argv[])
{
if(argc != 15)
{
std::cout << "Incorrect arguments! Required args:" << std::endl
<< "InputFilename OutputFilename CreateMesh? Tx Ty Tz Rx Ry Rz (R in degrees - dictated by VTK) NumThetaPoints NumPhiPoints ThetaSpan(radians) PhiSpan(radians) StoreRays? " << std::endl;
exit(-1);
}
//get the input/output filenames from the command line
std::string InputFilename = argv[1];
std::string OutputFilename = argv[2];
std::string strCreateMesh = argv[3];
std::string strTx = argv[4];
std::string strTy = argv[5];
std::string strTz = argv[6];
std::string strRx = argv[7];
std::string strRy = argv[8];
std::string strRz = argv[9];
std::string strThetaPoints = argv[10];
std::string strPhiPoints = argv[11];
std::string strThetaSpan = argv[12];
std::string strPhiSpan = argv[13];
std::string strStoreRays = argv[14];
std::cout << "InputFilename: " << InputFilename << std::endl;
std::cout << "OutputFilename: " << OutputFilename << std::endl;
//convert string to bool
std::stringstream ssCreateMesh(strCreateMesh);
bool CreateMesh;
ssCreateMesh >> CreateMesh;
std::cout << "Create Mesh? " << CreateMesh << std::endl;
//convert strings to doubles
double Tx, Ty, Tz, Rx, Ry, Rz;
std::stringstream ssTx(strTx);
ssTx >> Tx;
std::stringstream ssTy(strTy);
ssTy >> Ty;
std::stringstream ssTz(strTz);
ssTz >> Tz;
std::stringstream ssRx(strRx);
ssRx >> Rx;
std::stringstream ssRy(strRy);
ssRy >> Ry;
std::stringstream ssRz(strRz);
ssRz >> Rz;
std::cout << "Tx: " << Tx << std::endl;
std::cout << "Ty: " << Ty << std::endl;
std::cout << "Tz: " << Tz << std::endl;
std::cout << "Rx: " << Rx << std::endl;
std::cout << "Ry: " << Ry << std::endl;
std::cout << "Rz: " << Rz << std::endl;
unsigned int ThetaPoints;
unsigned int PhiPoints;
std::stringstream ssThetaPoints(strThetaPoints);
ssThetaPoints >> ThetaPoints;
std::stringstream ssPhiPoints(strPhiPoints);
ssPhiPoints >> PhiPoints;
std::cout << "Theta points: " << ThetaPoints << std::endl;
std::cout << "Phi points: " << PhiPoints << std::endl;
double ThetaSpan, PhiSpan;
std::stringstream ssThetaSpan(strThetaSpan);
ssThetaSpan >> ThetaSpan;
std::stringstream ssPhiSpan(strPhiSpan);
ssPhiSpan >> PhiSpan;
//convert string to bool
std::stringstream ssStoreRays(strStoreRays);
bool StoreRays;
ssStoreRays >> StoreRays;
std::cout << "Store Rays? " << StoreRays << std::endl;
//read the input vtp file
vtkXMLPolyDataReader* reader = vtkXMLPolyDataReader::New();
reader->SetFileName(InputFilename.c_str());
reader->Update();
//construct a vtkLidarScanner and set all of its parameters
vtkLidarScanner* Scanner = vtkLidarScanner::New();
//Scanner->WriteScanner("scanner_original.vtp");
//double testAngle = vtkMath::Pi()/4.0;
Scanner->SetPhiSpan(PhiSpan);
Scanner->SetThetaSpan(ThetaSpan);
Scanner->SetNumberOfThetaPoints(ThetaPoints);
Scanner->SetNumberOfPhiPoints(PhiPoints);
Scanner->SetStoreRays(StoreRays);
//"aim" the scanner. This is a very simple translation, but any transformation will work
vtkTransform* transform = vtkTransform::New();
transform->PostMultiply();
transform->RotateX(Rx);
transform->RotateY(Ry);
transform->RotateZ(Rz);
transform->Translate(Tx, Ty, Tz);
Scanner->SetTransform(transform);
Scanner->WriteScanner("scanner_transformed.vtp");
Scanner->MakeSphericalGrid(); //indicate to use uniform spherical spacing
Scanner->SetCreateMesh(CreateMesh);
Scanner->SetInput(reader->GetOutput());
Scanner->Update();
vtkPolyData* poly = Scanner->GetOutput();
//create a writer and write the output vtp file
vtkSmartPointer<vtkXMLPolyDataWriter> writer = vtkSmartPointer<vtkXMLPolyDataWriter>::New();
writer->SetFileName(OutputFilename.c_str());
writer->SetInput(poly);
writer->Write();
return EXIT_SUCCESS;
}
int main(int argc, char* argv[])
{
if(argc != 15)
{
std::cout << "Incorrect arguments! Required args:" << std::endl
<< "InputFilename OutputFilename CreateMesh? Tx Ty Tz Rx Ry Rz (R in degrees - dictated by VTK) NumThetaPoints NumPhiPoints ThetaSpan(radians) PhiSpan(radians) StoreRays? " << std::endl;
return EXIT_FAILURE;
}
// Get the input/output filenames from the command line
std::string inputFilename = argv[1];
std::string outputFilename = argv[2];
std::string strCreateMesh = argv[3];
std::string strTx = argv[4];
std::string strTy = argv[5];
std::string strTz = argv[6];
std::string strRx = argv[7];
std::string strRy = argv[8];
std::string strRz = argv[9];
std::string strThetaPoints = argv[10];
std::string strPhiPoints = argv[11];
std::string strThetaSpan = argv[12];
std::string strPhiSpan = argv[13];
std::string strStoreRays = argv[14];
std::cout << "InputFilename: " << inputFilename << std::endl;
std::cout << "OutputFilename: " << outputFilename << std::endl;
// Convert string to bool
std::stringstream ssCreateMesh(strCreateMesh);
bool createMesh;
ssCreateMesh >> createMesh;
std::cout << "Create Mesh? " << createMesh << std::endl;
// Convert strings to doubles
double tx, ty, tz, rx, ry, rz;
std::stringstream ssTx(strTx);
ssTx >> tx;
std::stringstream ssTy(strTy);
ssTy >> ty;
std::stringstream ssTz(strTz);
ssTz >> tz;
std::stringstream ssRx(strRx);
ssRx >> rx;
std::stringstream ssRy(strRy);
ssRy >> ry;
std::stringstream ssRz(strRz);
ssRz >> rz;
std::cout << "Tx: " << tx << std::endl;
std::cout << "Ty: " << ty << std::endl;
std::cout << "Tz: " << tz << std::endl;
std::cout << "Rx: " << rx << std::endl;
std::cout << "Ry: " << ry << std::endl;
std::cout << "Rz: " << rz << std::endl;
unsigned int thetaPoints;
unsigned int phiPoints;
std::stringstream ssThetaPoints(strThetaPoints);
ssThetaPoints >> thetaPoints;
std::stringstream ssPhiPoints(strPhiPoints);
ssPhiPoints >> phiPoints;
std::cout << "Theta points: " << thetaPoints << std::endl;
std::cout << "Phi points: " << phiPoints << std::endl;
double thetaSpan;
double phiSpan;
std::stringstream ssThetaSpan(strThetaSpan);
ssThetaSpan >> thetaSpan;
std::cout << "Theta span: " << thetaSpan << std::endl;
std::stringstream ssPhiSpan(strPhiSpan);
ssPhiSpan >> phiSpan;
std::cout << "Phi span: " << phiSpan << std::endl;
// Convert string to bool
std::stringstream ssStoreRays(strStoreRays);
bool storeRays;
ssStoreRays >> storeRays;
std::cout << "Store Rays? " << storeRays << std::endl;
// Read the input vtp file
vtkSmartPointer<vtkXMLPolyDataReader> reader =
vtkSmartPointer<vtkXMLPolyDataReader>::New();
reader->SetFileName(inputFilename.c_str());
reader->Update();
// Construct a vtkLidarScanner and set all of its parameters
vtkSmartPointer<vtkLidarScanner> scanner =
vtkSmartPointer<vtkLidarScanner>::New();
//Scanner->WriteScanner("scanner_original.vtp");
//double testAngle = vtkMath::Pi()/4.0;
scanner->SetPhiSpan(phiSpan);
scanner->SetThetaSpan(thetaSpan);
scanner->SetNumberOfThetaPoints(thetaPoints);
scanner->SetNumberOfPhiPoints(phiPoints);
scanner->SetStoreRays(storeRays);
// "Aim" the scanner. This is a very simple translation, but any transformation will work
vtkSmartPointer<vtkTransform> transform =
vtkSmartPointer<vtkTransform>::New();
transform->PostMultiply();
transform->RotateX(rx);
transform->RotateY(ry);
transform->RotateZ(rz);
transform->Translate(tx, ty, tz);
scanner->SetTransform(transform);
scanner->WriteScanner("scanner_transformed.vtp");
scanner->SetCreateMesh(createMesh);
scanner->SetInputConnection(reader->GetOutputPort());
scanner->Update();
// Create a writer and write the output vtp file
vtkSmartPointer<vtkXMLPolyDataWriter> writer =
vtkSmartPointer<vtkXMLPolyDataWriter>::New();
writer->SetFileName(outputFilename.c_str());
writer->SetInputConnection(scanner->GetOutputPort());
writer->Write();
return EXIT_SUCCESS;
}
