"""Extract points from a dxf file, and save in csv format Usage: dxf_to_csv [-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: > dxf_to_csv drawing.dxf -l Convert all layers: > dxf_to_csv drawing.dxf Convert specific layer: > dxf_to_csv 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) # Ensure dataframe is not empty if points.shape[0] == 0: if layer_name: raise ValueError('No objects found on layer {}'.format(layer_name)) else: raise ValueError('No objects found.') # 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) # Print usage if no arguments are provided if len(sys.argv) == 1: parser.print_help(sys.stderr) sys.exit(1) # Parse arguments args = parser.parse_args() 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()