Сделан и подключен генератор PDDL-доменов

cg/freecad/Frames/modelExport.py

@@ -0,0 +1,169 @@
+'''
+Эта штука создается с целью проводить все необходимые проверки и манипуляции при движении моделей
+по пайплайну
+
+Планируемые фичи:
+    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
+#