123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909 |
- /**
- * Marlin 3D Printer Firmware
- * Copyright (C) 2016 MarlinFirmware [https://github.com/MarlinFirmware/Marlin]
- *
- * Based on Sprinter and grbl.
- * Copyright (C) 2011 Camiel Gubbels / Erik van der Zalm
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- *
- */
-
- #include "MarlinConfig.h"
-
- #if ENABLED(SDSUPPORT)
-
- #include "cardreader.h"
-
- #include "ultralcd.h"
- #include "stepper.h"
- #include "language.h"
-
- #define LONGEST_FILENAME (longFilename[0] ? longFilename : filename)
-
- CardReader::CardReader() {
- #if ENABLED(SDCARD_SORT_ALPHA)
- sort_count = 0;
- #if ENABLED(SDSORT_GCODE)
- sort_alpha = true;
- sort_folders = FOLDER_SORTING;
- //sort_reverse = false;
- #endif
- #endif
- sdprinting = cardOK = saving = logging = false;
- filesize = 0;
- sdpos = 0;
- file_subcall_ctr = 0;
-
- workDirDepth = 0;
- ZERO(workDirParents);
-
- autostart_stilltocheck = true; //the SD start is delayed, because otherwise the serial cannot answer fast enough to make contact with the host software.
- autostart_index = 0;
-
- //power to SD reader
- #if SDPOWER > -1
- OUT_WRITE(SDPOWER, HIGH);
- #endif // SDPOWER
-
- next_autostart_ms = millis() + 5000;
- }
-
- char *createFilename(char *buffer, const dir_t &p) { //buffer > 12characters
- char *pos = buffer;
- for (uint8_t i = 0; i < 11; i++) {
- if (p.name[i] == ' ') continue;
- if (i == 8) *pos++ = '.';
- *pos++ = p.name[i];
- }
- *pos++ = 0;
- return buffer;
- }
-
- /**
- * Dive into a folder and recurse depth-first to perform a pre-set operation lsAction:
- * LS_Count - Add +1 to nrFiles for every file within the parent
- * LS_GetFilename - Get the filename of the file indexed by nrFile_index
- * LS_SerialPrint - Print the full path and size of each file to serial output
- */
-
- uint16_t nrFile_index;
-
- void CardReader::lsDive(const char *prepend, SdFile parent, const char * const match/*=NULL*/) {
- dir_t p;
- uint8_t cnt = 0;
-
- // Read the next entry from a directory
- while (parent.readDir(p, longFilename) > 0) {
-
- // If the entry is a directory and the action is LS_SerialPrint
- if (DIR_IS_SUBDIR(&p) && lsAction != LS_Count && lsAction != LS_GetFilename) {
-
- // Get the short name for the item, which we know is a folder
- char lfilename[FILENAME_LENGTH];
- createFilename(lfilename, p);
-
- // Allocate enough stack space for the full path to a folder, trailing slash, and nul
- bool prepend_is_empty = (prepend[0] == '\0');
- int len = (prepend_is_empty ? 1 : strlen(prepend)) + strlen(lfilename) + 1 + 1;
- char path[len];
-
- // Append the FOLDERNAME12/ to the passed string.
- // It contains the full path to the "parent" argument.
- // We now have the full path to the item in this folder.
- strcpy(path, prepend_is_empty ? "/" : prepend); // root slash if prepend is empty
- strcat(path, lfilename); // FILENAME_LENGTH-1 characters maximum
- strcat(path, "/"); // 1 character
-
- // Serial.print(path);
-
- // Get a new directory object using the full path
- // and dive recursively into it.
- SdFile dir;
- if (!dir.open(parent, lfilename, O_READ)) {
- if (lsAction == LS_SerialPrint) {
- SERIAL_ECHO_START();
- SERIAL_ECHOPGM(MSG_SD_CANT_OPEN_SUBDIR);
- SERIAL_ECHOLN(lfilename);
- }
- }
- lsDive(path, dir);
- // close() is done automatically by destructor of SdFile
- }
- else {
- uint8_t pn0 = p.name[0];
- if (pn0 == DIR_NAME_FREE) break;
- if (pn0 == DIR_NAME_DELETED || pn0 == '.') continue;
- if (longFilename[0] == '.') continue;
-
- if (!DIR_IS_FILE_OR_SUBDIR(&p) || (p.attributes & DIR_ATT_HIDDEN)) continue;
-
- filenameIsDir = DIR_IS_SUBDIR(&p);
-
- if (!filenameIsDir && (p.name[8] != 'G' || p.name[9] == '~')) continue;
-
- switch (lsAction) { // 1 based file count
- case LS_Count:
- nrFiles++;
- break;
-
- case LS_SerialPrint:
- createFilename(filename, p);
- SERIAL_PROTOCOL(prepend);
- SERIAL_PROTOCOL(filename);
- SERIAL_PROTOCOLCHAR(' ');
- SERIAL_PROTOCOLLN(p.fileSize);
- break;
-
- case LS_GetFilename:
- createFilename(filename, p);
- if (match != NULL) {
- if (strcasecmp(match, filename) == 0) return;
- }
- else if (cnt == nrFile_index) return; // 0 based index
- cnt++;
- break;
- }
-
- }
- } // while readDir
- }
-
- void CardReader::ls() {
- lsAction = LS_SerialPrint;
- root.rewind();
- lsDive("", root);
- }
-
- #if ENABLED(LONG_FILENAME_HOST_SUPPORT)
-
- /**
- * Get a long pretty path based on a DOS 8.3 path
- */
- void CardReader::printLongPath(char *path) {
- lsAction = LS_GetFilename;
-
- int i, pathLen = strlen(path);
-
- // SERIAL_ECHOPGM("Full Path: "); SERIAL_ECHOLN(path);
-
- // Zero out slashes to make segments
- for (i = 0; i < pathLen; i++) if (path[i] == '/') path[i] = '\0';
-
- SdFile diveDir = root; // start from the root for segment 1
- for (i = 0; i < pathLen;) {
-
- if (path[i] == '\0') i++; // move past a single nul
-
- char *segment = &path[i]; // The segment after most slashes
-
- // If a segment is empty (extra-slash) then exit
- if (!*segment) break;
-
- // Go to the next segment
- while (path[++i]) { }
-
- // SERIAL_ECHOPGM("Looking for segment: "); SERIAL_ECHOLN(segment);
-
- // Find the item, setting the long filename
- diveDir.rewind();
- lsDive("", diveDir, segment);
-
- // Print /LongNamePart to serial output
- SERIAL_PROTOCOLCHAR('/');
- SERIAL_PROTOCOL(longFilename[0] ? longFilename : "???");
-
- // If the filename was printed then that's it
- if (!filenameIsDir) break;
-
- // SERIAL_ECHOPGM("Opening dir: "); SERIAL_ECHOLN(segment);
-
- // Open the sub-item as the new dive parent
- SdFile dir;
- if (!dir.open(diveDir, segment, O_READ)) {
- SERIAL_EOL();
- SERIAL_ECHO_START();
- SERIAL_ECHOPGM(MSG_SD_CANT_OPEN_SUBDIR);
- SERIAL_ECHO(segment);
- break;
- }
-
- diveDir.close();
- diveDir = dir;
-
- } // while i<pathLen
-
- SERIAL_EOL();
- }
-
- #endif // LONG_FILENAME_HOST_SUPPORT
-
- void CardReader::initsd() {
- cardOK = false;
- if (root.isOpen()) root.close();
-
- #ifndef SPI_SPEED
- #define SPI_SPEED SPI_FULL_SPEED
- #endif
-
- if (!card.init(SPI_SPEED, SDSS)
- #if defined(LCD_SDSS) && (LCD_SDSS != SDSS)
- && !card.init(SPI_SPEED, LCD_SDSS)
- #endif
- ) {
- //if (!card.init(SPI_HALF_SPEED,SDSS))
- SERIAL_ECHO_START();
- SERIAL_ECHOLNPGM(MSG_SD_INIT_FAIL);
- }
- else if (!volume.init(&card)) {
- SERIAL_ERROR_START();
- SERIAL_ERRORLNPGM(MSG_SD_VOL_INIT_FAIL);
- }
- else if (!root.openRoot(&volume)) {
- SERIAL_ERROR_START();
- SERIAL_ERRORLNPGM(MSG_SD_OPENROOT_FAIL);
- }
- else {
- cardOK = true;
- SERIAL_ECHO_START();
- SERIAL_ECHOLNPGM(MSG_SD_CARD_OK);
- }
- setroot();
- }
-
- void CardReader::setroot() {
- /*if (!workDir.openRoot(&volume)) {
- SERIAL_ECHOLNPGM(MSG_SD_WORKDIR_FAIL);
- }*/
- workDir = root;
- curDir = &workDir;
- #if ENABLED(SDCARD_SORT_ALPHA)
- presort();
- #endif
- }
-
- void CardReader::release() {
- sdprinting = false;
- cardOK = false;
- }
-
- void CardReader::openAndPrintFile(const char *name) {
- char cmd[4 + strlen(name) + 1]; // Room for "M23 ", filename, and null
- sprintf_P(cmd, PSTR("M23 %s"), name);
- for (char *c = &cmd[4]; *c; c++) *c = tolower(*c);
- enqueue_and_echo_command(cmd);
- enqueue_and_echo_commands_P(PSTR("M24"));
- }
-
- void CardReader::startFileprint() {
- if (cardOK) {
- sdprinting = true;
- #if ENABLED(SDCARD_SORT_ALPHA)
- flush_presort();
- #endif
- }
- }
-
- void CardReader::stopSDPrint() {
- sdprinting = false;
- if (isFileOpen()) file.close();
- }
-
- void CardReader::openLogFile(char* name) {
- logging = true;
- openFile(name, false);
- }
-
- void appendAtom(SdFile &file, char *& dst, uint8_t &cnt) {
- file.getFilename(dst);
- while (*dst && cnt < MAXPATHNAMELENGTH) { dst++; cnt++; }
- if (cnt < MAXPATHNAMELENGTH) { *dst = '/'; dst++; cnt++; }
- }
-
- void CardReader::getAbsFilename(char *t) {
- *t++ = '/'; // Root folder
- uint8_t cnt = 1;
-
- for (uint8_t i = 0; i < workDirDepth; i++) // Loop to current work dir
- appendAtom(workDirParents[i], t, cnt);
-
- if (cnt < MAXPATHNAMELENGTH - (FILENAME_LENGTH)) {
- appendAtom(file, t, cnt);
- --t;
- }
- *t = '\0';
- }
-
- void CardReader::openFile(char* name, const bool read, const bool subcall/*=false*/) {
-
- if (!cardOK) return;
-
- uint8_t doing = 0;
- if (isFileOpen()) { // Replacing current file or doing a subroutine
- if (subcall) {
- if (file_subcall_ctr > SD_PROCEDURE_DEPTH - 1) {
- SERIAL_ERROR_START();
- SERIAL_ERRORPGM("trying to call sub-gcode files with too many levels. MAX level is:");
- SERIAL_ERRORLN(SD_PROCEDURE_DEPTH);
- kill(PSTR(MSG_KILLED));
- return;
- }
-
- // Store current filename (based on workDirParents) and position
- getAbsFilename(proc_filenames[file_subcall_ctr]);
- filespos[file_subcall_ctr] = sdpos;
-
- SERIAL_ECHO_START();
- SERIAL_ECHOPAIR("SUBROUTINE CALL target:\"", name);
- SERIAL_ECHOPAIR("\" parent:\"", proc_filenames[file_subcall_ctr]);
- SERIAL_ECHOLNPAIR("\" pos", sdpos);
- file_subcall_ctr++;
- }
- else
- doing = 1;
- }
- else if (subcall) { // Returning from a subcall?
- SERIAL_ECHO_START();
- SERIAL_ECHOLNPGM("END SUBROUTINE");
- }
- else { // Opening fresh file
- doing = 2;
- file_subcall_ctr = 0; // Reset procedure depth in case user cancels print while in procedure
- }
-
- if (doing) {
- SERIAL_ECHO_START();
- SERIAL_ECHOPGM("Now ");
- serialprintPGM(doing == 1 ? PSTR("doing") : PSTR("fresh"));
- SERIAL_ECHOLNPAIR(" file: ", name);
- }
-
- stopSDPrint();
-
- SdFile myDir;
- curDir = &root;
- char *fname = name;
- char *dirname_start, *dirname_end;
-
- if (name[0] == '/') {
- dirname_start = &name[1];
- while (dirname_start != NULL) {
- dirname_end = strchr(dirname_start, '/');
- //SERIAL_ECHOPGM("start:");SERIAL_ECHOLN((int)(dirname_start - name));
- //SERIAL_ECHOPGM("end :");SERIAL_ECHOLN((int)(dirname_end - name));
- if (dirname_end != NULL && dirname_end > dirname_start) {
- char subdirname[FILENAME_LENGTH];
- strncpy(subdirname, dirname_start, dirname_end - dirname_start);
- subdirname[dirname_end - dirname_start] = '\0';
- if (!myDir.open(curDir, subdirname, O_READ)) {
- SERIAL_PROTOCOLPGM(MSG_SD_OPEN_FILE_FAIL);
- SERIAL_PROTOCOL(subdirname);
- SERIAL_PROTOCOLCHAR('.');
- return;
- }
- else {
- //SERIAL_ECHOLNPGM("dive ok");
- }
-
- curDir = &myDir;
- dirname_start = dirname_end + 1;
- }
- else { // the remainder after all /fsa/fdsa/ is the filename
- fname = dirname_start;
- //SERIAL_ECHOLNPGM("remainder");
- //SERIAL_ECHOLN(fname);
- break;
- }
- }
- }
- else
- curDir = &workDir; // Relative paths start in current directory
-
- if (read) {
- if (file.open(curDir, fname, O_READ)) {
- filesize = file.fileSize();
- sdpos = 0;
- SERIAL_PROTOCOLPAIR(MSG_SD_FILE_OPENED, fname);
- SERIAL_PROTOCOLLNPAIR(MSG_SD_SIZE, filesize);
- SERIAL_PROTOCOLLNPGM(MSG_SD_FILE_SELECTED);
- getfilename(0, fname);
- lcd_setstatus(longFilename[0] ? longFilename : fname);
- }
- else {
- SERIAL_PROTOCOLPAIR(MSG_SD_OPEN_FILE_FAIL, fname);
- SERIAL_PROTOCOLCHAR('.');
- SERIAL_EOL();
- }
- }
- else { //write
- if (!file.open(curDir, fname, O_CREAT | O_APPEND | O_WRITE | O_TRUNC)) {
- SERIAL_PROTOCOLPAIR(MSG_SD_OPEN_FILE_FAIL, fname);
- SERIAL_PROTOCOLCHAR('.');
- SERIAL_EOL();
- }
- else {
- saving = true;
- SERIAL_PROTOCOLLNPAIR(MSG_SD_WRITE_TO_FILE, name);
- lcd_setstatus(fname);
- }
- }
- }
-
- void CardReader::removeFile(const char * const name) {
- if (!cardOK) return;
-
- stopSDPrint();
-
- SdFile myDir;
- curDir = &root;
- const char *fname = name;
-
- char *dirname_start, *dirname_end;
- if (name[0] == '/') {
- dirname_start = strchr(name, '/') + 1;
- while (dirname_start != NULL) {
- dirname_end = strchr(dirname_start, '/');
- //SERIAL_ECHOPGM("start:");SERIAL_ECHOLN((int)(dirname_start - name));
- //SERIAL_ECHOPGM("end :");SERIAL_ECHOLN((int)(dirname_end - name));
- if (dirname_end != NULL && dirname_end > dirname_start) {
- char subdirname[FILENAME_LENGTH];
- strncpy(subdirname, dirname_start, dirname_end - dirname_start);
- subdirname[dirname_end - dirname_start] = 0;
- SERIAL_ECHOLN(subdirname);
- if (!myDir.open(curDir, subdirname, O_READ)) {
- SERIAL_PROTOCOLPAIR(MSG_SD_OPEN_FILE_FAIL, subdirname);
- SERIAL_PROTOCOLCHAR('.');
- SERIAL_EOL();
- return;
- }
-
- curDir = &myDir;
- dirname_start = dirname_end + 1;
- }
- else {
- fname = dirname_start;
- break;
- }
- }
- }
- else // Relative paths are rooted in the current directory
- curDir = &workDir;
-
- if (file.remove(curDir, fname)) {
- SERIAL_PROTOCOLPGM("File deleted:");
- SERIAL_PROTOCOLLN(fname);
- sdpos = 0;
- #if ENABLED(SDCARD_SORT_ALPHA)
- presort();
- #endif
- }
- else {
- SERIAL_PROTOCOLPGM("Deletion failed, File: ");
- SERIAL_PROTOCOL(fname);
- SERIAL_PROTOCOLCHAR('.');
- }
- }
-
- void CardReader::getStatus() {
- if (cardOK) {
- SERIAL_PROTOCOLPGM(MSG_SD_PRINTING_BYTE);
- SERIAL_PROTOCOL(sdpos);
- SERIAL_PROTOCOLCHAR('/');
- SERIAL_PROTOCOLLN(filesize);
- }
- else
- SERIAL_PROTOCOLLNPGM(MSG_SD_NOT_PRINTING);
- }
-
- void CardReader::write_command(char *buf) {
- char* begin = buf;
- char* npos = NULL;
- char* end = buf + strlen(buf) - 1;
-
- file.writeError = false;
- if ((npos = strchr(buf, 'N')) != NULL) {
- begin = strchr(npos, ' ') + 1;
- end = strchr(npos, '*') - 1;
- }
- end[1] = '\r';
- end[2] = '\n';
- end[3] = '\0';
- file.write(begin);
- if (file.writeError) {
- SERIAL_ERROR_START();
- SERIAL_ERRORLNPGM(MSG_SD_ERR_WRITE_TO_FILE);
- }
- }
-
- void CardReader::checkautostart(bool force) {
- if (!force && (!autostart_stilltocheck || PENDING(millis(), next_autostart_ms)))
- return;
-
- autostart_stilltocheck = false;
-
- if (!cardOK) {
- initsd();
- if (!cardOK) return; // fail
- }
-
- char autoname[10];
- sprintf_P(autoname, PSTR("auto%i.g"), autostart_index);
- for (int8_t i = 0; i < (int8_t)strlen(autoname); i++) autoname[i] = tolower(autoname[i]);
-
- dir_t p;
-
- root.rewind();
-
- bool found = false;
- while (root.readDir(p, NULL) > 0) {
- for (int8_t i = (int8_t)strlen((char*)p.name); i--;) p.name[i] = tolower(p.name[i]);
- if (p.name[9] != '~' && strncmp((char*)p.name, autoname, 5) == 0) {
- openAndPrintFile(autoname);
- found = true;
- }
- }
- if (!found)
- autostart_index = -1;
- else
- autostart_index++;
- }
-
- void CardReader::closefile(bool store_location) {
- file.sync();
- file.close();
- saving = logging = false;
-
- if (store_location) {
- //future: store printer state, filename and position for continuing a stopped print
- // so one can unplug the printer and continue printing the next day.
- }
- }
-
- /**
- * Get the name of a file in the current directory by index
- */
- void CardReader::getfilename(uint16_t nr, const char * const match/*=NULL*/) {
- #if ENABLED(SDSORT_CACHE_NAMES)
- if (match != NULL) {
- while (nr < sort_count) {
- if (strcasecmp(match, sortshort[nr]) == 0) break;
- nr++;
- }
- }
- if (nr < sort_count) {
- strcpy(filename, sortshort[nr]);
- strcpy(longFilename, sortnames[nr]);
- filenameIsDir = TEST(isDir[nr>>3], nr & 0x07);
- return;
- }
- #endif // SDSORT_CACHE_NAMES
- curDir = &workDir;
- lsAction = LS_GetFilename;
- nrFile_index = nr;
- curDir->rewind();
- lsDive("", *curDir, match);
- }
-
- uint16_t CardReader::getnrfilenames() {
- curDir = &workDir;
- lsAction = LS_Count;
- nrFiles = 0;
- curDir->rewind();
- lsDive("", *curDir);
- //SERIAL_ECHOLN(nrFiles);
- return nrFiles;
- }
-
- void CardReader::chdir(const char * relpath) {
- SdFile newDir;
- SdFile *parent = &root;
-
- if (workDir.isOpen()) parent = &workDir;
-
- if (!newDir.open(*parent, relpath, O_READ)) {
- SERIAL_ECHO_START();
- SERIAL_ECHOPGM(MSG_SD_CANT_ENTER_SUBDIR);
- SERIAL_ECHOLN(relpath);
- }
- else {
- workDir = newDir;
- if (workDirDepth < MAX_DIR_DEPTH)
- workDirParents[workDirDepth++] = workDir;
- #if ENABLED(SDCARD_SORT_ALPHA)
- presort();
- #endif
- }
- }
-
- int8_t CardReader::updir() {
- if (workDirDepth > 0) { // At least 1 dir has been saved
- workDir = --workDirDepth ? workDirParents[workDirDepth - 1] : root; // Use parent, or root if none
- #if ENABLED(SDCARD_SORT_ALPHA)
- presort();
- #endif
- }
- return workDirDepth;
- }
-
- #if ENABLED(SDCARD_SORT_ALPHA)
-
- /**
- * Get the name of a file in the current directory by sort-index
- */
- void CardReader::getfilename_sorted(const uint16_t nr) {
- getfilename(
- #if ENABLED(SDSORT_GCODE)
- sort_alpha &&
- #endif
- (nr < sort_count) ? sort_order[nr] : nr
- );
- }
-
- /**
- * Read all the files and produce a sort key
- *
- * We can do this in 3 ways...
- * - Minimal RAM: Read two filenames at a time sorting along...
- * - Some RAM: Buffer the directory just for this sort
- * - Most RAM: Buffer the directory and return filenames from RAM
- */
- void CardReader::presort() {
-
- // Sorting may be turned off
- #if ENABLED(SDSORT_GCODE)
- if (!sort_alpha) return;
- #endif
-
- // Throw away old sort index
- flush_presort();
-
- // If there are files, sort up to the limit
- uint16_t fileCnt = getnrfilenames();
- if (fileCnt > 0) {
-
- // Never sort more than the max allowed
- // If you use folders to organize, 20 may be enough
- if (fileCnt > SDSORT_LIMIT) fileCnt = SDSORT_LIMIT;
-
- // Sort order is always needed. May be static or dynamic.
- #if ENABLED(SDSORT_DYNAMIC_RAM)
- sort_order = new uint8_t[fileCnt];
- #endif
-
- // Use RAM to store the entire directory during pre-sort.
- // SDSORT_LIMIT should be set to prevent over-allocation.
- #if ENABLED(SDSORT_USES_RAM)
-
- // If using dynamic ram for names, allocate on the heap.
- #if ENABLED(SDSORT_CACHE_NAMES)
- #if ENABLED(SDSORT_DYNAMIC_RAM)
- sortshort = new char*[fileCnt];
- sortnames = new char*[fileCnt];
- #endif
- #elif ENABLED(SDSORT_USES_STACK)
- char sortnames[fileCnt][SORTED_LONGNAME_MAXLEN];
- #endif
-
- // Folder sorting needs 1 bit per entry for flags.
- #if HAS_FOLDER_SORTING
- #if ENABLED(SDSORT_DYNAMIC_RAM)
- isDir = new uint8_t[(fileCnt + 7) >> 3];
- #elif ENABLED(SDSORT_USES_STACK)
- uint8_t isDir[(fileCnt + 7) >> 3];
- #endif
- #endif
-
- #else // !SDSORT_USES_RAM
-
- // By default re-read the names from SD for every compare
- // retaining only two filenames at a time. This is very
- // slow but is safest and uses minimal RAM.
- char name1[LONG_FILENAME_LENGTH + 1];
-
- #endif
-
- if (fileCnt > 1) {
-
- // Init sort order.
- for (uint16_t i = 0; i < fileCnt; i++) {
- sort_order[i] = i;
- // If using RAM then read all filenames now.
- #if ENABLED(SDSORT_USES_RAM)
- getfilename(i);
- #if ENABLED(SDSORT_DYNAMIC_RAM)
- // Use dynamic method to copy long filename
- sortnames[i] = strdup(LONGEST_FILENAME);
- #if ENABLED(SDSORT_CACHE_NAMES)
- // When caching also store the short name, since
- // we're replacing the getfilename() behavior.
- sortshort[i] = strdup(filename);
- #endif
- #else
- // Copy filenames into the static array
- #if SORTED_LONGNAME_MAXLEN != LONG_FILENAME_LENGTH
- strncpy(sortnames[i], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
- sortnames[i][SORTED_LONGNAME_MAXLEN - 1] = '\0';
- #else
- strncpy(sortnames[i], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
- #endif
- #if ENABLED(SDSORT_CACHE_NAMES)
- strcpy(sortshort[i], filename);
- #endif
- #endif
- // char out[30];
- // sprintf_P(out, PSTR("---- %i %s %s"), i, filenameIsDir ? "D" : " ", sortnames[i]);
- // SERIAL_ECHOLN(out);
- #if HAS_FOLDER_SORTING
- const uint16_t bit = i & 0x07, ind = i >> 3;
- if (bit == 0) isDir[ind] = 0x00;
- if (filenameIsDir) isDir[ind] |= _BV(bit);
- #endif
- #endif
- }
-
- // Bubble Sort
- for (uint16_t i = fileCnt; --i;) {
- bool didSwap = false;
- for (uint16_t j = 0; j < i; ++j) {
- const uint16_t o1 = sort_order[j], o2 = sort_order[j + 1];
-
- // Compare names from the array or just the two buffered names
- #if ENABLED(SDSORT_USES_RAM)
- #define _SORT_CMP_NODIR() (strcasecmp(sortnames[o1], sortnames[o2]) > 0)
- #else
- #define _SORT_CMP_NODIR() (strcasecmp(name1, name2) > 0)
- #endif
-
- #if HAS_FOLDER_SORTING
- #if ENABLED(SDSORT_USES_RAM)
- // Folder sorting needs an index and bit to test for folder-ness.
- const uint8_t ind1 = o1 >> 3, bit1 = o1 & 0x07,
- ind2 = o2 >> 3, bit2 = o2 & 0x07;
- #define _SORT_CMP_DIR(fs) \
- (((isDir[ind1] & _BV(bit1)) != 0) == ((isDir[ind2] & _BV(bit2)) != 0) \
- ? _SORT_CMP_NODIR() \
- : (isDir[fs > 0 ? ind1 : ind2] & (fs > 0 ? _BV(bit1) : _BV(bit2))) != 0)
- #else
- #define _SORT_CMP_DIR(fs) ((dir1 == filenameIsDir) ? _SORT_CMP_NODIR() : (fs > 0 ? dir1 : !dir1))
- #endif
- #endif
-
- // The most economical method reads names as-needed
- // throughout the loop. Slow if there are many.
- #if DISABLED(SDSORT_USES_RAM)
- getfilename(o1);
- strcpy(name1, LONGEST_FILENAME); // save (or getfilename below will trounce it)
- #if HAS_FOLDER_SORTING
- bool dir1 = filenameIsDir;
- #endif
- getfilename(o2);
- char *name2 = LONGEST_FILENAME; // use the string in-place
- #endif // !SDSORT_USES_RAM
-
- // Sort the current pair according to settings.
- if (
- #if HAS_FOLDER_SORTING
- #if ENABLED(SDSORT_GCODE)
- sort_folders ? _SORT_CMP_DIR(sort_folders) : _SORT_CMP_NODIR()
- #else
- _SORT_CMP_DIR(FOLDER_SORTING)
- #endif
- #else
- _SORT_CMP_NODIR()
- #endif
- ) {
- sort_order[j] = o2;
- sort_order[j + 1] = o1;
- didSwap = true;
- }
- }
- if (!didSwap) break;
- }
- // Using RAM but not keeping names around
- #if ENABLED(SDSORT_USES_RAM) && DISABLED(SDSORT_CACHE_NAMES)
- #if ENABLED(SDSORT_DYNAMIC_RAM)
- for (uint16_t i = 0; i < fileCnt; ++i) free(sortnames[i]);
- #if HAS_FOLDER_SORTING
- free(isDir);
- #endif
- #endif
- #endif
- }
- else {
- sort_order[0] = 0;
- #if ENABLED(SDSORT_USES_RAM) && ENABLED(SDSORT_CACHE_NAMES)
- getfilename(0);
- #if ENABLED(SDSORT_DYNAMIC_RAM)
- sortnames = new char*[1];
- sortnames[0] = strdup(LONGEST_FILENAME); // malloc
- #if ENABLED(SDSORT_CACHE_NAMES)
- sortshort = new char*[1];
- sortshort[0] = strdup(filename); // malloc
- #endif
- isDir = new uint8_t[1];
- #else
- #if SORTED_LONGNAME_MAXLEN != LONG_FILENAME_LENGTH
- strncpy(sortnames[0], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
- sortnames[0][SORTED_LONGNAME_MAXLEN - 1] = '\0';
- #else
- strncpy(sortnames[0], LONGEST_FILENAME, SORTED_LONGNAME_MAXLEN);
- #endif
- #if ENABLED(SDSORT_CACHE_NAMES)
- strcpy(sortshort[0], filename);
- #endif
- #endif
- isDir[0] = filenameIsDir ? 0x01 : 0x00;
- #endif
- }
-
- sort_count = fileCnt;
- }
- }
-
- void CardReader::flush_presort() {
- if (sort_count > 0) {
- #if ENABLED(SDSORT_DYNAMIC_RAM)
- delete sort_order;
- #if ENABLED(SDSORT_CACHE_NAMES)
- for (uint8_t i = 0; i < sort_count; ++i) {
- free(sortshort[i]); // strdup
- free(sortnames[i]); // strdup
- }
- delete sortshort;
- delete sortnames;
- #endif
- #endif
- sort_count = 0;
- }
- }
-
- #endif // SDCARD_SORT_ALPHA
-
- uint16_t CardReader::get_num_Files() {
- return
- #if ENABLED(SDCARD_SORT_ALPHA) && SDSORT_USES_RAM && SDSORT_CACHE_NAMES
- nrFiles // no need to access the SD card for filenames
- #else
- getnrfilenames()
- #endif
- ;
- }
-
- void CardReader::printingHasFinished() {
- stepper.synchronize();
- file.close();
- if (file_subcall_ctr > 0) { // Heading up to a parent file that called current as a procedure.
- file_subcall_ctr--;
- openFile(proc_filenames[file_subcall_ctr], true, true);
- setIndex(filespos[file_subcall_ctr]);
- startFileprint();
- }
- else {
- sdprinting = false;
- if (SD_FINISHED_STEPPERRELEASE)
- enqueue_and_echo_commands_P(PSTR(SD_FINISHED_RELEASECOMMAND));
- print_job_timer.stop();
- if (print_job_timer.duration() > 60)
- enqueue_and_echo_commands_P(PSTR("M31"));
- #if ENABLED(SDCARD_SORT_ALPHA)
- presort();
- #endif
-
- #if ENABLED(SD_REPRINT_LAST_SELECTED_FILE)
- lcd_reselect_last_file();
- #endif
- }
- }
-
- #endif // SDSUPPORT
|