// Class for controlling each extruder
//
// Adrian Bowyer 14 May 2009

#ifndef EXTRUDER_H
#define EXTRUDER_H

#define EXTRUDER_COUNT 1

void manage_all_extruders();

void new_extruder(byte e);

/**********************************************************************************************

* Sanguino/RepRap Motherboard v 1.0

*/

#if MOTHERBOARD < 2

#define EXTRUDER_FORWARD true
#define EXTRUDER_REVERSE false

class extruder
{
private:

//these our the default values for the extruder.
    byte e_speed;
    int target_celsius;
    int max_celsius;
    byte heater_low;
    byte heater_high;
    byte heater_current;
    int extrude_step_count;

// These are used for temperature control    
    byte count ;
    int oldT, newT;
    
//this is for doing encoder based extruder control
//    int rpm;
//    long e_delay;
//    int error;
//    int last_extruder_error;
//    int error_delta;
    bool e_direction;
    bool valve_open;

// The pins we control
    byte motor_dir_pin, motor_speed_pin, heater_pin, fan_pin, temp_pin, valve_dir_pin, valve_en_pin, step_en_pin;
    
     byte wait_till_hot();
     //byte wait_till_cool();
     void temperature_error(); 
     int sample_temperature();
     
public:

   extruder(byte md_pin, byte ms_pin, byte h_pin, byte f_pin, byte t_pin, byte vd_pin, byte ve_pin, byte se_pin);
   void wait_for_temperature();
   void valve_set(bool open, int dTime);
   void set_direction(bool direction);
//   void set_speed(float es);
   void set_cooler(byte e_speed);
   void set_temperature(int temp);
   int get_temperature();
   void manage();
// Interrupt setup and handling functions for stepper-driven extruders
   
//   void interrupt();
   void step();

   void enableStep();
   void disableStep();
   
};

inline void extruder::enableStep()
{
  if(step_en_pin < 0)
    return;
  digitalWrite(step_en_pin, ENABLE_ON); 
}

inline void extruder::disableStep()
{
  if(step_en_pin < 0)
    return;
  digitalWrite(step_en_pin, !ENABLE_ON);
}

inline void extruder::step()
{
   digitalWrite(motor_speed_pin, HIGH);
   delayMicrosecondsInterruptible(5); 
   digitalWrite(motor_speed_pin, LOW);
}

inline void extruder::temperature_error()
{
      Serial.print("E: ");
      Serial.println(get_temperature());  
}

//warmup if we're too cold; cool down if we're too hot
inline void extruder::wait_for_temperature()
{
/*
  if(wait_till_cool())
   {
      temperature_error();
      return;
   }
*/
   if(wait_till_hot())
     temperature_error();
}

inline void extruder::set_direction(bool dir)
{
	e_direction = dir;
	digitalWrite(motor_dir_pin, e_direction);
}

inline void extruder::set_cooler(byte sp)
{
  if(step_en_pin >= 0) // Step enable conflicts with the fan
    return;
  analogWrite(fan_pin, sp);
}

/**********************************************************************************************

* RepRap Motherboard v 1.x (x >= 1)
* Extruder is now on RS485

*/

#else

#define WAIT_T 'W'        // wait_for_temperature();
#define VALVE 'V'          // valve_set(bool open, int dTime);
#define DIRECTION 'D'   // set_direction(bool direction);
#define COOL 'C'           // set_cooler(byte e_speed);
#define SET_T 'T'           // set_temperature(int temp);
#define GET_T 't'           // get_temperature();
#define STEP 'S'             // step();
#define ENABLE 'E'       // enableStep();
#define DISABLE 'e'       // disableStep();
#define PING 'P'            // Just acknowledge

class extruder
{
private:

  char my_name;
  char commandBuffer[RS485_BUF_LEN];
  char* reply;
  
public:
   extruder(char name);
   void wait_for_temperature();
   void valve_set(bool open, int dTime);
   void set_direction(bool direction);
   void set_cooler(byte e_speed);
   void set_temperature(int temp);
   int get_temperature();
   void manage();
   void step();
   void enableStep();
   void disableStep();
   
   void buildCommand(char c);   
   void buildCommand(char c, char v);
   void buildNumberCommand(char c, int v);  
};

inline extruder::extruder(char name)
{
  my_name = name;
}

inline void extruder::buildCommand(char c)
{
  commandBuffer[0] = c;
  commandBuffer[1] = 0;  
}

inline void extruder::buildCommand(char c, char v)
{
  commandBuffer[0] = c;
  commandBuffer[1] = v;
  commandBuffer[2] = 0;  
}

inline void extruder::buildNumberCommand(char c, int v)
{
  commandBuffer[0] = c;
  itoa(v, &commandBuffer[1], 10);
}


inline  void extruder::wait_for_temperature()
{
  
}

inline  void extruder::valve_set(bool open, int dTime)
{
   if(open)
     buildCommand(VALVE, '1');
   else
     buildCommand(VALVE, '0');
   talker.sendPacketAndGetReply(my_name, commandBuffer);
   
   delay(dTime);
}

inline void extruder::set_direction(bool direction)
{
   if(direction)
     buildCommand(DIRECTION, '1');
   else
     buildCommand(DIRECTION, '0');
   talker.fireAndForget(my_name, commandBuffer);
}

inline  void extruder::set_cooler(byte e_speed)
{   
   buildNumberCommand(COOL, e_speed);
   talker.sendPacketAndGetReply(my_name, commandBuffer);
}

inline  void extruder::set_temperature(int temp)
{
   buildNumberCommand(SET_T, temp);
   talker.sendPacketAndGetReply(my_name, commandBuffer);  
}

inline  int extruder::get_temperature()
{
   //buildCommand(GET_T);
   //char* reply = talker.sendPacketAndGetReply(my_name, commandBuffer);
   //return(atoi(reply));
   return 10;   
}

inline  void extruder::manage()
{

}

inline  void extruder::step()
{
   buildCommand(STEP);
   talker.fireAndForget(my_name, commandBuffer);
}

inline  void extruder::enableStep()
{
  // Not needed
  /*
   buildCommand(ENABLE);
   talker.sendPacketAndGetReply(my_name, commandBuffer);
  */
}

inline  void extruder::disableStep()
{
  buildCommand(DISABLE);
  talker.sendPacketAndGetReply(my_name, commandBuffer);
}

#endif

extern extruder* ex[];
extern byte extruder_in_use;

#endif