Unverified Commit 80aa1ea6 authored by Oleg Alexandrov's avatar Oleg Alexandrov Committed by GitHub
Browse files

Add the tool usgscsm_cam_test (#373) 

* Add usgscsm_cam_test tool

* .gitignore: Do not ignore bin as there will be an executable there

* Flesh out a bit the proposed test tool

* usgscsm_cam_test: Finish loading functionality

* usgscsm_cam_test: Print note of failure if could not load a model.

* usgscsm_cam_test: Use the more generic CSM approach

* Add a test for the test program

* Move test
parent be2a14d6

.gitignore

+0 −3
Original line number Diff line number Diff line
@@ -6,9 +6,6 @@ include/csm/* 
# iPython checkpoint items

.ipynb_checkpoints



# Ignore any executables

bin/*



# Ignore any Mac stuff

.DS_Store

CMakeLists.txt

+5 −1
Original line number Diff line number Diff line
@@ -80,10 +80,13 @@ target_link_libraries(usgscsm 
                      ${ALE_TARGET}

                      nlohmann_json::nlohmann_json)



add_executable(usgscsm_cam_test bin/usgscsm_cam_test.cc)

target_link_libraries(usgscsm_cam_test

    usgscsm)



install(TARGETS usgscsm LIBRARY DESTINATION ${CMAKE_INSTALL_LIBDIR})

install(DIRECTORY ${USGSCSM_INCLUDE_DIRS} DESTINATION ${CMAKE_INSTALL_INCLUDEDIR})



install(TARGETS usgscsm_cam_test RUNTIME DESTINATION ${CMAKE_INSTALL_BINDIR})



# Optional build tests

option (USGSCSM_BUILD_TESTS "Build tests" ON)
@@ -100,4 +103,5 @@ if(USGSCSM_BUILD_TESTS) 
  include(CTest)

  enable_testing()

  add_subdirectory(tests)



endif()

bin/usgscsm_cam_test.cc

0 → 100644
+86 −0
Original line number Diff line number Diff line 
// A tool to perform some basic tests and operations on a CSM camera model.

// 

// Functionality:

//

// - Load a CSM model in ISD format.

//

// Future functionality:

// 

// - Test projecting rays from the camera to ground and vice-versa.

// - Load a CSM model state (stored in a .json file, just like

//   an ISD model).

// - Ability to export a CSM model in ISD format to a CSM model state file. 



#include <usgscsm/UsgsAstroPlugin.h>

#include <csm/RasterGM.h>

#include <UsgsAstroLsSensorModel.h>



#include <iostream>

int main(int argc, char **argv) {



  if (argc != 2) {

    std::cerr << "Usage: " << argv[0] << " <model file>" << std::endl;

    return 1;

  }



  // This is needed to trigger loading libusgscsm.so. Otherwise 0

  // plugins are detected.

  UsgsAstroLsSensorModel lsModel;



  // Load the isd

  std::string model_file = argv[1];

  csm::Isd isd(model_file);

  std::cout << "Loading model: " << model_file << std::endl;

  

  // Check if loading the model worked

  bool success = false;



  std::shared_ptr<csm::RasterGM> model;

  

  // Try all detected plugins and models for each plugin

  csm::PluginList plugins = csm::Plugin::getList();

  for (auto iter = plugins.begin(); iter != plugins.end(); iter++) {

    

    const csm::Plugin* csm_plugin = (*iter);



    std::cout << "Detected CSM plugin: " << csm_plugin->getPluginName()  << "\n";

    // For each plugin, loop through the available models.

    size_t num_models = csm_plugin->getNumModels();

    std::cout << "Number of models for this plugin: " << num_models << "\n";

    for (size_t i = 0; i < num_models; i++) {



      std::string model_name = (*iter)->getModelName(i);



      csm::WarningList* warnings = NULL;



      if (csm_plugin->canModelBeConstructedFromISD(isd, model_name)) {



        csm::Model *csm = csm_plugin->constructModelFromISD(isd, model_name, warnings);

        csm::RasterGM *modelPtr = dynamic_cast<csm::RasterGM*>(csm);



        if (modelPtr == NULL) {

          std::cerr << "Could not load correctly a CSM model of type: "

                    << model_name << "\n";

          return 1;

        } else {

          // Assign it to a smart pointer which will handle its deallocation

          model = std::shared_ptr<csm::RasterGM>(modelPtr);

          success = true;

          std::cout << "Loaded CSM model of type " << model_name

                    << " from " << model_file << ".\n";

        }

      }

    }

  }

  

  if (!success) {

    std::cerr << "Failed to load a CSM model from: " << model_file << ".\n";

    return 1;

  }



  csm::ImageVector image_size = model->getImageSize();

  std::cout << "Camera image rows and columns: "

            << image_size.samp << ' ' << image_size.line << "\n";



  return 0;

}

tests/CMakeLists.txt

+5 −0
Original line number Diff line number Diff line
@@ -17,4 +17,9 @@ else() 
    target_link_libraries(runCSMCameraModelTests usgscsm ${GTEST_LIBRARIES} ${GTEST_MAIN_LIBRARIES} pthread)

endif()



# Test the test_usgscsm_cam_test program

add_test(NAME test_usgscsm_cam_test_linescan

    COMMAND usgscsm_cam_test data/orbitalLineScan.json

    WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}/tests)

      

gtest_discover_tests(runCSMCameraModelTests WORKING_DIRECTORY ${PROJECT_SOURCE_DIR}/tests)