/* ------------------------------------------------------------------
 * Copyright (C) 1998-2009 PacketVideo
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
 * express or implied.
 * See the License for the specific language governing permissions
 * and limitations under the License.
 * -------------------------------------------------------------------
 */
/*
    This PVA_FF_MediaAtom Class contains all the objects that declare information
    about the media data within the stream.
*/

#ifndef __MediaAtom_H__
#define __MediaAtom_H__

#include "atom.h"
#include "mediaheaderatom.h"
#include "handleratom.h"
#include "mediainformationatom.h"
#include "esdescriptor.h"

class PVA_FF_MediaAtom : public PVA_FF_Atom
{

    public:
        // The protocol parameter is used when creating a HINT track.  It is passed
        // down to the PVA_FF_SampleDescriptionAtom so it can create the appropriate PVA_FF_HintSampleEntry.
        // If this constructor is used to create any other typ eof track atom, the protocol
        // parameter is ignored - hence the default parameter value.
        PVA_FF_MediaAtom(int32 mediaType,
                         PVA_FF_MP4_CODEC_TYPE codecType,
                         uint8 version,
                         uint32 fileAuthoringFlags,
                         uint32 protocol = 0,
                         uint8 profile = 1,
                         uint8 profileComp = 0xFF,
                         uint8 level = 0xFF); // Constructor

        virtual ~PVA_FF_MediaAtom();

        // Member get and set methods
        PVA_FF_MediaHeaderAtom *getMediaHeaderAtomPtr()
        {
            return _pmediaHeader;
        }

        // Update members based on next sample received
        void nextSample(int32 mediaType,
                        PVMP4FFComposerSampleParam *pSampleParam,
                        bool oChunkStart = false);

        void nextTextSample(int32 mediaType,
                            PVMP4FFComposerSampleParam *pSampleParam,
                            bool oChunkStart = false);

        bool reAuthorFirstSample(uint32 size,
                                 uint32 baseOffset);
        void updateLastTSEntry(uint32 ts);

        // Called to update the member atoms if they depend on the actual
        // file offsets where the media data atom (for which this track stores
        // the meta data) is stored
        void updateAtomFileOffsets(uint32 offset)
        {
            _pmediaInformation->updateAtomFileOffsets(offset);
        }

        void setTimeScale(uint32 ts)
        {
            _pmediaInformation->setTimeScale(ts);
            _pmediaHeader->setTimeScale(ts);
        }

        uint32 getTimeScale() const
        {
            return _pmediaHeader->getTimeScale();
        }

        // Stream properties
        void setTargetBitrate(uint32 avgBitRate, uint32 maxBitRate, uint32 bufferSizeDB)
        {
            _pmediaInformation->setTargetBitrate(avgBitRate, maxBitRate, bufferSizeDB);
        }

        // Rendering the PVA_FF_Atom in proper format (bitlengths, etc.) to an ostream
        virtual bool renderToFileStream(MP4_AUTHOR_FF_FILE_IO_WRAP *fp);

        // Getting and setting the Mpeg4 VOL header
        void addDecoderSpecificInfo(PVA_FF_DecoderSpecificInfo *pinfo)
        {
            _pmediaInformation->addDecoderSpecificInfo(pinfo);
        }

        // Getting and setting the Mpeg4 VOL header in timed text case
        void addTextDecoderSpecificInfo(PVA_FF_TextSampleDescInfo *pinfo)
        {
            _pmediaInformation->addTextDecoderSpecificInfo(pinfo);
        }

        void setMaxBufferSizeDB(uint32 max)
        {
            _pmediaInformation->setMaxBufferSizeDB(max);
        }

        void setVideoParams(uint32 frame_width, uint32 frame_height)
        {
            _pmediaInformation->setVideoParams(frame_width, frame_height);
        }

        void setAudioEncodeParams(PVMP4FFComposerAudioEncodeParams &audioParams)
        {
            _pmediaInformation->setAudioEncodeParams(audioParams);
        }

        void setH263ProfileLevel(uint8 profile, uint8 level)
        {
            _pmediaInformation->setH263ProfileLevel(profile, level);
        }

        void setDuration(uint32 duration)
        {
            _pmediaHeader->setDuration(duration);
        }

        void setLanguage(uint32 language)
        {
            _pmediaHeader->setLanguage(language);
        }

        void setESID(uint16 esid)
        {
            _pmediaInformation->setESID(esid);
        }

        void SetMaxSampleSize(uint32);
        void writeMaxSampleSize(MP4_AUTHOR_FF_FILE_IO_WRAP*);

        uint32 getSampleCount() const
        {
            return _pmediaInformation->getSampleCount();
        }

    private:
        virtual void recomputeSize();

        PVA_FF_MediaHeaderAtom *_pmediaHeader;
        PVA_FF_HandlerAtom *_phandler;
        PVA_FF_MediaInformationAtom *_pmediaInformation;

};



#endif

