#include <gst/gst.h>
#include <iostream>
#include <string>
#include <gst/app/gstappsink.h>
/************* 
global values 
*****    ***********/ 

GstElement *pipeline; 
GMainLoop *loop; 
GstBus *bus; 
GstMessage* msg; 
GstState state; 
GstClock* PipeClock;
bool gRetrievedConnCheck;
int bufferCount;
int preBufferCount;
#define GST_PLAYER_UDP "udpsrc name=source port=5555 caps=\"application/x-rtp, media=(string)video,clock-rate=(int)90000, encoding-name=(string)H264, payload=(int)96\" ! rtph264depay ! tee name=tee ! queue ! decodebin ! videoconvert ! autovideosink tee. ! queue ! appsink emit-signals=true name=appsink" 

static gboolean bus_cb (GstBus *bus, GstMessage *msg, gpointer *data); 
    static gboolean timeout_cb (gpointer user_data);
	static gboolean TestingConnection(gpointer user_data);
	static void ImmediateLoopQuit();
	static GstFlowReturn appsink_signal (GstElement* object,gpointer user_data);
	void VideoHasFinished();

int main() 
{ 
	GError* error;
	bufferCount = 0;
	preBufferCount = 0;
	gst_debug_set_threshold_from_string("*:3",TRUE);
        /* Initialize GStreamer */ 
        gst_init (NULL,NULL); 

        /* Build the pipeline */ 
        pipeline = gst_parse_launch (GST_PLAYER_UDP, NULL);
        
        loop = g_main_loop_new (NULL, FALSE); 

        /* Start playing */ 
        gst_element_set_state (pipeline, GST_STATE_PLAYING); 
		PipeClock = gst_pipeline_get_clock(GST_PIPELINE (pipeline));

        /* Wait until error or EOS */ 
         bus = gst_pipeline_get_bus (GST_PIPELINE (pipeline)); 
         if(bus == NULL) 
         { 
                std::cout << "\nNo Bus\n"; 
                return 0; 
         } 
        gst_bus_add_signal_watch(bus); 
        g_signal_connect (bus, "message", G_CALLBACK (bus_cb), pipeline); 
        
		GstElement* appsink = gst_bin_get_by_name(GST_BIN(pipeline),"appsink");
		g_signal_connect(appsink, "new-sample", G_CALLBACK(appsink_signal),loop);

        g_timeout_add_seconds (1, timeout_cb, loop); 

        g_main_loop_run(loop); 
        
        /* Free resources */ 
        gst_object_unref (bus); 
        gst_element_set_state (pipeline, GST_STATE_NULL); 
        gst_object_unref (pipeline); 
		VideoHasFinished();
		return 0;
} 

void VideoHasFinished()
{
	std::cout << "\n-----------------------------------------------\n";
	std::cout << "\n-----------------------------------------------\n";
	std::cout << "\n---------Video Has Finished--------------------\n";
	std::cout << "\n-----------------------------------------------\n";
	std::cout << "\n-----------------------------------------------\n";
	std::cout << "\n-----------------------------------------------\n";
	

}

GstFlowReturn appsink_signal (GstElement* object,gpointer user_data)
{

	GstFlowReturn flow;
	std::string TheFile = "";
	GstSample* _app_sample;
	GstMapInfo _app_map;
	GstBuffer* _app_buffer;
	flow = GST_FLOW_OK;
	try{
	g_signal_emit_by_name(object,"pull-sample",&_app_sample);
	
	if(_app_sample != NULL)
	{		
		_app_buffer = gst_sample_get_buffer(_app_sample);
		
		if(_app_buffer != NULL)
		{
			++bufferCount;
		}

		if(_app_sample != NULL)
		{
			gst_sample_unref(_app_sample);
			_app_sample = NULL;
		}

	}
	else
	{
		std::cout << "\nNo Sample\n";
	}
	if(gst_app_sink_is_eos ((GstAppSink*)object))
	{
		std::cout << "\nEos Received\n";
		return GST_FLOW_EOS;
	}
	}
	catch(std::exception ex)
	{
		std::string error = ex.what();
	}
	return flow;    
}


/********************** 
    bus method 
******************/ 
gboolean bus_cb (GstBus *bus, GstMessage *msg, gpointer *data) { 
	gboolean ret = FALSE;
  switch (GST_MESSAGE_TYPE (msg)) 
  { 
        case GST_MESSAGE_ERROR: 
			{ 
      GError *err; 
      gchar *debug; 

      gst_message_parse_error (msg, &err, &debug); 
      g_print ("Error: %s\n", err->message); 
      g_error_free (err); 
      g_free (debug); 
      gst_element_set_state (pipeline, GST_STATE_READY); 
      g_main_loop_quit (loop); 
		} 
		return FALSE;
    case GST_MESSAGE_EOS: 
      /* end-of-stream */ 
      gst_element_set_state (pipeline, GST_STATE_READY); 
      g_main_loop_quit (loop); 
	  return FALSE;
    case GST_MESSAGE_STATE_CHANGED: 
        { 
        GstElement* source = gst_bin_get_by_name(GST_BIN(pipeline),"source"); 

                if (GST_MESSAGE_SRC (msg) == GST_OBJECT(source)) { 
            GstState old_state, new_state, pending_state; 
            gst_message_parse_state_changed (msg, &old_state, &new_state, &pending_state);
                state = new_state; 
                gst_object_unref(source); 
				}
        } 
        break; 
    default: 
      
      break; 
    
  }
      return TRUE;
} 

gboolean timeout_cb (gpointer user_data) 
{ 
        static int time = 0; 
        static int timeout = 0; 
		static int SameBufferCount = 0;
        int MaxTimeout = 10;	
		if(PipeClock)
		{
                switch (state) 
                { 
                case GST_STATE_READY: 
                { 
                   std::cout << "\nSTATE Ready\n"; 
                 } 
                case GST_STATE_PLAYING: 
                        { 
                               std::cout << "\nSTATE Playing\n";
							   if(SameBufferCount == 4)
								   ImmediateLoopQuit();
							   else if(bufferCount == preBufferCount)
								   ++SameBufferCount;
							   else
							   {
								   preBufferCount = bufferCount;
								   SameBufferCount = 0;
							   }

                        } 
                        break; 
                case GST_STATE_PAUSED: 
                        { 
                               std::cout << "\nSTATE Paused\n";
                                break; 
                        } 
                case GST_STATE_NULL: 
                               std::cout << "\nSTATE Null\n";
                        return FALSE; 
                default: 
                        break; 
                } 
  return TRUE; 
		}
		else
		{
			std::cout << "No Clock";
		}
}

void ImmediateLoopQuit()
{
	g_main_loop_quit(loop);
}