create cnpy package

recipes/cnpy/all/CMakeLists.txt

@@ -0,0 +1,7 @@
+cmake_minimum_required(VERSION 3.15.0)
+project(cmake_wrapper)
+
+include(conanbuildinfo.cmake)
+conan_basic_setup()
+
+add_subdirectory("source_subfolder")

recipes/cnpy/all/conandata.yml

@@ -0,0 +1,4 @@
+sources:
+  "1.0":
+    url: "https://github.com/hongyx11/cnpy/archive/v1.0.tar.gz"
+    sha256: "a61b209c268db20a00240876df9fd8314a73d3c9b4139d00a0df048fe9c8e59f"

recipes/cnpy/all/conanfile.py

@@ -0,0 +1,66 @@
+import os
+
+from conans import ConanFile, CMake, tools
+
+class CnpyConan(ConanFile):
+  name = "cnpy"
+  description = "cnpy conan package"
+  license = "MIT License"
+  topics = ("conan", "cnpy")
+  homepage = "https://github.com/hongyx11/cnpy"
+  url = "https://github.com/conan-io/conan-center-index"
+  exports_sources = ["CMakeLists.txt"]
+  generators = "cmake"
+  settings = "os", "arch", "compiler", "build_type"
+  _cmake = None
+
+
+  options = {
+    "shared": [True, False],
+    "fPIC": [True, False]
+  }
+  default_options = {
+    "shared": False,
+    "fPIC": True
+  }
+
+  @property
+  def _source_subfolder(self):
+    return "source_subfolder"
+
+  @property
+  def _build_subfolder(self):
+    return "build_subfolder"
+
+  def config_options(self):
+    if self.settings.os == "Windows":
+      del self.options.fPIC
+
+  def configure(self):
+    del self.settings.compiler.libcxx
+    del self.settings.compiler.cppstd
+
+  def source(self):
+    tools.get(**self.conan_data["sources"][self.version])
+    os.rename(self.name + "-" + self.version, self._source_subfolder)
+
+  def build(self):
+    cmake = self._configure_cmake()
+    cmake.build()
+
+  def _configure_cmake(self):
+    if self._cmake:
+      return self._cmake
+    self._cmake = CMake(self)
+    self._cmake.configure(build_folder=self._build_subfolder)
+    return self._cmake
+
+  def package(self):
+    self.copy("LICENSE", dst="licenses", src=self._source_subfolder)
+    cmake = self._configure_cmake()
+    cmake.install()
+
+  def package_info(self):
+    self.cpp_info.names["cmake_find_package"] = "cnpy"
+    self.cpp_info.names["cmake_find_package_multi"] = "cnpy"
+    self.cpp_info.libs = tools.collect_libs(self)

recipes/cnpy/all/test_package/CMakeLists.txt

@@ -0,0 +1,8 @@
+cmake_minimum_required(VERSION 3.15.0)
+project(test_package CXX)
+
+include(${CMAKE_BINARY_DIR}/conanbuildinfo.cmake)
+conan_basic_setup()
+set(CMAKE_CXX_STANDARD 11)
+add_executable(${PROJECT_NAME} example1.cpp)
+target_link_libraries(${PROJECT_NAME} ${CONAN_LIBS})

recipes/cnpy/all/test_package/conanfile.py

@@ -0,0 +1,19 @@
+import os
+
+from conans import ConanFile, CMake, tools
+
+class TestPackageConan(ConanFile):
+  settings = "os", "compiler", "build_type", "arch"
+  generators = "cmake"
+  requires = [
+    "zlib/1.2.11"
+  ]
+  def build(self):
+    cmake = CMake(self)
+    cmake.configure()
+    cmake.build()
+
+  def test(self):
+    if not tools.cross_building(self.settings):
+      bin_path = os.path.join("bin", "test_package")
+      self.run(bin_path, run_environment=True)

recipes/cnpy/all/test_package/example1.cpp

@@ -0,0 +1,55 @@
+#include"cnpy.h"
+#include<complex>
+#include<cstdlib>
+#include<iostream>
+#include<map>
+#include<string>
+
+const int Nx = 128;
+const int Ny = 64;
+const int Nz = 32;
+
+int main()
+{
+    //set random seed so that result is reproducible (for testing)
+    srand(0);
+    //create random data
+    std::vector<std::complex<double>> data(Nx*Ny*Nz);
+    for(int i = 0;i < Nx*Ny*Nz;i++) data[i] = std::complex<double>(rand(),rand());
+
+    //save it to file
+    cnpy::npy_save("arr1.npy",&data[0],{Nz,Ny,Nx},"w");
+
+    //load it into a new array
+    cnpy::NpyArray arr = cnpy::npy_load("arr1.npy");
+    std::complex<double>* loaded_data = arr.data<std::complex<double>>();
+
+    //make sure the loaded data matches the saved data
+    assert(arr.word_size == sizeof(std::complex<double>));
+    assert(arr.shape.size() == 3 && arr.shape[0] == Nz && arr.shape[1] == Ny && arr.shape[2] == Nx);
+    for(int i = 0; i < Nx*Ny*Nz;i++) assert(data[i] == loaded_data[i]);
+
+    //append the same data to file
+    //npy array on file now has shape (Nz+Nz,Ny,Nx)
+    cnpy::npy_save("arr1.npy",&data[0],{Nz,Ny,Nx},"a");
+
+    //now write to an npz file
+    //non-array variables are treated as 1D arrays with 1 element
+    double myVar1 = 1.2;
+    char myVar2 = 'a';
+    cnpy::npz_save("out.npz","myVar1",&myVar1,{1},"w"); //"w" overwrites any existing file
+    cnpy::npz_save("out.npz","myVar2",&myVar2,{1},"a"); //"a" appends to the file we created above
+    cnpy::npz_save("out.npz","arr1",&data[0],{Nz,Ny,Nx},"a"); //"a" appends to the file we created above
+
+    //load a single var from the npz file
+    cnpy::NpyArray arr2 = cnpy::npz_load("out.npz","arr1");
+
+    //load the entire npz file
+    cnpy::npz_t my_npz = cnpy::npz_load("out.npz");
+
+    //check that the loaded myVar1 matches myVar1
+    cnpy::NpyArray arr_mv1 = my_npz["myVar1"];
+    double* mv1 = arr_mv1.data<double>();
+    assert(arr_mv1.shape.size() == 1 && arr_mv1.shape[0] == 1);
+    assert(mv1[0] == myVar1);
+}

recipes/cnpy/config.yml

@@ -0,0 +1,3 @@
+versions:
+  "1.0":
+    folder: all