Added SPI write enable command

Author: bbogush

firmware/programmer/spi_flash.c

@@ -26,12 +26,15 @@
 /* 4th addressing cycle */
 #define ADDR_4th_CYCLE(ADDR) (uint8_t)(((ADDR)& 0xFF000000) >> 24)
 
+#define UNDEFINED_CMD 0xFF
+
 typedef struct __attribute__((__packed__))
 {
     uint8_t page_offset;
     uint8_t read_cmd;
     uint8_t read_id_cmd;
     uint8_t write_cmd;
+    uint8_t write_en_cmd;
     uint8_t erase_cmd;
     uint8_t status_cmd;
     uint8_t busy_bit;
@@ -241,11 +244,23 @@ static void spi_flash_read_id(chip_id_t *chip_id)
     spi_flash_deselect_chip();
 }
 
+static void spi_flash_write_enable()
+{
+    if (spi_conf.write_en_cmd == UNDEFINED_CMD)
+        return;
+
+    spi_flash_select_chip();
+    spi_flash_send_byte(spi_conf.write_en_cmd);
+    spi_flash_deselect_chip();
+}
+
 static void spi_flash_write_page_async(uint8_t *buf, uint32_t page,
     uint32_t page_size)
 {
     uint32_t i;
 
+    spi_flash_write_enable();
+
     spi_flash_select_chip();
 
     spi_flash_send_byte(spi_conf.write_cmd);
@@ -302,6 +317,8 @@ static uint32_t spi_flash_erase_block(uint32_t page)
 {
     uint32_t addr = page << spi_conf.page_offset;
 
+    spi_flash_write_enable();
+
     spi_flash_select_chip();
 
     spi_flash_send_byte(spi_conf.erase_cmd);

qt/spi_chip_db.h

@@ -36,6 +36,7 @@ class SpiChipDb : public ChipDb
         CHIP_PARAM_READ_CMD,
         CHIP_PARAM_READ_ID_CMD,
         CHIP_PARAM_WRITE_CMD,
+        CHIP_PARAM_WRITE_EN_CMD,
         CHIP_PARAM_ERASE_CMD,
         CHIP_PARAM_STATUS_CMD,
         CHIP_PARAM_BUSY_BIT,

qt/spi_chip_db_table_model.cpp

@@ -65,6 +65,10 @@ QVariant SpiChipDbTableModel::data(const QModelIndex &index, int role) const
         chipDb->getHexStringFromParam(chipDb->getChipParam(index.row(),
             SpiChipInfo::CHIP_PARAM_WRITE_CMD), paramStr);
         return paramStr;
+    case SpiChipDb::CHIP_PARAM_WRITE_EN_CMD:
+        chipDb->getHexStringFromOptParam(chipDb->getChipParam(index.row(),
+            SpiChipInfo::CHIP_PARAM_WRITE_EN_CMD), paramStr);
+        return paramStr;
     case SpiChipDb::CHIP_PARAM_ERASE_CMD:
         chipDb->getHexStringFromParam(chipDb->getChipParam(index.row(),
             SpiChipInfo::CHIP_PARAM_ERASE_CMD), paramStr);
@@ -122,6 +126,7 @@ QVariant SpiChipDbTableModel::headerData(int section,
         case SpiChipDb::CHIP_PARAM_READ_CMD: return tr("Read com.");
         case SpiChipDb::CHIP_PARAM_READ_ID_CMD: return tr("Read ID com.");
         case SpiChipDb::CHIP_PARAM_WRITE_CMD: return tr("Write com.");
+        case SpiChipDb::CHIP_PARAM_WRITE_EN_CMD: return tr("Write en. com.");
         case SpiChipDb::CHIP_PARAM_ERASE_CMD: return tr("Erase com.");
         case SpiChipDb::CHIP_PARAM_STATUS_CMD: return tr("Status com.");
         case SpiChipDb::CHIP_PARAM_BUSY_BIT: return tr("Busy bit");
@@ -155,6 +160,8 @@ QVariant SpiChipDbTableModel::headerData(int section,
             return tr("Read ID command");
         case SpiChipDb::CHIP_PARAM_WRITE_CMD:
             return tr("Page write command");
+        case SpiChipDb::CHIP_PARAM_WRITE_EN_CMD:
+            return tr("Write enable command");
         case SpiChipDb::CHIP_PARAM_ERASE_CMD:
             return tr("Block erase command");
         case SpiChipDb::CHIP_PARAM_STATUS_CMD:
@@ -246,6 +253,14 @@ bool SpiChipDbTableModel::setData(const QModelIndex &index,
         chipDb->setChipParam(index.row(), SpiChipInfo::CHIP_PARAM_WRITE_CMD,
             paramVal);
         return true;
+    case SpiChipDb::CHIP_PARAM_WRITE_EN_CMD:
+        if (chipDb->getOptParamFromHexString(value.toString(), paramVal))
+            return false;
+        if (!chipDb->isOptParamValid(paramVal, 0x00, 0xFF))
+            return false;
+        chipDb->setChipParam(index.row(), SpiChipInfo::CHIP_PARAM_WRITE_EN_CMD,
+            paramVal);
+        return true;
     case SpiChipDb::CHIP_PARAM_ERASE_CMD:
         if (chipDb->getParamFromHexString(value.toString(), paramVal))
             return false;

qt/spi_chip_info.cpp

@@ -11,6 +11,7 @@ typedef struct __attribute__((__packed__))
     uint8_t read_cmd;
     uint8_t read_id_cmd;
     uint8_t write_cmd;
+    uint8_t write_en_cmd;
     uint8_t erase_cmd;
     uint8_t status_cmd;
     uint8_t busy_bit;
@@ -35,6 +36,7 @@ const QByteArray &SpiChipInfo::getHalConf()
     conf.read_cmd = static_cast<uint8_t>(params[CHIP_PARAM_READ_CMD]);
     conf.read_id_cmd = static_cast<uint8_t>(params[CHIP_PARAM_READ_ID_CMD]);
     conf.write_cmd = static_cast<uint8_t>(params[CHIP_PARAM_WRITE_CMD]);
+    conf.write_en_cmd = static_cast<uint8_t>(params[CHIP_PARAM_WRITE_EN_CMD]);
     conf.erase_cmd = static_cast<uint8_t>(params[CHIP_PARAM_ERASE_CMD]);
     conf.status_cmd = static_cast<uint8_t>(params[CHIP_PARAM_STATUS_CMD]);
     conf.busy_bit = static_cast<uint8_t>(params[CHIP_PARAM_BUSY_BIT]);

qt/spi_chip_info.h

@@ -17,6 +17,7 @@ class SpiChipInfo : public ChipInfo
         CHIP_PARAM_READ_CMD,
         CHIP_PARAM_READ_ID_CMD,
         CHIP_PARAM_WRITE_CMD,
+        CHIP_PARAM_WRITE_EN_CMD,
         CHIP_PARAM_ERASE_CMD,
         CHIP_PARAM_STATUS_CMD,
         CHIP_PARAM_BUSY_BIT,