manimgeo.components.vector.vector 源代码

from __future__ import annotations

from pydantic import Field, model_validator
from typing import TYPE_CHECKING, List, Any
import numpy as np

from ..base import BaseGeometry
from .adapter import VectorAdapter
from .args import *

if TYPE_CHECKING:
    from ..line import LineSegment
    from ..point import Point



class Vector(BaseGeometry):
    attrs: List[str] = Field(default=["vec", "norm", "unit_direction"], description="向量属性列表", init=False)
    vec: np.ndarray = Field(default=np.zeros(2), description="向量坐标", init=False)
    norm: Number = Field(default=0.0, description="向量模长", init=False)
    unit_direction: np.ndarray = Field(default=np.zeros(2), description="向量单位方向", init=False)

    args: VectorConstructArgs = Field(discriminator='construct_type', description="向量构造参数")



    @model_validator(mode='before')
    @classmethod
    def set_adapter_before_validation(cls, data: Any) -> Any:
        """在验证前设置 adapter 字段"""
        if isinstance(data, dict) and 'args' in data:
            # 假设 args 已经是 Pydantic 模型或可以被 VectorAdapter 接受
            data['adapter'] = VectorAdapter(args=data['args'])
        return data


    @property
    def construct_type(self) -> VectorConstructType:
        return self.args.construct_type



    def model_post_init(self, __context: Any):
        """模型初始化后，更新名字并添加依赖关系"""
        self.adapter = VectorAdapter(args=self.args)
        self.name = self.get_name(self.name)
        # 添加依赖关系
        self._extract_dependencies_from_args(self.args)
        self.update() # 首次计算


    def __add__(self, other: Vector):
        return Vector(
            name=f"{self.name} + {other.name}",
            args=AddVVArgs(vec1=self, vec2=other),
        )

    def __sub__(self, other: Vector):
        return Vector(
            name=f"{self.name} - {other.name}",
            args=SubVVArgs(vec1=self, vec2=other),
        )

    def __mul__(self, other: Number):
        return Vector(
            name=f"{other} * {self.name}",
            args=MulNVArgs(factor=other, vec=self),
        )

    # 构造方法


    @classmethod
    def PP(cls, start: Point, end: Point, name: str = ""):
        """
        通过两点构造向量

        - `start`: 起点
        - `end`: 终点
        """
        return Vector(
            name=name,
            args=PPArgs(start=start, end=end)
        )




    @classmethod
    def L(cls, line: LineSegment, name: str = ""):
        """
        通过线段构造向量

        - `line`: 线段
        """
        return Vector(
            name=name,
            args=LArgs(line=line)
        )




    @classmethod
    def N(cls, vec: np.ndarray, name: str = ""):
        """
        （数值）构造向量

        - `vec`: 向量数值
        """
        return Vector(
            name=name,
            args=NArgs(vec=vec)
        )




    @classmethod
    def NPP(cls, start: np.ndarray, end: np.ndarray, name: str = ""):
        """
        通过两点（数值）构造向量

        - `start`: 起点
        - `end`: 终点
        """
        return Vector(
            name=name,
            args=NPPArgs(start=start, end=end)
        )




    @classmethod
    def NNormDirection(cls, norm: Number, direction: np.ndarray, name: str = ""):
        """
        通过模长与方向构造向量

        - `norm`: 模长
        - `direction`: 方向
        """
        return Vector(
            name=name,
            args=NNormDirectionArgs(norm=norm, direction=direction)
        )