Commit d2abc0f0 authored by Mathieu Doucet's avatar Mathieu Doucet
Browse files

add fit

parent 58f8969f
Loading
Loading
Loading
Loading
+623 −0

File added.

Preview size limit exceeded, changes collapsed.

+6 −3
Original line number Diff line number Diff line
@@ -105,7 +105,7 @@ class RCurveGenerator:
        return sample

    @staticmethod
    def get_prob() -> QProbe:
    def get_prob(data=None, errors=None) -> QProbe:
        """Define the probe for reflectivity calculation.

        Returns
@@ -123,9 +123,12 @@ class RCurveGenerator:

        dq = q_resolution * q_range / 2.355

        zeros = np.zeros_like(q_range)
        if data is None:
            data = np.zeros_like(q_range)
        if errors is None:
            errors = np.zeros_like(q_range)

        probe = QProbe(q_range, dq, data=(zeros, zeros))
        probe = QProbe(q_range, dq, data=(data, errors))
        probe.background = Parameter(value=0.0, name="background")

        return probe