TableWriterBase.h

//* This file is a part of PRISM: Plasma Reaction Input SysteM,
//* A library for parcing chemical reaction networks for plasma chemistry
//* https://github.com/NCSU-ComPS-Group/prism
//*
//* Licensed under MIT, please see LICENSE for details
//* https://opensource.org/license/mit
//*
//* Copyright 2024, North Carolina State University
//* ALL RIGHTS RESERVED
//*
#pragma once
 
#include <sstream>
#include <string>
#include <vector>
#include <map>
#include <memory>
namespace prism
{
class Reaction;
class TableWriterBase
{
public:
  TableWriterBase(unsigned int max_rows = 32)
    : _max_rows_per_table(max_rows), _rxn_count(0), _note_count(0)
  {
  }

  void clear()
  {
    _rxn_count = 0;
    _note_count = 0;
    _table_str.str("");
    _table_str.clear();
  }

  unsigned int maxRows() { return _max_rows_per_table; }
  std::ostringstream & tableString() { return _table_str; }
  virtual void beginDocument(const std::vector<std::string> & bib_files) = 0;
  virtual void endDocument() = 0;
  virtual void beginRateBasedSection() = 0;
  virtual void beginXSecBasedSection() = 0;
  virtual void beginFunctionalTable() = 0;
  virtual void endFunctionalTable() = 0;
  virtual void beginTabulatedTable() = 0;
  virtual void endTabulatedTable() = 0;
  virtual void addFunctionalReaction(const std::shared_ptr<const Reaction> & r) = 0;
  virtual void addTabulatedReaction(const std::shared_ptr<const Reaction> & r) = 0;
  virtual void addNotes(const std::vector<std::string> & notes) = 0;
 
protected:
  const unsigned int _max_rows_per_table;
  unsigned int _rxn_count;
  unsigned int _note_count;
  std::ostringstream _table_str;
  std::map<std::string, unsigned int> _note_numbers;
  std::map<unsigned int, std::string> _inverse_note_numbers;
};
}