def addCatheter(self, catheter=None, coords=None):
print("cathType {}".format(type(catheter)))
if catheter is None:
row, col = coords
catheter = CatheterObj(rowNumber=float(row), colLetter=col)
catheter.makePlottable()
if type(catheter) is tuple:
catheter = catheter[0]
if type(catheter) is CatheterObj:
catheter.setParent(self)
self.CatheterList.append(catheter)
print('adding catheter {} {}'.format(len(self.CatheterList),
self.CatheterList))
self.activeCath = catheter
return catheter
# [79.329114, 33.750000, -236.000000]]))
# measurements.append(np.array([[40.0, 20.0, 4.0]]))
colLetters = ['B', 'f', 'e']
rowInts = [6, 5, 3.5]
# colLetters = ['B', 'b', 'C',
# 'c', 'D', 'd',
# 'E', 'e', 'F',
# 'C', 'C', 'E', 'E']
# rowInts = [3, 2.5, 2,
# 2, 2, 2,
# 2, 2.5, 3,
# 5, 5.5, 5, 5.5]
for ind, colLetter in enumerate(colLetters):
newCath = CatheterObj(rowNumber=rowInts[ind], colLetter=colLetter)
newCath.addDescribingPoint(measurements[ind])
# newCath.setTemplatePosition(row=4, col='b')
cathList.append(newCath)
writer.setCatheterList(cathList)
# exported_plan_path = r'C:\Users\Mark\Documents\Sunnybrook\semple'
exported_plan_path = r'P:\USERS\PUBLIC\Mark Semple\EARTh\tests for importing plans\exported sample plan'
writer.import_plan(exported_plan_path)
writer.execute()
if coords is not None:
print("FILL IN HERE")
pass
def removeCatheter(self, catheter=None):
targetUID = catheter.uid
for ind, cath in enumerate(self.CatheterList):
if cath.uid == targetUID:
print("a match! catheter being removed")
del (self.CatheterList[ind])
N = len(self.CatheterList)
print("There are {} catheters remaining".format(N))
def export_as_CHA(self, sourcePath, destPath):
# self.plan_exporter. DO THING ()
# sort catheters by bottom row, left to right, then up
#
pass
if __name__ == "__main__":
myPlan = Patient_Plan()
myCath = CatheterObj()
myCath.setTemplatePosition_byCode(row=4, col='B')
myCath.addDescribingPoint([44, 39, -40])
myCath.finishMeasuring()
myPlan.addCatheter(myCath)
print(myPlan.CatheterList[0].measurements)
def test_template_assignment():
cath = CatheterObj(row, col)
assert cath.getTemplateAlphanumeric() == template_coord_string
def test_get_VirtualCatheter():
cath = CatheterObj(row, col)
assert cath.to_virtual_catheter().shape[0] == 4
def test_premature_closure():
cath = CatheterObj(row, col)
assert cath.finishMeasuring() is False
def test_close_catheter():
cath = CatheterObj(row, col)
assert cath.add_raw_measurement(dummyPoints) is True
assert cath.finishMeasuring() is True
assert cath.add_raw_measurement(dummyPoints) is False
def test_catheter_adding_multi_points():
cath = CatheterObj(row, col)
assert cath.add_raw_measurement(dummyPoints) is True
assert cath.has_data() is True
assert cath.get_N_raw_pts() == 5
assert cath.get_raw_data().shape == dummyShape
def test_catheter_add_single_point():
cath = CatheterObj(row, col)
cath.add_raw_measurement(dummyPoint)
assert cath.has_data() is True
assert cath.get_N_raw_pts() == 1
def test_catheter_create_empty():
cath = CatheterObj(row, col)
assert cath.has_data() is False