python/include/boost/python/numeric.hpp at develop · jwakely/python · GitHub

History

242 lines (198 loc) · 6.9 KB

Raw

100

101

102

103

104

105

106

107

108

109

110

111

112

113

114

115

116

117

118

119

120

121

122

123

124

125

126

127

128

129

130

131

132

133

134

135

136

137

138

139

140

141

142

143

144

145

146

147

148

149

150

151

152

153

154

155

156

157

158

159

160

161

162

163

164

165

166

167

168

169

170

171

172

173

174

175

176

177

178

179

180

181

182

183

184

185

186

187

188

189

190

191

192

193

194

195

196

197

198

199

200

201

202

203

204

205

206

207

208

209

210

211

212

213

214

215

216

217

218

219

220

221

222

223

224

225

226

227

228

229

230

231

232

233

234

235

236

237

238

239

240

241

242

// Copyright David Abrahams 2002.

// Distributed under the Boost Software License, Version 1.0. (See

// accompanying file LICENSE_1_0.txt or copy at

// http://www.boost.org/LICENSE_1_0.txt)

#ifndef NUMARRAY_DWA2002922_HPP

# define NUMARRAY_DWA2002922_HPP

# include <boost/python/detail/prefix.hpp>

# include <boost/python/tuple.hpp>

# include <boost/python/str.hpp>

# include <boost/preprocessor/iteration/local.hpp>

# include <boost/preprocessor/cat.hpp>

# include <boost/preprocessor/repetition/enum.hpp>

# include <boost/preprocessor/repetition/enum_params.hpp>

# include <boost/preprocessor/repetition/enum_binary_params.hpp>

