itkPSMProcrustesFunction2DTest.cxx

/*=========================================================================
 *
 *  Copyright Insight Software Consortium
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *         http://www.apache.org/licenses/LICENSE-2.0.txt
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 *=========================================================================*/

#include <iostream>
#include <fstream>
#include "itkPSMProcrustesFunction.h"
#include "itkCommand.h"
#include "itkPSMProjectReader.h"

int itkPSMProcrustesFunction2DTest( int argc, char* argv[] )
{
  bool passed = true;
  double value1, value2, value3;
  itk::PSMProcrustesFunction<2>::PointType pt;
  std::string errstring = "";
  std::string output_path = "";
  std::string input_path_prefix = "";
  itk::PSMProcrustesFunction<2>::ShapeType s;
  itk::PSMProcrustesFunction<2>::ShapeListType sl;
  
  // Check for proper arguments
  if (argc < 2)
    {
    std::cout << "Wrong number of arguments. \nUse: "
              << "itkPSMProcrustesFunctionTest parameter_file [output_path] [input_path]\n"
              << "See itk::PSMParameterFileReader for documentation on the parameter file format.\n"
              <<" Note that input_path will be prefixed to any file names and paths in the xml parameter file.\n"
              << std::endl;
    return EXIT_FAILURE;
    }
  
  if (argc >2)
    {
    output_path = std::string(argv[2]);
    }
  
  if (argc >3)
    {
    input_path_prefix = std::string(argv[3]);
    }
  
  try
    {
    // Read the project parameters
    itk::PSMProjectReader::Pointer xmlreader = itk::PSMProjectReader::New();
    xmlreader->SetFileName(argv[1]);
    xmlreader->Update();
    
    itk::PSMProject::Pointer project = xmlreader->GetOutput();
    
    // Load the point files
    const std::vector<std::string> &pt_files = project->GetModel(std::string("initialization"));
    std::cout << "Reading the point files to be registered ..." << std::endl;
    for (unsigned int i = 0; i < pt_files.size(); i++)
      {
      // Read the points for this file and add as a list
      s.clear();
      // Open the ascii file.
      std::ifstream in;
      in.open( (input_path_prefix + pt_files[i]).c_str() );
      if ( !in )
        {
        errstring += "Could not open point file for input.";
        passed = false;
        break;
        }
      
      // Read all of the points, one point per line.
      while (in)
        {
        in>>value1>>value2>>value3;
        pt[0] = value1;
        pt[1] = value2;
        s.push_back(pt);
        }
      // This algorithm pushes the last point twice
      s.pop_back();
      
      // Store the shape in the list
      sl.push_back(s);
      in.close();
      std::cout << "  " << pt_files[i] << std::endl;
      }
    
    std::cout << "Done!" << std::endl;
    
    itk::PSMProcrustesFunction<2>::SimilarityTransformListType transforms;
    itk::PSMProcrustesFunction<2>::Pointer procrustes = itk::PSMProcrustesFunction<2>::New();
    procrustes->RunGeneralizedProcrustes(transforms, sl);
    
    // Check whether shapes are correctly registered by checking each point.
    // Reference shape is chosen to be the first shape in the list.
    itk::PSMProcrustesFunction<2>::ShapeType reference_shape;
    itk::PSMProcrustesFunction<2>::ShapeIteratorType it_ref;
    itk::PSMProcrustesFunction<2>::ShapeIteratorType it;
    reference_shape = sl[0];
    for (unsigned int i = 1; i < pt_files.size(); i++)
      {
      s = sl[i];
      for(it = s.begin(),it_ref = reference_shape.begin();it != s.end(),it_ref != reference_shape.end(); it++,it_ref++)
        {
        itk::PSMProcrustesFunction<2>::PointType & point1 = (*it_ref);
        itk::PSMProcrustesFunction<2>::PointType & point2 = (*it);
        // Subtract value of each point from the reference points
        for(int j = 0; j<2; j++)
          {
          float diff = point1[j] - point2[j];
          if(diff > 1e-6)
            {
            passed = false;
            break;
            }
          }
        }
      }     
      
    // Print out the outputs
    // Load the output model names
    const std::vector<std::string> &out_files = project->GetModel(std::string("optimized"));
    
    for (unsigned int i = 0; i < out_files.size(); i++)
      {            
      // Open the output file.
      std::string fname = output_path + out_files[i];
      std::ofstream out;
      out.open( fname.c_str() );
      s = sl[i];
      if ( !out )
        {
        errstring += "Could not open point file for output: ";
        }
      else
        {                
        for(itk::PSMProcrustesFunction<2>::ShapeIteratorType it = s.begin(); it != s.end(); it++)
          {
          itk::PSMProcrustesFunction<2>::PointType & point = (*it);
          // Extra point printed to allow visualizing shape in SWViewer.
          out << point[0] << " " << point[1] << " " << 0.0 << std::endl;
          }
        out.close();
        }
      }
    passed = true;
    // TODO: How to check if shapes have been registered correctly?
    }
  
  catch(itk::ExceptionObject &e)
    {
    errstring = "ITK exception with description: " + std::string(e.GetDescription())
      + std::string("\n at location:") + std::string(e.GetLocation())
      + std::string("\n in file:") + std::string(e.GetFile());
    passed = false;
    }
  catch(...)
    {
    errstring = "Unknown exception thrown";
    passed = false;
    }
  
  if (passed)
    {
    std::cout << "All tests passed" << std::endl;
    return EXIT_SUCCESS;
    }
  else
    {
    std::cout << "Test failed with the following error:" << std::endl;
    std::cout << errstring << std::endl;
    return EXIT_FAILURE;
    }
}