Svenska ElektronikForumet

Kod: Markera allt

#include "stm32f10x_rcc.h"
#include "stm32f10x_spi.h"
#include "stm32f10x_gpio.h"
#include "misc.h"
#include "nrf_spi.h"
#include "cBuffer.h"
#include "led.h"


static CBUFFER nrfTxBuffer, nrfRxBuffer;
static unsigned char nrfRawTxBuffer[64], nrfRawTxBufferElementsLeft=0;
static unsigned char nrfRawRxBuffer[64], nrfRawRxBufferWritePointer=0,nrfRawRxBufferElementsToRead=0;

static unsigned char nrfBusy=0;
static unsigned char nrfIsInTxMode=0;
static unsigned char nrfStatusRegister=0;

void nrfInit()
{

	//Init buffers
	CBufferNew(&nrfRxBuffer, sizeof(RADIO_PKG_TYPE), NRF_BUFFER_LENGTH);
	CBufferNew(&nrfTxBuffer, sizeof(RADIO_PKG_TYPE), NRF_BUFFER_LENGTH);


	//Start clocks
	RCC_APB2PeriphClockCmd(RCC_APB2Periph_SPI1,ENABLE);
	RCC_APB2PeriphClockCmd(RCC_APB2Periph_GPIOA,ENABLE);
	RCC_APB2PeriphClockCmd(RCC_APB2Periph_GPIOB,ENABLE);

	//Init I/O
	GPIO_InitTypeDef GPIO_InitStruct;

	GPIO_InitStruct.GPIO_Mode=GPIO_Mode_AF_PP;
	GPIO_InitStruct.GPIO_Pin=GPIO_Pin_5 | GPIO_Pin_6 | GPIO_Pin_7;
	GPIO_InitStruct.GPIO_Speed=GPIO_Speed_50MHz;
	GPIO_Init(GPIOA,&GPIO_InitStruct);

	GPIO_InitStruct.GPIO_Mode=GPIO_Mode_IPU;
	GPIO_InitStruct.GPIO_Pin=GPIO_Pin_4;
	GPIO_InitStruct.GPIO_Speed=GPIO_Speed_50MHz;
	GPIO_Init(GPIOA,&GPIO_InitStruct);

	GPIO_InitStruct.GPIO_Mode=GPIO_Mode_Out_PP;
	GPIO_InitStruct.GPIO_Pin=GPIO_Pin_0 | GPIO_Pin_1;
	GPIO_InitStruct.GPIO_Speed=GPIO_Speed_50MHz;
	GPIO_Init(GPIOB,&GPIO_InitStruct);

	nrfSetCSN();
	nrfSetCE();

	//Init SPI
	SPI_InitTypeDef SPI_InitStruct;
	SPI_StructInit(&SPI_InitStruct);
	SPI_InitStruct.SPI_NSS=SPI_NSS_Soft;
	SPI_InitStruct.SPI_Mode=SPI_Mode_Master;
	SPI_InitStruct.SPI_BaudRatePrescaler=SPI_BaudRatePrescaler_256;
	SPI_Init(SPI1,&SPI_InitStruct);
	SPI_I2S_ITConfig(SPI1, SPI_I2S_IT_RXNE,ENABLE);
	SPI_Cmd(SPI1,ENABLE);

	//Init interrupts
    NVIC_InitTypeDef NVIC_InitStruct;
    NVIC_InitStruct.NVIC_IRQChannel=SPI1_IRQn;
    NVIC_InitStruct.NVIC_IRQChannelCmd=ENABLE;
    NVIC_InitStruct.NVIC_IRQChannelPreemptionPriority=7;//Semi low priority, this is not timing critical
    NVIC_InitStruct.NVIC_IRQChannelSubPriority=7;
    NVIC_Init(&NVIC_InitStruct);


    nrfSendCommand(NRF_WRITE_MASK | 0, &((unsigned char){0b00111111}), 1);//Enable
    while(nrfBusy);
    nrfSendCommand(NRF_WRITE_MASK | 1, &((unsigned char){0}), 1);//No autoack
    while(nrfBusy);
    nrfSendCommand(NRF_WRITE_MASK | 2, &((unsigned char){1}), 1);//Datapipe 0
    while(nrfBusy);
    nrfSendCommand(NRF_WRITE_MASK | 3, &((unsigned char){3}), 1);//3Byte addr
    while(nrfBusy);
    nrfSendCommand(NRF_WRITE_MASK | 4, &((unsigned char){0}), 1);//No Retransmit
    while(nrfBusy);
    nrfSendCommand(NRF_WRITE_MASK | 5, &((unsigned char){0x7F}), 1);//Highest frequency
    while(nrfBusy);
    nrfSendCommand(NRF_WRITE_MASK | 6, &((unsigned char){0b00100110}), 1);//Max power minimum speed
    while(nrfBusy);
    nrfSendCommand(NRF_WRITE_MASK | 17, &((unsigned char){sizeof(RADIO_PKG_TYPE)}), 1);//5Bytes in each package
    while(nrfBusy);
}

unsigned char nrfPendingRxPackage()
{
	if(GPIO_ReadInputDataBit(GPIOA,GPIO_Pin_4))
		return 0;
	return 1;
}

