在 Cython 模块之间共享声明

原文: http://docs.cython.org/en/latest/src/userguide/sharing_declarations.html

本节描述如何在一个 Cython 模块中使 C 声明,函数和扩展类型可用于另一个 Cython 模块。这些工具以 Python 导入机制为模型,可以被认为是它的编译时版本。

定义和实施文件

Cython 模块可以分为两部分:带有.pxd后缀的定义文件,包含可供其他 Cython 模块使用的 C 声明,以及带有.pyx后缀的实现文件,其中包含其他所有内容。当模块想要使用在另一个模块的定义文件中声明的内容时,它会使用 cimport 语句导入它。

仅包含 extern 声明的.pxd文件不需要与实际的.pyx文件或 Python 模块相对应。这可以使它成为一个放置常见声明的便利位置,例如来自 外部库 的函数声明,这些函数需要在多个模块中使用。

什么定义文件包含

定义文件可以包含:

  • 任何类型的 C 类型声明。
  • extern C 函数或变量声明。
  • 模块中定义的 C 函数声明。
  • 扩展类型的定义部分(见下文)。

它不能包含任何 C 或 Python 函数的实现,也不能包含任何 Python 类定义或任何可执行语句。当想要访问 cdef 属性和方法,或从本模块中定义的 cdef 类继承时,需要它。

注意

您不需要(也不应该)在声明文件 public 中声明任何内容,以使其可供其他 Cython 模块使用;它只是存在于定义文件中。如果要为外部 C 代码提供某些内容,则只需要公开声明。

实现文件包含什么

实现文件可以包含任何类型的 Cython 语句,但是如果相应的定义文件也定义了该类型,则对扩展类型的实现部分有一些限制(见下文)。如果这个模块不需要 cimport ,那么这是唯一需要的文件。

cimport 声明

cimport 语句用于定义或实现文件中,以访问在另一个定义文件中声明的名称。它的语法与普通的 Python import 语句完全相同:

  1. cimport module [, module...]
  2. from module cimport name [as name] [, name [as name] ...]

这是一个例子。 dishes.pxd是导出 C 数据类型的定义文件。 restaurant.pyx是一个导入并使用它的实现文件。

dishes.pxd

  1. cdef enum otherstuff:
  2. sausage, eggs, lettuce
  3. cdef struct spamdish:
  4. int oz_of_spam
  5. otherstuff filler

restaurant.pyx

  1. from __future__ import print_function
  2. cimport dishes
  3. from dishes cimport spamdish
  4. cdef void prepare(spamdish *d):
  5. d.oz_of_spam = 42
  6. d.filler = dishes.sausage
  7. def serve():
  8. cdef spamdish d
  9. prepare(&d)
  10. print(f'{d.oz_of_spam} oz spam, filler no. {d.filler}')

重要的是要理解 cimport 语句只能用于导入 C 数据类型,C 函数和变量以及扩展类型。它不能用于导入任何 Python 对象,并且(有一个例外)它并不意味着在运行时导入任何 Python。如果要引用已经 cimported 的模块中的任何 Python 名称,则还必须为其包含常规 import 语句。

例外情况是,当您使用 cimport 导入扩展类型时,其类型对象将在运行时导入,并由您导入它的名称提供。使用 cimport 导入扩展类型将在下面详细介绍。

如果.pxd文件发生变化,则可能需要重新编译 cimport 的所有模块。 Cython.Build.cythonize实用程序可以为您解决此问题。

定义文件的搜索路径

当您 cimport 一个名为modulename的模块时,Cython 编译器会搜索名为modulename.pxd的文件。它沿着包含文件的路径搜索此文件(由-I命令行选项或cythonize()include_path选项指定)以及sys.path

使用package_datasetup.py脚本中安装.pxd文件允许其他软件包将模块中的项目作为依赖项进行处理。

此外,每当编译文件modulename.pyx时,首先沿着包含路径搜索相应的定义文件modulename.pxd(但不是sys.path),如果找到,则在处理.pyx文件之前对其进行处理。

