filecollection.cpp

Go to the documentation of this file.

/*
  Zipios -- a small C++ library that provides easy access to .zip files.
 
  Copyright (C) 2000-2007  Thomas Sondergaard
  Copyright (c) 2015-2022  Made to Order Software Corp.  All Rights Reserved
 
  This library is free software; 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; either
  version 2.1 of the License, or (at your option) any later version.
 
  This library 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
  Lesser General Public License for more details.
 
  You should have received a copy of the GNU Lesser General Public
  License along with this library; if not, write to the Free Software
  Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
*/
 
#include "zipios/filecollection.hpp"
 
#include "zipios/zipiosexceptions.hpp"
 
#include <algorithm>
 
 
namespace zipios
{
 
 
 
namespace
{
 
char const * g_default_filename = "-";
 
 
class MatchName
{
public:
    explicit MatchName(std::string const & name)
        : m_name(name)
    {
    }
 
    bool operator() (FileEntry::pointer_t entry) const
    {
        return entry->getName() == m_name;
    }
 
private:
    std::string const       m_name;
};
 
 
class MatchFileName
{
public:
    explicit MatchFileName(std::string const & name)
        : m_name(name)
    {
    }
 
    bool operator() (FileEntry::pointer_t entry) const
    {
        return entry->getFileName() == m_name;
    }
 
private:
    std::string const       m_name;
};
 
 
} // no name namespace
 
 
 
FileCollection::FileCollection(std::string const & filename)
    : m_filename(filename.empty() ? g_default_filename : filename)
{
}
 
 
FileCollection::FileCollection(FileCollection const & rhs)
    : m_filename(rhs.m_filename)
    , m_valid(rhs.m_valid)
{
    m_entries.reserve(rhs.m_entries.size());
    for(auto it = rhs.m_entries.begin(); it != rhs.m_entries.end(); ++it)
    {
        m_entries.push_back((*it)->clone());
    }
}
 
 
FileCollection & FileCollection::operator = (FileCollection const & rhs)
{
    if(this != &rhs)
    {
        m_filename = rhs.m_filename;
 
        m_entries.clear();
        m_entries.reserve(rhs.m_entries.size());
        for(auto it(rhs.m_entries.begin()); it != rhs.m_entries.end(); ++it)
        {
            m_entries.push_back((*it)->clone());
        }
 
        m_valid = rhs.m_valid;
    }
 
    return *this;
}
 
 
FileCollection::~FileCollection()
{
}
 
 
void FileCollection::addEntry(FileEntry const & entry)
{
    m_entries.push_back(entry.clone());
}
 
 
void FileCollection::close()
{
    m_entries.clear();
    m_filename = g_default_filename;
    m_valid = false;
}
 
 
FileEntry::vector_t FileCollection::entries() const
{
    mustBeValid();
 
    return m_entries;
}
 
 
FileEntry::pointer_t FileCollection::getEntry(std::string const & name, MatchPath matchpath) const
{
    // make sure the entries were loaded if necessary
    entries();
 
    mustBeValid();
 
    FileEntry::vector_t::const_iterator iter;
    if(matchpath == MatchPath::MATCH)
    {
        iter = std::find_if(m_entries.begin(), m_entries.end(), MatchName(name));
    }
    else
    {
        iter = std::find_if(m_entries.begin(), m_entries.end(), MatchFileName(name));
    }
 
    return iter == m_entries.end() ? FileEntry::pointer_t() : *iter;
}
 
 
std::string FileCollection::getName() const
{
    mustBeValid();
    return m_filename;
}
 
 
size_t FileCollection::size() const
{
    // make sure the entries were loaded if necessary
    entries();
 
    mustBeValid();
    return m_entries.size();
}
 
 
bool FileCollection::isValid() const
{
    return m_valid;
}
 
 
void FileCollection::mustBeValid() const
{
    if(!m_valid)
    {
        throw InvalidStateException("Attempted to access an invalid FileCollection");
    }
}
 
 
void FileCollection::setMethod(
      std::size_t limit
    , StorageMethod small_storage_method
    , StorageMethod large_storage_method)
{
    // make sure the entries were loaded if necessary
    entries();
 
    mustBeValid();
 
    for(auto it(m_entries.begin()); it != m_entries.end(); ++it)
    {
        if((*it)->getSize() > limit)
        {
            (*it)->setMethod(large_storage_method);
        }
        else
        {
            (*it)->setMethod(small_storage_method);
        }
    }
}
 
 
void FileCollection::setLevel(
      std::size_t limit
    , FileEntry::CompressionLevel small_compression_level
    , FileEntry::CompressionLevel large_compression_level)
{
    // make sure the entries were loaded if necessary
    entries();
 
    mustBeValid();
 
    for(auto it(m_entries.begin()); it != m_entries.end(); ++it)
    {
        if((*it)->getSize() > limit)
        {
            (*it)->setLevel(large_compression_level);
        }
        else
        {
            (*it)->setLevel(small_compression_level);
        }
    }
}
 
 
std::ostream & operator << (std::ostream & os, FileCollection const & collection)
{
    os << "collection '" << collection.getName() << "' {";
    FileEntry::vector_t entries(collection.entries());
    char const *sep("");
    for(auto it = entries.begin(); it != entries.end(); ++it)
    {
        os << sep;
        sep = ", ";
        os << (*it)->getName();
    }
    os << "}";
    return os;
}
 
 
} // zipios namespace
 
// Local Variables:
// mode: cpp
// indent-tabs-mode: nil
// c-basic-offset: 4
// tab-width: 4
// End:
 
// vim: ts=4 sw=4 et