Welcome to mirror list, hosted at ThFree Co, Russian Federation.

mesh_select_by_pi.py « mesh_select_tools « mesh_extra_tools - git.blender.org/blender-addons.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
blob: 6f2d44189e7c57f97a6ed2a4ba41dbc6d83a48ca (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
# mesh_select_by_pi.py Copyright (C) 2011, Dolf Veenvliet
# Extrude a selection from a mesh multiple times

# ***** BEGIN GPL LICENSE BLOCK *****
#
# This program 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.
#
# This program 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 this program; if not, write to the Free Software Foundation,
# Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#
# ***** END GPL LICENCE BLOCK *****

'''
bl_info = {
    "name": "Select by pi",
    "author": "Dolf Veenvliet",
    "version": 1,
    "blender": (2, 56, 0),
    "location": "View3D > Select",
    "description": "Select fake random based on pi",
    "warning": "",
    "wiki_url": "",
    "tracker_url": "",
    "category": "Mesh"}

"""
Usage:

Additional links:
    Author Site: http://www.macouno.com
    e-mail: dolf {at} macouno {dot} com
"""
'''

import bpy
from bpy.types import Operator
from bpy.props import BoolProperty


class Select_by_pi():

    # Initialise the class
    def __init__(self, context, e, invert, extend, start_new):

        self.ob = context.active_object
        # keep or not the original selection (helps with selected all)
        if start_new:
            bpy.ops.mesh.select_all(action='DESELECT')

        bpy.ops.object.mode_set(mode='OBJECT')

        self.invert = invert

        # Make pi as a list of integers
        if e:
            self.pi = list('27182818284590452353602874713526624977572470936999')
        else:
            self.pi = list('31415926535897932384626433832795028841971693993751')

        self.piLen = len(self.pi)
        self.piPos = 0

        vertSelect = bpy.context.tool_settings.mesh_select_mode[0]
        edgeSelect = bpy.context.tool_settings.mesh_select_mode[1]
        faceSelect = bpy.context.tool_settings.mesh_select_mode[2]

        # Vert select
        if vertSelect:
            hasSelected = self.hasSelected(self.ob.data.vertices)

            for v in self.ob.data.vertices:
                s = self.selectCheck(v.select, hasSelected, extend)
                d = self.deselectCheck(v.select, hasSelected, extend)

                # Check if the verts match any of the directions
                if s and self.choose():
                    v.select = True

                if d and not self.choose():
                    v.select = False

        # Edge select
        if edgeSelect:
            hasSelected = self.hasSelected(self.ob.data.edges)

            for e in self.ob.data.edges:
                s = self.selectCheck(e.select, hasSelected, extend)
                d = self.deselectCheck(e.select, hasSelected, extend)

                if s and self.choose():
                    e.select = True

                if d and not self.choose():
                    e.select = False

        # Face select
        if faceSelect:
            hasSelected = self.hasSelected(self.ob.data.polygons)

            # Loop through all the given faces
            for f in self.ob.data.polygons:
                s = self.selectCheck(f.select, hasSelected, extend)
                d = self.deselectCheck(f.select, hasSelected, extend)

                # Check if the faces match any of the directions
                if s and self.choose():
                    f.select = True

                if d and not self.choose():
                    f.select = False

        bpy.ops.object.mode_set(mode='EDIT')

    # Choose by pi
    def choose(self):
        choice = True

        # We just choose the odd numbers
        if int(self.pi[self.piPos]) % 2:
            choice = False

        if self.invert:
            choice = not choice

        self.incrementPiPos()
        return choice

    # Increment the pi position
    def incrementPiPos(self):
        self.piPos += 1
        if self.piPos == self.piLen:
            self.piPos = 0

    # See if the current item should be selected or not
    def selectCheck(self, isSelected, hasSelected, extend):

        # If the current item is not selected we may want to select
        if not isSelected:

            # If we are extending or nothing is selected we want to select
            if extend or not hasSelected:
                return True

        return False

    # See if the current item should be deselected or not
    def deselectCheck(self, isSelected, hasSelected, extend):
        # If the current item is selected we may want to deselect
        if isSelected:
            # If something is selected and we're not extending we want to deselect
            if hasSelected and not extend:
                return True

        return False

    # See if there is at least one selected item
    def hasSelected(self, items):
        for item in items:
            if item.select:
                return True

        return False


class Select_init(Operator):
    bl_idname = "mesh.select_by_pi"
    bl_label = "Select by Pi or e"
    bl_description = ("Select Vertices/Edges/Faces based on pi or e for a random-like selection\n"
                      "Number Pi (3.14 etc.) or e (2.71828 - Euler's number)")
    bl_options = {'REGISTER', 'UNDO'}

    e = BoolProperty(
                name="Use e",
                description="Use e as the base of selection instead of pi",
                default=False
                )
    invert = BoolProperty(
                name="Invert",
                description="Invert the selection result",
                default=False
                )
    extend = BoolProperty(
                name="Extend",
                description="Extend the current selection",
                default=False
                )
    start_new = BoolProperty(
            name="Fresh Start",
            default=False,
            description="Start from no previous selection"
            )

    @classmethod
    def poll(cls, context):
        obj = context.active_object
        return (obj and obj.type == 'MESH')

    def execute(self, context):
        Select_by_pi(context, self.e, self.invert, self.extend, self.start_new)

        return {'FINISHED'}