/*****************************************************************************/
/*                                                                           */
/*                                 dataseg.c                                 */
/*                                                                           */
/*                          Data segment structure                           */
/*                                                                           */
/*                                                                           */
/*                                                                           */
/* (C) 2001-2009, Ullrich von Bassewitz                                      */
/*                Roemerstrasse 52                                           */
/*                D-70794 Filderstadt                                        */
/* EMail:         uz@cc65.org                                                */
/*                                                                           */
/*                                                                           */
/* This software is provided 'as-is', without any expressed or implied       */
/* warranty.  In no event will the authors be held liable for any damages    */
/* arising from the use of this software.                                    */
/*                                                                           */
/* Permission is granted to anyone to use this software for any purpose,     */
/* including commercial applications, and to alter it and redistribute it    */
/* freely, subject to the following restrictions:                            */
/*                                                                           */
/* 1. The origin of this software must not be misrepresented; you must not   */
/*    claim that you wrote the original software. If you use this software   */
/*    in a product, an acknowledgment in the product documentation would be  */
/*    appreciated but is not required.                                       */
/* 2. Altered source versions must be plainly marked as such, and must not   */
/*    be misrepresented as being the original software.                      */
/* 3. This notice may not be removed or altered from any source              */
/*    distribution.                                                          */
/*                                                                           */
/*****************************************************************************/



/* common */
#include "check.h"
#include "xmalloc.h"
#include "xsprintf.h"

/* cc65 */
#include "dataseg.h"
#include "error.h"
#include "output.h"



/*****************************************************************************/
/*                                   Code                                    */
/*****************************************************************************/



DataSeg* NewDataSeg (const char* Name, SymEntry* Func)
/* Create a new data segment, initialize and return it */
{
    /* Allocate memory */
    DataSeg* S  = xmalloc (sizeof (DataSeg));

    /* Initialize the fields */
    S->SegName  = xstrdup (Name);
    S->Func     = Func;
    InitCollection (&S->Lines);

    /* Return the new struct */
    return S;
}



void DS_Append (DataSeg* Target, const DataSeg* Source)
/* Append the data from Source to Target */
{
    unsigned I;

    /* Append all lines from Source to Target */
    unsigned Count = CollCount (&Source->Lines);
    for (I = 0; I < Count; ++I) {
        CollAppend (&Target->Lines, xstrdup (CollConstAt (&Source->Lines, I)));
    }
}



void DS_AddVLine (DataSeg* S, const char* Format, va_list ap)
/* Add a line to the given data segment */
{
    /* Format the line */
    char Buf [256];
    xvsprintf (Buf, sizeof (Buf), Format, ap);

    /* Add a copy to the data segment */
    CollAppend (&S->Lines, xstrdup (Buf));
}



void DS_AddLine (DataSeg* S, const char* Format, ...)
/* Add a line to the given data segment */
{
    va_list ap;
    va_start (ap, Format);
    DS_AddVLine (S, Format, ap);
    va_end (ap);
}



void DS_Output (const DataSeg* S)
/* Output the data segment data to the output file */
{
    unsigned I;

    /* Get the number of entries in this segment */
    unsigned Count = CollCount (&S->Lines);

    /* If the segment is actually empty, bail out */
    if (Count == 0) {
        return;
    }

    /* Output the segment directive */
    WriteOutput (".segment\t\"%s\"\n\n", S->SegName);

    /* Output all entries */
    for (I = 0; I < Count; ++I) {
        WriteOutput ("%s\n", (const char*) CollConstAt (&S->Lines, I));
    }

    /* Add an additional newline after the segment output */
    WriteOutput ("\n");
}