Source code for openglider.vector

#! /usr/bin/python2
# -*- coding: utf-8; -*-
#
# (c) 2013 booya (http://booya.at)
#
# This file is part of the OpenGlider project.
#
# OpenGlider is free software; you can redistribute it and/or modify
# it under the terms of the GNU General Public License as published by
# the Free Software Foundation; either version 2 of the License, or
# (at your option) any later version.
#
# OpenGlider is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with OpenGlider.  If not, see <http://www.gnu.org/licenses/>.

import numpy as np

from openglider.vector.functions import (
    norm,
    norm_squared,
    normalize,
    rangefrom,
    vector_angle,
)
from openglider.vector.polyline import PolyLine, PolyLine2D
from openglider.vector.polygon import Polygon2D
from openglider.vector.plane import Plane
from openglider.vector.interpolate import Interpolation




def depth(arg):
    try:
        return max([depth(i) for i in arg]) + 1
    except TypeError:  # Not a list anymore
        return 1





def arrtype(arg):
    """
    return type of a vector list: 2d-point (1), list of 2d-points (2), 3d-point (3), list of 3d-points (4)
    """
    ##2d-point//argof2d-points//3d-point//argof3d-points
    ##2d-p: depth 1
    ##equivalent np.rank?

    # TODO: Room for improvement here!
    # todo: remove!

    if depth(arg) == 2:
        if len(arg) == 2:
            return 1
        elif len(arg) == 3:
            return 3
        else:
            return 0
    elif depth(arg) == 3:
        if [depth(i) for i in arg] == [2 for i in arg]:
            if [len(i) for i in arg] == [2 for i in arg]:
                return 2
            elif [len(i) for i in arg] == [3 for i in arg]:
                return 4
            else:
                return 0
        else:
            return 0
    else:
        return 0





class mirror_func:
    def __init__(self, direction=None):
        if direction is None:
            direction = [1.0, 0.0, 0.0]
        if len(direction) == 2:
            x, y = normalize(direction)
            self.matrix = np.array(
                [[1 - 2 * x**2, -2 * x * y], [-2 * x * y, 1 - 2 * y**2]]
            )
        else:
            x, y, z = normalize(direction)
            self.matrix = np.array(
                [
                    [1 - 2 * x**2, -2 * x * y, -2 * x * z],
                    [-2 * x * y, 1 - 2 * y**2, -2 * y * z],
                    [-2 * x * z, -2 * y * z, 1 - 2 * z**2],
                ]
            )

    def __call__(self, vec):
        if len(vec) == 2 and not isinstance(vec[0], (np.ndarray, list, tuple)):
            return np.dot(vec, self.matrix).tolist()
        else:
            return np.array([self(i) for i in vec]).tolist()



mirror2D_x = mirror_func(direction=[1.0, 0.0])
mirror_x = mirror_func(direction=[1.0, 0.0, 0.0])