/data/gitorious.org/libyui/libyui-master/src/FSize.h

/**************************************************************************
Copyright (C) 2000 - 2010 Novell, Inc.
All Rights Reserved.

This program is free software; you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation; either version 2 of the License, or
(at your option) any later version.

This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
GNU General Public License for more details.

You should have received a copy of the GNU General Public License along
with this program; if not, write to the Free Software Foundation, Inc.,
51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.

**************************************************************************/


/*---------------------------------------------------------------------\
|                                                                      |
|                      __   __    ____ _____ ____                      |
|                      \ \ / /_ _/ ___|_   _|___ \                     |
|                       \ V / _` \___ \ | |   __) |                    |
|                        | | (_| |___) || |  / __/                     |
|                        |_|\__,_|____/ |_| |_____|                    |
|                                                                      |
|                               core system                            |
|                                                        (C) SuSE GmbH |
\----------------------------------------------------------------------/

  File:       FSize.h

  Author:     Michael Andres <ma@suse.de>
  Maintainer: Michael Andres <ma@suse.de>

  Purpose: Store and operate on (file/package/partition) sizes (long long).

/-*/
#ifndef _FSize_h_
#define _FSize_h_

#include <iosfwd>
#include <string>

//
//      CLASS NAME : FSize
//
class FSize {

  public:

    enum Unit { B = 0, K, M, G, T };

  private:

    long long _size;

  public:

    static const long long KB = 1024;
    static const long long MB = 1024 * KB;
    static const long long GB = 1024 * MB;
    static const long long TB = 1024 * GB;

    static long long factor( const Unit unit_r ) {
      switch ( unit_r ) {
      case T: return TB;
      case G: return GB;
      case M: return MB;
      case K: return KB;
      case B: break;
      }
      return 1;
    }

    static const char * unit( const Unit unit_r ) {
      switch ( unit_r ) {
      case T: return "TB";
      case G: return "GB";
      case M: return "MB";
      case K: return "kB";
      case B: break;
      }
      return "B";
    }

  public:

    FSize( const long long size_r = 0 )
      : _size( size_r )
    {}

    FSize( const long long size_r, const Unit unit_r )
      : _size( size_r * factor( unit_r ) )
    {}

    FSize( const std::string &sizeStr, const Unit unit_r = B );

    operator long long() const { return _size; }

    FSize & operator+=( const long long rhs ) { _size += rhs; return *this; }
    FSize & operator-=( const long long rhs ) { _size -= rhs; return *this; }
    FSize & operator*=( const long long rhs ) { _size *= rhs; return *this; }
    FSize & operator/=( const long long rhs ) { _size /= rhs; return *this; }

    FSize & operator++(/*prefix*/) { _size += 1; return *this; }
    FSize & operator--(/*prefix*/) { _size -= 1; return *this; }

    FSize operator++(int/*postfix*/) { return _size++; }
    FSize operator--(int/*postfix*/) { return _size--; }

    FSize & fillBlock( FSize blocksize_r = KB );

    FSize fullBlock( FSize blocksize_r = KB ) const { FSize ret( _size ); return ret.fillBlock(  blocksize_r ); }

    long long operator()( const Unit unit_r ) const { return _size / factor( unit_r ); }

    Unit bestUnit() const;

    static const unsigned bestPrec = (unsigned)-1;

    std::string form( const Unit unit_r, unsigned fw = 0, unsigned prec = bestPrec, const bool showunit = true ) const;

    std::string form( unsigned fw = 0, unsigned prec = bestPrec, const bool showunit = true ) const {
      return form( bestUnit(), fw, prec, showunit );
    }

    std::string asString() const;
};


#endif // _FSize_h_