|
1 | 1 | from dynamic_graph import plug
|
2 | 2 |
|
3 |
| -from .selector import Selector |
| 3 | +from .selector import Selector as Base |
4 | 4 |
|
5 | 5 | print("With meta selector")
|
6 | 6 |
|
|
11 | 11 | # % ,['matrixHomo','pg_H_sf',pg.rightfootref,pg.leftfootref]
|
12 | 12 | # % ,['matrixHomo','wa_H_sf',geom.rf,geom.lf]
|
13 | 13 | # % )
|
14 |
| -Selector.basicInit = Selector.__init__ |
15 | 14 |
|
16 | 15 |
|
17 |
| -def metaSelectorInit(self, name, *signalPlug): |
18 |
| - self.basicInit(name) |
19 |
| - if len(signalPlug): |
20 |
| - nbOut = len(signalPlug) |
21 |
| - nbIn = len(signalPlug[0]) - 2 |
22 |
| - self.reset(nbIn, nbOut) |
23 |
| - idxOut = 0 |
24 |
| - for sigOut in signalPlug: |
25 |
| - typeSig = sigOut[0] |
26 |
| - nameSig = sigOut[1] |
27 |
| - self.create(typeSig, nameSig, idxOut) |
28 |
| - idxIn = 0 |
29 |
| - for sigIn in sigOut[2:]: |
30 |
| - plug(sigIn, self.signal(nameSig + str(idxIn))) |
31 |
| - idxIn += 1 |
32 |
| - idxOut += 1 |
33 |
| - |
34 |
| - |
35 |
| -Selector.__init__ = metaSelectorInit |
| 16 | +class Selector: |
| 17 | + def __init__(self, name, *signalPlug): |
| 18 | + self = Base(name) |
| 19 | + if len(signalPlug): |
| 20 | + nbOut = len(signalPlug) |
| 21 | + nbIn = len(signalPlug[0]) - 2 |
| 22 | + self.reset(nbIn, nbOut) |
| 23 | + idxOut = 0 |
| 24 | + for sigOut in signalPlug: |
| 25 | + typeSig = sigOut[0] |
| 26 | + nameSig = sigOut[1] |
| 27 | + self.create(typeSig, nameSig, idxOut) |
| 28 | + idxIn = 0 |
| 29 | + for sigIn in sigOut[2:]: |
| 30 | + plug(sigIn, self.signal(nameSig + str(idxIn))) |
| 31 | + idxIn += 1 |
| 32 | + idxOut += 1 |
0 commit comments