169 lines
No EOL
8.2 KiB
Python
169 lines
No EOL
8.2 KiB
Python
'''
|
||
Эта штука создается с целью проводить все необходимые проверки и манипуляции при движении моделей
|
||
по пайплайну
|
||
|
||
Планируемые фичи:
|
||
1. Материалы:
|
||
1.1 Проверка на уникальность
|
||
1.2 Проверка на повторяемость
|
||
1.3 Проверки на соответствие материала требованиям
|
||
1.4 Изменение цветов деталей по факту назначения материалов
|
||
1.5 Расчет массы по плотностям деталей и прописывание ее в свойствах детали
|
||
1.6 Указание наименования материала в свойствах детали
|
||
|
||
2. Вспомогательные обьекты и тела:
|
||
2.1. Вспомогательные обьекты имеют отметки о своем статусе
|
||
2.2. Несолиды имеют пометку несолидовости
|
||
2.3 Детали без материалов должны быть с материалами
|
||
|
||
3. [на будущее] Проверка существования разметки
|
||
|
||
4. Дополнение информации в бом-лист (это нужно засунуть в бом, а не сюда)
|
||
|
||
|
||
|
||
'''
|
||
|
||
import FreeCAD as App
|
||
import FreeCADGui as Gui
|
||
from helper.is_solid import is_object_solid
|
||
import json
|
||
|
||
doc = App.ActiveDocument
|
||
|
||
#0 Проверка существования директорий
|
||
#должны быть директории на материалы, на технологические операции, на вспомогательные обьекты
|
||
#если у нас есть директория "анализ" от Fem, то мы переименуем ее в свою директорию и сделаем ей пометку.
|
||
#если мы находим материалы в дереве, которые не относятся к этой директории, мы должны засунуть их в нашу директорию
|
||
#если материалов нет, мы должны указать, что их нет
|
||
|
||
|
||
def materialExistenceCheck(doc):
|
||
|
||
# if "Materials" not in doc.Objects:
|
||
if doc.getObjectsByLabel('Materials') == 0:
|
||
material_group = doc.addObject("App::DocumentObjectGroup", "Materials")
|
||
App.Console.PrintMessage("Директория материалов не обнаружена и была создана вновь\n")
|
||
|
||
|
||
else:
|
||
material_group = doc.getObject("Materials")
|
||
|
||
fem_materials = [obj for obj in doc.Objects if (hasattr(obj, 'Proxy') and hasattr(obj.Proxy, 'Type') and obj.Proxy.Type == 'Fem::MaterialCommon')]
|
||
|
||
if len(fem_materials) > 0:
|
||
for material in fem_materials:
|
||
material_group.addObject(material)
|
||
print('Материалы успешно скомпонованы')
|
||
else:
|
||
|
||
App.Console.PrintMessage("FEM-материалы не заданы. Укажите FEM-материалы и произведите проверку\n")
|
||
Gui.updateGui()
|
||
|
||
return material_group, fem_materials #шоб не пропадало зазря
|
||
|
||
|
||
[material_group, fem_materials ] = materialExistenceCheck(doc) # укомпоновали материалы
|
||
|
||
|
||
#проверим, что материалы правильно описаны:
|
||
# Открываем файл JSON с требованиями и загружаем его содержимое
|
||
# with open('material_requirements.json', 'r') as f:
|
||
# material_requirements = json.load(f)
|
||
material_requirements = ["Density",
|
||
"DiffuseColor",
|
||
"EmissiveColor"
|
||
]
|
||
|
||
|
||
|
||
|
||
def material_isCorrect(fem_materials, material_requirements):
|
||
femmat_trobles = {}
|
||
for femmat in fem_materials:
|
||
for req in material_requirements:
|
||
if not ( req in femmat.Material and femmat.Material[req]):
|
||
if femmat in femmat_trobles:
|
||
femmat_trobles[femmat].append(req)
|
||
else:
|
||
femmat_trobles[femmat] = [req]
|
||
|
||
print(femmat_trobles)
|
||
|
||
|
||
material_isCorrect(fem_materials, material_requirements)
|
||
|
||
|
||
#1
|
||
#1.1 Проверка на уникальность
|
||
|
||
#присвоим каждому телу имя материала и значение плотности
|
||
|
||
def matProperties2bodies(doc, fem_materials):
|
||
#проходимся по всем материалам и присваиваем деталюхам параметры
|
||
#нужно кстати еще цвет им переназначить
|
||
for femmat in fem_materials:
|
||
|
||
body = doc.getObject(femmat.References[0][0].Name).Label #вот эта вот абоба добывает нам тело из перечисления
|
||
#в это тело нам надо добавить два свойства
|
||
|
||
#нужно проверить, есть ли материальные свойства у тела и, если нет, добавить их
|
||
#в следующей серии будем проверять все тела, есть ли у них материал
|
||
if 'MaterialName' in body.PropertiesList:
|
||
if body.getPropertyByName('MaterialName') != femmat.Material['CardName']: #если не совпадает
|
||
print('ВНИМАНИЕ! Тело ' + body.Label + ' имеет переназначение материалов. Проверьте и укажите один (1) правильный материал!')
|
||
body.setProperty('MaterialName', femmat.Material['CardName']) #пусть перепишется на новый
|
||
body.setProperty('Density', femmat.Material['Density']) #плотность
|
||
else:
|
||
body.addObjectProperty('MaterialName', femmat.Material['CardName']) #добавляем название материала
|
||
body.addObjectProperty('Density', femmat.Material['Density']) #добавляем плотность
|
||
|
||
|
||
#считаем массу деталюх
|
||
density, unit = parse_values(femmat.Material['Density'])
|
||
volume = body.Shape.Volume
|
||
body_mass = volume * density
|
||
body.addObjectProperty('Mass', body_mass) #масса деталюхи
|
||
|
||
def bodyWithoutMaterialsCheck(doc):
|
||
objs = doc.Objects
|
||
for obj in objs:
|
||
#проверка на солидовость
|
||
if (is_object_solid(obj) and ('MaterialName' not in obj.PropertiesList)):
|
||
print('Деталь ' + obj.Label + 'не имеет назначения материала')
|
||
|
||
matProperties2bodies(doc, fem_materials)
|
||
bodyWithoutMaterialsCheck(doc)
|
||
|
||
#после выполнения предыдущей функции у нас появились деталюхи с обозначениями материалов
|
||
#все деталюхи без этой позиции нужно как-то выделить, а
|
||
|
||
#гпт предлагает парсить значение единиц измерения
|
||
#я с ним соглашусь
|
||
|
||
def parse_values(value_str):
|
||
value = ""
|
||
unit = ""
|
||
|
||
# Проверяем каждый символ в строке плотности
|
||
for char in value_str:
|
||
if char.isdigit() or char == ".":
|
||
# Символ является цифрой или точкой - добавляем его к числу
|
||
value += char
|
||
else:
|
||
# Символ не является цифрой или точкой - считаем его частью единицы измерения
|
||
unit += char
|
||
|
||
# Преобразуем значение плотности в число
|
||
density = float(value)
|
||
|
||
return value, unit
|
||
|
||
|
||
#todo:
|
||
# сделать проверку на солидовые тела
|
||
#укомпоновать все вспомогательные обьекты в отдельную штуку
|
||
#wip
|
||
|
||
#подготавливаем действия и состояния для PDDL
|
||
# |