a d@sddlmZmZddlZddlmZmZmZmZddl m Z ddl m Z m Z mZddgZGdd d ZGd ddeZd eeeed d dZdS))OptionalUnionN)Module ParameterTensor normalize) squared_norm) KORNIA_CHECKKORNIA_CHECK_IS_TENSORKORNIA_CHECK_SHAPEParametrizedLinefit_linec@seZdZdZdS) HyperplanezeNot implemented yet: https://gitlab.com/libeigen/eigen/-/blob/master/Eigen/src/Geometry/Hyperplane.h.N)__name__ __module__ __qualname____doc__rr]/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/kornia/geometry/line.pyrsrcseZdZdZeeddfdd ZedddZedd d Zedd d Z d dZ e edddZ e edddZ edddZeddddZeeddddZeeefedddZeedddZeedd d!Zeedd"d#ZZS)$r zClass that describes a parametrize line. A parametrized line is defined by an origin point :math:`o` and a unit direction vector :math:`d` such that the line corresponds to the set .. math:: l(t) = o + t * d N)origin directionreturncs"tt||_t||_dS)alInitializes a parametrized line of direction and origin. Args: origin: any point on the line of any dimension. direction: the normalized vector direction of any dimension. Example: >>> o = torch.tensor([0.0, 0.0]) >>> d = torch.tensor([1.0, 1.0]) >>> l = ParametrizedLine(o, d) N)super__init__r_origin _direction)selfrr __class__rrr s  zParametrizedLine.__init__)rcCsd|jd|jS)NzOrigin: z Direction: rrrrrr__str__0szParametrizedLine.__str__cCst|SN)strr rrr__repr__3szParametrizedLine.__repr__cCs|dkr|jS|jS)Nrr)ridxrrr __getitem__6szParametrizedLine.__getitem__ccs|j|jfEdHdSr"rr rrr__iter__9szParametrizedLine.__iter__cCs|jS)zReturn the line origin point.)rr rrrr<szParametrizedLine.origincCs|jS)z!Return the line direction vector.)rr rrrrAszParametrizedLine.directioncCs |jjdS)z-Return the dimension in which the line holds.)rshaper rrrdimFszParametrizedLine.dimcCst|t||dddS)aConstructs a parametrized line going from a point :math:`p0` to :math:`p1`. Args: p0: tensor with first point :math:`(B, D)` where `D` is the point dimension. p1: tensor with second point :math:`(B, D)` where `D` is the point dimension. Example: >>> p0 = torch.tensor([0.0, 0.0]) >>> p1 = torch.tensor([1.0, 1.0]) >>> l = ParametrizedLine.through(p0, p1) r()pr*)r r)clsZp0p1rrrthroughJs zParametrizedLine.through)planercCstd|ddS)NzPlane not implemented yet .)NotImplementedError)r-r0rrrfrom_hyperplaneYsz ParametrizedLine.from_hyperplane)trcCs|j|j|S)a_The point at :math:`t` along this line. Args: t: step along the line. Return: tensor with the point. Example: >>> p0 = torch.tensor([0.0, 0.0]) >>> p1 = torch.tensor([1.0, 1.0]) >>> l = ParametrizedLine.through(p0, p1) >>> p2 = l.point_at(0.1) r)rr4rrrpoint_at]szParametrizedLine.point_at)pointrcCs|j|j||j|jS)zuReturn the projection of a point onto the line. Args: point: the point to be projected. rrr6rrr projectionnszParametrizedLine.projectioncCs"||j}t||j||jS)zReturn the squared distance of a point to its projection onte the line. Args: point: the point to calculate the distance onto the line. )rrr)rr6diffrrrsquared_distancews z!ParametrizedLine.squared_distancecCs||S)zReturn the distance of a point to its projections onto the line. Args: point: the point to calculate the distance into the line. )r:sqrtr7rrrdistanceszParametrizedLine.distance)rrrrrrr#r!r$r&r'propertyrrintr* classmethodr/rr3rfloatr5r8r:r< __classcell__rrrrr s&   )pointsweightsrcCst|dt|gd|dd}||}|dur~t|dt|dgt|jd|jdk|dd t||}n|dd |}tj |\}}}|d dddf}|d dddf}t ||S) afFit a line from a set of points. Args: points: tensor containing a batch of sets of n-dimensional points. The expected shape of the tensor is :math:`(B,N,D)`. weights: weights to use to solve the equations system. The expected shape of the tensor is :math:`(B,N)`. Return: A tensor containing the direction of the fited line of shape :math:`(B,D)`. Example: >>> points = torch.rand(2, 10, 3) >>> weights = torch.ones(2, 10) >>> line = fit_line(points, weights) >>> line.direction.shape torch.Size([2, 3]) zpoints must be a tensor)BNDTNzweights must be a tensorrDrr(.) r r meanr r)Z transposetorchZ diag_embedZlinalgZsvdr )rBrCrHA_Vrrrrrr s    )N)typingrrrIZ kornia.corerrrrZkornia.geometry.linalgrZkornia.testingr r r __all__rr r rrrrs z