All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator
MachineSpecs.cpp
00001 /*********************************************************************
00002 * Software License Agreement (BSD License)
00003 *
00004 *  Copyright (c) 2010, Rice University
00005 *  All rights reserved.
00006 *
00007 *  Redistribution and use in source and binary forms, with or without
00008 *  modification, are permitted provided that the following conditions
00009 *  are met:
00010 *
00011 *   * Redistributions of source code must retain the above copyright
00012 *     notice, this list of conditions and the following disclaimer.
00013 *   * Redistributions in binary form must reproduce the above
00014 *     copyright notice, this list of conditions and the following
00015 *     disclaimer in the documentation and/or other materials provided
00016 *     with the distribution.
00017 *   * Neither the name of the Rice University nor the names of its
00018 *     contributors may be used to endorse or promote products derived
00019 *     from this software without specific prior written permission.
00020 *
00021 *  THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
00022 *  "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
00023 *  LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
00024 *  FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
00025 *  COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
00026 *  INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
00027 *  BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
00028 *  LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
00029 *  CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
00030 *  LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
00031 *  ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
00032 *  POSSIBILITY OF SUCH DAMAGE.
00033 *********************************************************************/
00034 
00035 /* Author: The Internet */
00036 
00037 #include "ompl/tools/benchmark/MachineSpecs.h"
00038 #include "ompl/util/Console.h"
00039 #include "unistd.h"
00041 
00042 #if defined _WIN32
00043 
00044 // Windows 2000 or newer
00045 #include <winsock2.h>
00046 #include <windows.h>
00047 #include <stdio.h>
00048 #include <psapi.h>
00049 
00050 ompl::machine::MemUsage_t getProcessMemoryUsageAux(void)
00051 {
00052     HANDLE hProcess;
00053     PROCESS_MEMORY_COUNTERS pmc;
00054 
00055     hProcess = OpenProcess(  PROCESS_QUERY_INFORMATION |
00056                              PROCESS_VM_READ,
00057                              false,
00058                              GetCurrentProcessId() );
00059 
00060     ompl::machine::MemUsage_t result = 0;
00061 
00062     if (NULL != hProcess)
00063     {
00064         if ( GetProcessMemoryInfo( hProcess, &pmc, sizeof(pmc)) )
00065             result = pmc.WorkingSetSize;
00066         CloseHandle( hProcess );
00067     }
00068 
00069     return result;
00070 }
00071 
00072 #else
00073 #if defined __APPLE__
00074 
00075 // Mac OS 10.2 or newer
00076 #include <mach/mach_init.h>
00077 #include <mach/task.h>
00078 #include <sys/time.h>
00079 #include <sys/resource.h>
00080 #include <stdint.h>
00081 #include <cstring>
00082 #include <unistd.h>
00083 
00084 ompl::machine::MemUsage_t getProcessMemoryUsageAux(void)
00085 {
00086 
00087     task_basic_info         info;
00088     kern_return_t           rval = 0;
00089     mach_port_t             task = mach_task_self();
00090     mach_msg_type_number_t  tcnt = TASK_BASIC_INFO_COUNT;
00091     task_info_t             tptr = (task_info_t) &info;
00092 
00093     memset(&info, 0, sizeof(info));
00094 
00095     rval = task_info(task, TASK_BASIC_INFO, tptr, &tcnt);
00096     if (!(rval == KERN_SUCCESS)) return 0;
00097     return info.resident_size;
00098 }
00099 
00100 #else
00101 #if defined _POSIX_VERSION
00102 // we need a posix compliant os that exposes /proc/self/stat
00103 
00104 #include <unistd.h>
00105 #include <ios>
00106 #include <iostream>
00107 #include <fstream>
00108 
00109 ompl::machine::MemUsage_t getProcessMemoryUsageAux(void)
00110 {
00111    using std::ios_base;
00112    using std::ifstream;
00113    using std::string;
00114 
00115    // 'file' stat seems to give the most reliable results
00116    //
00117    ifstream stat_stream("/proc/self/stat",ios_base::in);
00118 
00119    if (stat_stream.good() && !stat_stream.eof())
00120    {
00121        // dummy vars for leading entries in stat that we don't care about
00122        //
00123        string pid, comm, state, ppid, pgrp, session, tty_nr;
00124        string tpgid, flags, minflt, cminflt, majflt, cmajflt;
00125        string utime, stime, cutime, cstime, priority, nice;
00126        string O, itrealvalue, starttime;
00127 
00128 
00129        // the two fields we want
00130        //
00131        unsigned long vsize;
00132        long rss;
00133 
00134        stat_stream >> pid >> comm >> state >> ppid >> pgrp >> session >> tty_nr
00135                    >> tpgid >> flags >> minflt >> cminflt >> majflt >> cmajflt
00136                    >> utime >> stime >> cutime >> cstime >> priority >> nice
00137                    >> O >> itrealvalue >> starttime >> vsize >> rss; // don't care about the rest
00138 
00139        ompl::machine::MemUsage_t page_size = sysconf(_SC_PAGE_SIZE);
00140        return rss * page_size;
00141    }
00142    return 0;
00143 }
00144 
00145 #else
00146 // if we have no idea what to do, we return 0
00147 ompl::machine::MemUsage_t getProcessMemoryUsageAux(void)
00148 {
00149     return 0;
00150 }
00151 #endif // posix
00152 #endif // apple
00153 #endif // windows
00154 
00155 ompl::machine::MemUsage_t ompl::machine::getProcessMemoryUsage(void)
00156 {
00157     MemUsage_t result = getProcessMemoryUsageAux();
00158     if (result == 0)
00159     {
00160         static msg::Interface memMsg;
00161         memMsg.warn("Unable to get memory usage");
00162     }
00163     return result;
00164 }
00165 
00166 std::string ompl::machine::getHostname(void)
00167 {
00168     static const int BUF_SIZE = 1024;
00169     char buffer[BUF_SIZE];
00170     int len = gethostname(buffer, sizeof(buffer));
00171     if (len != 0)
00172         return std::string();
00173     else
00174     {
00175         buffer[BUF_SIZE - 1] = '\0';
00176         return std::string(buffer);
00177     }
00178 }
00179 
 All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends