apps/astreamer/streamer.cpp

Go to the documentation of this file.

#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>

#include "streamer.h"
#include "helpers.h"

#include <iostream>

using namespace std;

#undef RTSP_DEBUG
//#define RTSP_DEBUG

#ifdef RTSP_DEBUG
        #define D(a) a
#else
        #define D(a)
#endif

Streamer *Streamer::_streamer = NULL;

Streamer::Streamer(const map<string, string> &_args) {
        if(_streamer != NULL)
                throw("can't make an another instance of streamer");
        else
                _streamer = this;
        session = new Session();
        args = _args;
        session->process_audio = false;
        if(opt_present("a")) {
                session->process_audio = true;
                int channels = 0;
                int sample_rate = 0;
                string _sample_rate, _channels;
                String::split(args["a"], '/', _sample_rate, _channels);
                if(_sample_rate != "")
                        sample_rate = atol(_sample_rate.c_str());
                if(_channels != "")
                        channels = atol(_channels.c_str());
                audio = new Audio(true, sample_rate, channels);
        } else {
                audio = new Audio(false);

        }

        video = new Video();
        float fps = 0;
        if(opt_present("f")) {
                fps = atof(args["f"].c_str());
                if(fps < 0.1)
                        fps = 0;
        }
D(cout << "use fps: " << fps << endl;)
        video->fps(fps);
        rtsp_server=NULL;
}

Streamer::~Streamer(void) {
        delete video;
        delete audio;
}

int Streamer::f_handler(void *ptr, RTSP_Server *rtsp_server, RTSP_Server::event event) {
        Streamer *__this = (Streamer *)ptr;
        return __this->handler(rtsp_server, event);
}

int Streamer::handler(RTSP_Server *rtsp_server, RTSP_Server::event event) {
static bool _play = false;
        if(session->rtp_out.multicast)
//              return;
                return 0;
D(      cerr << "event: running= " << running << " ";)
        switch(event) {
        case RTSP_Server::DESCRIBE:  

            session->video.width = video->width();
            session->video.height = video->height();
            session->video.type = video->ptype();
            session->video.fps = video->fps();
                   break;
        case RTSP_Server::PLAY:
                if(running) {
D(cerr << ".......................... already play....." << endl;)
                        break;
                }
D(              cerr << "==PLAY==";)
//              video->Start(session->rtp_out.ip, 20020);
//              audio->Start(session->rtp_out.ip, 20020 + 2);
/*
                video->Start(session->rtp_out.ip, atol(session->rtp_out.port.c_str()));
                audio->Start(session->rtp_out.ip, atol(session->rtp_out.port.c_str()) + 2);
*/
//              video->Start(session->rtp_out.ip, atol(session->rtp_out.port.c_str()) - 2);
//              audio->Start(session->rtp_out.ip, atol(session->rtp_out.port.c_str()) + 0);
                video->Start(session->rtp_out.ip, session->rtp_out.port_video);
                audio->Start(session->rtp_out.ip, session->rtp_out.port_audio);

                _play = true;
                running=true;
                break;
        case RTSP_Server::PAUSE:
D(              cerr << "PAUSE";)
                video->Stop();
                audio->Stop();
                _play = false;
                running=false;
                break;
        case RTSP_Server::TEARDOWN:
D(              cerr << "TEARDOWN";)
                if (!running) {
D(              cerr << " was not running";)
                   break;
                }
                video->Stop();
                audio->Stop();
                _play = false;
                running=false;
                break;
        case RTSP_Server::IS_DAEMON_ENABLED:
D(              cerr << "IS_DAEMON_ENABLED video->isDaemonEnabled(-1)=" << video->isDaemonEnabled(-1) << endl;)
                return video->isDaemonEnabled(-1);
                break;
        default:
D(              cerr << "unknown == " << event;)
                break;
        }
D(      cerr << endl;)
  return 0;
}

void Streamer::Main(void) {
        string def_mcast = "232.1.1.1";
        int def_port = 20020;
        string def_ttl = "2";

        in_addr_t a_min = inet_addr("224.0.0.0");
        in_addr_t a_max = inet_addr("239.255.255.255");
        if(a_min > a_max) {
                in_addr_t a = a_min;
                a_min = a_max;
                a_max = a;
        }
        int port_min = 256;
        while (true) {
          if (! video->waitDaemonEnabled(-1)) continue; 

        // make session parameters from session description...
          session->rtp_out.ip = def_mcast;
//        session->rtp_out.port = def_port;
          session->rtp_out.port_video = def_port;
          session->rtp_out.port_audio = def_port + 2;
          session->rtp_out.ttl = def_ttl;
          if(opt_present("m")) {
                session->rtp_out.multicast = true;
                string m_addr, m_port;
                String::split(args["m"], ':', m_addr, m_port);
                if(m_addr != "") {
//                      string mcast = args["a"];
                        string mcast = m_addr;
                        in_addr_t a = inet_addr(mcast.c_str());
                        if(a >= a_min && a <= a_max)
                                session->rtp_out.ip = mcast;
                }
                if(m_port != "") {
                        int p = atoi(m_port.c_str());
                        if(p >= port_min) {
                                session->rtp_out.port_video = p;
                                session->rtp_out.port_audio = p + 2;
//                              session->rtp_out.port = m_port;
                        }
                }
          } else
                session->rtp_out.multicast = false;

/*
        if(opt_present("a")) {
                string mcast = args["a"];
                in_addr_t a = inet_addr(mcast.c_str());
                if(a >= a_min && a <= a_max)
                        session->rtp_out.ip = mcast;
        }
        if(opt_present("p")) {
                int p = atoi(args["p"].c_str());
                if(p >= port_min)
                        session->rtp_out.port = args["p"];
        }
*/
          if(opt_present("t")) {
                int t = atoi(args["t"].c_str());
                if(t < 1)
                        session->rtp_out.ttl = "1";
                else
                        session->rtp_out.ttl = args["t"];
          }

          session->video.width = video->width();
          session->video.height = video->height();
          session->video.type = video->ptype();
          session->video.fps = video->fps();
D(cout << "audio: " << audio->present() << endl;)
          if(audio->present()) {
                session->audio.sample_rate = audio->sample_rate();
                session->audio.channels = audio->channels();
                session->audio.type = audio->ptype();
          } else
                session->audio.type = -1;
          if(session->rtp_out.multicast) {
                video->Start(session->rtp_out.ip, session->rtp_out.port_video);
                audio->Start(session->rtp_out.ip, session->rtp_out.port_audio);
//              video->Start(session->rtp_out.ip, atol(session->rtp_out.port.c_str()));
//              audio->Start(session->rtp_out.ip, atol(session->rtp_out.port.c_str()) + 2);
//              audio->Start(session->rtp_out.ip, atol(session->rtp_out.port.c_str()) + 2);
          }
        // start RTSP server
D(cout << "start server" << endl;)
          rtsp_server = new RTSP_Server(Streamer::f_handler, (void *)this, session);
          rtsp_server->main();
        }
}

---