/*
 * Copyright 2010-2012, Axel Dörfler, axeld@pinc-software.de.
 * Copyright 2002-2003, Marcus Overhagen, <Marcus@Overhagen.de>.
 * Distributed under the terms of the MIT License.
 */


#include <Buffer.h>
#include <BufferConsumer.h>
#include <BufferGroup.h>
#include <BufferProducer.h>

#include "MediaDebug.h"
#include "DataExchange.h"
#include "MediaMisc.h"


// #pragma mark - protected BBufferProducer


BBufferProducer::~BBufferProducer()
{
        CALLED();
}


// #pragma mark - public BBufferProducer


/*static*/ status_t
BBufferProducer::ClipDataToRegion(int32 format, int32 size, const void* data,
        BRegion* region)
{
        CALLED();

        if (format != B_CLIP_SHORT_RUNS)
                return B_MEDIA_BAD_CLIP_FORMAT;

        return clip_shorts_to_region((const int16*)data, size / sizeof(int16),
                region);
}


media_type
BBufferProducer::ProducerType()
{
        CALLED();
        return fProducerType;
}


// #pragma mark - protected BBufferProducer


BBufferProducer::BBufferProducer(media_type producer_type)
        :
        BMediaNode("called by BBufferProducer"),
        fProducerType(producer_type),
        fInitialLatency(0),
        fInitialFlags(0),
        fDelay(0)
{
        CALLED();

        AddNodeKind(B_BUFFER_PRODUCER);
}


status_t
BBufferProducer::VideoClippingChanged(const media_source& source,
        int16 numShorts, int16* clipData, const media_video_display_info& display,
        int32* /*_deprecated_*/)
{
        CALLED();
        // may be implemented by derived classes
        return B_ERROR;
}


status_t
BBufferProducer::GetLatency(bigtime_t* _latency)
{
        CALLED();
        // The default implementation of GetLatency() finds the maximum
        // latency of your currently-available outputs by iterating over
        // them, and returns that value in outLatency

        int32 cookie;
        bigtime_t latency;
        media_output output;
        media_node_id unused;

        *_latency = 0;
        cookie = 0;
        while (GetNextOutput(&cookie, &output) == B_OK) {
                if (output.destination == media_destination::null)
                        continue;

                if (output.node.node == fNodeID) {
                        // avoid port writes (deadlock) if loopback connection
                        if (fConsumerThis == NULL)
                                fConsumerThis = dynamic_cast<BBufferConsumer*>(this);
                        if (fConsumerThis == NULL)
                                continue;

                        latency = 0;
                        if (fConsumerThis->GetLatencyFor(output.destination, &latency,
                                        &unused) == B_OK && latency > *_latency) {
                                *_latency = latency;
                        }
                } else if (FindLatencyFor(output.destination, &latency, &unused)
                                == B_OK &&  latency > *_latency) {
                        *_latency = latency;
                }
        }
        printf("BBufferProducer::GetLatency: node %" B_PRId32 ", name \"%s\" has "
                "max latency %" B_PRId64 "\n", fNodeID, fName, *_latency);
        return B_OK;
}


status_t
BBufferProducer::SetPlayRate(int32 numer, int32 denom)
{
        CALLED();
        // may be implemented by derived classes
        return B_ERROR;
}


