home *** CD-ROM | disk | FTP | other *** search
/ Chip 2005 June / ccd0605.iso / LINUX / gopchop-1.1.7.tar.tar / gopchop-1.1.7.tar / gopchop-1.1.7 / src / FirstPack.cpp < prev    next >
C/C++ Source or Header  |  2005-05-01  |  5KB  |  172 lines
  1. /*
  2. #
  3. # FirstPack class create a dummy first pack length 2KB with a valid system header.
  4. # A valid system header is obtain by the first system header found in the program stream.
  5. # If no system header is found, nothing is done.
  6. #
  7. # Copyright (C) 2005 Tiziano Cappellari and Igor Baldachini
  8. # t1z1an0@tele2.it
  9. #
  10. # This program is free software; you can redistribute it and/or
  11. # modify it under the terms of the GNU General Public License
  12. # as published by the Free Software Foundation; either version 2
  13. # of the License, or (at your option) any later version.
  14. # 
  15. # This program is distributed in the hope that it will be useful,
  16. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  17. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  18. # GNU General Public License for more details.
  19. # 
  20. # You should have received a copy of the GNU General Public License
  21. # along with this program; if not, write to the Free Software
  22. # Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
  23. # http://www.gnu.org/copyleft/gpl.html
  24. #
  25. */
  26.  
  27. #ifdef HAVE_CONFIG_H
  28. # include "config.h"
  29. #endif
  30.  
  31. #include <string.h>
  32.     
  33. #include "FirstPack.h"    
  34. #include "mpeg2consts.h"
  35.  
  36. FirstPack::FirstPack()
  37. {
  38.     memset((void *)&pack_header, 0, sizeof(pack_header));    
  39.     is_pack_header = false;
  40.     memset((void *)&system_header, 0, sizeof(system_header));    
  41.     pstream_id=NULL;
  42.     num_stream_id=0;
  43.     is_system_header=false;
  44.     
  45.     index_stream_ids=0;
  46. }
  47.  
  48.  
  49. FirstPack::~FirstPack()
  50. {
  51.     if (pstream_id) delete []pstream_id;
  52. }
  53.  
  54.  
  55. void FirstPack::setSystemHeader(system_header_t *header)
  56. {
  57.     if (!is_system_header)      //first system_header found
  58.     {
  59.         system_header=*header;  //save system header
  60.         
  61.         //get the number of stream ids
  62.         uint16_t header_length = system_header.header_length[0] * 256 + system_header.header_length[1];
  63.         num_stream_id = (header_length - sizeof(system_header.bits)) / sizeof(stream_id_t);
  64.  
  65.         if (num_stream_id)      //there are stream_id to save: alloc memory (addStreamId() do the job)
  66.         {
  67.             pstream_id = new stream_id_t[num_stream_id];
  68.         }
  69.  
  70.         is_system_header = true; //no more system headers to save
  71.     }
  72. }
  73.  
  74.  
  75. void FirstPack::addStreamId(stream_id_t *stream)
  76. {
  77.     if (index_stream_ids < num_stream_id)
  78.     {
  79.         pstream_id[index_stream_ids++] = *stream;  // save stream ids
  80.     }
  81. }
  82.  
  83.  
  84. void FirstPack::setPackHeader(pack_header_t *header)
  85. {
  86.     if (!is_pack_header)        //first pack header found
  87.     {
  88.         pack_header=*header;    //save pack header
  89.  
  90.         // force SCR to 0
  91.         pack_header.bits[0] = 0x44;
  92.         pack_header.bits[1] = 0x00;
  93.         pack_header.bits[2] = 0x04;
  94.         pack_header.bits[3] = 0x00;
  95.         pack_header.bits[4] = 0x04;
  96.         pack_header.bits[5] = 0x01;
  97.         
  98.         is_pack_header = true;  //no more pack headers to save
  99.     }
  100. }
  101.  
  102.  
  103. /*
  104.  * Save Dummy First Pack with a valid system header
  105.  */
  106. off_t FirstPack::saveFirstPack(FILE *mpeg2)
  107. {
  108.     off_t written_bytes = 0;
  109.  
  110.     if (is_system_header && is_pack_header)     //enough info to do the job
  111.     {
  112.         fwrite(&pack_header,sizeof(pack_header), 1, mpeg2);     //write pack header
  113.         written_bytes+=sizeof(pack_header);
  114.         
  115.         uint16_t data_length = pack_header.bits[9] & 0x07;      //get the num of stuffing bytes and...
  116.         written_bytes += data_length;
  117.         while (data_length--)                                   //...fill it
  118.         {
  119.             fputc(0xFF, mpeg2);
  120.         }
  121.         
  122.         fwrite(&system_header, sizeof(system_header), 1, mpeg2); //write system header
  123.         written_bytes+=sizeof(system_header);
  124.                     
  125.         if (num_stream_id)  //write (if any) stream ids
  126.         {
  127.             fwrite(pstream_id, sizeof(stream_id_t) * num_stream_id, 1, mpeg2);
  128.             written_bytes+=sizeof(stream_id_t) * num_stream_id;
  129.         }
  130.  
  131.         /* Fill what leaves to 2 KB */
  132.         /* with packet(s) stream id = 0xBF and with one packet length no more than 1KB */
  133.         if ((0x800 - written_bytes) > 1024) //more than 1 KB to fill, so make 2 packets
  134.         {
  135.             /* first packet: fill what leaves to 1 KB */
  136.             data_length = (0x800 - written_bytes) - 6 - 1024;
  137.             fwrite(private_stream2_buf,sizeof(private_stream2_buf),1,mpeg2);
  138.             fputc(data_length / 256, mpeg2);
  139.             fputc(data_length % 256, mpeg2);
  140.             while (data_length--)
  141.             {
  142.                 fputc(0x00, mpeg2);
  143.             }
  144.             
  145.             /* second packet: fill the remaining 1 KB */
  146.             data_length = 1018;
  147.             fwrite(private_stream2_buf,sizeof(private_stream2_buf),1,mpeg2);
  148.             fputc(data_length / 256, mpeg2);
  149.             fputc(data_length % 256, mpeg2);
  150.             while (data_length--)
  151.             {
  152.                 fputc(0x00, mpeg2);
  153.             }
  154.         }
  155.         else //less than 1 KB, so make just 1 packet
  156.         {
  157.             data_length = (0x800 - written_bytes) - 6;
  158.             fwrite(private_stream2_buf,sizeof(private_stream2_buf),1,mpeg2);
  159.             fputc(data_length / 256, mpeg2);
  160.             fputc(data_length % 256, mpeg2);
  161.             while (data_length--)
  162.             {
  163.                 fputc(0x00, mpeg2);
  164.             }
  165.         }
  166.         
  167.         written_bytes = 0x800;
  168.         }
  169.         return written_bytes;
  170. }
  171.  
  172.