merge
This commit is contained in:
commit
153d82223a
5 changed files with 60 additions and 43 deletions
1
.gitignore
vendored
1
.gitignore
vendored
|
@ -1,4 +1,5 @@
|
|||
.eggs/
|
||||
build/
|
||||
dist/
|
||||
cnorxz.egg-info/
|
||||
*.so
|
||||
|
|
|
@ -24,10 +24,16 @@ namespace CNORXZ
|
|||
virtual void readFile(cer::Format f, const String& fname) const = 0;
|
||||
#endif
|
||||
virtual int typenum() const = 0;
|
||||
virtual void* data() const = 0;
|
||||
virtual const void* data() const = 0;
|
||||
virtual SizeT datasize() const = 0;
|
||||
};
|
||||
|
||||
class ArrayWrapperBase : public CArrayWrapperBase
|
||||
{
|
||||
public:
|
||||
virtual void* data() = 0;
|
||||
};
|
||||
|
||||
template <typename T>
|
||||
struct Typenum
|
||||
{
|
||||
|
@ -59,7 +65,7 @@ namespace CNORXZ
|
|||
};
|
||||
|
||||
template <typename T>
|
||||
class ArrayWrapper : public CArrayWrapperBase
|
||||
class ArrayWrapper : public ArrayWrapperBase
|
||||
{
|
||||
private:
|
||||
Sptr<ArrayBase<T>> mArr;
|
||||
|
@ -67,6 +73,7 @@ namespace CNORXZ
|
|||
public:
|
||||
ArrayWrapper() : mArr( std::make_shared<MArray<T>>() ) {}
|
||||
ArrayWrapper(const RangePtr& r) : mArr( std::make_shared<MArray<T>>(r) ) {}
|
||||
ArrayWrapper(const Sptr<ArrayBase<T>>& arr) : mArr(arr) {}
|
||||
|
||||
virtual RangePtr range() const override final { return mArr->range(); }
|
||||
virtual SizeT size() const override final { return mArr->size(); }
|
||||
|
@ -92,7 +99,9 @@ namespace CNORXZ
|
|||
}
|
||||
#endif
|
||||
virtual int typenum() const override final { return Typenum<T>::value; }
|
||||
virtual void* data() const override final
|
||||
virtual const void* data() const override final
|
||||
{ return reinterpret_cast<const void*>( mArr->data() ); }
|
||||
virtual void* data() override final
|
||||
{ return reinterpret_cast<void*>( mArr->data() ); }
|
||||
virtual SizeT datasize() const override final { return size() * sizeof(T); }
|
||||
};
|
||||
|
|
|
@ -17,6 +17,8 @@ namespace CNORXZ
|
|||
inline SizeT dim() const { return mR->dim(); }
|
||||
inline RangePtr sub(SizeT pos) { return mR->sub(pos); }
|
||||
inline String stringMeta(SizeT pos) const { return mR->stringMeta(pos); }
|
||||
inline MArray<RangePtr> sub() const { return mR->sub(); }
|
||||
inline RangePtr sub(SizeT pos) const { return mR->sub(pos); }
|
||||
|
||||
private:
|
||||
RangePtr mR;
|
||||
|
@ -35,6 +37,7 @@ PyObject* PyCxRange_size(PyCxRange* self);
|
|||
PyObject* PyCxRange_dim(PyCxRange* self);
|
||||
PyObject* PyCxRange_sub(PyCxRange* self, PyObject* args, PyObject* kwds);
|
||||
PyObject* PyCxRange_stringMeta(PyCxRange* self, PyObject* args, PyObject* kwds);
|
||||
PyObject* PyCxRange_sub(PyCxRange* self, PyObject* args, PyObject* kwds);
|
||||
PyTypeObject* PyCxRangeType_init();
|
||||
|
||||
extern PyMethodDef PyCxRange_methods[];
|
||||
|
|
|
@ -1,4 +1,5 @@
|
|||
|
||||
#include "array_wrapper.h"
|
||||
#include "range_wrapper.h"
|
||||
|
||||
using namespace CNORXZ;
|
||||
|
@ -7,6 +8,7 @@ int PyCxRange_init(PyCxRange* self, PyObject* args, PyObject* kwds)
|
|||
{
|
||||
|
||||
//!!!
|
||||
return 0;
|
||||
}
|
||||
|
||||
void PyCxRange_dealloc(PyCxRange* self)
|
||||
|
@ -49,6 +51,14 @@ PyObject* PyCxRange_sub(PyCxRange* self, PyObject* args, PyObject* kwds)
|
|||
return PyErr_Format(PyExc_RuntimeError, "requested sub-range position (%d) exceeds range dimension (%d)", pos, dim);
|
||||
}
|
||||
|
||||
// entire set of sub-ranges:
|
||||
/*
|
||||
PyCxArrayB* o;
|
||||
o = PyObject_New(PyCxArrayB, &PyCxArrayBType);
|
||||
o->ptrObj = new ArrayWrapper<RangePtr>( std::make_shared<MArray<RangePtr>>(self->ptrObj->sub()) );
|
||||
return Py_BuildValue("O", o);
|
||||
*/
|
||||
|
||||
const RangePtr r = self->ptrObj->sub(pos);
|
||||
PyCxRange* o = PyObject_New(PyCxRange, &PyCxRangeType);
|
||||
o->ptrObj = new RangeWrapper(r);
|
||||
|
|
70
setup.py
70
setup.py
|
@ -1,47 +1,57 @@
|
|||
#!/usr/bin/env python
|
||||
|
||||
#import numpy
|
||||
# imports:
|
||||
|
||||
import setuptools
|
||||
from distutils.core import setup, Extension
|
||||
import os
|
||||
import sysconfig
|
||||
import subprocess
|
||||
|
||||
# basic variables:
|
||||
|
||||
libname = "cnorxz"
|
||||
author = "Christian Zimmermann"
|
||||
author_email = "chizeta@f3l.de"
|
||||
path_to_cnorxz = "/home/chizeta/repos/cnorxz/install"
|
||||
expected_cnorxz_version = "1.0.0" # dummy for now; in the future this is supposed to be the c++ version which is guaranteed to be compatible with py cnorxz
|
||||
path_to_cnorxz = ""
|
||||
if 'CNORXZ' in os.environ:
|
||||
path_to_cnorxz = os.environ['CNORXZ']
|
||||
version = "0.0.0"
|
||||
have_numpy = True
|
||||
|
||||
git_hash = subprocess.run(['git','rev-parse','HEAD'],stdout=subprocess.PIPE).stdout.decode('ascii')[:-1]
|
||||
git_tags = subprocess.run(['git', 'tag', '-l', '--sort=refname', 'v*'],stdout=subprocess.PIPE).stdout.decode('ascii').split()
|
||||
cnorxz_version = subprocess.run([path_to_cnorxz+"/bin/cnorxz-config",'--version'],stdout=subprocess.PIPE).stdout.decode('ascii')[:-1]
|
||||
# flags:
|
||||
|
||||
assert len(git_tags) != 0, "no version tags found"
|
||||
git_tag = git_tags[-1]
|
||||
version = git_tag[1:] + 'dev'
|
||||
git_hash_tag = subprocess.run(['git','rev-list','-n','1',git_tag],stdout=subprocess.PIPE).stdout.decode('ascii')[:-1]
|
||||
if git_hash_tag == git_hash and expected_cnorxz_version != cnorxz_version:
|
||||
version = git_tag[1:]
|
||||
config_bin = path_to_cnorxz+"/bin/cnorxz-config"
|
||||
extra_compile_args = sysconfig.get_config_var('CFLAGS').split()
|
||||
cnorxz_version = subprocess.run([config_bin,'--version'],stdout=subprocess.PIPE).stdout.decode('ascii')[:-1]
|
||||
cnorxz_flags = subprocess.run([config_bin,'--flags'],stdout=subprocess.PIPE).stdout.decode('ascii').split()
|
||||
cnorxz_definitions = subprocess.run([config_bin,'--definitions'],stdout=subprocess.PIPE).stdout.decode('ascii').split()
|
||||
cnorxz_flags.remove("-Werror")
|
||||
cnorxz_flags.append("-Wno-write-strings")
|
||||
extra_compile_args += cnorxz_flags
|
||||
|
||||
# includes:
|
||||
|
||||
inc_dirs = list()
|
||||
if path_to_cnorxz != "":
|
||||
inc_dirs.append( path_to_cnorxz + "/include/cnorxz" )
|
||||
inc_dirs.append( "cnorxz/core/include" )
|
||||
#inc_dirs.append( numpy.get_include() )
|
||||
inc_dirs.append( "/home/chizeta/my_env/lib/python3.11/site-packages/numpy/core/include" )
|
||||
|
||||
# library dirs:
|
||||
|
||||
lib_dirs = list()
|
||||
lib_dirs.append( "/usr/lib" )
|
||||
lib_dirs.append( "/usr/local/lib" )
|
||||
if path_to_cnorxz != "":
|
||||
inc_dirs.append( path_to_cnorxz + "/lib" )
|
||||
lib_dirs.append( "/home/chizeta/repos/cnorxz/install/lib" )
|
||||
lib_dirs.append( path_to_cnorxz + "/lib" )
|
||||
|
||||
extra_compile_args = sysconfig.get_config_var('CFLAGS').split()
|
||||
cnorxz_flags = subprocess.run([path_to_cnorxz+"/bin/cnorxz-config",'--flags'],stdout=subprocess.PIPE).stdout.decode('ascii').split()
|
||||
cnorxz_flags.remove("-Werror")
|
||||
cnorxz_flags.append("-Wno-write-strings")
|
||||
extra_compile_args += cnorxz_flags
|
||||
# optional dependencies:
|
||||
|
||||
if have_numpy:
|
||||
import numpy
|
||||
inc_dirs.append( numpy.get_include() )
|
||||
|
||||
# main
|
||||
|
||||
default_extension_args = dict(
|
||||
language = "c++",
|
||||
|
@ -49,6 +59,7 @@ default_extension_args = dict(
|
|||
libraries = [libname],
|
||||
library_dirs = lib_dirs,
|
||||
extra_compile_args = extra_compile_args,
|
||||
define_macros = [(mac,None) for mac in cnorxz_definitions]
|
||||
)
|
||||
|
||||
if __name__ == "__main__":
|
||||
|
@ -59,21 +70,6 @@ if __name__ == "__main__":
|
|||
for d in lib_dirs:
|
||||
print("\t",d)
|
||||
|
||||
#extensions = list()
|
||||
#extensions.append(
|
||||
# setuptools.Extension(
|
||||
# #name = libname + ".core",
|
||||
# name = libname,
|
||||
# sources = [
|
||||
# "cnorxz/core/core.cpp",
|
||||
# "cnorxz/core/lib/array_wrapper.cpp"
|
||||
# ],
|
||||
# define_macros=[('HAVE_CEREAL',None)],
|
||||
# **default_extension_args
|
||||
# )
|
||||
#)
|
||||
# append further extensions (cereal, hdf5, mpi ...) here
|
||||
|
||||
setup(
|
||||
name = libname,
|
||||
packages = setuptools.find_packages(),
|
||||
|
@ -81,10 +77,8 @@ if __name__ == "__main__":
|
|||
include_dirs = inc_dirs,
|
||||
ext_modules = [Extension(
|
||||
'cnorxz', ['cnorxz/core/core.cpp','cnorxz/core/lib/array_wrapper.cpp','cnorxz/core/lib/range_wrapper.cpp'],
|
||||
define_macros=[('HAVE_CEREAL',None)],
|
||||
**default_extension_args
|
||||
)],
|
||||
#ext_modules = Cython.Build.cythonize(extensions),
|
||||
author = author,
|
||||
author_email = author_email,
|
||||
)
|
||||
|
|
Loading…
Reference in a new issue