Merge common, func, abstract, stream as pxi to _ta_lib.pyx

Author: mckelvin

setup.py

@@ -80,16 +80,15 @@
 if has_cython:
     cmdclass['build_ext'] = build_ext
 
-ext_modules = []
-for name in ['common', 'func', 'abstract', 'stream']:
-    ext = Extension(
-        'talib.%s' % name,
-        [('talib/%s.pyx' if has_cython else 'talib/%s.c') % name],
-        include_dirs = include_dirs,
-        library_dirs = lib_talib_dirs,
-        libraries = [lib_talib_name]
+ext_modules = [
+    Extension(
+        'talib._ta_lib',
+        ['talib/_ta_lib.pyx' if has_cython else 'talib/_ta_lib.c'],
+        include_dirs=include_dirs,
+        library_dirs=lib_talib_dirs,
+        libraries=[lib_talib_name]
     )
-    ext_modules.append(ext)
+]
 
 setup(
     name = 'TA-Lib',

talib/__init__.py

@@ -1,13 +1,16 @@
 
 import atexit
 
-from . import common
-from . import abstract
-from . import stream
-from .common import MA_Type, __ta_version__
-from .common import _ta_set_unstable_period as set_unstable_period
-from .common import _ta_get_unstable_period as get_unstable_period
-from .func import *
+from ._ta_lib import (
+    _ta_initialize, _ta_shutdown, MA_Type, __ta_version__,
+    _ta_set_unstable_period as set_unstable_period,
+    _ta_get_unstable_period as get_unstable_period,
+    __TA_FUNCTION_NAMES__
+)
+
+func = __import__("_ta_lib", globals(), locals(), __TA_FUNCTION_NAMES__, level=1)
+for func_name in __TA_FUNCTION_NAMES__:
+    globals()[func_name] = getattr(func, func_name)
 
 __version__ = '0.4.10'
 
@@ -18,8 +21,8 @@
 # functions are called. Finally, when the python process exits, we shutdown
 # the underlying TA-Lib.
 
-common._ta_initialize()
-atexit.register(common._ta_shutdown)
+_ta_initialize()
+atexit.register(_ta_shutdown)
 
 __function_groups__ = {
     'Cycle Indicators': [

talib/abstract.pyx renamed to talib/_abstract.pxi

@@ -2,9 +2,6 @@
 This file Copyright (c) 2013 Brian A Cappello <briancappello at gmail>
 '''
 import math
-
-from . import func as func_c
-from .common import _ta_check_success, MA_Type
 try:
     from collections import OrderedDict
 except ImportError: # handle python 2.6 and earlier
@@ -14,11 +11,11 @@ import numpy
 import sys
 
 cimport numpy as np
-cimport libta_lib as lib
+cimport _ta_lib as lib
+# NOTE: _ta_check_success, MA_Type is defined in _common.pxi
 
 lib.TA_Initialize()
 
-__FUNCTION_NAMES = set(func_c.__all__)
 
 __INPUT_ARRAYS_DEFAULTS = {'open':   None,
                            'high':   None,
@@ -90,11 +87,9 @@ class Function(object):
     - FunctionInstance([input_arrays,] [param_args_andor_kwargs]) # calls set_function_args and returns self.outputs
     """
 
-    def __init__(self, function_name, *args, **kwargs):
+    def __init__(self, function_name, func_object, *args, **kwargs):
         # make sure the function_name is valid and define all of our variables
         self.__name = function_name.upper()
-        if self.__name not in __FUNCTION_NAMES:
-            raise Exception('%s not supported by TA-LIB.' % self.__name)
         self.__namestr = self.__name
         self.__name = str2bytes(self.__name)
         self.__info = None
@@ -109,6 +104,7 @@ class Function(object):
         # finish initializing: query the TALIB abstract interface and set arguments
         self.__initialize_function_info()
         self.set_function_args(*args, **kwargs)
+        self.func_object = func_object
 
     def __initialize_function_info(self):
         # function info
@@ -386,7 +382,7 @@ class Function(object):
             args.append(value)
 
         # Use the func module to actually call the function.
-        results = func_c.__getattribute__(self.__namestr)(*args)
+        results = self.func_object(*args)
         if isinstance(results, np.ndarray):
             keys = self.__outputs.keys()
             if not isinstance(keys, list):
@@ -682,10 +678,3 @@ cdef int __ta_getLookback(lib.TA_ParamHolder *holder):
     retCode = lib.TA_GetLookback(holder, &lookback)
     _ta_check_success('TA_GetLookback', retCode)
     return lookback
-
-# Configure all the available TA-Lib functions to be exported as
-# an abstract function wrapper for convenient import.
-for name in __FUNCTION_NAMES:
-    exec "%s = Function('%s')" % (name, name)
-
-__all__ = ['Function'] + list(__FUNCTION_NAMES)

talib/common.pyx renamed to talib/_common.pxi

@@ -1,6 +1,5 @@
-
-cimport libta_lib as lib
-from libta_lib cimport TA_RetCode, TA_FuncUnstId
+cimport _ta_lib as lib
+from _ta_lib cimport TA_RetCode, TA_FuncUnstId
 
 __ta_version__ = lib.TA_GetVersionString()
 

talib/func.pyx renamed to talib/_func.pxi

@@ -2,7 +2,7 @@ cimport numpy as np
 from numpy import nan
 from cython import boundscheck, wraparound
 
-from .common cimport _ta_check_success
+# _ta_check_success: defined in _common.pxi
 
 cdef double NaN = nan
 
@@ -14,10 +14,9 @@ cdef extern from "numpy/arrayobject.h":
 
 np.import_array() # Initialize the NumPy C API
 
-cimport libta_lib as lib
-from libta_lib cimport TA_RetCode
+cimport _ta_lib as lib
+from _ta_lib cimport TA_RetCode
 
-lib.TA_Initialize()
 
 @wraparound(False)  # turn off relative indexing from end of lists
 @boundscheck(False) # turn off bounds-checking for entire function
@@ -11243,4 +11242,4 @@ def WMA( np.ndarray real not None , int timeperiod=-2**31 ):
     _ta_check_success("TA_WMA", retCode)
     return outreal 
 
-__all__ = ["ACOS","AD","ADD","ADOSC","ADX","ADXR","APO","AROON","AROONOSC","ASIN","ATAN","ATR","AVGPRICE","BBANDS","BETA","BOP","CCI","CDL2CROWS","CDL3BLACKCROWS","CDL3INSIDE","CDL3LINESTRIKE","CDL3OUTSIDE","CDL3STARSINSOUTH","CDL3WHITESOLDIERS","CDLABANDONEDBABY","CDLADVANCEBLOCK","CDLBELTHOLD","CDLBREAKAWAY","CDLCLOSINGMARUBOZU","CDLCONCEALBABYSWALL","CDLCOUNTERATTACK","CDLDARKCLOUDCOVER","CDLDOJI","CDLDOJISTAR","CDLDRAGONFLYDOJI","CDLENGULFING","CDLEVENINGDOJISTAR","CDLEVENINGSTAR","CDLGAPSIDESIDEWHITE","CDLGRAVESTONEDOJI","CDLHAMMER","CDLHANGINGMAN","CDLHARAMI","CDLHARAMICROSS","CDLHIGHWAVE","CDLHIKKAKE","CDLHIKKAKEMOD","CDLHOMINGPIGEON","CDLIDENTICAL3CROWS","CDLINNECK","CDLINVERTEDHAMMER","CDLKICKING","CDLKICKINGBYLENGTH","CDLLADDERBOTTOM","CDLLONGLEGGEDDOJI","CDLLONGLINE","CDLMARUBOZU","CDLMATCHINGLOW","CDLMATHOLD","CDLMORNINGDOJISTAR","CDLMORNINGSTAR","CDLONNECK","CDLPIERCING","CDLRICKSHAWMAN","CDLRISEFALL3METHODS","CDLSEPARATINGLINES","CDLSHOOTINGSTAR","CDLSHORTLINE","CDLSPINNINGTOP","CDLSTALLEDPATTERN","CDLSTICKSANDWICH","CDLTAKURI","CDLTASUKIGAP","CDLTHRUSTING","CDLTRISTAR","CDLUNIQUE3RIVER","CDLUPSIDEGAP2CROWS","CDLXSIDEGAP3METHODS","CEIL","CMO","CORREL","COS","COSH","DEMA","DIV","DX","EMA","EXP","FLOOR","HT_DCPERIOD","HT_DCPHASE","HT_PHASOR","HT_SINE","HT_TRENDLINE","HT_TRENDMODE","KAMA","LINEARREG","LINEARREG_ANGLE","LINEARREG_INTERCEPT","LINEARREG_SLOPE","LN","LOG10","MA","MACD","MACDEXT","MACDFIX","MAMA","MAVP","MAX","MAXINDEX","MEDPRICE","MFI","MIDPOINT","MIDPRICE","MIN","MININDEX","MINMAX","MINMAXINDEX","MINUS_DI","MINUS_DM","MOM","MULT","NATR","OBV","PLUS_DI","PLUS_DM","PPO","ROC","ROCP","ROCR","ROCR100","RSI","SAR","SAREXT","SIN","SINH","SMA","SQRT","STDDEV","STOCH","STOCHF","STOCHRSI","SUB","SUM","T3","TAN","TANH","TEMA","TRANGE","TRIMA","TRIX","TSF","TYPPRICE","ULTOSC","VAR","WCLPRICE","WILLR","WMA"]
+__TA_FUNCTION_NAMES__ = ["ACOS","AD","ADD","ADOSC","ADX","ADXR","APO","AROON","AROONOSC","ASIN","ATAN","ATR","AVGPRICE","BBANDS","BETA","BOP","CCI","CDL2CROWS","CDL3BLACKCROWS","CDL3INSIDE","CDL3LINESTRIKE","CDL3OUTSIDE","CDL3STARSINSOUTH","CDL3WHITESOLDIERS","CDLABANDONEDBABY","CDLADVANCEBLOCK","CDLBELTHOLD","CDLBREAKAWAY","CDLCLOSINGMARUBOZU","CDLCONCEALBABYSWALL","CDLCOUNTERATTACK","CDLDARKCLOUDCOVER","CDLDOJI","CDLDOJISTAR","CDLDRAGONFLYDOJI","CDLENGULFING","CDLEVENINGDOJISTAR","CDLEVENINGSTAR","CDLGAPSIDESIDEWHITE","CDLGRAVESTONEDOJI","CDLHAMMER","CDLHANGINGMAN","CDLHARAMI","CDLHARAMICROSS","CDLHIGHWAVE","CDLHIKKAKE","CDLHIKKAKEMOD","CDLHOMINGPIGEON","CDLIDENTICAL3CROWS","CDLINNECK","CDLINVERTEDHAMMER","CDLKICKING","CDLKICKINGBYLENGTH","CDLLADDERBOTTOM","CDLLONGLEGGEDDOJI","CDLLONGLINE","CDLMARUBOZU","CDLMATCHINGLOW","CDLMATHOLD","CDLMORNINGDOJISTAR","CDLMORNINGSTAR","CDLONNECK","CDLPIERCING","CDLRICKSHAWMAN","CDLRISEFALL3METHODS","CDLSEPARATINGLINES","CDLSHOOTINGSTAR","CDLSHORTLINE","CDLSPINNINGTOP","CDLSTALLEDPATTERN","CDLSTICKSANDWICH","CDLTAKURI","CDLTASUKIGAP","CDLTHRUSTING","CDLTRISTAR","CDLUNIQUE3RIVER","CDLUPSIDEGAP2CROWS","CDLXSIDEGAP3METHODS","CEIL","CMO","CORREL","COS","COSH","DEMA","DIV","DX","EMA","EXP","FLOOR","HT_DCPERIOD","HT_DCPHASE","HT_PHASOR","HT_SINE","HT_TRENDLINE","HT_TRENDMODE","KAMA","LINEARREG","LINEARREG_ANGLE","LINEARREG_INTERCEPT","LINEARREG_SLOPE","LN","LOG10","MA","MACD","MACDEXT","MACDFIX","MAMA","MAVP","MAX","MAXINDEX","MEDPRICE","MFI","MIDPOINT","MIDPRICE","MIN","MININDEX","MINMAX","MINMAXINDEX","MINUS_DI","MINUS_DM","MOM","MULT","NATR","OBV","PLUS_DI","PLUS_DM","PPO","ROC","ROCP","ROCR","ROCR100","RSI","SAR","SAREXT","SIN","SINH","SMA","SQRT","STDDEV","STOCH","STOCHF","STOCHRSI","SUB","SUM","T3","TAN","TANH","TEMA","TRANGE","TRIMA","TRIX","TSF","TYPPRICE","ULTOSC","VAR","WCLPRICE","WILLR","WMA"]