/*
 Function Generator and Frequency Counter with SCPI parser

  setup: 

 1. programm your Arduino
 2. addapt the serial port in your python script to the serial port used on your PC
 3. run the python script 

 PC: python with tkinter and pyserial
 MC: any Arduino

 YY.MM.DD
 22.05.06 first version of function generator by mchris
 22.05.08 V0.2, SCPI parser lib included by mchris
 22.05.14 V0.2, frequency counter added by mchris

 Use SCPI library from
 https://github.com/Vrekrer/Vrekrer_scpi_parser

*/


#include "Arduino.h"
#include "Vrekrer_scpi_parser.h"

SCPI_Parser my_instrument;

#define COUNTERINPUT_PIN 6
#define FREQENCY_PIN 8


void setup()
{
  //We change the `hash_magic_number` variable before registering the commands
  my_instrument.hash_magic_number = 16; //16 will generate hash crashes
  //The default value is 37 and good values are prime numbers (up to 113)

  my_instrument.RegisterCommand(F("*IDN?"), &Identify);   //*IDN?
  
  my_instrument.SetCommandTreeBase(F("GENErator:"));
    my_instrument.RegisterCommand(F(":FREQency"), &SetGeneratorFrequency);  //GENE:FREQ 100
    my_instrument.RegisterCommand(F(":FREQency?"), &GetGeneratorFrequency); //GENE:FREQ?
    
  my_instrument.SetCommandTreeBase(F("COUNter:"));
    my_instrument.RegisterCommand(F(":FREQency?"), &GetCounterFrequency); //GENE:FREQ?
    
  Serial.begin(115200);
  while (!Serial) {;}
  
  //my_instrument.PrintDebugInfo();

}

void loop()
{
  my_instrument.ProcessInput(Serial, "\n");
}

void Identify(SCPI_C commands, SCPI_P parameters, Stream& interface) 
{
  interface.println(F("MCHRIS, Arduino Frequency Counter and Generator V0.3"));
}

uint32_t Frequency=0;

void SetGeneratorFrequency(SCPI_C commands, SCPI_P parameters, Stream& interface) 
{
  String str=parameters[0];
  Frequency=str.toInt();
  Serial.print("f set: ");  Serial.println(Frequency);
  tone(FREQENCY_PIN,Frequency);
}

void GetGeneratorFrequency(SCPI_C commands, SCPI_P parameters, Stream& interface) 
{
  Serial.print("f read: ");  Serial.println(Frequency);
}

#define MEASUREMENT_TIME_US 1000000L


uint32_t Counter=0;
void GetCounterFrequency(SCPI_C commands, SCPI_P parameters, Stream& interface) 
{
  float frequency;
  uint32_t startTime=micros();

  bool state=0;

  frequency=Counter;


  // measure next frequency
  Counter=0;
  
  while((micros()-startTime)<MEASUREMENT_TIME_US)
  {
    if(state==0)
    {
      if(digitalRead(COUNTERINPUT_PIN)==HIGH) state=1;
    }
    if(state==1)
    {
      if(digitalRead(COUNTERINPUT_PIN)==LOW) 
      {
        Counter++;
        state=0;
      }
    }
  }
  Serial.println(Counter);
  Serial.flush();

}


/*


Hauptfenster.geometry("1024x640")
Hauptfenster.title('frequency controller')

PrintTxt_portName=ttk.Label(Hauptfenster, text="benutzer serieller Anschluss:  "+McSeriallPort)   
PrintTxt_portName.place(x=100,y=10)

PrintTxt_Verwendung=ttk.Label(Hauptfenster, text="frequency output: pin 8 , frequency input: pin6"   )
PrintTxt_Verwendung.place(x=100,y=40)

PrintTxt_comRead=ttk.Label(Hauptfenster, text="empfangener text:  ") 
PrintTxt_comRead.place(x=100, y=300)

def sendValue(txt):
    arduino.read_all()
    txtLf=txt+" \r\n"
    print("send: "+txtLf),
    arduino.write(bytes(txtLf, 'utf-8'))
    data = arduino.readline()
    print("receive: "),
    print(data)
    PrintTxt_comRead.configure(text=data) 
    
def getFrequency():
    arduino.read_all()
    txt='COUN:FREQ?'
    txtLf=txt+" \r\n"
    print("send: "+txtLf),
    arduino.write(bytes(txtLf, 'utf-8'))
    time.sleep(1.1)
    data = arduino.readline()
    print("receive: "),
    print(data)
    PrintTxt_comRead.configure(text=data) 
    
    
#===================== Button ======================================

button1 = ttk.Button(Hauptfenster, text="generate 100Hz", command=lambda: sendValue('GENE:FREQ 100'))
button1.place(x=100, y=100)

button2 = ttk.Button(Hauptfenster, text="generate 1000Hz", command=lambda: sendValue('GENE:FREQ 1000'))
button2.place(x=100, y=150)

#button_GetGeneratorFrequency = ttk.Button(Hauptfenster, text="get frequency", command=lambda: sendValue('GENE:FREQ?'))
#button_GetGeneratorFrequency.place(x=100, y=200)

button_GetCounterFrequency = ttk.Button(Hauptfenster, text="measure frequency", command=lambda: getFrequency())
button_GetCounterFrequency.place(x=400, y=150)

button_getIdentifier = ttk.Button(Hauptfenster, text="SCPI device info", command=lambda: sendValue('*IDN?'))
button_getIdentifier.place(x=100, y=400)


Hauptfenster.mainloop()


*/