[c9636f7] | 1 | |
---|
| 2 | from sans.models.BaseComponent import BaseComponent |
---|
| 3 | import numpy, math |
---|
| 4 | |
---|
| 5 | class MultiplicationModel(BaseComponent): |
---|
| 6 | """ |
---|
| 7 | Use for S(Q)*P(Q). |
---|
| 8 | Perform multplication of two models. |
---|
| 9 | Contains the models parameters combined. |
---|
| 10 | """ |
---|
| 11 | def __init__(self, model1, model2 ): |
---|
| 12 | BaseComponent.__init__(self) |
---|
| 13 | |
---|
[996fd35] | 14 | |
---|
[c9636f7] | 15 | ## Name of the model |
---|
[996fd35] | 16 | self.name="" |
---|
[c9636f7] | 17 | ## model description |
---|
[996fd35] | 18 | self.description="" |
---|
| 19 | if model1.name != "NoStructure"and model2.name != "NoStructure": |
---|
| 20 | self.name = model1.name +" * "+ model2.name |
---|
| 21 | self.description= self.name+"\n" |
---|
| 22 | self.description +="see %s description and %s description"%( model1.name, |
---|
| 23 | model2.name ) |
---|
| 24 | elif model2.name != "NoStructure": |
---|
| 25 | self.name = model2.name |
---|
| 26 | self.description= model2.description |
---|
| 27 | else : |
---|
| 28 | self.name = model1.name |
---|
| 29 | self.description= model1.description |
---|
| 30 | |
---|
| 31 | |
---|
| 32 | |
---|
[c9636f7] | 33 | self.model1= model1 |
---|
| 34 | self.model2= model2 |
---|
| 35 | ## dispersion |
---|
| 36 | self.dispersion = {} |
---|
| 37 | self._set_dispersion() |
---|
| 38 | ## Define parameters |
---|
| 39 | self.params = {} |
---|
| 40 | self._set_params() |
---|
| 41 | ## Parameter details [units, min, max] |
---|
| 42 | self.details = {} |
---|
| 43 | self._set_details() |
---|
| 44 | #list of parameter that can be fitted |
---|
| 45 | self.fixed= [] |
---|
| 46 | self._set_fixed_params() |
---|
| 47 | |
---|
| 48 | |
---|
| 49 | def _set_dispersion(self): |
---|
| 50 | """ |
---|
| 51 | combined the two models dispersions |
---|
| 52 | """ |
---|
| 53 | for name , value in self.model1.dispersion.iteritems(): |
---|
| 54 | self.dispersion[name]= value |
---|
| 55 | |
---|
| 56 | for name , value in self.model2.dispersion.iteritems(): |
---|
| 57 | if not name in self.dispersion.keys(): |
---|
| 58 | self.dispersion[name]= value |
---|
| 59 | |
---|
| 60 | |
---|
| 61 | def _set_params(self): |
---|
| 62 | """ |
---|
| 63 | Concatenate the parameters of the two models to create |
---|
| 64 | this model parameters |
---|
| 65 | """ |
---|
| 66 | for name , value in self.model1.params.iteritems(): |
---|
| 67 | self.params[name]= value |
---|
| 68 | |
---|
| 69 | for name , value in self.model2.params.iteritems(): |
---|
| 70 | if not name in self.params.keys(): |
---|
| 71 | self.params[name]= value |
---|
| 72 | |
---|
| 73 | def _set_details(self): |
---|
| 74 | """ |
---|
| 75 | Concatenate details of the two models to create |
---|
| 76 | this model details |
---|
| 77 | """ |
---|
| 78 | for name ,detail in self.model1.details.iteritems(): |
---|
| 79 | self.details[name]= detail |
---|
| 80 | |
---|
| 81 | for name , detail in self.model2.details.iteritems(): |
---|
| 82 | if not name in self.details.keys(): |
---|
| 83 | self.details[name]= detail |
---|
| 84 | |
---|
| 85 | def _set_fixed_params(self): |
---|
| 86 | """ |
---|
| 87 | fill the self.fixed list with the two models fixed list |
---|
| 88 | """ |
---|
| 89 | for item in self.model1.fixed: |
---|
| 90 | self.fixed.append(item) |
---|
| 91 | |
---|
| 92 | for item in self.model2.fixed: |
---|
| 93 | if not item in self.fixed: |
---|
| 94 | self.fixed.append(item) |
---|
| 95 | self.fixed.sort() |
---|
| 96 | |
---|
| 97 | |
---|
| 98 | def run(self, x = 0.0): |
---|
| 99 | """ Evaluate the model |
---|
| 100 | @param x: input q-value (float or [float, float] as [r, theta]) |
---|
| 101 | @return: (DAB value) |
---|
| 102 | """ |
---|
| 103 | return self.model1.run(x)* self.model2.run(x) |
---|
| 104 | |
---|
| 105 | def runXY(self, x = 0.0): |
---|
| 106 | """ Evaluate the model |
---|
| 107 | @param x: input q-value (float or [float, float] as [qx, qy]) |
---|
| 108 | @return: DAB value |
---|
| 109 | """ |
---|
| 110 | return self.model1.runXY(x)* self.model2.runXY(x) |
---|
| 111 | |
---|
| 112 | def set_dispersion(self, parameter, dispersion): |
---|
| 113 | """ |
---|
| 114 | Set the dispersion object for a model parameter |
---|
| 115 | @param parameter: name of the parameter [string] |
---|
| 116 | @dispersion: dispersion object of type DispersionModel |
---|
| 117 | """ |
---|
| 118 | if parameter in self.model1.getParamList(): |
---|
| 119 | return self.model1.set_dispersion(parameter, dispersion) |
---|
| 120 | else: |
---|
| 121 | return self.model2.set_dispersion(parameter, dispersion) |
---|
| 122 | |
---|
| 123 | |
---|
| 124 | |
---|
| 125 | |
---|