My Marlin configs for Fabrikator Mini and CTC i3 Pro B
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

cardreader.cpp 27KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943
  1. /**
  2. * Marlin 3D Printer Firmware
  3. * Copyright (C) 2016 MarlinFirmware [https://github.com/MarlinFirmware/Marlin]
  4. *
  5. * Based on Sprinter and grbl.
  6. * Copyright (C) 2011 Camiel Gubbels / Erik van der Zalm
  7. *
  8. * This program is free software: you can redistribute it and/or modify
  9. * it under the terms of the GNU General Public License as published by
  10. * the Free Software Foundation, either version 3 of the License, or
  11. * (at your option) any later version.
  12. *
  13. * This program is distributed in the hope that it will be useful,
  14. * but WITHOUT ANY WARRANTY; without even the implied warranty of
  15. * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  16. * GNU General Public License for more details.
  17. *
  18. * You should have received a copy of the GNU General Public License
  19. * along with this program. If not, see <http://www.gnu.org/licenses/>.
  20. *
  21. */
  22. #include "../inc/MarlinConfig.h"
  23. #if ENABLED(SDSUPPORT)
  24. #include "cardreader.h"
  25. #include "../Marlin.h"
  26. #include "../lcd/ultralcd.h"
  27. #include "../module/stepper.h"
  28. #include "../module/printcounter.h"
  29. #include "../core/language.h"
  30. #include "../gcode/queue.h"
  31. #include <ctype.h>
  32. #define LONGEST_FILENAME (longFilename[0] ? longFilename : filename)
  33. CardReader::CardReader() {
  34. #if ENABLED(SDCARD_SORT_ALPHA)
  35. sort_count = 0;
  36. #if ENABLED(SDSORT_GCODE)
  37. sort_alpha = true;
  38. sort_folders = FOLDER_SORTING;
  39. //sort_reverse = false;
  40. #endif
  41. #endif
  42. sdprinting = cardOK = saving = logging = false;
  43. filesize = 0;
  44. sdpos = 0;
  45. file_subcall_ctr = 0;
  46. workDirDepth = 0;
  47. ZERO(workDirParents);
  48. autostart_stilltocheck = true; //the SD start is delayed, because otherwise the serial cannot answer fast enough to make contact with the host software.
  49. autostart_index = 0;
  50. //power to SD reader
  51. #if SDPOWER > -1
  52. OUT_WRITE(SDPOWER, HIGH);
  53. #endif // SDPOWER
  54. next_autostart_ms = millis() + 5000;
  55. }
  56. char *createFilename(char *buffer, const dir_t &p) { //buffer > 12characters
  57. char *pos = buffer;
  58. for (uint8_t i = 0; i < 11; i++) {
  59. if (p.name[i] == ' ') continue;
  60. if (i == 8) *pos++ = '.';
  61. *pos++ = p.name[i];
  62. }
  63. *pos++ = 0;
  64. return buffer;
  65. }
  66. /**
  67. * Dive into a folder and recurse depth-first to perform a pre-set operation lsAction:
  68. * LS_Count - Add +1 to nrFiles for every file within the parent
  69. * LS_GetFilename - Get the filename of the file indexed by nrFile_index
  70. * LS_SerialPrint - Print the full path and size of each file to serial output
  71. */
  72. uint16_t nrFile_index;
  73. void CardReader::lsDive(const char *prepend, SdFile parent, const char * const match/*=NULL*/
  74. #if NUM_SERIAL > 1
  75. , const int8_t port/*= -1*/
  76. #endif
  77. ) {
  78. dir_t p;
  79. uint8_t cnt = 0;
  80. // Read the next entry from a directory
  81. while (parent.readDir(p, longFilename) > 0) {
  82. // If the entry is a directory and the action is LS_SerialPrint
  83. if (DIR_IS_SUBDIR(&p) && lsAction != LS_Count && lsAction != LS_GetFilename) {
  84. // Get the short name for the item, which we know is a folder
  85. char lfilename[FILENAME_LENGTH];
  86. createFilename(lfilename, p);
  87. // Allocate enough stack space for the full path to a folder, trailing slash, and nul
  88. bool prepend_is_empty = (prepend[0] == '\0');
  89. int len = (prepend_is_empty ? 1 : strlen(prepend)) + strlen(lfilename) + 1 + 1;
  90. char path[len];
  91. // Append the FOLDERNAME12/ to the passed string.
  92. // It contains the full path to the "parent" argument.
  93. // We now have the full path to the item in this folder.
  94. strcpy(path, prepend_is_empty ? "/" : prepend); // root slash if prepend is empty
  95. strcat(path, lfilename); // FILENAME_LENGTH-1 characters maximum
  96. strcat(path, "/"); // 1 character
  97. // Serial.print(path);
  98. // Get a new directory object using the full path
  99. // and dive recursively into it.
  100. SdFile dir;
  101. if (!dir.open(parent, lfilename, O_READ)) {
  102. if (lsAction == LS_SerialPrint) {
  103. SERIAL_ECHO_START_P(port);
  104. SERIAL_ECHOPGM_P(port, MSG_SD_CANT_OPEN_SUBDIR);
  105. SERIAL_ECHOLN_P(port, lfilename);
  106. }
  107. }
  108. lsDive(path, dir
  109. #if NUM_SERIAL > 1
  110. , NULL, port
  111. #endif
  112. );
  113. // close() is done automatically by destructor of SdFile
  114. }
  115. else {
  116. uint8_t pn0 = p.name[0];
  117. if (pn0 == DIR_NAME_FREE) break;
  118. if (pn0 == DIR_NAME_DELETED || pn0 == '.') continue;
  119. if (longFilename[0] == '.') continue;
  120. if (!DIR_IS_FILE_OR_SUBDIR(&p) || (p.attributes & DIR_ATT_HIDDEN)) continue;
  121. filenameIsDir = DIR_IS_SUBDIR(&p);
  122. if (!filenameIsDir && (p.name[8] != 'G' || p.name[9] == '~')) continue;
  123. switch (lsAction) { // 1 based file count
  124. case LS_Count:
  125. nrFiles++;
  126. break;
  127. case LS_SerialPrint:
  128. createFilename(filename, p);
  129. SERIAL_PROTOCOL_P(port, prepend);
  130. SERIAL_PROTOCOL_P(port, filename);
  131. SERIAL_PROTOCOLCHAR_P(port, ' ');
  132. SERIAL_PROTOCOLLN_P(port, p.fileSize);
  133. break;
  134. case LS_GetFilename:
  135. createFilename(filename, p);
  136. if (match != NULL) {
  137. if (strcasecmp(match, filename) == 0) return;
  138. }
  139. else if (cnt == nrFile_index) return; // 0 based index
  140. cnt++;
  141. break;
  142. }
  143. }
  144. } // while readDir
  145. }
  146. void CardReader::ls(
  147. #if NUM_SERIAL > 1
  148. const int8_t port
  149. #endif
  150. ) {
  151. lsAction = LS_SerialPrint;
  152. root.rewind();
  153. lsDive("", root
  154. #if NUM_SERIAL > 1
  155. , NULL, port
  156. #endif
  157. );
  158. }
  159. #if ENABLED(LONG_FILENAME_HOST_SUPPORT)
  160. /**
  161. * Get a long pretty path based on a DOS 8.3 path
  162. */
  163. void CardReader::printLongPath(char *path
  164. #if NUM_SERIAL > 1
  165. , const int8_t port/*= -1*/
  166. #endif
  167. ) {
  168. lsAction = LS_GetFilename;
  169. int i, pathLen = strlen(path);
  170. // SERIAL_ECHOPGM_P(port, "Full Path: "); SERIAL_ECHOLN_P(port, path);
  171. // Zero out slashes to make segments
  172. for (i = 0; i < pathLen; i++) if (path[i] == '/') path[i] = '\0';
  173. SdFile diveDir = root; // start from the root for segment 1
  174. for (i = 0; i < pathLen;) {
  175. if (path[i] == '\0') i++; // move past a single nul
  176. char *segment = &path[i]; // The segment after most slashes
  177. // If a segment is empty (extra-slash) then exit
  178. if (!*segment) break;
  179. // Go to the next segment
  180. while (path[++i]) { }
  181. // SERIAL_ECHOPGM_P(port, "Looking for segment: "); SERIAL_ECHOLN_P(port, segment);
  182. // Find the item, setting the long filename
  183. diveDir.rewind();
  184. lsDive("", diveDir, segment
  185. #if NUM_SERIAL > 1
  186. , port
  187. #endif
  188. );
  189. // Print /LongNamePart to serial output
  190. SERIAL_PROTOCOLCHAR_P(port, '/');
  191. SERIAL_PROTOCOL_P(port, longFilename[0] ? longFilename : "???");
  192. // If the filename was printed then that's it
  193. if (!filenameIsDir) break;
  194. // SERIAL_ECHOPGM_P(port, "Opening dir: "); SERIAL_ECHOLN_P(port, segment);
  195. // Open the sub-item as the new dive parent
  196. SdFile dir;
  197. if (!dir.open(diveDir, segment, O_READ)) {
  198. SERIAL_EOL_P(port);
  199. SERIAL_ECHO_START_P(port);
  200. SERIAL_ECHOPGM_P(port, MSG_SD_CANT_OPEN_SUBDIR);
  201. SERIAL_ECHO_P(port, segment);
  202. break;
  203. }
  204. diveDir.close();
  205. diveDir = dir;
  206. } // while i<pathLen
  207. SERIAL_EOL_P(port);
  208. }
  209. #endif // LONG_FILENAME_HOST_SUPPORT
  210. void CardReader::initsd() {
  211. cardOK = false;
  212. if (root.isOpen()) root.close();
  213. #ifndef SPI_SPEED
  214. #define SPI_SPEED SPI_FULL_SPEED
  215. #endif
  216. if (!card.init(SPI_SPEED, SDSS)
  217. #if defined(LCD_SDSS) && (LCD_SDSS != SDSS)
  218. && !card.init(SPI_SPEED, LCD_SDSS)
  219. #endif
  220. ) {
  221. //if (!card.init(SPI_HALF_SPEED,SDSS))
  222. SERIAL_ECHO_START();
  223. SERIAL_ECHOLNPGM(MSG_SD_INIT_FAIL);
  224. }
  225. else if (!volume.init(&card)) {
  226. SERIAL_ERROR_START();
  227. SERIAL_ERRORLNPGM(MSG_SD_VOL_INIT_FAIL);
  228. }
  229. else if (!root.openRoot(&volume)) {
  230. SERIAL_ERROR_START();
  231. SERIAL_ERRORLNPGM(MSG_SD_OPENROOT_FAIL);
  232. }
  233. else {
  234. cardOK = true;
  235. SERIAL_ECHO_START();
  236. SERIAL_ECHOLNPGM(MSG_SD_CARD_OK);
  237. }
  238. setroot();
  239. }
  240. void CardReader::setroot() {
  241. /*if (!workDir.openRoot(&volume)) {
  242. SERIAL_ECHOLNPGM(MSG_SD_WORKDIR_FAIL);
  243. }*/
  244. workDir = root;
  245. curDir = &workDir;
  246. #if ENABLED(SDCARD_SORT_ALPHA)
  247. presort();
  248. #endif
  249. }
  250. void CardReader::release() {
  251. sdprinting = false;
  252. cardOK = false;
  253. }
  254. void CardReader::openAndPrintFile(const char *name) {
  255. char cmd[4 + strlen(name) + 1]; // Room for "M23 ", filename, and null
  256. sprintf_P(cmd, PSTR("M23 %s"), name);
  257. for (char *c = &cmd[4]; *c; c++) *c = tolower(*c);
  258. enqueue_and_echo_command_now(cmd);
  259. enqueue_and_echo_commands_P(PSTR("M24"));
  260. }
  261. void CardReader::startFileprint() {
  262. if (cardOK) {
  263. sdprinting = true;
  264. #if ENABLED(SDCARD_SORT_ALPHA)
  265. flush_presort();
  266. #endif
  267. }
  268. }
  269. void CardReader::stopSDPrint() {
  270. sdprinting = false;
  271. if (isFileOpen()) file.close();
  272. }
  273. void CardReader::openLogFile(char* name) {
  274. logging = true;
  275. openFile(name, false);
  276. }
  277. void appendAtom(SdFile &file, char *& dst, uint8_t &cnt) {
  278. file.getFilename(dst);
  279. while (*dst && cnt < MAXPATHNAMELENGTH) { dst++; cnt++; }
  280. if (cnt < MAXPATHNAMELENGTH) { *dst = '/'; dst++; cnt++; }
  281. }
  282. void CardReader::getAbsFilename(char *t) {
  283. *t++ = '/'; // Root folder
  284. uint8_t cnt = 1;
  285. for (uint8_t i = 0; i < workDirDepth; i++) // Loop to current work dir
  286. appendAtom(workDirParents[i], t, cnt);
  287. if (cnt < MAXPATHNAMELENGTH - (FILENAME_LENGTH)) {
  288. appendAtom(file, t, cnt);
  289. --t;
  290. }
  291. *t = '\0';
  292. }
  293. void CardReader::openFile(char* name, const bool read, const bool subcall/*=false*/) {
  294. if (!cardOK) return;
  295. uint8_t doing = 0;
  296. if (isFileOpen()) { // Replacing current file or doing a subroutine
  297. if (subcall) {
  298. if (file_subcall_ctr > SD_PROCEDURE_DEPTH - 1) {
  299. SERIAL_ERROR_START();
  300. SERIAL_ERRORPGM("trying to call sub-gcode files with too many levels. MAX level is:");
  301. SERIAL_ERRORLN((int)SD_PROCEDURE_DEPTH);
  302. kill(PSTR(MSG_KILLED));
  303. return;
  304. }
  305. // Store current filename (based on workDirParents) and position
  306. getAbsFilename(proc_filenames[file_subcall_ctr]);
  307. filespos[file_subcall_ctr] = sdpos;
  308. SERIAL_ECHO_START();
  309. SERIAL_ECHOPAIR("SUBROUTINE CALL target:\"", name);
  310. SERIAL_ECHOPAIR("\" parent:\"", proc_filenames[file_subcall_ctr]);
  311. SERIAL_ECHOLNPAIR("\" pos", sdpos);
  312. file_subcall_ctr++;
  313. }
  314. else
  315. doing = 1;
  316. }
  317. else if (subcall) { // Returning from a subcall?
  318. SERIAL_ECHO_START();
  319. SERIAL_ECHOLNPGM("END SUBROUTINE");
  320. }
  321. else { // Opening fresh file
  322. doing = 2;
  323. file_subcall_ctr = 0; // Reset procedure depth in case user cancels print while in procedure
  324. }
  325. if (doing) {
  326. SERIAL_ECHO_START();
  327. SERIAL_ECHOPGM("Now ");
  328. serialprintPGM(doing == 1 ? PSTR("doing") : PSTR("fresh"));
  329. SERIAL_ECHOLNPAIR(" file: ", name);
  330. }
  331. stopSDPrint();
  332. SdFile myDir;
  333. curDir = &root;
  334. char *fname = name;
  335. char *dirname_start, *dirname_end;
  336. if (name[0] == '/') {
  337. dirname_start = &name[1];
  338. while (dirname_start != NULL) {
  339. dirname_end = strchr(dirname_start, '/');
  340. //SERIAL_ECHOPGM("start:");SERIAL_ECHOLN((int)(dirname_start - name));
  341. //SERIAL_ECHOPGM("end :");SERIAL_ECHOLN((int)(dirname_end - name));
  342. if (dirname_end != NULL && dirname_end > dirname_start) {
  343. char subdirname[FILENAME_LENGTH];
  344. strncpy(subdirname, dirname_start, dirname_end - dirname_start);
  345. subdirname[dirname_end - dirname_start] = '\0';
  346. if (!myDir.open(curDir, subdirname, O_READ)) {
  347. SERIAL_PROTOCOLPGM(MSG_SD_OPEN_FILE_FAIL);
  348. SERIAL_PROTOCOL(subdirname);
  349. SERIAL_PROTOCOLCHAR('.');
  350. return;
  351. }
  352. else {
  353. //SERIAL_ECHOLNPGM("dive ok");
  354. }
  355. curDir = &myDir;
  356. dirname_start = dirname_end + 1;
  357. }
  358. else { // the remainder after all /fsa/fdsa/ is the filename
  359. fname = dirname_start;
  360. //SERIAL_ECHOLNPGM("remainder");
  361. //SERIAL_ECHOLN(fname);
  362. break;
  363. }
  364. }
  365. }
  366. else
  367. curDir = &workDir; // Relative paths start in current directory
  368. if (read) {
  369. if (file.open(curDir, fname, O_READ)) {
  370. filesize = file.fileSize();
  371. sdpos = 0;
  372. SERIAL_PROTOCOLPAIR(MSG_SD_FILE_OPENED, fname);
  373. SERIAL_PROTOCOLLNPAIR(MSG_SD_SIZE, filesize);
  374. SERIAL_PROTOCOLLNPGM(MSG_SD_FILE_SELECTED);
  375. getfilename(0, fname);
  376. lcd_setstatus(longFilename[0] ? longFilename : fname);
  377. }
  378. else {
  379. SERIAL_PROTOCOLPAIR(MSG_SD_OPEN_FILE_FAIL, fname);
  380. SERIAL_PROTOCOLCHAR('.');
  381. SERIAL_EOL();
  382. }
  383. }
  384. else { //write
  385. if (!file.open(curDir, fname, O_CREAT | O_APPEND | O_WRITE | O_TRUNC)) {
  386. SERIAL_PROTOCOLPAIR(MSG_SD_OPEN_FILE_FAIL, fname);
  387. SERIAL_PROTOCOLCHAR('.');
  388. SERIAL_EOL();
  389. }
  390. else {
  391. saving = true;
  392. SERIAL_PROTOCOLLNPAIR(MSG_SD_WRITE_TO_FILE, name);
  393. lcd_setstatus(fname);
  394. }
  395. }
  396. }
  397. void CardReader::removeFile(const char * const name) {
  398. if (!cardOK) return;
  399. stopSDPrint();
  400. SdFile myDir;
  401. curDir = &root;
  402. const char *fname = name;
  403. char *dirname_start, *dirname_end;
  404. if (name[0] == '/') {
  405. dirname_start = strchr(name, '/') + 1;
  406. while (dirname_start != NULL) {
  407. dirname_end = strchr(dirname_start, '/');
  408. //SERIAL_ECHOPGM("start:");SERIAL_ECHOLN((int)(dirname_start - name));
  409. //SERIAL_ECHOPGM("end :");SERIAL_ECHOLN((int)(dirname_end - name));
  410. if (dirname_end != NULL && dirname_end > dirname_start) {
  411. char subdirname[FILENAME_LENGTH];
  412. strncpy(subdirname, dirname_start, dirname_end - dirname_start);
  413. subdirname[dirname_end - dirname_start] = 0;
  414. SERIAL_ECHOLN(subdirname);
  415. if (!myDir.open(curDir, subdirname, O_READ)) {
  416. SERIAL_PROTOCOLPAIR(MSG_SD_OPEN_FILE_FAIL, subdirname);
  417. SERIAL_PROTOCOLCHAR('.');
  418. SERIAL_EOL();
  419. return;
  420. }
  421. curDir = &myDir;
  422. dirname_start = dirname_end + 1;
  423. }
  424. else {
  425. fname = dirname_start;
  426. break;
  427. }
  428. }
  429. }
  430. else // Relative paths are rooted in the current directory
  431. curDir = &workDir;
  432. if (file.remove(curDir, fname)) {
  433. SERIAL_PROTOCOLPGM("File deleted:");
  434. SERIAL_PROTOCOLLN(fname);
  435. sdpos = 0;
  436. #if ENABLED(SDCARD_SORT_ALPHA)
  437. presort();
  438. #endif
  439. }
  440. else {
  441. SERIAL_PROTOCOLPGM("Deletion failed, File: ");
  442. SERIAL_PROTOCOL(fname);
  443. SERIAL_PROTOCOLCHAR('.');
  444. }
  445. }
  446. void CardReader::getStatus(
  447. #if NUM_SERIAL > 1
  448. const int8_t port/*= -1*/
  449. #endif
  450. ) {
  451. if (cardOK) {
  452. SERIAL_PROTOCOLPGM_P(port, MSG_SD_PRINTING_BYTE);
  453. SERIAL_PROTOCOL_P(port, sdpos);
  454. SERIAL_PROTOCOLCHAR_P(port, '/');
  455. SERIAL_PROTOCOLLN_P(port, filesize);
  456. }
  457. else
  458. SERIAL_PROTOCOLLNPGM_P(port, MSG_SD_NOT_PRINTING);
  459. }
  460. void CardReader::write_command(char *buf) {
  461. char* begin = buf;
  462. char* npos = NULL;
  463. char* end = buf + strlen(buf) - 1;
  464. file.writeError = false;
  465. if ((npos = strchr(buf, 'N')) != NULL) {
  466. begin = strchr(npos, ' ') + 1;
  467. end = strchr(npos, '*') - 1;
  468. }
  469. end[1] = '\r';
  470. end[2] = '\n';
  471. end[3] = '\0';
  472. file.write(begin);
  473. if (file.writeError) {
  474. SERIAL_ERROR_START();
  475. SERIAL_ERRORLNPGM(MSG_SD_ERR_WRITE_TO_FILE);
  476. }
  477. }
  478. void CardReader::checkautostart(bool force) {
  479. if (!force && (!autostart_stilltocheck || PENDING(millis(), next_autostart_ms)))
  480. return;
  481. autostart_stilltocheck = false;
  482. if (!cardOK) {
  483. initsd();
  484. if (!cardOK) return; // fail
  485. }
  486. char autoname[10];
  487. sprintf_P(autoname, PSTR("auto%i.g"), autostart_index);
  488. for (int8_t i = 0; i < (int8_t)strlen(autoname); i++) autoname[i] = tolower(autoname[i]);
  489. dir_t p;
  490. root.rewind();
  491. bool found = false;
  492. while (root.readDir(p, NULL) > 0) {
  493. for (int8_t i = (int8_t)strlen((char*)p.name); i--;) p.name[i] = tolower(p.name[i]);
  494. if (p.name[9] != '~' && strncmp((char*)p.name, autoname, 5) == 0) {
  495. openAndPrintFile(autoname);
  496. found = true;
  497. }
  498. }
  499. if (!found)
  500. autostart_index = -1;
  501. else
  502. autostart_index++;
  503. }
  504. void CardReader::closefile(bool store_location) {
  505. file.sync();
  506. file.close();
  507. saving = logging = false;
  508. if (store_location) {
  509. //future: store printer state, filename and position for continuing a stopped print
  510. // so one can unplug the printer and continue printing the next day.
  511. }
  512. }
  513. /**
  514. * Get the name of a file in the current directory by index
  515. */
  516. void CardReader::getfilename(uint16_t nr, const char * const match/*=NULL*/) {
  517. #if ENABLED(SDSORT_CACHE_NAMES)
  518. if (match != NULL) {
  519. while (nr < sort_count) {
  520. if (strcasecmp(match, sortshort[nr]) == 0) break;
  521. nr++;
  522. }
  523. }
  524. if (nr < sort_count) {
  525. strcpy(filename, sortshort[nr]);
  526. strcpy(longFilename, sortnames[nr]);
  527. filenameIsDir = TEST(isDir[nr>>3], nr & 0x07);
  528. return;
  529. }
  530. #endif // SDSORT_CACHE_NAMES
  531. curDir = &workDir;
  532. lsAction = LS_GetFilename;
  533. nrFile_index = nr;
  534. curDir->rewind();
  535. lsDive("", *curDir, match);
  536. }
  537. uint16_t CardReader::getnrfilenames() {
  538. curDir = &workDir;
  539. lsAction = LS_Count;
  540. nrFiles = 0;
  541. curDir->rewind();
  542. lsDive("", *curDir);
  543. //SERIAL_ECHOLN(nrFiles);
  544. return nrFiles;
  545. }
  546. void CardReader::chdir(const char * relpath) {
  547. SdFile newDir;
  548. SdFile *parent = &root;
  549. if (workDir.isOpen()) parent = &workDir;
  550. if (!newDir.open(*parent, relpath, O_READ)) {
  551. SERIAL_ECHO_START();
  552. SERIAL_ECHOPGM(MSG_SD_CANT_ENTER_SUBDIR);
  553. SERIAL_ECHOLN(relpath);
  554. }
  555. else {
  556. workDir = newDir;
  557. if (workDirDepth < MAX_DIR_DEPTH)
  558. workDirParents[workDirDepth++] = workDir;
  559. #if ENABLED(SDCARD_SORT_ALPHA)
  560. presort();
  561. #endif
  562. }
  563. }
  564. int8_t CardReader::updir() {
  565. if (workDirDepth > 0) { // At least 1 dir has been saved
  566. workDir = --workDirDepth ? workDirParents[workDirDepth - 1] : root; // Use parent, or root if none
  567. #if ENABLED(SDCARD_SORT_ALPHA)
  568. presort();
  569. #endif
  570. }
  571. return workDirDepth;
  572. }
  573. #if ENABLED(SDCARD_SORT_ALPHA)
  574. /**
  575. * Get the name of a file in the current directory by sort-index
  576. */
  577. void CardReader::getfilename_sorted(const uint16_t nr) {
  578. getfilename(
  579. #if ENABLED(SDSORT_GCODE)
  580. sort_alpha &&
  581. #endif
  582. (nr < sort_count) ? sort_order[nr] : nr
  583. );
  584. }
  585. /**
  586. * Read all the files and produce a sort key
  587. *
  588. * We can do this in 3 ways...
  589. * - Minimal RAM: Read two filenames at a time sorting along...
  590. * - Some RAM: Buffer the directory just for this sort
  591. * - Most RAM: Buffer the directory and return filenames from RAM
  592. */
  593. void CardReader::presort() {
  594. // Sorting may be turned off
  595. #if ENABLED(SDSORT_GCODE)
  596. if (!sort_alpha) return;
  597. #endif
  598. // Throw away old sort index
  599. flush_presort();
  600. // If there are files, sort up to the limit
  601. uint16_t fileCnt = getnrfilenames();
  602. if (fileCnt > 0) {
  603. // Never sort more than the max allowed
  604. // If you use folders to organize, 20 may be enough
  605. if (fileCnt > SDSORT_LIMIT) fileCnt = SDSORT_LIMIT;
  606. // Sort order is always needed. May be static or dynamic.
  607. #if ENABLED(SDSORT_DYNAMIC_RAM)
  608. sort_order = new uint8_t[fileCnt];
  609. #endif
  610. // Use RAM to store the entire directory during pre-sort.
  611. // SDSORT_LIMIT should be set to prevent over-allocation.
  612. #if ENABLED(SDSORT_USES_RAM)
  613. // If using dynamic ram for names, allocate on the heap.
  614. #if ENABLED(SDSORT_CACHE_NAMES)
  615. #if ENABLED(SDSORT_DYNAMIC_RAM)
  616. sortshort = new char*[fileCnt];
  617. sortnames = new char*[fileCnt];
  618. #endif
  619. #elif ENABLED(SDSORT_USES_STACK)
  620. char sortnames[fileCnt][SORTED_LONGNAME_MAXLEN];
  621. #endif
  622. // Folder sorting needs 1 bit per entry for flags.
  623. #if HAS_FOLDER_SORTING
  624. #if ENABLED(SDSORT_DYNAMIC_RAM)
  625. isDir = new uint8_t[(fileCnt + 7) >> 3];
  626. #elif ENABLED(SDSORT_USES_STACK)
  627. uint8_t isDir[(fileCnt + 7) >> 3];
  628. #endif
  629. #endif
  630. #else // !SDSORT_USES_RAM
  631. // By default re-read the names from SD for every compare
  632. // retaining only two filenames at a time. This is very
  633. // slow but is safest and uses minimal RAM.
  634. char name1[LONG_FILENAME_LENGTH + 1];
  635. #endif
  636. if (fileCnt > 1) {
  637. // Init sort order.
  638. for (uint16_t i = 0; i < fileCnt; i++) {
  639. sort_order[i] = i;
  640. // If using RAM then read all filenames now.
  641. #if ENABLED(SDSORT_USES_RAM)
  642. getfilename(i);
  643. #if ENABLED(SDSORT_DYNAMIC_RAM)
  644. // Use dynamic method to copy long filename
  645. sortnames[i] = strdup(LONGEST_FILENAME);
  646. #if ENABLED(SDSORT_CACHE_NAMES)
  647. // When caching also store the short name, since
  648. // we're replacing the getfilename() behavior.
  649. sortshort[i] = strdup(filename);
  650. #endif
  651. #else
  652. // Copy filenames into the static array
  653. #if SORTED_LONGNAME_MAXLEN != LONG_FILENAME_LENGTH
  654. strncpy(sortnames[i], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
  655. sortnames[i][SORTED_LONGNAME_MAXLEN - 1] = '\0';
  656. #else
  657. strncpy(sortnames[i], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
  658. #endif
  659. #if ENABLED(SDSORT_CACHE_NAMES)
  660. strcpy(sortshort[i], filename);
  661. #endif
  662. #endif
  663. // char out[30];
  664. // sprintf_P(out, PSTR("---- %i %s %s"), i, filenameIsDir ? "D" : " ", sortnames[i]);
  665. // SERIAL_ECHOLN(out);
  666. #if HAS_FOLDER_SORTING
  667. const uint16_t bit = i & 0x07, ind = i >> 3;
  668. if (bit == 0) isDir[ind] = 0x00;
  669. if (filenameIsDir) isDir[ind] |= _BV(bit);
  670. #endif
  671. #endif
  672. }
  673. // Bubble Sort
  674. for (uint16_t i = fileCnt; --i;) {
  675. bool didSwap = false;
  676. for (uint16_t j = 0; j < i; ++j) {
  677. const uint16_t o1 = sort_order[j], o2 = sort_order[j + 1];
  678. // Compare names from the array or just the two buffered names
  679. #if ENABLED(SDSORT_USES_RAM)
  680. #define _SORT_CMP_NODIR() (strcasecmp(sortnames[o1], sortnames[o2]) > 0)
  681. #else
  682. #define _SORT_CMP_NODIR() (strcasecmp(name1, name2) > 0)
  683. #endif
  684. #if HAS_FOLDER_SORTING
  685. #if ENABLED(SDSORT_USES_RAM)
  686. // Folder sorting needs an index and bit to test for folder-ness.
  687. const uint8_t ind1 = o1 >> 3, bit1 = o1 & 0x07,
  688. ind2 = o2 >> 3, bit2 = o2 & 0x07;
  689. #define _SORT_CMP_DIR(fs) \
  690. (((isDir[ind1] & _BV(bit1)) != 0) == ((isDir[ind2] & _BV(bit2)) != 0) \
  691. ? _SORT_CMP_NODIR() \
  692. : (isDir[fs > 0 ? ind1 : ind2] & (fs > 0 ? _BV(bit1) : _BV(bit2))) != 0)
  693. #else
  694. #define _SORT_CMP_DIR(fs) ((dir1 == filenameIsDir) ? _SORT_CMP_NODIR() : (fs > 0 ? dir1 : !dir1))
  695. #endif
  696. #endif
  697. // The most economical method reads names as-needed
  698. // throughout the loop. Slow if there are many.
  699. #if DISABLED(SDSORT_USES_RAM)
  700. getfilename(o1);
  701. strcpy(name1, LONGEST_FILENAME); // save (or getfilename below will trounce it)
  702. #if HAS_FOLDER_SORTING
  703. bool dir1 = filenameIsDir;
  704. #endif
  705. getfilename(o2);
  706. char *name2 = LONGEST_FILENAME; // use the string in-place
  707. #endif // !SDSORT_USES_RAM
  708. // Sort the current pair according to settings.
  709. if (
  710. #if HAS_FOLDER_SORTING
  711. #if ENABLED(SDSORT_GCODE)
  712. sort_folders ? _SORT_CMP_DIR(sort_folders) : _SORT_CMP_NODIR()
  713. #else
  714. _SORT_CMP_DIR(FOLDER_SORTING)
  715. #endif
  716. #else
  717. _SORT_CMP_NODIR()
  718. #endif
  719. ) {
  720. sort_order[j] = o2;
  721. sort_order[j + 1] = o1;
  722. didSwap = true;
  723. }
  724. }
  725. if (!didSwap) break;
  726. }
  727. // Using RAM but not keeping names around
  728. #if ENABLED(SDSORT_USES_RAM) && DISABLED(SDSORT_CACHE_NAMES)
  729. #if ENABLED(SDSORT_DYNAMIC_RAM)
  730. for (uint16_t i = 0; i < fileCnt; ++i) free(sortnames[i]);
  731. #if HAS_FOLDER_SORTING
  732. free(isDir);
  733. #endif
  734. #endif
  735. #endif
  736. }
  737. else {
  738. sort_order[0] = 0;
  739. #if ENABLED(SDSORT_USES_RAM) && ENABLED(SDSORT_CACHE_NAMES)
  740. getfilename(0);
  741. #if ENABLED(SDSORT_DYNAMIC_RAM)
  742. sortnames = new char*[1];
  743. sortnames[0] = strdup(LONGEST_FILENAME); // malloc
  744. #if ENABLED(SDSORT_CACHE_NAMES)
  745. sortshort = new char*[1];
  746. sortshort[0] = strdup(filename); // malloc
  747. #endif
  748. isDir = new uint8_t[1];
  749. #else
  750. #if SORTED_LONGNAME_MAXLEN != LONG_FILENAME_LENGTH
  751. strncpy(sortnames[0], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
  752. sortnames[0][SORTED_LONGNAME_MAXLEN - 1] = '\0';
  753. #else
  754. strncpy(sortnames[0], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
  755. #endif
  756. #if ENABLED(SDSORT_CACHE_NAMES)
  757. strcpy(sortshort[0], filename);
  758. #endif
  759. #endif
  760. isDir[0] = filenameIsDir ? 0x01 : 0x00;
  761. #endif
  762. }
  763. sort_count = fileCnt;
  764. }
  765. }
  766. void CardReader::flush_presort() {
  767. if (sort_count > 0) {
  768. #if ENABLED(SDSORT_DYNAMIC_RAM)
  769. delete sort_order;
  770. #if ENABLED(SDSORT_CACHE_NAMES)
  771. for (uint8_t i = 0; i < sort_count; ++i) {
  772. free(sortshort[i]); // strdup
  773. free(sortnames[i]); // strdup
  774. }
  775. delete sortshort;
  776. delete sortnames;
  777. #endif
  778. #endif
  779. sort_count = 0;
  780. }
  781. }
  782. #endif // SDCARD_SORT_ALPHA
  783. uint16_t CardReader::get_num_Files() {
  784. return
  785. #if ENABLED(SDCARD_SORT_ALPHA) && SDSORT_USES_RAM && SDSORT_CACHE_NAMES
  786. nrFiles // no need to access the SD card for filenames
  787. #else
  788. getnrfilenames()
  789. #endif
  790. ;
  791. }
  792. void CardReader::printingHasFinished() {
  793. stepper.synchronize();
  794. file.close();
  795. if (file_subcall_ctr > 0) { // Heading up to a parent file that called current as a procedure.
  796. file_subcall_ctr--;
  797. openFile(proc_filenames[file_subcall_ctr], true, true);
  798. setIndex(filespos[file_subcall_ctr]);
  799. startFileprint();
  800. }
  801. else {
  802. sdprinting = false;
  803. #if ENABLED(SD_FINISHED_STEPPERRELEASE) && defined(SD_FINISHED_RELEASECOMMAND)
  804. stepper.cleaning_buffer_counter = 1; // The command will fire from the Stepper ISR
  805. #endif
  806. print_job_timer.stop();
  807. if (print_job_timer.duration() > 60)
  808. enqueue_and_echo_commands_P(PSTR("M31"));
  809. #if ENABLED(SDCARD_SORT_ALPHA)
  810. presort();
  811. #endif
  812. #if ENABLED(SD_REPRINT_LAST_SELECTED_FILE)
  813. lcd_reselect_last_file();
  814. #endif
  815. }
  816. }
  817. #endif // SDSUPPORT