Leaked source code of windows server 2003
You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
 
 
 
 
 
 

366 lines
8.8 KiB

/*--------------------------------------------------------------------------*
*
* Microsoft Windows
* Copyright (C) Microsoft Corporation, 1992 - 1998.
*
* File: stgio.inl
*
* Contents: Inlines file structured storage I/O utilities
*
* History: 25-Jun-98 jeffro Created
*
*--------------------------------------------------------------------------*/
#ifndef STGIO_INL
#define STGIO_INL
#pragma once
#include <functional>
#include <algorithm>
/*+-------------------------------------------------------------------------*
* stream_insert
*
* Functor to insert an object in an IStream.
*--------------------------------------------------------------------------*/
template<class T>
struct stream_insert :
public std::binary_function<IStream*, T, IStream&>
{
IStream& operator() (IStream* pstm, const T& t) const
{ return (*pstm << t); }
};
/*+-------------------------------------------------------------------------*
* stream_extract
*
* Functor to extract an object from an IStream.
*--------------------------------------------------------------------------*/
template<class T>
struct stream_extract :
public std::binary_function<IStream*, T, IStream&>
{
IStream& operator() (IStream* pstm, T& t) const
{ return (*pstm >> t); }
};
/*+-------------------------------------------------------------------------*
* insert_each
*
* Inserts each item in a collection in an IStream.
*--------------------------------------------------------------------------*/
template<class Collection>
void insert_each (IStream* pstm, const Collection& c)
{
std::for_each (c.begin(), c.end(),
std::bind1st (stream_insert<Collection::value_type>(), pstm));
}
/*+-------------------------------------------------------------------------*
* insert_collection
*
* Inserts an entire collection into an IStream.
*--------------------------------------------------------------------------*/
template<class Collection>
void insert_collection (IStream* pstm, const Collection& c)
{
/*
* write the size
*/
*pstm << (DWORD) c.size();
/*
* write the elements
*/
insert_each (pstm, c);
}
/*+-------------------------------------------------------------------------*
* extract_collection
*
* Extracts an entire collection (written by insert_collection) from an IStream.
*--------------------------------------------------------------------------*/
template<class Collection>
void extract_collection (IStream* pstm, Collection& c)
{
/*
* clear out the current container
*/
c.clear();
ASSERT (c.empty());
/*
* read the number of items
*/
DWORD cItems;
*pstm >> cItems;
/*
* read each item
*/
while (cItems-- > 0)
{
/*
* read the item
*/
Collection::value_type t;
*pstm >> t;
/*
* put it in the container
*/
c.push_back (t);
}
}
/*+-------------------------------------------------------------------------*
* extract_vector
*
* Extracts an entire vector (written by insert_collection) from an IStream.
*--------------------------------------------------------------------------*/
template<class T>
void extract_vector (IStream* pstm, std::vector<T>& v)
{
/*
* clear out the current container
*/
v.clear();
/*
* read the number of items
*/
DWORD cItems;
*pstm >> cItems;
/*
* pre-allocate the appropriate number of items (specialization for vector)
*/
v.reserve (cItems);
ASSERT (v.empty());
ASSERT (v.capacity() >= cItems);
/*
* read each item
*/
while (cItems-- > 0)
{
/*
* read the item
*/
T t;
*pstm >> t;
/*
* put it in the container
*/
v.push_back (t);
}
}
// specialized scalar vector prototypes (stgio.cpp)
#define DeclareScalarVectorStreamFunctions(scalar_type) \
void extract_vector (IStream* pstm, std::vector<scalar_type>& v); \
void insert_collection (IStream* pstm, const std::vector<scalar_type>& v);
DeclareScalarVectorStreamFunctions (bool);
DeclareScalarVectorStreamFunctions ( char);
DeclareScalarVectorStreamFunctions (unsigned char);
DeclareScalarVectorStreamFunctions ( short);
DeclareScalarVectorStreamFunctions (unsigned short);
DeclareScalarVectorStreamFunctions ( int);
DeclareScalarVectorStreamFunctions (unsigned int);
DeclareScalarVectorStreamFunctions ( long);
DeclareScalarVectorStreamFunctions (unsigned long);
DeclareScalarVectorStreamFunctions ( __int64);
DeclareScalarVectorStreamFunctions (unsigned __int64);
DeclareScalarVectorStreamFunctions (float);
DeclareScalarVectorStreamFunctions (double);
DeclareScalarVectorStreamFunctions (long double);
/*+-------------------------------------------------------------------------*
* extract_set_or_map
*
* Extracts an entire set or map (written by insert_collection) from an IStream.
*--------------------------------------------------------------------------*/
template<class Collection>
void extract_set_or_map (IStream* pstm, Collection& c)
{
/*
* clear out the current container
*/
c.clear();
ASSERT (c.empty());
/*
* read the number of items
*/
DWORD cItems;
*pstm >> cItems;
/*
* read each item
*/
while (cItems-- > 0)
{
/*
* read the item
*/
Collection::value_type t;
*pstm >> t;
/*
* put it in the container
*/
c.insert (t);
}
}
/*+-------------------------------------------------------------------------*
* operator<<, operator>>
*
* Stream insertion and extraction operators for various types
*--------------------------------------------------------------------------*/
// std::pair<>
template<class T1, class T2>
IStream& operator>> (IStream& stm, std::pair<T1, T2>& p)
{
return (stm >> p.first >> p.second);
}
template<class T1, class T2>
IStream& operator<< (IStream& stm, const std::pair<T1, T2>& p)
{
return (stm << p.first << p.second);
}
// std::list<>
template<class T, class Al>
IStream& operator>> (IStream& stm, std::list<T, Al>& l)
{
extract_collection (&stm, l);
return (stm);
}
template<class T, class Al>
IStream& operator<< (IStream& stm, const std::list<T, Al>& l)
{
insert_collection (&stm, l);
return (stm);
}
// std::deque<>
template<class T, class Al>
IStream& operator>> (IStream& stm, std::deque<T, Al>& l)
{
extract_collection (&stm, l);
return (stm);
}
template<class T, class Al>
IStream& operator<< (IStream& stm, const std::deque<T, Al>& l)
{
insert_collection (&stm, l);
return (stm);
}
// std::vector<>
template<class T, class Al>
IStream& operator>> (IStream& stm, std::vector<T, Al>& v)
{
extract_vector (&stm, v);
return (stm);
}
template<class T, class Al>
IStream& operator<< (IStream& stm, const std::vector<T, Al>& v)
{
insert_collection (&stm, v);
return (stm);
}
// std::set<>
template<class T, class Pr, class Al>
IStream& operator>> (IStream& stm, std::set<T, Pr, Al>& s)
{
extract_set_or_map (&stm, s);
return (stm);
}
template<class T, class Pr, class Al>
IStream& operator<< (IStream& stm, const std::set<T, Pr, Al>& s)
{
insert_collection (&stm, s);
return (stm);
}
// std::multiset<>
template<class T, class Pr, class Al>
IStream& operator>> (IStream& stm, std::multiset<T, Pr, Al>& s)
{
extract_set_or_map (&stm, s);
return (stm);
}
template<class T, class Pr, class Al>
IStream& operator<< (IStream& stm, const std::multiset<T, Pr, Al>& s)
{
insert_collection (&stm, s);
return (stm);
}
// std::map<>
template<class K, class T, class Pr, class Al>
IStream& operator>> (IStream& stm, std::map<K, T, Pr, Al>& m)
{
extract_set_or_map (&stm, m);
return (stm);
}
template<class K, class T, class Pr, class Al>
IStream& operator<< (IStream& stm, const std::map<K, T, Pr, Al>& m)
{
insert_collection (&stm, m);
return (stm);
}
// std::multimap<>
template<class K, class T, class Pr, class Al>
IStream& operator>> (IStream& stm, std::multimap<K, T, Pr, Al>& m)
{
extract_set_or_map (&stm, m);
return (stm);
}
template<class K, class T, class Pr, class Al>
IStream& operator<< (IStream& stm, const std::multimap<K, T, Pr, Al>& m)
{
insert_collection (&stm, m);
return (stm);
}
#endif /* STGIO_INL */