void nrfSendCommand(unsigned char addr, unsigned char *data, unsigned char length)
{
	if(nrfRawTxBufferElementsLeft)
		return;
	nrfRawTxBufferElementsLeft=length+1;
	nrfRawRxBufferElementsToRead=length+1;
	nrfRawRxBufferWritePointer=0;
	nrfRawTxBuffer[length]=addr;
	while(length--)
		nrfRawTxBuffer[length]=*data++;
	SPI_I2S_ITConfig(SPI1,SPI_I2S_IT_TXE,ENABLE);
}

void nrfSendPackage(unsigned char addr, unsigned char valueAddr, unsigned short value)
{
	RADIO_PKG_TYPE package;
	package.readRequest=0;
	package.unitAddr=NRF_MY_ADDR;
	package.value=value;
	package.valueAddr=valueAddr;
	CBufferPush(&nrfTxBuffer,&package);
}

RADIO_PKG_TYPE nrfReadPackage()
{
	RADIO_PKG_TYPE package;
	package.readRequest=0;
	package.unitAddr=0;
	package.value=0;
	package.valueAddr=0;
	if(!CBufferUsed(&nrfRxBuffer))
		return package;
	CBufferPull(&nrfRxBuffer,&package);
	return package;
}

void nrfSetCE()
{
	GPIO_SetBits(GPIOB,GPIO_Pin_1);
}

void nrfResetCE()
{
	GPIO_ResetBits(GPIOB,GPIO_Pin_1);
}

void nrfSetCSN()
{
	GPIO_SetBits(GPIOB,GPIO_Pin_0);
}

void nrfResetCSN()
{
	GPIO_ResetBits(GPIOB,GPIO_Pin_0);
}

void nrfParseRxBuffer()
{
	nrfStatusRegister=nrfRawRxBuffer[0];
}

void SPI1_IRQHandler()
{
	if(SPI_I2S_GetITStatus(SPI1,SPI_I2S_IT_RXNE))
	{
		unsigned char data;
		data=SPI_I2S_ReceiveData(SPI1);

		if(nrfRawRxBufferElementsToRead)
		{
			nrfRawRxBuffer[nrfRawRxBufferWritePointer++]=data;
			nrfRawRxBufferElementsToRead--;
		}

		if(!nrfRawRxBufferElementsToRead)//All done here
		{
			nrfParseRxBuffer();
			nrfSetCSN();
			nrfBusy=0;
		}
		//SPI_I2S_ClearITPendingBit(SPI1,SPI_I2S_IT_RXNE);
	}
	if(SPI_I2S_GetITStatus(SPI1,SPI_I2S_IT_TXE))
	{
		if(nrfRawTxBufferElementsLeft)
		{
			nrfBusy=1;
			nrfResetCSN();
			while(SPI_I2S_GetITStatus(SPI1,SPI_I2S_IT_TXE) && nrfRawTxBufferElementsLeft)
				SPI_I2S_SendData(SPI1,nrfRawTxBuffer[--nrfRawTxBufferElementsLeft]);

			if(!nrfRawTxBufferElementsLeft)
				SPI_I2S_ITConfig(SPI1,SPI_I2S_IT_TXE,DISABLE);
		}
		//SPI_I2S_ClearITPendingBit(SPI1,SPI_I2S_IT_TXE);
	}

}

//Here we want to move packages from tx buffer to raw tx buffer
void nrfProcess1000Hz()
{
	static unsigned char state=0;
	if(!nrfBusy && !nrfRawTxBufferElementsLeft && CBufferUsed(&nrfTxBuffer))//If we can send data and have data to send
	{
		if(state==2)
		{
			RADIO_PKG_TYPE package;
			CBufferPull(&nrfTxBuffer,&package);
			nrfSendCommand(0xA0, (unsigned char*)(&package), sizeof(RADIO_PKG_TYPE));//Send data
			state=0;
			nrfSetCE();
		}
		else if(state==1)
		{
			nrfSendCommand(0xE1, 0, 0);//Flush Buffer
			nrfIsInTxMode=1;
			state=2;
		}
		else if(state==0)
		{
			nrfResetCE();
			nrfSendCommand(NRF_WRITE_MASK | 0, &((unsigned char){0b00111110}), 1);//Enter TX mode
			nrfIsInTxMode=1;
			state=1;
		}
	}
	else if(nrfIsInTxMode)
	{
		nrfSendCommand(NRF_WRITE_MASK | 0, &((unsigned char){0b00111111}), 1);//Enter RX mode
		nrfIsInTxMode=0;
		state=0;
	}

}

Kod: Markera allt

#ifndef __NRF_SPI_H_
#define __NRF_SPI_H_

/*The RADIO_CMD_TYPE works as following
 * unitAddr is a unique addr for each unit
 * readRequest is 1 when the package is a read request, otherwise it is 0
 * valueAddr is the addr of the value that is to be read or is present in the package
 * value is the value, this value is irrelevant for readRequest=1
 */


typedef struct {
	unsigned char unitAddr;
	unsigned char readRequest;
	unsigned char valueAddr;
	unsigned short value;
} RADIO_PKG_TYPE;

#define NRF_BUFFER_LENGTH	32
#define NRF_MY_ADDR	15


#define NRF_WRITE_MASK	0b00100000


void nrfSendCommand(unsigned char , unsigned char *, unsigned char );
void nrfInit();
void nrfSetCE();
void nrfResetCE();
void nrfSetCSN();
void nrfResetCSN();
void nrfProcess1000Hz();
unsigned char nrfPendingRxPackage();
void nrfSendPackage(unsigned char, unsigned char, unsigned short);
#endif