Spaces:
Sleeping
Sleeping
File size: 3,104 Bytes
5c8d4fd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
from fastapi import FastAPI, UploadFile, File
from fastapi.responses import JSONResponse, FileResponse
from fastapi.middleware.cors import CORSMiddleware
import os
app = FastAPI()
app.add_middleware(
CORSMiddleware,
allow_origins=["*"], allow_credentials=True,
allow_methods=["*"], allow_headers=["*"],
)
@app.get("/")
def serve_frontend():
return FileResponse("index.html")
@app.post("/parse")
async def parse_file(file: UploadFile = File(...)):
content = await file.read()
return JSONResponse(content=parse_dcm_assembly(content))
def parse_dcm_assembly(file_bytes):
lines = file_bytes.decode("utf-8").splitlines()
blocks, current = [], []
for line in lines:
if line.startswith("CLASS"):
if current:
blocks.append(current)
current = []
current.append(line.strip())
if current: blocks.append(current)
parts, geometry, constraints = [], {}, []
plane_map, tag_to_plane, label_map = {}, {}, {}
for block in blocks:
head = block[0]
if "CLASS SET " in head:
part, members = {}, []
for line in block:
if "DCM3_TEXT_LABEL" in line:
part['name'] = line.split("DCM3_TEXT_LABEL", 1)[1].strip()
elif "FMEMBER TAGS" in line:
members = list(map(int, line.split()[2:]))
if 'name' in part:
part['member_tags'] = members
parts.append(part)
elif "CLASS PLANE " in head:
pid = int(head.split()[-1])
base, norm = None, None
for line in block:
if "FBASE_POINT" in line:
base = tuple(map(float, line.split()[1:]))
elif "FNORMAL" in line:
norm = tuple(map(float, line.split()[1:]))
if base and norm:
plane_map[pid] = {'base_point': base, 'normal': norm}
elif "CLASS WORK_PLANE " in head:
wid = int(head.split()[-1])
tag, label = None, None
for line in block:
if "FMY_PLANE TAG" in line:
tag = int(line.split()[-1])
elif "DCM3_TEXT_LABEL" in line:
label = line.split("DCM3_TEXT_LABEL", 1)[1].strip()
if tag:
tag_to_plane[wid] = tag
label_map[wid] = label
elif "CLASS CONSTRAINT " in head:
con = {}
for line in block:
if "FEND_1 TAG" in line:
con['from'] = int(line.split()[-1])
elif "FEND_2 TAG" in line:
con['to'] = int(line.split()[-1])
elif "DCM3_TEXT_LABEL" in line:
con['label'] = line.split("DCM3_TEXT_LABEL", 1)[1].strip()
if con: constraints.append(con)
for wid, pid in tag_to_plane.items():
if pid in plane_map:
geometry[wid] = {**plane_map[pid], 'label': label_map.get(wid)}
return { "parts": parts, "geometry": geometry, "constraints": constraints }
|