Initial commit

.gitignore

 build/*
+*.pyc
+__pycache__/*
+.pytest_cache/*

CMakeLists.txt

+cmake_minimum_required(VERSION 3.10)
+
+find_package(SWIG REQUIRED)
+include(${SWIG_USE_FILE})
+
+add_subdirectory(python)
\ No newline at end of file

csm.i

+%module(package="csmapi") csm
+%{
+    #include "csm.h"
+%}
+
+%ignore CSM_UNKNOWN;
+%ignore CSM_SENSOR_TYPE_UNKNOWN;     
+%ignore CSM_SENSOR_TYPE_EO;            
+%ignore CSM_SENSOR_TYPE_IR;            
+%ignore CSM_SENSOR_TYPE_MWIR;          
+%ignore CSM_SENSOR_TYPE_LWIR;          
+%ignore CSM_SENSOR_TYPE_SAR;           
+%ignore CSM_SENSOR_TYPE_EOIRSC;        
+%ignore CSM_SENSOR_MODE_UNKNOWN;       
+%ignore CSM_SENSOR_MODE_FRAME;        
+%ignore CSM_SENSOR_MODE_PULSE;         
+%ignore CSM_SENSOR_MODE_PB;            
+%ignore CSM_SENSOR_MODE_WB;            
+%ignore CSM_SENSOR_MODE_SPOT;          
+%ignore CSM_SENSOR_MODE_STRIP;         
+%ignore CSM_SENSOR_MODE_SCAN;          
+%ignore CSM_SENSOR_MODE_VIDEO;         
+%ignore CSM_SENSOR_MODE_BODY_POINTING; 
+
+%include "csm.h"

csmapi.i

+%module csmapi
+%include "csm.i"
+%include "version.i"
+%include "ellipsoid.i"
+%include "warning.i"
+%include "isd.i"
+%include "rastergm.i"
+%include "plugin.i"
+%include "model.i"
\ No newline at end of file

ellipsoid.i

+%module(package="csmapi") ellipsoid
+%{
+    #include "Ellipsoid.h"
+%}
+
+%ignore CSM_WGS84_SEMI_MAJOR_AXIS;
+%ignore CSM_WGS84_SEMI_MINOR_AXIS;
+
+%include "Ellipsoid.h"

isd.i

+%module(package="csmapi") isd
+%{
+    #include "Isd.h"
+%}
+
+%ignore IMAGE_ID_PARAM;
+%ignore IMAGE_ID_PARAM;      
+%ignore IMAGE_INDEX_PARAM;
+%ignore LOGICAL_INDEX_PARAM;
+%ignore MODEL_NAME_PARAM;
+
+%include <std_string.i>
+
+%include "Isd.h"
+%pythoncode %{
+  import json
+  import numpy as np
+  @classmethod
+  def loads(cls, stream):
+      isd = cls()
+      if not isinstance(stream, dict):
+          stream = json.loads(stream)
+      for k, v in stream.items():
+          if isinstance(v, np.ndarray):
+              v = v.tolist()
+          if isinstance(v, list):
+              for i in v:
+                  isd.addParam(k, str(i))
+          isd.addParam(k, str(v))
+      return isd
+
+  @classmethod
+  def load(cls, fp):
+      return cls.loads(fp.read())
+
+  Isd.load = load
+  Isd.loads = loads
+%}
\ No newline at end of file

model.i

+%module(package="csmapi") model
+%{
+    #include "Model.h"
+%}
+
+%include Model.h
+

plugin.i

+%module(package="csmapi") plugin
+%{
+    #include "Plugin.h"
+%}
+
+%include <std_string.i>
+%include <std_list.i>
+%include typemaps.i
+%include Plugin.h
+
+
+
+
+%template(PluginList) std::list<const csm::Plugin*>;
+
+%{
+    namespace swig {
+    template <> struct traits<csm::Plugin>
+    {
+        typedef pointer_category category;
+        static const char* type_name()
+        {
+            return "Plugin";
+        }
+    };
+    }
+%}
+
+

python/CMakeLists.txt

+# Include python
+find_package(PythonLibs REQUIRED)
+include_directories(${PYTHON_INCLUDE_PATH})
+
+
+set(CMAKE_SWIG_FLAGS "")
+set_source_files_properties(../csmapi.i 
+                            PROPERTIES CPLUSPLUS ON)
+
+find_path(CSM_INCLUDE_DIR NAMES csm.h PATH_SUFFIXES csm)
+find_library(CSM_LIBRARY NAMES csmapi)
+
+include_directories(${CSM_INCLUDE_DIR})
+
+# Add swig module
+swig_add_library(csmapi
+                 LANGUAGE python 
+                 SOURCES ../csmapi.i)
+
+
+swig_link_libraries(csmapi ${CSM_LIBRARY} ${PYTHON_LIBRARIES})
+
+# Files to install with Python
+set(PYTHON_INSTALL_FILES
+    ${CMAKE_CURRENT_BINARY_DIR}/csmapi.py
+    ${CMAKE_CURRENT_BINARY_DIR}/_csmapi.so)
+
+# Configure setup.py and copy to output directory
+set(SETUP_PY_IN ${CMAKE_CURRENT_SOURCE_DIR}/setup.py.in)
+set(SETUP_PY_OUT ${CMAKE_CURRENT_BINARY_DIR}/setup.py)
+configure_file(${SETUP_PY_IN} ${SETUP_PY_OUT})
+
+# Declare install target for python
+#install(TARGETS swig_example
+#        COMMAND "${PYTHON_EXECUTABLE} setup.py"
+#        COMPONENT swig-python)
+
+# Install target to call setup.py
+add_custom_target(install-python
+        DEPENDS _csmapi
+COMMAND python ${SETUP_PY_OUT} install)

python/setup.py.in

+import setuptools.command.install
+import shutil
+from distutils.sysconfig import get_python_lib
+
+if __name__ == '__main__':
+    setuptools.setup(
+        name='csmapi',
+        version='0.1.0',
+        py_modules=['csmapi'],
+        package_data={'csmapi':['_csmapi.so']},
+        license='UnLicense',
+        author='jlaura',
+        author_email='jlaura@usgs.gov'
+)

rastergm.i

+%module(package="csmapi") rastergm
+%{
+    #include "RasterGM.h"
+%}
+
+%ignore CSM_RASTER_FAMILY;
+
+%include RasterGM.h
\ No newline at end of file

tests/test_functional.py

+import ctypes
+from distutils import dir_util
+import json
+import os
+
+import pytest
+
+import csmapi
+
+# Load a plugin with CSM compliant sensors
+lib = ctypes.CDLL('/data/big/github/CSM-CameraModel/build/libusgscsm.so')
+
+@pytest.fixture
+def datadir(tmpdir, request):
+    '''
+    Fixture responsible for searching a folder with the same name of test
+    module and, if available, moving all contents to a temporary directory so
+    tests can use them freely.
+    '''
+    filename = request.module.__file__
+    test_dir, _ = os.path.splitext(filename)
+
+    if os.path.isdir(test_dir):
+        dir_util.copy_tree(test_dir, str(tmpdir))
+    return tmpdir
+
+@pytest.fixture
+def isd(datadir):
+    with open(datadir.join('simpleFramerISD.json')) as f:
+        i = csmapi.Isd.load(f)
+    return i
+    
+
+@pytest.fixture
+def plugin():
+    plugin = csmapi.Plugin.findPlugin('UsgsAstroFramePluginCSM')
+    return plugin 
+
+def test_isd_to_model_to_ground(isd, plugin):
+    model_name = "USGS_ASTRO_FRAME_SENSOR_MODEL"
+    assert plugin.canModelBeConstructedFromISD(isd, model_name)
+    assert plugin.canISDBeConvertedToModelState(isd, model_name)
+    
+    model = plugin.constructModelFromISD(isd, model_name)
+    assert model.getVersion().version() == '0.1.0'
+    print(dir(model))
+    assert False
+    
\ No newline at end of file

tests/test_functional/simpleFramerISD.json

+{
+    "boresight": [
+        0.0,
+        0.0,
+        1.0
+    ],
+    "ccd_center": [
+        7.5,
+        7.5
+    ],
+    "ephemeris_time": 100.0,
+    "focal_length": 500,
+    "focal_length_epsilon": 1.0,
+    "ifov": 6.0,
+    "model_name": "UsgsAstroFramePluginCSM",
+    "spacecraft_name": "TEST_CRAFT",
+    "instrument_id": "TEST_SENSOR",
+    "target_name": "TEST_BALL",
+    "pixel_pitch": 0.1,
+    "itrans_line": [
+        0.0,
+        0.0,
+        10
+    ],
+    "itrans_sample": [
+        0.0,
+        10,
+        0.0
+    ],
+    "transx": [
+        0.0,
+        0.1,
+        0.0
+    ],
+    "transy": [
+        0.0,
+        0.0,
+        0.1
+    ],
+    "min_elevation": -1,
+    "max_elevation": 1,
+    "nlines": 16,
+    "nsamples": 16,
+    "original_half_lines": 8.0,
+    "original_half_samples": 8.0,
+    "omega": 0,
+    "phi": 0,
+    "kappa": 0,
+    "semi_major_axis": 10,
+    "semi_minor_axis": 10,
+    "transx": [
+        0.0,
+        0.1,
+        0.0
+    ],
+    "transy": [
+        0.0,
+        0.0,
+        0.1
+    ],
+    "x_sensor_origin": 1000,
+    "y_sensor_origin": 0,
+    "z_sensor_origin": 0,
+    "x_sensor_velocity": 1,
+    "y_sensor_velocity": 0,
+    "z_sensor_velocity": 0,
+    "x_sun_position": 100,
+    "y_sun_position": 100,
+    "z_sun_position": 0,
+    "odt_x": [
+        0.0,
+        1.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0
+    ],
+    "odt_y": [
+        0.0,
+        0.0,
+        1.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0,
+        0.0
+    ],
+    "starting_detector_line": 0.0,
+    "starting_detector_sample": 0.0
+}

tests/test_plugin.py

+import pytest
+
+import csmapi
+import ctypes
+
+# Load a plugin with CSM compliant sensors
+lib = ctypes.CDLL('/data/big/github/CSM-CameraModel/build/libusgscsm.so')
+
+@pytest.fixture
+def plugin():
+    return csmapi.Plugin
+
+@pytest.fixture
+def model(plugin):
+    pl = plugin.findPlugin('UsgsAstroFramePluginCSM')
+    return pl  
+
+def test_plugin_size(plugin):
+    plugin_list = plugin.getList()
+    assert len(plugin_list) == 2
+
+@pytest.mark.parametrize("plugin, index, expected_name",
+                         [(plugin(), 0, 'UsgsAstroFramePluginCSM'),
+                          (plugin(), 1, 'USGS_ASTRO_LINE_SCANNER_PLUGIN')])
+def test_plugin_name(plugin, index, expected_name):
+    pl = plugin.getList()[index]
+    plugin_name = pl.getPluginName()
+    assert plugin_name == expected_name
+
+def test_model_getNumModels(model):
+    assert model.getNumModels() == 1
+
+def test_model_getModelName(model):
+    assert model.getModelName(0) == 'USGS_ASTRO_FRAME_SENSOR_MODEL'
\ No newline at end of file

usgs.i

+%module(package="csmapi") usgs
+%{
+    #include "UsgsAstroLsPlugin.h"
+%}
+
+%import "UsgsAstroLsPlugin.h"
\ No newline at end of file

version.i

+%module(package="csmapi") version
+%{
+    #include "Version.h"
+%}
+
+%include <std_string.i>
+%include "Version.h"

warning.i

+%module(package="csmapi") warning
+%{
+    #include "Warning.h"
+%}
+
+%include Warning.h
\ No newline at end of file