kopia lustrzana https://github.com/torrinworx/Blend_My_NFTs
101 wiersze
3.2 KiB
Python
101 wiersze
3.2 KiB
Python
# Purpose:
|
|
# The purpose of this file is to apply the materials a user sets in a given .json file to the Variant collection objects
|
|
# also specified in the .json file. The Materialized DNA is then returned in the following format: 1-1-1:1-1-1
|
|
# Where the numbers right of the ":" are the material numbers applied to the respective Variants to the left of the ":"
|
|
|
|
import bpy
|
|
|
|
import json
|
|
import random
|
|
|
|
|
|
def select_material(materialList):
|
|
"""Selects a material from a passed material list. """
|
|
|
|
number_List_Of_i = []
|
|
rarity_List_Of_i = []
|
|
ifZeroBool = None
|
|
|
|
for material in materialList:
|
|
|
|
material_order_num = material.split("_")[1]
|
|
number_List_Of_i.append(material_order_num)
|
|
|
|
material_rarity_percent = material.split("_")[1]
|
|
rarity_List_Of_i.append(float(material_rarity_percent))
|
|
|
|
for x in rarity_List_Of_i:
|
|
if x == 0:
|
|
ifZeroBool = True
|
|
break
|
|
elif x != 0:
|
|
ifZeroBool = False
|
|
|
|
if ifZeroBool:
|
|
selected_material = random.choices(number_List_Of_i, k=1)
|
|
elif not ifZeroBool:
|
|
selected_material = random.choices(number_List_Of_i, weights=rarity_List_Of_i, k=1)
|
|
|
|
return selected_material[0]
|
|
|
|
def get_variant_att_index(variant, hierarchy):
|
|
variant_attribute = None
|
|
|
|
for attribute in hierarchy:
|
|
for variant_h in hierarchy[attribute]:
|
|
if variant_h == variant:
|
|
variant_attribute = attribute
|
|
|
|
attribute_index = list(hierarchy.keys()).index(variant_attribute)
|
|
variant_order_num = variant.split("_")[1]
|
|
return attribute_index, variant_order_num
|
|
|
|
def match_DNA_to_Variant(hierarchy, singleDNA):
|
|
"""
|
|
Matches each DNA number separated by "-" to its attribute, then its variant.
|
|
"""
|
|
|
|
listAttributes = list(hierarchy.keys())
|
|
listDnaDecunstructed = singleDNA.split('-')
|
|
dnaDictionary = {}
|
|
|
|
for i, j in zip(listAttributes, listDnaDecunstructed):
|
|
dnaDictionary[i] = j
|
|
|
|
for x in dnaDictionary:
|
|
for k in hierarchy[x]:
|
|
kNum = hierarchy[x][k]["number"]
|
|
if kNum == dnaDictionary[x]:
|
|
dnaDictionary.update({x: k})
|
|
return dnaDictionary
|
|
|
|
def apply_materials(hierarchy, singleDNA, materialsFile):
|
|
"""
|
|
DNA with applied material example: "1-1:1-1" <Normal DNA>:<Selected Material for each Variant>
|
|
|
|
The Material DNA will select the material for the Variant order number in the NFT DNA based on the Variant Material
|
|
list in the Variant_Material.json file.
|
|
"""
|
|
|
|
singleDNADict = match_DNA_to_Variant(hierarchy, singleDNA)
|
|
materialsFile = json.load(open(materialsFile))
|
|
deconstructed_MaterialDNA = {}
|
|
|
|
for a in singleDNADict:
|
|
complete = False
|
|
for b in materialsFile:
|
|
if singleDNADict[a] == b:
|
|
mat = select_material(materialsFile[b]['Material List'])
|
|
deconstructed_MaterialDNA[a] = mat
|
|
complete = True
|
|
if not complete:
|
|
deconstructed_MaterialDNA[a] = "0"
|
|
|
|
material_DNA = ""
|
|
for a in deconstructed_MaterialDNA:
|
|
num = "-" + str(deconstructed_MaterialDNA[a])
|
|
material_DNA += num
|
|
material_DNA = ''.join(material_DNA.split('-', 1))
|
|
|
|
return f"{singleDNA}:{material_DNA}"
|