Skip to content

Commit 7e3171a

Browse files
Add files via upload
1 parent 889d4fd commit 7e3171a

1 file changed

Lines changed: 36 additions & 0 deletions

File tree

mousearm/getMuscleLengths.py

Lines changed: 36 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,36 @@
1+
import opensim as osim
2+
import os as os
3+
import glob
4+
import shutil
5+
6+
def runFiberReport(path_to_model, path_to_sto, storage_path):
7+
analyze = osim.AnalyzeTool()
8+
analyze.setName("analyze")
9+
analyze.setModelFilename(path_to_model)
10+
analyze.setStatesFileName(path_to_sto)
11+
analyze.updAnalysisSet().cloneAndAppend(osim.MuscleAnalysis())
12+
analyze.updControllerSet().cloneAndAppend(
13+
osim.PrescribedController(path_to_sto))
14+
analyze.printToXML(storage_path)
15+
analyze = osim.AnalyzeTool(storage_path)
16+
analyze.run()
17+
18+
rset_name = "reachset_1/" # change this accordingly
19+
reach_dir= "C:\\Users\\matte\\Downloads\\MouseArmProject_v1_0\\analysis_and_processing_scripts\\sim scripts\\data\\Predicted EMG\\Test Reaches\\reachset_1" # change to your path
20+
base_dir = "C:\\Users\\matte\\Downloads\\MouseArmProject_v1_0\\analysis_and_processing_scripts\\sim scripts\\data\\Predicted EMG\\Test Reaches\\" # change to your path
21+
model_path = "C:\\Users\\matte\\Downloads\\MouseArmProject_v1_0\\model\\model_toScale.osim" # change to your path
22+
for g2 in glob.glob(reach_dir+"//muscle_solution_*"):
23+
print(g2)
24+
gend = g2.split("_")
25+
gend = gend[-1]
26+
gend = gend.split(".")
27+
gend = gend[0]
28+
s2 = str(g2).replace(".sto",".xml").replace("muscle_solution","fiber_results")
29+
print(s2)
30+
runFiberReport(model_path, g2,s2)
31+
32+
for ana in glob.glob(base_dir+"*"):
33+
if "analyze" in ana:
34+
ana2 = str(ana).replace("analyze_MuscleAnalysis_",rset_name).replace(".sto","_"+str(gend)+".sto").replace("\\","/")
35+
shutil.copy(ana, ana2)
36+
print(ana,ana2)

0 commit comments

Comments
 (0)