bool View::CIndexArrayElem::SetColor
(unsigned int id_es_v, unsigned int id_ns_v, const Fem::Field::CFieldWorld& world,
const std::auto_ptr<CColorMap>& color_map )
{
if( itype == Fem::Field::TRI ) // TRI
{
if( !world.IsIdEA(id_ea) ) return false;
const CElemAry& ea = world.GetEA(id_ea);
if( !ea.IsSegID(id_es) ) return false;
if( ea.ElemType() != TRI ) return false;
const CElemAry::CElemSeg& es_v = ea.GetSeg(id_es_v);
// assert( es_v.Size() == 1 );
unsigned int id_na_v = es_v.GetIdNA();
assert( world.IsIdNA(id_na_v) );
const CNodeAry& na = world.GetNA(id_na_v);
assert( es_v.GetMaxNoes() < na.Size() );
assert( na.IsSegID(id_ns_v) );
const CNodeAry::CNodeSeg& ns = na.GetSeg(id_ns_v);
assert( ns.Length() == 1 );
if( this->pColor == 0 ){ this->pColor = new float [nElem*3]; }
for(unsigned int itri=0;itri<nElem;itri++){
unsigned int inode0;
es_v.GetNodes(itri,&inode0);
double val;
ns.GetValue(inode0,&val);
color_map->GetColor(pColor+itri*3,val);
}
return true;
}
else if( itype == Fem::Field::QUAD )
{
if( !world.IsIdEA(id_ea) ) return false;
const CElemAry& ea = world.GetEA(id_ea);
if( !ea.IsSegID(id_es) ) return false;
if( ea.ElemType() != QUAD ) return false;
const CElemAry::CElemSeg& es_v = ea.GetSeg(id_es_v);
assert( es_v.Length() == 1 );
unsigned int id_na_v = es_v.GetIdNA();
assert( world.IsIdNA(id_na_v) );
const CNodeAry& na = world.GetNA(id_na_v);
assert( es_v.GetMaxNoes() < na.Size() );
assert( na.IsSegID(id_ns_v) );
const CNodeAry::CNodeSeg& ns = na.GetSeg(id_ns_v);
assert( ns.Length() == 1 );
if( this->pColor == 0 ){ this->pColor = new float [nElem*3]; }
for(unsigned int iquad=0;iquad<nElem;iquad++){
unsigned int inode0;
es_v.GetNodes(iquad,&inode0);
double val;
ns.GetValue(inode0,&val);
color_map->GetColor(pColor+iquad*3,val);
}
return true;
}
else if( itype == Fem::Field::TET )
{
if( !world.IsIdEA(id_ea) ) return false;
const CElemAry& ea = world.GetEA(id_ea);
const unsigned int ntet = ea.Size();
if( !ea.IsSegID(id_es) ) return false;
if( ea.ElemType() != TET ) return false;
const CElemAry::CElemSeg& es_v = ea.GetSeg(id_es_v);
assert( es_v.Length() == 1 );
unsigned int id_na_v = es_v.GetIdNA();
assert( world.IsIdNA(id_na_v) );
const CNodeAry& na = world.GetNA(id_na_v);
assert( es_v.GetMaxNoes() < na.Size() );
assert( na.IsSegID(id_ns_v) );
const CNodeAry::CNodeSeg& ns = na.GetSeg(id_ns_v);
assert( ns.Length() == 1 );
if( this->pColor == 0 ){ this->pColor = new float [nElem*3]; }
for(unsigned int iface=0;iface<nElem;iface++){
unsigned int itet = aIndElem[iface];
assert( itet < ntet );
unsigned int inode0;
es_v.GetNodes(itet,&inode0);
double val;
ns.GetValue(inode0,&val);
color_map->GetColor(pColor+iface*3,val);
}
return true;
}
else if( itype == Fem::Field::HEX )
{
if( !world.IsIdEA(id_ea) ) return false;
const CElemAry& ea = world.GetEA(id_ea);
const unsigned int nhex = ea.Size();
if( !ea.IsSegID(id_es) ) return false;
if( ea.ElemType() != HEX ) return false;
const CElemAry::CElemSeg& es_v = ea.GetSeg(id_es_v);
assert( es_v.Length() == 1 );
unsigned int id_na_v = es_v.GetIdNA();
assert( world.IsIdNA(id_na_v) );
const CNodeAry& na = world.GetNA(id_na_v);
assert( es_v.GetMaxNoes() < na.Size() );
assert( na.IsSegID(id_ns_v) );
const CNodeAry::CNodeSeg& ns = na.GetSeg(id_ns_v);
assert( ns.Length() == 1 );
if( this->pColor == 0 ){ this->pColor = new float [nElem*3]; }
for(unsigned int iface=0;iface<nElem;iface++){
unsigned int ihex = aIndElem[iface];
assert( ihex < nhex );
unsigned int inode0;
es_v.GetNodes(ihex,&inode0);
double val;
ns.GetValue(inode0,&val);
color_map->GetColor(pColor+iface*3,val);
}
return true;
}
return true;
}
