DelimitedFile.hpp

Go to the documentation of this file.

// Copyright 2010 Complete Genomics, Inc.
//
// Licensed under the Apache License, Version 2.0 (the "License"); you
// may not use this file except in compliance with the License. You
// may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or
// implied. See the License for the specific language governing
// permissions and limitations under the License.

#ifndef CGATOOLS_UTIL_DELIMITEDFILE_HPP_
#define CGATOOLS_UTIL_DELIMITEDFILE_HPP_ 1


#include "cgatools/core.hpp"
#include "cgatools/util/DelimitedLineParser.hpp"

#include <boost/noncopyable.hpp>

namespace cgatools { namespace util {

    class DelimitedFileMetadata
    {
    public:
        int getFormatVersion() const;

        bool hasKey(const std::string& key) const;
        const std::string& get(const std::string& key) const;

        void set(const std::string& key, const std::string& value);
        void add(const std::string& key, const std::string& value);
        void removeAll(const std::string& key);

        const std::vector< std::pair<std::string, std::string> >& getMap() const
        {
            return kv_;
        }

        void initDefaults();

        void transfer(const DelimitedFileMetadata& src,
                      const std::string& keys,
                      const std::string& prefix = "");

        void sort();
    private:
        std::vector< std::pair<std::string, std::string> > kv_;
    };

    std::ostream& operator<< (std::ostream& out, const DelimitedFileMetadata& meta);

    class DelimitedFile : boost::noncopyable
    {
    public:
        typedef DelimitedFileMetadata Metadata;
        typedef DelimitedLineParser::EmptyFieldHandling EmptyFieldHandling;
        typedef DelimitedLineParser::StrictnessChecking StrictnessChecking;

        explicit DelimitedFile(
            std::istream& in,
            char delimiter = '\t',
            EmptyFieldHandling emptyHandling = DelimitedLineParser::PROCESS_EMPTY_FIELDS,
            StrictnessChecking strictnessChecking = DelimitedLineParser::RELAXED_CHECKING);

        enum FieldParserType
        {
            REQUIRED = 0,
            OPTIONAL = 1
        };
        template <class Field>
        void addField(const Field& parser, FieldParserType ft = REQUIRED)
        {
            int count = 0;
            for(size_t ii=0; ii<columnHeaders_.size(); ii++)
            {
                if (columnHeadersEqual(columnHeaders_[ii], parser.getName()))
                {
                    lp_.setField(ii, parser);
                    count++;
                }
            }
            if (0 == count && OPTIONAL != ft)
                throw Exception("missing required field: "+parser.getName());
            if (count > 1)
                throw Exception("multiple fields with same name: "+parser.getName());
        }

        void addAllFields(std::vector<std::string>& fields);

        bool next();

        const Metadata& getMetadata() const
        {
            return metadata_;
        }

        const std::vector<std::string>& getColumnHeaders() const
        {
            return columnHeaders_;
        }

        size_t getFieldOffset(const std::string& fieldName) const;

        bool hasField(const std::string& fieldName) const;

        DelimitedLineParser& getDelimitedLineParser()
        {
            return lp_;
        }

        const std::string& getLine() const
        {
            return line_;
        }

    private:
        void readHeader();
        bool columnHeadersEqual(const std::string& h1, const std::string& h2) const;

        std::istream& in_;
        Metadata metadata_;
        std::vector<std::string> columnHeaders_;
        DelimitedLineParser lp_;
        std::string line_;
        char delimiter_;
        EmptyFieldHandling emptyHandling_;
        StrictnessChecking strictnessChecking_;
        size_t lineNo_;
    };

} } // cgatools::util

#endif // CGATOOLS_UTIL_DELIMITEDFILE_HPP_

---