summaryrefslogtreecommitdiff
path: root/trunk/Lib/ruby/rubycomplex.swg
blob: afdb15e7ed3c83f857e7cae15e54b2aace580b82 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
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
/*
  Defines the As/From conversors for double/float complex, you need to
  provide complex Type, the Name you want to use in the conversors,
  the complex Constructor method, and the Real and Imag complex
  accesor methods.

  See the std_complex.i and ccomplex.i for concrete examples.
*/

/*
  Ruby does not have native complex numbers.  They are an extension in the
  STD library.
*/
%{
  static VALUE swig_rb_cComplex = Qnil;
  static ID    swig_real_id = 0;
  static ID    swig_imag_id = 0;

  int Ruby_Is_Complex( VALUE obj )
    {
      return ( (rb_respond_to( obj, swig_real_id ) == Qtrue) &&
	       (rb_respond_to( obj, swig_imag_id ) == Qtrue) );
    }
%}

%init {
  rb_require("complex");
  swig_rb_cComplex = rb_const_get( rb_cObject, rb_intern("Complex") );
  if( swig_rb_cComplex == Qnil )
    rb_warn("Ruby's complex.so not found");
  swig_real_id = rb_intern("real");
  swig_imag_id = rb_intern("imag");
}

/* the common from conversor */
%define %swig_fromcplx_conv(Type, Real, Imag)
%fragment(SWIG_From_frag(Type),"header")
{
SWIGINTERNINLINE VALUE
SWIG_From(Type)(%ifcplusplus(const Type&, Type) c)
{
  VALUE args[] = { 
    rb_float_new(Real(c)), 
    rb_float_new(Imag(c))
  };
  return rb_class_new_instance(2, args, swig_rb_cComplex);
}
}
%enddef

/* the double case */
%define %swig_cplxdbl_conv(Type, Constructor, Real, Imag)
%fragment(SWIG_AsVal_frag(Type),"header",
	  fragment=SWIG_AsVal_frag(double))
{
SWIGINTERN int
SWIG_AsVal(Type) (VALUE o, Type* val)
{
  if ( Ruby_Is_Complex( o ) ) {
    if (val) {
      VALUE real = rb_funcall(o, swig_real_id, 0 );
      VALUE imag = rb_funcall(o, swig_imag_id, 0 );
      double re = 0;
      SWIG_AsVal_double( real, &re );
      double im = 0;
      SWIG_AsVal_double( imag, &im );
      *val = Constructor(re, im);
    }
    return SWIG_OK;
  } else {
    double d;    
    int res = SWIG_AddCast(SWIG_AsVal(double)(o, &d));
    if (SWIG_IsOK(res)) {
      if (val) *val = Constructor(d, 0.0);
      return res;
    }
  }
  return SWIG_TypeError;
}
}
%swig_fromcplx_conv(Type, Real, Imag);
%enddef

/* the float case */
%define %swig_cplxflt_conv(Type, Constructor, Real, Imag)
%fragment(SWIG_AsVal_frag(Type),"header",
          fragment=SWIG_AsVal_frag(float),
          fragment=SWIG_AsVal_frag(double)) {
SWIGINTERN int
SWIG_AsVal(Type)(VALUE o, Type *val)
{
  if ( Ruby_Is_Complex( o ) ) {
    VALUE real = rb_funcall(o, swig_real_id, 0 );
    VALUE imag = rb_funcall(o, swig_imag_id, 0 );
    double re = 0;
    SWIG_AsVal_double( real, &re );
    double im = 0;
    SWIG_AsVal_double( imag, &im );
    if ((-FLT_MAX <= re && re <= FLT_MAX) && 
	(-FLT_MAX <= im && im <= FLT_MAX)) {
      if (val) *val = Constructor(%numeric_cast(re, float),
				  %numeric_cast(im, float));
      return SWIG_OK;
    } else {
      return SWIG_OverflowError;
    }
  } else {
    float re;
    int res = SWIG_AddCast(SWIG_AsVal(float)(o, &re));
    if (SWIG_IsOK(res)) {
      if (val) *val = Constructor(re, 0.0);
      return res;
    }
  }
  return SWIG_TypeError;
}
}

%swig_fromcplx_conv(Type, Real, Imag);
%enddef

#define %swig_cplxflt_convn(Type, Constructor, Real, Imag) \
%swig_cplxflt_conv(Type, Constructor, Real, Imag)


#define %swig_cplxdbl_convn(Type, Constructor, Real, Imag) \
%swig_cplxdbl_conv(Type, Constructor, Real, Imag)