使用 cimport 解决命名冲突

cimport 机制提供了一种简洁明了的方法来解决使用相同名称的 Python 函数包装外部 C 函数的问题。您需要做的就是将外部 C 声明放入假想模块的.pxd文件中,将 cimport 放入该模块。然后,您可以通过使用模块名称对它们进行限定来引用 C 函数。这是一个例子:

c_lunch.pxd

  1. cdef extern from "lunch.h":
  2. void eject_tomato(float)

lunch.pyx

  1. cimport c_lunch
  2. def eject_tomato(float speed):
  3. c_lunch.eject_tomato(speed)

您不需要任何c_lunch.pyx文件,因为c_lunch.pxd中定义的唯一内容是外部 C 实体。在运行时不会有任何实际的c_lunch模块,但这无关紧要; c_lunch.pxd文件完成了在编译时提供额外命名空间的工作。

共享 C 函数

可以通过 cimport.pxd文件中为它们添加标题,使模块顶层定义的 C 函数可用,例如:

volume.pxd

  1. cdef float cube(float)

volume.pyx

  1. cdef float cube(float x):
  2. return x * x * x

spammery.pyx

  1. from __future__ import print_function
  2. from volume cimport cube
  3. def menu(description, size):
  4. print(description, ":", cube(size),
  5. "cubic metres of spam")
  6. menu("Entree", 1)
  7. menu("Main course", 3)
  8. menu("Dessert", 2)

Note

当模块以这种方式导出 C 函数时,对象将出现在函数名称下的模块字典中。但是,您不能使用 Python 中的此对象,也不能使用普通的 import 语句从 Cython 中使用它;你必须使用 cimport

共享扩展类型

可以通过 cimport 将扩展类型分为两部分,一部分在定义文件中,另一部分在相应的实现文件中。

扩展类型的定义部分只能声明 C 属性和 C 方法,而不能声明 Python 方法,并且它必须声明所有类型的 C 属性和 C 方法。

实现部分必须实现定义部分中声明的所有 C 方法,并且可能不会添加任何其他 C 属性。它也可以定义 Python 方法。

以下是定义和导出扩展类型的模块示例,以及使用它的另一个模块:

shrubbing.pxd

  1. cdef class Shrubbery:
  2. cdef int width
  3. cdef int length

shrubbing.pyx

  1. cdef class Shrubbery:
  2. def __cinit__(self, int w, int l):
  3. self.width = w
  4. self.length = l
  5. def standard_shrubbery():
  6. return Shrubbery(3, 7)

landscaping.pyx

  1. cimport shrubbing
  2. import shrubbing
  3. def main():
  4. cdef shrubbing.Shrubbery sh
  5. sh = shrubbing.standard_shrubbery()
  6. print("Shrubbery size is", sh.width, 'x', sh.length)

然后,人们需要编译这两个模块,例如运用

setup.py

  1. from distutils.core import setup
  2. from Cython.Build import cythonize
  3. setup(ext_modules=cythonize(["landscaping.pyx", "shrubbing.pyx"]))

有关此示例的一些注意事项:

  • Shrubbing.pxdShrubbing.pyx中都有 cdef 类灌木声明。编译 Shrubbing 模块时,这两个声明合并为一个。
  • 在 Landscaping.pyx 中, cimport Shrubbing 声明允许我们将灌木类型称为Shrubbing.Shrubbery。但它在运行时没有绑定 Landscaping 模块命名空间中的名称 Shrubbing,因此要访问Shrubbing.standard_shrubbery()我们还需要import Shrubbing
  • 如果您使用 setuptools 而不是 distutils,则需要注意,运行python setup.py install时的默认操作是创建一个压缩的egg文件,当您尝试从依赖包中使用它们时,这些文件无法与pxd文件一起用于pxd文件。为防止这种情况,请在setup()的参数中包含zip_safe=False