Prepare 0.3.1 release
parent
b5a314588c
commit
0557d85403
@ -1,134 +0,0 @@
|
||||
"""Extract points from a dxf file, and save in csv format
|
||||
|
||||
Usage:
|
||||
dxf_to_csv.py [-h] [-n LAYER_NAME] [-l] dxf_name
|
||||
|
||||
Positional arguments:
|
||||
dxf_name path to dxf file
|
||||
|
||||
Optional arguments:
|
||||
-h, --help show this help message and exit
|
||||
-n LAYER_NAME name of layer
|
||||
-l show list of layers
|
||||
|
||||
Examples:
|
||||
|
||||
Show available layers:
|
||||
> python dxf_to_csv.py drawing.dxf -l
|
||||
|
||||
Convert all layers:
|
||||
> python dxf_to_csv.py drawing.dxf
|
||||
|
||||
Convert specific layer:
|
||||
> python dxf_to_csv.py drawing.dxf -n LAYER_NAME
|
||||
|
||||
"""
|
||||
|
||||
import os
|
||||
import sys
|
||||
import numpy as np
|
||||
import pandas as pd
|
||||
import matplotlib.pyplot as plt
|
||||
import argparse
|
||||
import dxfgrabber
|
||||
|
||||
|
||||
def point_to_row(point, layer, dxf_type, dxf_id):
|
||||
"""Create a dict from a point with layer and e properties"""
|
||||
row = {key: val for key, val in zip(['x', 'y', 'z'], [*point])}
|
||||
row['layer'] = layer
|
||||
row['type'] = dxf_type
|
||||
row['id'] = dxf_id
|
||||
|
||||
return row
|
||||
|
||||
|
||||
def print_layers(dxf_name):
|
||||
dxf = dxfgrabber.readfile(dxf_name)
|
||||
|
||||
# Get data from DTM
|
||||
print('Available layers in {}'.format(dxf_name))
|
||||
layers = sorted([layer.name for layer in dxf.layers])
|
||||
for layer in layers:
|
||||
print(layer)
|
||||
|
||||
|
||||
def convert(dxf_name, layer_name):
|
||||
# Read dxf file
|
||||
dxf = dxfgrabber.readfile(dxf_name)
|
||||
|
||||
if layer_name:
|
||||
print("\nExtracting data from layer '{}'".format(layer_name))
|
||||
entities = [
|
||||
e for e in dxf.entities if e.layer.lower() == layer_name.lower()
|
||||
]
|
||||
else:
|
||||
# Take all layers if no layer name given
|
||||
print('\nExtracing data from all layers')
|
||||
entities = [e for e in dxf.entities]
|
||||
|
||||
# Get coordinates of entities
|
||||
data = []
|
||||
for e in entities:
|
||||
if e.dxftype == 'POINT':
|
||||
point = e.point
|
||||
row = point_to_row(point, e.layer, e.dxftype, e.handle)
|
||||
data.append(row)
|
||||
elif e.dxftype == 'LINE':
|
||||
points = [e.start, e.end]
|
||||
for point in points:
|
||||
row = point_to_row(point, e.layer, e.dxftype, e.handle)
|
||||
data.append(row)
|
||||
elif e.dxftype == 'POLYLINE':
|
||||
for vertex in e.vertices:
|
||||
point = vertex.location
|
||||
row = point_to_row(point, e.layer, e.dxftype, e.handle)
|
||||
data.append(row)
|
||||
elif e.dxftype == 'LWPOLYLINE':
|
||||
for point in e.points:
|
||||
row = point_to_row(point, e.layer, e.dxftype, e.handle)
|
||||
data.append(row)
|
||||
|
||||
# Add to dataframe
|
||||
points = pd.DataFrame(data)
|
||||
|
||||
# Check if there are z coordinates
|
||||
if 'z' not in points.columns:
|
||||
points['z'] = np.nan
|
||||
|
||||
# Reorder columns
|
||||
points = points[['layer', 'type', 'id', 'x', 'y', 'z']]
|
||||
|
||||
# Export points
|
||||
if layer_name:
|
||||
csv_name = dxf_name.replace('.dxf', '-{}.csv'.format(layer_name))
|
||||
else:
|
||||
csv_name = dxf_name.replace('.dxf', '.csv')
|
||||
points.to_csv(csv_name, index=False)
|
||||
|
||||
|
||||
def main():
|
||||
# Set up command line arguments
|
||||
parser = argparse.ArgumentParser()
|
||||
parser.add_argument('dxf_name', help='path to dxf file', default=None)
|
||||
parser.add_argument(
|
||||
'-n', metavar='LAYER_NAME', help='name of layer', default=None)
|
||||
parser.add_argument(
|
||||
'-l', action='store_true', help='show list of layers', default=False)
|
||||
args = parser.parse_args()
|
||||
|
||||
# Parse arguments
|
||||
dxf_name = args.dxf_name
|
||||
layer_name = args.n
|
||||
show_layers_only = args.l
|
||||
|
||||
if show_layers_only:
|
||||
# Print list of layers only
|
||||
print_layers(dxf_name)
|
||||
else:
|
||||
# Convert to csv
|
||||
convert(dxf_name, layer_name)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
main()
|
||||
Loading…
Reference in New Issue