namespace boost { namespace python { namespace numeric {

class array;

namespace aux

{

struct BOOST_PYTHON_DECL array_base : object

{

# define BOOST_PP_LOCAL_MACRO(n) \

array_base(BOOST_PP_ENUM_PARAMS_Z(1, n, object const& x));

# define BOOST_PP_LOCAL_LIMITS (1, 7)

# include BOOST_PP_LOCAL_ITERATE()

object argmax(long axis=-1);

object argmin(long axis=-1);

object argsort(long axis=-1);

object astype(object const& type = object());

void byteswap();

object copy() const;

object diagonal(long offset = 0, long axis1 = 0, long axis2 = 1) const;

void info() const;

bool is_c_array() const;

bool isbyteswapped() const;

array new_(object type) const;

void sort();

object trace(long offset = 0, long axis1 = 0, long axis2 = 1) const;

object type() const;

char typecode() const;

object factory(

object const& sequence = object()

, object const& typecode = object()

, bool copy = true

, bool savespace = false

, object type = object()

, object shape = object());

object getflat() const;

long getrank() const;

object getshape() const;

bool isaligned() const;

bool iscontiguous() const;

long itemsize() const;

long nelements() const;

object nonzero() const;

void put(object const& indices, object const& values);

void ravel();

object repeat(object const& repeats, long axis=0);

void resize(object const& shape);

void setflat(object const& flat);

void setshape(object const& shape);

void swapaxes(long axis1, long axis2);

object take(object const& sequence, long axis = 0) const;

void tofile(object const& file) const;

str tostring() const;

void transpose(object const& axes = object());

object view() const;

public: // implementation detail - do not touch.

BOOST_PYTHON_FORWARD_OBJECT_CONSTRUCTORS(array_base, object);

};

struct BOOST_PYTHON_DECL array_object_manager_traits

{

static bool check(PyObject* obj);

static detail::new_non_null_reference adopt(PyObject* obj);

static PyTypeObject const* get_pytype() ;

};

} // namespace aux

class array : public aux::array_base

{

typedef aux::array_base base;

public:

object astype() { return base::astype(); }

template <class Type>

object astype(Type const& type_)

{

return base::astype(object(type_));

}

template <class Type>

array new_(Type const& type_) const

{

return base::new_(object(type_));

}

template <class Sequence>

void resize(Sequence const& x)

{

base::resize(object(x));

}

# define BOOST_PP_LOCAL_MACRO(n) \

void resize(BOOST_PP_ENUM_PARAMS_Z(1, n, long x)) \

{ \

resize(make_tuple(BOOST_PP_ENUM_PARAMS_Z(1, n, x))); \

}

# define BOOST_PP_LOCAL_LIMITS (1, BOOST_PYTHON_MAX_ARITY)

# include BOOST_PP_LOCAL_ITERATE()

template <class Sequence>

void setshape(Sequence const& x)

{

base::setshape(object(x));

}

# define BOOST_PP_LOCAL_MACRO(n) \

void setshape(BOOST_PP_ENUM_PARAMS_Z(1, n, long x)) \

{ \

setshape(make_tuple(BOOST_PP_ENUM_PARAMS_Z(1, n, x))); \

}

# define BOOST_PP_LOCAL_LIMITS (1, BOOST_PYTHON_MAX_ARITY)

# include BOOST_PP_LOCAL_ITERATE()

template <class Indices, class Values>

void put(Indices const& indices, Values const& values)

{

base::put(object(indices), object(values));

}

template <class Sequence>

object take(Sequence const& sequence, long axis = 0)

{

return base::take(object(sequence), axis);

}

template <class File>

void tofile(File const& f) const

{

base::tofile(object(f));

}

object factory()

{

return base::factory();

}

template <class Sequence>

object factory(Sequence const& sequence)

{

return base::factory(object(sequence));

}

template <class Sequence, class Typecode>

object factory(

Sequence const& sequence

, Typecode const& typecode_

, bool copy = true

, bool savespace = false

)

{

return base::factory(object(sequence), object(typecode_), copy, savespace);

}

template <class Sequence, class Typecode, class Type>

object factory(

Sequence const& sequence

, Typecode const& typecode_

, bool copy

, bool savespace

, Type const& type

)

{

return base::factory(object(sequence), object(typecode_), copy, savespace, object(type));

}

template <class Sequence, class Typecode, class Type, class Shape>

object factory(

Sequence const& sequence

, Typecode const& typecode_

, bool copy

, bool savespace

, Type const& type

, Shape const& shape

)

{

return base::factory(object(sequence), object(typecode_), copy, savespace, object(type), object(shape));

}

# define BOOST_PYTHON_ENUM_AS_OBJECT(z, n, x) object(BOOST_PP_CAT(x,n))

# define BOOST_PP_LOCAL_MACRO(n) \

template <BOOST_PP_ENUM_PARAMS_Z(1, n, class T)> \

explicit array(BOOST_PP_ENUM_BINARY_PARAMS_Z(1, n, T, const& x)) \

: base(BOOST_PP_ENUM_1(n, BOOST_PYTHON_ENUM_AS_OBJECT, x)) \

{}

# define BOOST_PP_LOCAL_LIMITS (1, 7)

# include BOOST_PP_LOCAL_ITERATE()

# undef BOOST_PYTHON_AS_OBJECT

static BOOST_PYTHON_DECL void set_module_and_type(char const* package_name = 0, char const* type_attribute_name = 0);

static BOOST_PYTHON_DECL std::string get_module_name();

public: // implementation detail -- for internal use only

BOOST_PYTHON_FORWARD_OBJECT_CONSTRUCTORS(array, base);

};

} // namespace boost::python::numeric

namespace converter

{

template <>

struct object_manager_traits< numeric::array >

: numeric::aux::array_object_manager_traits

{

BOOST_STATIC_CONSTANT(bool, is_specialized = true);

};

}

}} // namespace boost::python

#endif // NUMARRAY_DWA2002922_HPP

python/include/boost/python/numeric.hpp at develop · jwakely/python · GitHub

Expand file tree

Breadcrumbs

numeric.hpp

Latest commit

History

Breadcrumbs

numeric.hpp

File metadata and controls