Made displacement optional

    @smproperty.intvector(name="Displace Points", default_values=1)

    @smdomain.xml('<BooleanDomain name="bool"/>')

    def SetPositionDirection(self, direction):

        self._looks_down = direction == 1

    def SetDisplacementMode(self, displace):

        self._displace_points = displace == 1

        self.Modified()

    def _get_timesteps(self):

        y = np.repeat(y_pos,self._bins)

        z = np.zeros(self._bins)

        for i in range(self._bins):

            if(self._displace_points):

                x[i] += self._displacement_vectors[i][0]

                y[i] += self._displacement_vectors[i][1]

            z[i] = self._pos_depth + ((z_zero + (i * z_offset)) * (-1.0 if self._looks_down else 1.0))

            displacement[i] = [offset[0], offset[1],0]

            self._displacement_vectors[i] = (prev[0] + offset[0], prev[1] + offset[1])

        print(repr(self._displacement_vectors))

        #print(repr(self._displacement_vectors))

        output.SetPoints(points)

        output.PointData.append(magnitudes, "Magnitude (mm/s)")