#pragma once

#ifndef TSCANNER_H
#define TSCANNER_H

#include "trasterimage.h"
#include <set>
#include <QString>
#undef DVAPI
#undef DVVAR
#ifdef TNZBASE_EXPORTS
#define DVAPI DV_EXPORT_API
#define DVVAR DV_EXPORT_VAR
#else
#define DVAPI DV_IMPORT_API
#define DVVAR DV_IMPORT_VAR
#endif

// forward declaration
class TOStream;
class TIStream;
class TFilePath;

struct DVAPI TScanParam {
  TScanParam()
      : m_supported(false)
      , m_min(0)
      , m_max(0)
      , m_def(0)
      , m_step(0)
      , m_value(0) {}

  TScanParam(float _min, float _max, float _def, float _step)
      : m_supported(true)
      , m_min(_min)
      , m_max(_max)
      , m_def(_def)
      , m_step(1)
      , m_value(_def) {}

  bool m_supported;
  float m_min, m_max, m_def, m_step,
      m_value;  // questi sono float per seguire lo standard TWAIN !

  void update(const TScanParam &model);
  // assegna min,max,def,step e supported da model a this
  // mantiene value a patto che sia contenuto nel nuovo range, diversamente ne
  // fa il crop
};

class DVAPI TScannerParameters {
public:
  enum ScanType { None, BW, GR8, RGB24 };

private:
  // Supported scan types: Black and white, graytones, color
  bool m_bw, m_gray, m_rgb;

  // Current scan type
  ScanType m_scanType;

  std::string m_paperFormat;  // e.g. "A4 paper"
  TRectD m_scanArea;  // in mm /* TWAIN preferirebbe gli inch, ma uso i mm per
                      // seguire tnz4.x*/
  TRectD m_cropBox;   // in mm /* TWAIN preferirebbe gli inch, ma uso i mm per
                      // seguire tnz4.x*/
  bool m_isPreview;
  TDimensionD m_maxPaperSize;  // in mm /* TWAIN preferirebbe gli inch, ma uso i
                               // mm per seguire tnz4.x*/

  bool m_paperOverflow;  // vale true se la scanArea e' stata tagliata rispetto
                         // alle dimensioni della carta
                         // per rispettare la maxPaperSize

  bool m_validatedByCurrentScanner;
  // vale false se bisogna ancora chiamare adaptToCurrentScanner()
  // l'idea e' di chiamare questo metodo il meno possibile

public:
  TScanParam m_brightness;
  TScanParam m_contrast;
  TScanParam m_threshold;
  TScanParam m_dpi;
  TScanParam m_paperFeeder;  // value==1.0 => use paper feeder

private:
  // other useful info ?!
  std::string m_twainVersion;
  std::string m_manufacturer;
  std::string m_prodFamily;
  std::string m_productName;
  std::string m_version;

  bool m_reverseOrder;  // if true then scan levels starting from last frame

  void cropScanArea();  // assicura che scanArea sia dentro maxPaperSize

public:
  TScannerParameters();
  ~TScannerParameters();

  void setSupportedTypes(bool bw, bool gray, bool rgb);
  bool isSupported(ScanType) const;

  void setMaxPaperSize(double maxWidth,
                       double maxHeight);  // note: possibly update m_scanArea

  std::string getPaperFormat() const { return m_paperFormat; }
  void setPaperFormat(std::string paperFormat);
  // assert(TPaperFormatManager::instance()->isValidFormat(paperFormat));
  // updates scanArea (cropping with maxPaperSize)

  void updatePaperFormat();
  // if paperFormat=="" set default paper format; recompute (to be sure!)
  // scanArea

  bool getPaperOverflow() const {
    return m_paperOverflow;
  }  // true iff paperSize > maxPaperSize

  TRectD getScanArea() const { return m_scanArea; }
  void setCropBox(const TRectD &cropBox) { m_cropBox = cropBox * m_scanArea; }
  TRectD getCropBox() const { return m_cropBox; }
  void setIsPreview(bool isPreview) { m_isPreview = isPreview; };
  bool isPreview() const { return m_isPreview; }

  bool isReverseOrder() const { return m_reverseOrder; }
  void setReverseOrder(bool reverseOrder) { m_reverseOrder = reverseOrder; }

  bool isPaperFeederEnabled() const { return m_paperFeeder.m_value == 1.0; }
  void enablePaperFeeder(bool on) { m_paperFeeder.m_value = on ? 1.0f : 0.0f; }

  void setScanType(ScanType scanType);
  ScanType getScanType() const { return m_scanType; }

  // se e' stato definito uno scanner aggiorna lo stato (enabled/disabled,
  // range, etc.)
  // dei parametri rispetto a quest'ultimo
  void adaptToCurrentScanner();
  void adaptToCurrentScannerIfNeeded() {
    if (!m_validatedByCurrentScanner) adaptToCurrentScanner();
  }

  void assign(const TScannerParameters *params);
  void saveData(TOStream &os) const;
  void loadData(TIStream &is);
};

//----------------------------------------

class TScannerListener {
public:
  virtual void onImage(const TRasterImageP &) = 0;
  virtual void onError()                      = 0;
  virtual void onNextPaper()                  = 0;
  virtual void onAutomaticallyNextPaper()     = 0;
  virtual bool isCanceled()                   = 0;
  virtual ~TScannerListener() {}
};

//----------------------------------------

class DVAPI TScanner {
  std::set<TScannerListener *> m_listeners;

protected:
  TScanParam m_brightness, m_contrast, m_threshold, m_dpi;
  int m_paperLeft;
  QString m_scannerName;

public:
  TScanner();
  virtual ~TScanner();

  static bool m_isTwain;  // brutto, brutto :(
  static TScanner *instance();

  virtual void selectDevice()      = 0;
  virtual bool isDeviceAvailable() = 0;
  virtual bool isDeviceSelected() { return false; }

  virtual void updateParameters(TScannerParameters &parameters) = 0;
  // aggiorna i parametri 'parameters' in funzione del tipo di scanner
  // selezionato
  // se possibile non modifica i valori correnti, ma cambia solo quello che non
  // e' piu' adatto
  // abilita/disabilita i parametri "opzionali".
  // n.b. i parametri opzionali che vengono disabilitati mantengono il loro
  // valore
  // (cosi' se fai save default settings dopo aver cambiato scanner non ti perdi
  // i vecchi settaggi)

  virtual void acquire(const TScannerParameters &param, int paperCount) = 0;

  void addListener(TScannerListener *);
  void removeListener(TScannerListener *);

  void notifyImageDone(const TRasterImageP &image);
  void notifyNextPaper();
  void notifyAutomaticallyNextPaper();
  void notifyError();
  bool isScanningCanceled();

  int getPaperLeftCount() const { return m_paperLeft; }
  void setPaperLeftCount(int count) { m_paperLeft = count; }
  void decrementPaperLeftCount() { --m_paperLeft; }

  QString getName() const { return m_scannerName; }
  void setName(const QString &name) { m_scannerName = name; }
};

//---------------------------------------------------------

class DVAPI TPaperFormatManager {  // singleton
public:
  class Format {
  public:
    TDimensionD m_size;
    Format() : m_size(0, 0) {}
    Format(const TDimensionD &size) : m_size(size) {}
  };

private:
  typedef std::map<std::string, Format> FormatTable;
  FormatTable m_formats;

  TPaperFormatManager();
  void readPaperFormat(const TFilePath &fp);
  void readPaperFormats();

public:
  static TPaperFormatManager *instance();

  // resitutisce la lista dei formati
  void getFormats(std::vector<std::string> &names) const;

  bool isValidFormat(std::string name) const;
  std::string getDefaultFormat() const;

  // nome formato --> dimensione
  TDimensionD getSize(std::string name) const;
};

#endif