text: // Copyright 2011-2021 Google LLC
  //
  // 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 STATISTICS_WRITER_H_
  #define STATISTICS_WRITER_H_
  
  #include <fstream>
  
  #include "third_party/absl/container/flat_hash_map.h"
  #include "third_party/zynamics/binexport/writer.h"
  
  namespace security::binexport {
  
  class StatisticsWriter : public Writer {
   public:
    explicit StatisticsWriter(std::ostream& stream);
    explicit StatisticsWriter(const std::string& filename);
  
    void GenerateStatistics(
        const CallGraph& call_graph, const FlowGraph& flow_graph,
        absl::flat_hash_map<std::string, size_t>* statistics) const;
  
    absl::Status Write(const CallGraph& call_graph, const FlowGraph& flow_graph,
                       const Instructions&, const AddressReferences&,
                       const AddressSpace&) override;
  
   private:
    std::ofstream file_;
    std::ostream& stream_;
  };
  
  }  // namespace security::binexport
  
  #endif  // STATISTICS_WRITER_H_
  