DtTextureAtlasIndexParser.h

Go to the documentation of this file.

/******************************************************************************
** Copyright (c) 2023 MAK Technologies, Inc.
** All rights reserved.
******************************************************************************/


#pragma once

#include <vrvCore/vrvCore.h>
#include <vrvMath/DtVector2.h>
#include <vrvUtil/DtVirtualBaseClassCreator.h>

#include <unordered_map>
#include <mutex>
#include <string>

namespace makVrv
{
   class DtTextureAtlasIndexParser;

   typedef DtVirtualBaseClassCreator<DtTextureAtlasIndexParser> DtTextureAtlasIndexParserCreator;

   template <>
   inline const char* creatorTypeName<DtTextureAtlasIndexParser>(void) { return "DtTextureAtlasIndexParserCreator"; }

   class DT_DLL_VRVCORE DtTextureAtlasIndexParser : public DtVirtualBaseClass
   {
   public:
      struct AtlasResult
      {
         std::string annotation;
         float uOffset;
         float vOffset;
      };

      friend DtTextureAtlasIndexParserCreator;

      struct StringOrFloatAttrib{
         std::string myText;
         float myValue;
         StringOrFloatAttrib(const std::string& name) : myText(name), myValue(0) { ; }
         StringOrFloatAttrib(const std::string& name, float value) : myText(name), myValue(value) { ; }
      };
      typedef std::pair<std::string, StringOrFloatAttrib> AttrValuePair;
      
      typedef std::vector<AttrValuePair> AttrValuePairVec;

      virtual ~DtTextureAtlasIndexParser();

      virtual bool parseAtlasIfNecessary(const std::string& inputFile);

      virtual bool retrieveTextureAtlasUV(const std::string& annotation,
         const AttrValuePairVec& searchStringPairs,
         float& u, float& v);

      virtual bool retrieveTextureAtlasUV(const AttrValuePairVec & categoryValuePairs,
         std::vector<AtlasResult>& results);

      virtual bool hasSeen(const std::string& fileName);

      virtual void setCurrentFile(const std::string& fileName);

      virtual const std::string& currentFile() const;

      virtual void setDefaultUV(float u, float v);

      virtual void defaultUV(float& u, float& v);

      virtual void setDefaultAttribute(const std::string& annotation);

      virtual const std::string& defaultAttribute() const;

      virtual void setDefaultValue(const std::string& value);

      virtual const std::string& defaultValue() const;

   protected:

      DtTextureAtlasIndexParser(DtDe& de);

   protected:
      // valueUVMap["texture"] => { ("COLOUR", "Red"), .. ("U", "0.3"), ("V", "0.2") }
      typedef std::unordered_multimap<std::string, AttrValuePairVec > AnnotVecMap; 
      typedef std::unordered_multimap<std::string, AttrValuePairVec >::iterator AnnotVecMapItr;

      // valueUVMap = FileMap["$(DATA)/objects/buoys/pillarBuoy.dtxc"]
      typedef std::unordered_map<std::string, AnnotVecMap> FileMap;
      typedef std::unordered_map<std::string, AnnotVecMap>::iterator FileMapItr;

      // For categories
      typedef std::set<std::string> AttributeSet;
      // Annotation -> categories ex. ( "COLOUR", "COLPAT", ... )
      typedef std::unordered_map<std::string, AttributeSet> AnnotAttrMap;
      typedef std::unordered_map<std::string, AnnotAttrMap> AttrMap;
      
      // map[file][annotation] = { ("COLOUR", "Red"), .. ("U", "0.3"), ("V", "0.2") }
      FileMap myFileAttrValueUVMap;

      // map[file][annotation1] = { COLOUR, COLPAT, CATSPM }
      // map[file][annotation2] = { CATSPM, TOPSHP }
      AttrMap myAttributeOnlyMap;
      
      std::string myCurrentAtlasIndex;

      // If we can't find anything, provide a global default
      float myDefaultU;
      float myDefaultV;

      // If we can't find our pair, look for DEFAULT, default
      std::string myDefaultAnnotation;
      std::string myDefaultValue;

      std::mutex myDataMutex;
      DtDe& myDe;
   };
   
   inline bool operator==(const DtTextureAtlasIndexParser::StringOrFloatAttrib& lhs, const DtTextureAtlasIndexParser::StringOrFloatAttrib& rhs) {
      return lhs.myText == rhs.myText;
   }
}