/****************************************************************************
 * Copyright (C) 2009-2015 EPAM Systems
 *
 * This file is part of Indigo toolkit.
 *
 * This file may be distributed and/or modified under the terms of the
 * GNU General Public License version 3 as published by the Free Software
 * Foundation and appearing in the file LICENSE.GPL included in the
 * packaging of this file.
 *
 * This file is provided AS IS with NO WARRANTY OF ANY KIND, INCLUDING THE
 * WARRANTY OF DESIGN, MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE.
 ***************************************************************************/

#ifndef __indigo_deconvolution__
#define __indigo_deconvolution__

#include "indigo_internal.h"
#include "molecule/query_molecule.h"
#include "molecule/molecule.h"
#include "molecule/molecule_arom_match.h"
#include "molecule/molecule_substructure_matcher.h"
#include "base_cpp/obj_list.h"
#include "base_cpp/properties_map.h"

#ifdef _WIN32
#pragma warning(push)
#pragma warning(disable:4251)
#endif

class IndigoDeconvolutionElem;
class IndigoDecompositionMatch;

class DLLEXPORT IndigoDeconvolution : public IndigoObject {
private:
   enum {
      SHIFT_IDX = 2
   };
public:
   IndigoDeconvolution();
   virtual ~IndigoDeconvolution(){}

   void addMolecule(Molecule& mol, PropertiesMap& props, int idx);

   void setScaffold (QueryMolecule& scaffold);
   void makeRGroups (QueryMolecule& scaffold);
   void makeRGroup (IndigoDeconvolutionElem& elem, bool all_matches, bool change_scaffold);

   QueryMolecule& getDecomposedScaffold() { return _fullScaffold; }
   ObjArray<IndigoDeconvolutionElem>& getItems () {return _deconvolutionElems;}
   /*
    * Save AP as sepearate atoms
    */
   bool save_ap_bond_orders;
   /*
    * Ignore match errors
    */
   bool ignore_errors;
   /*
    * Aromatize
    */
   bool aromatize;

   int (*cbEmbedding) (const int *sub_vert_map, const int *sub_edge_map, const void* info, void* userdata);
   void *embeddingUserdata;

public:
   
   class DecompositionEnumerator {
   public:
      DecompositionEnumerator():all_matches(false), remove_rsites(false), deco(0){}
      ~DecompositionEnumerator(){}

      AutoPtr<AromaticityMatcher> am;
      AutoPtr<MoleculeSubstructureMatcher::FragmentMatchCache> fmcache;
       
      void calculateAutoMaps(Graph& sub);
      bool shouldContinue(int* map, int size);
      void addMatch(IndigoDecompositionMatch& match, Graph& sub, Graph& super);

      bool all_matches;
      bool remove_rsites;
      IndigoDeconvolution* deco;
      ObjArray<IndigoDecompositionMatch> contexts;
   private:
      DecompositionEnumerator(const DecompositionEnumerator&); //no implicit copy
      bool _foundOrder(ObjArray< Array<int> >& rsite_orders, Array<int>& swap_order);
      void _swapIndexes(IndigoDecompositionMatch&, int old_idx, int new_idx);
      void _refineAutoMaps(ObjList<Array<int> >& auto_maps, Graph& sub, Graph& super, Array<int>& scaf_map);
      void _addAllRsites(QueryMolecule&, IndigoDecompositionMatch&, RedBlackMap<int, int>&);

      static bool _cbAutoCheckAutomorphism (Graph &graph, const Array<int> &mapping, const void *context);
      ObjList< Array<int> > _autoMaps;
      ObjList< Array<int> > _scafAutoMaps;
   };

   void addCompleteRGroup(IndigoDecompositionMatch& emb_context, bool change_scaffold, Array<int>* rg_map);
   void createRgroups(IndigoDecompositionMatch& emb_context, bool change_scaffold);
   
   DECL_ERROR;
private:
   void _parseOptions(const char* options);
   
