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
|
namespace meep {
#ifndef SWIG_PYTHON_THREAD_SCOPED_BLOCK
#define SWIG_PYTHON_THREAD_SCOPED_BLOCK SWIG_PYTHON_THREAD_BEGIN_BLOCK
#endif
// like custom_src_time, but using Python function object, with proper reference counting
class custom_py_src_time : public src_time {
public:
custom_py_src_time(PyObject *fun, double st = -infinity, double et = infinity,
std::complex<double> f = 0, double fw = 0)
: func(fun), freq(f), start_time(float(st)), end_time(float(et)), fwidth(fw) {
SWIG_PYTHON_THREAD_SCOPED_BLOCK;
Py_INCREF(func);
}
virtual ~custom_py_src_time() {
SWIG_PYTHON_THREAD_SCOPED_BLOCK;
Py_DECREF(func);
}
virtual std::complex<double> current(double time, double dt) const {
if (is_integrated)
return src_time::current(time, dt);
else
return dipole(time);
}
virtual std::complex<double> dipole(double time) const {
SWIG_PYTHON_THREAD_SCOPED_BLOCK;
float rtime = float(time);
if (rtime >= start_time && rtime <= end_time) {
PyObject *py_t = PyFloat_FromDouble(time);
PyObject *pyres = PyObject_CallFunctionObjArgs(func, py_t, NULL);
double real = PyComplex_RealAsDouble(pyres);
double imag = PyComplex_ImagAsDouble(pyres);
std::complex<double> ret(real, imag);
Py_DECREF(py_t);
Py_DECREF(pyres);
return ret;
}
else
return 0.0;
}
virtual double last_time() const { return end_time; };
virtual src_time *clone() const {
SWIG_PYTHON_THREAD_SCOPED_BLOCK;
Py_INCREF(func); // default copy constructor doesn't incref
return new custom_py_src_time(*this);
}
virtual bool is_equal(const src_time &t) const {
const custom_py_src_time *tp = dynamic_cast<const custom_py_src_time *>(&t);
if (tp)
return (tp->start_time == start_time && tp->end_time == end_time && tp->func == func &&
tp->freq == freq && tp->fwidth == fwidth);
else
return 0;
}
virtual std::complex<double> frequency() const { return freq; }
virtual void set_frequency(std::complex<double> f) { freq = f; }
virtual double get_fwidth() const { return fwidth; };
virtual void set_fwidth(double fw) { fwidth = fw; }
private:
PyObject *func;
std::complex<double> freq;
double start_time, end_time, fwidth;
};
} // namespace meep
|
