cgal/Timer/include/CGAL/Memory_sizer.h

134 lines
3.7 KiB
C++

// Copyright (c) 2004 INRIA Sophia-Antipolis (France).
// All rights reserved.
//
// This file is part of CGAL (www.cgal.org); you can redistribute it and/or
// modify it under the terms of the GNU Lesser General Public License as
// published by the Free Software Foundation; version 2.1 of the License.
// See the file LICENSE.LGPL distributed with CGAL.
//
// Licensees holding a valid commercial license may use this file in
// accordance with the commercial license agreement provided with the software.
//
// This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
// WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
//
// $URL$
// $Id$
//
// Author(s) : Sylvain Pion, Andreas Fabri
#ifndef CGAL_MEMORY_SIZER_H
#define CGAL_MEMORY_SIZER_H
#include <CGAL/basic.h>
// This has only been implemented for Linux and VC++ for now.
#if !defined _MSC_VER && !defined __linux__
#include <iostream>
CGAL_BEGIN_NAMESPACE
struct Memory_sizer
{
typedef std::size_t size_type;
size_type virtual_size() const { return 0; }
size_type resident_size() const { return 0; }
};
CGAL_END_NAMESPACE
#else // defined _MSC_VER || defined __linux__
#if defined _MSC_VER
# include <windows.h>
# include "psapi.h"
#elif defined __linux__
# include <cstddef>
# include <cstdio>
# include <sys/types.h>
# include <sys/stat.h>
# include <fcntl.h>
# include <unistd.h>
#endif
CGAL_BEGIN_NAMESPACE
// A class giving access to the memory currently used by the process.
// Both the virtual memory size and the resident size.
// I put it in a class instead of free functions for similarity with Timer,
// and in case we want to store some state.
struct Memory_sizer
{
typedef std::size_t size_type;
size_type virtual_size() const { return get(true); }
size_type resident_size() const { return get(false); }
private:
size_type get (bool virtual_size) const
{
#ifdef _MSC_VER
DWORD pid = GetCurrentProcessId();
size_type result;
HANDLE hProcess;
PROCESS_MEMORY_COUNTERS pmc;
hProcess = OpenProcess( PROCESS_QUERY_INFORMATION |
PROCESS_VM_READ,
FALSE, pid );
if ( GetProcessMemoryInfo( hProcess, &pmc, sizeof(pmc)) )
{
result = (virtual_size)? pmc.PagefileUsage : pmc.WorkingSetSize;
}
CloseHandle( hProcess );
return result;
#else
// Extract of "man proc" under Linux :
//
// vsize %u Virtual memory size
//
// rss %u Resident Set Size: number of pages
// the process has in real memory,
// minus 3 for administrative purposes.
// This is just the pages which count
// towards text, data, or stack space.
// This does not include pages which
// have not been demand-loaded in, or
// which are swapped out.
int pid;
char name[1024];
char state;
int ppid, pgrp, session, tty, tpgid;
unsigned flags, minflt, cminflt, majflt, cmajflt;
int utime, stime, cutime, cstime, counter, priority, timeout;
unsigned itrealvalue, starttime, vsize = 0, rss = 0;
FILE *f = fopen("/proc/self/stat", "r");
CGAL_assertion(f);
fscanf(f, "%d %s %c %d %d %d %d %d %u "
"%u %u %u %u %d %d %d "
"%d %d %d %u %u %d "
"%u %u",
&pid, name, &state, &ppid, &pgrp, &session, &tty, &tpgid, &flags,
&minflt, &cminflt, &majflt, &cmajflt, &utime, &stime, &cutime,
&cstime, &counter, &priority, &timeout, &itrealvalue, &starttime,
&vsize, &rss);
fclose(f);
return virtual_size ? vsize : rss * getpagesize();
#endif
}
};
CGAL_END_NAMESPACE
#endif
#endif