asnio.cpp

Go to the documentation of this file.

Go to the SVN repository for this file.

/*  $Id: asnio.cpp 33815 2007-05-04 17:18:18Z kazimird $
* ===========================================================================
*
*                            PUBLIC DOMAIN NOTICE
*               National Center for Biotechnology Information
*
*  This software/database is a "United States Government Work" under the
*  terms of the United States Copyright Act.  It was written as part of
*  the author's official duties as a United States Government employee and
*  thus cannot be copyrighted.  This software/database is freely available
*  to the public for use. The National Library of Medicine and the U.S.
*  Government have not placed any restriction on its use or reproduction.
*
*  Although all reasonable efforts have been taken to ensure the accuracy
*  and reliability of the software and data, the NLM and the U.S.
*  Government do not and cannot warrant the performance or results that
*  may be obtained by using this software or data. The NLM and the U.S.
*  Government disclaim all warranties, express or implied, including
*  warranties of performance, merchantability or fitness for any particular
*  purpose.
*
*  Please cite the author in any work or product based on this material.
*
* ===========================================================================
*
* Author: Eugene Vasilchenko
*
* File Description:
*   !!! PUT YOUR DESCRIPTION HERE !!!
*/
 
#include <ncbi_pch.hpp>
#include <corelib/ncbistd.hpp>
#include <ctools/asn/asnio.hpp>
#include <algorithm>
 
 
BEGIN_NCBI_SCOPE
 
 
/////////////////////////////////////////////////////////////////////////////
//  AsnMemoryRead::
//
 
AsnMemoryRead::AsnMemoryRead(Uint2 mode, const string& str)
    : m_Source(str), m_Data(str.c_str()), m_Size(str.size()), m_mode(mode)
{
    Init();
}
 
AsnMemoryRead::AsnMemoryRead(Uint2 mode, const char* data, size_t size)
    : m_Data(data), m_Size(size), m_mode(mode)
{
    Init();
}
 
AsnMemoryRead::~AsnMemoryRead(void)
{
    AsnIoClose(m_In);
}
 
size_t AsnMemoryRead::Read(char* buffer, size_t size)
{
    size_t count = min(size, m_Size - m_Ptr);
    memcpy(buffer, m_Data + m_Ptr, count);
    m_Ptr += count;
    return count;
}
 
static Int2 LIBCALLBACK ReadAsn(Pointer data, CharPtr buffer, Uint2 size)
{
    if ( !data || !buffer )
        return -1;
 
    return Int2(static_cast<AsnMemoryRead*>(data)->Read(buffer, size));
}
 
void AsnMemoryRead::Init(void)
{
    m_Ptr = 0;
    m_In = AsnIoNew(m_mode | ASNIO_IN, 0, this,
            reinterpret_cast<IoFuncType>(ReadAsn), 0);
}
 
 
 
/////////////////////////////////////////////////////////////////////////////
//  AsnMemoryWrite::
//
 
static Int2 LIBCALLBACK WriteAsn(Pointer data, CharPtr buffer, Uint2 size)
{
    if ( !data || !buffer )
        return -1;
 
    return Int2(static_cast<AsnMemoryWrite*>(data)->Write(buffer, size));
}
 
AsnMemoryWrite::AsnMemoryWrite(Uint2 mode)
    : m_Data(new char[512]), m_Size(512), m_Ptr(0)
{
    m_Out = AsnIoNew(mode | ASNIO_OUT, 0, this,
             0, reinterpret_cast<IoFuncType>(WriteAsn));
}
 
AsnMemoryWrite::~AsnMemoryWrite(void)
{
    AsnIoClose(m_Out);
    delete[] m_Data;
}
 
size_t AsnMemoryWrite::Write(const char* buffer, size_t size)
{
    if ( m_Size - m_Ptr < size ) { // not enough space
        // new buffer
        char* data = new char[m_Size *= 2];
        if ( m_Ptr ) // copy old data
            memcpy(data, m_Data, m_Ptr);
        // delete old buffer
        delete[] m_Data;
        // set new buffer
        m_Data = data;
    }
    // append data
    memcpy(m_Data + m_Ptr, buffer, size);
    // increase size
    m_Ptr += size;
    return size;
}
 
void AsnMemoryWrite::flush(void) const
{
    AsnIoFlush(m_Out);
}
 
END_NCBI_SCOPE