Выбрать главу

zmax = 24

ds = 2.0/xmax 

def makeFlag(name, origin, invert):

    # Добавление нового меша, который будет флагом

    me = bpy.data.meshes.new(name)

    flag = bpy.data.objects.new(name, me)

    scn = bpy.context.scene

    scn.objects.link(flag)

    scn.objects.active = flag  

    # Построение меша флага

    verts = []

    faces = []

    for x in range(xmax):

        for z in range(zmax):

            verts.append(((x+0.5)*ds, 0, z*ds))

            if x > 0 and z > 0:

                faces.append(((x-1)*zmax+(z-1), (x-1)*zmax+z, x*zmax+z, x*zmax+(z-1)))

    me.from_pydata(verts, [], faces)

    me.update(calc_edges=True)

    flag.location = origin 

    # Добавление групп вершин

    grp = flag.vertex_groups.new('Pole')

    for v in me.vertices:

        w = 1.5 - 7*v.co[0]

        if invert:

            if w > 1:

                 grp.add([v.index], 0.0, 'REPLACE')

            else:

                 grp.add([v.index], 1-w, 'REPLACE')

        else:

            if w > 1:

                 grp.add([v.index], 1.0, 'REPLACE')

            elif w > 0:

                 grp.add([v.index], w, 'REPLACE')

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

    bpy.ops.object.shade_smooth()

    return flag 

def makePole(origin):

    bpy.ops.mesh.primitive_cylinder_add(

        vertices=32,

        radius=ds/2,

        depth=1,

        cap_ends=True)

    bpy.ops.transform.resize(value=(1, 1, 2.5))

    pole = bpy.context.object

    pole.location = origin

    return pole 

def addSoftBodyModifier(ob):

    mod = ob.modifiers.new(name='SoftBody', type='SOFT_BODY')

    sbset = mod.settings 

    # Мягкое тело

    sbset.friction = 0.3

    sbset.speed = 1.4

    sbset.mass = 0.9 

    # Цель

    sbset.goal_default = 0.3

    sbset.goal_spring = 0.1

    sbset.goal_friction = 0.1

    sbset.vertex_group_goal = 'Pole'  

    # Рёбра мягкого тела

    sbset.pull = 0.1

    sbset.push = 0.1

    sbset.bend = 0.1

    sbset.aerodynamics_type = 'LIFT_FORCE'

    sbset.aero = 0.5 

    #Веса эффектора

    ew = sbset.effector_weights

    ew.gravity = 0.1

    ew.wind = 0.8

    return 

def addClothModifier(ob):

    cloth = ob.modifiers.new(name='Cloth', type='CLOTH')

    cset = cloth.settings 

    cset.quality = 4

    cset.mass = 0.2

    cset.structural_stiffness = 0.5

    cset.bending_stiffness = 0.05

    cset.spring_damping = 0

    cset.air_damping = 0.3

    cset.use_pin_cloth = True

    cset.vertex_group_mass = 'Pole'  

    #Веса эффектора

    ew = cset.effector_weights

    ew.gravity = 0.1

    ew.wind = 1.0 return  

def addWindEffector(origin):

    # Добавление эффектора ветра

    bpy.ops.object.effector_add(

        type='WIND',

        location=origin,

        rotation=(pi/2,0,0))

    wind = bpy.context.object

    fset = wind.field

    fset.strength = -2.0

    fset.noise = 10.0

    fset.flow = 0.8

    fset.shape = 'PLANE'

    return 

def addFlagMaterial(name, ob, color1, color2):

    # Текстура флага

    tex = bpy.data.textures.new('Flag', type = 'WOOD')

    tex.noise_basis_2 = 'TRI'

    tex.wood_type = 'RINGS' 

    # Создание материала

    mat = bpy.data.materials.new(name)

    mat.diffuse_color = color1 

    # Добавление текстурного слота для текстуры цвета

    mtex = mat.texture_slots.add()

    mtex.texture = tex

    mtex.texture_coords = 'ORCO'

    mtex.use_map_color_diffuse = True

    mtex.color = color2 

    # Добавление материала к флагу

    ob.data.materials.append(mat)

    return mat 

def createDisplacementTexture(mat):

    tex = bpy.data.textures.new('Flag', type = 'WOOD')

    tex.noise_basis_2 = 'SIN'

    tex.wood_type = 'BANDNOISE'

    tex.noise_type = 'SOFT_NOISE'

    tex.noise_scale = 0.576

    tex.turbulence = 9.0

    # Сохранение текстуры в материале для легкого доступа. Не необходимо на самом деле.

    mtex = mat.texture_slots.add()

    mtex.texture = tex

    mat.use_textures[1] = False

    return tex 

def addDisplacementModifier(ob, tex, vgrp, empty):

    mod = ob.modifiers.new('Displace', 'DISPLACE')

    mod.texture = tex

mod.vertex_group = vgrp

    mod.direction = 'NORMAL'

    mod.texture_coords = 'OBJECT'

    mod.texture_coords_object = empty

    mod.mid_level = 0.0

    mod.strength = 0.1

    print("'%s' '%s'" % (vgrp, mod.vertex_group))

    mod.vertex_group = vgrp

    print("'%s' '%s'" % (vgrp, mod.vertex_group))

    return mod 

def createAndAnimateEmpty(origin):

    bpy.ops.object.add(type='EMPTY', location=origin)

    empty = bpy.context.object

    scn = bpy.context.scene

    scn.frame_current = 1