Fg3CmdControls.cpp

//
// Copyright (c) Singular Inversions Inc. 2012
//
// Authors:     Andrew Beatty
// Created:     Feb 16, 2012
//
 
#include "stdafx.h"
 
#include "Fg3Face.hpp"
#include "Fg3Controls.hpp"
#include "FgCommand.hpp"
 
#include "FgSerial.hpp"
#include "FgFileSystem.hpp"
#include "FgTestUtils.hpp"
 
using namespace std;
 
namespace Fg {
 
namespace {
 
void                tween(CLArgs const & args)
{
    Syntax    syn(args,
        "<in0>.fg <in1>.fg <val> <out>.fg                                 \n"
        "    <val> = interpolation value: 0 gives <in0>.fg, 1 gives <in1>.fg"
        );
 
    if (args.size() == 1)
        syn.error();
    Face3         in0(syn.next()),
                    in1(syn.next());
    float           iv = syn.nextAs<float>();
    Face3         out = in0;
    out.coord = in0.coord * (1.0f - iv) + in1.coord * iv;
    out.save(syn.next());
}
 
void                linearList(CLArgs const &)
{
    Sam3Controls         ctls(dataDir()+"main/si.ctl");
    for (uint tt=0; tt<2; ++tt) {
        PushIndent           optt(fg3ModeTypeStrs.at(tt) + ":");
        for (uint ss=0; ss<2; ++ss) {
            PushIndent           opss(fg3ModeSymmStrs.at(ss) + ":");
                size_t              numCtls = ctls.getSliderNum(FanTypeE(tt),FanSymmE(ss));
            for (size_t ii=0; ii<numCtls; ++ii)
                fgout << fgnl << toStrDigits(ii,2) << " " << ctls.getSliderName(FanTypeE(tt),FanSymmE(ss),ii);
        }
    }
}
 
void                linearEdit(CLArgs const & args)
{
    Syntax        syn(args,
        "<faceIn>.fg [<type> <symm> <index> [<value> <faceOut>.fg] ]\n"
        "    <type>    - ( shape | color )\n"
        "    <symm>    - ( symm | asym )     // Note that there are no 'asym' 'color' modes\n"
        "    <index>   - Index of desired control to view / modify:\n"
        "NOTES:\n"
        "    * If no <type> is given, all controls will be listed by index, value, name.\n"
        "    * If no <value> is given, the selected control's value and name will be output.\n"
        "    * Otherwise, the specified control will be set to <value> and stored in <faceOut>.fg"
        );
    Sam3Controls         ctls(dataDir()+"main/si.ctl");
    Face3             face(syn.next());
    if (!syn.more()) {
        for (uint tt=0; tt<2; ++tt) {
            PushIndent           optt(fg3ModeTypeStrs.at(tt) + ":");
            for (uint ss=0; ss<2; ++ss) {
                PushIndent           opss(fg3ModeSymmStrs.at(ss) + ":");
                    size_t              numCtls = ctls.getSliderNum(FanTypeE(tt),FanSymmE(ss));
                for (size_t ii=0; ii<numCtls; ++ii) {
                    fgout << fgnl << toStrDigits(ii,2) << "  "
                        << ctls.getSliderVal(face.coord,FanTypeE(tt),FanSymmE(ss),ii) << "  "
                        << ctls.getSliderName(FanTypeE(tt),FanSymmE(ss),ii);
                }
            }
        }
        return;
    }
    FanTypeE        type = FanTypeE(syn.nextSelectionIndex(fg3ModeTypeStrs,"<type>"));
    FanSymmE        symm = FanSymmE(syn.nextSelectionIndex(fg3ModeSymmStrs,"<symm>"));
    size_t          index = syn.nextAs<size_t>();
    float           valInitial = ctls.getSliderVal(face.coord,type,symm,index);
    if (!syn.more()) {
        fgout << fgnl << toStrDigits(index,2) << "  " << valInitial << "  " << ctls.getSliderName(type,symm,index);
        return;
    }
    float           value = syn.nextAs<float>();
    ctls.setSliderVal_(face.coord,type,symm,index,value);
    face.save(syn.next());
    fgout << fgnl << toStrDigits(index,2) << " " << ctls.getSliderName(type,symm,index)
        << fgnl << "Value changed from " << valInitial << " to " << ctls.getSliderVal(face.coord,type,symm,index);
}
 
void                linearExport(CLArgs const & args)
{
    Syntax            syn(args,"<face>.fg <base>\n"
        "NOTES:\n"
        "    The following text files will be created, each listing the index, value and name of the\n"
        "    respective linear controls:\n"
        "        <base>_shape_symm.txt\n"
        "        <base>_shape_asym.txt\n"
        "        <base>_color_symm.txt"
        );
    Sam3Controls        ctls(dataDir()+"main/si.ctl");
    Face3             face(syn.next());
    string              base(syn.next());
    for (uint tt=0; tt<2; ++tt) {
        string              base_type = base + "_" + fg3ModeTypeStrs.at(tt);
        for (uint ss=0; ss<2; ++ss) {
            string              base_type_symm = base_type + "_" + fg3ModeSymmStrs.at(ss) + ".txt";
            Ofstream          ofs(base_type_symm);
                size_t              numCtls = ctls.getSliderNum(FanTypeE(tt),FanSymmE(ss));
            for (size_t ii=0; ii<numCtls; ++ii) {
                ofs << toStrDigits(ii,2) << "  "
                    << ctls.getSliderVal(face.coord,FanTypeE(tt),FanSymmE(ss),ii) <<  "  "
                    << ctls.getSliderName(FanTypeE(tt),FanSymmE(ss),ii) << "\n";
            }
        }
    }
}
 
void                linear(CLArgs const & args)
{
    Cmds  cmds;
    cmds.push_back(Cmd(linearList,"list","List all linear controls by index and name"));
    cmds.push_back(Cmd(linearEdit,"edit","View or set control values from the 'Shape' and 'Color' tabs"));
    cmds.push_back(Cmd(linearExport,"export","Export linear control values of a face to text files"));
    doMenu(args,cmds);
}
 
string              getDemogVals(Sam3Controls const & ctls,Sam3Coord const & coord,FanTypeE type)
{
    ostringstream   os;
    os  << "Age " << ctls.getAge(coord,type) << "\n"
           "Gender " << ctls.getGender(coord,type) << "\n"
           "CaricatureShape " << ctls.getCaricature(coord,FANRACE_ALL,FANTYPE_GEO) << "\n"
           "CaricatureColor " << ctls.getCaricature(coord,FANRACE_ALL,FANTYPE_TEX) << "\n"
           "Asymmetry " << ctls.getAsymmetry(coord) << "\n"
           "African " << ctls.getRace(coord,FANRACE_ALL,FANRACE_AFRO,type) << "\n"
           "EastAsian " << ctls.getRace(coord,FANRACE_ALL,FANRACE_EASIA,type) << "\n"
           "SouthAsian " << ctls.getRace(coord,FANRACE_ALL,FANRACE_SASIA,type) << "\n"
           "European " << ctls.getRace(coord,FANRACE_ALL,FANRACE_EURO,type) << "\n";
    return os.str();
}
 
void                cmdDemogEdit(CLArgs const & args)
{
    Syntax        syn {args,R"([-s | -c] <in>.fg [<control> <value> <out>.fg]
    -s              - only view or edit the shape subspace of this control (age, gender & racial groups only)
    -c              - only view or edit the color subspace of this control (age, gender & racial groups only)
    <in>.fg         - face coordinate to view or edit
    <control>       - ( age | gender | caricatureshape | caricaturecolor |
                      asymmetry | african | eastasian | southasian | european)
    <value>         - The new value to set the control to:
        age (years)
        gender (-1 average male, 1 average female)
        caricatureshape (1 is average. Must be > 0)
        caricaturecolor (1 is average. Must be > 0)
        asymmetry (1 is average. Must be > 0)
        african | eastasian | southasian | european (1 is similar)
NOTES:
    * If only <in>.fg is specified, the values of all controls will be printed.)"
    };
    Sam3Controls        ctls {dataDir()+"main/si.ctl"};
    FanTypeE            type = FANTYPE_SIZE;        // view/edit combined shape & color
    if (syn.peekNext() == "-s") {
        type = FANTYPE_GEO;
        syn.next();
    }
    else if (syn.peekNext() == "-c") {
        type = FANTYPE_TEX;
        syn.next();
    }
    Face3               face {syn.next()};
    if (!syn.more()) {
        fgout << '\n' << getDemogVals(ctls,face.coord,type);
        return;
    }
    String              control = syn.next();
    float               valInitial,
                        value = syn.nextAs<float>();
    if (control == "age") {
        valInitial = ctls.getAge(face.coord,type);
        ctls.setAge_(face.coord,type,value);
    }
    else if (control == "gender") {
        valInitial = ctls.getGender(face.coord);
        ctls.setGender_(face.coord,type,value);
    }
    else if (control == "caricatureshape") {
        valInitial = ctls.getCaricature(face.coord,FANRACE_ALL,FANTYPE_GEO);
        ctls.setCaricature_(face.coord,FANRACE_ALL,FANTYPE_GEO,value);
    }
    else if (control == "caricaturecolor") {
        valInitial = ctls.getCaricature(face.coord,FANRACE_ALL,FANTYPE_TEX);
        ctls.setCaricature_(face.coord,FANRACE_ALL,FANTYPE_TEX,value);
    }
    else if (control == "asymmetry") {
        valInitial = ctls.getAsymmetry(face.coord);
        ctls.setAsymmetry_(face.coord,value);
    }
    else if (control == "african") {
        valInitial = ctls.getRace(face.coord,FANRACE_ALL,FANRACE_AFRO,type);
        ctls.setRace_(face.coord,FANRACE_ALL,FANRACE_AFRO,value,type);
    }
    else if (control == "eastasian") {
        valInitial = ctls.getRace(face.coord,FANRACE_ALL,FANRACE_EASIA,type);
        ctls.setRace_(face.coord,FANRACE_ALL,FANRACE_EASIA,value,type);
    }
    else if (control == "southasian") {
        valInitial = ctls.getRace(face.coord,FANRACE_ALL,FANRACE_SASIA,type);
        ctls.setRace_(face.coord,FANRACE_ALL,FANRACE_SASIA,value,type);
    }
    else if (control == "european") {
        valInitial = ctls.getRace(face.coord,FANRACE_ALL,FANRACE_EURO,type);
        ctls.setRace_(face.coord,FANRACE_ALL,FANRACE_EURO,value,type);
    }
    else {
        valInitial = 0.0f;      // Avoid gcc warning - below statement throws anyway.
        syn.error("Invalid value for <control>",control);
    }
    face.save(syn.next());
    fgout << fgnl << control << " value changed from " << valInitial << " to " << value;
}
 
void                cmdDemogExport(CLArgs const & args)
{
    Syntax              syn {args,"<face>.fg <out>.txt\n"
        "    Creates a <out>.txt containing the demographic control values of <face>.fg"
    };
    Sam3Controls        ctls {dataDir()+"main/si.ctl"};
    Face3               face {syn.next()};
    Ofstream            ofs {syn.next()};
    ofs << getDemogVals(ctls,face.coord,FANTYPE_SIZE);
}
 
void                cmdDemog(CLArgs const & args)
{
    Cmds            cmds {
        {cmdDemogEdit,"edit","View or set demographics control values"},
        {cmdDemogExport,"export","Export linear control values of a face to text files"},
    };
    doMenu(args,cmds);
}
 
void                genetic(CLArgs const & args)
{
    Syntax            syn(args,"<faceIn>.fg <randomness> <faceOut>.fg\n"
        "    <randomness> - Amount of perturbation in per-dimension standard deviations in the range (0,1]."
        );
    Sam3Controls        ctls(dataDir()+"main/si.ctl");
    Face3               face(syn.next());
    float               randomness = strTo<float>(syn.next());
    if ((randomness <= 0.0f) || (randomness > 1.0f))
        fgThrow("Randomness must be in the range (0,1]",syn.curr());
    face.coord = ctls.genCtrls.geneticRandom(randomness,face.coord);
    face.save(syn.next());
}
 
}
 
void                cmd3Controls(CLArgs const & args)
{
    Cmds            cmds {
        {tween,"tween","Interpolate between two face coordinates"},
        {linear,"linear","Controls from the 'Shape' and 'Color' tabs in our GUI products"},
        {cmdDemog,"demographic","Controls from the 'Demographics' tab in our GUI products"},
        {genetic,"genetic","Randomly perturb face coordinates for a genetic interface"},
    };
    doMenu(args,cmds);
}
 
}