o
    h                     @   s   d dl Z d dlZd dlZd dlZd dlmZ d dlmZ d dlm	Z	m
Z
mZmZ edZedZddgZdd
dZdd Zdd ZdS )    N)import_module)skip)compile_link_import_py_extcompile_link_import_stringscompile_sourcesget_abspathnumpycython)z	sigmoid.cz
#include <math.h>

void sigmoid(int n, const double * const restrict in,
             double * const restrict out, double lim){
    for (int i=0; i<n; ++i){
        const double x = in[i];
        out[i] = x*pow(pow(x/lim, 8)+1, -1./8.);
    }
}
)z_sigmoid.pyxaw  
import numpy as np
cimport numpy as cnp

cdef extern void c_sigmoid "sigmoid" (int, const double * const,
                                      double * const, double)

def sigmoid(double [:] inp, double lim=350.0):
    cdef cnp.ndarray[cnp.float64_t, ndim=1] out = np.empty(
        inp.size, dtype=np.float64)
    c_sigmoid(inp.size, &inp[0], &out[0], lim)
    return out
     u@c                 C   s   | | | d d d  S )N      g      ? )datalimr   r   g/var/www/vscode/kcb/lib/python3.10/site-packages/sympy/utilities/_compilation/tests/test_compilation.pynpy)   s   r   c               
   C   s   t std tstd ddlm}  |  std dt  gd}d }z3tt|d\}}t jd	}|	|}t
|}t ||sCJ W |rS|d
 rUt|d
  d S d S d S |re|d
 rft|d
  w w w )Nznumpy not installed.zcython not installed.r   has_cNo C compiler found.c99)stdinclude_dirs)compile_kwargsi   	build_dir)r   r   r	   sympy.utilities._compilationr   get_includer   	_sources1randomsigmoidr   allcloseshutilrmtree)r   
compile_kwinfomodr   res_modres_npyr   r   r    test_compile_link_import_strings-   s(   
r'   c                 C   sR  ddl m} | std t| }tjd|d\}}t|d}|d W d    n1 s/w   Y  t|g|d\}t	||d}t
j|sJJ z	td	d
g}W n
 tjy]   Y nw td	|g}	d|	dv snJ tsrd S tjd|d\}}
t|
d}|d W d    n1 sw   Y  t|
g|g|d}|ddksJ d S )Nr   r   r   z.c)dirwtzH
        int foo(int bar) {
            return 2*bar;
        }
        )cwdnmz--helpfoozutf-8z.pyxz;cdef extern int foo(int)
def _foo(arg):
    return foo(arg))
extra_objsr      *   )r   r   r   strtempfilemkstempopenwriter   r   ospathexists
subprocesscheck_outputCalledProcessErrordecoder	   r   _foo)tmpdirr   r   _handle	file_pathofhobjobj_path_nm_outpyx_pathr$   r   r   r   test_compile_sourcesD   s4   rF   )r
   )r    r5   r8   r1   sympy.externalr   sympy.testing.pytestr   (sympy.utilities._compilation.compilationr   r   r   r   r   r	   r   r   r'   rF   r   r   r   r   <module>   s    
