1
0

helicalDosecalcSetup7.m 22 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532
  1. %%
  2. % This version was modified by Peter Ferjancic to accept .nrrd file format
  3. % for target. It was also rollbacked on some parameters back to the
  4. % clinical system settings.
  5. % This file has been modified by Surendra Prajapati especially to run
  6. % WiscPlan for KV beams. Works for running locally as well as in Server.
  7. %
  8. % RTF 11/4/05
  9. % Sets up the run files needed for running a Condor convolution
  10. % superposition dose calculation with photon beams.
  11. % Versions of the convolution code that run in both Windows XP, Condor, and
  12. % directly through Matlab can all be created here. The only real
  13. % differences between the files are in the access conventions, which just%
  14. % This has to do with switching forward slashes with backslashes.
  15. % Surendra edits: num_batches, SAD, pitch, xpmin/max, ypmin/max, Mxp, Nphi
  16. % Also edit:
  17. % Kernel should always be named Kernels.mat
  18. function num_batches = helicalDosecalcSetup7(patient_dir)
  19. % -- INPUT:
  20. % patient_dir: specify where kernel and [geometry] files are located
  21. iso = [0 0 0]; % Point about which gantry rotations begin
  22. SAD = 85; % source-axis distance for the x-ray source ##
  23. pitch = 0.86; % fraction of beam with couch translates per rotation
  24. %% --- make the figure prompt for number of angles and beamlets
  25. str = inputdlg({'Enter number of calc cores', 'Enter number of angles (51 default)', ...
  26. 'Enter number of beamlets (64 default)'}, 'input', [1,35], {'3', '51', '64'});
  27. num_batches = str2double(str{1}); % number of cores you want to run the beam calculation
  28. % -- (3 for a 4-core comp to prevent lockdown)
  29. N_angles = str2double(str{2}); % 51 for full resolution
  30. Mxp = str2double(str{3}); % Mxp = 64; number of MLC leaves;
  31. Nyp = 1; % always 1 for Tomo due to binary mlc
  32. % define the overall beam field size for each beam angle
  33. % beam is 40 cm wide in transverse direction and 1-5 cm (usually 2) in y
  34. % direction.
  35. % isocenter is 85 cm from source, ends of jaws are 23 cm from source
  36. xpmin = -20.0; % -field width / 2
  37. xpmax = 20.0; % +field width / 2
  38. % ypmin = -0.3125; % total jaw width is 0.625 cm
  39. % ypmax = 0.3125;
  40. % ypmin = -0.5; % total jaw width is 1 cm
  41. % ypmax = 0.5;
  42. ypmin = -1.25; % total jaw width is 2.5 cm
  43. ypmax = 1.25;
  44. % y-prime points in the z-direction in the CT coordinate system
  45. % ============================================= End of user-supplied inputs
  46. % executable_path = 'C:\010-work\003_localGit\WiscPlan_v2\WiscPlanPhotonkV125\WiscPlanEXE\RyanCsphoton.x86.exe';
  47. executable_path = mfilename('fullpath');
  48. executable_path = [executable_path(1:end-37), 'WiscPlanEXE\RyanCsphoton.x86.exe']
  49. kernel_file = 'Kernels.mat';
  50. geometry_file = fullfile(patient_dir, 'matlab_files\Geometry.mat');
  51. load(geometry_file);
  52. ROI_names = cellfun(@(c)c.name, Geometry.ROIS, 'UniformOutput', false);
  53. [target_idx, okay] = listdlg('ListString', ROI_names, ...
  54. 'SelectionMode', 'single', 'Name', 'Target Selection', ...
  55. 'PromptString', 'Please select the target ROI for beamlet calc. ');
  56. if okay ~= 1
  57. msgbox('Plan creation aborted');
  58. return;
  59. end
  60. targetMask = zeros(size(Geometry.data));
  61. targetMask(Geometry.ROIS{target_idx}.ind) = 1;
  62. % Grozomah - targetMask needs to get a 'double' matrix with the location of
  63. % the target
  64. targetMaskZ = sum(sum(targetMask,1),2);
  65. zBow = (find(targetMaskZ>0, 1, 'first')-1)*Geometry.voxel_size(3) + Geometry.start(3) + ypmin;
  66. zStern = (find(targetMaskZ>0, 1, 'last')+1)*Geometry.voxel_size(3) + Geometry.start(3) + ypmax;
  67. [subi, subj, subk] = ind2sub(size(Geometry.data), Geometry.ROIS{target_idx}.ind);
  68. iso = [Geometry.start(1)+Geometry.voxel_size(1)*mean(subi) ...
  69. Geometry.start(2)+Geometry.voxel_size(2)*mean(subj) 0];
  70. % shift = [0 8 0] % Y X Z
  71. % iso = [iso(1)+Geometry.voxel_size(1)*shift(1) ...
  72. % iso(2)+Geometry.voxel_size(2)*shift(2) ...
  73. % iso(3)+Geometry.voxel_size(3)*shift(3)];
  74. % flags used to select which calculations will be set up
  75. Condor_flag = 1;
  76. ptvInd = target_idx; % PTV index in Geometry.ROIS
  77. fieldWidth = ypmax - ypmin;
  78. % total number of rotations required for treatment
  79. Nrot = ceil(abs(zBow - zStern)/(pitch*fieldWidth));
  80. % Nphi = Nrot*51; % number of angles used in the calculation
  81. Nphi = Nrot * N_angles; % Grozomah
  82. % define the limits of the angles that will be used for the calculation
  83. % ##phimin = 0; % starting angle in radians
  84. % ##phimax = 2*pi*Nphi;
  85. phi = [0:Nphi-1]/Nphi *2*pi*Nrot;
  86. condor_folder = patient_dir;
  87. winxp_folder = 'winxp';
  88. % create names for condor input and output folders
  89. input_folder = '.';
  90. output_folder = '.';
  91. % name of the convolution/superposition executable, which will be in the
  92. % 'code' folder of each respective run type folder
  93. condor_exectuable_name = 'convolutionCondor'; % relative path on the cluster where code will be
  94. winxp_executable_name = 'convolution.exe';
  95. matlab_executable_name = 'convolution_mex'; % name of the Matlab version of the dose calculation code
  96. % set the beam parameters, assuming a helical beam trajectory
  97. % folders that will be inside the 'input' folder
  98. beamspec_folder = 'beamspecfiles'; % directory where beam files will be stored
  99. beamspec_batches_folder = 'beamspecbatches';
  100. beamspec_batch_base_name = 'beamspecbatch'; % base name for a beamlet batch file
  101. kernel_folder = 'kernelfiles'; % folder where kernel information will be saved
  102. kernel_filenames_condor = 'kernelFilenamesCondor.txt';
  103. kernel_filenames_winxp = 'kernelFilenamesWinXP.txt';
  104. % output folders
  105. beamlet_batch_base_name = 'beamletbatch'; % base name for a dose batch file
  106. geometry_header_filename = 'geometryHeader.txt';
  107. geometry_density_filename = 'density.bin'; % save the density, not the Hounsfield units!
  108. % end of user-defined parameters
  109. % check the validity of the user-defined variables
  110. if xpmin >= xpmax
  111. error('xpmin must be less than xpmax.');
  112. end
  113. if ypmin >= ypmax
  114. error('ypmin must be less than ypmax.');b
  115. end
  116. % if phimin > phimax
  117. % error('phimin must be less than or equal to phimax.');
  118. % end
  119. if Mxp <= 0 || Nyp <= 0 || Nphi <= 0
  120. error('Mxp, Nyp, and Nphi must be greater than zero.');
  121. end
  122. if SAD < 50
  123. error('It is recommended that the SAD be greater than 50 cm.');
  124. end
  125. % the xy plane is perpendicular to the isocenter axis of the linac gantry
  126. % size of each beam aperture, making them vectors so extension to
  127. % non-uniform aperture sizes becomes obvious
  128. del_xp = (xpmax - xpmin)/Mxp;
  129. del_yp = (ypmax - ypmin)/Nyp;
  130. % Calculate the xp and yp offsets, which lie at the centers of the
  131. % apertures.
  132. xp = [xpmin:del_xp:xpmax-del_xp] + del_xp/2;
  133. yp = [ypmin:del_yp:ypmax-del_yp] + del_yp/2;
  134. [M,N,Q] = size(Geometry.rhomw);
  135. START = single(Geometry.start - iso);
  136. INC = single(Geometry.voxel_size);
  137. % Grozomah ##
  138. % START(1) = START(1)/10;
  139. % START(2) = START(2)/10;
  140. % INC(1) = INC(1)/10;
  141. % INC(2) = INC(2)/10;
  142. % END= START+[32,32,40].*INC
  143. % define the tumor mask
  144. tumorMask = zeros(size(Geometry.rhomw),'single');
  145. tumorMask(Geometry.ROIS{ptvInd}.ind) = 1;
  146. BW = bwdist(tumorMask);
  147. tumorMaskExp = tumorMask;
  148. tumorMaskExp(BW <= 4) = 1;
  149. P = zeros(Mxp,Nphi);
  150. fprintf('Checking beam''s eye view ...\n');
  151. for p=1:Nphi
  152. % ir and jr form the beam's eye view (BEV)
  153. ir = [-sin(phi(p)); cos(phi(p)); 0];
  154. jr = [0 0 1]';
  155. % kr denotes the beam direction
  156. kr = [cos(phi(p)); sin(phi(p)); 0];
  157. for m=1:Mxp
  158. point1 = single(-kr*SAD + [0 0 zBow + pitch*fieldWidth*phi(p)/(2*pi)]'); % source point
  159. point2 = single(point1 + (SAD*kr + ir*xp(m))*10);
  160. [indVisited,deffVisited] = singleRaytraceClean(tumorMaskExp,START,INC,point1,point2);
  161. if ~isempty(indVisited)
  162. P(m,p) = max(deffVisited);
  163. end
  164. end
  165. end
  166. fprintf('Finished checking BEV\n');
  167. % load data required for the dose calculator
  168. load(kernel_file);
  169. Geometry.rhomw(Geometry.rhomw < 0) = 0;
  170. Geometry.rhomw(Geometry.rhomw < 0.0013) = 0.0013; % fill blank voxels with air
  171. % convert Geometry and kernels to single
  172. f = fieldnames(Kernels);
  173. for k=1:length(f)
  174. if isnumeric(getfield(Kernels,f{k}))
  175. Kernels = setfield(Kernels,f{k},single(getfield(Kernels,f{k})));
  176. end
  177. end
  178. f = fieldnames(Geometry);
  179. for k=1:length(f)
  180. if isnumeric(getfield(Geometry,f{k}))
  181. Geometry = setfield(Geometry,f{k},single(getfield(Geometry,f{k})));
  182. end
  183. end
  184. % account for isocenter
  185. Geometry.start = single(Geometry.start - iso);
  186. % find the total number of beams
  187. Nbeam = Nphi*Mxp*Nyp;
  188. batch_num = 0; % start the count for the number of total batches
  189. % fill up a cell array of beam structures, grouped by batch
  190. clear batches;
  191. batch_num = 0;
  192. batches = cell(1,Nrot); % start the batches cell array (cell array of beam batches)
  193. rotNum = 0;
  194. % calculate beams for all source directions and apertures
  195. for k=1:Nphi % loop through all gantry angles
  196. % calculate the source location for a helical trajectory
  197. beam.SAD = single(SAD);
  198. % the kp vector is the beam direction, ip and jp span the beam's eye view
  199. beam.ip = single([-sin(phi(k)) cos(phi(k)) 0]);
  200. beam.jp = single([0 0 1]);
  201. beam.kp = single([cos(phi(k)) sin(phi(k)) 0]);
  202. beam.y_vec = single(-beam.kp*SAD + [0 0 zBow + pitch*fieldWidth*phi(k)/(2*pi)]);
  203. rotNumOld = rotNum;
  204. rotNum = floor(k/51) + 1; % current rotation number
  205. if rotNum - rotNumOld > 0
  206. beam_num = 0; % if the rotation number has changed, start the beam count over
  207. end
  208. for m=1:Mxp % loop through all apertures in the xp-direction
  209. % calculate the beam if the tomotherapy fluence value is non-zero
  210. if P(m,k) > 0
  211. num = m + (k-1)*Mxp - 1; % beamlet number (overall)
  212. beam_num = beam_num + 1;
  213. % set the beam aperture parameters
  214. beam.del_xp = single(del_xp);
  215. beam.del_yp = single(del_yp);
  216. beam.xp = single(xp(m));
  217. beam.yp = single(0);
  218. beam.num = single(num); % record the beam number to avoid any later ambiguity
  219. batches{rotNum}{beam_num} = beam;
  220. end
  221. end
  222. end
  223. % merge/split batches
  224. all_beams = horzcat(batches{:});
  225. num_beams_per_batch = ceil(numel(all_beams)/num_batches);
  226. batches = cell(num_batches,1);
  227. for k = 1:(num_batches)
  228. beams_idx = 1+num_beams_per_batch*(k-1):num_beams_per_batch*k;
  229. beams_idx (beams_idx>numel(all_beams)) = [];
  230. batches{k} = all_beams(beams_idx);
  231. end
  232. % batches{num_batches} = all_beams(1+num_beams_per_batch*(k):end);
  233. % Everything else in this file is related to saving the batches in a
  234. % useable form.
  235. if Condor_flag == 1
  236. % delete the old submission file
  237. err = rmdir(fullfile(condor_folder,beamspec_batches_folder),'s');
  238. err = rmdir(fullfile(condor_folder,kernel_folder),'s');
  239. % create folders where batch information will be sent
  240. mkdir([condor_folder '/' input_folder '/' beamspec_batches_folder]);
  241. % save the kernels
  242. save_kernels(Kernels,[condor_folder '/' input_folder '/' kernel_folder]);
  243. fprintf(['Successfully saved Condor kernels to ' input_folder '/' kernel_folder '\n']);
  244. % create kernel filenames files
  245. kernel_filenames_CHTC = 'kernelFilenamesCHTC.txt';
  246. kernel_filenames_condor = 'kernelFilenamesCondor.txt';
  247. fid = fopen([condor_folder '/' input_folder '/' kernel_filenames_condor],'w');
  248. fid2 = fopen([condor_folder '/' input_folder '/' kernel_filenames_CHTC],'w');
  249. fprintf(fid,'kernel_header\n');
  250. % fprintf(fid,['./' input_folder '/' kernel_folder '/kernel_header.txt\n']);
  251. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'kernel_header.txt'));
  252. fprintf(fid2,'kernel_header\n');
  253. fprintf(fid2, '%s/%s\n', kernel_folder,'kernel_header.txt');
  254. fprintf(fid,'kernel_radii\n');
  255. % fprintf(fid,['./' input_folder '/' kernel_folder '/radii.bin\n']);
  256. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'radii.bin'));
  257. fprintf(fid2,'kernel_radii\n');
  258. fprintf(fid2, '%s/%s\n', kernel_folder,'radii.bin');
  259. fprintf(fid,'kernel_angles\n');
  260. % fprintf(fid,['./' input_folder '/' kernel_folder '/angles.bin\n']);
  261. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'angles.bin'));
  262. fprintf(fid2,'kernel_angles\n');
  263. fprintf(fid2, '%s/%s\n', kernel_folder,'angles.bin');
  264. fprintf(fid,'kernel_energies\n');
  265. % fprintf(fid,['./' input_folder '/' kernel_folder '/energies.bin\n']);
  266. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'energies.bin'));
  267. fprintf(fid2,'kernel_energies\n');
  268. fprintf(fid2, '%s/%s\n', kernel_folder,'energies.bin');
  269. fprintf(fid,'kernel_primary\n');
  270. % fprintf(fid,['./' input_folder '/' kernel_folder '/primary.bin\n']);
  271. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'primary.bin'));
  272. fprintf(fid2,'kernel_primary\n');
  273. fprintf(fid2, '%s/%s\n', kernel_folder,'primary.bin');
  274. fprintf(fid,'kernel_first_scatter\n');
  275. % fprintf(fid,['./' input_folder '/' kernel_folder '/first_scatter.bin\n']);
  276. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'first_scatter.bin'));
  277. fprintf(fid2,'kernel_first_scatter\n');
  278. fprintf(fid2, '%s/%s\n', kernel_folder,'first_scatter.bin');
  279. fprintf(fid,'kernel_second_scatter\n');
  280. % fprintf(fid,['./' input_folder '/' kernel_folder '/second_scatter.bin\n']);
  281. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'second_scatter.bin'));
  282. fprintf(fid2,'kernel_second_scatter\n');
  283. fprintf(fid2, '%s/%s\n', kernel_folder,'second_scatter.bin');
  284. fprintf(fid,'kernel_multiple_scatter\n');
  285. % fprintf(fid,['./' input_folder '/' kernel_folder '/multiple_scatter.bin\n']);
  286. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'multiple_scatter.bin'));
  287. fprintf(fid2,'kernel_multiple_scatter\n');
  288. fprintf(fid2, '%s/%s\n', kernel_folder,'multiple_scatter.bin');
  289. fprintf(fid,'kernel_brem_annih\n');
  290. % fprintf(fid,['./' input_folder '/' kernel_folder '/brem_annih.bin\n']);
  291. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'brem_annih.bin'));
  292. fprintf(fid2,'kernel_brem_annih\n');
  293. fprintf(fid2, '%s/%s\n', kernel_folder,'brem_annih.bin');
  294. fprintf(fid,'kernel_total\n');
  295. % fprintf(fid,['./' input_folder '/' kernel_folder '/total.bin\n']);
  296. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'total.bin'));
  297. fprintf(fid2,'kernel_total\n');
  298. fprintf(fid2, '%s/%s\n', kernel_folder,'total.bin');
  299. fprintf(fid,'kernel_fluence\n');
  300. % fprintf(fid,['./' input_folder '/' kernel_folder '/fluence.bin\n']);
  301. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'fluence.bin'));
  302. fprintf(fid2,'kernel_fluence\n');
  303. fprintf(fid2, '%s/%s\n', kernel_folder,'fluence.bin');
  304. fprintf(fid,'kernel_mu\n');
  305. % fprintf(fid,['./' input_folder '/' kernel_folder '/mu.bin\n']);
  306. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'mu.bin'));
  307. fprintf(fid2,'kernel_mu\n');
  308. fprintf(fid2, '%s/%s\n', kernel_folder,'mu.bin');
  309. fprintf(fid,'kernel_mu_en\n');
  310. % fprintf(fid,['./' input_folder '/' kernel_folder '/mu_en.bin\n']);
  311. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,kernel_folder,'mu_en.bin'));
  312. fprintf(fid2,'kernel_mu_en\n');
  313. fprintf(fid2, '%s/%s\n', kernel_folder,'mu_en.bin');
  314. fclose(fid);
  315. end
  316. % name for the condor submit file that will be used
  317. condor_submit_file = 'convolutionSubmit.txt';
  318. geometry_filenames_condor = 'geometryFilenamesCondor.txt';
  319. geometry_filenames_CHTC = 'geometryFilenamesCHTC.txt';
  320. % check the geometry file to ensure that it's not in Hounsfield units
  321. if length(find(Geometry.rhomw > 20)) || length(find(Geometry.rhomw < 0))
  322. error('Double check the Geometry structure, it may still be in Hounsfield units!');
  323. end
  324. geometry_folder = 'geometryfiles';
  325. batch_output_folder = 'batchoutput'; % folder to which stdout will be printed
  326. beamlet_batches_folder = 'beamletbatches'; % folder where resulting beamlet batches will be stored
  327. if Condor_flag == 1
  328. mkdir([condor_folder '/' output_folder '/' beamlet_batches_folder]);
  329. mkdir([condor_folder '/' output_folder '/' batch_output_folder]);
  330. save_geometry(Geometry,[condor_folder '/' input_folder '/' geometry_folder],geometry_header_filename,geometry_density_filename);
  331. fprintf(['Successfully saved Condor geometry to ' input_folder '/' geometry_folder '\n']);
  332. % create geometry filenames files
  333. fid = fopen([condor_folder '/' input_folder '/' geometry_filenames_condor],'w');
  334. fid2 = fopen([condor_folder '/' input_folder '/' geometry_filenames_CHTC],'w');
  335. fprintf(fid,'geometry_header\n');
  336. % fprintf(fid,['./' input_folder '/' geometry_folder '/' geometry_header_filename '\n']);
  337. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,geometry_folder,geometry_header_filename));
  338. fprintf(fid2,'geometry_header\n');
  339. fprintf(fid2, '%s/%s\n', geometry_folder,'geometryHeader.txt');
  340. fprintf(fid,'geometry_density\n');
  341. % fprintf(fid,['./' input_folder '/' geometry_folder '/' geometry_density_filename '\n']);
  342. fprintf(fid,'%s\n',fullfile(patient_dir,input_folder,geometry_folder,geometry_density_filename));
  343. fprintf(fid2,'geometry_density\n');
  344. fprintf(fid2, '%s/%s\n', geometry_folder,'density.bin');
  345. fclose(fid);
  346. % write command file
  347. % TODO consistent naming throughout script
  348. for k = 1:numel(batches)
  349. fid = fopen(fullfile(patient_dir,sprintf('run%d.cmd',k-1)), 'w');
  350. fprintf(fid, '"%s" "%s" "%s" "%s" "%s"', executable_path,...
  351. fullfile(patient_dir, kernel_filenames_condor),...
  352. fullfile(patient_dir, geometry_filenames_condor),...
  353. fullfile(patient_dir, 'beamspecbatches', sprintf('beamspecbatch%d.txt',k-1)),...
  354. fullfile(patient_dir, sprintf('batch_dose%d.bin',k-1)));
  355. fclose(fid);
  356. end
  357. % write the condor submit file
  358. % beamspec_batch_filename = ['./' input_folder '/' beamspec_batches_folder '/' beamspec_batch_base_name '$(Process).txt'];
  359. % beamlet_batch_filename = ['./' output_folder '/' beamlet_batches_folder '/' beamlet_batch_base_name '$(Process).bin'];
  360. % fid = fopen([condor_folder '/' condor_submit_file],'w');
  361. % fprintf(fid,'###############################################################\n');
  362. % fprintf(fid,'# Condor submission script for convolution/superposition code\n');
  363. % fprintf(fid,'###############################################################\n\n');
  364. % fprintf(fid,'copy_to_spool = false\n');
  365. % fprintf(fid,['Executable = ' code_folder '/' condor_exectuable_name '\n']);
  366. % fprintf(fid,['arguments = ' input_folder '/' kernel_filenames_condor ' ' input_folder '/' geometry_filenames_condor ' ' beamspec_batch_filename ' ' beamlet_batch_filename '\n']);
  367. % fprintf(fid,['Output = ./' output_folder '/' batch_output_folder '/batchout$(Process).txt\n']);
  368. % fprintf(fid,['Log = ./' output_folder '/' batch_output_folder '/log.txt\n']);
  369. % fprintf(fid,['Queue ' num2str(Nrot)]);
  370. % fclose(fid);
  371. % % write the condor submit file
  372. % beamspec_batch_filename = ['./' input_folder '/' beamspec_batches_folder '/' beamspec_batch_base_name '$(Process).txt'];
  373. % beamlet_batch_filename = ['./' output_folder '/' beamlet_batches_folder '/' beamlet_batch_base_name '$(Process).bin'];
  374. fid = fopen([condor_folder '/' condor_submit_file],'w');
  375. fprintf(fid,'###############################################################\n');
  376. fprintf(fid,'# Condor submission script for convolution/superposition code\n');
  377. fprintf(fid,'###############################################################\n\n');
  378. fprintf(fid,'copy_to_spool = false\n');
  379. fprintf(fid,['Executable = ' condor_exectuable_name '\n']);
  380. fprintf(fid,['Arguments = ' kernel_filenames_CHTC ' ' geometry_filenames_CHTC ' ' beamspec_batch_base_name '$(Process).txt ' 'batch_dose$(Process).bin\n']);
  381. fprintf(fid,['Transfer_input_files = ' kernel_folder ',' geometry_folder ',' beamspec_batches_folder '/' beamspec_batch_base_name '$(Process).txt' ',' kernel_filenames_CHTC ',' geometry_filenames_CHTC '\n']);
  382. fprintf(fid,['Request_memory = 1000' '\n']);
  383. fprintf(fid,['Request_disk = 500000' '\n']);
  384. fprintf(fid,['Output = $(Cluster).out' '\n']);
  385. fprintf(fid,['Log = $(Cluster).log' '\n']);
  386. fprintf(fid,['Error = $(Cluster).err' '\n']);
  387. fprintf(fid,['Queue ' num2str(num_batches) '\n']);
  388. % fclose(fid);
  389. end
  390. % write the batches to files
  391. for n=1:numel(batches)
  392. batch = batches{n}; % current batch
  393. if Condor_flag == 1
  394. save_beamspec_batch(batch,[condor_folder '/' input_folder '/' beamspec_batches_folder],[beamspec_batch_base_name num2str(n-1) '.txt']);
  395. end
  396. end
  397. all_beams{1}.Mxp = Mxp;
  398. all_beams{1}.N_angles = N_angles;
  399. all_beams{1}.num_batches = num_batches;
  400. save([patient_dir '\all_beams.mat'], 'all_beams');
  401. % for k = 1:numel(batches)
  402. % system([fullfile(patient_dir,sprintf('run%d.cmd',k-1)) ' &']);
  403. % end
  404. % Ask for User option to run the dose calculation locally on the computer
  405. % or just to get necessary files for CHTC server
  406. % 'y' means run locally, 'n' means not to run locally on the computer
  407. strBeamlet = '';
  408. while(1)
  409. if strcmpi('y',strBeamlet)
  410. break;
  411. elseif strcmpi('n',strBeamlet)
  412. break;
  413. end
  414. strBeamlet = input('Run beamlet batches dose calculation locally? y/n \n','s');
  415. end
  416. t = datetime('now');
  417. disp(['Calculating ' num2str(size(all_beams, 2)) ' beamlets in ' num2str(size(batches, 1))...
  418. ' batches. Start: ' datestr(t)])
  419. if(strcmpi('y',strBeamlet))
  420. for k = 1:numel(batches)
  421. system([fullfile(patient_dir,sprintf('run%d.cmd',k-1)) ' &']);
  422. end
  423. end
  424. end