   void _addFullRGroup(IndigoDecompositionMatch& deco_match, Array<int>& auto_map, int rg_idx, int new_rg_idx);

   static int _rGroupsEmbedding(Graph &g1, Graph &g2, int *core1, int *core2, void *userdata);

   static bool _matchAtoms (Graph &g1, Graph &g2, const int *, int sub_idx, int super_idx, void* userdata);
   static bool _matchBonds (Graph &subgraph, Graph &supergraph, int sub_idx, int super_idx, void* userdata);
   static void _addBond (Graph &subgraph, Graph &supergraph, int sub_idx, int super_idx, void *userdata);
   static void _removeAtom (Graph &subgraph, int sub_idx, void *userdata);
   void _makeInvertMap(Array<int>& map, Array<int>& invmap);
   int _createRgMap(IndigoDecompositionMatch& deco_match, int aut_idx,
      RedBlackStringObjMap<Array<int> >& match_rgroups, Array<int>* rg_map_buf, bool change_scaffold);
   int _getRgScore(Array<int>& rg_map) const;


   QueryMolecule _scaffold;
   QueryMolecule _fullScaffold;
   bool _userDefinedScaffold;
   ObjArray<IndigoDeconvolutionElem> _deconvolutionElems;

};

class DLLEXPORT IndigoDeconvolutionElem : public IndigoObject
{
public:
   IndigoDeconvolutionElem (Molecule& mol);
   IndigoDeconvolutionElem (Molecule& mol, PropertiesMap& props);
   IndigoDeconvolutionElem (Molecule& mol, int* index);
   IndigoDeconvolutionElem (IndigoDeconvolutionElem& elem);

   virtual ~IndigoDeconvolutionElem ();

   virtual int getIndex () {return idx;}
   virtual indigo::PropertiesMap& getProperties() { return _properties;}
   int idx;

   Molecule mol_in;
   IndigoDeconvolution::DecompositionEnumerator deco_enum;

   indigo::PropertiesMap _properties;

};

class DLLEXPORT IndigoDecompositionMatch : public IndigoObject {
public:
   IndigoDecompositionMatch();
   Array<int> visitedAtoms;
   Array<int> scaffoldBonds;
   Array<int> scaffoldAtoms;
   Array<int> lastMapping;
   Array<int> lastInvMapping;
   ObjArray< Array<int> > attachmentOrder;
   ObjArray< Array<int> > attachmentIndex;
   ObjList< Array<int> > scafAutoMaps;

   int getRgroupNumber() const {
      return attachmentIndex.size() - 1;
   }

   void renumber(Array<int>& map, Array<int>& inv_map);
   void copy(IndigoDecompositionMatch& other);
   void removeRsitesFromMaps(Graph& query_graph);
   void copyScafAutoMaps(ObjList< Array<int> >& autoMaps);
   void completeScaffold();

   Molecule mol_out;
   Molecule rgroup_mol;
   Molecule mol_scaffold;

   IndigoDeconvolution* deco;
private:
   IndigoDecompositionMatch(const IndigoDecompositionMatch&); //no implicit copy

   bool _completeScaffold;
};

class DLLEXPORT IndigoDeconvolutionIter : public IndigoObject {
public:

   IndigoDeconvolutionIter(ObjArray<IndigoDeconvolutionElem>& items);
   virtual ~IndigoDeconvolutionIter();

   virtual IndigoObject * next ();
   virtual bool hasNext ();

protected:
   int _index;
   ObjArray<IndigoDeconvolutionElem>& _items;
};
class DLLEXPORT IndigoDecompositionMatchIter : public IndigoObject {
public:

   IndigoDecompositionMatchIter(ObjArray<IndigoDecompositionMatch>& matches);
   virtual ~IndigoDecompositionMatchIter(){}

   virtual IndigoObject * next ();
   virtual bool hasNext ();
   virtual int getIndex() {return _index;}

protected:
   int _index;
   ObjArray<IndigoDecompositionMatch>& _matches;
};



#ifdef _WIN32
#pragma warning(pop)
#endif

#endif