FgApply.hpp

//
// Copyright (c) 2023 Singular Inversions Inc. (facegen.com)
//
// Apply a face to Daz Genesis or Metahuman DNA
//
 
#ifndef FG_APPLY_HPP
#define FG_APPLY_HPP
 
#include "Fg3dMesh.hpp"
#include "FgTransform.hpp"
#include "FgGuiApi.hpp"
#include "Fg3Face.hpp"
#include "Fg3Sam.hpp"
#include "FgImageIo.hpp"
#include "FgAnthropometry.hpp"
 
namespace Fg {
 
Arr<ScaleTrans3F,2> cEyesTransform(FaceLmPos3Fs const & lmBases,Vec3Fs const & allTargs);
 
// returns mean RGB of 'img' over pixels where 'region' is 255:
Arr3D               sampleColor(
    ImgRgba8 const &    img,
    ImgUC const &       region);        // [0,255] aspect ratio must be same as 'img'
 
// additive color shift modulated over given region:
void                shiftColor_(
    Arr3D               delta,
    ImgUC const &       region,         // [0,255] aspect ration must be same as 'img'
    ImgRgba8 &          img);           // MODIFIED
 
enum struct DazGenesisVer { G1, G2, G3, G8, G81, G9 };
 
struct      DazGenesis
{
    DazGenesisVer       ver;
    String              str;    // just the number part of the above symbols
 
    bool                operator==(DazGenesisVer v) const {return (v==ver); }
    bool                operator==(String const & s) const {return (s==str); }
};
 
Svec<DazGenesis>    cDazGenesisInfo();
std::ostream &      operator<<(std::ostream &,DazGenesisVer);
// Returns exported map names for given Genesis version other than face and eyes:
Strings             dazGenesisBodyMapNames(DazGenesisVer ver);
String              dazRelMorphDir(DazGenesisVer ver,bool female);
// returns a de-duped list of DS content dirs stored by DS in Windows registry (if exists) followed by a 
// de-duped list of DS public beta content dirs stored by DSPB in Windows registry, in their given order,
// filtering out any that do not exist:
String8s            getDazContentDirs();
// return the first daz content library containing the given relative path within it (file or directory).
// Returns empty if none do:
String8             getDazContentDirContaining(String const & relPath);
// there are only 2 possible Daz Connect dirs (DS and DSPB) but within each of these are the versioned data directories
// which contain the compressed content; this searches DS first, and from most recent to find the first instance of 'relPath':
String8             getDazConnectDirContaining(String const & relPath);
 
typedef Sfun<void(const Img2F &,Mesh &)>        EmbossMesh;
typedef Sfun<void(const Img2F &,ImgRgba8 &)>    TattooImage;
 
// Applies the given face coordinate to the given Daz Genesis mesh and stores in the given content library.
// If provided, applies the given detail texture to the color maps and also stores in the given content library.
// Returns true for success or user cancellation, false if unable to write to the given content library,
// and throws for unexpected conditions such as not being able to locate data files:
void                dazExport(
    DazGenesisVer           ver,
    bool                    female,         // true = female, false = male
    String8 const &         contentDir,     // Daz content library for writing (don't use Daz Connect library).
    Sam3Coord const &       coord,          // given face
    Bytes const &           detail,         // given detail texture
    bool                    sphericalEyes,  // Apply spherical eye correction
    String8 const &         group,          // Daz grouping path for morph
    String8 const &         name,           // Export name
    // Filenames of original color maps. First must be face, then in order of 'dazGenesisBodyMapNames'.
    // These are used in the creation of the customized maps. If not supplied, no maps will be exported:
    String8s const &        baseMapNames,
    // Optional. If non-empty, this callback will be invoked 3 times with 'true' as the argument
    // (indicating each of the 3 milestones). The callback should return immediately (blocking call)
    // and if it returns true this function will terminate without completion:
    WorkerCallback const &  callback,
    EmbossMesh const &      embossMesh={},      // Ignore
    TattooImage const &     tattooImage={});    // Ignore
 
// Genesis 9 CS is X - subject's left, Y - subject's up, Z - subect's forward, in inches, with origin at base of feet 
// on saggital plane:
void                dazExportShapeG9(
    Sam3Coord const &   coord,
    String8 const &     contentDir,         // valid DS4 content library absolute path
    // parameter group specifies path withing Head shape in DS. It can contain delimiters for sub-paths.
    // it is not used for the output file location:
    String8 const &     group,
    String8 const &     name,               // non-empty, no delimiters. Used for both control and output file name.
    IdxVec3Fs const &   deltas={});         // Ignore
 
// extract the relative (to DCL) paths of the color maps for the given parts from the material DUF as long as it uses
// the same naming as the G9 starter essentials base maps:
Strings             dazGetMapPathsG9(String8 const & materialDuf,Strings const & parts);
 
void                dazExportColorG9(
    Sam3Coord const &   coord,
    Bytes const &       detail,
    String8 const &     contentDir,     // valid DS4 content library absolute path to save in
    String8s const &    baseMapPaths,   // absolute paths of Head, Arms, Body, Legs, Eyes base map images in that order
    // parameter group specifies path withing Head shape in DS. It can contain delimiters for sub-paths.
    // it is not used for the output file location:
    String8 const &     group,         
    String8 const &     name,           // output name; non-empty, no delimiters
    bool                adjustIray,     // apply the color adjustment for the iRay renderer.
    WorkerCallback const & updateFn,    // As above. Albedo map export takes most of the time.
    TattooImage const & tattooImage={}); // Ignore
 
// each mesh has the name given to it in the DNA file (modified for improvement):
Meshes              dnaImport(Bytes const & dnaFile);
// inject the given vertex positions into the dna file overwriting the existing:
void                dnaInject_(
    Vec3Fss const &     shapes,     // must be 1-1 with the meshes and vertex counts in the file below
    Bytes &             dnaFile);   // modified
 
}
 
#endif