#ifndef PROBABILITY_CALCULATION_CONTROLLER_H
#define PROBABILITY_CALCULATION_CONTROLLER_H

#include <escher.h>
#include "law/law.h"
#include "law_curve_view.h"
#include "image_table_view.h"
#include "calculation/calculation.h"
#include "../shared/text_field_delegate.h"

namespace Probability {

class CalculationController : public ViewController, public Shared::TextFieldDelegate {
public:
  CalculationController(Responder * parentResponder);
  View * view() override;
  const char * title() override;
  void reload();
  void setLaw(Law * law);
  void setCalculationAccordingToIndex(int index);
  bool handleEvent(Ion::Events::Event event) override;
  void didBecomeFirstResponder() override;
  void selectSubview(int subviewIndex);
  bool textFieldDidFinishEditing(TextField * textField, const char * text) override;
private:
  void updateTitle();
  Shared::TextFieldDelegateApp * textFieldDelegateApp() override;
  Calculation * m_calculation;
  class ContentView : public View {
  public:
    ContentView(Responder * parentResponder, CalculationController * calculationController, Calculation * Calculation);
    void setLaw(Law * law);
    void setCalculation(Calculation * calculation, int index);
    void layoutSubviews() override;
    void drawRect(KDContext * ctx, KDRect rect) const override;
    LawCurveView * lawCurveView();
    ImageTableView * imageTableView();
    EditableTextCell * calculationCellAtIndex(int index);
    void willDisplayEditableCellAtIndex(int index);
    constexpr static int k_maxNumberOfEditableFields = 3;
  private:
    constexpr static KDCoordinate k_textFieldWidth = 50;
    constexpr static KDCoordinate k_textWidthMargin = 5;
    constexpr static KDCoordinate k_titleHeightMargin = 5;
    int numberOfSubviews() const override;
    View * subviewAtIndex(int index) override;
    MessageTextView m_titleView;
    LawCurveView m_lawCurveView;
    ImageTableView m_imageTableView;
    MessageTextView m_text[k_maxNumberOfEditableFields];
    char m_draftTextBuffer[EditableTextCell::k_bufferLength];
    EditableTextCell m_calculationCell[k_maxNumberOfEditableFields];
    Calculation * m_calculation;
  };
  ContentView m_contentView;
  Law * m_law;
  int m_highlightedSubviewIndex;
  constexpr static int k_maxNumberOfTitleCharacters = 30;
  char m_titleBuffer[k_maxNumberOfTitleCharacters];
};

}

#endif