status_t
BBufferProducer::HandleMessage(int32 message, const void* data, size_t size)
{
        PRINT(4, "BBufferProducer::HandleMessage %#lx, node %ld\n", message,
                fNodeID);

        switch (message) {
                case PRODUCER_SET_RUN_MODE_DELAY:
                {
                        const producer_set_run_mode_delay_command* command
                                = static_cast<const producer_set_run_mode_delay_command*>(data);
                        // when changing this, also change NODE_SET_RUN_MODE
                        fDelay = command->delay;
                        fRunMode = command->mode;

                        TRACE("PRODUCER_SET_RUN_MODE_DELAY: fDelay now %" B_PRId64 "\n",
                                fDelay);

                        SetRunMode(fRunMode);
                        return B_OK;
                }

                case PRODUCER_FORMAT_SUGGESTION_REQUESTED:
                {
                        const producer_format_suggestion_requested_request* request
                                = static_cast<
                                        const producer_format_suggestion_requested_request*>(data);
                        producer_format_suggestion_requested_reply reply;
                        status_t status = FormatSuggestionRequested(request->type,
                                request->quality, &reply.format);
                        request->SendReply(status, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_FORMAT_PROPOSAL:
                {
                        const producer_format_proposal_request* request
                                = static_cast<const producer_format_proposal_request*>(data);
                        producer_format_proposal_reply reply;
                        reply.format = request->format;
                        status_t status = FormatProposal(request->output, &reply.format);
                        request->SendReply(status, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_PREPARE_TO_CONNECT:
                {
                        const producer_prepare_to_connect_request* request
                                = static_cast<const producer_prepare_to_connect_request*>(data);
                        producer_prepare_to_connect_reply reply;
                        reply.format = request->format;
                        reply.out_source = request->source;
                        memcpy(reply.name, request->name, B_MEDIA_NAME_LENGTH);
                        status_t status = PrepareToConnect(request->source,
                                request->destination, &reply.format, &reply.out_source,
                                reply.name);
                        request->SendReply(status, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_CONNECT:
                {
                        const producer_connect_request* request
                                = static_cast<const producer_connect_request*>(data);
                        producer_connect_reply reply;
                        memcpy(reply.name, request->name, B_MEDIA_NAME_LENGTH);
                        Connect(request->error, request->source, request->destination,
                                request->format, reply.name);
                        request->SendReply(B_OK, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_DISCONNECT:
                {
                        const producer_disconnect_request* request
                                = static_cast<const producer_disconnect_request*>(data);
                        producer_disconnect_reply reply;
                        Disconnect(request->source, request->destination);
                        request->SendReply(B_OK, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_GET_INITIAL_LATENCY:
                {
                        const producer_get_initial_latency_request* request
                                = static_cast<
                                        const producer_get_initial_latency_request*>(data);
                        producer_get_initial_latency_reply reply;
                        reply.initial_latency = fInitialLatency;
                        reply.flags = fInitialFlags;
                        request->SendReply(B_OK, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_SET_PLAY_RATE:
                {
                        const producer_set_play_rate_request* request
                                = static_cast<const producer_set_play_rate_request*>(data);
                        producer_set_play_rate_reply reply;
                        status_t status = SetPlayRate(request->numer, request->denom);
                        request->SendReply(status, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_GET_LATENCY:
                {
                        const producer_get_latency_request* request
                                = static_cast<const producer_get_latency_request*>(data);
                        producer_get_latency_reply reply;
                        status_t status = GetLatency(&reply.latency);
                        request->SendReply(status, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_GET_NEXT_OUTPUT:
                {
                        const producer_get_next_output_request* request
                                = static_cast<const producer_get_next_output_request*>(data);
                        producer_get_next_output_reply reply;
                        reply.cookie = request->cookie;
                        status_t status = GetNextOutput(&reply.cookie, &reply.output);
                        request->SendReply(status, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_DISPOSE_OUTPUT_COOKIE:
                {
                        const producer_dispose_output_cookie_request*request
                                = static_cast<
                                        const producer_dispose_output_cookie_request*>(data);
                        producer_dispose_output_cookie_reply reply;
                        DisposeOutputCookie(request->cookie);
                        request->SendReply(B_OK, &reply, sizeof(reply));
                        return B_OK;
                }

                case PRODUCER_SET_BUFFER_GROUP:
                {
                        const producer_set_buffer_group_command* command
                                = static_cast<const producer_set_buffer_group_command*>(data);
                        node_request_completed_command replycommand;
                        BBufferGroup *group;
                        group = command->buffer_count != 0
                                ? new BBufferGroup(command->buffer_count, command->buffers)
                                : NULL;

                        if (group != NULL && group->InitCheck() != B_OK) {
                                ERROR("BBufferProducer::HandleMessage PRODUCER_SET_BUFFER_GROUP"
                                        " group InitCheck() failed.\n");
                                delete group;
                                group = NULL;
                        }
                        status_t status = SetBufferGroup(command->source, group);
                        if (command->destination == media_destination::null)
                                return B_OK;
                        replycommand.info.what
                                = media_request_info::B_SET_OUTPUT_BUFFERS_FOR;
                        replycommand.info.change_tag = command->change_tag;
                        replycommand.info.status = status;
                        replycommand.info.cookie = group;
                        replycommand.info.user_data = command->user_data;
                        replycommand.info.source = command->source;
                        replycommand.info.destination = command->destination;
                        SendToPort(command->destination.port, NODE_REQUEST_COMPLETED,
                                &replycommand, sizeof(replycommand));
                        return B_OK;
                }

                case PRODUCER_FORMAT_CHANGE_REQUESTED:
                {
                        const producer_format_change_requested_command* command
                                = static_cast<
                                        const producer_format_change_requested_command*>(data);
                        node_request_completed_command replycommand;
                        replycommand.info.format = command->format;
                        status_t status = FormatChangeRequested(command->source,
                                command->destination, &replycommand.info.format, NULL);
                        if (command->destination == media_destination::null)
                                return B_OK;
                        replycommand.info.what
                                = media_request_info::B_REQUEST_FORMAT_CHANGE;
                        replycommand.info.change_tag = command->change_tag;
                        replycommand.info.status = status;
                        //replycommand.info.cookie
                        replycommand.info.user_data = command->user_data;
                        replycommand.info.source = command->source;
                        replycommand.info.destination = command->destination;
                        SendToPort(command->destination.port, NODE_REQUEST_COMPLETED,
                                &replycommand, sizeof(replycommand));
                        return B_OK;
                }

                case PRODUCER_VIDEO_CLIPPING_CHANGED:
                {
                        const producer_video_clipping_changed_command* command
                                = static_cast<
                                        const producer_video_clipping_changed_command*>(data);
                        node_request_completed_command replycommand;
                        status_t status = VideoClippingChanged(command->source,
                                command->short_count, (int16 *)command->shorts,
                                command->display, NULL);
                        if (command->destination == media_destination::null)
                                return B_OK;
                        replycommand.info.what
                                = media_request_info::B_SET_VIDEO_CLIPPING_FOR;
                        replycommand.info.change_tag = command->change_tag;
                        replycommand.info.status = status;
                        //replycommand.info.cookie
                        replycommand.info.user_data = command->user_data;
                        replycommand.info.source = command->source;
                        replycommand.info.destination = command->destination;
                        replycommand.info.format.type = B_MEDIA_RAW_VIDEO;
                        replycommand.info.format.u.raw_video.display = command->display;
                        SendToPort(command->destination.port, NODE_REQUEST_COMPLETED,
                                &replycommand, sizeof(replycommand));
                        return B_OK;
                }

                case PRODUCER_ADDITIONAL_BUFFER_REQUESTED:
                {
                        const producer_additional_buffer_requested_command* command
                                = static_cast<
                                        const producer_additional_buffer_requested_command*>(data);
                        AdditionalBufferRequested(command->source, command->prev_buffer,
                                command->prev_time, command->has_seek_tag
                                        ? &command->prev_tag : NULL);
                        return B_OK;
                }

                case PRODUCER_LATENCY_CHANGED:
                {
                        const producer_latency_changed_command* command
                                = static_cast<const producer_latency_changed_command*>(data);
                        LatencyChanged(command->source, command->destination,
                                command->latency, command->flags);
                        return B_OK;
                }

                case PRODUCER_LATE_NOTICE_RECEIVED:
                {
                        const producer_late_notice_received_command* command
                                = static_cast<
                                        const producer_late_notice_received_command*>(data);
                        LateNoticeReceived(command->source, command->how_much,
                                command->performance_time);
                        return B_OK;
                }

                case PRODUCER_ENABLE_OUTPUT:
                {
                        const producer_enable_output_command* command
                                = static_cast<const producer_enable_output_command*>(data);
                        node_request_completed_command replycommand;
                        EnableOutput(command->source, command->enabled, NULL);
                        if (command->destination == media_destination::null)
                                return B_OK;

                        replycommand.info.what = media_request_info::B_SET_OUTPUT_ENABLED;
                        replycommand.info.change_tag = command->change_tag;
                        replycommand.info.status = B_OK;
                        //replycommand.info.cookie
                        replycommand.info.user_data = command->user_data;
                        replycommand.info.source = command->source;
                        replycommand.info.destination = command->destination;
                        //replycommand.info.format
                        SendToPort(command->destination.port, NODE_REQUEST_COMPLETED,
                                &replycommand, sizeof(replycommand));
                        return B_OK;
                }
        }

        return B_ERROR;
}


void
BBufferProducer::AdditionalBufferRequested(const media_source& source,
        media_buffer_id previousBuffer, bigtime_t previousTime,
        const media_seek_tag* previousTag)
{
        CALLED();
        // may be implemented by derived classes
}


void
BBufferProducer::LatencyChanged(const media_source& source,
        const media_destination& destination, bigtime_t newLatency, uint32 flags)
{
        CALLED();
        // may be implemented by derived classes
}


status_t
BBufferProducer::SendBuffer(BBuffer* buffer, const media_source& source,
        const media_destination& destination)
{
        CALLED();
        if (destination == media_destination::null)
                return B_MEDIA_BAD_DESTINATION;
        if (source == media_source::null)
                return B_MEDIA_BAD_SOURCE;
        if (buffer == NULL)
                return B_BAD_VALUE;

        consumer_buffer_received_command command;
        command.buffer = buffer->ID();
        command.header = *buffer->Header();
        command.header.buffer = command.buffer;
        command.header.source_port = source.port;
        command.header.source = source.id;
        command.header.destination = destination.id;
        command.header.owner = 0; // XXX fill with "buffer owner info area"
        command.header.start_time += fDelay;
                // time compensation as set by BMediaRoster::SetProducerRunModeDelay()

        //printf("BBufferProducer::SendBuffer     node %2ld, buffer %2ld, start_time %12Ld with lateness %6Ld\n", ID(), buffer->Header()->buffer, command.header.start_time, TimeSource()->Now() - command.header.start_time);

        return SendToPort(destination.port, CONSUMER_BUFFER_RECEIVED, &command,
                sizeof(command));
}


status_t
BBufferProducer::SendDataStatus(int32 status,
        const media_destination& destination, bigtime_t atTime)
{
        CALLED();
        if (IS_INVALID_DESTINATION(destination))
                return B_MEDIA_BAD_DESTINATION;

        consumer_producer_data_status_command command;
        command.for_whom = destination;
        command.status = status;
        command.at_performance_time = atTime;

        return SendToPort(destination.port, CONSUMER_PRODUCER_DATA_STATUS, &command,
                sizeof(command));
}


status_t
BBufferProducer::ProposeFormatChange(media_format* format,
        const media_destination& destination)
{
        CALLED();
        if (IS_INVALID_DESTINATION(destination))
                return B_MEDIA_BAD_DESTINATION;

        consumer_accept_format_request request;
        consumer_accept_format_reply reply;

        request.dest = destination;
        request.format = *format;
        status_t status = QueryPort(destination.port, CONSUMER_ACCEPT_FORMAT,
                &request, sizeof(request), &reply, sizeof(reply));
        if (status != B_OK)
                return status;

        *format = reply.format;
        return B_OK;
}


status_t
BBufferProducer::ChangeFormat(const media_source& source,
        const media_destination& destination, media_format* format)
{
        CALLED();
        if (IS_INVALID_SOURCE(source))
                return B_MEDIA_BAD_SOURCE;
        if (IS_INVALID_DESTINATION(destination))
                return B_MEDIA_BAD_DESTINATION;

        consumer_format_changed_request request;
        consumer_format_changed_reply reply;

        request.producer = source;
        request.consumer = destination;
        request.format = *format;

        // we use a request/reply to make this synchronous
        return QueryPort(destination.port, CONSUMER_FORMAT_CHANGED, &request,
                sizeof(request), &reply, sizeof(reply));
}


status_t
BBufferProducer::FindLatencyFor(const media_destination& destination,
        bigtime_t* _latency, media_node_id* _timesource)
{
        CALLED();
        if (IS_INVALID_DESTINATION(destination))
                return B_MEDIA_BAD_DESTINATION;

        consumer_get_latency_for_request request;
        consumer_get_latency_for_reply reply;

        request.for_whom = destination;

        status_t status = QueryPort(destination.port, CONSUMER_GET_LATENCY_FOR,
                &request, sizeof(request), &reply, sizeof(reply));
        if (status != B_OK)
                return status;

        *_latency = reply.latency;
        *_timesource = reply.timesource;
        return B_OK;
}


status_t
BBufferProducer::FindSeekTag(const media_destination& destination,
        bigtime_t targetTime, media_seek_tag* _tag, bigtime_t* _tagged_time,
        uint32* _flags, uint32 flags)
{
        CALLED();
        if (IS_INVALID_DESTINATION(destination))
                return B_MEDIA_BAD_DESTINATION;

        consumer_seek_tag_requested_request request;
        consumer_seek_tag_requested_reply reply;

        request.destination = destination;
        request.target_time = targetTime;
        request.flags = flags;

        status_t status = QueryPort(destination.port, CONSUMER_SEEK_TAG_REQUESTED,
                &request, sizeof(request), &reply, sizeof(reply));
        if (status != B_OK)
                return status;

        *_tag = reply.seek_tag;
        *_tagged_time = reply.tagged_time;
        *_flags = reply.flags;
        return B_OK;
}


void
BBufferProducer::SetInitialLatency(bigtime_t initialLatency, uint32 flags)
{
        fInitialLatency = initialLatency;
        fInitialFlags = flags;
}


// #pragma mark - private BBufferProducer


/*
private unimplemented
BBufferProducer::BBufferProducer()
BBufferProducer::BBufferProducer(const BBufferProducer &clone)
BBufferProducer & BBufferProducer::operator=(const BBufferProducer &clone)
*/

status_t BBufferProducer::_Reserved_BufferProducer_0(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_1(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_2(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_3(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_4(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_5(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_6(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_7(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_8(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_9(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_10(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_11(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_12(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_13(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_14(void*) { return B_ERROR; }
status_t BBufferProducer::_Reserved_BufferProducer_15(void*) { return B_ERROR; }


//! Deprecated.
status_t
BBufferProducer::SendBuffer(BBuffer* buffer,
        const media_destination& destination)
{
        CALLED();

        // Try to find the source - this is the best we can do
        media_output output;
        int32 cookie = 0;
        status_t status = GetNextOutput(&cookie, &output);
        if (status != B_OK)
                return status;

        return SendBuffer(buffer, output.source, destination);
}


status_t
BBufferProducer::clip_shorts_to_region(const int16* data, int count,
        BRegion* output)
{
        UNIMPLEMENTED();
        return B_ERROR;
}


status_t
BBufferProducer::clip_region_to_shorts(const BRegion* input, int16* data,
        int maxCount, int* _count)
{
        UNIMPLEMENTED();
        return B_ERROR;
}