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);
}

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

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

void Streamer::handler(RTSP_Server *rtsp_server, RTSP_Server::event event) {
static bool _play = false;
        if(session->rtp_out.multicast)
                return;
D(      cerr << "event: ";)
        switch(event) {
        case RTSP_Server::PLAY:
                if(_play) {
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;
                break;
        case RTSP_Server::PAUSE:
D(              cerr << "PAUSE";)
                video->Stop();
                audio->Stop();
                _play = false;
                break;
        case RTSP_Server::TEARDOWN:
D(              cerr << "TEARDOWN";)
                video->Stop();
                audio->Stop();
                _play = false;
                break;
        default:
D(              cerr << "unknown == " << event;)
                break;
        }
D(      cerr << endl;)
}

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;

        // 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();
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
cout << "start server" << endl;
        rtsp_server = new RTSP_Server(Streamer::f_handler, (void *)this, session);
        rtsp_server->main();
}

---