firebird-mirror/src/include/fb_blk.h

#ifndef INCLUDE_FB_BLK
#define INCLUDE_FB_BLK

#include "../common/classes/alloc.h"

enum BlockType
{
	type_unknown = 0,

	type_vec,
	type_dbb,
	type_bcb,
	type_bdb,
	type_pre,
	type_lck,
	type_fil,
	type_pgc,
	type_rel,
	type_fmt,
	type_vcl,
	type_req,
	type_tra,
	type_nod,
	type_csb,
	type_lls,
	type_rec,
	type_rsb,
	type_bms,
	type_dfw,
	type_tfb,
	type_str,
	type_dcc,
	type_smb,
	type_blb,
	type_irb,
	type_scl,
	type_fld,
	type_ext,
	type_mfb,
	type_riv,
	type_usr,
	type_att,
	type_sym,
	type_fun,
	type_irl,
	type_acc,
	type_idl,
	type_rsc,
	type_sdw,
	type_vct,
	type_blf,
	type_arr,
	type_map,
	type_log,
	type_dls,
	type_prc,
	type_prm,
	type_sav,
	type_xcp,
	type_idb,
	type_tpc,
	type_svc,
	type_lwt,
	type_vcx,
	type_srpb,
	type_opt,
	type_prf,
	type_rse,
	type_lit,
	type_asb,
	type_ctl,

	type_PageSpace,
	type_PageManager,

	dsql_type_ctx,
	dsql_type_par,
	dsql_type_map,
	dsql_type_req,
	dsql_type_dbb,
	dsql_type_rel,
	dsql_type_fld,
	dsql_type_fil,
	dsql_type_nod,
	dsql_type_msg,
	dsql_type_lls,
	dsql_type_str,
	dsql_type_sym,
	dsql_type_err,
	dsql_type_tra,
	dsql_type_udf,
	dsql_type_var,
	dsql_type_blb,
	dsql_type_prc,
	dsql_type_intlsym,
	dsql_type_vec,
	dsql_type_imp_join,

	alice_type_frb,
	alice_type_hnk,
	alice_type_plb,
	alice_type_vec,
	alice_type_vcl,
	alice_type_tdr,
	alice_type_str,
	alice_type_lls,
};


template<BlockType BLOCK_TYPE>
class EngineType
{
public:
	EngineType() : blockType(BLOCK_TYPE) {}

	EngineType<BLOCK_TYPE>& operator= (const EngineType<BLOCK_TYPE>& from)
	{
		blockType = from.blockType;
		return *this;
	}

	BlockType getType() const
	{
		return blockType;
	}

	bool checkHandle() const
	{
		if (!this)
		{
			return false;
		}

		return (blockType == BLOCK_TYPE);
	}

private:
	BlockType blockType;
};

template<BlockType BLOCK_TYPE = type_unknown>
class pool_alloc : public EngineType<BLOCK_TYPE>
{
    public:
#ifdef DEBUG_GDS_ALLOC
        void* operator new(size_t s, MemoryPool& p, char* file, int line)
            { return p.calloc(s, file, line); }
        void* operator new[](size_t s, MemoryPool& p, char* file, int line)
            { return p.calloc(s, file, line); }
#else
        void* operator new(size_t s, MemoryPool& p )
            { return p.calloc(s); }
        void* operator new[](size_t s, MemoryPool& p)
            { return p.calloc(s); }
#endif

        void operator delete(void* mem, MemoryPool& p)
            { if (mem) p.deallocate(mem); }
        void operator delete[](void* mem, MemoryPool& p)
            { if (mem) p.deallocate(mem); }

        void operator delete(void* mem) { if (mem) MemoryPool::globalFree(mem); }
        void operator delete[](void* mem) { if (mem) MemoryPool::globalFree(mem); }

private:
    /* These operators are off-limits */
	void* operator new(size_t s) { return 0; }
    void* operator new[](size_t s) { return 0; }
};

template<typename RPT, BlockType BLOCK_TYPE = type_unknown>
class pool_alloc_rpt : public EngineType<BLOCK_TYPE>
{
    public:
		typedef RPT blk_repeat_type;
#ifdef DEBUG_GDS_ALLOC
        void* operator new(size_t s, MemoryPool& p, int rpt, char* file, int line)
            { return p.calloc(s + sizeof(RPT) * rpt, file, line); }
#else
        void* operator new(size_t s, MemoryPool& p, int rpt)
            { return p.calloc(s + sizeof(RPT) * rpt); }
#endif
        void operator delete(void* mem, MemoryPool& p, int rpt)
            { if (mem) p.deallocate(mem); }
        void operator delete(void* mem) { if (mem) MemoryPool::globalFree(mem); }

    private:
        // These operations are not supported on static repeat-base objects
        void* operator new[](size_t s, MemoryPool& p)
            { return 0; }
        void operator delete[](void* mem, MemoryPool& p)
            { }
        void operator delete[](void* mem) { }

private:
    /* These operators are off-limits */
	void* operator new(size_t s) { return 0; }
    void* operator new[](size_t s) { return 0; }
};

#endif	/* INCLUDE_FB_BLK */