// Copyright (c) 2014-2015 Dropbox, 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 PYSTON_ASMWRITING_DISASSEMBLE_H
#define PYSTON_ASMWRITING_DISASSEMBLE_H

#include <unordered_map>

#include "llvm/ADT/Twine.h"
#include "llvm/Support/raw_ostream.h"

namespace pyston {
namespace assembler {

void disassemblyInitialize();

class AssemblyLogger {
private:
    std::unordered_map<size_t, std::vector<std::string>> comments;

    void append_comments(llvm::raw_string_ostream&, size_t pos) const;

public:
    void log_comment(const llvm::Twine&, size_t offset);
    std::string finalize_log(uint8_t const* start_addr, uint8_t const* end_addr) const;
};
}
}

#endif
