Line data    Source code

       1             : /* -*- Mode: C; tab-width: 4; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
       2             : /* This Source Code Form is subject to the terms of the Mozilla Public
       3             :  * License, v. 2.0. If a copy of the MPL was not distributed with this
       4             :  * file, You can obtain one at http://mozilla.org/MPL/2.0/. */
       5             : #ifndef nsCyrillicDetector_h__
       6             : #define nsCyrillicDetector_h__
       7             : 
       8             : #include "nsCyrillicClass.h"
       9             : 
      10             : 
      11             : 
      12             : 
      13             : // {2002F781-3960-11d3-B3C3-00805F8A6670}
      14             : #define NS_RU_PROBDETECTOR_CID \
      15             : { 0x2002f781, 0x3960, 0x11d3, { 0xb3, 0xc3, 0x0, 0x80, 0x5f, 0x8a, 0x66, 0x70 } }
      16             : 
      17             : 
      18             : // {2002F782-3960-11d3-B3C3-00805F8A6670}
      19             : #define NS_UK_PROBDETECTOR_CID \
      20             : { 0x2002f782, 0x3960, 0x11d3, { 0xb3, 0xc3, 0x0, 0x80, 0x5f, 0x8a, 0x66, 0x70 } }
      21             : 
      22             : // {2002F783-3960-11d3-B3C3-00805F8A6670}
      23             : #define NS_RU_STRING_PROBDETECTOR_CID \
      24             : { 0x2002f783, 0x3960, 0x11d3, { 0xb3, 0xc3, 0x0, 0x80, 0x5f, 0x8a, 0x66, 0x70 } }
      25             : 
      26             : // {2002F784-3960-11d3-B3C3-00805F8A6670}
      27             : #define NS_UK_STRING_PROBDETECTOR_CID \
      28             : { 0x2002f784, 0x3960, 0x11d3, { 0xb3, 0xc3, 0x0, 0x80, 0x5f, 0x8a, 0x66, 0x70 } }
      29             : 
      30             : static const uint8_t *gCyrillicCls[5] =
      31             : {
      32             :    CP1251Map,
      33             :    KOI8Map,
      34             :    ISO88595Map,
      35             :    MacCyrillicMap,
      36             :    IBM866Map
      37             : };
      38             : 
      39             : static const char * gRussian[5] = {
      40             :   "windows-1251",
      41             :   "KOI8-R",
      42             :   "ISO-8859-5",
      43             :   "x-mac-cyrillic",
      44             :   "IBM866"
      45             : };
      46             : 
      47             : static const char * gUkrainian[5] = {
      48             :   "windows-1251",
      49             :   "KOI8-U",
      50             :   "ISO-8859-5",
      51             :   "x-mac-cyrillic",
      52             :   "IBM866"
      53             : };
      54             : 
      55             : #define NUM_CYR_CHARSET 5
      56             : 
      57             : class nsCyrillicDetector
      58             : {
      59             :   public:
      60           0 :     nsCyrillicDetector(uint8_t aItems,
      61             :                       const uint8_t ** aCyrillicClass,
      62           0 :                       const char **aCharsets) {
      63           0 :       mItems = aItems;
      64           0 :       mCyrillicClass = aCyrillicClass;
      65           0 :       mCharsets = aCharsets;
      66           0 :       for(unsigned i=0;i<mItems;i++)
      67           0 :         mProb[i] = mLastCls[i] =0;
      68           0 :       mDone = false;
      69           0 :     }
      70           0 :     virtual ~nsCyrillicDetector() {}
      71             :     virtual void HandleData(const char* aBuf, uint32_t aLen);
      72             :     virtual void   DataEnd();
      73             :   protected:
      74             :     virtual void Report(const char* aCharset) = 0;
      75             :     bool    mDone;
      76             : 
      77             :   private:
      78             :     uint8_t  mItems;
      79             :     const uint8_t ** mCyrillicClass;
      80             :     const char** mCharsets;
      81             :     uint32_t mProb[NUM_CYR_CHARSET];
      82             :     uint8_t mLastCls[NUM_CYR_CHARSET];
      83             : };
      84             : 
      85             : class nsCyrXPCOMDetector :
      86             :       public nsCyrillicDetector,
      87             :       public nsICharsetDetector
      88             : {
      89             :   public:
      90             :     // nsISupports interface
      91             :     NS_DECL_ISUPPORTS
      92             :     nsCyrXPCOMDetector(uint8_t aItems,
      93             :                       const uint8_t ** aCyrillicClass,
      94             :                       const char **aCharsets);
      95             :     NS_IMETHOD Init(nsICharsetDetectionObserver* aObserver) override;
      96             :     NS_IMETHOD DoIt(const char* aBuf, uint32_t aLen, bool *oDontFeedMe) override;
      97             :     NS_IMETHOD Done() override;
      98             :   protected:
      99             :     virtual ~nsCyrXPCOMDetector();
     100             :     virtual void Report(const char* aCharset) override;
     101             :   private:
     102             :     nsCOMPtr<nsICharsetDetectionObserver> mObserver;
     103             : };
     104             : 
     105             : class nsCyrXPCOMStringDetector :
     106             :       public nsCyrillicDetector,
     107             :       public nsIStringCharsetDetector
     108             : {
     109             :   public:
     110             :     // nsISupports interface
     111             :     NS_DECL_ISUPPORTS
     112             :     nsCyrXPCOMStringDetector(uint8_t aItems,
     113             :                       const uint8_t ** aCyrillicClass,
     114             :                       const char **aCharsets);
     115             :     NS_IMETHOD DoIt(const char* aBuf, uint32_t aLen,
     116             :                      const char** oCharset, nsDetectionConfident &oConf) override;
     117             :   protected:
     118             :     virtual ~nsCyrXPCOMStringDetector();
     119             :     virtual void Report(const char* aCharset) override;
     120             :   private:
     121             :     nsCOMPtr<nsICharsetDetectionObserver> mObserver;
     122             :     const char* mResult;
     123             : };
     124             : 
     125           0 : class nsRUProbDetector : public nsCyrXPCOMDetector
     126             : {
     127             :   public:
     128           0 :     nsRUProbDetector()
     129           0 :       : nsCyrXPCOMDetector(5, gCyrillicCls, gRussian) {}
     130             : };
     131             : 
     132           0 : class nsRUStringProbDetector : public nsCyrXPCOMStringDetector
     133             : {
     134             :   public:
     135           0 :     nsRUStringProbDetector()
     136           0 :       : nsCyrXPCOMStringDetector(5, gCyrillicCls, gRussian) {}
     137             : };
     138             : 
     139           0 : class nsUKProbDetector : public nsCyrXPCOMDetector
     140             : {
     141             :   public:
     142           0 :     nsUKProbDetector()
     143           0 :       : nsCyrXPCOMDetector(5, gCyrillicCls, gUkrainian) {}
     144             : };
     145             : 
     146           0 : class nsUKStringProbDetector : public nsCyrXPCOMStringDetector
     147             : {
     148             :   public:
     149           0 :     nsUKStringProbDetector()
     150           0 :       : nsCyrXPCOMStringDetector(5, gCyrillicCls, gUkrainian) {}
     151             : };
     152             : 
     153             : #endif