174 lines
5.9 KiB
Python
174 lines
5.9 KiB
Python
# ***************************************************************************
|
|
# * *
|
|
# * Copyright (c) 2013-2015 - Juergen Riegel <FreeCAD@juergen-riegel.net> *
|
|
# * *
|
|
# * This program is free software; you can redistribute it and/or modify *
|
|
# * it under the terms of the GNU Lesser General Public License (LGPL) *
|
|
# * as published by the Free Software Foundation; either version 2 of *
|
|
# * the License, or (at your option) any later version. *
|
|
# * for detail see the LICENCE text file. *
|
|
# * *
|
|
# * 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 Library General Public License for more details. *
|
|
# * *
|
|
# * You should have received a copy of the GNU Library General Public *
|
|
# * License along with this program; if not, write to the Free Software *
|
|
# * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 *
|
|
# * USA *
|
|
# * *
|
|
# ***************************************************************************
|
|
|
|
# here the usage description if you use this tool from the command line ("__main__")
|
|
CommandlineUsage = """Material - Tool to work with FreeCAD Material definition cards
|
|
|
|
Usage:
|
|
Material [Options] card-file-name
|
|
|
|
Options:
|
|
-c, --output-csv=file-name write a comma separated grid with the material data
|
|
|
|
Exit:
|
|
0 No Error or Warning found
|
|
1 Argument error, wrong or less Arguments given
|
|
|
|
Tool to work with FreeCAD Material definition cards
|
|
|
|
Examples:
|
|
|
|
Material "StandardMaterial/Steel.FCMat"
|
|
|
|
Author:
|
|
(c) 2013 Juergen Riegel
|
|
mail@juergen-riegel.net
|
|
Licence: LGPL
|
|
|
|
Version:
|
|
0.1
|
|
"""
|
|
|
|
|
|
# see comments in module importFCMat, there is a independent parser implementaion for reading and writing FCMat files
|
|
# inside FreeCAD a mixture of these parsers and the ones in importFCMat.py is used
|
|
|
|
|
|
def importFCMat(fileName):
|
|
"Read a FCMat file into a dictionary"
|
|
try:
|
|
import ConfigParser as configparser
|
|
except ImportError:
|
|
import configparser
|
|
|
|
Config = configparser.RawConfigParser()
|
|
Config.optionxform = str
|
|
Config.read(fileName)
|
|
dict1 = {}
|
|
for section in Config.sections():
|
|
options = Config.options(section)
|
|
for option in options:
|
|
dict1[option] = Config.get(section, option)
|
|
|
|
return dict1
|
|
|
|
|
|
def exportFCMat(fileName, matDict):
|
|
"Write a material dictionary to a FCMat file"
|
|
try:
|
|
import ConfigParser as configparser
|
|
except ImportError:
|
|
import configparser
|
|
import string
|
|
Config = configparser.RawConfigParser()
|
|
|
|
# create groups
|
|
for x in matDict.keys():
|
|
grp, key = string.split(x, sep='_')
|
|
if not Config.has_section(grp):
|
|
Config.add_section(grp)
|
|
|
|
# fill groups
|
|
for x in matDict.keys():
|
|
grp, key = string.split(x, sep='_')
|
|
Config.set(grp, key, matDict[x])
|
|
|
|
Preamble = "# This is a FreeCAD material-card file\n\n"
|
|
# Writing our configuration file to 'example.cfg'
|
|
with open(fileName, 'wb') as configfile:
|
|
configfile.write(Preamble)
|
|
Config.write(configfile)
|
|
|
|
|
|
def getMaterialAttributeStructure(withSpaces=None):
|
|
|
|
''''''
|
|
|
|
# material properties
|
|
# see the following resources in the FreeCAD wiki for more information about the material specific properties:
|
|
# https://www.freecadweb.org/wiki/Material_data_model
|
|
# https://www.freecadweb.org/wiki/Material
|
|
|
|
import os
|
|
import xml.etree.ElementTree as ElementTree
|
|
|
|
infile = os.path.dirname(__file__) + os.sep + "MatPropDict.xml"
|
|
tree = ElementTree.parse(infile)
|
|
|
|
if withSpaces:
|
|
# on attributes, add a space before a capital letter, will be used for better display in the ui
|
|
import re
|
|
root = tree.getroot()
|
|
for group in root.getchildren():
|
|
for proper in group.getchildren():
|
|
proper.set('Name', re.sub(r"(\w)([A-Z]+)", r"\1 \2",
|
|
proper.attrib['Name']))
|
|
|
|
return tree
|
|
|
|
|
|
def read_cards_from_path(cards_path):
|
|
from os import listdir
|
|
from os.path import isfile, join, basename, splitext
|
|
from importFCMat import read
|
|
only_files = [f for f in listdir(cards_path) if isfile(join(cards_path, f))]
|
|
mat_files = [f for f in only_files if basename(splitext(f)[1]) == '.FCMat' or basename(splitext(f)[1]) == '.fcmat']
|
|
# print(mat_files)
|
|
mat_cards = []
|
|
for f in sorted(mat_files):
|
|
mat_cards.append(read(join(cards_path, f)))
|
|
return mat_cards
|
|
|
|
|
|
def write_cards_to_path(cards_path, cards_data):
|
|
from importFCMat import write
|
|
from os.path import join
|
|
for card_data in cards_data:
|
|
card_path = join(cards_path, (card_data['CardName'] + '.FCMat'))
|
|
print(card_path)
|
|
write(card_path, card_data)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
import sys
|
|
import getopt
|
|
try:
|
|
opts, args = getopt.getopt(sys.argv[1:], "c:", ["output-csv="])
|
|
except getopt.GetoptError:
|
|
# print help information and exit:
|
|
sys.stderr.write(CommandlineUsage)
|
|
sys.exit(1)
|
|
|
|
# checking on the options
|
|
for o, a in opts:
|
|
if o in ("-c", "--output-csv"):
|
|
print("writing file: " + a + "\n")
|
|
OutPath = a
|
|
|
|
# running through the files
|
|
FileName = args[0]
|
|
|
|
kv_map = importFCMat(FileName)
|
|
for k in kv_map.keys():
|
|
print(repr(k) + " : " + repr(kv_map[k]))
|
|
sys.exit(0) # no error
|