Arduino for STM32

Code: Select all

/* spi_dma_ws.h */


void spi_dma_transmit();

extern uint16_t adc_pa5;

// some colours
const uint8_t ws_off[3] = { 0, 0, 0 };
const uint8_t ws_green[3] = { 25, 0, 0 };
const uint8_t ws_red[3] = { 0, 25, 0 };
const uint8_t ws_blue[3] = { 0, 0, 25 };
const uint8_t ws_white[3] = { 25, 25, 25 };
const uint8_t ws_purple[3] = { 0, 25, 25 };
const uint8_t ws_yell[3] = { 25, 25, 0 };
const uint8_t ws_turc[3] = { 25, 0, 25 };


uint8_t ws3[nb_ws][3];        // has the 3 colours of each RGB chip
uint8_t ws33[nb_ws * 3 * 3];  // has the 3 SPI bytes for each the 3 colours of each RGB chip
///  example with some colours
/*
void ws_3() {
  for (int c = 0; c < 3; c++) {
    ws3[0][c] = ws_green[c];
    ws3[1][c] = ws_red[c];
    ws3[2][c] = ws_blue[c];
    ws3[3][c] = ws_white[c];
    ws3[4][c] = ws_off[c];
    ws3[5][c] = ws_yell[c];
    ws3[6][c] = ws_turc[c];
    ws3[7][c] = ws_purple[c];
  }
}
*/
void ws3_green(uint8_t ws_chip) {
    for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = ws_green[c];
  }
}
void ws3_red(uint8_t ws_chip) {
  for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = ws_red[c];
  }
}
void ws3_blue(uint8_t ws_chip) {
  for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = ws_blue[c];
  }
}
void ws3_white(uint8_t ws_chip) {
  for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = ws_green[c];
  }
}
void ws3_off(uint8_t ws_chip) {
  for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = 0;
  }
}
void ws3_yell(uint8_t ws_chip) {
  for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = ws_yell[c];
  }
}
void ws3_turc(uint8_t ws_chip) {
  for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = ws_turc[c];
  }
}
void ws3_purple(uint8_t ws_chip) {
  for (int c = 0; c < 3; c++) {
    ws3[ws_chip-1][c] = ws_purple[c];
  }
}

//  7 6 5 4 3 2 1 0 7 6 5 4 3 2 1 0 7 6 5 4 3 2 1 0
// |‾|_._|‾|_._|‾|_._|‾|_._|‾|_._|‾|_._|‾|_._|‾|_._  // 146 73 36
//  128   64    32    16    8     4     2     1
// |‾.‾|_|‾.‾|_|‾.‾|_|‾.‾|_|‾.‾|_|‾.‾|_|‾.‾|_|‾.‾|_  // 219 109 182
void ws_33() {

  uint8_t p = 0, w;
  ///ws_3();
  ///ws_green_adc[0] = adc_pa5 / 4;
  for (int ws = 0; ws < sizeof(ws3) / 3; ws++) {
    for (int c = 0; c < 3; c++) {
      w = ws3[ws][c];
      ws33[p] = 0b10010010;
      if (w & 128) ws33[p] += 0b01000000;
      if (w & 64) ws33[p] += 0b00001000;
      if (w & 32) ws33[p] += 1;
      p++;
      ws33[p] = 0b01001001;
      if (w & 16) ws33[p] += 0b00100000;
      if (w & 8) ws33[p] += 0b00000100;
      p++;
      ws33[p] = 0b00100100;
      if (w & 4) ws33[p] += 0b10000000;
      if (w & 2) ws33[p] += 0b00010000;
      if (w & 1) ws33[p] += 0b00000010;
      p++;
    }
  }
}


void send_ws() {
  ws_33();
  spi_dma_transmit();
}


void spi_dma_init() {
  // Enable Port A clock // Enable SPI Clock
  RCC->APB2ENR |= RCC_APB2ENR_IOPAEN | RCC_APB2ENR_SPI1EN;
  // Mode: Output, Speed: 10MHz // Alternate function push-pull
  GPIOA->CRL &= ~(GPIO_CRL_CNF7 | GPIO_CRL_MODE7);  // MOSI = PA7
  GPIOA->CRL |= (GPIO_CRL_CNF7_1 | GPIO_CRL_MODE7_0);  // 10.01

  // 100: BaudRate = fPCLK/32 // + CPOL + CPHA
  SPI1->CR1 |= SPI_CR1_BR_2 | SPI_CR1_CPOL | SPI_CR1_CPHA;  ///
  // Set TXDMA bit in CR2 → enable DMA transmit
  SPI1->CR2 |= SPI_CR2_TXDMAEN | SPI_CR2_SSOE; /// pourquoi le SSOE ??
  // SPI Mode: Master // SPI Enable
  SPI1->CR1 |= SPI_CR1_MSTR | SPI_CR1_SPE;

  // DMA Configuration  // Enable DMA Clock
  RCC->AHBENR |= RCC_AHBENR_DMA1EN;
  // Priority Level = 0b10 → High
  DMA1_Channel3->CCR |= DMA_CCR_PL_1;
  // Memory Increment mode + Direction Read from Memory
  DMA1_Channel3->CCR |= DMA_CCR_MINC | DMA_CCR_DIR;  // no need for interrupt
  // Set Memory Address → Peripheral Address
  DMA1_Channel3->CMAR = (uint32_t)ws33;
  DMA1_Channel3->CPAR = (uint32_t)&SPI1->DR;
}

void spi_dma_transmit() {
  // stop DMA to refill CNDTR
  DMA1_Channel3->CCR &= ~DMA_CCR_EN;
  DMA1_Channel3->CNDTR = sizeof(ws33);
  // launch DMA
  DMA1_Channel3->CCR |= DMA_CCR_EN;
  ///Serial.print(sizeof(ws33));
}