cvalloc.cpp.svn-base

来自「非结构化路识别」· SVN-BASE 代码 · 共 151 行

/*M///////////////////////////////////////////////////////////////////////////////////////
//
//  IMPORTANT: READ BEFORE DOWNLOADING, COPYING, INSTALLING OR USING.
//
//  By downloading, copying, installing or using the software you agree to this license.
//  If you do not agree to this license, do not download, install,
//  copy or use the software.
//
//
//                        Intel License Agreement
//                For Open Source Computer Vision Library
//
// Copyright (C) 2000, Intel Corporation, all rights reserved.
// Third party copyrights are property of their respective owners.
//
// Redistribution and use in source and binary forms, with or without modification,
// are permitted provided that the following conditions are met:
//
//   * Redistribution's of source code must retain the above copyright notice,
//     this list of conditions and the following disclaimer.
//
//   * Redistribution's in binary form must reproduce the above copyright notice,
//     this list of conditions and the following disclaimer in the documentation
//     and/or other materials provided with the distribution.
//
//   * The name of Intel Corporation may not be used to endorse or promote products
//     derived from this software without specific prior written permission.
//
// This software is provided by the copyright holders and contributors "as is" and
// any express or implied warranties, including, but not limited to, the implied
// warranties of merchantability and fitness for a particular purpose are disclaimed.
// In no event shall the Intel Corporation or contributors be liable for any direct,
// indirect, incidental, special, exemplary, or consequential damages
// (including, but not limited to, procurement of substitute goods or services;
// loss of use, data, or profits; or business interruption) however caused
// and on any theory of liability, whether in contract, strict liability,
// or tort (including negligence or otherwise) arising in any way out of
// the use of this software, even if advised of the possibility of such damage.
//
//M*/

#include "_cv.h"

// default <malloc>
CV_IMPL void* icvDefaultAlloc( size_t size, const char*, int)
{
    char *ptr, *ptr0 = (char*)malloc(
        (size_t)(size + CV_MALLOC_ALIGN*((size >= 4096) + 1) + sizeof(char*)*2));

    if( !ptr0 )
        return 0;

    // align the pointer
    ptr = ptr0 + sizeof(char*)*2 + 1;
    ptr = (char*)icvAlignPtr(ptr, CV_MALLOC_ALIGN);
    *(char**)(ptr - sizeof(char*)) = ptr0;

    return ptr;
}


// default <free>
CV_IMPL CVStatus icvDefaultFree( void** pptr, const char*, int )
{
    /* Checking for bad pointer */
    if( pptr && *pptr )
    {
        char* ptr = (char*)*pptr;

        /* Pointer must be align by CV_MALLOC_ALIGN */
        if( ((size_t)ptr & (CV_MALLOC_ALIGN-1)) != 0 )
            return CV_BADARG_ERR;

        *pptr = 0;
        free( *(char**)(ptr - sizeof(char*)) );
    }

    return CV_OK;
}


// pointers to allocation functions, initially set to default
CvAllocFunc p_cvAlloc = icvDefaultAlloc;
CvFreeFunc p_cvFree = icvDefaultFree;


IPCVAPI_IMPL( void*, icvAllocEx, ( size_t size, const char* file, int line  ))
{
    return  p_cvAlloc( size, file, line );
}

IPCVAPI_IMPL( CVStatus, icvFreeEx, ( void** ptr, const char* file, int line ))
{
    return  p_cvFree( ptr, file, line );
}


CV_IMPL void cvSetMemoryManager( CvAllocFunc alloc_func, CvFreeFunc free_func )
{
    CV_FUNCNAME( "cvSetMemoryManager" );

    __BEGIN__;
    
    if( (alloc_func == 0) ^ (free_func == 0) )
        CV_ERROR( CV_StsNullPtr, "Either both pointers should be NULL or none of them");

    p_cvAlloc = alloc_func ? alloc_func : icvDefaultAlloc;
    p_cvFree = free_func ? free_func : icvDefaultFree;

    __END__;
}


CV_IMPL  void*  cvAlloc( size_t size )
{
    void* ptr = 0;
    
    CV_FUNCNAME( "cvAlloc" );

    __BEGIN__;

    if( (unsigned)size > CV_MAX_ALLOC_SIZE )
        CV_ERROR( CV_StsOutOfRange,
                  "Negative or too large argument of cvAlloc function" );

    ptr = icvAlloc( size );
    if( !ptr )
        CV_ERROR( CV_StsNoMem, "Out of memory" );

    __END__;

    return ptr;
}


CV_IMPL  void  cvFree( void** ptr )
{
    CV_FUNCNAME( "cvFree" );

    __BEGIN__;

    CVStatus status = icvFree( ptr );
    if( status < 0 )
        CV_ERROR( status, "" );

    __END__;
}


/* End of file. */